+++ /dev/null
-/* Array object implementation */\r
-\r
-/* An array is a uniform list -- all items have the same type.\r
- The item type is restricted to simple C types like int or float */\r
-\r
-#define PY_SSIZE_T_CLEAN\r
-#include "Python.h"\r
-#include "structmember.h"\r
-\r
-#ifdef STDC_HEADERS\r
-#include <stddef.h>\r
-#else /* !STDC_HEADERS */\r
-#ifdef HAVE_SYS_TYPES_H\r
-#include <sys/types.h> /* For size_t */\r
-#endif /* HAVE_SYS_TYPES_H */\r
-#endif /* !STDC_HEADERS */\r
-\r
-struct arrayobject; /* Forward */\r
-\r
-/* All possible arraydescr values are defined in the vector "descriptors"\r
- * below. That's defined later because the appropriate get and set\r
- * functions aren't visible yet.\r
- */\r
-struct arraydescr {\r
- int typecode;\r
- int itemsize;\r
- PyObject * (*getitem)(struct arrayobject *, Py_ssize_t);\r
- int (*setitem)(struct arrayobject *, Py_ssize_t, PyObject *);\r
-};\r
-\r
-typedef struct arrayobject {\r
- PyObject_VAR_HEAD\r
- char *ob_item;\r
- Py_ssize_t allocated;\r
- struct arraydescr *ob_descr;\r
- PyObject *weakreflist; /* List of weak references */\r
-} arrayobject;\r
-\r
-static PyTypeObject Arraytype;\r
-\r
-#define array_Check(op) PyObject_TypeCheck(op, &Arraytype)\r
-#define array_CheckExact(op) (Py_TYPE(op) == &Arraytype)\r
-\r
-static int\r
-array_resize(arrayobject *self, Py_ssize_t newsize)\r
-{\r
- char *items;\r
- size_t _new_size;\r
-\r
- /* Bypass realloc() when a previous overallocation is large enough\r
- to accommodate the newsize. If the newsize is 16 smaller than the\r
- current size, then proceed with the realloc() to shrink the list.\r
- */\r
-\r
- if (self->allocated >= newsize &&\r
- Py_SIZE(self) < newsize + 16 &&\r
- self->ob_item != NULL) {\r
- Py_SIZE(self) = newsize;\r
- return 0;\r
- }\r
-\r
- /* This over-allocates proportional to the array size, making room\r
- * for additional growth. The over-allocation is mild, but is\r
- * enough to give linear-time amortized behavior over a long\r
- * sequence of appends() in the presence of a poorly-performing\r
- * system realloc().\r
- * The growth pattern is: 0, 4, 8, 16, 25, 34, 46, 56, 67, 79, ...\r
- * Note, the pattern starts out the same as for lists but then\r
- * grows at a smaller rate so that larger arrays only overallocate\r
- * by about 1/16th -- this is done because arrays are presumed to be more\r
- * memory critical.\r
- */\r
-\r
- _new_size = (newsize >> 4) + (Py_SIZE(self) < 8 ? 3 : 7) + newsize;\r
- items = self->ob_item;\r
- /* XXX The following multiplication and division does not optimize away\r
- like it does for lists since the size is not known at compile time */\r
- if (_new_size <= ((~(size_t)0) / self->ob_descr->itemsize))\r
- PyMem_RESIZE(items, char, (_new_size * self->ob_descr->itemsize));\r
- else\r
- items = NULL;\r
- if (items == NULL) {\r
- PyErr_NoMemory();\r
- return -1;\r
- }\r
- self->ob_item = items;\r
- Py_SIZE(self) = newsize;\r
- self->allocated = _new_size;\r
- return 0;\r
-}\r
-\r
-/****************************************************************************\r
-Get and Set functions for each type.\r
-A Get function takes an arrayobject* and an integer index, returning the\r
-array value at that index wrapped in an appropriate PyObject*.\r
-A Set function takes an arrayobject, integer index, and PyObject*; sets\r
-the array value at that index to the raw C data extracted from the PyObject*,\r
-and returns 0 if successful, else nonzero on failure (PyObject* not of an\r
-appropriate type or value).\r
-Note that the basic Get and Set functions do NOT check that the index is\r
-in bounds; that's the responsibility of the caller.\r
-****************************************************************************/\r
-\r
-static PyObject *\r
-c_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyString_FromStringAndSize(&((char *)ap->ob_item)[i], 1);\r
-}\r
-\r
-static int\r
-c_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- char x;\r
- if (!PyArg_Parse(v, "c;array item must be char", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((char *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-b_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- long x = ((char *)ap->ob_item)[i];\r
- if (x >= 128)\r
- x -= 256;\r
- return PyInt_FromLong(x);\r
-}\r
-\r
-static int\r
-b_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- short x;\r
- /* PyArg_Parse's 'b' formatter is for an unsigned char, therefore\r
- must use the next size up that is signed ('h') and manually do\r
- the overflow checking */\r
- if (!PyArg_Parse(v, "h;array item must be integer", &x))\r
- return -1;\r
- else if (x < -128) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "signed char is less than minimum");\r
- return -1;\r
- }\r
- else if (x > 127) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "signed char is greater than maximum");\r
- return -1;\r
- }\r
- if (i >= 0)\r
- ((char *)ap->ob_item)[i] = (char)x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-BB_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- long x = ((unsigned char *)ap->ob_item)[i];\r
- return PyInt_FromLong(x);\r
-}\r
-\r
-static int\r
-BB_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- unsigned char x;\r
- /* 'B' == unsigned char, maps to PyArg_Parse's 'b' formatter */\r
- if (!PyArg_Parse(v, "b;array item must be integer", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((char *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-#ifdef Py_USING_UNICODE\r
-static PyObject *\r
-u_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyUnicode_FromUnicode(&((Py_UNICODE *) ap->ob_item)[i], 1);\r
-}\r
-\r
-static int\r
-u_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- Py_UNICODE *p;\r
- Py_ssize_t len;\r
-\r
- if (!PyArg_Parse(v, "u#;array item must be unicode character", &p, &len))\r
- return -1;\r
- if (len != 1) {\r
- PyErr_SetString(PyExc_TypeError,\r
- "array item must be unicode character");\r
- return -1;\r
- }\r
- if (i >= 0)\r
- ((Py_UNICODE *)ap->ob_item)[i] = p[0];\r
- return 0;\r
-}\r
-#endif\r
-\r
-static PyObject *\r
-h_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyInt_FromLong((long) ((short *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-h_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- short x;\r
- /* 'h' == signed short, maps to PyArg_Parse's 'h' formatter */\r
- if (!PyArg_Parse(v, "h;array item must be integer", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((short *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-HH_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyInt_FromLong((long) ((unsigned short *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-HH_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- int x;\r
- /* PyArg_Parse's 'h' formatter is for a signed short, therefore\r
- must use the next size up and manually do the overflow checking */\r
- if (!PyArg_Parse(v, "i;array item must be integer", &x))\r
- return -1;\r
- else if (x < 0) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned short is less than minimum");\r
- return -1;\r
- }\r
- else if (x > USHRT_MAX) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned short is greater than maximum");\r
- return -1;\r
- }\r
- if (i >= 0)\r
- ((short *)ap->ob_item)[i] = (short)x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-i_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyInt_FromLong((long) ((int *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-i_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- int x;\r
- /* 'i' == signed int, maps to PyArg_Parse's 'i' formatter */\r
- if (!PyArg_Parse(v, "i;array item must be integer", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((int *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-II_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyLong_FromUnsignedLong(\r
- (unsigned long) ((unsigned int *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-II_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- unsigned long x;\r
- if (PyLong_Check(v)) {\r
- x = PyLong_AsUnsignedLong(v);\r
- if (x == (unsigned long) -1 && PyErr_Occurred())\r
- return -1;\r
- }\r
- else {\r
- long y;\r
- if (!PyArg_Parse(v, "l;array item must be integer", &y))\r
- return -1;\r
- if (y < 0) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned int is less than minimum");\r
- return -1;\r
- }\r
- x = (unsigned long)y;\r
-\r
- }\r
- if (x > UINT_MAX) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned int is greater than maximum");\r
- return -1;\r
- }\r
-\r
- if (i >= 0)\r
- ((unsigned int *)ap->ob_item)[i] = (unsigned int)x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-l_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyInt_FromLong(((long *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-l_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- long x;\r
- if (!PyArg_Parse(v, "l;array item must be integer", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((long *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-LL_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyLong_FromUnsignedLong(((unsigned long *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-LL_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- unsigned long x;\r
- if (PyLong_Check(v)) {\r
- x = PyLong_AsUnsignedLong(v);\r
- if (x == (unsigned long) -1 && PyErr_Occurred())\r
- return -1;\r
- }\r
- else {\r
- long y;\r
- if (!PyArg_Parse(v, "l;array item must be integer", &y))\r
- return -1;\r
- if (y < 0) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned long is less than minimum");\r
- return -1;\r
- }\r
- x = (unsigned long)y;\r
-\r
- }\r
- if (x > ULONG_MAX) {\r
- PyErr_SetString(PyExc_OverflowError,\r
- "unsigned long is greater than maximum");\r
- return -1;\r
- }\r
-\r
- if (i >= 0)\r
- ((unsigned long *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-f_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyFloat_FromDouble((double) ((float *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-f_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- float x;\r
- if (!PyArg_Parse(v, "f;array item must be float", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((float *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-static PyObject *\r
-d_getitem(arrayobject *ap, Py_ssize_t i)\r
-{\r
- return PyFloat_FromDouble(((double *)ap->ob_item)[i]);\r
-}\r
-\r
-static int\r
-d_setitem(arrayobject *ap, Py_ssize_t i, PyObject *v)\r
-{\r
- double x;\r
- if (!PyArg_Parse(v, "d;array item must be float", &x))\r
- return -1;\r
- if (i >= 0)\r
- ((double *)ap->ob_item)[i] = x;\r
- return 0;\r
-}\r
-\r
-/* Description of types */\r
-static struct arraydescr descriptors[] = {\r
- {'c', sizeof(char), c_getitem, c_setitem},\r
- {'b', sizeof(char), b_getitem, b_setitem},\r
- {'B', sizeof(char), BB_getitem, BB_setitem},\r
-#ifdef Py_USING_UNICODE\r
- {'u', sizeof(Py_UNICODE), u_getitem, u_setitem},\r
-#endif\r
- {'h', sizeof(short), h_getitem, h_setitem},\r
- {'H', sizeof(short), HH_getitem, HH_setitem},\r
- {'i', sizeof(int), i_getitem, i_setitem},\r
- {'I', sizeof(int), II_getitem, II_setitem},\r
- {'l', sizeof(long), l_getitem, l_setitem},\r
- {'L', sizeof(long), LL_getitem, LL_setitem},\r
- {'f', sizeof(float), f_getitem, f_setitem},\r
- {'d', sizeof(double), d_getitem, d_setitem},\r
- {'\0', 0, 0, 0} /* Sentinel */\r
-};\r
-\r
-/****************************************************************************\r
-Implementations of array object methods.\r
-****************************************************************************/\r
-\r
-static PyObject *\r
-newarrayobject(PyTypeObject *type, Py_ssize_t size, struct arraydescr *descr)\r
-{\r
- arrayobject *op;\r
- size_t nbytes;\r
-\r
- if (size < 0) {\r
- PyErr_BadInternalCall();\r
- return NULL;\r
- }\r
-\r
- nbytes = size * descr->itemsize;\r
- /* Check for overflow */\r
- if (nbytes / descr->itemsize != (size_t)size) {\r
- return PyErr_NoMemory();\r
- }\r
- op = (arrayobject *) type->tp_alloc(type, 0);\r
- if (op == NULL) {\r
- return NULL;\r
- }\r
- op->ob_descr = descr;\r
- op->allocated = size;\r
- op->weakreflist = NULL;\r
- Py_SIZE(op) = size;\r
- if (size <= 0) {\r
- op->ob_item = NULL;\r
- }\r
- else {\r
- op->ob_item = PyMem_NEW(char, nbytes);\r
- if (op->ob_item == NULL) {\r
- Py_DECREF(op);\r
- return PyErr_NoMemory();\r
- }\r
- }\r
- return (PyObject *) op;\r
-}\r
-\r
-static PyObject *\r
-getarrayitem(PyObject *op, Py_ssize_t i)\r
-{\r
- register arrayobject *ap;\r
- assert(array_Check(op));\r
- ap = (arrayobject *)op;\r
- assert(i>=0 && i<Py_SIZE(ap));\r
- return (*ap->ob_descr->getitem)(ap, i);\r
-}\r
-\r
-static int\r
-ins1(arrayobject *self, Py_ssize_t where, PyObject *v)\r
-{\r
- char *items;\r
- Py_ssize_t n = Py_SIZE(self);\r
- if (v == NULL) {\r
- PyErr_BadInternalCall();\r
- return -1;\r
- }\r
- if ((*self->ob_descr->setitem)(self, -1, v) < 0)\r
- return -1;\r
-\r
- if (array_resize(self, n+1) == -1)\r
- return -1;\r
- items = self->ob_item;\r
- if (where < 0) {\r
- where += n;\r
- if (where < 0)\r
- where = 0;\r
- }\r
- if (where > n)\r
- where = n;\r
- /* appends don't need to call memmove() */\r
- if (where != n)\r
- memmove(items + (where+1)*self->ob_descr->itemsize,\r
- items + where*self->ob_descr->itemsize,\r
- (n-where)*self->ob_descr->itemsize);\r
- return (*self->ob_descr->setitem)(self, where, v);\r
-}\r
-\r
-/* Methods */\r
-\r
-static void\r
-array_dealloc(arrayobject *op)\r
-{\r
- if (op->weakreflist != NULL)\r
- PyObject_ClearWeakRefs((PyObject *) op);\r
- if (op->ob_item != NULL)\r
- PyMem_DEL(op->ob_item);\r
- Py_TYPE(op)->tp_free((PyObject *)op);\r
-}\r
-\r
-static PyObject *\r
-array_richcompare(PyObject *v, PyObject *w, int op)\r
-{\r
- arrayobject *va, *wa;\r
- PyObject *vi = NULL;\r
- PyObject *wi = NULL;\r
- Py_ssize_t i, k;\r
- PyObject *res;\r
-\r
- if (!array_Check(v) || !array_Check(w)) {\r
- Py_INCREF(Py_NotImplemented);\r
- return Py_NotImplemented;\r
- }\r
-\r
- va = (arrayobject *)v;\r
- wa = (arrayobject *)w;\r
-\r
- if (Py_SIZE(va) != Py_SIZE(wa) && (op == Py_EQ || op == Py_NE)) {\r
- /* Shortcut: if the lengths differ, the arrays differ */\r
- if (op == Py_EQ)\r
- res = Py_False;\r
- else\r
- res = Py_True;\r
- Py_INCREF(res);\r
- return res;\r
- }\r
-\r
- /* Search for the first index where items are different */\r
- k = 1;\r
- for (i = 0; i < Py_SIZE(va) && i < Py_SIZE(wa); i++) {\r
- vi = getarrayitem(v, i);\r
- wi = getarrayitem(w, i);\r
- if (vi == NULL || wi == NULL) {\r
- Py_XDECREF(vi);\r
- Py_XDECREF(wi);\r
- return NULL;\r
- }\r
- k = PyObject_RichCompareBool(vi, wi, Py_EQ);\r
- if (k == 0)\r
- break; /* Keeping vi and wi alive! */\r
- Py_DECREF(vi);\r
- Py_DECREF(wi);\r
- if (k < 0)\r
- return NULL;\r
- }\r
-\r
- if (k) {\r
- /* No more items to compare -- compare sizes */\r
- Py_ssize_t vs = Py_SIZE(va);\r
- Py_ssize_t ws = Py_SIZE(wa);\r
- int cmp;\r
- switch (op) {\r
- case Py_LT: cmp = vs < ws; break;\r
- case Py_LE: cmp = vs <= ws; break;\r
- case Py_EQ: cmp = vs == ws; break;\r
- case Py_NE: cmp = vs != ws; break;\r
- case Py_GT: cmp = vs > ws; break;\r
- case Py_GE: cmp = vs >= ws; break;\r
- default: return NULL; /* cannot happen */\r
- }\r
- if (cmp)\r
- res = Py_True;\r
- else\r
- res = Py_False;\r
- Py_INCREF(res);\r
- return res;\r
- }\r
-\r
- /* We have an item that differs. First, shortcuts for EQ/NE */\r
- if (op == Py_EQ) {\r
- Py_INCREF(Py_False);\r
- res = Py_False;\r
- }\r
- else if (op == Py_NE) {\r
- Py_INCREF(Py_True);\r
- res = Py_True;\r
- }\r
- else {\r
- /* Compare the final item again using the proper operator */\r
- res = PyObject_RichCompare(vi, wi, op);\r
- }\r
- Py_DECREF(vi);\r
- Py_DECREF(wi);\r
- return res;\r
-}\r
-\r
-static Py_ssize_t\r
-array_length(arrayobject *a)\r
-{\r
- return Py_SIZE(a);\r
-}\r
-\r
-static PyObject *\r
-array_item(arrayobject *a, Py_ssize_t i)\r
-{\r
- if (i < 0 || i >= Py_SIZE(a)) {\r
- PyErr_SetString(PyExc_IndexError, "array index out of range");\r
- return NULL;\r
- }\r
- return getarrayitem((PyObject *)a, i);\r
-}\r
-\r
-static PyObject *\r
-array_slice(arrayobject *a, Py_ssize_t ilow, Py_ssize_t ihigh)\r
-{\r
- arrayobject *np;\r
- if (ilow < 0)\r
- ilow = 0;\r
- else if (ilow > Py_SIZE(a))\r
- ilow = Py_SIZE(a);\r
- if (ihigh < 0)\r
- ihigh = 0;\r
- if (ihigh < ilow)\r
- ihigh = ilow;\r
- else if (ihigh > Py_SIZE(a))\r
- ihigh = Py_SIZE(a);\r
- np = (arrayobject *) newarrayobject(&Arraytype, ihigh - ilow, a->ob_descr);\r
- if (np == NULL)\r
- return NULL;\r
- memcpy(np->ob_item, a->ob_item + ilow * a->ob_descr->itemsize,\r
- (ihigh-ilow) * a->ob_descr->itemsize);\r
- return (PyObject *)np;\r
-}\r
-\r
-static PyObject *\r
-array_copy(arrayobject *a, PyObject *unused)\r
-{\r
- return array_slice(a, 0, Py_SIZE(a));\r
-}\r
-\r
-PyDoc_STRVAR(copy_doc,\r
-"copy(array)\n\\r
-\n\\r
- Return a copy of the array.");\r
-\r
-static PyObject *\r
-array_concat(arrayobject *a, PyObject *bb)\r
-{\r
- Py_ssize_t size;\r
- arrayobject *np;\r
- if (!array_Check(bb)) {\r
- PyErr_Format(PyExc_TypeError,\r
- "can only append array (not \"%.200s\") to array",\r
- Py_TYPE(bb)->tp_name);\r
- return NULL;\r
- }\r
-#define b ((arrayobject *)bb)\r
- if (a->ob_descr != b->ob_descr) {\r
- PyErr_BadArgument();\r
- return NULL;\r
- }\r
- if (Py_SIZE(a) > PY_SSIZE_T_MAX - Py_SIZE(b)) {\r
- return PyErr_NoMemory();\r
- }\r
- size = Py_SIZE(a) + Py_SIZE(b);\r
- np = (arrayobject *) newarrayobject(&Arraytype, size, a->ob_descr);\r
- if (np == NULL) {\r
- return NULL;\r
- }\r
- memcpy(np->ob_item, a->ob_item, Py_SIZE(a)*a->ob_descr->itemsize);\r
- memcpy(np->ob_item + Py_SIZE(a)*a->ob_descr->itemsize,\r
- b->ob_item, Py_SIZE(b)*b->ob_descr->itemsize);\r
- return (PyObject *)np;\r
-#undef b\r
-}\r
-\r
-static PyObject *\r
-array_repeat(arrayobject *a, Py_ssize_t n)\r
-{\r
- Py_ssize_t i;\r
- Py_ssize_t size;\r
- arrayobject *np;\r
- char *p;\r
- Py_ssize_t nbytes;\r
- if (n < 0)\r
- n = 0;\r
- if ((Py_SIZE(a) != 0) && (n > PY_SSIZE_T_MAX / Py_SIZE(a))) {\r
- return PyErr_NoMemory();\r
- }\r
- size = Py_SIZE(a) * n;\r
- np = (arrayobject *) newarrayobject(&Arraytype, size, a->ob_descr);\r
- if (np == NULL)\r
- return NULL;\r
- p = np->ob_item;\r
- nbytes = Py_SIZE(a) * a->ob_descr->itemsize;\r
- for (i = 0; i < n; i++) {\r
- memcpy(p, a->ob_item, nbytes);\r
- p += nbytes;\r
- }\r
- return (PyObject *) np;\r
-}\r
-\r
-static int\r
-array_ass_slice(arrayobject *a, Py_ssize_t ilow, Py_ssize_t ihigh, PyObject *v)\r
-{\r
- char *item;\r
- Py_ssize_t n; /* Size of replacement array */\r
- Py_ssize_t d; /* Change in size */\r
-#define b ((arrayobject *)v)\r
- if (v == NULL)\r
- n = 0;\r
- else if (array_Check(v)) {\r
- n = Py_SIZE(b);\r
- if (a == b) {\r
- /* Special case "a[i:j] = a" -- copy b first */\r
- int ret;\r
- v = array_slice(b, 0, n);\r
- if (!v)\r
- return -1;\r
- ret = array_ass_slice(a, ilow, ihigh, v);\r
- Py_DECREF(v);\r
- return ret;\r
- }\r
- if (b->ob_descr != a->ob_descr) {\r
- PyErr_BadArgument();\r
- return -1;\r
- }\r
- }\r
- else {\r
- PyErr_Format(PyExc_TypeError,\r
- "can only assign array (not \"%.200s\") to array slice",\r
- Py_TYPE(v)->tp_name);\r
- return -1;\r
- }\r
- if (ilow < 0)\r
- ilow = 0;\r
- else if (ilow > Py_SIZE(a))\r
- ilow = Py_SIZE(a);\r
- if (ihigh < 0)\r
- ihigh = 0;\r
- if (ihigh < ilow)\r
- ihigh = ilow;\r
- else if (ihigh > Py_SIZE(a))\r
- ihigh = Py_SIZE(a);\r
- item = a->ob_item;\r
- d = n - (ihigh-ilow);\r
- if (d < 0) { /* Delete -d items */\r
- memmove(item + (ihigh+d)*a->ob_descr->itemsize,\r
- item + ihigh*a->ob_descr->itemsize,\r
- (Py_SIZE(a)-ihigh)*a->ob_descr->itemsize);\r
- Py_SIZE(a) += d;\r
- PyMem_RESIZE(item, char, Py_SIZE(a)*a->ob_descr->itemsize);\r
- /* Can't fail */\r
- a->ob_item = item;\r
- a->allocated = Py_SIZE(a);\r
- }\r
- else if (d > 0) { /* Insert d items */\r
- PyMem_RESIZE(item, char,\r
- (Py_SIZE(a) + d)*a->ob_descr->itemsize);\r
- if (item == NULL) {\r
- PyErr_NoMemory();\r
- return -1;\r
- }\r
- memmove(item + (ihigh+d)*a->ob_descr->itemsize,\r
- item + ihigh*a->ob_descr->itemsize,\r
- (Py_SIZE(a)-ihigh)*a->ob_descr->itemsize);\r
- a->ob_item = item;\r
- Py_SIZE(a) += d;\r
- a->allocated = Py_SIZE(a);\r
- }\r
- if (n > 0)\r
- memcpy(item + ilow*a->ob_descr->itemsize, b->ob_item,\r
- n*b->ob_descr->itemsize);\r
- return 0;\r
-#undef b\r
-}\r
-\r
-static int\r
-array_ass_item(arrayobject *a, Py_ssize_t i, PyObject *v)\r
-{\r
- if (i < 0 || i >= Py_SIZE(a)) {\r
- PyErr_SetString(PyExc_IndexError,\r
- "array assignment index out of range");\r
- return -1;\r
- }\r
- if (v == NULL)\r
- return array_ass_slice(a, i, i+1, v);\r
- return (*a->ob_descr->setitem)(a, i, v);\r
-}\r
-\r
-static int\r
-setarrayitem(PyObject *a, Py_ssize_t i, PyObject *v)\r
-{\r
- assert(array_Check(a));\r
- return array_ass_item((arrayobject *)a, i, v);\r
-}\r
-\r
-static int\r
-array_iter_extend(arrayobject *self, PyObject *bb)\r
-{\r
- PyObject *it, *v;\r
-\r
- it = PyObject_GetIter(bb);\r
- if (it == NULL)\r
- return -1;\r
-\r
- while ((v = PyIter_Next(it)) != NULL) {\r
- if (ins1(self, Py_SIZE(self), v) != 0) {\r
- Py_DECREF(v);\r
- Py_DECREF(it);\r
- return -1;\r
- }\r
- Py_DECREF(v);\r
- }\r
- Py_DECREF(it);\r
- if (PyErr_Occurred())\r
- return -1;\r
- return 0;\r
-}\r
-\r
-static int\r
-array_do_extend(arrayobject *self, PyObject *bb)\r
-{\r
- Py_ssize_t size;\r
- char *old_item;\r
-\r
- if (!array_Check(bb))\r
- return array_iter_extend(self, bb);\r
-#define b ((arrayobject *)bb)\r
- if (self->ob_descr != b->ob_descr) {\r
- PyErr_SetString(PyExc_TypeError,\r
- "can only extend with array of same kind");\r
- return -1;\r
- }\r
- if ((Py_SIZE(self) > PY_SSIZE_T_MAX - Py_SIZE(b)) ||\r
- ((Py_SIZE(self) + Py_SIZE(b)) > PY_SSIZE_T_MAX / self->ob_descr->itemsize)) {\r
- PyErr_NoMemory();\r
- return -1;\r
- }\r
- size = Py_SIZE(self) + Py_SIZE(b);\r
- old_item = self->ob_item;\r
- PyMem_RESIZE(self->ob_item, char, size*self->ob_descr->itemsize);\r
- if (self->ob_item == NULL) {\r
- self->ob_item = old_item;\r
- PyErr_NoMemory();\r
- return -1;\r
- }\r
- memcpy(self->ob_item + Py_SIZE(self)*self->ob_descr->itemsize,\r
- b->ob_item, Py_SIZE(b)*b->ob_descr->itemsize);\r
- Py_SIZE(self) = size;\r
- self->allocated = size;\r
-\r
- return 0;\r
-#undef b\r
-}\r
-\r
-static PyObject *\r
-array_inplace_concat(arrayobject *self, PyObject *bb)\r
-{\r
- if (!array_Check(bb)) {\r
- PyErr_Format(PyExc_TypeError,\r
- "can only extend array with array (not \"%.200s\")",\r
- Py_TYPE(bb)->tp_name);\r
- return NULL;\r
- }\r
- if (array_do_extend(self, bb) == -1)\r
- return NULL;\r
- Py_INCREF(self);\r
- return (PyObject *)self;\r
-}\r
-\r
-static PyObject *\r
-array_inplace_repeat(arrayobject *self, Py_ssize_t n)\r
-{\r
- char *items, *p;\r
- Py_ssize_t size, i;\r
-\r
- if (Py_SIZE(self) > 0) {\r
- if (n < 0)\r
- n = 0;\r
- items = self->ob_item;\r
- if ((self->ob_descr->itemsize != 0) &&\r
- (Py_SIZE(self) > PY_SSIZE_T_MAX / self->ob_descr->itemsize)) {\r
- return PyErr_NoMemory();\r
- }\r
- size = Py_SIZE(self) * self->ob_descr->itemsize;\r
- if (n == 0) {\r
- PyMem_FREE(items);\r
- self->ob_item = NULL;\r
- Py_SIZE(self) = 0;\r
- self->allocated = 0;\r
- }\r
- else {\r
- if (size > PY_SSIZE_T_MAX / n) {\r
- return PyErr_NoMemory();\r
- }\r
- PyMem_RESIZE(items, char, n * size);\r
- if (items == NULL)\r
- return PyErr_NoMemory();\r
- p = items;\r
- for (i = 1; i < n; i++) {\r
- p += size;\r
- memcpy(p, items, size);\r
- }\r
- self->ob_item = items;\r
- Py_SIZE(self) *= n;\r
- self->allocated = Py_SIZE(self);\r
- }\r
- }\r
- Py_INCREF(self);\r
- return (PyObject *)self;\r
-}\r
-\r
-\r
-static PyObject *\r
-ins(arrayobject *self, Py_ssize_t where, PyObject *v)\r
-{\r
- if (ins1(self, where, v) != 0)\r
- return NULL;\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-static PyObject *\r
-array_count(arrayobject *self, PyObject *v)\r
-{\r
- Py_ssize_t count = 0;\r
- Py_ssize_t i;\r
-\r
- for (i = 0; i < Py_SIZE(self); i++) {\r
- PyObject *selfi = getarrayitem((PyObject *)self, i);\r
- int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);\r
- Py_DECREF(selfi);\r
- if (cmp > 0)\r
- count++;\r
- else if (cmp < 0)\r
- return NULL;\r
- }\r
- return PyInt_FromSsize_t(count);\r
-}\r
-\r
-PyDoc_STRVAR(count_doc,\r
-"count(x)\n\\r
-\n\\r
-Return number of occurrences of x in the array.");\r
-\r
-static PyObject *\r
-array_index(arrayobject *self, PyObject *v)\r
-{\r
- Py_ssize_t i;\r
-\r
- for (i = 0; i < Py_SIZE(self); i++) {\r
- PyObject *selfi = getarrayitem((PyObject *)self, i);\r
- int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);\r
- Py_DECREF(selfi);\r
- if (cmp > 0) {\r
- return PyInt_FromLong((long)i);\r
- }\r
- else if (cmp < 0)\r
- return NULL;\r
- }\r
- PyErr_SetString(PyExc_ValueError, "array.index(x): x not in list");\r
- return NULL;\r
-}\r
-\r
-PyDoc_STRVAR(index_doc,\r
-"index(x)\n\\r
-\n\\r
-Return index of first occurrence of x in the array.");\r
-\r
-static int\r
-array_contains(arrayobject *self, PyObject *v)\r
-{\r
- Py_ssize_t i;\r
- int cmp;\r
-\r
- for (i = 0, cmp = 0 ; cmp == 0 && i < Py_SIZE(self); i++) {\r
- PyObject *selfi = getarrayitem((PyObject *)self, i);\r
- cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);\r
- Py_DECREF(selfi);\r
- }\r
- return cmp;\r
-}\r
-\r
-static PyObject *\r
-array_remove(arrayobject *self, PyObject *v)\r
-{\r
- int i;\r
-\r
- for (i = 0; i < Py_SIZE(self); i++) {\r
- PyObject *selfi = getarrayitem((PyObject *)self,i);\r
- int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);\r
- Py_DECREF(selfi);\r
- if (cmp > 0) {\r
- if (array_ass_slice(self, i, i+1,\r
- (PyObject *)NULL) != 0)\r
- return NULL;\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
- }\r
- else if (cmp < 0)\r
- return NULL;\r
- }\r
- PyErr_SetString(PyExc_ValueError, "array.remove(x): x not in list");\r
- return NULL;\r
-}\r
-\r
-PyDoc_STRVAR(remove_doc,\r
-"remove(x)\n\\r
-\n\\r
-Remove the first occurrence of x in the array.");\r
-\r
-static PyObject *\r
-array_pop(arrayobject *self, PyObject *args)\r
-{\r
- Py_ssize_t i = -1;\r
- PyObject *v;\r
- if (!PyArg_ParseTuple(args, "|n:pop", &i))\r
- return NULL;\r
- if (Py_SIZE(self) == 0) {\r
- /* Special-case most common failure cause */\r
- PyErr_SetString(PyExc_IndexError, "pop from empty array");\r
- return NULL;\r
- }\r
- if (i < 0)\r
- i += Py_SIZE(self);\r
- if (i < 0 || i >= Py_SIZE(self)) {\r
- PyErr_SetString(PyExc_IndexError, "pop index out of range");\r
- return NULL;\r
- }\r
- v = getarrayitem((PyObject *)self,i);\r
- if (array_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {\r
- Py_DECREF(v);\r
- return NULL;\r
- }\r
- return v;\r
-}\r
-\r
-PyDoc_STRVAR(pop_doc,\r
-"pop([i])\n\\r
-\n\\r
-Return the i-th element and delete it from the array. i defaults to -1.");\r
-\r
-static PyObject *\r
-array_extend(arrayobject *self, PyObject *bb)\r
-{\r
- if (array_do_extend(self, bb) == -1)\r
- return NULL;\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(extend_doc,\r
-"extend(array or iterable)\n\\r
-\n\\r
- Append items to the end of the array.");\r
-\r
-static PyObject *\r
-array_insert(arrayobject *self, PyObject *args)\r
-{\r
- Py_ssize_t i;\r
- PyObject *v;\r
- if (!PyArg_ParseTuple(args, "nO:insert", &i, &v))\r
- return NULL;\r
- return ins(self, i, v);\r
-}\r
-\r
-PyDoc_STRVAR(insert_doc,\r
-"insert(i,x)\n\\r
-\n\\r
-Insert a new item x into the array before position i.");\r
-\r
-\r
-static PyObject *\r
-array_buffer_info(arrayobject *self, PyObject *unused)\r
-{\r
- PyObject* retval = NULL;\r
- retval = PyTuple_New(2);\r
- if (!retval)\r
- return NULL;\r
-\r
- PyTuple_SET_ITEM(retval, 0, PyLong_FromVoidPtr(self->ob_item));\r
- PyTuple_SET_ITEM(retval, 1, PyInt_FromLong((long)(Py_SIZE(self))));\r
-\r
- return retval;\r
-}\r
-\r
-PyDoc_STRVAR(buffer_info_doc,\r
-"buffer_info() -> (address, length)\n\\r
-\n\\r
-Return a tuple (address, length) giving the current memory address and\n\\r
-the length in items of the buffer used to hold array's contents\n\\r
-The length should be multiplied by the itemsize attribute to calculate\n\\r
-the buffer length in bytes.");\r
-\r
-\r
-static PyObject *\r
-array_append(arrayobject *self, PyObject *v)\r
-{\r
- return ins(self, Py_SIZE(self), v);\r
-}\r
-\r
-PyDoc_STRVAR(append_doc,\r
-"append(x)\n\\r
-\n\\r
-Append new value x to the end of the array.");\r
-\r
-\r
-static PyObject *\r
-array_byteswap(arrayobject *self, PyObject *unused)\r
-{\r
- char *p;\r
- Py_ssize_t i;\r
-\r
- switch (self->ob_descr->itemsize) {\r
- case 1:\r
- break;\r
- case 2:\r
- for (p = self->ob_item, i = Py_SIZE(self); --i >= 0; p += 2) {\r
- char p0 = p[0];\r
- p[0] = p[1];\r
- p[1] = p0;\r
- }\r
- break;\r
- case 4:\r
- for (p = self->ob_item, i = Py_SIZE(self); --i >= 0; p += 4) {\r
- char p0 = p[0];\r
- char p1 = p[1];\r
- p[0] = p[3];\r
- p[1] = p[2];\r
- p[2] = p1;\r
- p[3] = p0;\r
- }\r
- break;\r
- case 8:\r
- for (p = self->ob_item, i = Py_SIZE(self); --i >= 0; p += 8) {\r
- char p0 = p[0];\r
- char p1 = p[1];\r
- char p2 = p[2];\r
- char p3 = p[3];\r
- p[0] = p[7];\r
- p[1] = p[6];\r
- p[2] = p[5];\r
- p[3] = p[4];\r
- p[4] = p3;\r
- p[5] = p2;\r
- p[6] = p1;\r
- p[7] = p0;\r
- }\r
- break;\r
- default:\r
- PyErr_SetString(PyExc_RuntimeError,\r
- "don't know how to byteswap this array type");\r
- return NULL;\r
- }\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(byteswap_doc,\r
-"byteswap()\n\\r
-\n\\r
-Byteswap all items of the array. If the items in the array are not 1, 2,\n\\r
-4, or 8 bytes in size, RuntimeError is raised.");\r
-\r
-static PyObject *\r
-array_reverse(arrayobject *self, PyObject *unused)\r
-{\r
- register Py_ssize_t itemsize = self->ob_descr->itemsize;\r
- register char *p, *q;\r
- /* little buffer to hold items while swapping */\r
- char tmp[256]; /* 8 is probably enough -- but why skimp */\r
- assert((size_t)itemsize <= sizeof(tmp));\r
-\r
- if (Py_SIZE(self) > 1) {\r
- for (p = self->ob_item,\r
- q = self->ob_item + (Py_SIZE(self) - 1)*itemsize;\r
- p < q;\r
- p += itemsize, q -= itemsize) {\r
- /* memory areas guaranteed disjoint, so memcpy\r
- * is safe (& memmove may be slower).\r
- */\r
- memcpy(tmp, p, itemsize);\r
- memcpy(p, q, itemsize);\r
- memcpy(q, tmp, itemsize);\r
- }\r
- }\r
-\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(reverse_doc,\r
-"reverse()\n\\r
-\n\\r
-Reverse the order of the items in the array.");\r
-\r
-static PyObject *\r
-array_fromfile(arrayobject *self, PyObject *args)\r
-{\r
- PyObject *f;\r
- Py_ssize_t n;\r
- FILE *fp;\r
- if (!PyArg_ParseTuple(args, "On:fromfile", &f, &n))\r
- return NULL;\r
- fp = PyFile_AsFile(f);\r
- if (fp == NULL) {\r
- PyErr_SetString(PyExc_TypeError, "arg1 must be open file");\r
- return NULL;\r
- }\r
- if (n > 0) {\r
- char *item = self->ob_item;\r
- Py_ssize_t itemsize = self->ob_descr->itemsize;\r
- size_t nread;\r
- Py_ssize_t newlength;\r
- size_t newbytes;\r
- /* Be careful here about overflow */\r
- if ((newlength = Py_SIZE(self) + n) <= 0 ||\r
- (newbytes = newlength * itemsize) / itemsize !=\r
- (size_t)newlength)\r
- goto nomem;\r
- PyMem_RESIZE(item, char, newbytes);\r
- if (item == NULL) {\r
- nomem:\r
- PyErr_NoMemory();\r
- return NULL;\r
- }\r
- self->ob_item = item;\r
- Py_SIZE(self) += n;\r
- self->allocated = Py_SIZE(self);\r
- nread = fread(item + (Py_SIZE(self) - n) * itemsize,\r
- itemsize, n, fp);\r
- if (nread < (size_t)n) {\r
- Py_SIZE(self) -= (n - nread);\r
- PyMem_RESIZE(item, char, Py_SIZE(self)*itemsize);\r
- self->ob_item = item;\r
- self->allocated = Py_SIZE(self);\r
- if (ferror(fp)) {\r
- PyErr_SetFromErrno(PyExc_IOError);\r
- clearerr(fp);\r
- }\r
- else {\r
- PyErr_SetString(PyExc_EOFError,\r
- "not enough items in file");\r
- }\r
- return NULL;\r
- }\r
- }\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(fromfile_doc,\r
-"fromfile(f, n)\n\\r
-\n\\r
-Read n objects from the file object f and append them to the end of the\n\\r
-array. Also called as read.");\r
-\r
-\r
-static PyObject *\r
-array_fromfile_as_read(arrayobject *self, PyObject *args)\r
-{\r
- if (PyErr_WarnPy3k("array.read() not supported in 3.x; "\r
- "use array.fromfile()", 1) < 0)\r
- return NULL;\r
- return array_fromfile(self, args);\r
-}\r
-\r
-\r
-static PyObject *\r
-array_tofile(arrayobject *self, PyObject *f)\r
-{\r
- FILE *fp;\r
-\r
- fp = PyFile_AsFile(f);\r
- if (fp == NULL) {\r
- PyErr_SetString(PyExc_TypeError, "arg must be open file");\r
- return NULL;\r
- }\r
- if (self->ob_size > 0) {\r
- if (fwrite(self->ob_item, self->ob_descr->itemsize,\r
- self->ob_size, fp) != (size_t)self->ob_size) {\r
- PyErr_SetFromErrno(PyExc_IOError);\r
- clearerr(fp);\r
- return NULL;\r
- }\r
- }\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(tofile_doc,\r
-"tofile(f)\n\\r
-\n\\r
-Write all items (as machine values) to the file object f. Also called as\n\\r
-write.");\r
-\r
-\r
-static PyObject *\r
-array_tofile_as_write(arrayobject *self, PyObject *f)\r
-{\r
- if (PyErr_WarnPy3k("array.write() not supported in 3.x; "\r
- "use array.tofile()", 1) < 0)\r
- return NULL;\r
- return array_tofile(self, f);\r
-}\r
-\r
-\r
-static PyObject *\r
-array_fromlist(arrayobject *self, PyObject *list)\r
-{\r
- Py_ssize_t n;\r
- Py_ssize_t itemsize = self->ob_descr->itemsize;\r
-\r
- if (!PyList_Check(list)) {\r
- PyErr_SetString(PyExc_TypeError, "arg must be list");\r
- return NULL;\r
- }\r
- n = PyList_Size(list);\r
- if (n > 0) {\r
- char *item = self->ob_item;\r
- Py_ssize_t i;\r
- PyMem_RESIZE(item, char, (Py_SIZE(self) + n) * itemsize);\r
- if (item == NULL) {\r
- PyErr_NoMemory();\r
- return NULL;\r
- }\r
- self->ob_item = item;\r
- Py_SIZE(self) += n;\r
- self->allocated = Py_SIZE(self);\r
- for (i = 0; i < n; i++) {\r
- PyObject *v = PyList_GetItem(list, i);\r
- if ((*self->ob_descr->setitem)(self,\r
- Py_SIZE(self) - n + i, v) != 0) {\r
- Py_SIZE(self) -= n;\r
- if (itemsize && (self->ob_size > PY_SSIZE_T_MAX / itemsize)) {\r
- return PyErr_NoMemory();\r
- }\r
- PyMem_RESIZE(item, char,\r
- Py_SIZE(self) * itemsize);\r
- self->ob_item = item;\r
- self->allocated = Py_SIZE(self);\r
- return NULL;\r
- }\r
- }\r
- }\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(fromlist_doc,\r
-"fromlist(list)\n\\r
-\n\\r
-Append items to array from list.");\r
-\r
-\r
-static PyObject *\r
-array_tolist(arrayobject *self, PyObject *unused)\r
-{\r
- PyObject *list = PyList_New(Py_SIZE(self));\r
- Py_ssize_t i;\r
-\r
- if (list == NULL)\r
- return NULL;\r
- for (i = 0; i < Py_SIZE(self); i++) {\r
- PyObject *v = getarrayitem((PyObject *)self, i);\r
- if (v == NULL) {\r
- Py_DECREF(list);\r
- return NULL;\r
- }\r
- PyList_SetItem(list, i, v);\r
- }\r
- return list;\r
-}\r
-\r
-PyDoc_STRVAR(tolist_doc,\r
-"tolist() -> list\n\\r
-\n\\r
-Convert array to an ordinary list with the same items.");\r
-\r
-\r
-static PyObject *\r
-array_fromstring(arrayobject *self, PyObject *args)\r
-{\r
- char *str;\r
- Py_ssize_t n;\r
- int itemsize = self->ob_descr->itemsize;\r
- if (!PyArg_ParseTuple(args, "s#:fromstring", &str, &n))\r
- return NULL;\r
- if (n % itemsize != 0) {\r
- PyErr_SetString(PyExc_ValueError,\r
- "string length not a multiple of item size");\r
- return NULL;\r
- }\r
- n = n / itemsize;\r
- if (n > 0) {\r
- char *item = self->ob_item;\r
- if ((n > PY_SSIZE_T_MAX - Py_SIZE(self)) ||\r
- ((Py_SIZE(self) + n) > PY_SSIZE_T_MAX / itemsize)) {\r
- return PyErr_NoMemory();\r
- }\r
- PyMem_RESIZE(item, char, (Py_SIZE(self) + n) * itemsize);\r
- if (item == NULL) {\r
- PyErr_NoMemory();\r
- return NULL;\r
- }\r
- self->ob_item = item;\r
- Py_SIZE(self) += n;\r
- self->allocated = Py_SIZE(self);\r
- memcpy(item + (Py_SIZE(self) - n) * itemsize,\r
- str, itemsize*n);\r
- }\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(fromstring_doc,\r
-"fromstring(string)\n\\r
-\n\\r
-Appends items from the string, interpreting it as an array of machine\n\\r
-values,as if it had been read from a file using the fromfile() method).");\r
-\r
-\r
-static PyObject *\r
-array_tostring(arrayobject *self, PyObject *unused)\r
-{\r
- if (self->ob_size <= PY_SSIZE_T_MAX / self->ob_descr->itemsize) {\r
- return PyString_FromStringAndSize(self->ob_item,\r
- Py_SIZE(self) * self->ob_descr->itemsize);\r
- } else {\r
- return PyErr_NoMemory();\r
- }\r
-}\r
-\r
-PyDoc_STRVAR(tostring_doc,\r
-"tostring() -> string\n\\r
-\n\\r
-Convert the array to an array of machine values and return the string\n\\r
-representation.");\r
-\r
-\r
-\r
-#ifdef Py_USING_UNICODE\r
-static PyObject *\r
-array_fromunicode(arrayobject *self, PyObject *args)\r
-{\r
- Py_UNICODE *ustr;\r
- Py_ssize_t n;\r
-\r
- if (!PyArg_ParseTuple(args, "u#:fromunicode", &ustr, &n))\r
- return NULL;\r
- if (self->ob_descr->typecode != 'u') {\r
- PyErr_SetString(PyExc_ValueError,\r
- "fromunicode() may only be called on "\r
- "type 'u' arrays");\r
- return NULL;\r
- }\r
- if (n > 0) {\r
- Py_UNICODE *item = (Py_UNICODE *) self->ob_item;\r
- if (Py_SIZE(self) > PY_SSIZE_T_MAX - n) {\r
- return PyErr_NoMemory();\r
- }\r
- PyMem_RESIZE(item, Py_UNICODE, Py_SIZE(self) + n);\r
- if (item == NULL) {\r
- PyErr_NoMemory();\r
- return NULL;\r
- }\r
- self->ob_item = (char *) item;\r
- Py_SIZE(self) += n;\r
- self->allocated = Py_SIZE(self);\r
- memcpy(item + Py_SIZE(self) - n,\r
- ustr, n * sizeof(Py_UNICODE));\r
- }\r
-\r
- Py_INCREF(Py_None);\r
- return Py_None;\r
-}\r
-\r
-PyDoc_STRVAR(fromunicode_doc,\r
-"fromunicode(ustr)\n\\r
-\n\\r
-Extends this array with data from the unicode string ustr.\n\\r
-The array must be a type 'u' array; otherwise a ValueError\n\\r
-is raised. Use array.fromstring(ustr.decode(...)) to\n\\r
-append Unicode data to an array of some other type.");\r
-\r
-\r
-static PyObject *\r
-array_tounicode(arrayobject *self, PyObject *unused)\r
-{\r
- if (self->ob_descr->typecode != 'u') {\r
- PyErr_SetString(PyExc_ValueError,\r
- "tounicode() may only be called on type 'u' arrays");\r
- return NULL;\r
- }\r
- return PyUnicode_FromUnicode((Py_UNICODE *) self->ob_item, Py_SIZE(self));\r
-}\r
-\r
-PyDoc_STRVAR(tounicode_doc,\r
-"tounicode() -> unicode\n\\r
-\n\\r
-Convert the array to a unicode string. The array must be\n\\r
-a type 'u' array; otherwise a ValueError is raised. Use\n\\r
-array.tostring().decode() to obtain a unicode string from\n\\r
-an array of some other type.");\r
-\r
-#endif /* Py_USING_UNICODE */\r
-\r
-static PyObject *\r
-array_reduce(arrayobject *array)\r
-{\r
- PyObject *dict, *result, *list;\r
-\r
- dict = PyObject_GetAttrString((PyObject *)array, "__dict__");\r
- if (dict == NULL) {\r
- if (!PyErr_ExceptionMatches(PyExc_AttributeError))\r
- return NULL;\r
- PyErr_Clear();\r
- dict = Py_None;\r
- Py_INCREF(dict);\r
- }\r
- /* Unlike in Python 3.x, we never use the more efficient memory\r
- * representation of an array for pickling. This is unfortunately\r
- * necessary to allow array objects to be unpickled by Python 3.x,\r
- * since str objects from 2.x are always decoded to unicode in\r
- * Python 3.x.\r
- */\r
- list = array_tolist(array, NULL);\r
- if (list == NULL) {\r
- Py_DECREF(dict);\r
- return NULL;\r
- }\r
- result = Py_BuildValue(\r
- "O(cO)O", Py_TYPE(array), array->ob_descr->typecode, list, dict);\r
- Py_DECREF(list);\r
- Py_DECREF(dict);\r
- return result;\r
-}\r
-\r
-PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");\r
-\r
-static PyObject *\r
-array_sizeof(arrayobject *self, PyObject *unused)\r
-{\r
- Py_ssize_t res;\r
- res = sizeof(arrayobject) + self->allocated * self->ob_descr->itemsize;\r
- return PyLong_FromSsize_t(res);\r
-}\r
-\r
-PyDoc_STRVAR(sizeof_doc,\r
-"__sizeof__() -> int\n\\r
-\n\\r
-Size of the array in memory, in bytes.");\r
-\r
-static PyObject *\r
-array_get_typecode(arrayobject *a, void *closure)\r
-{\r
- char tc = a->ob_descr->typecode;\r
- return PyString_FromStringAndSize(&tc, 1);\r
-}\r
-\r
-static PyObject *\r
-array_get_itemsize(arrayobject *a, void *closure)\r
-{\r
- return PyInt_FromLong((long)a->ob_descr->itemsize);\r
-}\r
-\r
-static PyGetSetDef array_getsets [] = {\r
- {"typecode", (getter) array_get_typecode, NULL,\r
- "the typecode character used to create the array"},\r
- {"itemsize", (getter) array_get_itemsize, NULL,\r
- "the size, in bytes, of one array item"},\r
- {NULL}\r
-};\r
-\r
-static PyMethodDef array_methods[] = {\r
- {"append", (PyCFunction)array_append, METH_O,\r
- append_doc},\r
- {"buffer_info", (PyCFunction)array_buffer_info, METH_NOARGS,\r
- buffer_info_doc},\r
- {"byteswap", (PyCFunction)array_byteswap, METH_NOARGS,\r
- byteswap_doc},\r
- {"__copy__", (PyCFunction)array_copy, METH_NOARGS,\r
- copy_doc},\r
- {"count", (PyCFunction)array_count, METH_O,\r
- count_doc},\r
- {"__deepcopy__",(PyCFunction)array_copy, METH_O,\r
- copy_doc},\r
- {"extend", (PyCFunction)array_extend, METH_O,\r
- extend_doc},\r
- {"fromfile", (PyCFunction)array_fromfile, METH_VARARGS,\r
- fromfile_doc},\r
- {"fromlist", (PyCFunction)array_fromlist, METH_O,\r
- fromlist_doc},\r
- {"fromstring", (PyCFunction)array_fromstring, METH_VARARGS,\r
- fromstring_doc},\r
-#ifdef Py_USING_UNICODE\r
- {"fromunicode", (PyCFunction)array_fromunicode, METH_VARARGS,\r
- fromunicode_doc},\r
-#endif\r
- {"index", (PyCFunction)array_index, METH_O,\r
- index_doc},\r
- {"insert", (PyCFunction)array_insert, METH_VARARGS,\r
- insert_doc},\r
- {"pop", (PyCFunction)array_pop, METH_VARARGS,\r
- pop_doc},\r
- {"read", (PyCFunction)array_fromfile_as_read, METH_VARARGS,\r
- fromfile_doc},\r
- {"__reduce__", (PyCFunction)array_reduce, METH_NOARGS,\r
- reduce_doc},\r
- {"remove", (PyCFunction)array_remove, METH_O,\r
- remove_doc},\r
- {"reverse", (PyCFunction)array_reverse, METH_NOARGS,\r
- reverse_doc},\r
-/* {"sort", (PyCFunction)array_sort, METH_VARARGS,\r
- sort_doc},*/\r
- {"tofile", (PyCFunction)array_tofile, METH_O,\r
- tofile_doc},\r
- {"tolist", (PyCFunction)array_tolist, METH_NOARGS,\r
- tolist_doc},\r
- {"tostring", (PyCFunction)array_tostring, METH_NOARGS,\r
- tostring_doc},\r
-#ifdef Py_USING_UNICODE\r
- {"tounicode", (PyCFunction)array_tounicode, METH_NOARGS,\r
- tounicode_doc},\r
-#endif\r
- {"write", (PyCFunction)array_tofile_as_write, METH_O,\r
- tofile_doc},\r
- {"__sizeof__", (PyCFunction)array_sizeof, METH_NOARGS,\r
- sizeof_doc},\r
- {NULL, NULL} /* sentinel */\r
-};\r
-\r
-static PyObject *\r
-array_repr(arrayobject *a)\r
-{\r
- char buf[256], typecode;\r
- PyObject *s, *t, *v = NULL;\r
- Py_ssize_t len;\r
-\r
- len = Py_SIZE(a);\r
- typecode = a->ob_descr->typecode;\r
- if (len == 0) {\r
- PyOS_snprintf(buf, sizeof(buf), "array('%c')", typecode);\r
- return PyString_FromString(buf);\r
- }\r
-\r
- if (typecode == 'c')\r
- v = array_tostring(a, NULL);\r
-#ifdef Py_USING_UNICODE\r
- else if (typecode == 'u')\r
- v = array_tounicode(a, NULL);\r
-#endif\r
- else\r
- v = array_tolist(a, NULL);\r
- t = PyObject_Repr(v);\r
- Py_XDECREF(v);\r
-\r
- PyOS_snprintf(buf, sizeof(buf), "array('%c', ", typecode);\r
- s = PyString_FromString(buf);\r
- PyString_ConcatAndDel(&s, t);\r
- PyString_ConcatAndDel(&s, PyString_FromString(")"));\r
- return s;\r
-}\r
-\r
-static PyObject*\r
-array_subscr(arrayobject* self, PyObject* item)\r
-{\r
- if (PyIndex_Check(item)) {\r
- Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);\r
- if (i==-1 && PyErr_Occurred()) {\r
- return NULL;\r
- }\r
- if (i < 0)\r
- i += Py_SIZE(self);\r
- return array_item(self, i);\r
- }\r
- else if (PySlice_Check(item)) {\r
- Py_ssize_t start, stop, step, slicelength, cur, i;\r
- PyObject* result;\r
- arrayobject* ar;\r
- int itemsize = self->ob_descr->itemsize;\r
-\r
- if (PySlice_GetIndicesEx((PySliceObject*)item, Py_SIZE(self),\r
- &start, &stop, &step, &slicelength) < 0) {\r
- return NULL;\r
- }\r
-\r
- if (slicelength <= 0) {\r
- return newarrayobject(&Arraytype, 0, self->ob_descr);\r
- }\r
- else if (step == 1) {\r
- PyObject *result = newarrayobject(&Arraytype,\r
- slicelength, self->ob_descr);\r
- if (result == NULL)\r
- return NULL;\r
- memcpy(((arrayobject *)result)->ob_item,\r
- self->ob_item + start * itemsize,\r
- slicelength * itemsize);\r
- return result;\r
- }\r
- else {\r
- result = newarrayobject(&Arraytype, slicelength, self->ob_descr);\r
- if (!result) return NULL;\r
-\r
- ar = (arrayobject*)result;\r
-\r
- for (cur = start, i = 0; i < slicelength;\r
- cur += step, i++) {\r
- memcpy(ar->ob_item + i*itemsize,\r
- self->ob_item + cur*itemsize,\r
- itemsize);\r
- }\r
-\r
- return result;\r
- }\r
- }\r
- else {\r
- PyErr_SetString(PyExc_TypeError,\r
- "array indices must be integers");\r
- return NULL;\r
- }\r
-}\r
-\r
-static int\r
-array_ass_subscr(arrayobject* self, PyObject* item, PyObject* value)\r
-{\r
- Py_ssize_t start, stop, step, slicelength, needed;\r
- arrayobject* other;\r
- int itemsize;\r
-\r
- if (PyIndex_Check(item)) {\r
- Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);\r
-\r
- if (i == -1 && PyErr_Occurred())\r
- return -1;\r
- if (i < 0)\r
- i += Py_SIZE(self);\r
- if (i < 0 || i >= Py_SIZE(self)) {\r
- PyErr_SetString(PyExc_IndexError,\r
- "array assignment index out of range");\r
- return -1;\r
- }\r
- if (value == NULL) {\r
- /* Fall through to slice assignment */\r
- start = i;\r
- stop = i + 1;\r
- step = 1;\r
- slicelength = 1;\r
- }\r
- else\r
- return (*self->ob_descr->setitem)(self, i, value);\r
- }\r
- else if (PySlice_Check(item)) {\r
- if (PySlice_GetIndicesEx((PySliceObject *)item,\r
- Py_SIZE(self), &start, &stop,\r
- &step, &slicelength) < 0) {\r
- return -1;\r
- }\r
- }\r
- else {\r
- PyErr_SetString(PyExc_TypeError,\r
- "array indices must be integer");\r
- return -1;\r
- }\r
- if (value == NULL) {\r
- other = NULL;\r
- needed = 0;\r
- }\r
- else if (array_Check(value)) {\r
- other = (arrayobject *)value;\r
- needed = Py_SIZE(other);\r
- if (self == other) {\r
- /* Special case "self[i:j] = self" -- copy self first */\r
- int ret;\r
- value = array_slice(other, 0, needed);\r
- if (value == NULL)\r
- return -1;\r
- ret = array_ass_subscr(self, item, value);\r
- Py_DECREF(value);\r
- return ret;\r
- }\r
- if (other->ob_descr != self->ob_descr) {\r
- PyErr_BadArgument();\r
- return -1;\r
- }\r
- }\r
- else {\r
- PyErr_Format(PyExc_TypeError,\r
- "can only assign array (not \"%.200s\") to array slice",\r
- Py_TYPE(value)->tp_name);\r
- return -1;\r
- }\r
- itemsize = self->ob_descr->itemsize;\r
- /* for 'a[2:1] = ...', the insertion point is 'start', not 'stop' */\r
- if ((step > 0 && stop < start) ||\r
- (step < 0 && stop > start))\r
- stop = start;\r
- if (step == 1) {\r
- if (slicelength > needed) {\r
- memmove(self->ob_item + (start + needed) * itemsize,\r
- self->ob_item + stop * itemsize,\r
- (Py_SIZE(self) - stop) * itemsize);\r
- if (array_resize(self, Py_SIZE(self) +\r
- needed - slicelength) < 0)\r
- return -1;\r
- }\r
- else if (slicelength < needed) {\r
- if (array_resize(self, Py_SIZE(self) +\r
- needed - slicelength) < 0)\r
- return -1;\r
- memmove(self->ob_item + (start + needed) * itemsize,\r
- self->ob_item + stop * itemsize,\r
- (Py_SIZE(self) - start - needed) * itemsize);\r
- }\r
- if (needed > 0)\r
- memcpy(self->ob_item + start * itemsize,\r
- other->ob_item, needed * itemsize);\r
- return 0;\r
- }\r
- else if (needed == 0) {\r
- /* Delete slice */\r
- size_t cur;\r
- Py_ssize_t i;\r
-\r
- if (step < 0) {\r
- stop = start + 1;\r
- start = stop + step * (slicelength - 1) - 1;\r
- step = -step;\r
- }\r
- for (cur = start, i = 0; i < slicelength;\r
- cur += step, i++) {\r
- Py_ssize_t lim = step - 1;\r
-\r
- if (cur + step >= (size_t)Py_SIZE(self))\r
- lim = Py_SIZE(self) - cur - 1;\r
- memmove(self->ob_item + (cur - i) * itemsize,\r
- self->ob_item + (cur + 1) * itemsize,\r
- lim * itemsize);\r
- }\r
- cur = start + slicelength * step;\r
- if (cur < (size_t)Py_SIZE(self)) {\r
- memmove(self->ob_item + (cur-slicelength) * itemsize,\r
- self->ob_item + cur * itemsize,\r
- (Py_SIZE(self) - cur) * itemsize);\r
- }\r
- if (array_resize(self, Py_SIZE(self) - slicelength) < 0)\r
- return -1;\r
- return 0;\r
- }\r
- else {\r
- Py_ssize_t cur, i;\r
-\r
- if (needed != slicelength) {\r
- PyErr_Format(PyExc_ValueError,\r
- "attempt to assign array of size %zd "\r
- "to extended slice of size %zd",\r
- needed, slicelength);\r
- return -1;\r
- }\r
- for (cur = start, i = 0; i < slicelength;\r
- cur += step, i++) {\r
- memcpy(self->ob_item + cur * itemsize,\r
- other->ob_item + i * itemsize,\r
- itemsize);\r
- }\r
- return 0;\r
- }\r
-}\r
-\r
-static PyMappingMethods array_as_mapping = {\r
- (lenfunc)array_length,\r
- (binaryfunc)array_subscr,\r
- (objobjargproc)array_ass_subscr\r
-};\r
-\r
-static const void *emptybuf = "";\r
-\r
-static Py_ssize_t\r
-array_buffer_getreadbuf(arrayobject *self, Py_ssize_t index, const void **ptr)\r
-{\r
- if ( index != 0 ) {\r
- PyErr_SetString(PyExc_SystemError,\r
- "Accessing non-existent array segment");\r
- return -1;\r
- }\r
- *ptr = (void *)self->ob_item;\r
- if (*ptr == NULL)\r
- *ptr = emptybuf;\r
- return Py_SIZE(self)*self->ob_descr->itemsize;\r
-}\r
-\r
-static Py_ssize_t\r
-array_buffer_getwritebuf(arrayobject *self, Py_ssize_t index, const void **ptr)\r
-{\r
- if ( index != 0 ) {\r
- PyErr_SetString(PyExc_SystemError,\r
- "Accessing non-existent array segment");\r
- return -1;\r
- }\r
- *ptr = (void *)self->ob_item;\r
- if (*ptr == NULL)\r
- *ptr = emptybuf;\r
- return Py_SIZE(self)*self->ob_descr->itemsize;\r
-}\r
-\r
-static Py_ssize_t\r
-array_buffer_getsegcount(arrayobject *self, Py_ssize_t *lenp)\r
-{\r
- if ( lenp )\r
- *lenp = Py_SIZE(self)*self->ob_descr->itemsize;\r
- return 1;\r
-}\r
-\r
-static PySequenceMethods array_as_sequence = {\r
- (lenfunc)array_length, /*sq_length*/\r
- (binaryfunc)array_concat, /*sq_concat*/\r
- (ssizeargfunc)array_repeat, /*sq_repeat*/\r
- (ssizeargfunc)array_item, /*sq_item*/\r
- (ssizessizeargfunc)array_slice, /*sq_slice*/\r
- (ssizeobjargproc)array_ass_item, /*sq_ass_item*/\r
- (ssizessizeobjargproc)array_ass_slice, /*sq_ass_slice*/\r
- (objobjproc)array_contains, /*sq_contains*/\r
- (binaryfunc)array_inplace_concat, /*sq_inplace_concat*/\r
- (ssizeargfunc)array_inplace_repeat /*sq_inplace_repeat*/\r
-};\r
-\r
-static PyBufferProcs array_as_buffer = {\r
- (readbufferproc)array_buffer_getreadbuf,\r
- (writebufferproc)array_buffer_getwritebuf,\r
- (segcountproc)array_buffer_getsegcount,\r
- NULL,\r
-};\r
-\r
-static PyObject *\r
-array_new(PyTypeObject *type, PyObject *args, PyObject *kwds)\r
-{\r
- char c;\r
- PyObject *initial = NULL, *it = NULL;\r
- struct arraydescr *descr;\r
-\r
- if (type == &Arraytype && !_PyArg_NoKeywords("array.array()", kwds))\r
- return NULL;\r
-\r
- if (!PyArg_ParseTuple(args, "c|O:array", &c, &initial))\r
- return NULL;\r
-\r
- if (!(initial == NULL || PyList_Check(initial)\r
- || PyString_Check(initial) || PyTuple_Check(initial)\r
- || (c == 'u' && PyUnicode_Check(initial)))) {\r
- it = PyObject_GetIter(initial);\r
- if (it == NULL)\r
- return NULL;\r
- /* We set initial to NULL so that the subsequent code\r
- will create an empty array of the appropriate type\r
- and afterwards we can use array_iter_extend to populate\r
- the array.\r
- */\r
- initial = NULL;\r
- }\r
- for (descr = descriptors; descr->typecode != '\0'; descr++) {\r
- if (descr->typecode == c) {\r
- PyObject *a;\r
- Py_ssize_t len;\r
-\r
- if (initial == NULL || !(PyList_Check(initial)\r
- || PyTuple_Check(initial)))\r
- len = 0;\r
- else\r
- len = PySequence_Size(initial);\r
-\r
- a = newarrayobject(type, len, descr);\r
- if (a == NULL)\r
- return NULL;\r
-\r
- if (len > 0) {\r
- Py_ssize_t i;\r
- for (i = 0; i < len; i++) {\r
- PyObject *v =\r
- PySequence_GetItem(initial, i);\r
- if (v == NULL) {\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- if (setarrayitem(a, i, v) != 0) {\r
- Py_DECREF(v);\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- Py_DECREF(v);\r
- }\r
- } else if (initial != NULL && PyString_Check(initial)) {\r
- PyObject *t_initial, *v;\r
- t_initial = PyTuple_Pack(1, initial);\r
- if (t_initial == NULL) {\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- v = array_fromstring((arrayobject *)a,\r
- t_initial);\r
- Py_DECREF(t_initial);\r
- if (v == NULL) {\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- Py_DECREF(v);\r
-#ifdef Py_USING_UNICODE\r
- } else if (initial != NULL && PyUnicode_Check(initial)) {\r
- Py_ssize_t n = PyUnicode_GET_DATA_SIZE(initial);\r
- if (n > 0) {\r
- arrayobject *self = (arrayobject *)a;\r
- char *item = self->ob_item;\r
- item = (char *)PyMem_Realloc(item, n);\r
- if (item == NULL) {\r
- PyErr_NoMemory();\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- self->ob_item = item;\r
- Py_SIZE(self) = n / sizeof(Py_UNICODE);\r
- memcpy(item, PyUnicode_AS_DATA(initial), n);\r
- self->allocated = Py_SIZE(self);\r
- }\r
-#endif\r
- }\r
- if (it != NULL) {\r
- if (array_iter_extend((arrayobject *)a, it) == -1) {\r
- Py_DECREF(it);\r
- Py_DECREF(a);\r
- return NULL;\r
- }\r
- Py_DECREF(it);\r
- }\r
- return a;\r
- }\r
- }\r
- PyErr_SetString(PyExc_ValueError,\r
- "bad typecode (must be c, b, B, u, h, H, i, I, l, L, f or d)");\r
- return NULL;\r
-}\r
-\r
-\r
-PyDoc_STRVAR(module_doc,\r
-"This module defines an object type which can efficiently represent\n\\r
-an array of basic values: characters, integers, floating point\n\\r
-numbers. Arrays are sequence types and behave very much like lists,\n\\r
-except that the type of objects stored in them is constrained. The\n\\r
-type is specified at object creation time by using a type code, which\n\\r
-is a single character. The following type codes are defined:\n\\r
-\n\\r
- Type code C Type Minimum size in bytes \n\\r
- 'c' character 1 \n\\r
- 'b' signed integer 1 \n\\r
- 'B' unsigned integer 1 \n\\r
- 'u' Unicode character 2 \n\\r
- 'h' signed integer 2 \n\\r
- 'H' unsigned integer 2 \n\\r
- 'i' signed integer 2 \n\\r
- 'I' unsigned integer 2 \n\\r
- 'l' signed integer 4 \n\\r
- 'L' unsigned integer 4 \n\\r
- 'f' floating point 4 \n\\r
- 'd' floating point 8 \n\\r
-\n\\r
-The constructor is:\n\\r
-\n\\r
-array(typecode [, initializer]) -- create a new array\n\\r
-");\r
-\r
-PyDoc_STRVAR(arraytype_doc,\r
-"array(typecode [, initializer]) -> array\n\\r
-\n\\r
-Return a new array whose items are restricted by typecode, and\n\\r
-initialized from the optional initializer value, which must be a list,\n\\r
-string or iterable over elements of the appropriate type.\n\\r
-\n\\r
-Arrays represent basic values and behave very much like lists, except\n\\r
-the type of objects stored in them is constrained.\n\\r
-\n\\r
-Methods:\n\\r
-\n\\r
-append() -- append a new item to the end of the array\n\\r
-buffer_info() -- return information giving the current memory info\n\\r
-byteswap() -- byteswap all the items of the array\n\\r
-count() -- return number of occurrences of an object\n\\r
-extend() -- extend array by appending multiple elements from an iterable\n\\r
-fromfile() -- read items from a file object\n\\r
-fromlist() -- append items from the list\n\\r
-fromstring() -- append items from the string\n\\r
-index() -- return index of first occurrence of an object\n\\r
-insert() -- insert a new item into the array at a provided position\n\\r
-pop() -- remove and return item (default last)\n\\r
-read() -- DEPRECATED, use fromfile()\n\\r
-remove() -- remove first occurrence of an object\n\\r
-reverse() -- reverse the order of the items in the array\n\\r
-tofile() -- write all items to a file object\n\\r
-tolist() -- return the array converted to an ordinary list\n\\r
-tostring() -- return the array converted to a string\n\\r
-write() -- DEPRECATED, use tofile()\n\\r
-\n\\r
-Attributes:\n\\r
-\n\\r
-typecode -- the typecode character used to create the array\n\\r
-itemsize -- the length in bytes of one array item\n\\r
-");\r
-\r
-static PyObject *array_iter(arrayobject *ao);\r
-\r
-static PyTypeObject Arraytype = {\r
- PyVarObject_HEAD_INIT(NULL, 0)\r
- "array.array",\r
- sizeof(arrayobject),\r
- 0,\r
- (destructor)array_dealloc, /* tp_dealloc */\r
- 0, /* tp_print */\r
- 0, /* tp_getattr */\r
- 0, /* tp_setattr */\r
- 0, /* tp_compare */\r
- (reprfunc)array_repr, /* tp_repr */\r
- 0, /* tp_as_number*/\r
- &array_as_sequence, /* tp_as_sequence*/\r
- &array_as_mapping, /* tp_as_mapping*/\r
- 0, /* tp_hash */\r
- 0, /* tp_call */\r
- 0, /* tp_str */\r
- PyObject_GenericGetAttr, /* tp_getattro */\r
- 0, /* tp_setattro */\r
- &array_as_buffer, /* tp_as_buffer*/\r
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */\r
- arraytype_doc, /* tp_doc */\r
- 0, /* tp_traverse */\r
- 0, /* tp_clear */\r
- array_richcompare, /* tp_richcompare */\r
- offsetof(arrayobject, weakreflist), /* tp_weaklistoffset */\r
- (getiterfunc)array_iter, /* tp_iter */\r
- 0, /* tp_iternext */\r
- array_methods, /* tp_methods */\r
- 0, /* tp_members */\r
- array_getsets, /* tp_getset */\r
- 0, /* tp_base */\r
- 0, /* tp_dict */\r
- 0, /* tp_descr_get */\r
- 0, /* tp_descr_set */\r
- 0, /* tp_dictoffset */\r
- 0, /* tp_init */\r
- PyType_GenericAlloc, /* tp_alloc */\r
- array_new, /* tp_new */\r
- PyObject_Del, /* tp_free */\r
-};\r
-\r
-\r
-/*********************** Array Iterator **************************/\r
-\r
-typedef struct {\r
- PyObject_HEAD\r
- Py_ssize_t index;\r
- arrayobject *ao;\r
- PyObject * (*getitem)(struct arrayobject *, Py_ssize_t);\r
-} arrayiterobject;\r
-\r
-static PyTypeObject PyArrayIter_Type;\r
-\r
-#define PyArrayIter_Check(op) PyObject_TypeCheck(op, &PyArrayIter_Type)\r
-\r
-static PyObject *\r
-array_iter(arrayobject *ao)\r
-{\r
- arrayiterobject *it;\r
-\r
- if (!array_Check(ao)) {\r
- PyErr_BadInternalCall();\r
- return NULL;\r
- }\r
-\r
- it = PyObject_GC_New(arrayiterobject, &PyArrayIter_Type);\r
- if (it == NULL)\r
- return NULL;\r
-\r
- Py_INCREF(ao);\r
- it->ao = ao;\r
- it->index = 0;\r
- it->getitem = ao->ob_descr->getitem;\r
- PyObject_GC_Track(it);\r
- return (PyObject *)it;\r
-}\r
-\r
-static PyObject *\r
-arrayiter_next(arrayiterobject *it)\r
-{\r
- assert(PyArrayIter_Check(it));\r
- if (it->index < Py_SIZE(it->ao))\r
- return (*it->getitem)(it->ao, it->index++);\r
- return NULL;\r
-}\r
-\r
-static void\r
-arrayiter_dealloc(arrayiterobject *it)\r
-{\r
- PyObject_GC_UnTrack(it);\r
- Py_XDECREF(it->ao);\r
- PyObject_GC_Del(it);\r
-}\r
-\r
-static int\r
-arrayiter_traverse(arrayiterobject *it, visitproc visit, void *arg)\r
-{\r
- Py_VISIT(it->ao);\r
- return 0;\r
-}\r
-\r
-static PyTypeObject PyArrayIter_Type = {\r
- PyVarObject_HEAD_INIT(NULL, 0)\r
- "arrayiterator", /* tp_name */\r
- sizeof(arrayiterobject), /* tp_basicsize */\r
- 0, /* tp_itemsize */\r
- /* methods */\r
- (destructor)arrayiter_dealloc, /* tp_dealloc */\r
- 0, /* tp_print */\r
- 0, /* tp_getattr */\r
- 0, /* tp_setattr */\r
- 0, /* tp_compare */\r
- 0, /* tp_repr */\r
- 0, /* tp_as_number */\r
- 0, /* tp_as_sequence */\r
- 0, /* tp_as_mapping */\r
- 0, /* tp_hash */\r
- 0, /* tp_call */\r
- 0, /* tp_str */\r
- PyObject_GenericGetAttr, /* tp_getattro */\r
- 0, /* tp_setattro */\r
- 0, /* tp_as_buffer */\r
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */\r
- 0, /* tp_doc */\r
- (traverseproc)arrayiter_traverse, /* tp_traverse */\r
- 0, /* tp_clear */\r
- 0, /* tp_richcompare */\r
- 0, /* tp_weaklistoffset */\r
- PyObject_SelfIter, /* tp_iter */\r
- (iternextfunc)arrayiter_next, /* tp_iternext */\r
- 0, /* tp_methods */\r
-};\r
-\r
-\r
-/*********************** Install Module **************************/\r
-\r
-/* No functions in array module. */\r
-static PyMethodDef a_methods[] = {\r
- {NULL, NULL, 0, NULL} /* Sentinel */\r
-};\r
-\r
-\r
-PyMODINIT_FUNC\r
-initarray(void)\r
-{\r
- PyObject *m;\r
-\r
- Arraytype.ob_type = &PyType_Type;\r
- PyArrayIter_Type.ob_type = &PyType_Type;\r
- m = Py_InitModule3("array", a_methods, module_doc);\r
- if (m == NULL)\r
- return;\r
-\r
- Py_INCREF((PyObject *)&Arraytype);\r
- PyModule_AddObject(m, "ArrayType", (PyObject *)&Arraytype);\r
- Py_INCREF((PyObject *)&Arraytype);\r
- PyModule_AddObject(m, "array", (PyObject *)&Arraytype);\r
- /* No need to check the error here, the caller will do that */\r
-}\r