python - 有效地将字符串(或元组)转换为 ctypes 数组

Question

我有获取 PIL 图像并将其转换为 ctypes 数组以传递给 C 函数的代码:

w_px, h_px = img.size
pixels = struct.unpack('%dI'%(w_px*h_px), img.convert('RGBA').tostring())
pixels_array = (ctypes.c_int * len(pixels))(*pixels)

但我正在处理大图像，将那么多项目解包到函数参数中似乎非常慢。要获得合理的加速，我能做的最简单的事情是什么？

我只是将转换为元组作为中间步骤，所以如果没有必要，那就更好了。

Answer 1

可以先构建一个未初始化的数组:

pixarray = (ctypes.c_int * (w_px * h_px))()

然后将图像的内容复制到其中:

# dylib in MacOSX, cdll.wincrt in Win, libc.so.? in Unix, ...
clib = ctypes.CDLL('libc.dylib')

_ = clib.memcpy(pixarray, im.tostring(), w_px * h_px * 4)

memcpy 的返回值是一个你不关心的地址，所以我通过将它分配给名称“单下划线”(按照惯例意味着“我不关心这个”)来“吞下”它；- ).

编辑:正如@Mu Mind 在评论中指出的那样，后一个片段可以有效地简化为使用 ctypes.memmove无需依赖平台来搜索 clib:只需执行即可

_ = ctypes.memmove(pixarray, im.tostring(), w_px * h_px * 4)