Cython - 如何使用引用正确调用运算符

Cython - how to properly call operator with references

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

# 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 反之亦然。问题是我需要能够以这种方式将两个 SymMonomials 相乘。但是由于某种原因,我无法掌握如何正确执行它的窍门。有什么建议吗?

您有很多问题:

  1. 您不能 return C++ 对象直接指向 Python - 您需要 return 您的包装器类型(分配给 thisptr包装纸)

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

  3. 您还需要从 other 获取 thisptr,而不是仅仅将 Python 包装器 class 传递给您的 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