python - 在 Cython 中按名称调用 cdef 函数

Question

我在 Cython 中有一堆 cdef 函数，它们由 pyx 文件中的 def 函数调用，例如:

cdef inline void myfunc_c(...):
    (...)
    return

def wrapper(...):
    myfunc_c(...)
    return

这很好用。但是为了简化不必为每个 cdef 函数都有一个 python 包装器，我试图通过将它们分配给字典或类似的东西来按名称索引 cdef 函数:

def wrapper(operation):
    if operation == 'my_func':
        func = myfunc_c
    func(...)
    return

但这行不通。 Cython 提示它不知道 myfunc_c 的类型。

有什么方法可以通过名称索引或调用 cpdef 函数(例如使用字符串)？我也尝试过类似 locals()['myfunc_c'] 的方法，但这也不起作用。

Answer 1

对于 general cdef 函数，这是不可能的 - 它们只定义了 C 接口(interface)而不是 Python 接口(interface)，因此没有可用的内省(introspection)。

对于 cdef 函数 declared with api (例如 cdef api funcname())这实际上是可能的。有一个未记录的字典 __pyx_capi__。这定义了包含函数指针的 PyCapsules 字典(按名称索引)。然后你会做

capsule_name = PyCapsule_GetName(obj)
func = PyCapsule_GetPointer(obj, capsule_name)

(其中 PyCapsule_* 是从 Python C API 导入的函数)。 func 是一个 void*，您可以将其转换为适当类型的函数指针。正确使用类型很重要，这取决于您!

虽然未记录，但 __pyx_capi__ 接口(interface)相对稳定，并被 Scipy 使用，例如其 LowLevelCallable 特性。

cpdef 函数定义了 Python 和 C 接口(interface)。在一个函数中，它们将在 globals() 而不是 locals() 中可用(因为 locals() 只给出在该函数中定义的变量.)

---

不过，我实际上并不认为您想要这样做。我认为您只想使用 cpdef 而不是 cdef 因为这会自动为该函数生成一个 Python 包装器。