1991-02-19 12:39:46 +00:00
|
|
|
/***********************************************************
|
2000-06-30 23:50:40 +00:00
|
|
|
Copyright (c) 2000, BeOpen.com.
|
|
|
|
Copyright (c) 1995-2000, Corporation for National Research Initiatives.
|
|
|
|
Copyright (c) 1990-1995, Stichting Mathematisch Centrum.
|
|
|
|
All rights reserved.
|
|
|
|
|
|
|
|
See the file "Misc/COPYRIGHT" for information on usage and
|
|
|
|
redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
|
1991-02-19 12:39:46 +00:00
|
|
|
******************************************************************/
|
|
|
|
|
1990-10-14 12:07:46 +00:00
|
|
|
/* List object implementation */
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
#include "Python.h"
|
|
|
|
|
1994-08-29 12:45:32 +00:00
|
|
|
#ifdef STDC_HEADERS
|
|
|
|
#include <stddef.h>
|
|
|
|
#else
|
|
|
|
#include <sys/types.h> /* For size_t */
|
|
|
|
#endif
|
1990-10-14 12:07:46 +00:00
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
#define ROUNDUP(n, PyTryBlock) \
|
|
|
|
((((n)+(PyTryBlock)-1)/(PyTryBlock))*(PyTryBlock))
|
1995-01-26 22:59:43 +00:00
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
roundupsize(int n)
|
1995-01-26 22:59:43 +00:00
|
|
|
{
|
|
|
|
if (n < 500)
|
|
|
|
return ROUNDUP(n, 10);
|
|
|
|
else
|
|
|
|
return ROUNDUP(n, 100);
|
|
|
|
}
|
|
|
|
|
1997-08-25 18:36:23 +00:00
|
|
|
#define NRESIZE(var, type, nitems) PyMem_RESIZE(var, type, roundupsize(nitems))
|
1995-01-26 22:59:43 +00:00
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_New(int size)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
1997-05-02 03:12:38 +00:00
|
|
|
PyListObject *op;
|
1994-08-29 12:45:32 +00:00
|
|
|
size_t nbytes;
|
1990-10-14 12:07:46 +00:00
|
|
|
if (size < 0) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_BadInternalCall();
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
nbytes = size * sizeof(PyObject *);
|
1992-08-19 16:46:30 +00:00
|
|
|
/* Check for overflow */
|
1997-05-02 03:12:38 +00:00
|
|
|
if (nbytes / sizeof(PyObject *) != (size_t)size) {
|
|
|
|
return PyErr_NoMemory();
|
1992-08-19 16:46:30 +00:00
|
|
|
}
|
2000-05-03 23:44:39 +00:00
|
|
|
/* PyObject_NewVar is inlined */
|
2000-06-23 19:37:02 +00:00
|
|
|
op = (PyListObject *) PyObject_MALLOC(sizeof(PyListObject)
|
2000-06-30 05:02:53 +00:00
|
|
|
+ PyGC_HEAD_SIZE);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (op == NULL) {
|
1997-05-02 03:12:38 +00:00
|
|
|
return PyErr_NoMemory();
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
2000-06-30 05:02:53 +00:00
|
|
|
op = (PyListObject *) PyObject_FROM_GC(op);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (size <= 0) {
|
|
|
|
op->ob_item = NULL;
|
|
|
|
}
|
|
|
|
else {
|
2000-05-03 23:44:39 +00:00
|
|
|
op->ob_item = (PyObject **) PyMem_MALLOC(nbytes);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (op->ob_item == NULL) {
|
2000-07-15 03:31:31 +00:00
|
|
|
PyObject_FREE(PyObject_AS_GC(op));
|
1997-05-02 03:12:38 +00:00
|
|
|
return PyErr_NoMemory();
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
}
|
2000-05-03 23:44:39 +00:00
|
|
|
PyObject_INIT_VAR(op, &PyList_Type, size);
|
1990-10-14 12:07:46 +00:00
|
|
|
for (i = 0; i < size; i++)
|
|
|
|
op->ob_item[i] = NULL;
|
2000-06-30 05:02:53 +00:00
|
|
|
PyObject_GC_Init(op);
|
1997-05-02 03:12:38 +00:00
|
|
|
return (PyObject *) op;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_Size(PyObject *op)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(op)) {
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-14 12:07:46 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
else
|
1997-05-02 03:12:38 +00:00
|
|
|
return ((PyListObject *)op) -> ob_size;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *indexerr;
|
1996-08-09 20:51:27 +00:00
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_GetItem(PyObject *op, int i)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(op)) {
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
|
1996-08-09 20:51:27 +00:00
|
|
|
if (indexerr == NULL)
|
1997-05-02 03:12:38 +00:00
|
|
|
indexerr = PyString_FromString(
|
|
|
|
"list index out of range");
|
|
|
|
PyErr_SetObject(PyExc_IndexError, indexerr);
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return ((PyListObject *)op) -> ob_item[i];
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_SetItem(register PyObject *op, register int i,
|
|
|
|
register PyObject *newitem)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
register PyObject *olditem;
|
|
|
|
register PyObject **p;
|
|
|
|
if (!PyList_Check(op)) {
|
|
|
|
Py_XDECREF(newitem);
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
|
|
|
|
Py_XDECREF(newitem);
|
|
|
|
PyErr_SetString(PyExc_IndexError,
|
|
|
|
"list assignment index out of range");
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
p = ((PyListObject *)op) -> ob_item + i;
|
1995-03-09 12:12:50 +00:00
|
|
|
olditem = *p;
|
|
|
|
*p = newitem;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_XDECREF(olditem);
|
1990-10-14 12:07:46 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
ins1(PyListObject *self, int where, PyObject *v)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject **items;
|
1990-10-21 22:15:08 +00:00
|
|
|
if (v == NULL) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_BadInternalCall();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
items = self->ob_item;
|
1997-05-02 03:12:38 +00:00
|
|
|
NRESIZE(items, PyObject *, self->ob_size+1);
|
1990-10-21 22:15:08 +00:00
|
|
|
if (items == NULL) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_NoMemory();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
if (where < 0)
|
|
|
|
where = 0;
|
|
|
|
if (where > self->ob_size)
|
|
|
|
where = self->ob_size;
|
|
|
|
for (i = self->ob_size; --i >= where; )
|
|
|
|
items[i+1] = items[i];
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(v);
|
1990-10-14 12:07:46 +00:00
|
|
|
items[where] = v;
|
|
|
|
self->ob_item = items;
|
|
|
|
self->ob_size++;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_Insert(PyObject *op, int where, PyObject *newitem)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(op)) {
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return ins1((PyListObject *)op, where, newitem);
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_Append(PyObject *op, PyObject *newitem)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(op)) {
|
|
|
|
PyErr_BadInternalCall();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return ins1((PyListObject *)op,
|
|
|
|
(int) ((PyListObject *)op)->ob_size, newitem);
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Methods */
|
|
|
|
|
|
|
|
static void
|
2000-07-09 15:16:51 +00:00
|
|
|
list_dealloc(PyListObject *op)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
2000-03-13 16:01:29 +00:00
|
|
|
Py_TRASHCAN_SAFE_BEGIN(op)
|
2000-06-30 05:02:53 +00:00
|
|
|
PyObject_GC_Fini(op);
|
1995-01-19 12:09:27 +00:00
|
|
|
if (op->ob_item != NULL) {
|
1999-06-09 15:19:34 +00:00
|
|
|
/* Do it backwards, for Christian Tismer.
|
|
|
|
There's a simple test case where somehow this reduces
|
|
|
|
thrashing when a *very* large list is created and
|
|
|
|
immediately deleted. */
|
|
|
|
i = op->ob_size;
|
|
|
|
while (--i >= 0) {
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_XDECREF(op->ob_item[i]);
|
1995-01-19 12:09:27 +00:00
|
|
|
}
|
2000-05-03 23:44:39 +00:00
|
|
|
PyMem_FREE(op->ob_item);
|
1995-01-19 12:09:27 +00:00
|
|
|
}
|
2000-07-01 01:00:38 +00:00
|
|
|
op = (PyListObject *) PyObject_AS_GC(op);
|
2000-05-03 23:44:39 +00:00
|
|
|
PyObject_DEL(op);
|
2000-03-13 16:01:29 +00:00
|
|
|
Py_TRASHCAN_SAFE_END(op)
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1991-06-07 16:10:43 +00:00
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_print(PyListObject *op, FILE *fp, int flags)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
1998-04-10 22:47:27 +00:00
|
|
|
|
|
|
|
i = Py_ReprEnter((PyObject*)op);
|
|
|
|
if (i != 0) {
|
|
|
|
if (i < 0)
|
|
|
|
return i;
|
|
|
|
fprintf(fp, "[...]");
|
|
|
|
return 0;
|
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
fprintf(fp, "[");
|
1991-06-07 16:10:43 +00:00
|
|
|
for (i = 0; i < op->ob_size; i++) {
|
|
|
|
if (i > 0)
|
1990-10-14 12:07:46 +00:00
|
|
|
fprintf(fp, ", ");
|
1998-04-10 22:47:27 +00:00
|
|
|
if (PyObject_Print(op->ob_item[i], fp, 0) != 0) {
|
|
|
|
Py_ReprLeave((PyObject *)op);
|
1991-06-07 16:10:43 +00:00
|
|
|
return -1;
|
1998-04-10 22:47:27 +00:00
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
fprintf(fp, "]");
|
1998-04-10 22:47:27 +00:00
|
|
|
Py_ReprLeave((PyObject *)op);
|
1991-06-07 16:10:43 +00:00
|
|
|
return 0;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_repr(PyListObject *v)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *s, *comma;
|
1990-10-14 12:07:46 +00:00
|
|
|
int i;
|
1998-04-10 22:47:27 +00:00
|
|
|
|
|
|
|
i = Py_ReprEnter((PyObject*)v);
|
|
|
|
if (i != 0) {
|
|
|
|
if (i > 0)
|
|
|
|
return PyString_FromString("[...]");
|
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
s = PyString_FromString("[");
|
|
|
|
comma = PyString_FromString(", ");
|
1990-10-14 12:07:46 +00:00
|
|
|
for (i = 0; i < v->ob_size && s != NULL; i++) {
|
|
|
|
if (i > 0)
|
1997-05-02 03:12:38 +00:00
|
|
|
PyString_Concat(&s, comma);
|
|
|
|
PyString_ConcatAndDel(&s, PyObject_Repr(v->ob_item[i]));
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_XDECREF(comma);
|
|
|
|
PyString_ConcatAndDel(&s, PyString_FromString("]"));
|
1998-04-10 22:47:27 +00:00
|
|
|
Py_ReprLeave((PyObject *)v);
|
1990-10-14 12:07:46 +00:00
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_compare(PyListObject *v, PyListObject *w)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
1998-05-28 20:18:46 +00:00
|
|
|
for (i = 0; i < v->ob_size && i < w->ob_size; i++) {
|
1997-05-02 03:12:38 +00:00
|
|
|
int cmp = PyObject_Compare(v->ob_item[i], w->ob_item[i]);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (cmp != 0)
|
|
|
|
return cmp;
|
|
|
|
}
|
|
|
|
return v->ob_size - w->ob_size;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_length(PyListObject *a)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
return a->ob_size;
|
|
|
|
}
|
|
|
|
|
2000-04-27 21:41:03 +00:00
|
|
|
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_contains(PyListObject *a, PyObject *el)
|
2000-04-27 21:41:03 +00:00
|
|
|
{
|
|
|
|
int i, cmp;
|
|
|
|
|
|
|
|
for (i = 0; i < a->ob_size; ++i) {
|
|
|
|
cmp = PyObject_Compare(el, PyList_GET_ITEM(a, i));
|
|
|
|
if (cmp == 0)
|
|
|
|
return 1;
|
|
|
|
if (PyErr_Occurred())
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_item(PyListObject *a, int i)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1996-08-09 20:51:27 +00:00
|
|
|
if (indexerr == NULL)
|
1997-05-02 03:12:38 +00:00
|
|
|
indexerr = PyString_FromString(
|
|
|
|
"list index out of range");
|
|
|
|
PyErr_SetObject(PyExc_IndexError, indexerr);
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(a->ob_item[i]);
|
1990-10-14 12:07:46 +00:00
|
|
|
return a->ob_item[i];
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_slice(PyListObject *a, int ilow, int ihigh)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyListObject *np;
|
1990-10-14 12:07:46 +00:00
|
|
|
int i;
|
|
|
|
if (ilow < 0)
|
|
|
|
ilow = 0;
|
|
|
|
else if (ilow > a->ob_size)
|
|
|
|
ilow = a->ob_size;
|
|
|
|
if (ihigh < ilow)
|
|
|
|
ihigh = ilow;
|
|
|
|
else if (ihigh > a->ob_size)
|
|
|
|
ihigh = a->ob_size;
|
1997-05-02 03:12:38 +00:00
|
|
|
np = (PyListObject *) PyList_New(ihigh - ilow);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (np == NULL)
|
|
|
|
return NULL;
|
|
|
|
for (i = ilow; i < ihigh; i++) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *v = a->ob_item[i];
|
|
|
|
Py_INCREF(v);
|
1990-10-14 12:07:46 +00:00
|
|
|
np->ob_item[i - ilow] = v;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return (PyObject *)np;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_GetSlice(PyObject *a, int ilow, int ihigh)
|
1993-06-17 12:35:49 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(a)) {
|
|
|
|
PyErr_BadInternalCall();
|
1993-06-17 12:35:49 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return list_slice((PyListObject *)a, ilow, ihigh);
|
1993-06-17 12:35:49 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_concat(PyListObject *a, PyObject *bb)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int size;
|
|
|
|
int i;
|
1997-05-02 03:12:38 +00:00
|
|
|
PyListObject *np;
|
|
|
|
if (!PyList_Check(bb)) {
|
2000-06-01 03:12:13 +00:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
2000-06-01 14:31:03 +00:00
|
|
|
"can only concatenate list (not \"%.200s\") to list",
|
|
|
|
bb->ob_type->tp_name);
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
#define b ((PyListObject *)bb)
|
1990-10-14 12:07:46 +00:00
|
|
|
size = a->ob_size + b->ob_size;
|
1997-05-02 03:12:38 +00:00
|
|
|
np = (PyListObject *) PyList_New(size);
|
1990-10-14 12:07:46 +00:00
|
|
|
if (np == NULL) {
|
1991-06-07 16:10:43 +00:00
|
|
|
return NULL;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
for (i = 0; i < a->ob_size; i++) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *v = a->ob_item[i];
|
|
|
|
Py_INCREF(v);
|
1990-10-14 12:07:46 +00:00
|
|
|
np->ob_item[i] = v;
|
|
|
|
}
|
|
|
|
for (i = 0; i < b->ob_size; i++) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *v = b->ob_item[i];
|
|
|
|
Py_INCREF(v);
|
1990-10-14 12:07:46 +00:00
|
|
|
np->ob_item[i + a->ob_size] = v;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return (PyObject *)np;
|
1990-10-14 12:07:46 +00:00
|
|
|
#undef b
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_repeat(PyListObject *a, int n)
|
1991-03-06 13:07:53 +00:00
|
|
|
{
|
|
|
|
int i, j;
|
|
|
|
int size;
|
1997-05-02 03:12:38 +00:00
|
|
|
PyListObject *np;
|
|
|
|
PyObject **p;
|
1991-03-06 13:07:53 +00:00
|
|
|
if (n < 0)
|
|
|
|
n = 0;
|
|
|
|
size = a->ob_size * n;
|
1997-05-02 03:12:38 +00:00
|
|
|
np = (PyListObject *) PyList_New(size);
|
1991-03-06 13:07:53 +00:00
|
|
|
if (np == NULL)
|
|
|
|
return NULL;
|
|
|
|
p = np->ob_item;
|
|
|
|
for (i = 0; i < n; i++) {
|
|
|
|
for (j = 0; j < a->ob_size; j++) {
|
|
|
|
*p = a->ob_item[j];
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(*p);
|
1991-03-06 13:07:53 +00:00
|
|
|
p++;
|
|
|
|
}
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return (PyObject *) np;
|
1991-03-06 13:07:53 +00:00
|
|
|
}
|
|
|
|
|
1990-10-14 12:07:46 +00:00
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1995-01-17 10:21:11 +00:00
|
|
|
/* Because [X]DECREF can recursively invoke list operations on
|
|
|
|
this list, we must postpone all [X]DECREF activity until
|
|
|
|
after the list is back in its canonical shape. Therefore
|
|
|
|
we must allocate an additional array, 'recycle', into which
|
|
|
|
we temporarily copy the items that are deleted from the
|
|
|
|
list. :-( */
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject **recycle, **p;
|
|
|
|
PyObject **item;
|
1990-10-14 12:07:46 +00:00
|
|
|
int n; /* Size of replacement list */
|
|
|
|
int d; /* Change in size */
|
|
|
|
int k; /* Loop index */
|
1997-05-02 03:12:38 +00:00
|
|
|
#define b ((PyListObject *)v)
|
1990-10-14 12:07:46 +00:00
|
|
|
if (v == NULL)
|
|
|
|
n = 0;
|
1997-05-02 03:12:38 +00:00
|
|
|
else if (PyList_Check(v)) {
|
1990-10-14 12:07:46 +00:00
|
|
|
n = b->ob_size;
|
1991-12-24 13:27:34 +00:00
|
|
|
if (a == b) {
|
|
|
|
/* Special case "a[i:j] = a" -- copy b first */
|
|
|
|
int ret;
|
|
|
|
v = list_slice(b, 0, n);
|
|
|
|
ret = list_ass_slice(a, ilow, ihigh, v);
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_DECREF(v);
|
1991-12-24 13:27:34 +00:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
}
|
1990-10-21 22:15:08 +00:00
|
|
|
else {
|
2000-06-01 03:12:13 +00:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"must assign list (not \"%.200s\") to slice",
|
|
|
|
v->ob_type->tp_name);
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
if (ilow < 0)
|
|
|
|
ilow = 0;
|
|
|
|
else if (ilow > a->ob_size)
|
|
|
|
ilow = a->ob_size;
|
|
|
|
if (ihigh < ilow)
|
|
|
|
ihigh = ilow;
|
|
|
|
else if (ihigh > a->ob_size)
|
|
|
|
ihigh = a->ob_size;
|
|
|
|
item = a->ob_item;
|
|
|
|
d = n - (ihigh-ilow);
|
1995-01-17 10:21:11 +00:00
|
|
|
if (ihigh > ilow)
|
1997-05-02 03:12:38 +00:00
|
|
|
p = recycle = PyMem_NEW(PyObject *, (ihigh-ilow));
|
1995-01-17 10:21:11 +00:00
|
|
|
else
|
|
|
|
p = recycle = NULL;
|
|
|
|
if (d <= 0) { /* Delete -d items; recycle ihigh-ilow items */
|
1990-10-14 12:07:46 +00:00
|
|
|
for (k = ilow; k < ihigh; k++)
|
1995-01-17 10:21:11 +00:00
|
|
|
*p++ = item[k];
|
1990-10-14 12:07:46 +00:00
|
|
|
if (d < 0) {
|
|
|
|
for (/*k = ihigh*/; k < a->ob_size; k++)
|
|
|
|
item[k+d] = item[k];
|
|
|
|
a->ob_size += d;
|
1997-05-02 03:12:38 +00:00
|
|
|
NRESIZE(item, PyObject *, a->ob_size); /* Can't fail */
|
1990-10-14 12:07:46 +00:00
|
|
|
a->ob_item = item;
|
|
|
|
}
|
|
|
|
}
|
1995-01-17 10:21:11 +00:00
|
|
|
else { /* Insert d items; recycle ihigh-ilow items */
|
1997-05-02 03:12:38 +00:00
|
|
|
NRESIZE(item, PyObject *, a->ob_size + d);
|
1990-10-21 22:15:08 +00:00
|
|
|
if (item == NULL) {
|
2000-05-03 23:44:39 +00:00
|
|
|
if (recycle != NULL)
|
|
|
|
PyMem_DEL(recycle);
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_NoMemory();
|
1990-10-21 22:15:08 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
for (k = a->ob_size; --k >= ihigh; )
|
|
|
|
item[k+d] = item[k];
|
|
|
|
for (/*k = ihigh-1*/; k >= ilow; --k)
|
1995-01-17 10:21:11 +00:00
|
|
|
*p++ = item[k];
|
1990-10-14 12:07:46 +00:00
|
|
|
a->ob_item = item;
|
|
|
|
a->ob_size += d;
|
|
|
|
}
|
|
|
|
for (k = 0; k < n; k++, ilow++) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *w = b->ob_item[k];
|
|
|
|
Py_XINCREF(w);
|
1990-10-14 12:07:46 +00:00
|
|
|
item[ilow] = w;
|
|
|
|
}
|
1995-01-17 10:21:11 +00:00
|
|
|
if (recycle) {
|
|
|
|
while (--p >= recycle)
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_XDECREF(*p);
|
|
|
|
PyMem_DEL(recycle);
|
1995-01-17 10:21:11 +00:00
|
|
|
}
|
1990-10-14 12:07:46 +00:00
|
|
|
return 0;
|
|
|
|
#undef b
|
|
|
|
}
|
|
|
|
|
1993-06-17 12:35:49 +00:00
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_SetSlice(PyObject *a, int ilow, int ihigh, PyObject *v)
|
1993-06-17 12:35:49 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyList_Check(a)) {
|
|
|
|
PyErr_BadInternalCall();
|
1993-07-29 08:25:09 +00:00
|
|
|
return -1;
|
1993-06-17 12:35:49 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return list_ass_slice((PyListObject *)a, ilow, ihigh, v);
|
1993-06-17 12:35:49 +00:00
|
|
|
}
|
|
|
|
|
1991-04-03 19:05:18 +00:00
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
list_ass_item(PyListObject *a, int i, PyObject *v)
|
1991-04-03 19:05:18 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *old_value;
|
1991-04-03 19:05:18 +00:00
|
|
|
if (i < 0 || i >= a->ob_size) {
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_SetString(PyExc_IndexError,
|
|
|
|
"list assignment index out of range");
|
1991-04-03 19:05:18 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (v == NULL)
|
|
|
|
return list_ass_slice(a, i, i+1, v);
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(v);
|
1995-01-02 19:07:15 +00:00
|
|
|
old_value = a->ob_item[i];
|
1991-04-03 19:05:18 +00:00
|
|
|
a->ob_item[i] = v;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_DECREF(old_value);
|
1991-04-03 19:05:18 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
ins(PyListObject *self, int where, PyObject *v)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
if (ins1(self, where, v) != 0)
|
|
|
|
return NULL;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listinsert(PyListObject *self, PyObject *args)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
|
|
|
int i;
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *v;
|
2000-02-24 21:48:29 +00:00
|
|
|
if (!PyArg_ParseTuple(args, "iO:insert", &i, &v))
|
1990-10-14 12:07:46 +00:00
|
|
|
return NULL;
|
1993-02-08 15:49:17 +00:00
|
|
|
return ins(self, i, v);
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
2000-03-13 15:41:59 +00:00
|
|
|
/* Define NO_STRICT_LIST_APPEND to enable multi-argument append() */
|
|
|
|
|
|
|
|
#ifndef NO_STRICT_LIST_APPEND
|
|
|
|
#define PyArg_ParseTuple_Compat1 PyArg_ParseTuple
|
|
|
|
#else
|
|
|
|
#define PyArg_ParseTuple_Compat1(args, format, ret) \
|
|
|
|
( \
|
|
|
|
PyTuple_GET_SIZE(args) > 1 ? (*ret = args, 1) : \
|
|
|
|
PyTuple_GET_SIZE(args) == 1 ? (*ret = PyTuple_GET_ITEM(args, 0), 1) : \
|
|
|
|
PyArg_ParseTuple(args, format, ret) \
|
|
|
|
)
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listappend(PyListObject *self, PyObject *args)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *v;
|
2000-03-13 15:41:59 +00:00
|
|
|
if (!PyArg_ParseTuple_Compat1(args, "O:append", &v))
|
1993-02-08 15:49:17 +00:00
|
|
|
return NULL;
|
|
|
|
return ins(self, (int) self->ob_size, v);
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1998-10-09 16:37:25 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listextend(PyListObject *self, PyObject *args)
|
1998-10-09 16:37:25 +00:00
|
|
|
{
|
|
|
|
PyObject *b = NULL, *res = NULL;
|
|
|
|
PyObject **items;
|
|
|
|
int selflen = PyList_GET_SIZE(self);
|
|
|
|
int blen;
|
|
|
|
register int i;
|
|
|
|
|
2000-02-24 21:48:29 +00:00
|
|
|
if (!PyArg_ParseTuple(args, "O:extend", &b))
|
1998-10-09 16:37:25 +00:00
|
|
|
return NULL;
|
|
|
|
|
2000-06-18 18:43:14 +00:00
|
|
|
b = PySequence_Fast(b, "list.extend() argument must be a sequence");
|
|
|
|
if (!b)
|
1998-10-09 16:37:25 +00:00
|
|
|
return NULL;
|
2000-06-18 18:43:14 +00:00
|
|
|
|
2000-07-12 13:05:33 +00:00
|
|
|
if (PyObject_Size(b) == 0)
|
1998-10-09 16:37:25 +00:00
|
|
|
/* short circuit when b is empty */
|
2000-06-18 18:43:14 +00:00
|
|
|
goto ok;
|
|
|
|
|
1998-10-09 16:37:25 +00:00
|
|
|
if (self == (PyListObject*)b) {
|
|
|
|
/* as in list_ass_slice() we must special case the
|
|
|
|
* situation: a.extend(a)
|
|
|
|
*
|
|
|
|
* XXX: I think this way ought to be faster than using
|
|
|
|
* list_slice() the way list_ass_slice() does.
|
|
|
|
*/
|
2000-06-18 18:43:14 +00:00
|
|
|
Py_DECREF(b);
|
1998-10-09 16:37:25 +00:00
|
|
|
b = PyList_New(selflen);
|
|
|
|
if (!b)
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < selflen; i++) {
|
|
|
|
PyObject *o = PyList_GET_ITEM(self, i);
|
|
|
|
Py_INCREF(o);
|
|
|
|
PyList_SET_ITEM(b, i, o);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2000-07-12 13:05:33 +00:00
|
|
|
blen = PyObject_Size(b);
|
2000-06-18 18:43:14 +00:00
|
|
|
|
1998-10-09 16:37:25 +00:00
|
|
|
/* resize a using idiom */
|
|
|
|
items = self->ob_item;
|
|
|
|
NRESIZE(items, PyObject*, selflen + blen);
|
2000-06-18 18:43:14 +00:00
|
|
|
if (items == NULL) {
|
1998-10-09 16:37:25 +00:00
|
|
|
PyErr_NoMemory();
|
2000-06-18 18:43:14 +00:00
|
|
|
goto failed;
|
1998-10-09 16:37:25 +00:00
|
|
|
}
|
|
|
|
self->ob_item = items;
|
|
|
|
|
2000-06-18 18:43:14 +00:00
|
|
|
/* populate the end of self with b's items */
|
1998-10-09 16:37:25 +00:00
|
|
|
for (i = 0; i < blen; i++) {
|
2000-06-18 18:43:14 +00:00
|
|
|
PyObject *o = PySequence_Fast_GET_ITEM(b, i);
|
1998-10-09 16:37:25 +00:00
|
|
|
Py_INCREF(o);
|
|
|
|
PyList_SET_ITEM(self, self->ob_size++, o);
|
|
|
|
}
|
2000-06-18 18:43:14 +00:00
|
|
|
ok:
|
1998-10-09 16:37:25 +00:00
|
|
|
res = Py_None;
|
|
|
|
Py_INCREF(res);
|
2000-06-18 18:43:14 +00:00
|
|
|
failed:
|
|
|
|
Py_DECREF(b);
|
1998-10-09 16:37:25 +00:00
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-06-30 15:36:32 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listpop(PyListObject *self, PyObject *args)
|
1998-06-30 15:36:32 +00:00
|
|
|
{
|
|
|
|
int i = -1;
|
|
|
|
PyObject *v;
|
2000-02-24 21:48:29 +00:00
|
|
|
if (!PyArg_ParseTuple(args, "|i:pop", &i))
|
1998-06-30 15:36:32 +00:00
|
|
|
return NULL;
|
|
|
|
if (self->ob_size == 0) {
|
|
|
|
/* Special-case most common failure cause */
|
|
|
|
PyErr_SetString(PyExc_IndexError, "pop from empty list");
|
|
|
|
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 = self->ob_item[i];
|
|
|
|
Py_INCREF(v);
|
|
|
|
if (list_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {
|
|
|
|
Py_DECREF(v);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return v;
|
|
|
|
}
|
|
|
|
|
1996-12-10 23:55:39 +00:00
|
|
|
/* New quicksort implementation for arrays of object pointers.
|
|
|
|
Thanks to discussions with Tim Peters. */
|
|
|
|
|
|
|
|
/* CMPERROR is returned by our comparison function when an error
|
|
|
|
occurred. This is the largest negative integer (0x80000000 on a
|
|
|
|
32-bit system). */
|
1997-03-05 00:45:43 +00:00
|
|
|
#define CMPERROR ( (int) ((unsigned int)1 << (8*sizeof(int) - 1)) )
|
1996-12-10 23:55:39 +00:00
|
|
|
|
|
|
|
/* Comparison function. Takes care of calling a user-supplied
|
|
|
|
comparison function (any callable Python object). Calls the
|
1998-06-16 15:18:28 +00:00
|
|
|
standard comparison function, PyObject_Compare(), if the user-
|
|
|
|
supplied function is NULL. */
|
1996-12-10 23:55:39 +00:00
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
docompare(PyObject *x, PyObject *y, PyObject *compare)
|
1996-12-10 23:55:39 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *args, *res;
|
1996-12-10 23:55:39 +00:00
|
|
|
int i;
|
|
|
|
|
1997-05-23 00:06:51 +00:00
|
|
|
if (compare == NULL) {
|
|
|
|
i = PyObject_Compare(x, y);
|
|
|
|
if (i && PyErr_Occurred())
|
|
|
|
i = CMPERROR;
|
|
|
|
return i;
|
|
|
|
}
|
1996-12-10 23:55:39 +00:00
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
args = Py_BuildValue("(OO)", x, y);
|
1996-12-10 23:55:39 +00:00
|
|
|
if (args == NULL)
|
|
|
|
return CMPERROR;
|
1997-05-02 03:12:38 +00:00
|
|
|
res = PyEval_CallObject(compare, args);
|
|
|
|
Py_DECREF(args);
|
1996-12-10 23:55:39 +00:00
|
|
|
if (res == NULL)
|
|
|
|
return CMPERROR;
|
1997-05-02 03:12:38 +00:00
|
|
|
if (!PyInt_Check(res)) {
|
|
|
|
Py_DECREF(res);
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
1999-04-19 17:44:39 +00:00
|
|
|
"comparison function must return int");
|
1996-12-10 23:55:39 +00:00
|
|
|
return CMPERROR;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
i = PyInt_AsLong(res);
|
|
|
|
Py_DECREF(res);
|
1996-12-10 23:55:39 +00:00
|
|
|
if (i < 0)
|
|
|
|
return -1;
|
|
|
|
if (i > 0)
|
|
|
|
return 1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
/* MINSIZE is the smallest array that will get a full-blown samplesort
|
|
|
|
treatment; smaller arrays are sorted using binary insertion. It must
|
|
|
|
be at least 7 for the samplesort implementation to work. Binary
|
|
|
|
insertion does fewer compares, but can suffer O(N**2) data movement.
|
|
|
|
The more expensive compares, the larger MINSIZE should be. */
|
|
|
|
#define MINSIZE 100
|
|
|
|
|
|
|
|
/* MINPARTITIONSIZE is the smallest array slice samplesort will bother to
|
|
|
|
partition; smaller slices are passed to binarysort. It must be at
|
|
|
|
least 2, and no larger than MINSIZE. Setting it higher reduces the #
|
|
|
|
of compares slowly, but increases the amount of data movement quickly.
|
|
|
|
The value here was chosen assuming a compare costs ~25x more than
|
|
|
|
swapping a pair of memory-resident pointers -- but under that assumption,
|
|
|
|
changing the value by a few dozen more or less has aggregate effect
|
|
|
|
under 1%. So the value is crucial, but not touchy <wink>. */
|
|
|
|
#define MINPARTITIONSIZE 40
|
|
|
|
|
|
|
|
/* MAXMERGE is the largest number of elements we'll always merge into
|
|
|
|
a known-to-be sorted chunk via binary insertion, regardless of the
|
|
|
|
size of that chunk. Given a chunk of N sorted elements, and a group
|
|
|
|
of K unknowns, the largest K for which it's better to do insertion
|
|
|
|
(than a full-blown sort) is a complicated function of N and K mostly
|
|
|
|
involving the expected number of compares and data moves under each
|
|
|
|
approach, and the relative cost of those operations on a specific
|
|
|
|
architecure. The fixed value here is conservative, and should be a
|
|
|
|
clear win regardless of architecture or N. */
|
|
|
|
#define MAXMERGE 15
|
1996-12-10 23:55:39 +00:00
|
|
|
|
|
|
|
/* STACKSIZE is the size of our work stack. A rough estimate is that
|
1998-06-16 15:18:28 +00:00
|
|
|
this allows us to sort arrays of size N where
|
|
|
|
N / ln(N) = MINPARTITIONSIZE * 2**STACKSIZE, so 60 is more than enough
|
|
|
|
for arrays of size 2**64. Because we push the biggest partition
|
|
|
|
first, the worst case occurs when all subarrays are always partitioned
|
|
|
|
exactly in two. */
|
|
|
|
#define STACKSIZE 60
|
|
|
|
|
|
|
|
|
|
|
|
#define SETK(X,Y) if ((k = docompare(X,Y,compare))==CMPERROR) goto fail
|
|
|
|
|
|
|
|
/* binarysort is the best method for sorting small arrays: it does
|
|
|
|
few compares, but can do data movement quadratic in the number of
|
|
|
|
elements.
|
1998-06-17 14:15:44 +00:00
|
|
|
[lo, hi) is a contiguous slice of a list, and is sorted via
|
1998-06-16 15:18:28 +00:00
|
|
|
binary insertion.
|
|
|
|
On entry, must have lo <= start <= hi, and that [lo, start) is already
|
|
|
|
sorted (pass start == lo if you don't know!).
|
|
|
|
If docompare complains (returns CMPERROR) return -1, else 0.
|
|
|
|
Even in case of error, the output slice will be some permutation of
|
|
|
|
the input (nothing is lost or duplicated).
|
|
|
|
*/
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
binarysort(PyObject **lo, PyObject **hi, PyObject **start, PyObject *compare)
|
|
|
|
/* compare -- comparison function object, or NULL for default */
|
1998-06-16 15:18:28 +00:00
|
|
|
{
|
|
|
|
/* assert lo <= start <= hi
|
|
|
|
assert [lo, start) is sorted */
|
|
|
|
register int k;
|
|
|
|
register PyObject **l, **p, **r;
|
|
|
|
register PyObject *pivot;
|
|
|
|
|
|
|
|
if (lo == start)
|
|
|
|
++start;
|
|
|
|
for (; start < hi; ++start) {
|
|
|
|
/* set l to where *start belongs */
|
|
|
|
l = lo;
|
|
|
|
r = start;
|
|
|
|
pivot = *r;
|
|
|
|
do {
|
|
|
|
p = l + ((r - l) >> 1);
|
|
|
|
SETK(pivot, *p);
|
|
|
|
if (k < 0)
|
|
|
|
r = p;
|
|
|
|
else
|
|
|
|
l = p + 1;
|
|
|
|
} while (l < r);
|
|
|
|
/* Pivot should go at l -- slide over to make room.
|
|
|
|
Caution: using memmove is much slower under MSVC 5;
|
|
|
|
we're not usually moving many slots. */
|
|
|
|
for (p = start; p > l; --p)
|
|
|
|
*p = *(p-1);
|
|
|
|
*l = pivot;
|
|
|
|
}
|
|
|
|
return 0;
|
1996-12-10 23:55:39 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
fail:
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* samplesortslice is the sorting workhorse.
|
1998-06-17 14:15:44 +00:00
|
|
|
[lo, hi) is a contiguous slice of a list, to be sorted in place.
|
1998-06-16 15:18:28 +00:00
|
|
|
On entry, must have lo <= hi,
|
|
|
|
If docompare complains (returns CMPERROR) return -1, else 0.
|
|
|
|
Even in case of error, the output slice will be some permutation of
|
|
|
|
the input (nothing is lost or duplicated).
|
|
|
|
|
|
|
|
samplesort is basically quicksort on steroids: a power of 2 close
|
|
|
|
to n/ln(n) is computed, and that many elements (less 1) are picked at
|
|
|
|
random from the array and sorted. These 2**k - 1 elements are then
|
|
|
|
used as preselected pivots for an equal number of quicksort
|
|
|
|
partitioning steps, partitioning the slice into 2**k chunks each of
|
|
|
|
size about ln(n). These small final chunks are then usually handled
|
|
|
|
by binarysort. Note that when k=1, this is roughly the same as an
|
|
|
|
ordinary quicksort using a random pivot, and when k=2 this is roughly
|
|
|
|
a median-of-3 quicksort. From that view, using k ~= lg(n/ln(n)) makes
|
|
|
|
this a "median of n/ln(n)" quicksort. You can also view it as a kind
|
|
|
|
of bucket sort, where 2**k-1 bucket boundaries are picked dynamically.
|
|
|
|
|
|
|
|
The large number of samples makes a quadratic-time case almost
|
|
|
|
impossible, and asymptotically drives the average-case number of
|
|
|
|
compares from quicksort's 2 N ln N (or 12/7 N ln N for the median-of-
|
|
|
|
3 variant) down to N lg N.
|
|
|
|
|
|
|
|
We also play lots of low-level tricks to cut the number of compares.
|
|
|
|
|
|
|
|
Very obscure: To avoid using extra memory, the PPs are stored in the
|
|
|
|
array and shuffled around as partitioning proceeds. At the start of a
|
|
|
|
partitioning step, we'll have 2**m-1 (for some m) PPs in sorted order,
|
|
|
|
adjacent (either on the left or the right!) to a chunk of X elements
|
|
|
|
that are to be partitioned: P X or X P. In either case we need to
|
|
|
|
shuffle things *in place* so that the 2**(m-1) smaller PPs are on the
|
|
|
|
left, followed by the PP to be used for this step (that's the middle
|
|
|
|
of the PPs), followed by X, followed by the 2**(m-1) larger PPs:
|
|
|
|
P X or X P -> Psmall pivot X Plarge
|
|
|
|
and the order of the PPs must not be altered. It can take a while
|
|
|
|
to realize this isn't trivial! It can take even longer <wink> to
|
|
|
|
understand why the simple code below works, using only 2**(m-1) swaps.
|
|
|
|
The key is that the order of the X elements isn't necessarily
|
|
|
|
preserved: X can end up as some cyclic permutation of its original
|
|
|
|
order. That's OK, because X is unsorted anyway. If the order of X
|
|
|
|
had to be preserved too, the simplest method I know of using O(1)
|
|
|
|
scratch storage requires len(X) + 2**(m-1) swaps, spread over 2 passes.
|
|
|
|
Since len(X) is typically several times larger than 2**(m-1), that
|
|
|
|
would slow things down.
|
|
|
|
*/
|
|
|
|
|
|
|
|
struct SamplesortStackNode {
|
|
|
|
/* Represents a slice of the array, from (& including) lo up
|
|
|
|
to (but excluding) hi. "extra" additional & adjacent elements
|
|
|
|
are pre-selected pivots (PPs), spanning [lo-extra, lo) if
|
|
|
|
extra > 0, or [hi, hi-extra) if extra < 0. The PPs are
|
|
|
|
already sorted, but nothing is known about the other elements
|
|
|
|
in [lo, hi). |extra| is always one less than a power of 2.
|
|
|
|
When extra is 0, we're out of PPs, and the slice must be
|
|
|
|
sorted by some other means. */
|
|
|
|
PyObject **lo;
|
|
|
|
PyObject **hi;
|
|
|
|
int extra;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* The number of PPs we want is 2**k - 1, where 2**k is as close to
|
1998-06-17 14:15:44 +00:00
|
|
|
N / ln(N) as possible. So k ~= lg(N / ln(N)). Calling libm routines
|
1998-06-16 15:18:28 +00:00
|
|
|
is undesirable, so cutoff values are canned in the "cutoff" table
|
|
|
|
below: cutoff[i] is the smallest N such that k == CUTOFFBASE + i. */
|
|
|
|
#define CUTOFFBASE 4
|
1999-01-14 19:01:53 +00:00
|
|
|
static long cutoff[] = {
|
1998-06-16 15:18:28 +00:00
|
|
|
43, /* smallest N such that k == 4 */
|
|
|
|
106, /* etc */
|
|
|
|
250,
|
|
|
|
576,
|
|
|
|
1298,
|
|
|
|
2885,
|
|
|
|
6339,
|
|
|
|
13805,
|
|
|
|
29843,
|
|
|
|
64116,
|
|
|
|
137030,
|
|
|
|
291554,
|
|
|
|
617916,
|
|
|
|
1305130,
|
|
|
|
2748295,
|
|
|
|
5771662,
|
|
|
|
12091672,
|
|
|
|
25276798,
|
|
|
|
52734615,
|
|
|
|
109820537,
|
|
|
|
228324027,
|
|
|
|
473977813,
|
|
|
|
982548444, /* smallest N such that k == 26 */
|
|
|
|
2034159050 /* largest N that fits in signed 32-bit; k == 27 */
|
|
|
|
};
|
1996-12-10 23:55:39 +00:00
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
samplesortslice(PyObject **lo, PyObject **hi, PyObject *compare)
|
|
|
|
/* compare -- comparison function object, or NULL for default */
|
1996-12-10 23:55:39 +00:00
|
|
|
{
|
1998-06-16 15:18:28 +00:00
|
|
|
register PyObject **l, **r;
|
1997-05-02 03:12:38 +00:00
|
|
|
register PyObject *tmp, *pivot;
|
1998-06-16 15:18:28 +00:00
|
|
|
register int k;
|
|
|
|
int n, extra, top, extraOnRight;
|
|
|
|
struct SamplesortStackNode stack[STACKSIZE];
|
|
|
|
|
|
|
|
/* assert lo <= hi */
|
|
|
|
n = hi - lo;
|
|
|
|
|
|
|
|
/* ----------------------------------------------------------
|
|
|
|
* Special cases
|
|
|
|
* --------------------------------------------------------*/
|
|
|
|
if (n < 2)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
/* Set r to the largest value such that [lo,r) is sorted.
|
|
|
|
This catches the already-sorted case, the all-the-same
|
|
|
|
case, and the appended-a-few-elements-to-a-sorted-list case.
|
|
|
|
If the array is unsorted, we're very likely to get out of
|
|
|
|
the loop fast, so the test is cheap if it doesn't pay off.
|
|
|
|
*/
|
|
|
|
/* assert lo < hi */
|
|
|
|
for (r = lo+1; r < hi; ++r) {
|
|
|
|
SETK(*r, *(r-1));
|
|
|
|
if (k < 0)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
/* [lo,r) is sorted, [r,hi) unknown. Get out cheap if there are
|
|
|
|
few unknowns, or few elements in total. */
|
|
|
|
if (hi - r <= MAXMERGE || n < MINSIZE)
|
1998-06-17 14:15:44 +00:00
|
|
|
return binarysort(lo, hi, r, compare);
|
1998-06-16 15:18:28 +00:00
|
|
|
|
|
|
|
/* Check for the array already being reverse-sorted. Typical
|
|
|
|
benchmark-driven silliness <wink>. */
|
|
|
|
/* assert lo < hi */
|
|
|
|
for (r = lo+1; r < hi; ++r) {
|
|
|
|
SETK(*(r-1), *r);
|
|
|
|
if (k < 0)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (hi - r <= MAXMERGE) {
|
|
|
|
/* Reverse the reversed prefix, then insert the tail */
|
|
|
|
PyObject **originalr = r;
|
|
|
|
l = lo;
|
|
|
|
do {
|
|
|
|
--r;
|
|
|
|
tmp = *l; *l = *r; *r = tmp;
|
|
|
|
++l;
|
|
|
|
} while (l < r);
|
1998-06-17 14:15:44 +00:00
|
|
|
return binarysort(lo, hi, originalr, compare);
|
1998-06-16 15:18:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* ----------------------------------------------------------
|
|
|
|
* Normal case setup: a large array without obvious pattern.
|
|
|
|
* --------------------------------------------------------*/
|
|
|
|
|
|
|
|
/* extra := a power of 2 ~= n/ln(n), less 1.
|
|
|
|
First find the smallest extra s.t. n < cutoff[extra] */
|
|
|
|
for (extra = 0;
|
|
|
|
extra < sizeof(cutoff) / sizeof(cutoff[0]);
|
|
|
|
++extra) {
|
|
|
|
if (n < cutoff[extra])
|
|
|
|
break;
|
|
|
|
/* note that if we fall out of the loop, the value of
|
|
|
|
extra still makes *sense*, but may be smaller than
|
|
|
|
we would like (but the array has more than ~= 2**31
|
|
|
|
elements in this case!) */
|
|
|
|
}
|
|
|
|
/* Now k == extra - 1 + CUTOFFBASE. The smallest value k can
|
|
|
|
have is CUTOFFBASE-1, so
|
|
|
|
assert MINSIZE >= 2**(CUTOFFBASE-1) - 1 */
|
|
|
|
extra = (1 << (extra - 1 + CUTOFFBASE)) - 1;
|
|
|
|
/* assert extra > 0 and n >= extra */
|
|
|
|
|
|
|
|
/* Swap that many values to the start of the array. The
|
|
|
|
selection of elements is pseudo-random, but the same on
|
|
|
|
every run (this is intentional! timing algorithm changes is
|
|
|
|
a pain if timing varies across runs). */
|
|
|
|
{
|
|
|
|
unsigned int seed = n / extra; /* arbitrary */
|
|
|
|
unsigned int i;
|
|
|
|
for (i = 0; i < (unsigned)extra; ++i) {
|
|
|
|
/* j := random int in [i, n) */
|
|
|
|
unsigned int j;
|
|
|
|
seed = seed * 69069 + 7;
|
|
|
|
j = i + seed % (n - i);
|
|
|
|
tmp = lo[i]; lo[i] = lo[j]; lo[j] = tmp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Recursively sort the preselected pivots. */
|
1998-06-17 14:15:44 +00:00
|
|
|
if (samplesortslice(lo, lo + extra, compare) < 0)
|
1998-06-16 15:18:28 +00:00
|
|
|
goto fail;
|
|
|
|
|
|
|
|
top = 0; /* index of available stack slot */
|
|
|
|
lo += extra; /* point to first unknown */
|
|
|
|
extraOnRight = 0; /* the PPs are at the left end */
|
|
|
|
|
|
|
|
/* ----------------------------------------------------------
|
|
|
|
* Partition [lo, hi), and repeat until out of work.
|
|
|
|
* --------------------------------------------------------*/
|
|
|
|
for (;;) {
|
|
|
|
/* assert lo <= hi, so n >= 0 */
|
1996-12-11 23:57:39 +00:00
|
|
|
n = hi - lo;
|
1998-05-26 15:06:32 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
/* We may not want, or may not be able, to partition:
|
|
|
|
If n is small, it's quicker to insert.
|
|
|
|
If extra is 0, we're out of pivots, and *must* use
|
|
|
|
another method.
|
|
|
|
*/
|
|
|
|
if (n < MINPARTITIONSIZE || extra == 0) {
|
|
|
|
if (n >= MINSIZE) {
|
|
|
|
/* assert extra == 0
|
|
|
|
This is rare, since the average size
|
|
|
|
of a final block is only about
|
|
|
|
ln(original n). */
|
1998-06-17 14:15:44 +00:00
|
|
|
if (samplesortslice(lo, hi, compare) < 0)
|
1998-06-16 15:18:28 +00:00
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
/* Binary insertion should be quicker,
|
|
|
|
and we can take advantage of the PPs
|
|
|
|
already being sorted. */
|
|
|
|
if (extraOnRight && extra) {
|
|
|
|
/* swap the PPs to the left end */
|
|
|
|
k = extra;
|
|
|
|
do {
|
|
|
|
tmp = *lo;
|
|
|
|
*lo = *hi;
|
|
|
|
*hi = tmp;
|
|
|
|
++lo; ++hi;
|
|
|
|
} while (--k);
|
|
|
|
}
|
|
|
|
if (binarysort(lo - extra, hi, lo,
|
1998-06-17 14:15:44 +00:00
|
|
|
compare) < 0)
|
1998-06-16 15:18:28 +00:00
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Find another slice to work on. */
|
|
|
|
if (--top < 0)
|
|
|
|
break; /* no more -- done! */
|
|
|
|
lo = stack[top].lo;
|
|
|
|
hi = stack[top].hi;
|
|
|
|
extra = stack[top].extra;
|
|
|
|
extraOnRight = 0;
|
|
|
|
if (extra < 0) {
|
|
|
|
extraOnRight = 1;
|
|
|
|
extra = -extra;
|
1998-05-26 15:06:32 +00:00
|
|
|
}
|
1996-12-10 23:55:39 +00:00
|
|
|
continue;
|
1998-05-26 15:06:32 +00:00
|
|
|
}
|
1996-12-10 23:55:39 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
/* Pretend the PPs are indexed 0, 1, ..., extra-1.
|
|
|
|
Then our preselected pivot is at (extra-1)/2, and we
|
|
|
|
want to move the PPs before that to the left end of
|
|
|
|
the slice, and the PPs after that to the right end.
|
|
|
|
The following section changes extra, lo, hi, and the
|
|
|
|
slice such that:
|
|
|
|
[lo-extra, lo) contains the smaller PPs.
|
|
|
|
*lo == our PP.
|
|
|
|
(lo, hi) contains the unknown elements.
|
|
|
|
[hi, hi+extra) contains the larger PPs.
|
|
|
|
*/
|
|
|
|
k = extra >>= 1; /* num PPs to move */
|
|
|
|
if (extraOnRight) {
|
|
|
|
/* Swap the smaller PPs to the left end.
|
|
|
|
Note that this loop actually moves k+1 items:
|
|
|
|
the last is our PP */
|
|
|
|
do {
|
|
|
|
tmp = *lo; *lo = *hi; *hi = tmp;
|
|
|
|
++lo; ++hi;
|
|
|
|
} while (k--);
|
1998-05-29 17:56:32 +00:00
|
|
|
}
|
1998-06-16 15:18:28 +00:00
|
|
|
else {
|
|
|
|
/* Swap the larger PPs to the right end. */
|
|
|
|
while (k--) {
|
|
|
|
--lo; --hi;
|
|
|
|
tmp = *lo; *lo = *hi; *hi = tmp;
|
1998-05-29 17:56:32 +00:00
|
|
|
}
|
|
|
|
}
|
1998-06-16 15:18:28 +00:00
|
|
|
--lo; /* *lo is now our PP */
|
|
|
|
pivot = *lo;
|
|
|
|
|
|
|
|
/* Now an almost-ordinary quicksort partition step.
|
|
|
|
Note that most of the time is spent here!
|
|
|
|
Only odd thing is that we partition into < and >=,
|
|
|
|
instead of the usual <= and >=. This helps when
|
|
|
|
there are lots of duplicates of different values,
|
|
|
|
because it eventually tends to make subfiles
|
|
|
|
"pure" (all duplicates), and we special-case for
|
|
|
|
duplicates later. */
|
|
|
|
l = lo + 1;
|
|
|
|
r = hi - 1;
|
|
|
|
/* assert lo < l < r < hi (small n weeded out above) */
|
|
|
|
|
|
|
|
do {
|
|
|
|
/* slide l right, looking for key >= pivot */
|
|
|
|
do {
|
1998-05-29 17:56:32 +00:00
|
|
|
SETK(*l, pivot);
|
1998-05-13 21:20:49 +00:00
|
|
|
if (k < 0)
|
1998-06-16 15:18:28 +00:00
|
|
|
++l;
|
|
|
|
else
|
1996-12-10 23:55:39 +00:00
|
|
|
break;
|
1998-06-16 15:18:28 +00:00
|
|
|
} while (l < r);
|
|
|
|
|
|
|
|
/* slide r left, looking for key < pivot */
|
|
|
|
while (l < r) {
|
1998-06-17 14:15:44 +00:00
|
|
|
register PyObject *rval = *r--;
|
|
|
|
SETK(rval, pivot);
|
|
|
|
if (k < 0) {
|
|
|
|
/* swap and advance */
|
|
|
|
r[1] = *l;
|
|
|
|
*l++ = rval;
|
1996-12-10 23:55:39 +00:00
|
|
|
break;
|
1998-06-17 14:15:44 +00:00
|
|
|
}
|
1997-12-10 15:14:24 +00:00
|
|
|
}
|
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
} while (l < r);
|
1998-05-13 21:20:49 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
/* assert lo < r <= l < hi
|
|
|
|
assert r == l or r+1 == l
|
|
|
|
everything to the left of l is < pivot, and
|
|
|
|
everything to the right of r is >= pivot */
|
1998-05-13 21:20:49 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
if (l == r) {
|
|
|
|
SETK(*r, pivot);
|
|
|
|
if (k < 0)
|
|
|
|
++l;
|
|
|
|
else
|
|
|
|
--r;
|
|
|
|
}
|
|
|
|
/* assert lo <= r and r+1 == l and l <= hi
|
|
|
|
assert r == lo or a[r] < pivot
|
|
|
|
assert a[lo] is pivot
|
|
|
|
assert l == hi or a[l] >= pivot
|
|
|
|
Swap the pivot into "the middle", so we can henceforth
|
|
|
|
ignore it.
|
1998-05-13 21:20:49 +00:00
|
|
|
*/
|
1998-06-16 15:18:28 +00:00
|
|
|
*lo = *r;
|
|
|
|
*r = pivot;
|
|
|
|
|
|
|
|
/* The following is true now, & will be preserved:
|
|
|
|
All in [lo,r) are < pivot
|
|
|
|
All in [r,l) == pivot (& so can be ignored)
|
|
|
|
All in [l,hi) are >= pivot */
|
|
|
|
|
|
|
|
/* Check for duplicates of the pivot. One compare is
|
|
|
|
wasted if there are no duplicates, but can win big
|
|
|
|
when there are.
|
|
|
|
Tricky: we're sticking to "<" compares, so deduce
|
|
|
|
equality indirectly. We know pivot <= *l, so they're
|
|
|
|
equal iff not pivot < *l.
|
|
|
|
*/
|
|
|
|
while (l < hi) {
|
|
|
|
/* pivot <= *l known */
|
|
|
|
SETK(pivot, *l);
|
|
|
|
if (k < 0)
|
|
|
|
break;
|
|
|
|
else
|
1998-05-26 15:06:32 +00:00
|
|
|
/* <= and not < implies == */
|
1998-06-16 15:18:28 +00:00
|
|
|
++l;
|
|
|
|
}
|
1998-05-26 15:06:32 +00:00
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
/* assert lo <= r < l <= hi
|
|
|
|
Partitions are [lo, r) and [l, hi) */
|
|
|
|
|
|
|
|
/* push fattest first; remember we still have extra PPs
|
|
|
|
to the left of the left chunk and to the right of
|
|
|
|
the right chunk! */
|
|
|
|
/* assert top < STACKSIZE */
|
|
|
|
if (r - lo <= hi - l) {
|
|
|
|
/* second is bigger */
|
|
|
|
stack[top].lo = l;
|
|
|
|
stack[top].hi = hi;
|
|
|
|
stack[top].extra = -extra;
|
|
|
|
hi = r;
|
|
|
|
extraOnRight = 0;
|
1996-12-10 23:55:39 +00:00
|
|
|
}
|
1998-06-16 15:18:28 +00:00
|
|
|
else {
|
|
|
|
/* first is bigger */
|
|
|
|
stack[top].lo = lo;
|
|
|
|
stack[top].hi = r;
|
|
|
|
stack[top].extra = extra;
|
|
|
|
lo = l;
|
|
|
|
extraOnRight = 1;
|
|
|
|
}
|
|
|
|
++top;
|
|
|
|
|
|
|
|
} /* end of partitioning loop */
|
1997-12-10 15:14:24 +00:00
|
|
|
|
1996-12-10 23:55:39 +00:00
|
|
|
return 0;
|
1998-05-29 17:56:32 +00:00
|
|
|
|
|
|
|
fail:
|
|
|
|
return -1;
|
1998-06-16 15:18:28 +00:00
|
|
|
}
|
1998-05-29 17:56:32 +00:00
|
|
|
|
|
|
|
#undef SETK
|
1998-06-16 15:18:28 +00:00
|
|
|
|
|
|
|
staticforward PyTypeObject immutable_list_type;
|
1996-12-10 23:55:39 +00:00
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listsort(PyListObject *self, PyObject *args)
|
1996-12-10 23:55:39 +00:00
|
|
|
{
|
1998-06-16 15:18:28 +00:00
|
|
|
int err;
|
2000-02-24 15:23:03 +00:00
|
|
|
PyObject *compare = NULL;
|
1998-06-16 15:18:28 +00:00
|
|
|
|
2000-02-24 15:23:03 +00:00
|
|
|
if (args != NULL) {
|
2000-02-24 21:48:29 +00:00
|
|
|
if (!PyArg_ParseTuple(args, "|O:sort", &compare))
|
2000-02-24 15:23:03 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1998-06-16 15:18:28 +00:00
|
|
|
self->ob_type = &immutable_list_type;
|
|
|
|
err = samplesortslice(self->ob_item,
|
|
|
|
self->ob_item + self->ob_size,
|
1998-06-17 14:15:44 +00:00
|
|
|
compare);
|
1998-06-16 15:18:28 +00:00
|
|
|
self->ob_type = &PyList_Type;
|
|
|
|
if (err < 0)
|
1996-12-10 23:55:39 +00:00
|
|
|
return NULL;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1996-12-10 23:55:39 +00:00
|
|
|
}
|
|
|
|
|
1998-06-16 15:18:28 +00:00
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_Sort(PyObject *v)
|
1998-06-16 15:18:28 +00:00
|
|
|
{
|
|
|
|
if (v == NULL || !PyList_Check(v)) {
|
|
|
|
PyErr_BadInternalCall();
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
v = listsort((PyListObject *)v, (PyObject *)NULL);
|
|
|
|
if (v == NULL)
|
|
|
|
return -1;
|
|
|
|
Py_DECREF(v);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listreverse(PyListObject *self, PyObject *args)
|
1991-03-06 13:07:53 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
register PyObject **p, **q;
|
|
|
|
register PyObject *tmp;
|
1991-03-06 13:07:53 +00:00
|
|
|
|
2000-02-24 21:48:29 +00:00
|
|
|
if (!PyArg_ParseTuple(args, ":reverse"))
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
|
|
|
|
|
|
|
if (self->ob_size > 1) {
|
|
|
|
for (p = self->ob_item, q = self->ob_item + self->ob_size - 1;
|
|
|
|
p < q; p++, q--) {
|
|
|
|
tmp = *p;
|
|
|
|
*p = *q;
|
|
|
|
*q = tmp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1991-03-06 13:07:53 +00:00
|
|
|
}
|
|
|
|
|
1995-01-17 16:34:45 +00:00
|
|
|
int
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_Reverse(PyObject *v)
|
1995-01-17 16:34:45 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
if (v == NULL || !PyList_Check(v)) {
|
|
|
|
PyErr_BadInternalCall();
|
1995-01-17 16:34:45 +00:00
|
|
|
return -1;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
v = listreverse((PyListObject *)v, (PyObject *)NULL);
|
1995-01-17 16:34:45 +00:00
|
|
|
if (v == NULL)
|
|
|
|
return -1;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_DECREF(v);
|
1995-01-17 16:34:45 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
PyList_AsTuple(PyObject *v)
|
1994-08-29 12:45:32 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
PyObject *w;
|
|
|
|
PyObject **p;
|
1994-08-29 12:45:32 +00:00
|
|
|
int n;
|
1997-05-02 03:12:38 +00:00
|
|
|
if (v == NULL || !PyList_Check(v)) {
|
|
|
|
PyErr_BadInternalCall();
|
1994-08-29 12:45:32 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
n = ((PyListObject *)v)->ob_size;
|
|
|
|
w = PyTuple_New(n);
|
1994-08-29 12:45:32 +00:00
|
|
|
if (w == NULL)
|
|
|
|
return NULL;
|
1997-05-02 03:12:38 +00:00
|
|
|
p = ((PyTupleObject *)w)->ob_item;
|
2000-07-25 12:56:38 +00:00
|
|
|
memcpy((void *)p,
|
|
|
|
(void *)((PyListObject *)v)->ob_item,
|
1997-05-02 03:12:38 +00:00
|
|
|
n*sizeof(PyObject *));
|
1994-08-29 12:45:32 +00:00
|
|
|
while (--n >= 0) {
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(*p);
|
1994-08-29 12:45:32 +00:00
|
|
|
p++;
|
|
|
|
}
|
|
|
|
return w;
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listindex(PyListObject *self, PyObject *args)
|
1991-03-06 13:07:53 +00:00
|
|
|
{
|
|
|
|
int i;
|
2000-02-24 15:23:03 +00:00
|
|
|
PyObject *v;
|
|
|
|
|
2000-03-13 15:41:59 +00:00
|
|
|
if (!PyArg_ParseTuple_Compat1(args, "O:index", &v))
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
2000-02-24 15:23:03 +00:00
|
|
|
if (PyObject_Compare(self->ob_item[i], v) == 0)
|
1997-05-02 03:12:38 +00:00
|
|
|
return PyInt_FromLong((long)i);
|
1997-05-23 00:06:51 +00:00
|
|
|
if (PyErr_Occurred())
|
|
|
|
return NULL;
|
1991-03-06 13:07:53 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_SetString(PyExc_ValueError, "list.index(x): x not in list");
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listcount(PyListObject *self, PyObject *args)
|
1991-10-20 20:20:40 +00:00
|
|
|
{
|
|
|
|
int count = 0;
|
|
|
|
int i;
|
2000-02-24 15:23:03 +00:00
|
|
|
PyObject *v;
|
|
|
|
|
2000-03-13 15:41:59 +00:00
|
|
|
if (!PyArg_ParseTuple_Compat1(args, "O:count", &v))
|
1991-10-20 20:20:40 +00:00
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
2000-02-24 15:23:03 +00:00
|
|
|
if (PyObject_Compare(self->ob_item[i], v) == 0)
|
1991-10-20 20:20:40 +00:00
|
|
|
count++;
|
1997-05-23 00:06:51 +00:00
|
|
|
if (PyErr_Occurred())
|
|
|
|
return NULL;
|
1991-10-20 20:20:40 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
return PyInt_FromLong((long)count);
|
1991-10-20 20:20:40 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
listremove(PyListObject *self, PyObject *args)
|
1991-03-06 13:07:53 +00:00
|
|
|
{
|
|
|
|
int i;
|
2000-02-24 15:23:03 +00:00
|
|
|
PyObject *v;
|
|
|
|
|
2000-03-13 15:41:59 +00:00
|
|
|
if (!PyArg_ParseTuple_Compat1(args, "O:remove", &v))
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < self->ob_size; i++) {
|
2000-02-24 15:23:03 +00:00
|
|
|
if (PyObject_Compare(self->ob_item[i], v) == 0) {
|
1997-05-02 03:12:38 +00:00
|
|
|
if (list_ass_slice(self, i, i+1,
|
|
|
|
(PyObject *)NULL) != 0)
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
1997-05-02 03:12:38 +00:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
1991-03-06 13:07:53 +00:00
|
|
|
}
|
1997-05-23 00:06:51 +00:00
|
|
|
if (PyErr_Occurred())
|
|
|
|
return NULL;
|
1991-03-06 13:07:53 +00:00
|
|
|
}
|
1997-05-02 03:12:38 +00:00
|
|
|
PyErr_SetString(PyExc_ValueError, "list.remove(x): x not in list");
|
1991-03-06 13:07:53 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2000-06-23 14:18:11 +00:00
|
|
|
static int
|
|
|
|
list_traverse(PyListObject *o, visitproc visit, void *arg)
|
|
|
|
{
|
|
|
|
int i, err;
|
|
|
|
PyObject *x;
|
|
|
|
|
|
|
|
for (i = o->ob_size; --i >= 0; ) {
|
|
|
|
x = o->ob_item[i];
|
|
|
|
if (x != NULL) {
|
|
|
|
err = visit(x, arg);
|
|
|
|
if (err)
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
list_clear(PyListObject *lp)
|
|
|
|
{
|
|
|
|
(void) PyList_SetSlice((PyObject *)lp, 0, lp->ob_size, 0);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1998-06-30 15:36:32 +00:00
|
|
|
static char append_doc[] =
|
|
|
|
"L.append(object) -- append object to end";
|
1998-10-09 16:37:25 +00:00
|
|
|
static char extend_doc[] =
|
|
|
|
"L.extend(list) -- extend list by appending list elements";
|
1998-06-30 15:36:32 +00:00
|
|
|
static char insert_doc[] =
|
|
|
|
"L.insert(index, object) -- insert object before index";
|
|
|
|
static char pop_doc[] =
|
|
|
|
"L.pop([index]) -> item -- remove and return item at index (default last)";
|
|
|
|
static char remove_doc[] =
|
|
|
|
"L.remove(value) -- remove first occurrence of value";
|
|
|
|
static char index_doc[] =
|
|
|
|
"L.index(value) -> integer -- return index of first occurrence of value";
|
|
|
|
static char count_doc[] =
|
|
|
|
"L.count(value) -> integer -- return number of occurrences of value";
|
|
|
|
static char reverse_doc[] =
|
|
|
|
"L.reverse() -- reverse *IN PLACE*";
|
|
|
|
static char sort_doc[] =
|
|
|
|
"L.sort([cmpfunc]) -- sort *IN PLACE*; if given, cmpfunc(x, y) -> -1, 0, 1";
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyMethodDef list_methods[] = {
|
2000-02-24 15:23:03 +00:00
|
|
|
{"append", (PyCFunction)listappend, 1, append_doc},
|
|
|
|
{"insert", (PyCFunction)listinsert, 1, insert_doc},
|
1998-10-09 16:37:25 +00:00
|
|
|
{"extend", (PyCFunction)listextend, 1, extend_doc},
|
1998-06-30 15:36:32 +00:00
|
|
|
{"pop", (PyCFunction)listpop, 1, pop_doc},
|
2000-02-24 15:23:03 +00:00
|
|
|
{"remove", (PyCFunction)listremove, 1, remove_doc},
|
|
|
|
{"index", (PyCFunction)listindex, 1, index_doc},
|
|
|
|
{"count", (PyCFunction)listcount, 1, count_doc},
|
|
|
|
{"reverse", (PyCFunction)listreverse, 1, reverse_doc},
|
|
|
|
{"sort", (PyCFunction)listsort, 1, sort_doc},
|
1990-10-14 12:07:46 +00:00
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
};
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
list_getattr(PyListObject *f, char *name)
|
1990-10-14 12:07:46 +00:00
|
|
|
{
|
1997-05-02 03:12:38 +00:00
|
|
|
return Py_FindMethod(list_methods, (PyObject *)f, name);
|
1990-10-14 12:07:46 +00:00
|
|
|
}
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
static PySequenceMethods list_as_sequence = {
|
1994-08-29 12:45:32 +00:00
|
|
|
(inquiry)list_length, /*sq_length*/
|
|
|
|
(binaryfunc)list_concat, /*sq_concat*/
|
|
|
|
(intargfunc)list_repeat, /*sq_repeat*/
|
|
|
|
(intargfunc)list_item, /*sq_item*/
|
|
|
|
(intintargfunc)list_slice, /*sq_slice*/
|
|
|
|
(intobjargproc)list_ass_item, /*sq_ass_item*/
|
|
|
|
(intintobjargproc)list_ass_slice, /*sq_ass_slice*/
|
2000-04-27 21:41:03 +00:00
|
|
|
(objobjproc)list_contains, /*sq_contains*/
|
1990-10-14 12:07:46 +00:00
|
|
|
};
|
|
|
|
|
1997-05-02 03:12:38 +00:00
|
|
|
PyTypeObject PyList_Type = {
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
1990-10-14 12:07:46 +00:00
|
|
|
0,
|
|
|
|
"list",
|
2000-06-30 05:02:53 +00:00
|
|
|
sizeof(PyListObject) + PyGC_HEAD_SIZE,
|
1990-10-14 12:07:46 +00:00
|
|
|
0,
|
1994-08-29 12:45:32 +00:00
|
|
|
(destructor)list_dealloc, /*tp_dealloc*/
|
|
|
|
(printfunc)list_print, /*tp_print*/
|
|
|
|
(getattrfunc)list_getattr, /*tp_getattr*/
|
1990-10-14 12:07:46 +00:00
|
|
|
0, /*tp_setattr*/
|
1994-08-29 12:45:32 +00:00
|
|
|
(cmpfunc)list_compare, /*tp_compare*/
|
|
|
|
(reprfunc)list_repr, /*tp_repr*/
|
1990-10-14 12:07:46 +00:00
|
|
|
0, /*tp_as_number*/
|
|
|
|
&list_as_sequence, /*tp_as_sequence*/
|
|
|
|
0, /*tp_as_mapping*/
|
2000-06-15 14:50:20 +00:00
|
|
|
0, /*tp_hash*/
|
|
|
|
0, /*tp_call*/
|
|
|
|
0, /*tp_str*/
|
|
|
|
0, /*tp_getattro*/
|
|
|
|
0, /*tp_setattro*/
|
|
|
|
0, /*tp_as_buffer*/
|
2000-06-23 19:37:02 +00:00
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
|
2000-06-23 14:18:11 +00:00
|
|
|
0, /* tp_doc */
|
|
|
|
(traverseproc)list_traverse, /* tp_traverse */
|
|
|
|
(inquiry)list_clear, /* tp_clear */
|
1990-10-14 12:07:46 +00:00
|
|
|
};
|
1998-06-16 15:18:28 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* During a sort, we really can't have anyone modifying the list; it could
|
|
|
|
cause core dumps. Thus, we substitute a dummy type that raises an
|
|
|
|
explanatory exception when a modifying operation is used. Caveat:
|
|
|
|
comparisons may behave differently; but I guess it's a bad idea anyway to
|
|
|
|
compare a list that's being sorted... */
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
immutable_list_op(void)
|
1998-06-16 15:18:28 +00:00
|
|
|
{
|
|
|
|
PyErr_SetString(PyExc_TypeError,
|
|
|
|
"a list cannot be modified while it is being sorted");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyMethodDef immutable_list_methods[] = {
|
|
|
|
{"append", (PyCFunction)immutable_list_op},
|
|
|
|
{"insert", (PyCFunction)immutable_list_op},
|
|
|
|
{"remove", (PyCFunction)immutable_list_op},
|
|
|
|
{"index", (PyCFunction)listindex},
|
|
|
|
{"count", (PyCFunction)listcount},
|
|
|
|
{"reverse", (PyCFunction)immutable_list_op},
|
|
|
|
{"sort", (PyCFunction)immutable_list_op},
|
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
};
|
|
|
|
|
|
|
|
static PyObject *
|
2000-07-09 15:16:51 +00:00
|
|
|
immutable_list_getattr(PyListObject *f, char *name)
|
1998-06-16 15:18:28 +00:00
|
|
|
{
|
|
|
|
return Py_FindMethod(immutable_list_methods, (PyObject *)f, name);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2000-07-09 15:16:51 +00:00
|
|
|
immutable_list_ass(void)
|
1998-06-16 15:18:28 +00:00
|
|
|
{
|
|
|
|
immutable_list_op();
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PySequenceMethods immutable_list_as_sequence = {
|
|
|
|
(inquiry)list_length, /*sq_length*/
|
|
|
|
(binaryfunc)list_concat, /*sq_concat*/
|
|
|
|
(intargfunc)list_repeat, /*sq_repeat*/
|
|
|
|
(intargfunc)list_item, /*sq_item*/
|
|
|
|
(intintargfunc)list_slice, /*sq_slice*/
|
|
|
|
(intobjargproc)immutable_list_ass, /*sq_ass_item*/
|
|
|
|
(intintobjargproc)immutable_list_ass, /*sq_ass_slice*/
|
2000-06-15 14:50:20 +00:00
|
|
|
(objobjproc)list_contains, /*sq_contains*/
|
1998-06-16 15:18:28 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
static PyTypeObject immutable_list_type = {
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
|
|
0,
|
|
|
|
"list (immutable, during sort)",
|
2000-06-30 05:02:53 +00:00
|
|
|
sizeof(PyListObject) + PyGC_HEAD_SIZE,
|
1998-06-16 15:18:28 +00:00
|
|
|
0,
|
|
|
|
0, /*tp_dealloc*/ /* Cannot happen */
|
|
|
|
(printfunc)list_print, /*tp_print*/
|
|
|
|
(getattrfunc)immutable_list_getattr, /*tp_getattr*/
|
|
|
|
0, /*tp_setattr*/
|
|
|
|
0, /*tp_compare*/ /* Won't be called */
|
|
|
|
(reprfunc)list_repr, /*tp_repr*/
|
|
|
|
0, /*tp_as_number*/
|
|
|
|
&immutable_list_as_sequence, /*tp_as_sequence*/
|
|
|
|
0, /*tp_as_mapping*/
|
2000-06-15 14:50:20 +00:00
|
|
|
0, /*tp_hash*/
|
|
|
|
0, /*tp_call*/
|
|
|
|
0, /*tp_str*/
|
|
|
|
0, /*tp_getattro*/
|
|
|
|
0, /*tp_setattro*/
|
|
|
|
0, /*tp_as_buffer*/
|
2000-06-23 19:37:02 +00:00
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
|
2000-06-23 14:18:11 +00:00
|
|
|
0, /* tp_doc */
|
|
|
|
(traverseproc)list_traverse, /* tp_traverse */
|
1998-06-16 15:18:28 +00:00
|
|
|
};
|