通过ctypes将C数组传递给python

Question

我有一个控制相机的C函数，我正在尝试通过要在GUI中使用的Ctypes将图像缓冲区发送到Python。

我可以抓住缓冲区，但是我基本上仍在坚持如何使它起作用，我想我可以研究一下numpy-c api，但看起来非常令人困惑。

到目前为止，我已经尝试了以下方法：

创建一个C数组（我猜它是通过指针引用），类似于：

   UINT16 * MyFunction()
    {

        ...grab image... and return Buffer

        size = 2088 * 2080 * sizeof( UINT16);
        array = (BYTE*) malloc(size);
        API_Write_to_buffer(Buffer, 0, array, size);

        API_Free_Buffer(Buffer);


        return array;
    }   

我可以尝试在python中获取数组的返回值：

Data = MyFunction() 
array = (c_ubyte * size).from_address(addressof(Data.contents))

MyFunction（）如下所示：

def MyFunction():

    MyFunction= lib.MyFunction
    MyFunction.restype = POINTER(c_ubyte)
    img_arr = MyFunction()
    return img_arr

另一种选择是使用指针逐行读取：

for(i=0;i<Width;i++)
{
    Ptr = (UINT16*)(bufferPointer + i*bufferWidth);
    for(j=0;j<Height;j++)
    {
    ...
    }
}

更新：

事实证明，我需要为数组分配一个指针，但它是C语言中的UINT16。当我尝试将数组转换为numpy数组时，python崩溃了。 我可以从包装器函数中得到它：

def Grab(nframes):

    Grab= 2112 * 2088 * 8   (size of frame width * height * pixelDepth)
    Grab= lib.MyFunction
    Grab.argtypes = [c_uint32]
    Grab.restype = POINTER(c_uint16 * array_length)
    r = Grab(nframes)
    return r

调用函数如下所示：

Data = Grab(1)
print Data

哪个返回：

<__main__.LP_c_ushort_Array_4409856 object at 0x000000000436FE48>

Answer 1

假设您有一个c_ubyte数组img_buffer并且它具有width和height ，因为它是一个图像。 我将假设您已经拥有了它，因为它看起来就像您在代码中所做的一样。

您对c_ubyte数组的转换应如下所示：

img_buffer = (c_ubyte * (height * bytes_per_pixel) *
         (width * bytes_per_pixel)).from_address(addressof(data.contents))

您可以使用numpy将其转换为ndarray，方法是：

import numpy as np

img = np.ndarray(buffer=img_buffer, dtype=np.uint16, shape=(width, height, 1))

一旦有了ndarray，就可以使用PIL或matplotlib或OpenCV进行显示。

编辑：看到您的图像是16位的，所以我更改了数据类型。