[英]Can I assign a cython class method to a gsl_function structure via wrapping or cython re-declaration of gsl_function?
这个问题是基于这里的问题
我有以下cython代码,该代码导入一个文件( cGslInteg.pxd
),其中包含gsl/gsl_integration.h
和gsl/gsl_math.h
头文件的相关部分的cython声明,然后定义一个用于计算积分的类:
from libc.math cimport log
from libc.math cimport sqrt
from cGslInteg cimport *
ctypedef double * double_ptr
cdef double integrand(double x, void * params):
"""integrand implemented outside class """
cdef double alpha = (<double_ptr> params)[0]
cdef double f = log(alpha*x) / sqrt(x)
return f
cdef class CalcSomething(object):
def integrate(self, double a, double b, double alpha):
"""This does the integral"""
cdef double result, error
cdef gsl_integration_workspace * W
W = gsl_integration_workspace_alloc(1000)
cdef gsl_function F
F.function = &integrand # works
# F.function = &self._integrand # doesn't work
cdef double params[1]
params[0] = alpha
F.params = ¶ms
cdef double epsabs = 0.
cdef double epsrel = 1e-7
gsl_integration_qags(&F, a, b, epsabs, epsrel, 1000, W, &result, &error)
gsl_integration_workspace_free(W)
return result
cdef double _integrand(self, double x, void * params):
"""integrand implemented inside class """
cdef double alpha = (<double_ptr> params)[0]
cdef double f = log(alpha*x) / sqrt(x)
return f
上面的代码在使用该类之外的 F.function = &integrand
函数时可以编译并正确运行( F.function = &integrand
)。 但是,将这一行更改为下面注释掉的那一行( F.function = &self._integrand
),以代替使用类内部的 F.function = &self._integrand
函数,会导致以下编译错误:
cython_class_gsl.pyx:31:21: Cannot assign type 'double (*)(CalcSomething, double, void *)' to 'double (*)(double, void *)'
这是有道理的,因为我在cGslInteg.pxd
对gsl_function
的cGslInteg.pxd
是:
cdef extern from "gsl/gsl_math.h":
# Definition of an arbitrary function with parameters
ctypedef struct gsl_function:
double (* function) (double x, void * params)
void * params
我的问题是:我可以重新声明gsl_function
以便它期望使用double (*)(PythonObject, double, void *)
类型的东西,还是可以包装self._integrand
使其看起来像double (*)( double, void *)
?
编辑:一些额外的说明,以使其更接近于现实生活中的问题/问题:
self._integrand
任何包装都必须在CalcSomething
类中定义。 alpha
不能是类变量(即必须在params
参数中传递给被积分数) 这就是params
目的。 你会做这样的事情(大大削减了例子...)
from libc.math cimport log,sqrt
cdef class CalcSomething:
cdef double integrand(self, double x, double alpha):
return log(alpha*x)/sqrt(x)
cdef double wrapped_integrand_func(double x, void* params):
cdef object params_as_object = <object>params
cdef CalcSomething instance = params_as_object[0] # we need to know the type to use cdef functions
alpha = params_as_object[1]
return instance.integrand(x,alpha)
def test(alpha):
cdef object o = (CalcSomething(),alpha) # store Calc something and alpha as a tuple
cdef void* o_ptr = <void*>o # be careful with reference counts here - o_ptr doesn't keep o alive!
print(wrapped_integrand_func(4.3,o_ptr))
这显然不是包装绑定方法的通用解决方案,但是在这种情况下,这正是gsl允许您传递void*
。
一个警告是,您必须确保o
保持生存状态, o_ptr
您想使用o_ptr
。
更新 (以匹配稍作修改的问题):
积分参数alpha不能是类变量(即必须在params参数中传递给被积分数)
这很容易做到。 在我的原始代码中,我使用了一个类变量,但是现在我将它们分别作为一个元组传递给了它们。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.