与 Go 之间的 C 指针转换

Question

我正在使用 FFmpeg 为 Windows 平台编写一个应用程序，它是 golang 包装器 goav，但我无法理解如何在 C 和 Go 之间传递 C 指针。

我已经删除了 C 代码的所有相关部分，包装器和我的代码，如下所示:

C 代码 - libavutil/frame.h

#include <stdint.h>

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

去 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))
}

我的 Go 代码

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)
}

当我尝试保存帧时，由于指针错误导致生成的图像出现乱码，我该如何解决？

Answer 1

AVFrame 结构的 data 字段是一个包含 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 array of an AVFrame struct 作为该参数，但更惯用将传递其第零个元素的地址——类似

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