如何使用Python脚本将指针变量从一个cython模块传输到另一个模块

Question

假设我们有一个cython类a，它的指针指向float，如

# A.pyx
cdef class A:
    cdef float * ptr

我们还在另一个模块中有一个cython类b，它需要访问ptr下的数据：

# B.pyx

cdef class B:
    cdef float * f_ptr

    cpdef submit(self, ptr_var):
        self.f_ptr= get_from( ptr_var ) # ???

使用A和B的相应Python代码可能类似于

from A import A
from B import B

a = A()
b = B()
ptr = a.get_ptr()
b.submit(ptr)

我们如何定义get_ptr()以及在b中，我们将对get_from使用什么？

Answer 1

解决方案是将指针变量包装到一个Python对象中。模块libc.stdint提供了一个名为uintptr_t的类型，它是一个足够大的整数，可以存储任何类型的指针。这样，解决方案可能如下所示。

from libc.stdint cimport uintptr_t
cdef class A:
    cdef float * ptr

    def get_ptr(self):
        return <uintptr_t>self.ptr

尖括号中的表达式 对应于对uintptr_t的强制转换。在类B中，我们必须将变量转换回一个要浮点的指针。

from libc.stdint cimport uintptr_t
cdef class B:
    cdef float * f_ptr

    cpdef submit(self, uintptr_t ptr_var):
        self.f_ptr= <float *>ptr_var

这适用于任何类型的指针，不仅适用于浮动的指针。必须确保这两个模块(a和B)处理的指针类型相同，因为一旦指针包装在UINTPTR_T中，该信息就会丢失。