#include "Python.h"

#include "pycore_modsupport.h" // _PyArg_NoKwnames()

#include "pycore_moduleobject.h" // _PyModule_GetState()

#include "pycore_pystate.h" // _PyInterpreterState_GET()

#include "pycore_tuple.h" // _PyTuple_ITEMS()

#include "pycore_typeobject.h" // _PyType_GetModuleState()

#include "pycore_unicodeobject.h" // _PyUnicode_InternMortal()

#include "clinic/_operator.c.h"

typedef struct {

PyObject *itemgetter_type;

PyObject *attrgetter_type;

PyObject *methodcaller_type;

} _operator_state;

static inline _operator_state*

get_operator_state(PyObject *module)

{

void *state = _PyModule_GetState(module);

assert(state != NULL);

return (_operator_state *)state;

}

PyDoc_STRVAR(operator_doc,

'__' are also provided for convenience.");

static int

_operator_truth_impl(PyObject *module, PyObject *a)

{

return PyObject_IsTrue(a);

}

static PyObject *

_operator_add_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Add(a, b);

}

static PyObject *

_operator_sub_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Subtract(a, b);

}

static PyObject *

_operator_mul_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Multiply(a, b);

}

static PyObject *

_operator_matmul_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_MatrixMultiply(a, b);

}

static PyObject *

_operator_floordiv_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_FloorDivide(a, b);

}

static PyObject *

_operator_truediv_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_TrueDivide(a, b);

}

static PyObject *

_operator_mod_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Remainder(a, b);

}

static PyObject *

_operator_neg(PyObject *module, PyObject *a)

{

return PyNumber_Negative(a);

}

static PyObject *

_operator_pos(PyObject *module, PyObject *a)

{

return PyNumber_Positive(a);

}

static PyObject *

_operator_abs(PyObject *module, PyObject *a)

{

return PyNumber_Absolute(a);

}

static PyObject *

_operator_inv(PyObject *module, PyObject *a)

{

return PyNumber_Invert(a);

}

static PyObject *

_operator_invert(PyObject *module, PyObject *a)

{

return PyNumber_Invert(a);

}

static PyObject *

_operator_lshift_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Lshift(a, b);

}

static PyObject *

_operator_rshift_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Rshift(a, b);

}

static int

_operator_not__impl(PyObject *module, PyObject *a)

{

return PyObject_Not(a);

}

static PyObject *

_operator_and__impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_And(a, b);

}

static PyObject *

_operator_xor_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Xor(a, b);

}

static PyObject *

_operator_or__impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Or(a, b);

}

static PyObject *

_operator_iadd_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceAdd(a, b);

}

static PyObject *

_operator_isub_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceSubtract(a, b);

}

static PyObject *

_operator_imul_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceMultiply(a, b);

}

static PyObject *

_operator_imatmul_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceMatrixMultiply(a, b);

}

static PyObject *

_operator_ifloordiv_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceFloorDivide(a, b);

}

static PyObject *

_operator_itruediv_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceTrueDivide(a, b);

}

static PyObject *

_operator_imod_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceRemainder(a, b);

}

static PyObject *

_operator_ilshift_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceLshift(a, b);

}

static PyObject *

_operator_irshift_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceRshift(a, b);

}

static PyObject *

_operator_iand_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceAnd(a, b);

}

static PyObject *

_operator_ixor_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceXor(a, b);

}

static PyObject *

_operator_ior_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlaceOr(a, b);

}

static PyObject *

_operator_concat_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PySequence_Concat(a, b);

}

static PyObject *

_operator_iconcat_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PySequence_InPlaceConcat(a, b);

}

static int

_operator_contains_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PySequence_Contains(a, b);

}

static Py_ssize_t

_operator_indexOf_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PySequence_Index(a, b);

}

static Py_ssize_t

_operator_countOf_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PySequence_Count(a, b);

}

static PyObject *

_operator_getitem_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_GetItem(a, b);

}

static PyObject *

_operator_setitem_impl(PyObject *module, PyObject *a, PyObject *b,

PyObject *c)

{

if (-1 == PyObject_SetItem(a, b, c))

return NULL;

Py_RETURN_NONE;

}

