86 lines
3.4 KiB
C++
86 lines
3.4 KiB
C++
// 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
|