| // Copyright David Abrahams 2002. |
| // Distributed under the Boost Software License, Version 1.0. (See |
| // accompanying file LICENSE_1_0.txt or copy at |
| // http://www.boost.org/LICENSE_1_0.txt) |
| |
| #include <boost/python/object_operators.hpp> |
| #include <boost/python/detail/raw_pyobject.hpp> |
| |
| namespace boost { namespace python { namespace api { |
| |
| # define BOOST_PYTHON_COMPARE_OP(op, opid) \ |
| BOOST_PYTHON_DECL object operator op(object const& l, object const& r) \ |
| { \ |
| return object( \ |
| detail::new_reference( \ |
| PyObject_RichCompare( \ |
| l.ptr(), r.ptr(), opid)) \ |
| ); \ |
| } |
| BOOST_PYTHON_COMPARE_OP(>, Py_GT) |
| BOOST_PYTHON_COMPARE_OP(>=, Py_GE) |
| BOOST_PYTHON_COMPARE_OP(<, Py_LT) |
| BOOST_PYTHON_COMPARE_OP(<=, Py_LE) |
| BOOST_PYTHON_COMPARE_OP(==, Py_EQ) |
| BOOST_PYTHON_COMPARE_OP(!=, Py_NE) |
| # undef BOOST_PYTHON_COMPARE_OP |
| |
| |
| #define BOOST_PYTHON_BINARY_OPERATOR(op, name) \ |
| BOOST_PYTHON_DECL object operator op(object const& l, object const& r) \ |
| { \ |
| return object( \ |
| detail::new_reference( \ |
| PyNumber_##name(l.ptr(), r.ptr())) \ |
| ); \ |
| } |
| |
| BOOST_PYTHON_BINARY_OPERATOR(+, Add) |
| BOOST_PYTHON_BINARY_OPERATOR(-, Subtract) |
| BOOST_PYTHON_BINARY_OPERATOR(*, Multiply) |
| #if PY_VERSION_HEX >= 0x03000000 |
| // We choose FloorDivide instead of TrueDivide to keep the semantic |
| // conform with C/C++'s '/' operator |
| BOOST_PYTHON_BINARY_OPERATOR(/, FloorDivide) |
| #else |
| BOOST_PYTHON_BINARY_OPERATOR(/, Divide) |
| #endif |
| BOOST_PYTHON_BINARY_OPERATOR(%, Remainder) |
| BOOST_PYTHON_BINARY_OPERATOR(<<, Lshift) |
| BOOST_PYTHON_BINARY_OPERATOR(>>, Rshift) |
| BOOST_PYTHON_BINARY_OPERATOR(&, And) |
| BOOST_PYTHON_BINARY_OPERATOR(^, Xor) |
| BOOST_PYTHON_BINARY_OPERATOR(|, Or) |
| #undef BOOST_PYTHON_BINARY_OPERATOR |
| |
| #define BOOST_PYTHON_INPLACE_OPERATOR(op, name) \ |
| BOOST_PYTHON_DECL object& operator op##=(object& l, object const& r) \ |
| { \ |
| return l = object( \ |
| (detail::new_reference) \ |
| PyNumber_InPlace##name(l.ptr(), r.ptr())); \ |
| } |
| |
| BOOST_PYTHON_INPLACE_OPERATOR(+, Add) |
| BOOST_PYTHON_INPLACE_OPERATOR(-, Subtract) |
| BOOST_PYTHON_INPLACE_OPERATOR(*, Multiply) |
| #if PY_VERSION_HEX >= 0x03000000 |
| // Same reason as above for choosing FloorDivide instead of TrueDivide |
| BOOST_PYTHON_INPLACE_OPERATOR(/, FloorDivide) |
| #else |
| BOOST_PYTHON_INPLACE_OPERATOR(/, Divide) |
| #endif |
| BOOST_PYTHON_INPLACE_OPERATOR(%, Remainder) |
| BOOST_PYTHON_INPLACE_OPERATOR(<<, Lshift) |
| BOOST_PYTHON_INPLACE_OPERATOR(>>, Rshift) |
| BOOST_PYTHON_INPLACE_OPERATOR(&, And) |
| BOOST_PYTHON_INPLACE_OPERATOR(^, Xor) |
| BOOST_PYTHON_INPLACE_OPERATOR(|, Or) |
| #undef BOOST_PYTHON_INPLACE_OPERATOR |
| |
| object::object(handle<> const& x) |
| : object_base(python::incref(python::expect_non_null(x.get()))) |
| {} |
| |
| }}} // namespace boost::python |