在Cython中包装C ++朋友非成员运算符

时间:2017-12-22 08:41:56

标签: c++ operator-overloading cython friend

我是c ++和cython的新手,并且对在Cython中包装C ++朋友非成员运算符感到困惑。这是我试图包装的一个小例子,但失败了。非常感谢

现在,我如何在pyx文件中声明友元运算符

all files can be found here, makefile for test   -

Rectangle.h

 namespace shapes {
    class Rectangle {
      public:
        int x0, y0, x1, y1;
        Rectangle(int x0=0, int y0=0, int x1=0, int y1=0);
        ~Rectangle();
        int getLength();
        Rectangle operator+(const Rectangle& target);
        friend Rectangle operator-(const Rectangle & left, const Rectangle & right);
    };
    }

Rectangle.cpp

#include "Rectangle.h"

using namespace shapes;

Rectangle::Rectangle(int X0, int Y0, int X1, int Y1) {
    x0 = X0;
    y0 = Y0;
    x1 = X1;
    y1 = Y1;
}


int Rectangle::getLength() {
    return (x1 - x0);
}

Rectangle::~Rectangle()
{

}

Rectangle Rectangle::operator+(const Rectangle & target) {
    return Rectangle(x0+target.x0, y0+target.y0,x1+target.x1,y1+target.y1);
}

Rectangle operator-(const Rectangle & left,const Rectangle & right) {
    return Rectangle(left.x0 - right.x0,
                     left.y0-right.y0,
                     left.x1-right.x1,
                     left.y1-right.y1);
}

pyx文件

from libcpp.vector cimport vector
from cython.operator cimport dereference as deref 

# c++ interface to cython
cdef extern from "Rectangle.h" namespace "shapes":
    cdef cppclass Rectangle:
        Rectangle() except +
        Rectangle(int, int, int, int) except +
        int x0, y0, x1, y1
        int getLength()
        Rectangle opadd "operator+"(Rectangle right)
        Rectangle opsub "operator-" (Rectangle right)
     #  Rectangle opsub "operator-"(Rectangle left ,Rectangle right)

# cdef extern from "Rectangle.h" namespace "shapes":
#     cdef Rectangle opsub "operator-"(Rectangle left ,Rectangle right) 

# creating a cython wrapper class
cdef class PyRectangle:
    cdef Rectangle *thisptr      # hold a C++ instance which we're wrapping
    def __cinit__(self, int x0=0, int y0=0, int x1=0, int y1=0):
        self.thisptr = new Rectangle(x0, y0, x1, y1)
    def __dealloc__(self):
        del self.thisptr
    def getLength(self):
        return self.thisptr.getLength()
    def __add__(PyRectangle left,PyRectangle right):
        cdef Rectangle rect = left.thisptr.opadd(right.thisptr[0])
        cdef PyRectangle sum = PyRectangle(rect.x0,rect.y0,rect.x1,rect.y1)
        return sum
    def __sub__(PyRectangle left,PyRectangle right):
        cdef Rectangle rect = left.thisptr.opsub(right.thisptr[0])
        # cdef Rectangle rect = opsub(left.thisptr[0],right.thisptr[0])
        cdef PyRectangle sub = PyRectangle(rect.x0,rect.y0,rect.x1,rect.y1)
        return sub
    def __repr__(self):
        return "PyRectangle[%s,%s,%s,%s]" % (
                self.thisptr.x0,
                self.thisptr.y0,
                self.thisptr.x1,
                self.thisptr.y1)

我尝试了serval方法,例如

cdef extern from "Rectangle.h" namespace "shapes":
    cdef Rectangle opsub "operator-"(Rectangle left ,Rectangle right) 

或者只是假装它是cppclass声明中的成员定义的运算符

Rectangle opsub "operator-" (Rectangle right)

我无法编译两种方式,因为它显示了

error: no member named 'operator-' in 'shapes::Rectangle'

1 个答案:

答案 0 :(得分:2)

抱歉,我在Rectangle.cpp中发现了一个很大的错误。运算符的定义需要以命名空间

作为前缀
    Rectangle shapes::operator-(const Rectangle & left,const Rectangle & right) {
    return Rectangle(left.x0 - right.x0,
                     left.y0-right.y0,
                     left.x1-right.x1,
                     left.y1-right.y1);
}

根据DavidW的提示,可以修改rect.pyx以直接调用操作符。

    # c++ interface to cython
cdef extern from "Rectangle.h" namespace "shapes":
    cdef cppclass Rectangle:
        Rectangle() except +
        Rectangle(int, int, int, int) except +
        int x0, y0, x1, y1
        int getLength()
        Rectangle operator+(Rectangle right)
        Rectangle operator-(Rectangle right)

####  nonmember operator can also work :
# cdef extern from "Rectangle.h" namespace "shapes":
    # cdef Rectangle operator-(Rectangle left ,Rectangle right) 

# creating a cython wrapper class
cdef class PyRectangle:
    cdef Rectangle *thisptr      # hold a C++ instance which we're wrapping
    def __cinit__(self, int x0=0, int y0=0, int x1=0, int y1=0):
        self.thisptr = new Rectangle(x0, y0, x1, y1)
    def __dealloc__(self):
        del self.thisptr
    def getLength(self):
        return self.thisptr.getLength()
    def __add__(PyRectangle left,PyRectangle right):
        cdef Rectangle rect = left.thisptr[0] + right.thisptr[0]
        cdef PyRectangle sum = PyRectangle(rect.x0,rect.y0,rect.x1,rect.y1)
        return sum
    def __sub__(PyRectangle left,PyRectangle right):
        cdef Rectangle rect = left.thisptr[0] - right.thisptr[0]
        # cdef Rectangle rect = opsub(left.thisptr[0],right.thisptr[0])
        cdef PyRectangle sub = PyRectangle(rect.x0,rect.y0,rect.x1,rect.y1)
        return sub