Cython - 将指针数组转换为Python对象

Question

好的，我快要完成这个项目了，感觉已经触手可及了。在过去的几周里，我一直在尝试创建一个Python扩展，用来通过Cython与一个用C++写的库进行交互。在这里得到了一些帮助，还有几个朋友的支持，我觉得自己已经完成了大约98%的工作。剩下的唯一问题是：我实在搞不懂怎么把一个指向无符号短整型数组的指针转换成Python对象（最好是列表）。

先说一下背景，我正在尝试与库中的一个设置回调函数的部分进行交互，这个我已经成功实现了，代码如下：

global callbackfunc

ctypedef unsigned short const_ushort "const uint16_t"

ctypedef void (*Function1)(const_ushort *data, unsigned width, unsigned height)

cdef extern from "lib.hpp":
    void SetCallback(Function1)

cdef void cSetCallback(Function1 function):
    SetCallback(function)

cdef void callcallback(const_ushort *data, unsigned width, unsigned height):
    global callbackfunc
    callbackfunc(data,width,height)

cSetCallback(callcallback)

def PySetCallback(callbackFunc):
    global callbackfunc
    callbackfunc = callbackFunc

问题出现在“callcallback”这个函数里，我遇到了一个错误：“无法将'const_ushort *'转换为Python对象”。我第一次尝试解决这个问题时，创建了一个新的Python列表，然后循环遍历数组，把每个元素放到Python列表里，像这样：

datalist = []
for i in range(width*height):
    datalist += data[i]

但是，结果是编译的Cython代码试图把一个类型定义为“const const unsigned short”，这显然是个问题。

接着我尝试了这个：

datalist = []
for i in data:
    datalist += i

结果提示我“C数组迭代需要已知的结束索引”。需要说明的是，我对C/C++了解得很少，所以大部分内容对我来说并不太明白。

总之，有没有什么有效的方法可以把这样的指针转换成Python对象（最好比循环遍历数组快，因为通常有大约57344个项目，而且这个过程对时间要求很高）

编辑：再多说一点，正如我提到的，我在处理回调，而库中的C++函数会发送一个指向“const uint_16”数组的指针，这就是我这样定义const_ushort的原因，因为否则类型就不一致了。我无法以任何方式修改这个库。

编辑2：看起来我搞定了。最后我不得不明确将数组转换为无符号短整型数组，而不是常量无符号短整型数组，这样我才能用非常量来索引它们。为此，我创建了另一个C++函数，像这样（是别人帮我写的，我对C++几乎一无所知）：

unsigned short *convert_short(const unsigned short *test){ return const_cast<unsigned short *>(test); }

这让我在我的类中创建了“getindex”函数，并根据这个函数返回正确的值。所以，Python似乎能够正确读取这些数组，问题看起来解决了。非常感谢大家。

性能优化 c# 回调函数类型转换数据转换 cython python扩展指针数组

Cython - 将指针数组转换为Python对象

2 个回答

撰写回答