Приведение указателя C к и от Go - PullRequest
Новая
       15

Приведение указателя C к и от Go

0 голосов
/ 26 апреля 2018

Я пишу приложение для платформы Windows, используя FFmpeg, и это gavang wrapper goav, но у меня возникают проблемы с пониманием того, как передавать указатели C между C и Go.

Я вычеркнулвсе соответствующие части кода C, оболочки и моего кода, показанные ниже:

Код C - libavutil / frame.h 

#include <stdint.h>

typedef struct AVFrame {
#define AV_NUM_DATA_POINTERS 8
    uint8_t *data[AV_NUM_DATA_POINTERS];
}

Go goav оболочка 

package avutil

/*
    #cgo pkg-config: libavutil
    #include <libavutil/frame.h>
    #include <stdlib.h>

    // C code I added:
    #include <stdio.h>

    void SaveFrame(const char* location, uint8_t *data, int width, int height) {
        FILE *pFile;
        int  y;

        // Open file
        pFile=fopen(location, "wb");
        if(pFile==NULL)
            return;

        // Write header
        fprintf(pFile, "P6\n%d %d\n255\n", width, height);

        // Write pixel data
        for(y=0; y<height; y++)
            fwrite(data+y*width, 1, width*3, pFile);

        // Close file
        fclose(pFile);
    }
*/
import "C"
import (
    "unsafe"
)

type Frame C.struct_AVFrame

func Data(f *Frame) *uint8 {
    // i think this is the function thats not working?
    return (*uint8)(unsafe.Pointer((*C.uint8_t)(unsafe.Pointer(&f.data))))
}
func SaveFrame(location string, data *uint8, width int, height int) {
    C.SaveFrame(C.CString(location), unsafe.Pointer(data), C.int(width), C.int(height))
}

My Go code 

package main

import "github.com/giorgisio/goav/avutil"

func main() {
    var frame *avutil.Frame
    var data *uint8

    //... initialize frame

    data = avutil.Data(frame)
    avutil.SaveFrame("frame0.ppm", data, 1920, 1080)
}

Когда я пытаюсь сохранить кадр, полученное изображение искажается из-за неправильного указателя, как мне это исправить?

1 Ответ

0 голосов
/ 26 апреля 2018

Поле data структуры AVFrame представляет собой массив из 8 указателей.

То есть это непрерывный кусок памяти, содержащий 8 слотов, смежных друг с другом, каждый из которых содержит указатель (назначение типа uint8_t).

IOW, объявление 

uint8_t *data[AV_NUM_DATA_POINTERS];

лучше интерпретировать как 

uint8_t* data[AV_NUM_DATA_POINTERS];

, а в Go вы объявите егочто-то вроде 

var data [AV_NUM_DATA_POINTERS]*uint8_t

Теперь ваша функция SaveFrame объявляет свои data аргументы типа uint8_t*.Можно передать адрес массива data структуры AVFrame в качестве этого аргумента, но более идиоматичным будет передать адрес его нулевого элемента - что-то вроде 

func Data(frame *Frame) *uint8 {
    return (*uint8)(unsafe.Pointer(&frame.data[0]))
}
...