Closes #15469: Correct __sizeof__ support for deque

This commit is contained in:
Jesus Cea 2012-08-03 14:49:42 +02:00
parent e9c5318967
commit 16e2fca47e
3 changed files with 39 additions and 1 deletions

View file

@ -932,6 +932,23 @@ deque_init(dequeobject *deque, PyObject *args, PyObject *kwdargs)
return 0;
}
static PyObject *
deque_sizeof(dequeobject *deque, void *unused)
{
Py_ssize_t res;
Py_ssize_t blocks;
res = sizeof(dequeobject);
blocks = (deque->leftindex + deque->len + BLOCKLEN - 1) / BLOCKLEN;
assert(deque->leftindex + deque->len - 1 ==
(blocks - 1) * BLOCKLEN + deque->rightindex);
res += blocks * sizeof(block);
return PyLong_FromSsize_t(res);
}
PyDoc_STRVAR(sizeof_doc,
"D.__sizeof__() -- size of D in memory, in bytes");
static PyObject *
deque_get_maxlen(dequeobject *deque)
{
@ -995,7 +1012,9 @@ static PyMethodDef deque_methods[] = {
{"reverse", (PyCFunction)deque_reverse,
METH_NOARGS, reverse_doc},
{"rotate", (PyCFunction)deque_rotate,
METH_VARARGS, rotate_doc},
METH_VARARGS, rotate_doc},
{"__sizeof__", (PyCFunction)deque_sizeof,
METH_NOARGS, sizeof_doc},
{NULL, NULL} /* sentinel */
};