Cython：基于调用签名的通用函数接口函数

Question

基本上，想要编写两个函数，然后使用第三个通用函数，该函数纯粹根据给定的输入参数选择使用给定参数调用两个原始函数中的哪一个。例如

cdef fun1(int a):
    #do something

cdef fun1(float a):
    #do something

#roughly stealing syntax from fortran (this is the part I don't know how to do)
interface generic_function:
    function fun1
    function fun2


#Calling the functions
cdef int a = 2
cdef float b = 1.3
generic_function(a) #runs fun1
generic_function(b) #runs fun2

，显然我可以使用 if 语句来做到这一点，但这对我来说似乎效率不高。 感谢任何和所有的帮助，干杯。

Answer 1

这可以使用 [fused types] (https://cython. readthedocs.io/en/latest/src/userguide/fusedtypes.html）

ctypedef fused int_or_float:
    int
    float

cdef generic_function(int_or_float a):
    if int_or_float is int:
        fun1(a)
    else:
        fun2(a)

if 语句在 Cython 编译函数时进行评估（您可以通过查看生成的 c 代码来验证这一点），因此不是性能问题。