static PyObject *

_operator_delitem_impl(PyObject *module, PyObject *a, PyObject *b)

{

if (-1 == PyObject_DelItem(a, b))

return NULL;

Py_RETURN_NONE;

}

static PyObject *

_operator_eq_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_EQ);

}

static PyObject *

_operator_ne_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_NE);

}

static PyObject *

_operator_lt_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_LT);

}

static PyObject *

_operator_le_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_LE);

}

static PyObject *

_operator_gt_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_GT);

}

static PyObject *

_operator_ge_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyObject_RichCompare(a, b, Py_GE);

}

static PyObject *

_operator_pow_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_Power(a, b, Py_None);

}

static PyObject *

_operator_ipow_impl(PyObject *module, PyObject *a, PyObject *b)

{

return PyNumber_InPlacePower(a, b, Py_None);

}

static PyObject *

_operator_index(PyObject *module, PyObject *a)

{

return PyNumber_Index(a);

}

static PyObject *

_operator_is__impl(PyObject *module, PyObject *a, PyObject *b)

{

PyObject *result = Py_Is(a, b) ? Py_True : Py_False;

return Py_NewRef(result);

}

static PyObject *

_operator_is_not_impl(PyObject *module, PyObject *a, PyObject *b)

{

PyObject *result;

result = (a != b) ? Py_True : Py_False;

return Py_NewRef(result);

}

static PyObject *

_operator_is_none(PyObject *module, PyObject *a)

{

PyObject *result = Py_IsNone(a) ? Py_True : Py_False;

return Py_NewRef(result);

}

static PyObject *

_operator_is_not_none(PyObject *module, PyObject *a)

{

PyObject *result = Py_IsNone(a) ? Py_False : Py_True;

return Py_NewRef(result);

}

static int

_tscmp(const unsigned char *a, const unsigned char *b,

Py_ssize_t len_a, Py_ssize_t len_b)

{

/* The volatile type declarations make sure that the compiler has no

volatile Py_ssize_t length;

volatile const unsigned char *left;

volatile const unsigned char *right;

Py_ssize_t i;

volatile unsigned char result;

/* loop count depends on length of b */

length = len_b;

left = NULL;

right = b;

/* don't use else here to keep the amount of CPU instructions constant,

if (len_a == length) {

left = *((volatile const unsigned char**)&a);

result = 0;

}

if (len_a != length) {

left = b;

result = 1;

}

for (i=0; i < length; i++) {

result |= *left++ ^ *right++;

}

return (result == 0);

}

static Py_ssize_t

_operator_length_hint_impl(PyObject *module, PyObject *obj,

Py_ssize_t default_value)

{

return PyObject_LengthHint(obj, default_value);

}

static PyObject *

_operator__compare_digest_impl(PyObject *module, PyObject *a, PyObject *b)

{

int rc;

/* ASCII unicode string */

if(PyUnicode_Check(a) && PyUnicode_Check(b)) {

if (!PyUnicode_IS_ASCII(a) || !PyUnicode_IS_ASCII(b)) {

PyErr_SetString(PyExc_TypeError,

"comparing strings with non-ASCII characters is "

"not supported");

return NULL;

}

rc = _tscmp(PyUnicode_DATA(a),

PyUnicode_DATA(b),

PyUnicode_GET_LENGTH(a),

PyUnicode_GET_LENGTH(b));

}

/* fallback to buffer interface for bytes, bytearray and other */

else {

Py_buffer view_a;

Py_buffer view_b;

if (PyObject_CheckBuffer(a) == 0 && PyObject_CheckBuffer(b) == 0) {

PyErr_Format(PyExc_TypeError,

"unsupported operand types(s) or combination of types: "

"'%.100s' and '%.100s'",

Py_TYPE(a)->tp_name, Py_TYPE(b)->tp_name);

return NULL;

}

if (PyObject_GetBuffer(a, &view_a, PyBUF_SIMPLE) == -1) {

return NULL;

}

if (view_a.ndim > 1) {

PyErr_SetString(PyExc_BufferError,

"Buffer must be single dimension");

PyBuffer_Release(&view_a);

return NULL;

}

if (PyObject_GetBuffer(b, &view_b, PyBUF_SIMPLE) == -1) {

PyBuffer_Release(&view_a);

return NULL;

}

if (view_b.ndim > 1) {

PyErr_SetString(PyExc_BufferError,

"Buffer must be single dimension");

PyBuffer_Release(&view_a);

PyBuffer_Release(&view_b);

return NULL;

}

rc = _tscmp((const unsigned char*)view_a.buf,

(const unsigned char*)view_b.buf,

view_a.len,

view_b.len);

PyBuffer_Release(&view_a);

PyBuffer_Release(&view_b);

}

return PyBool_FromLong(rc);

}

