Перегрузка Cython Специальные методы? - PullRequest
Новая
       7

Перегрузка Cython Специальные методы?

2 голосов
/ 09 августа 2011

Есть ли возможность перегрузить __cinit__ или __add__? Примерно так: 

cdef class Vector(Base):
    cdef double x, y, z

    def __cinit__(self, double all):
        self.x = self.y = self.z = all

    def __cinit__(self, double x, double y, double z):
        self.x  = x
        self.y  = y
        self.z  = z

    def __str__(self):
        return "Vector(%s, %s, %s)" % (self.x, self.y, self.z)

    def __add__(self, Vector other):
        return Vector(
            self.x + other.x,
            self.y + other.y,
            self.z + other.z,
        )

    def __add__(self, object other):
        other   = <double>other
        return Vector(
            self.x + other.x,
            self.y + other.y,
            self.z + other.z,
        )

Вызов Vector(0) + Vector(2, 4, 7) говорит мне, что здесь необходимо использовать число с плавающей точкой, поэтому кажется, что __add__(self, Vector other) не распознается как перегруженный метод.

Это потому, что специальные методы не должны определяться как cdef, и только cdef функции могут быть перегружены?

1 Ответ

3 голосов
/ 09 августа 2011

Я не думаю, что перегрузка операторов специальными функциями поддерживается в Cython.

Лучше всего вручную создать логику проверки типов и соответственно привести объект Python. 

def __add__(self, other):
    if type(other) is float:
        return self.__add__(<double> other)
    elif isinstance(other,Vector):
        return self.__add__(<Vector> other)
    ...
...