make deque methods faster

refer python/cpython@dd407d5006
This commit is contained in:
ahgamut 2021-11-05 02:33:33 +05:30
parent f49003241f
commit 4959ac1497

View file

@ -936,12 +936,18 @@ done:
} }
static PyObject * static PyObject *
deque_rotate(dequeobject *deque, PyObject *args) deque_rotate(dequeobject *deque, PyObject **args, Py_ssize_t nargs,
PyObject *kwnames)
{ {
Py_ssize_t n=1; Py_ssize_t n=1;
if (!PyArg_ParseTuple(args, "|n:rotate", &n)) if (!_PyArg_NoStackKeywords("rotate", kwnames)) {
return NULL; return NULL;
}
if (!_PyArg_ParseStack(args, nargs, "|n:rotate", &n)) {
return NULL;
}
if (!_deque_rotate(deque, n)) if (!_deque_rotate(deque, n))
Py_RETURN_NONE; Py_RETURN_NONE;
return NULL; return NULL;
@ -1070,7 +1076,8 @@ deque_len(dequeobject *deque)
} }
static PyObject * static PyObject *
deque_index(dequeobject *deque, PyObject *args) deque_index(dequeobject *deque, PyObject **args, Py_ssize_t nargs,
PyObject *kwnames)
{ {
Py_ssize_t i, n, start=0, stop=Py_SIZE(deque); Py_ssize_t i, n, start=0, stop=Py_SIZE(deque);
PyObject *v, *item; PyObject *v, *item;
@ -1079,10 +1086,15 @@ deque_index(dequeobject *deque, PyObject *args)
size_t start_state = deque->state; size_t start_state = deque->state;
int cmp; int cmp;
if (!PyArg_ParseTuple(args, "O|O&O&:index", &v, if (!_PyArg_NoStackKeywords("index", kwnames)) {
_PyEval_SliceIndexNotNone, &start,
_PyEval_SliceIndexNotNone, &stop))
return NULL; return NULL;
}
if (!_PyArg_ParseStack(args, nargs, "O|O&O&:index", &v,
_PyEval_SliceIndex, &start,
_PyEval_SliceIndex, &stop)) {
return NULL;
}
if (start < 0) { if (start < 0) {
start += Py_SIZE(deque); start += Py_SIZE(deque);
if (start < 0) if (start < 0)
@ -1145,15 +1157,21 @@ PyDoc_STRVAR(index_doc,
*/ */
static PyObject * static PyObject *
deque_insert(dequeobject *deque, PyObject *args) deque_insert(dequeobject *deque, PyObject **args, Py_ssize_t nargs,
PyObject *kwnames)
{ {
Py_ssize_t index; Py_ssize_t index;
Py_ssize_t n = Py_SIZE(deque); Py_ssize_t n = Py_SIZE(deque);
PyObject *value; PyObject *value;
PyObject *rv; PyObject *rv;
if (!PyArg_ParseTuple(args, "nO:insert", &index, &value)) if (!_PyArg_NoStackKeywords("insert", kwnames)) {
return NULL; return NULL;
}
if (!_PyArg_ParseStack(args, nargs, "nO:insert", &index, &value)) {
return NULL;
}
if (deque->maxlen == Py_SIZE(deque)) { if (deque->maxlen == Py_SIZE(deque)) {
PyErr_SetString(PyExc_IndexError, "deque already at its maximum size"); PyErr_SetString(PyExc_IndexError, "deque already at its maximum size");
return NULL; return NULL;
@ -1623,9 +1641,9 @@ static PyMethodDef deque_methods[] = {
{"extendleft", (PyCFunction)deque_extendleft, {"extendleft", (PyCFunction)deque_extendleft,
METH_O, extendleft_doc}, METH_O, extendleft_doc},
{"index", (PyCFunction)deque_index, {"index", (PyCFunction)deque_index,
METH_VARARGS, index_doc}, METH_FASTCALL, index_doc},
{"insert", (PyCFunction)deque_insert, {"insert", (PyCFunction)deque_insert,
METH_VARARGS, insert_doc}, METH_FASTCALL, insert_doc},
{"pop", (PyCFunction)deque_pop, {"pop", (PyCFunction)deque_pop,
METH_NOARGS, pop_doc}, METH_NOARGS, pop_doc},
{"popleft", (PyCFunction)deque_popleft, {"popleft", (PyCFunction)deque_popleft,
@ -1639,7 +1657,7 @@ static PyMethodDef deque_methods[] = {
{"reverse", (PyCFunction)deque_reverse, {"reverse", (PyCFunction)deque_reverse,
METH_NOARGS, reverse_doc}, METH_NOARGS, reverse_doc},
{"rotate", (PyCFunction)deque_rotate, {"rotate", (PyCFunction)deque_rotate,
METH_VARARGS, rotate_doc}, METH_FASTCALL, rotate_doc},
{"__sizeof__", (PyCFunction)deque_sizeof, {"__sizeof__", (PyCFunction)deque_sizeof,
METH_NOARGS, sizeof_doc}, METH_NOARGS, sizeof_doc},
{NULL, NULL} /* sentinel */ {NULL, NULL} /* sentinel */