PyDoc_STRVAR(_operator_call__doc__,

"Same as obj(*args, **kwargs).");

#define _OPERATOR_CALL_METHODDEF \

{"call", _PyCFunction_CAST(_operator_call), METH_FASTCALL | METH_KEYWORDS, _operator_call__doc__},

static PyObject *

_operator_call(PyObject *module, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)

{

if (!_PyArg_CheckPositional("call", nargs, 1, PY_SSIZE_T_MAX)) {

return NULL;

}

return PyObject_Vectorcall(

args[0],

&args[1], (PyVectorcall_NARGS(nargs) - 1) | PY_VECTORCALL_ARGUMENTS_OFFSET,

kwnames);

}

static struct PyMethodDef operator_methods[] = {

_OPERATOR_TRUTH_METHODDEF

_OPERATOR_CONTAINS_METHODDEF

_OPERATOR_INDEXOF_METHODDEF

_OPERATOR_COUNTOF_METHODDEF

_OPERATOR_IS__METHODDEF

_OPERATOR_IS_NOT_METHODDEF

_OPERATOR_IS_NONE_METHODDEF

_OPERATOR_IS_NOT_NONE_METHODDEF

_OPERATOR_INDEX_METHODDEF

_OPERATOR_ADD_METHODDEF

_OPERATOR_SUB_METHODDEF

_OPERATOR_MUL_METHODDEF

_OPERATOR_MATMUL_METHODDEF

_OPERATOR_FLOORDIV_METHODDEF

_OPERATOR_TRUEDIV_METHODDEF

_OPERATOR_MOD_METHODDEF

_OPERATOR_NEG_METHODDEF

_OPERATOR_POS_METHODDEF

_OPERATOR_ABS_METHODDEF

_OPERATOR_INV_METHODDEF

_OPERATOR_INVERT_METHODDEF

_OPERATOR_LSHIFT_METHODDEF

_OPERATOR_RSHIFT_METHODDEF

_OPERATOR_NOT__METHODDEF

_OPERATOR_AND__METHODDEF

_OPERATOR_XOR_METHODDEF

_OPERATOR_OR__METHODDEF

_OPERATOR_IADD_METHODDEF

_OPERATOR_ISUB_METHODDEF

_OPERATOR_IMUL_METHODDEF

_OPERATOR_IMATMUL_METHODDEF

_OPERATOR_IFLOORDIV_METHODDEF

_OPERATOR_ITRUEDIV_METHODDEF

_OPERATOR_IMOD_METHODDEF

_OPERATOR_ILSHIFT_METHODDEF

_OPERATOR_IRSHIFT_METHODDEF

_OPERATOR_IAND_METHODDEF

_OPERATOR_IXOR_METHODDEF

_OPERATOR_IOR_METHODDEF

_OPERATOR_CONCAT_METHODDEF

_OPERATOR_ICONCAT_METHODDEF

_OPERATOR_GETITEM_METHODDEF

_OPERATOR_SETITEM_METHODDEF

_OPERATOR_DELITEM_METHODDEF

_OPERATOR_POW_METHODDEF

_OPERATOR_IPOW_METHODDEF

_OPERATOR_EQ_METHODDEF

_OPERATOR_NE_METHODDEF

_OPERATOR_LT_METHODDEF

_OPERATOR_LE_METHODDEF

_OPERATOR_GT_METHODDEF

_OPERATOR_GE_METHODDEF

_OPERATOR__COMPARE_DIGEST_METHODDEF

_OPERATOR_LENGTH_HINT_METHODDEF

_OPERATOR_CALL_METHODDEF

{NULL, NULL} /* sentinel */

};