使用ctypes将python数组传递给c++函数

passing python array to c++ function using ctypes

本文关键字:c++ 函数 数组 ctypes python 使用      更新时间:2023-10-16

我对这种扩展还是个新手(尽管我在C++和python方面有很好的背景)

我想要的是使用ctypes模块将数据传递给C++函数到目前为止对我有效的例子:

test.dll文件

#define DLLEXPORT extern "C" __declspec(dllexport)
DLLEXPORT int sum(int a, int b) {
    return a + b;
}

sum_test.py文件:

from ctypes import cdll
mydll = cdll.LoadLibrary('test.dll')
mydll.sum(5, 3)

结果是8,效果很好。。。

注意:我正在使用它为blender开发一个插件,我将传递给C++函数的数据是一个巨大的数组(元组或任何类型的数组),这些数据将传递给GPU,用OpenCL 进行处理

C++中的工作流程主要是传递指针(将整个数组数据复制到GPU)

所以问题是:在C++中获取元组指针的最佳方法(也是最快的)(希望能解释python和C++中发生的事情)

最简单的方法是告诉ctypes函数将指针指向某个类型的数组,在许多情况下Python/ctypes会计算出其余部分。

libc = CDLL('libc.so.6') # on Debian/Linux
printf = libc.printf
printf.restype = c_int
printf.argtypes = [POINTER(c_char), c_int]
printf('< %08x >', 1234)

这假设函数将只接受一个指针。例如,如果您的函数正好取五个浮点值,则可以根据参数类型提供c_float * 5