如何在Cython中将函数指针传递给外部程序

时间:2012-01-10 09:15:16

标签: python cython

我正在尝试编写一个C程序的包装器,以便我可以从Python调用它。我正在使用Cython来做到这一点。 C函数将回调函数作为参数,但这个回调函数只能在python程序的运行时知道。我一直在寻找如何做到这一点,似乎没有简单的解决方案,但以下似乎工作:

#python.py

libc = cdll.LoadLibrary("myfunc.so") #Callback function is defined in myfunc.so 
....
c_wrapper(libc.fun, ...)

#c_wrapper.pyx

cdef extern void mainfunction(void *F, ...) #The intial C function we are wrapping
ctypedef void (*myfuncptr) () 

def c_wrapper(f, ...) # our function pointer is passed to the wrapper as a Python object
    cdef myfuncptr thisfunc
    thisfunc = (<myfuncptr*><size_t>addressof(f))[0]
    mainfunction(thisfunc, ...)

此方法适用于C和FORTRAN函数(我假设它适用于大多数编译的语言)和Python函数(使用C类型),但它似乎有点尴尬。在Cython中有没有更简单的方法呢?

由于

编辑:我无法更改我想要包装的C库

1 个答案:

答案 0 :(得分:5)

我猜你知道this

<击>   

是否可以从C调用我的Python代码?

     

答案:是的,很容易。按照演示/回调/中的示例进行操作   Cython源码分发   

所以,知道你可以卸载你的主要功能,让我们采取另一种方法。我写了几个愚蠢的函数来测试它:

/* lib.c -> lib.so */
#include <stdio.h>

void fn1(void) {
        puts("Called function 1");
}

void fn2(void) {
        puts("Called function 2");
}

然后,接受回调的函数

/* main.c -> main.so */

typedef void (*callback)();

void mainfunction(void *F) {
        ((callback)F)();
}

可以直接从Python传递:

>>> from ctypes import cdll
>>> lib = cdll.LoadLibrary('./lib.so')
>>> main = cdll.LoadLibrary('./main.so')
>>> main.mainfunction(lib.fn1)
Called function 1
>>> main.mainfunction(lib.fn2)
Called function 2

现在,让我们包装一个Python函数:

>>> from ctypes import CFUNCTYPE
>>> def pyfn():
...     print "Called the Python function"
... 
>>> CWRAPPER = CFUNCTYPE(None)

>>> wrapped_py_func = CWRAPPER(pyfn)
>>> main.mainfunction(wrapped_py_func)
Called the Python function