1

我有一些代码可以编译一个 C 函数以使用 PyUFunc_FromFuncAndData 创建一个 numpy 通用函数。我已经编写了一些 cython 来创建 ufunc，但如果可能的话，我想使用 ctypes 来完成它，因为我打算分发它并且我想避免用户进行编译步骤。

问题是 PyUFunc_FromFuncAndData 返回一个指向 PyObject 的指针。是否可以将其用作 python 代码中的对象？

基本上，我希望能够将以下 cython 代码转换为 python/ctypes：

from numpy cimport NPY_DOUBLE
from libc.stdlib cimport malloc, free

cdef extern from "numpy/ufuncobject.h":
    ctypedef void (*PyUFuncGenericFunction) (char **, Py_ssize_t *, Py_ssize_t *, void *)
    object PyUFunc_FromFuncAndData (PyUFuncGenericFunction *, void **, char *, int, int, int, int, char *, char *, int)
    void import_ufunc()

import_ufunc()


cdef class UFuncWrapper:

    cdef readonly object func
    cdef object _llvm_func
    cdef PyUFuncGenericFunction function
    cdef char *types
    cdef bytes name

    def __init__(self, func, ufunc, long long ptr):
        self._llvm_func = ufunc # keep a ref to prevent it from being gced
        cdef int num_args = len(func.args)
        self.types = <char*>malloc(sizeof(char)*(num_args+1))
        self.name = func.name
        cdef int i
        for i in range(num_args+1):
            self.types[i] = NPY_DOUBLE
        self.function = <PyUFuncGenericFunction>ptr
        self.func = PyUFunc_FromFuncAndData(
            &self.function,
            NULL,
            self.types,
            1,  #ntypes
            num_args,
            1,
            -1, # PyUFunc_None,
            self.name,
            self.name,   #FIXME: __doc__
            0)

    def __dealloc__(self):
        free(self.types)

    def __call__(self, *args):
        return self.func(*args)
4

1 回答 1

3

将该函数的 restype 设置为 ctypes.py_object。下面的示例使用对 python C-API 的调用，但它对其他任何东西都一样。

import ctypes
class Foo(object):
    bar='baz'

foo=ctypes.py_object(Foo)
print 'Memory adress of Foo.bar object:',
print ctypes.pythonapi.PyObject_GetAttrString(foo,'bar') # prints the pointer

ctypes.pythonapi.PyObject_GetAttrString.restype = ctypes.py_object

print 'Actual Foo.bar after we set restype correctly:', 
print ctypes.pythonapi.PyObject_GetAttrString(foo,'bar') # prints "baz"
于 2011-03-02T10:40:46.737 回答