Issue #15424: Add a __sizeof__ implementation for array objects.

Patch by Ludwig Hähne.
This commit is contained in:
Meador Inge 2012-08-10 22:05:45 -05:00
parent a939105a40
commit 2d639d5665
4 changed files with 32 additions and 0 deletions

View file

@ -1532,6 +1532,19 @@ array_reduce(arrayobject *array)
PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
static PyObject *
array_sizeof(arrayobject *self, PyObject *unused)
{
Py_ssize_t res;
res = sizeof(arrayobject) + self->allocated * self->ob_descr->itemsize;
return PyLong_FromSsize_t(res);
}
PyDoc_STRVAR(sizeof_doc,
"__sizeof__() -> int\n\
\n\
Size of the array in memory, in bytes.");
static PyObject *
array_get_typecode(arrayobject *a, void *closure)
{
@ -1606,6 +1619,8 @@ static PyMethodDef array_methods[] = {
#endif
{"write", (PyCFunction)array_tofile_as_write, METH_O,
tofile_doc},
{"__sizeof__", (PyCFunction)array_sizeof, METH_NOARGS,
sizeof_doc},
{NULL, NULL} /* sentinel */
};