static PyObject *
array_divmod(PyArrayObject *op1, PyObject *op2)
{
PyObject *divp, *modp, *result;
GIVE_UP_IF_HAS_RIGHT_BINOP(op1, op2, "__divmod__", "__rdivmod__", 0);
divp = array_floor_divide(op1, op2);
if (divp == NULL) {
return NULL;
}
else if(divp == Py_NotImplemented) {
return divp;
}
modp = array_remainder(op1, op2);
if (modp == NULL) {
Py_DECREF(divp);
return NULL;
}
else if(modp == Py_NotImplemented) {
Py_DECREF(divp);
return modp;
}
result = Py_BuildValue("OO", divp, modp);
Py_DECREF(divp);
Py_DECREF(modp);
return result;
}
static PyObject *
array_inplace_floor_divide(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__ifloordiv__", "__rfloordiv__", 1, nb_inplace_floor_divide);
return PyArray_GenericInplaceBinaryFunction(m1, m2,
n_ops.floor_divide);
}
static PyObject *
array_inplace_true_divide(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__itruediv__", "__rtruediv__", 1);
return PyArray_GenericInplaceBinaryFunction(m1, m2,
n_ops.true_divide);
}
static PyObject *
array_inplace_power(PyArrayObject *a1, PyObject *o2, PyObject *NPY_UNUSED(modulo))
{
/* modulo is ignored! */
PyObject *value;
GIVE_UP_IF_HAS_RIGHT_BINOP(a1, o2, "__ipow__", "__rpow__", 1);
value = fast_scalar_power(a1, o2, 1);
if (!value) {
value = PyArray_GenericInplaceBinaryFunction(a1, o2, n_ops.power);
}
return value;
}
/* Need this to be version dependent on account of the slot check */
static PyObject *
array_matrix_multiply(PyArrayObject *m1, PyObject *m2)
{
static PyObject *matmul = NULL;
npy_cache_pyfunc("numpy.core.multiarray", "matmul", &matmul);
if (matmul == NULL) {
return NULL;
}
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__matmul__", "__rmatmul__",
0, nb_matrix_multiply);
return PyArray_GenericBinaryFunction(m1, m2, matmul);
}
static PyObject *
array_remainder(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__mod__", "__rmod__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.remainder);
}
static PyObject *
array_divide(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__div__", "__rdiv__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.divide);
}
static PyObject *
array_multiply(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__mul__", "__rmul__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.multiply);
}
static PyObject *
array_subtract(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__sub__", "__rsub__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.subtract);
}
static PyObject *
array_add(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__add__", "__radd__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.add);
}
static PyObject *
array_inplace_bitwise_xor(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__ixor__", "__rxor__", 1);
return PyArray_GenericInplaceBinaryFunction(m1, m2, n_ops.bitwise_xor);
}
static PyObject *
array_inplace_right_shift(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__irshift__", "__rrshift__", 1);
return PyArray_GenericInplaceBinaryFunction(m1, m2, n_ops.right_shift);
}
static PyObject *
array_bitwise_or(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__or__", "__ror__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.bitwise_or);
}
static PyObject *
array_left_shift(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__lshift__", "__rlshift__", 0);
return PyArray_GenericBinaryFunction(m1, m2, n_ops.left_shift);
}
static PyObject *
array_inplace_add(PyArrayObject *m1, PyObject *m2)
{
GIVE_UP_IF_HAS_RIGHT_BINOP(m1, m2, "__iadd__", "__radd__", 1, nb_inplace_add);
return PyArray_GenericInplaceBinaryFunction(m1, m2, n_ops.add);
}
