1993-02-19 11:55:02 -04:00
|
|
|
/* Array object implementation */
|
|
|
|
|
|
|
|
/* An array is a uniform list -- all items have the same type.
|
|
|
|
The item type is restricted to simple C types like int or float */
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
#include "Python.h"
|
1996-12-09 18:24:19 -04:00
|
|
|
|
1994-08-19 09:01:32 -03:00
|
|
|
#ifdef STDC_HEADERS
|
|
|
|
#include <stddef.h>
|
1999-08-27 17:33:52 -03:00
|
|
|
#else /* !STDC_HEADERS */
|
|
|
|
#ifndef DONT_HAVE_SYS_TYPES_H
|
1994-08-01 08:34:53 -03:00
|
|
|
#include <sys/types.h> /* For size_t */
|
1999-08-27 17:33:52 -03:00
|
|
|
#endif /* DONT_HAVE_SYS_TYPES_H */
|
|
|
|
#endif /* !STDC_HEADERS */
|
1993-02-19 11:55:02 -04:00
|
|
|
|
|
|
|
struct arrayobject; /* Forward */
|
|
|
|
|
2000-09-10 02:22:54 -03:00
|
|
|
/* All possible arraydescr values are defined in the vector "descriptors"
|
|
|
|
* below. That's defined later because the appropriate get and set
|
|
|
|
* functions aren't visible yet.
|
|
|
|
*/
|
1993-02-19 11:55:02 -04:00
|
|
|
struct arraydescr {
|
|
|
|
int typecode;
|
|
|
|
int itemsize;
|
2000-07-09 00:09:57 -03:00
|
|
|
PyObject * (*getitem)(struct arrayobject *, int);
|
|
|
|
int (*setitem)(struct arrayobject *, int, PyObject *);
|
1993-02-19 11:55:02 -04:00
|
|
|
};
|
|
|
|
|
|
|
|
typedef struct arrayobject {
|
2002-03-01 06:27:01 -04:00
|
|
|
PyObject_HEAD
|
|
|
|
int ob_size;
|
1993-02-19 11:55:02 -04:00
|
|
|
char *ob_item;
|
|
|
|
struct arraydescr *ob_descr;
|
|
|
|
} arrayobject;
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
staticforward PyTypeObject Arraytype;
|
1993-02-19 11:55:02 -04:00
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
#define array_Check(op) PyObject_TypeCheck(op, &Arraytype)
|
|
|
|
#define array_CheckExact(op) ((op)->ob_type == &Arraytype)
|
1993-02-19 11:55:02 -04:00
|
|
|
|
2000-09-10 02:22:54 -03:00
|
|
|
/****************************************************************************
|
|
|
|
Get and Set functions for each type.
|
|
|
|
A Get function takes an arrayobject* and an integer index, returning the
|
|
|
|
array value at that index wrapped in an appropriate PyObject*.
|
|
|
|
A Set function takes an arrayobject, integer index, and PyObject*; sets
|
|
|
|
the array value at that index to the raw C data extracted from the PyObject*,
|
|
|
|
and returns 0 if successful, else nonzero on failure (PyObject* not of an
|
|
|
|
appropriate type or value).
|
|
|
|
Note that the basic Get and Set functions do NOT check that the index is
|
|
|
|
in bounds; that's the responsibility of the caller.
|
|
|
|
****************************************************************************/
|
1993-02-19 11:55:02 -04:00
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
c_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyString_FromStringAndSize(&((char *)ap->ob_item)[i], 1);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
c_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char x;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "c;array item must be char", &x))
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
2002-03-01 06:27:01 -04:00
|
|
|
((char *)ap->ob_item)[i] = x;
|
1993-02-19 11:55:02 -04:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
b_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
long x = ((char *)ap->ob_item)[i];
|
|
|
|
if (x >= 128)
|
|
|
|
x -= 256;
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyInt_FromLong(x);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
b_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2000-06-28 14:49:30 -03:00
|
|
|
short x;
|
|
|
|
/* PyArg_Parse's 'b' formatter is for an unsigned char, therefore
|
|
|
|
must use the next size up that is signed ('h') and manually do
|
|
|
|
the overflow checking */
|
|
|
|
if (!PyArg_Parse(v, "h;array item must be integer", &x))
|
|
|
|
return -1;
|
2000-06-30 21:38:19 -03:00
|
|
|
else if (x < -128) {
|
2000-06-28 14:49:30 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"signed char is less than minimum");
|
|
|
|
return -1;
|
|
|
|
}
|
2000-06-30 21:38:19 -03:00
|
|
|
else if (x > 127) {
|
2000-06-28 14:49:30 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"signed char is greater than maximum");
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
2000-06-28 14:49:30 -03:00
|
|
|
}
|
1993-02-19 11:55:02 -04:00
|
|
|
if (i >= 0)
|
2000-06-28 14:49:30 -03:00
|
|
|
((char *)ap->ob_item)[i] = (char)x;
|
1993-02-19 11:55:02 -04:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
BB_getitem(arrayobject *ap, int i)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
long x = ((unsigned char *)ap->ob_item)[i];
|
|
|
|
return PyInt_FromLong(x);
|
|
|
|
}
|
|
|
|
|
2000-06-28 14:49:30 -03:00
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
BB_setitem(arrayobject *ap, int i, PyObject *v)
|
2000-06-28 14:49:30 -03:00
|
|
|
{
|
|
|
|
unsigned char x;
|
|
|
|
/* 'B' == unsigned char, maps to PyArg_Parse's 'b' formatter */
|
|
|
|
if (!PyArg_Parse(v, "b;array item must be integer", &x))
|
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
2002-03-01 06:27:01 -04:00
|
|
|
((char *)ap->ob_item)[i] = x;
|
2000-06-28 14:49:30 -03:00
|
|
|
return 0;
|
|
|
|
}
|
1997-01-03 15:09:47 -04:00
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
static PyObject *
|
|
|
|
u_getitem(arrayobject *ap, int i)
|
|
|
|
{
|
|
|
|
return PyUnicode_FromUnicode(&((Py_UNICODE *) ap->ob_item)[i], 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
u_setitem(arrayobject *ap, int i, PyObject *v)
|
|
|
|
{
|
|
|
|
Py_UNICODE *p;
|
|
|
|
int len;
|
|
|
|
|
|
|
|
if (!PyArg_Parse(v, "u#;array item must be unicode character", &p, &len))
|
|
|
|
return -1;
|
|
|
|
if (len != 1) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "array item must be unicode character");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (i >= 0)
|
|
|
|
((Py_UNICODE *)ap->ob_item)[i] = p[0];
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
h_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyInt_FromLong((long) ((short *)ap->ob_item)[i]);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
h_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
short x;
|
2000-06-28 14:49:30 -03:00
|
|
|
/* 'h' == signed short, maps to PyArg_Parse's 'h' formatter */
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "h;array item must be integer", &x))
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
|
|
|
((short *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
HH_getitem(arrayobject *ap, int i)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
return PyInt_FromLong((long) ((unsigned short *)ap->ob_item)[i]);
|
|
|
|
}
|
|
|
|
|
2000-06-28 14:49:30 -03:00
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
HH_setitem(arrayobject *ap, int i, PyObject *v)
|
2000-06-28 14:49:30 -03:00
|
|
|
{
|
|
|
|
int x;
|
|
|
|
/* PyArg_Parse's 'h' formatter is for a signed short, therefore
|
|
|
|
must use the next size up and manually do the overflow checking */
|
|
|
|
if (!PyArg_Parse(v, "i;array item must be integer", &x))
|
|
|
|
return -1;
|
|
|
|
else if (x < 0) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned short is less than minimum");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
else if (x > USHRT_MAX) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned short is greater than maximum");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (i >= 0)
|
|
|
|
((short *)ap->ob_item)[i] = (short)x;
|
|
|
|
return 0;
|
|
|
|
}
|
1997-01-03 15:09:47 -04:00
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
i_getitem(arrayobject *ap, int i)
|
1993-11-03 11:01:26 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyInt_FromLong((long) ((int *)ap->ob_item)[i]);
|
1993-11-03 11:01:26 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
i_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-11-03 11:01:26 -04:00
|
|
|
{
|
|
|
|
int x;
|
2000-06-28 14:49:30 -03:00
|
|
|
/* 'i' == signed int, maps to PyArg_Parse's 'i' formatter */
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "i;array item must be integer", &x))
|
1993-11-03 11:01:26 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
|
|
|
((int *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
II_getitem(arrayobject *ap, int i)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
return PyLong_FromUnsignedLong(
|
|
|
|
(unsigned long) ((unsigned int *)ap->ob_item)[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
II_setitem(arrayobject *ap, int i, PyObject *v)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
unsigned long x;
|
|
|
|
if (PyLong_Check(v)) {
|
|
|
|
x = PyLong_AsUnsignedLong(v);
|
|
|
|
if (x == (unsigned long) -1 && PyErr_Occurred())
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
else {
|
2000-06-28 14:49:30 -03:00
|
|
|
long y;
|
|
|
|
if (!PyArg_Parse(v, "l;array item must be integer", &y))
|
1997-01-03 15:09:47 -04:00
|
|
|
return -1;
|
2000-06-28 14:49:30 -03:00
|
|
|
if (y < 0) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned int is less than minimum");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
x = (unsigned long)y;
|
2000-09-10 02:22:54 -03:00
|
|
|
|
2000-06-28 14:49:30 -03:00
|
|
|
}
|
|
|
|
if (x > UINT_MAX) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned int is greater than maximum");
|
|
|
|
return -1;
|
1997-01-03 15:09:47 -04:00
|
|
|
}
|
2000-06-28 14:49:30 -03:00
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
if (i >= 0)
|
2000-06-28 14:49:30 -03:00
|
|
|
((unsigned int *)ap->ob_item)[i] = (unsigned int)x;
|
1997-01-03 15:09:47 -04:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
l_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyInt_FromLong(((long *)ap->ob_item)[i]);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
l_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
long x;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "l;array item must be integer", &x))
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
|
|
|
((long *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
LL_getitem(arrayobject *ap, int i)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
return PyLong_FromUnsignedLong(((unsigned long *)ap->ob_item)[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
LL_setitem(arrayobject *ap, int i, PyObject *v)
|
1997-01-03 15:09:47 -04:00
|
|
|
{
|
|
|
|
unsigned long x;
|
|
|
|
if (PyLong_Check(v)) {
|
|
|
|
x = PyLong_AsUnsignedLong(v);
|
|
|
|
if (x == (unsigned long) -1 && PyErr_Occurred())
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
else {
|
2000-06-28 14:49:30 -03:00
|
|
|
long y;
|
|
|
|
if (!PyArg_Parse(v, "l;array item must be integer", &y))
|
|
|
|
return -1;
|
|
|
|
if (y < 0) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned long is less than minimum");
|
1997-01-03 15:09:47 -04:00
|
|
|
return -1;
|
2000-06-28 14:49:30 -03:00
|
|
|
}
|
|
|
|
x = (unsigned long)y;
|
2000-09-10 02:22:54 -03:00
|
|
|
|
2000-06-28 14:49:30 -03:00
|
|
|
}
|
|
|
|
if (x > ULONG_MAX) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"unsigned long is greater than maximum");
|
|
|
|
return -1;
|
1997-01-03 15:09:47 -04:00
|
|
|
}
|
2000-09-10 02:22:54 -03:00
|
|
|
|
1997-01-03 15:09:47 -04:00
|
|
|
if (i >= 0)
|
|
|
|
((unsigned long *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
f_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyFloat_FromDouble((double) ((float *)ap->ob_item)[i]);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
f_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
float x;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "f;array item must be float", &x))
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
|
|
|
((float *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
d_getitem(arrayobject *ap, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyFloat_FromDouble(((double *)ap->ob_item)[i]);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
d_setitem(arrayobject *ap, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
double x;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyArg_Parse(v, "d;array item must be float", &x))
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
if (i >= 0)
|
|
|
|
((double *)ap->ob_item)[i] = x;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Description of types */
|
1993-06-17 09:35:49 -03:00
|
|
|
static struct arraydescr descriptors[] = {
|
1993-02-19 11:55:02 -04:00
|
|
|
{'c', sizeof(char), c_getitem, c_setitem},
|
|
|
|
{'b', sizeof(char), b_getitem, b_setitem},
|
1997-05-16 13:21:38 -03:00
|
|
|
{'B', sizeof(char), BB_getitem, BB_setitem},
|
2002-03-01 06:27:01 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
{'u', sizeof(Py_UNICODE), u_getitem, u_setitem},
|
|
|
|
#endif
|
1993-02-19 11:55:02 -04:00
|
|
|
{'h', sizeof(short), h_getitem, h_setitem},
|
1997-05-16 13:21:38 -03:00
|
|
|
{'H', sizeof(short), HH_getitem, HH_setitem},
|
1993-11-03 11:01:26 -04:00
|
|
|
{'i', sizeof(int), i_getitem, i_setitem},
|
1997-05-16 13:21:38 -03:00
|
|
|
{'I', sizeof(int), II_getitem, II_setitem},
|
1993-02-19 11:55:02 -04:00
|
|
|
{'l', sizeof(long), l_getitem, l_setitem},
|
1997-05-16 13:21:38 -03:00
|
|
|
{'L', sizeof(long), LL_getitem, LL_setitem},
|
1993-02-19 11:55:02 -04:00
|
|
|
{'f', sizeof(float), f_getitem, f_setitem},
|
|
|
|
{'d', sizeof(double), d_getitem, d_setitem},
|
|
|
|
{'\0', 0, 0, 0} /* Sentinel */
|
|
|
|
};
|
2000-09-10 02:22:54 -03:00
|
|
|
|
|
|
|
/****************************************************************************
|
|
|
|
Implementations of array object methods.
|
|
|
|
****************************************************************************/
|
1993-02-19 11:55:02 -04:00
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2002-03-01 06:27:01 -04:00
|
|
|
newarrayobject(PyTypeObject *type, int size, struct arraydescr *descr)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
arrayobject *op;
|
1994-08-01 08:34:53 -03:00
|
|
|
size_t nbytes;
|
2002-03-01 06:27:01 -04:00
|
|
|
|
1993-02-19 11:55:02 -04:00
|
|
|
if (size < 0) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_BadInternalCall();
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
|
1993-02-19 11:55:02 -04:00
|
|
|
nbytes = size * descr->itemsize;
|
|
|
|
/* Check for overflow */
|
1997-04-11 17:44:04 -03:00
|
|
|
if (nbytes / descr->itemsize != (size_t)size) {
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
op = (arrayobject *) type->tp_alloc(type, 0);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (op == NULL) {
|
2002-03-01 06:27:01 -04:00
|
|
|
return NULL;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
op->ob_size = size;
|
1993-02-19 11:55:02 -04:00
|
|
|
if (size <= 0) {
|
|
|
|
op->ob_item = NULL;
|
|
|
|
}
|
|
|
|
else {
|
1996-12-09 16:10:36 -04:00
|
|
|
op->ob_item = PyMem_NEW(char, nbytes);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (op->ob_item == NULL) {
|
2000-05-03 20:44:39 -03:00
|
|
|
PyObject_Del(op);
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
op->ob_descr = descr;
|
1996-12-09 16:10:36 -04:00
|
|
|
return (PyObject *) op;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
getarrayitem(PyObject *op, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
register arrayobject *ap;
|
2002-03-01 06:27:01 -04:00
|
|
|
assert(array_Check(op));
|
1993-02-19 11:55:02 -04:00
|
|
|
ap = (arrayobject *)op;
|
|
|
|
if (i < 0 || i >= ap->ob_size) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_IndexError, "array index out of range");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return (*ap->ob_descr->getitem)(ap, i);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
ins1(arrayobject *self, int where, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char *items;
|
|
|
|
if (v == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_BadInternalCall();
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if ((*self->ob_descr->setitem)(self, -1, v) < 0)
|
|
|
|
return -1;
|
|
|
|
items = self->ob_item;
|
1996-12-09 18:24:19 -04:00
|
|
|
PyMem_RESIZE(items, char,
|
|
|
|
(self->ob_size+1) * self->ob_descr->itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (items == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (where < 0)
|
|
|
|
where = 0;
|
|
|
|
if (where > self->ob_size)
|
|
|
|
where = self->ob_size;
|
|
|
|
memmove(items + (where+1)*self->ob_descr->itemsize,
|
|
|
|
items + where*self->ob_descr->itemsize,
|
|
|
|
(self->ob_size-where)*self->ob_descr->itemsize);
|
|
|
|
self->ob_item = items;
|
|
|
|
self->ob_size++;
|
|
|
|
return (*self->ob_descr->setitem)(self, where, v);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Methods */
|
|
|
|
|
|
|
|
static void
|
2000-07-13 18:10:57 -03:00
|
|
|
array_dealloc(arrayobject *op)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
if (op->ob_item != NULL)
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_DEL(op->ob_item);
|
2002-03-01 06:27:01 -04:00
|
|
|
op->ob_type->tp_free((PyObject *)op);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
2001-01-17 21:02:55 -04:00
|
|
|
static PyObject *
|
|
|
|
array_richcompare(PyObject *v, PyObject *w, int op)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2001-01-17 21:02:55 -04:00
|
|
|
arrayobject *va, *wa;
|
|
|
|
PyObject *vi = NULL;
|
|
|
|
PyObject *wi = NULL;
|
|
|
|
int i, k;
|
|
|
|
PyObject *res;
|
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
if (!array_Check(v) || !array_Check(w)) {
|
2001-01-17 21:02:55 -04:00
|
|
|
Py_INCREF(Py_NotImplemented);
|
|
|
|
return Py_NotImplemented;
|
|
|
|
}
|
|
|
|
|
|
|
|
va = (arrayobject *)v;
|
|
|
|
wa = (arrayobject *)w;
|
|
|
|
|
|
|
|
if (va->ob_size != wa->ob_size && (op == Py_EQ || op == Py_NE)) {
|
|
|
|
/* Shortcut: if the lengths differ, the arrays differ */
|
|
|
|
if (op == Py_EQ)
|
|
|
|
res = Py_False;
|
|
|
|
else
|
|
|
|
res = Py_True;
|
|
|
|
Py_INCREF(res);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Search for the first index where items are different */
|
|
|
|
k = 1;
|
|
|
|
for (i = 0; i < va->ob_size && i < wa->ob_size; i++) {
|
|
|
|
vi = getarrayitem(v, i);
|
|
|
|
wi = getarrayitem(w, i);
|
|
|
|
if (vi == NULL || wi == NULL) {
|
|
|
|
Py_XDECREF(vi);
|
|
|
|
Py_XDECREF(wi);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
k = PyObject_RichCompareBool(vi, wi, Py_EQ);
|
|
|
|
if (k == 0)
|
|
|
|
break; /* Keeping vi and wi alive! */
|
|
|
|
Py_DECREF(vi);
|
|
|
|
Py_DECREF(wi);
|
|
|
|
if (k < 0)
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (k) {
|
|
|
|
/* No more items to compare -- compare sizes */
|
|
|
|
int vs = va->ob_size;
|
|
|
|
int ws = wa->ob_size;
|
1993-02-19 11:55:02 -04:00
|
|
|
int cmp;
|
2001-01-17 21:02:55 -04:00
|
|
|
switch (op) {
|
|
|
|
case Py_LT: cmp = vs < ws; break;
|
2001-01-25 18:12:43 -04:00
|
|
|
case Py_LE: cmp = vs <= ws; break;
|
2001-01-17 21:02:55 -04:00
|
|
|
case Py_EQ: cmp = vs == ws; break;
|
|
|
|
case Py_NE: cmp = vs != ws; break;
|
|
|
|
case Py_GT: cmp = vs > ws; break;
|
|
|
|
case Py_GE: cmp = vs >= ws; break;
|
|
|
|
default: return NULL; /* cannot happen */
|
|
|
|
}
|
|
|
|
if (cmp)
|
|
|
|
res = Py_True;
|
1993-02-19 11:55:02 -04:00
|
|
|
else
|
2001-01-17 21:02:55 -04:00
|
|
|
res = Py_False;
|
|
|
|
Py_INCREF(res);
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* We have an item that differs. First, shortcuts for EQ/NE */
|
|
|
|
if (op == Py_EQ) {
|
|
|
|
Py_INCREF(Py_False);
|
|
|
|
res = Py_False;
|
|
|
|
}
|
|
|
|
else if (op == Py_NE) {
|
|
|
|
Py_INCREF(Py_True);
|
|
|
|
res = Py_True;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
2001-01-17 21:02:55 -04:00
|
|
|
else {
|
|
|
|
/* Compare the final item again using the proper operator */
|
|
|
|
res = PyObject_RichCompare(vi, wi, op);
|
|
|
|
}
|
|
|
|
Py_DECREF(vi);
|
|
|
|
Py_DECREF(wi);
|
|
|
|
return res;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_length(arrayobject *a)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
return a->ob_size;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_item(arrayobject *a, int i)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_IndexError, "array index out of range");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
return getarrayitem((PyObject *)a, i);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_slice(arrayobject *a, int ilow, int ihigh)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
arrayobject *np;
|
|
|
|
if (ilow < 0)
|
|
|
|
ilow = 0;
|
|
|
|
else if (ilow > a->ob_size)
|
|
|
|
ilow = a->ob_size;
|
|
|
|
if (ihigh < 0)
|
|
|
|
ihigh = 0;
|
|
|
|
if (ihigh < ilow)
|
|
|
|
ihigh = ilow;
|
|
|
|
else if (ihigh > a->ob_size)
|
|
|
|
ihigh = a->ob_size;
|
2002-03-01 06:27:01 -04:00
|
|
|
np = (arrayobject *) newarrayobject(&Arraytype, ihigh - ilow, a->ob_descr);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (np == NULL)
|
|
|
|
return NULL;
|
|
|
|
memcpy(np->ob_item, a->ob_item + ilow * a->ob_descr->itemsize,
|
|
|
|
(ihigh-ilow) * a->ob_descr->itemsize);
|
1996-12-09 16:10:36 -04:00
|
|
|
return (PyObject *)np;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_concat(arrayobject *a, PyObject *bb)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
int size;
|
|
|
|
arrayobject *np;
|
2002-03-01 06:27:01 -04:00
|
|
|
if (!array_Check(bb)) {
|
2000-05-31 23:02:46 -03:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"can only append array (not \"%.200s\") to array",
|
|
|
|
bb->ob_type->tp_name);
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
#define b ((arrayobject *)bb)
|
|
|
|
if (a->ob_descr != b->ob_descr) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_BadArgument();
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
size = a->ob_size + b->ob_size;
|
2002-03-01 06:27:01 -04:00
|
|
|
np = (arrayobject *) newarrayobject(&Arraytype, size, a->ob_descr);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (np == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
memcpy(np->ob_item, a->ob_item, a->ob_size*a->ob_descr->itemsize);
|
|
|
|
memcpy(np->ob_item + a->ob_size*a->ob_descr->itemsize,
|
1993-11-05 06:16:27 -04:00
|
|
|
b->ob_item, b->ob_size*b->ob_descr->itemsize);
|
1996-12-09 16:10:36 -04:00
|
|
|
return (PyObject *)np;
|
1993-02-19 11:55:02 -04:00
|
|
|
#undef b
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_repeat(arrayobject *a, int n)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1995-01-02 15:30:30 -04:00
|
|
|
int i;
|
1993-02-19 11:55:02 -04:00
|
|
|
int size;
|
|
|
|
arrayobject *np;
|
|
|
|
char *p;
|
|
|
|
int nbytes;
|
|
|
|
if (n < 0)
|
|
|
|
n = 0;
|
|
|
|
size = a->ob_size * n;
|
2002-03-01 06:27:01 -04:00
|
|
|
np = (arrayobject *) newarrayobject(&Arraytype, size, a->ob_descr);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (np == NULL)
|
|
|
|
return NULL;
|
|
|
|
p = np->ob_item;
|
|
|
|
nbytes = a->ob_size * a->ob_descr->itemsize;
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
memcpy(p, a->ob_item, nbytes);
|
|
|
|
p += nbytes;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
return (PyObject *) np;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_ass_slice(arrayobject *a, int ilow, int ihigh, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char *item;
|
|
|
|
int n; /* Size of replacement array */
|
|
|
|
int d; /* Change in size */
|
|
|
|
#define b ((arrayobject *)v)
|
|
|
|
if (v == NULL)
|
|
|
|
n = 0;
|
2002-03-01 06:27:01 -04:00
|
|
|
else if (array_Check(v)) {
|
1993-02-19 11:55:02 -04:00
|
|
|
n = b->ob_size;
|
|
|
|
if (a == b) {
|
|
|
|
/* Special case "a[i:j] = a" -- copy b first */
|
|
|
|
int ret;
|
|
|
|
v = array_slice(b, 0, n);
|
|
|
|
ret = array_ass_slice(a, ilow, ihigh, v);
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_DECREF(v);
|
1993-02-19 11:55:02 -04:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
if (b->ob_descr != a->ob_descr) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_BadArgument();
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
2000-05-31 23:02:46 -03:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"can only assign array (not \"%.200s\") to array slice",
|
|
|
|
v->ob_type->tp_name);
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (ilow < 0)
|
|
|
|
ilow = 0;
|
|
|
|
else if (ilow > a->ob_size)
|
|
|
|
ilow = a->ob_size;
|
|
|
|
if (ihigh < 0)
|
|
|
|
ihigh = 0;
|
|
|
|
if (ihigh < ilow)
|
|
|
|
ihigh = ilow;
|
|
|
|
else if (ihigh > a->ob_size)
|
|
|
|
ihigh = a->ob_size;
|
|
|
|
item = a->ob_item;
|
|
|
|
d = n - (ihigh-ilow);
|
|
|
|
if (d < 0) { /* Delete -d items */
|
|
|
|
memmove(item + (ihigh+d)*a->ob_descr->itemsize,
|
|
|
|
item + ihigh*a->ob_descr->itemsize,
|
|
|
|
(a->ob_size-ihigh)*a->ob_descr->itemsize);
|
|
|
|
a->ob_size += d;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char, a->ob_size*a->ob_descr->itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
/* Can't fail */
|
|
|
|
a->ob_item = item;
|
|
|
|
}
|
|
|
|
else if (d > 0) { /* Insert d items */
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char,
|
|
|
|
(a->ob_size + d)*a->ob_descr->itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (item == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
memmove(item + (ihigh+d)*a->ob_descr->itemsize,
|
|
|
|
item + ihigh*a->ob_descr->itemsize,
|
|
|
|
(a->ob_size-ihigh)*a->ob_descr->itemsize);
|
|
|
|
a->ob_item = item;
|
|
|
|
a->ob_size += d;
|
|
|
|
}
|
|
|
|
if (n > 0)
|
|
|
|
memcpy(item + ilow*a->ob_descr->itemsize, b->ob_item,
|
|
|
|
n*b->ob_descr->itemsize);
|
|
|
|
return 0;
|
|
|
|
#undef b
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_ass_item(arrayobject *a, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_IndexError,
|
|
|
|
"array assignment index out of range");
|
1993-02-19 11:55:02 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (v == NULL)
|
|
|
|
return array_ass_slice(a, i, i+1, v);
|
|
|
|
return (*a->ob_descr->setitem)(a, i, v);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
setarrayitem(PyObject *a, int i, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2002-03-01 06:27:01 -04:00
|
|
|
assert(array_Check(a));
|
1993-02-19 11:55:02 -04:00
|
|
|
return array_ass_item((arrayobject *)a, i, v);
|
|
|
|
}
|
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
static int
|
|
|
|
array_do_extend(arrayobject *self, PyObject *bb)
|
|
|
|
{
|
|
|
|
int size;
|
|
|
|
|
|
|
|
if (!array_Check(bb)) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"can only extend array with array (not \"%.200s\")",
|
|
|
|
bb->ob_type->tp_name);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
#define b ((arrayobject *)bb)
|
|
|
|
if (self->ob_descr != b->ob_descr) {
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
"can only extend with array of same kind");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
size = self->ob_size + b->ob_size;
|
|
|
|
PyMem_RESIZE(self->ob_item, char, size*self->ob_descr->itemsize);
|
|
|
|
if (self->ob_item == NULL) {
|
|
|
|
PyObject_Del(self);
|
|
|
|
PyErr_NoMemory();
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
memcpy(self->ob_item + self->ob_size*self->ob_descr->itemsize,
|
|
|
|
b->ob_item, b->ob_size*b->ob_descr->itemsize);
|
|
|
|
self->ob_size = size;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
#undef b
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_inplace_concat(arrayobject *self, PyObject *bb)
|
|
|
|
{
|
|
|
|
if (array_do_extend(self, bb) == -1)
|
|
|
|
return NULL;
|
|
|
|
Py_INCREF(self);
|
|
|
|
return (PyObject *)self;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_inplace_repeat(arrayobject *self, int n)
|
|
|
|
{
|
|
|
|
char *items, *p;
|
|
|
|
int size, i;
|
|
|
|
|
|
|
|
if (self->ob_size > 0) {
|
|
|
|
if (n < 0)
|
|
|
|
n = 0;
|
|
|
|
items = self->ob_item;
|
|
|
|
size = self->ob_size * self->ob_descr->itemsize;
|
|
|
|
if (n == 0) {
|
|
|
|
PyMem_FREE(items);
|
|
|
|
self->ob_item = NULL;
|
|
|
|
self->ob_size = 0;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
PyMem_Resize(items, char, n * size);
|
|
|
|
if (items == NULL)
|
|
|
|
return PyErr_NoMemory();
|
|
|
|
p = items;
|
|
|
|
for (i = 1; i < n; i++) {
|
|
|
|
p += size;
|
|
|
|
memcpy(p, items, size);
|
|
|
|
}
|
|
|
|
self->ob_item = items;
|
|
|
|
self->ob_size *= n;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Py_INCREF(self);
|
|
|
|
return (PyObject *)self;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
ins(arrayobject *self, int where, PyObject *v)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
if (ins1(self, where, v) != 0)
|
|
|
|
return NULL;
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
2000-07-31 17:52:21 -03:00
|
|
|
static PyObject *
|
|
|
|
array_count(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
int count = 0;
|
|
|
|
int i;
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "O:count", &v))
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
|
|
|
PyObject *selfi = getarrayitem((PyObject *)self, i);
|
2001-01-17 21:02:55 -04:00
|
|
|
int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);
|
2000-07-31 17:52:21 -03:00
|
|
|
Py_DECREF(selfi);
|
2001-01-17 21:02:55 -04:00
|
|
|
if (cmp > 0)
|
|
|
|
count++;
|
|
|
|
else if (cmp < 0)
|
2000-07-31 17:52:21 -03:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return PyInt_FromLong((long)count);
|
|
|
|
}
|
|
|
|
|
|
|
|
static char count_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"count(x)\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
\n\
|
|
|
|
Return number of occurences of x in the array.";
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_index(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "O:index", &v))
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
|
|
|
PyObject *selfi = getarrayitem((PyObject *)self, i);
|
2001-01-17 21:02:55 -04:00
|
|
|
int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);
|
|
|
|
Py_DECREF(selfi);
|
|
|
|
if (cmp > 0) {
|
2000-07-31 17:52:21 -03:00
|
|
|
return PyInt_FromLong((long)i);
|
|
|
|
}
|
2001-01-17 21:02:55 -04:00
|
|
|
else if (cmp < 0)
|
2000-07-31 17:52:21 -03:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
PyErr_SetString(PyExc_ValueError, "array.index(x): x not in list");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char index_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"index(x)\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
\n\
|
|
|
|
Return index of first occurence of x in the array.";
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_remove(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "O:remove", &v))
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
|
|
|
PyObject *selfi = getarrayitem((PyObject *)self,i);
|
2001-01-17 21:02:55 -04:00
|
|
|
int cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);
|
|
|
|
Py_DECREF(selfi);
|
|
|
|
if (cmp > 0) {
|
2000-07-31 17:52:21 -03:00
|
|
|
if (array_ass_slice(self, i, i+1,
|
|
|
|
(PyObject *)NULL) != 0)
|
|
|
|
return NULL;
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
2001-01-17 21:02:55 -04:00
|
|
|
else if (cmp < 0)
|
2000-07-31 17:52:21 -03:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
PyErr_SetString(PyExc_ValueError, "array.remove(x): x not in list");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char remove_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"remove(x)\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
\n\
|
|
|
|
Remove the first occurence of x in the array.";
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_pop(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
int i = -1;
|
|
|
|
PyObject *v;
|
|
|
|
if (!PyArg_ParseTuple(args, "|i:pop", &i))
|
|
|
|
return NULL;
|
|
|
|
if (self->ob_size == 0) {
|
|
|
|
/* Special-case most common failure cause */
|
|
|
|
PyErr_SetString(PyExc_IndexError, "pop from empty array");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (i < 0)
|
|
|
|
i += self->ob_size;
|
|
|
|
if (i < 0 || i >= self->ob_size) {
|
|
|
|
PyErr_SetString(PyExc_IndexError, "pop index out of range");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
v = getarrayitem((PyObject *)self,i);
|
|
|
|
if (array_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {
|
|
|
|
Py_DECREF(v);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char pop_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"pop([i])\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
\n\
|
|
|
|
Return the i-th element and delete it from the array. i defaults to -1.";
|
|
|
|
|
|
|
|
static PyObject *
|
2000-08-31 02:18:54 -03:00
|
|
|
array_extend(arrayobject *self, PyObject *args)
|
2000-07-31 17:52:21 -03:00
|
|
|
{
|
|
|
|
PyObject *bb;
|
2000-09-10 02:22:54 -03:00
|
|
|
|
2000-07-31 17:52:21 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "O:extend", &bb))
|
|
|
|
return NULL;
|
2002-03-01 06:27:01 -04:00
|
|
|
if (array_do_extend(self, bb) == -1)
|
2000-07-31 17:52:21 -03:00
|
|
|
return NULL;
|
2002-03-01 06:27:01 -04:00
|
|
|
Py_INCREF(Py_None);
|
2000-07-31 17:52:21 -03:00
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char extend_doc [] =
|
|
|
|
"extend(array)\n\
|
|
|
|
\n\
|
|
|
|
Append array items to the end of the array.";
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_insert(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
int i;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "iO:insert", &i, &v))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
return ins(self, i, v);
|
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char insert_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"insert(i,x)\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
|
|
|
Insert a new item x into the array before position i.";
|
|
|
|
|
|
|
|
|
1997-08-12 11:55:56 -03:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_buffer_info(arrayobject *self, PyObject *args)
|
1997-08-12 11:55:56 -03:00
|
|
|
{
|
2000-09-16 19:31:29 -03:00
|
|
|
PyObject* retval = NULL;
|
|
|
|
if (!PyArg_ParseTuple(args, ":buffer_info"))
|
|
|
|
return NULL;
|
|
|
|
retval = PyTuple_New(2);
|
|
|
|
if (!retval)
|
|
|
|
return NULL;
|
2000-06-28 14:49:30 -03:00
|
|
|
|
|
|
|
PyTuple_SET_ITEM(retval, 0, PyLong_FromVoidPtr(self->ob_item));
|
|
|
|
PyTuple_SET_ITEM(retval, 1, PyInt_FromLong((long)(self->ob_size)));
|
|
|
|
|
|
|
|
return retval;
|
1997-08-12 11:55:56 -03:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char buffer_info_doc [] =
|
2000-09-16 19:31:29 -03:00
|
|
|
"buffer_info() -> (address, length)\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
|
|
|
Return a tuple (address, length) giving the current memory address and\n\
|
2001-07-27 13:05:32 -03:00
|
|
|
the length in items of the buffer used to hold array's contents\n\
|
|
|
|
The length should be multiplied by the itemsize attribute to calculate\n\
|
|
|
|
the buffer length in bytes.";
|
1998-10-13 11:27:22 -03:00
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_append(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "O:append", &v))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
return ins(self, (int) self->ob_size, v);
|
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char append_doc [] =
|
|
|
|
"append(x)\n\
|
|
|
|
\n\
|
|
|
|
Append new value x to the end of the array.";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_byteswap(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char *p;
|
|
|
|
int i;
|
1999-12-03 13:15:30 -04:00
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, ":byteswap"))
|
|
|
|
return NULL;
|
|
|
|
|
1993-02-19 11:55:02 -04:00
|
|
|
switch (self->ob_descr->itemsize) {
|
|
|
|
case 1:
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
for (p = self->ob_item, i = self->ob_size; --i >= 0; p += 2) {
|
|
|
|
char p0 = p[0];
|
|
|
|
p[0] = p[1];
|
|
|
|
p[1] = p0;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 4:
|
|
|
|
for (p = self->ob_item, i = self->ob_size; --i >= 0; p += 4) {
|
|
|
|
char p0 = p[0];
|
|
|
|
char p1 = p[1];
|
|
|
|
p[0] = p[3];
|
|
|
|
p[1] = p[2];
|
|
|
|
p[2] = p1;
|
|
|
|
p[3] = p0;
|
|
|
|
}
|
|
|
|
break;
|
1993-11-03 11:01:26 -04:00
|
|
|
case 8:
|
|
|
|
for (p = self->ob_item, i = self->ob_size; --i >= 0; p += 8) {
|
|
|
|
char p0 = p[0];
|
|
|
|
char p1 = p[1];
|
|
|
|
char p2 = p[2];
|
|
|
|
char p3 = p[3];
|
|
|
|
p[0] = p[7];
|
|
|
|
p[1] = p[6];
|
|
|
|
p[2] = p[5];
|
|
|
|
p[3] = p[4];
|
|
|
|
p[4] = p3;
|
|
|
|
p[5] = p2;
|
|
|
|
p[6] = p1;
|
|
|
|
p[7] = p0;
|
|
|
|
}
|
|
|
|
break;
|
1993-02-19 11:55:02 -04:00
|
|
|
default:
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_RuntimeError,
|
1993-02-19 11:55:02 -04:00
|
|
|
"don't know how to byteswap this array type");
|
|
|
|
return NULL;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char byteswap_doc [] =
|
1999-12-03 13:15:30 -04:00
|
|
|
"byteswap()\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
1999-12-03 13:15:30 -04:00
|
|
|
Byteswap all items of the array. If the items in the array are not 1, 2,\n\
|
|
|
|
4, or 8 bytes in size, RuntimeError is raised.";
|
1998-10-13 11:27:22 -03:00
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_reverse(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1993-11-03 11:01:26 -04:00
|
|
|
register int itemsize = self->ob_descr->itemsize;
|
|
|
|
register char *p, *q;
|
2000-09-16 19:31:29 -03:00
|
|
|
/* little buffer to hold items while swapping */
|
|
|
|
char tmp[256]; /* 8 is probably enough -- but why skimp */
|
|
|
|
assert(itemsize <= sizeof(tmp));
|
1993-11-03 11:01:26 -04:00
|
|
|
|
2000-09-16 19:31:29 -03:00
|
|
|
if (!PyArg_ParseTuple(args, ":reverse"))
|
|
|
|
return NULL;
|
1993-02-19 11:55:02 -04:00
|
|
|
|
|
|
|
if (self->ob_size > 1) {
|
1993-11-03 11:01:26 -04:00
|
|
|
for (p = self->ob_item,
|
|
|
|
q = self->ob_item + (self->ob_size - 1)*itemsize;
|
|
|
|
p < q;
|
|
|
|
p += itemsize, q -= itemsize) {
|
2000-09-16 19:31:29 -03:00
|
|
|
/* memory areas guaranteed disjoint, so memcpy
|
|
|
|
* is safe (& memmove may be slower).
|
|
|
|
*/
|
|
|
|
memcpy(tmp, p, itemsize);
|
|
|
|
memcpy(p, q, itemsize);
|
|
|
|
memcpy(q, tmp, itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
}
|
2000-09-10 02:22:54 -03:00
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
1993-11-03 11:01:26 -04:00
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char reverse_doc [] =
|
|
|
|
"reverse()\n\
|
|
|
|
\n\
|
|
|
|
Reverse the order of the items in the array.";
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_fromfile(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *f;
|
1993-02-19 11:55:02 -04:00
|
|
|
int n;
|
|
|
|
FILE *fp;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "Oi:fromfile", &f, &n))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
1996-12-09 16:10:36 -04:00
|
|
|
fp = PyFile_AsFile(f);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (fp == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_TypeError, "arg1 must be open file");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (n > 0) {
|
|
|
|
char *item = self->ob_item;
|
|
|
|
int itemsize = self->ob_descr->itemsize;
|
2000-06-28 18:27:21 -03:00
|
|
|
size_t nread;
|
1999-02-23 14:05:22 -04:00
|
|
|
int newlength;
|
|
|
|
size_t newbytes;
|
|
|
|
/* Be careful here about overflow */
|
|
|
|
if ((newlength = self->ob_size + n) <= 0 ||
|
1999-03-19 17:50:11 -04:00
|
|
|
(newbytes = newlength * itemsize) / itemsize !=
|
|
|
|
(size_t)newlength)
|
1999-02-23 14:05:22 -04:00
|
|
|
goto nomem;
|
|
|
|
PyMem_RESIZE(item, char, newbytes);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (item == NULL) {
|
1999-02-23 14:05:22 -04:00
|
|
|
nomem:
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
self->ob_item = item;
|
|
|
|
self->ob_size += n;
|
|
|
|
nread = fread(item + (self->ob_size - n) * itemsize,
|
|
|
|
itemsize, n, fp);
|
2000-06-28 18:27:21 -03:00
|
|
|
if (nread < (size_t)n) {
|
1993-02-19 11:55:02 -04:00
|
|
|
self->ob_size -= (n - nread);
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char, self->ob_size*itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
self->ob_item = item;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_EOFError,
|
|
|
|
"not enough items in file");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char fromfile_doc [] =
|
|
|
|
"fromfile(f, n)\n\
|
|
|
|
\n\
|
|
|
|
Read n objects from the file object f and append them to the end of the\n\
|
|
|
|
array. Also called as read.";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_tofile(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *f;
|
1993-02-19 11:55:02 -04:00
|
|
|
FILE *fp;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "O:tofile", &f))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
1996-12-09 16:10:36 -04:00
|
|
|
fp = PyFile_AsFile(f);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (fp == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_TypeError, "arg must be open file");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (self->ob_size > 0) {
|
2000-08-12 17:58:11 -03:00
|
|
|
if (fwrite(self->ob_item, self->ob_descr->itemsize,
|
|
|
|
self->ob_size, fp) != (size_t)self->ob_size) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
1993-02-19 11:55:02 -04:00
|
|
|
clearerr(fp);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char tofile_doc [] =
|
|
|
|
"tofile(f)\n\
|
|
|
|
\n\
|
|
|
|
Write all items (as machine values) to the file object f. Also called as\n\
|
|
|
|
write.";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_fromlist(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
int n;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *list;
|
1993-02-19 11:55:02 -04:00
|
|
|
int itemsize = self->ob_descr->itemsize;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "O:fromlist", &list))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (!PyList_Check(list)) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "arg must be list");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
n = PyList_Size(list);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (n > 0) {
|
|
|
|
char *item = self->ob_item;
|
|
|
|
int i;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char, (self->ob_size + n) * itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (item == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
self->ob_item = item;
|
|
|
|
self->ob_size += n;
|
|
|
|
for (i = 0; i < n; i++) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v = PyList_GetItem(list, i);
|
1993-02-19 11:55:02 -04:00
|
|
|
if ((*self->ob_descr->setitem)(self,
|
|
|
|
self->ob_size - n + i, v) != 0) {
|
|
|
|
self->ob_size -= n;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char,
|
|
|
|
self->ob_size * itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
self->ob_item = item;
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char fromlist_doc [] =
|
|
|
|
"fromlist(list)\n\
|
|
|
|
\n\
|
|
|
|
Append items to array from list.";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_tolist(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *list = PyList_New(self->ob_size);
|
1993-02-19 11:55:02 -04:00
|
|
|
int i;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, ":tolist"))
|
|
|
|
return NULL;
|
1993-02-19 11:55:02 -04:00
|
|
|
if (list == NULL)
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v = getarrayitem((PyObject *)self, i);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (v == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_DECREF(list);
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
PyList_SetItem(list, i, v);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
return list;
|
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char tolist_doc [] =
|
1998-10-13 23:52:31 -03:00
|
|
|
"tolist() -> list\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
|
|
|
Convert array to an ordinary list with the same items.";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_fromstring(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char *str;
|
|
|
|
int n;
|
|
|
|
int itemsize = self->ob_descr->itemsize;
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "s#:fromstring", &str, &n))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
if (n % itemsize != 0) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_ValueError,
|
1993-02-19 11:55:02 -04:00
|
|
|
"string length not a multiple of item size");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
n = n / itemsize;
|
|
|
|
if (n > 0) {
|
|
|
|
char *item = self->ob_item;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyMem_RESIZE(item, char, (self->ob_size + n) * itemsize);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (item == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_NoMemory();
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
self->ob_item = item;
|
|
|
|
self->ob_size += n;
|
1996-12-09 18:24:19 -04:00
|
|
|
memcpy(item + (self->ob_size - n) * itemsize,
|
|
|
|
str, itemsize*n);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char fromstring_doc [] =
|
|
|
|
"fromstring(string)\n\
|
|
|
|
\n\
|
|
|
|
Appends items from the string, interpreting it as an array of machine\n\
|
|
|
|
values,as if it had been read from a file using the fromfile() method).";
|
|
|
|
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_tostring(arrayobject *self, PyObject *args)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2000-07-31 18:57:30 -03:00
|
|
|
if (!PyArg_ParseTuple(args, ":tostring"))
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyString_FromStringAndSize(self->ob_item,
|
1993-02-19 11:55:02 -04:00
|
|
|
self->ob_size * self->ob_descr->itemsize);
|
|
|
|
}
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char tostring_doc [] =
|
|
|
|
"tostring() -> string\n\
|
|
|
|
\n\
|
|
|
|
Convert the array to an array of machine values and return the string\n\
|
|
|
|
representation.";
|
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
|
|
|
|
|
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
static PyObject *
|
|
|
|
array_fromunicode(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
Py_UNICODE *ustr;
|
|
|
|
int n;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "u#:fromunicode", &ustr, &n))
|
|
|
|
return NULL;
|
|
|
|
if (self->ob_descr->typecode != 'u') {
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
"fromunicode() may only be called on "
|
|
|
|
"type 'u' arrays");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (n > 0) {
|
|
|
|
Py_UNICODE *item = (Py_UNICODE *) self->ob_item;
|
|
|
|
PyMem_RESIZE(item, Py_UNICODE, self->ob_size + n);
|
|
|
|
if (item == NULL) {
|
|
|
|
PyErr_NoMemory();
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
self->ob_item = (char *) item;
|
|
|
|
self->ob_size += n;
|
|
|
|
memcpy(item + self->ob_size - n,
|
|
|
|
ustr, n * sizeof(Py_UNICODE));
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char fromunicode_doc[] =
|
|
|
|
"fromunicode(ustr)\n\
|
|
|
|
\n\
|
|
|
|
Extends this array with data from the unicode string ustr.\n\
|
|
|
|
The array must be a type 'u' array; otherwise a ValueError\n\
|
|
|
|
is raised. Use array.fromstring(ustr.decode(...)) to\n\
|
|
|
|
append Unicode data to an array of some other type.";
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_tounicode(arrayobject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
if (!PyArg_ParseTuple(args, ":tounicode"))
|
|
|
|
return NULL;
|
|
|
|
if (self->ob_descr->typecode != 'u') {
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
"tounicode() may only be called on type 'u' arrays");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return PyUnicode_FromUnicode((Py_UNICODE *) self->ob_item, self->ob_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
static char tounicode_doc [] =
|
|
|
|
"tounicode() -> unicode\n\
|
|
|
|
\n\
|
|
|
|
Convert the array to a unicode string. The array must be\n\
|
|
|
|
a type 'u' array; otherwise a ValueError is raised. Use\n\
|
|
|
|
array.tostring().decode() to obtain a unicode string from\n\
|
|
|
|
an array of some other type.";
|
|
|
|
|
|
|
|
#endif /* Py_USING_UNICODE */
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_get_typecode(arrayobject *a, void *closure)
|
|
|
|
{
|
|
|
|
char tc = a->ob_descr->typecode;
|
|
|
|
return PyString_FromStringAndSize(&tc, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
array_get_itemsize(arrayobject *a, void *closure)
|
|
|
|
{
|
|
|
|
return PyInt_FromLong((long)a->ob_descr->itemsize);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyGetSetDef array_getsets [] = {
|
|
|
|
{"typecode", (getter) array_get_typecode, NULL,
|
|
|
|
"the typecode character used to create the array"},
|
|
|
|
{"itemsize", (getter) array_get_itemsize, NULL,
|
|
|
|
"the size, in bytes, of one array item"},
|
|
|
|
{NULL}
|
|
|
|
};
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
PyMethodDef array_methods[] = {
|
2001-01-17 21:02:55 -04:00
|
|
|
{"append", (PyCFunction)array_append, METH_VARARGS,
|
|
|
|
append_doc},
|
|
|
|
{"buffer_info", (PyCFunction)array_buffer_info, METH_VARARGS,
|
|
|
|
buffer_info_doc},
|
|
|
|
{"byteswap", (PyCFunction)array_byteswap, METH_VARARGS,
|
|
|
|
byteswap_doc},
|
|
|
|
{"count", (PyCFunction)array_count, METH_VARARGS,
|
|
|
|
count_doc},
|
|
|
|
{"extend", (PyCFunction)array_extend, METH_VARARGS,
|
|
|
|
extend_doc},
|
|
|
|
{"fromfile", (PyCFunction)array_fromfile, METH_VARARGS,
|
|
|
|
fromfile_doc},
|
|
|
|
{"fromlist", (PyCFunction)array_fromlist, METH_VARARGS,
|
|
|
|
fromlist_doc},
|
|
|
|
{"fromstring", (PyCFunction)array_fromstring, METH_VARARGS,
|
|
|
|
fromstring_doc},
|
2002-03-01 06:27:01 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
{"fromunicode", (PyCFunction)array_fromunicode, METH_VARARGS,
|
|
|
|
fromunicode_doc},
|
|
|
|
#endif
|
2001-01-17 21:02:55 -04:00
|
|
|
{"index", (PyCFunction)array_index, METH_VARARGS,
|
|
|
|
index_doc},
|
|
|
|
{"insert", (PyCFunction)array_insert, METH_VARARGS,
|
|
|
|
insert_doc},
|
|
|
|
{"pop", (PyCFunction)array_pop, METH_VARARGS,
|
|
|
|
pop_doc},
|
|
|
|
{"read", (PyCFunction)array_fromfile, METH_VARARGS,
|
|
|
|
fromfile_doc},
|
|
|
|
{"remove", (PyCFunction)array_remove, METH_VARARGS,
|
|
|
|
remove_doc},
|
|
|
|
{"reverse", (PyCFunction)array_reverse, METH_VARARGS,
|
|
|
|
reverse_doc},
|
|
|
|
/* {"sort", (PyCFunction)array_sort, METH_VARARGS,
|
|
|
|
sort_doc},*/
|
|
|
|
{"tofile", (PyCFunction)array_tofile, METH_VARARGS,
|
|
|
|
tofile_doc},
|
|
|
|
{"tolist", (PyCFunction)array_tolist, METH_VARARGS,
|
|
|
|
tolist_doc},
|
|
|
|
{"tostring", (PyCFunction)array_tostring, METH_VARARGS,
|
|
|
|
tostring_doc},
|
2002-03-01 06:27:01 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
{"tounicode", (PyCFunction)array_tounicode, METH_VARARGS,
|
|
|
|
tounicode_doc},
|
|
|
|
#endif
|
2001-01-17 21:02:55 -04:00
|
|
|
{"write", (PyCFunction)array_tofile, METH_VARARGS,
|
|
|
|
tofile_doc},
|
1993-02-19 11:55:02 -04:00
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
};
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2000-07-13 18:10:57 -03:00
|
|
|
array_repr(arrayobject *a)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2002-03-01 06:27:01 -04:00
|
|
|
char buf[256], typecode;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *s, *t, *comma, *v;
|
1993-02-19 11:55:02 -04:00
|
|
|
int i, len;
|
2002-03-01 06:27:01 -04:00
|
|
|
|
1993-02-19 11:55:02 -04:00
|
|
|
len = a->ob_size;
|
2002-03-01 06:27:01 -04:00
|
|
|
typecode = a->ob_descr->typecode;
|
1993-02-19 11:55:02 -04:00
|
|
|
if (len == 0) {
|
2002-03-01 06:27:01 -04:00
|
|
|
PyOS_snprintf(buf, sizeof(buf), "array('%c')", typecode);
|
1996-12-09 16:10:36 -04:00
|
|
|
return PyString_FromString(buf);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
|
|
|
|
if (typecode == 'c' || typecode == 'u') {
|
2000-11-10 15:04:19 -04:00
|
|
|
PyObject *t_empty = PyTuple_New(0);
|
2002-03-01 06:27:01 -04:00
|
|
|
PyOS_snprintf(buf, sizeof(buf), "array('%c', ", typecode);
|
1996-12-09 16:10:36 -04:00
|
|
|
s = PyString_FromString(buf);
|
2002-03-01 06:27:01 -04:00
|
|
|
if (typecode == 'c')
|
|
|
|
v = array_tostring(a, t_empty);
|
|
|
|
else
|
|
|
|
v = array_tounicode(a, t_empty);
|
2000-11-10 15:04:19 -04:00
|
|
|
Py_DECREF(t_empty);
|
1996-12-09 16:10:36 -04:00
|
|
|
t = PyObject_Repr(v);
|
|
|
|
Py_XDECREF(v);
|
|
|
|
PyString_ConcatAndDel(&s, t);
|
|
|
|
PyString_ConcatAndDel(&s, PyString_FromString(")"));
|
1993-02-19 11:55:02 -04:00
|
|
|
return s;
|
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
PyOS_snprintf(buf, sizeof(buf), "array('%c', [", typecode);
|
1996-12-09 16:10:36 -04:00
|
|
|
s = PyString_FromString(buf);
|
|
|
|
comma = PyString_FromString(", ");
|
|
|
|
for (i = 0; i < len && !PyErr_Occurred(); i++) {
|
1994-08-01 08:34:53 -03:00
|
|
|
if (i > 0)
|
1996-12-09 16:10:36 -04:00
|
|
|
PyString_Concat(&s, comma);
|
1993-02-19 11:55:02 -04:00
|
|
|
v = (a->ob_descr->getitem)(a, i);
|
1996-12-09 16:10:36 -04:00
|
|
|
t = PyObject_Repr(v);
|
|
|
|
Py_XDECREF(v);
|
|
|
|
PyString_ConcatAndDel(&s, t);
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_XDECREF(comma);
|
|
|
|
PyString_ConcatAndDel(&s, PyString_FromString("])"));
|
1993-02-19 11:55:02 -04:00
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
1997-05-05 19:15:02 -03:00
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_buffer_getreadbuf(arrayobject *self, int index, const void **ptr)
|
1997-05-05 19:15:02 -03:00
|
|
|
{
|
|
|
|
if ( index != 0 ) {
|
1998-10-13 11:27:22 -03:00
|
|
|
PyErr_SetString(PyExc_SystemError,
|
|
|
|
"Accessing non-existent array segment");
|
1997-05-05 19:15:02 -03:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
*ptr = (void *)self->ob_item;
|
|
|
|
return self->ob_size*self->ob_descr->itemsize;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_buffer_getwritebuf(arrayobject *self, int index, const void **ptr)
|
1997-05-05 19:15:02 -03:00
|
|
|
{
|
|
|
|
if ( index != 0 ) {
|
1998-10-13 11:27:22 -03:00
|
|
|
PyErr_SetString(PyExc_SystemError,
|
|
|
|
"Accessing non-existent array segment");
|
1997-05-05 19:15:02 -03:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
*ptr = (void *)self->ob_item;
|
|
|
|
return self->ob_size*self->ob_descr->itemsize;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-13 18:10:57 -03:00
|
|
|
array_buffer_getsegcount(arrayobject *self, int *lenp)
|
1997-05-05 19:15:02 -03:00
|
|
|
{
|
|
|
|
if ( lenp )
|
|
|
|
*lenp = self->ob_size*self->ob_descr->itemsize;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PySequenceMethods array_as_sequence = {
|
1996-12-09 18:24:19 -04:00
|
|
|
(inquiry)array_length, /*sq_length*/
|
|
|
|
(binaryfunc)array_concat, /*sq_concat*/
|
1994-08-01 08:34:53 -03:00
|
|
|
(intargfunc)array_repeat, /*sq_repeat*/
|
1996-12-09 18:24:19 -04:00
|
|
|
(intargfunc)array_item, /*sq_item*/
|
1994-08-01 08:34:53 -03:00
|
|
|
(intintargfunc)array_slice, /*sq_slice*/
|
|
|
|
(intobjargproc)array_ass_item, /*sq_ass_item*/
|
|
|
|
(intintobjargproc)array_ass_slice, /*sq_ass_slice*/
|
2002-03-01 06:27:01 -04:00
|
|
|
NULL, /*sq_contains*/
|
|
|
|
(binaryfunc)array_inplace_concat, /*sq_inplace_concat*/
|
|
|
|
(intargfunc)array_inplace_repeat /*sq_inplace_repeat*/
|
1993-02-19 11:55:02 -04:00
|
|
|
};
|
|
|
|
|
1997-05-05 19:15:02 -03:00
|
|
|
static PyBufferProcs array_as_buffer = {
|
|
|
|
(getreadbufferproc)array_buffer_getreadbuf,
|
|
|
|
(getwritebufferproc)array_buffer_getwritebuf,
|
|
|
|
(getsegcountproc)array_buffer_getsegcount,
|
|
|
|
};
|
|
|
|
|
1996-12-09 16:10:36 -04:00
|
|
|
static PyObject *
|
2002-03-01 06:27:01 -04:00
|
|
|
array_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
|
|
|
char c;
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *initial = NULL;
|
1993-02-19 11:55:02 -04:00
|
|
|
struct arraydescr *descr;
|
2002-03-01 06:27:01 -04:00
|
|
|
|
|
|
|
if (kwds != NULL) {
|
|
|
|
int i = PyObject_Length(kwds);
|
|
|
|
if (i < 0)
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
2002-03-01 06:27:01 -04:00
|
|
|
else if (i > 0) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyErr_SetString(PyExc_TypeError,
|
2002-03-01 06:27:01 -04:00
|
|
|
"array.array constructor takes "
|
|
|
|
"no keyword arguments");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
2002-03-01 06:27:01 -04:00
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "c|O:array", &c, &initial))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
if (!(initial == NULL || PyList_Check(initial)
|
|
|
|
|| PyString_Check(initial)
|
|
|
|
|| (c == 'u' && PyUnicode_Check(initial)))) {
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
"array initializer must be list or string");
|
|
|
|
return NULL;
|
|
|
|
}
|
1993-02-19 11:55:02 -04:00
|
|
|
for (descr = descriptors; descr->typecode != '\0'; descr++) {
|
|
|
|
if (descr->typecode == c) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *a;
|
1993-02-19 11:55:02 -04:00
|
|
|
int len;
|
1996-12-09 16:10:36 -04:00
|
|
|
if (initial == NULL || !PyList_Check(initial))
|
1993-02-19 11:55:02 -04:00
|
|
|
len = 0;
|
|
|
|
else
|
1996-12-09 16:10:36 -04:00
|
|
|
len = PyList_Size(initial);
|
2002-03-01 06:27:01 -04:00
|
|
|
|
|
|
|
a = newarrayobject(type, len, descr);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (a == NULL)
|
|
|
|
return NULL;
|
2002-03-01 06:27:01 -04:00
|
|
|
|
1993-02-19 11:55:02 -04:00
|
|
|
if (len > 0) {
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < len; i++) {
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v =
|
2002-03-01 06:27:01 -04:00
|
|
|
PyList_GetItem(initial, i);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (setarrayitem(a, i, v) != 0) {
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_DECREF(a);
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
if (initial != NULL && PyString_Check(initial)) {
|
2001-01-17 21:02:55 -04:00
|
|
|
PyObject *t_initial = Py_BuildValue("(O)",
|
|
|
|
initial);
|
1996-12-09 16:10:36 -04:00
|
|
|
PyObject *v =
|
2001-01-17 21:02:55 -04:00
|
|
|
array_fromstring((arrayobject *)a,
|
|
|
|
t_initial);
|
2002-03-01 06:27:01 -04:00
|
|
|
Py_DECREF(t_initial);
|
1993-02-19 11:55:02 -04:00
|
|
|
if (v == NULL) {
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_DECREF(a);
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
1996-12-09 16:10:36 -04:00
|
|
|
Py_DECREF(v);
|
2002-03-01 06:27:01 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
} else if (initial != NULL && PyUnicode_Check(initial)) {
|
|
|
|
int n = PyUnicode_GET_DATA_SIZE(initial);
|
|
|
|
if (n > 0) {
|
|
|
|
arrayobject *self = (arrayobject *)a;
|
|
|
|
char *item = self->ob_item;
|
|
|
|
item = PyMem_Realloc(item, n);
|
|
|
|
if (item == NULL) {
|
|
|
|
PyErr_NoMemory();
|
|
|
|
Py_DECREF(a);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
self->ob_item = item;
|
|
|
|
self->ob_size = n / sizeof(Py_UNICODE);
|
|
|
|
memcpy(item, PyUnicode_AS_DATA(initial), n);
|
|
|
|
}
|
|
|
|
#endif
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
}
|
1996-12-09 18:24:19 -04:00
|
|
|
PyErr_SetString(PyExc_ValueError,
|
2002-03-01 06:27:01 -04:00
|
|
|
"bad typecode (must be c, b, B, u, h, H, i, I, l, L, f or d)");
|
1993-02-19 11:55:02 -04:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-10-13 11:27:22 -03:00
|
|
|
static char module_doc [] =
|
2002-03-01 06:27:01 -04:00
|
|
|
"This module defines an object type which can efficiently represent\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
an array of basic values: characters, integers, floating point\n\
|
|
|
|
numbers. Arrays are sequence types and behave very much like lists,\n\
|
|
|
|
except that the type of objects stored in them is constrained. The\n\
|
|
|
|
type is specified at object creation time by using a type code, which\n\
|
|
|
|
is a single character. The following type codes are defined:\n\
|
|
|
|
\n\
|
|
|
|
Type code C Type Minimum size in bytes \n\
|
|
|
|
'c' character 1 \n\
|
|
|
|
'b' signed integer 1 \n\
|
|
|
|
'B' unsigned integer 1 \n\
|
2002-03-01 06:27:01 -04:00
|
|
|
'u' Unicode character 2 \n\
|
1998-10-13 11:27:22 -03:00
|
|
|
'h' signed integer 2 \n\
|
|
|
|
'H' unsigned integer 2 \n\
|
|
|
|
'i' signed integer 2 \n\
|
|
|
|
'I' unsigned integer 2 \n\
|
|
|
|
'l' signed integer 4 \n\
|
|
|
|
'L' unsigned integer 4 \n\
|
|
|
|
'f' floating point 4 \n\
|
|
|
|
'd' floating point 8 \n\
|
|
|
|
\n\
|
2002-03-01 06:27:01 -04:00
|
|
|
The constructor is:\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
|
|
|
array(typecode [, initializer]) -- create a new array\n\
|
|
|
|
";
|
|
|
|
|
|
|
|
static char arraytype_doc [] =
|
2002-03-01 06:27:01 -04:00
|
|
|
"array(typecode [, initializer]) -> array\n\
|
|
|
|
\n\
|
|
|
|
Return a new array whose items are restricted by typecode, and\n\
|
|
|
|
initialized from the optional initializer value, which must be a list\n\
|
|
|
|
or a string.\n\
|
|
|
|
\n\
|
|
|
|
Arrays represent basic values and behave very much like lists, except\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
the type of objects stored in them is constrained.\n\
|
|
|
|
\n\
|
|
|
|
Methods:\n\
|
|
|
|
\n\
|
|
|
|
append() -- append a new item to the end of the array\n\
|
|
|
|
buffer_info() -- return information giving the current memory info\n\
|
|
|
|
byteswap() -- byteswap all the items of the array\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
count() -- return number of occurences of an object\n\
|
|
|
|
extend() -- extend array by appending array elements\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
fromfile() -- read items from a file object\n\
|
|
|
|
fromlist() -- append items from the list\n\
|
|
|
|
fromstring() -- append items from the string\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
index() -- return index of first occurence of an object\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
insert() -- insert a new item into the array at a provided position\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
pop() -- remove and return item (default last)\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
read() -- DEPRECATED, use fromfile()\n\
|
2000-07-31 17:52:21 -03:00
|
|
|
remove() -- remove first occurence of an object\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
reverse() -- reverse the order of the items in the array\n\
|
|
|
|
tofile() -- write all items to a file object\n\
|
|
|
|
tolist() -- return the array converted to an ordinary list\n\
|
|
|
|
tostring() -- return the array converted to a string\n\
|
|
|
|
write() -- DEPRECATED, use tofile()\n\
|
|
|
|
\n\
|
2002-03-01 06:27:01 -04:00
|
|
|
Attributes:\n\
|
1998-10-13 11:27:22 -03:00
|
|
|
\n\
|
|
|
|
typecode -- the typecode character used to create the array\n\
|
|
|
|
itemsize -- the length in bytes of one array item\n\
|
|
|
|
";
|
|
|
|
|
|
|
|
statichere PyTypeObject Arraytype = {
|
2000-02-04 16:33:49 -04:00
|
|
|
PyObject_HEAD_INIT(NULL)
|
1998-10-13 11:27:22 -03:00
|
|
|
0,
|
2001-12-08 14:02:58 -04:00
|
|
|
"array.array",
|
1998-10-13 11:27:22 -03:00
|
|
|
sizeof(arrayobject),
|
|
|
|
0,
|
2001-01-17 21:02:55 -04:00
|
|
|
(destructor)array_dealloc, /* tp_dealloc */
|
2002-03-04 05:38:52 -04:00
|
|
|
0, /* tp_print */
|
2002-03-01 06:27:01 -04:00
|
|
|
0, /* tp_getattr */
|
2001-01-17 21:02:55 -04:00
|
|
|
0, /* tp_setattr */
|
|
|
|
0, /* tp_compare */
|
|
|
|
(reprfunc)array_repr, /* tp_repr */
|
|
|
|
0, /* tp_as _number*/
|
|
|
|
&array_as_sequence, /* tp_as _sequence*/
|
|
|
|
0, /* tp_as _mapping*/
|
|
|
|
0, /* tp_hash */
|
|
|
|
0, /* tp_call */
|
|
|
|
0, /* tp_str */
|
2002-03-01 06:27:01 -04:00
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
2001-01-17 21:02:55 -04:00
|
|
|
0, /* tp_setattro */
|
2001-06-05 01:43:26 -03:00
|
|
|
&array_as_buffer, /* tp_as_buffer*/
|
2002-03-01 06:27:01 -04:00
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
|
2001-01-17 21:02:55 -04:00
|
|
|
arraytype_doc, /* tp_doc */
|
|
|
|
0, /* tp_traverse */
|
|
|
|
0, /* tp_clear */
|
|
|
|
array_richcompare, /* tp_richcompare */
|
2002-03-01 06:27:01 -04:00
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
0, /* tp_iter */
|
|
|
|
0, /* tp_iternext */
|
|
|
|
array_methods, /* tp_methods */
|
|
|
|
0, /* tp_members */
|
|
|
|
array_getsets, /* tp_getset */
|
|
|
|
0, /* tp_base */
|
|
|
|
0, /* tp_dict */
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
0, /* tp_descr_set */
|
|
|
|
0, /* tp_dictoffset */
|
|
|
|
0, /* tp_init */
|
|
|
|
PyType_GenericAlloc, /* tp_alloc */
|
|
|
|
array_new, /* tp_new */
|
|
|
|
_PyObject_Del, /* tp_free */
|
1998-10-13 11:27:22 -03:00
|
|
|
};
|
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
/* No functions in array module. */
|
|
|
|
static PyMethodDef a_methods[] = {
|
|
|
|
{NULL, NULL, 0, NULL} /* Sentinel */
|
|
|
|
};
|
|
|
|
|
|
|
|
|
1998-12-04 14:50:17 -04:00
|
|
|
DL_EXPORT(void)
|
2000-07-21 03:00:07 -03:00
|
|
|
initarray(void)
|
1993-02-19 11:55:02 -04:00
|
|
|
{
|
2002-03-31 23:45:06 -04:00
|
|
|
PyObject *m;
|
2000-02-04 16:33:49 -04:00
|
|
|
|
2002-03-01 06:27:01 -04:00
|
|
|
Arraytype.ob_type = &PyType_Type;
|
1998-10-13 11:27:22 -03:00
|
|
|
m = Py_InitModule3("array", a_methods, module_doc);
|
2002-03-31 23:45:06 -04:00
|
|
|
|
|
|
|
Py_INCREF((PyObject *)&Arraytype);
|
|
|
|
PyModule_AddObject(m, "ArrayType", (PyObject *)&Arraytype);
|
|
|
|
Py_INCREF((PyObject *)&Arraytype);
|
|
|
|
PyModule_AddObject(m, "array", (PyObject *)&Arraytype);
|
1998-10-14 10:45:06 -03:00
|
|
|
/* No need to check the error here, the caller will do that */
|
1993-02-19 11:55:02 -04:00
|
|
|
}
|