Cython - 如何正确调用带有引用的运算符

Cython - how to properly call operator with references

本文关键字:引用 运算符 调用 何正确 Cython      更新时间:2023-10-16

我的 Cython 包装器中有以下代码到C++代码:

# distutils: language = c++
# distutils: sources = symbolic.cpp
from libcpp.vector cimport vector
from libcpp.pair cimport pair
from libcpp.string cimport string
from libcpp cimport bool
cdef extern from "symbolic.h" namespace "metadiff::symbolic":
    cdef cppclass SymbolicMonomial:
        vector[pair[int, int]] powers
        long long coefficient;
        SymbolicMonomial()
        SymbolicMonomial(long)
        SymbolicMonomial(const SymbolicMonomial&)
        bool is_constant()
        long long int eval(vector[int]&)
        long long int eval()
        string to_string()
        string to_string_with_star() const
    cdef SymbolicMonomial mul_mm"operator*"(const SymbolicMonomial&, const SymbolicMonomial&)
    # SymbolicMonomial operator*(long long, const SymbolicMonomial&)
    # SymbolicMonomial operator*(const SymbolicMonomial&, long long)

cdef class SymMonomial:
    cdef SymbolicMonomial* thisptr      # hold a C++ instance which we're wrapping
    def __cinit__(self):
        self.thisptr = new SymbolicMonomial()
    def __cinit__(self, int value):
        self.thisptr = new SymbolicMonomial(value)
    def __dealloc__(self):
        del self.thisptr
    def is_constant(self):
        return self.thisptr.is_constant()
    def eval(self):
        return self.thisptr.eval()
    def __str__(self):
        return self.to_string_with_star()
    def to_string(self):
        return self.thisptr.to_string().decode('UTF-8')
    def to_string_with_star(self):
        return self.thisptr.to_string_with_star().decode('UTF-8')
    def __mul__(self, other):
        return mul_mm(self.thisptr, other)
def variable(variable_id):
    monomial = SymMonomial()
    monomial.thisptr.powers.push_back((variable_id, 1))
    return monomial

但是,我从来没有弄清楚如何正确调用mul_mm方法。它一直说Cannot convert 'SymbolicMonomial' to Python object反之亦然。问题是我需要能够以这种方式将两个对称项相乘。但是,由于某种原因,我无法掌握如何正确执行此操作的窍门。有什么建议吗?

您有许多问题:

  1. 你不能C++对象直接返回给 Python - 你需要返回你的包装器类型(分配给包装器的thisptr

  2. 在调用函数时,无法保证selfother的类型正确(请参阅 http://docs.cython.org/src/userguide/special_methods.html#arithmetic-methods 中的注释,了解如何按任一顺序使用操作数调用方法)。要使用 Cython 类的 C/C++ 成员,您需要确保 Cython 知道该对象确实属于该类。我建议使用 <Classname?> 样式转换(请注意问号),如果不匹配,则会引发异常。

  3. 您还需要从other获取thisptr,而不仅仅是将 Python 包装类传递给您的C++函数。

以下应该有效。

def __mul__(self,other):
    cdef SymMonomial tmp = SymMonomial()
    cdef SymMonomial self2, other2
    try:
        self2 = <SymMonomial?>self
        other2 = <SymMonomial?>other
    except TypeError:
        return NotImplemented # this is what Python expects for operators
                      # that don't know what to do
    tmp.thisptr[0] = mul_mm(self2.thisptr[0],other2.thisptr[0])
    return tmp