blob: 45b0951da3ebadc2e38bd95a92dd51749831ef34 [file] [log] [blame]
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001/* List object implementation */
2
Guido van Rossumc0b618a1997-05-02 03:12:38 +00003#include "Python.h"
4
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00005#ifdef STDC_HEADERS
6#include <stddef.h>
7#else
8#include <sys/types.h> /* For size_t */
9#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010
Guido van Rossuma46d51d1995-01-26 22:59:43 +000011static int
Fred Drakea2f55112000-07-09 15:16:51 +000012roundupsize(int n)
Guido van Rossuma46d51d1995-01-26 22:59:43 +000013{
Tim Peters65b8b842001-05-26 05:28:40 +000014 unsigned int nbits = 0;
15 unsigned int n2 = (unsigned int)n >> 5;
16
Tim Peters3b01a122002-07-19 02:35:45 +000017 /* Round up:
Tim Peters65b8b842001-05-26 05:28:40 +000018 * If n < 256, to a multiple of 8.
19 * If n < 2048, to a multiple of 64.
20 * If n < 16384, to a multiple of 512.
21 * If n < 131072, to a multiple of 4096.
22 * If n < 1048576, to a multiple of 32768.
23 * If n < 8388608, to a multiple of 262144.
24 * If n < 67108864, to a multiple of 2097152.
25 * If n < 536870912, to a multiple of 16777216.
26 * ...
27 * If n < 2**(5+3*i), to a multiple of 2**(3*i).
28 *
29 * This over-allocates proportional to the list size, making room
30 * for additional growth. The over-allocation is mild, but is
31 * enough to give linear-time amortized behavior over a long
32 * sequence of appends() in the presence of a poorly-performing
33 * system realloc() (which is a reality, e.g., across all flavors
34 * of Windows, with Win9x behavior being particularly bad -- and
35 * we've still got address space fragmentation problems on Win9x
36 * even with this scheme, although it requires much longer lists to
37 * provoke them than it used to).
38 */
39 do {
40 n2 >>= 3;
41 nbits += 3;
42 } while (n2);
43 return ((n >> nbits) + 1) << nbits;
44 }
Guido van Rossuma46d51d1995-01-26 22:59:43 +000045
Neal Norwitzd4e5be52002-05-22 23:19:17 +000046#define NRESIZE(var, type, nitems) \
47do { \
48 size_t _new_size = roundupsize(nitems); \
49 if (_new_size <= ((~(size_t)0) / sizeof(type))) \
50 PyMem_RESIZE(var, type, _new_size); \
51 else \
52 var = NULL; \
53} while (0)
Guido van Rossuma46d51d1995-01-26 22:59:43 +000054
Guido van Rossumc0b618a1997-05-02 03:12:38 +000055PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +000056PyList_New(int size)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000057{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000058 PyListObject *op;
Guido van Rossum6cd2fe01994-08-29 12:45:32 +000059 size_t nbytes;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000060 if (size < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000061 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000062 return NULL;
63 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000064 nbytes = size * sizeof(PyObject *);
Guido van Rossum1e28e5e1992-08-19 16:46:30 +000065 /* Check for overflow */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000066 if (nbytes / sizeof(PyObject *) != (size_t)size) {
67 return PyErr_NoMemory();
Guido van Rossum1e28e5e1992-08-19 16:46:30 +000068 }
Neil Schemenauere83c00e2001-08-29 23:54:21 +000069 op = PyObject_GC_New(PyListObject, &PyList_Type);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000070 if (op == NULL) {
Neil Schemenauere83c00e2001-08-29 23:54:21 +000071 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000072 }
73 if (size <= 0) {
74 op->ob_item = NULL;
75 }
76 else {
Guido van Rossumb18618d2000-05-03 23:44:39 +000077 op->ob_item = (PyObject **) PyMem_MALLOC(nbytes);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000078 if (op->ob_item == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000079 return PyErr_NoMemory();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000080 }
Neal Norwitz35fc7602002-06-13 21:11:11 +000081 memset(op->ob_item, 0, sizeof(*op->ob_item) * size);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000082 }
Neil Schemenauere83c00e2001-08-29 23:54:21 +000083 op->ob_size = size;
Neil Schemenauere83c00e2001-08-29 23:54:21 +000084 _PyObject_GC_TRACK(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000086}
87
88int
Fred Drakea2f55112000-07-09 15:16:51 +000089PyList_Size(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000090{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000091 if (!PyList_Check(op)) {
92 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000093 return -1;
94 }
95 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000096 return ((PyListObject *)op) -> ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000097}
98
Guido van Rossumc0b618a1997-05-02 03:12:38 +000099static PyObject *indexerr;
Guido van Rossum929f1b81996-08-09 20:51:27 +0000100
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000101PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000102PyList_GetItem(PyObject *op, int i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000103{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000104 if (!PyList_Check(op)) {
105 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000106 return NULL;
107 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000108 if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
Guido van Rossum929f1b81996-08-09 20:51:27 +0000109 if (indexerr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000110 indexerr = PyString_FromString(
111 "list index out of range");
112 PyErr_SetObject(PyExc_IndexError, indexerr);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000113 return NULL;
114 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000115 return ((PyListObject *)op) -> ob_item[i];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000116}
117
118int
Fred Drakea2f55112000-07-09 15:16:51 +0000119PyList_SetItem(register PyObject *op, register int i,
120 register PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000121{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000122 register PyObject *olditem;
123 register PyObject **p;
124 if (!PyList_Check(op)) {
125 Py_XDECREF(newitem);
126 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000127 return -1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000128 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000129 if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
130 Py_XDECREF(newitem);
131 PyErr_SetString(PyExc_IndexError,
132 "list assignment index out of range");
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000133 return -1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000134 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000135 p = ((PyListObject *)op) -> ob_item + i;
Guido van Rossum5fe60581995-03-09 12:12:50 +0000136 olditem = *p;
137 *p = newitem;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000138 Py_XDECREF(olditem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000139 return 0;
140}
141
142static int
Fred Drakea2f55112000-07-09 15:16:51 +0000143ins1(PyListObject *self, int where, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000144{
145 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000146 PyObject **items;
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000147 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000148 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000149 return -1;
150 }
Trent Micka5846642000-08-13 22:47:45 +0000151 if (self->ob_size == INT_MAX) {
152 PyErr_SetString(PyExc_OverflowError,
153 "cannot add more objects to list");
154 return -1;
155 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000156 items = self->ob_item;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000157 NRESIZE(items, PyObject *, self->ob_size+1);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000158 if (items == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000159 PyErr_NoMemory();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000160 return -1;
161 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000162 if (where < 0)
163 where = 0;
164 if (where > self->ob_size)
165 where = self->ob_size;
166 for (i = self->ob_size; --i >= where; )
167 items[i+1] = items[i];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000168 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000169 items[where] = v;
170 self->ob_item = items;
171 self->ob_size++;
172 return 0;
173}
174
175int
Fred Drakea2f55112000-07-09 15:16:51 +0000176PyList_Insert(PyObject *op, int where, PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000177{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000178 if (!PyList_Check(op)) {
179 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000180 return -1;
181 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000182 return ins1((PyListObject *)op, where, newitem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000183}
184
185int
Fred Drakea2f55112000-07-09 15:16:51 +0000186PyList_Append(PyObject *op, PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000187{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000188 if (!PyList_Check(op)) {
189 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000190 return -1;
191 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000192 return ins1((PyListObject *)op,
193 (int) ((PyListObject *)op)->ob_size, newitem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000194}
195
196/* Methods */
197
198static void
Fred Drakea2f55112000-07-09 15:16:51 +0000199list_dealloc(PyListObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000200{
201 int i;
Guido van Rossumff413af2002-03-28 20:34:59 +0000202 PyObject_GC_UnTrack(op);
Guido van Rossumd724b232000-03-13 16:01:29 +0000203 Py_TRASHCAN_SAFE_BEGIN(op)
Jack Jansen7874d1f1995-01-19 12:09:27 +0000204 if (op->ob_item != NULL) {
Guido van Rossumfa717011999-06-09 15:19:34 +0000205 /* Do it backwards, for Christian Tismer.
206 There's a simple test case where somehow this reduces
207 thrashing when a *very* large list is created and
208 immediately deleted. */
209 i = op->ob_size;
210 while (--i >= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000211 Py_XDECREF(op->ob_item[i]);
Jack Jansen7874d1f1995-01-19 12:09:27 +0000212 }
Guido van Rossumb18618d2000-05-03 23:44:39 +0000213 PyMem_FREE(op->ob_item);
Jack Jansen7874d1f1995-01-19 12:09:27 +0000214 }
Guido van Rossum9475a232001-10-05 20:51:39 +0000215 op->ob_type->tp_free((PyObject *)op);
Guido van Rossumd724b232000-03-13 16:01:29 +0000216 Py_TRASHCAN_SAFE_END(op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000217}
218
Guido van Rossum90933611991-06-07 16:10:43 +0000219static int
Fred Drakea2f55112000-07-09 15:16:51 +0000220list_print(PyListObject *op, FILE *fp, int flags)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000221{
222 int i;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000223
224 i = Py_ReprEnter((PyObject*)op);
225 if (i != 0) {
226 if (i < 0)
227 return i;
228 fprintf(fp, "[...]");
229 return 0;
230 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000231 fprintf(fp, "[");
Guido van Rossum90933611991-06-07 16:10:43 +0000232 for (i = 0; i < op->ob_size; i++) {
233 if (i > 0)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000234 fprintf(fp, ", ");
Guido van Rossumfb376de1998-04-10 22:47:27 +0000235 if (PyObject_Print(op->ob_item[i], fp, 0) != 0) {
236 Py_ReprLeave((PyObject *)op);
Guido van Rossum90933611991-06-07 16:10:43 +0000237 return -1;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000238 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000239 }
240 fprintf(fp, "]");
Guido van Rossumfb376de1998-04-10 22:47:27 +0000241 Py_ReprLeave((PyObject *)op);
Guido van Rossum90933611991-06-07 16:10:43 +0000242 return 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000243}
244
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000245static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000246list_repr(PyListObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000247{
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000248 int i;
Tim Petersa7259592001-06-16 05:11:17 +0000249 PyObject *s, *temp;
250 PyObject *pieces = NULL, *result = NULL;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000251
252 i = Py_ReprEnter((PyObject*)v);
253 if (i != 0) {
Tim Petersa7259592001-06-16 05:11:17 +0000254 return i > 0 ? PyString_FromString("[...]") : NULL;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000255 }
Tim Petersa7259592001-06-16 05:11:17 +0000256
257 if (v->ob_size == 0) {
258 result = PyString_FromString("[]");
259 goto Done;
260 }
261
262 pieces = PyList_New(0);
263 if (pieces == NULL)
264 goto Done;
265
266 /* Do repr() on each element. Note that this may mutate the list,
267 so must refetch the list size on each iteration. */
268 for (i = 0; i < v->ob_size; ++i) {
269 int status;
270 s = PyObject_Repr(v->ob_item[i]);
271 if (s == NULL)
272 goto Done;
273 status = PyList_Append(pieces, s);
274 Py_DECREF(s); /* append created a new ref */
275 if (status < 0)
276 goto Done;
277 }
278
279 /* Add "[]" decorations to the first and last items. */
280 assert(PyList_GET_SIZE(pieces) > 0);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000281 s = PyString_FromString("[");
Tim Petersa7259592001-06-16 05:11:17 +0000282 if (s == NULL)
283 goto Done;
284 temp = PyList_GET_ITEM(pieces, 0);
285 PyString_ConcatAndDel(&s, temp);
286 PyList_SET_ITEM(pieces, 0, s);
287 if (s == NULL)
288 goto Done;
289
290 s = PyString_FromString("]");
291 if (s == NULL)
292 goto Done;
293 temp = PyList_GET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1);
294 PyString_ConcatAndDel(&temp, s);
295 PyList_SET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1, temp);
296 if (temp == NULL)
297 goto Done;
298
299 /* Paste them all together with ", " between. */
300 s = PyString_FromString(", ");
301 if (s == NULL)
302 goto Done;
303 result = _PyString_Join(s, pieces);
Tim Peters3b01a122002-07-19 02:35:45 +0000304 Py_DECREF(s);
Tim Petersa7259592001-06-16 05:11:17 +0000305
306Done:
307 Py_XDECREF(pieces);
Guido van Rossumfb376de1998-04-10 22:47:27 +0000308 Py_ReprLeave((PyObject *)v);
Tim Petersa7259592001-06-16 05:11:17 +0000309 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000310}
311
312static int
Fred Drakea2f55112000-07-09 15:16:51 +0000313list_length(PyListObject *a)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000314{
315 return a->ob_size;
316}
317
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000318
319
320static int
Fred Drakea2f55112000-07-09 15:16:51 +0000321list_contains(PyListObject *a, PyObject *el)
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000322{
Raymond Hettingeraae59992002-09-05 14:23:49 +0000323 int i, cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000324
Raymond Hettingeraae59992002-09-05 14:23:49 +0000325 for (i = 0, cmp = 0 ; cmp == 0 && i < a->ob_size; ++i)
326 cmp = PyObject_RichCompareBool(el, PyList_GET_ITEM(a, i),
Guido van Rossum65e1cea2001-01-17 22:11:59 +0000327 Py_EQ);
Neal Norwitzbb9c5f52002-09-05 21:32:55 +0000328 return cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000329}
330
331
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000332static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000333list_item(PyListObject *a, int i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000334{
335 if (i < 0 || i >= a->ob_size) {
Guido van Rossum929f1b81996-08-09 20:51:27 +0000336 if (indexerr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000337 indexerr = PyString_FromString(
338 "list index out of range");
339 PyErr_SetObject(PyExc_IndexError, indexerr);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000340 return NULL;
341 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000342 Py_INCREF(a->ob_item[i]);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000343 return a->ob_item[i];
344}
345
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000346static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000347list_slice(PyListObject *a, int ilow, int ihigh)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000348{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000349 PyListObject *np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000350 int i;
351 if (ilow < 0)
352 ilow = 0;
353 else if (ilow > a->ob_size)
354 ilow = a->ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000355 if (ihigh < ilow)
356 ihigh = ilow;
357 else if (ihigh > a->ob_size)
358 ihigh = a->ob_size;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000359 np = (PyListObject *) PyList_New(ihigh - ilow);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000360 if (np == NULL)
361 return NULL;
362 for (i = ilow; i < ihigh; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000363 PyObject *v = a->ob_item[i];
364 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000365 np->ob_item[i - ilow] = v;
366 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000367 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000368}
369
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000370PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000371PyList_GetSlice(PyObject *a, int ilow, int ihigh)
Guido van Rossum234f9421993-06-17 12:35:49 +0000372{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000373 if (!PyList_Check(a)) {
374 PyErr_BadInternalCall();
Guido van Rossum234f9421993-06-17 12:35:49 +0000375 return NULL;
376 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000377 return list_slice((PyListObject *)a, ilow, ihigh);
Guido van Rossum234f9421993-06-17 12:35:49 +0000378}
379
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000380static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000381list_concat(PyListObject *a, PyObject *bb)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000382{
383 int size;
384 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000385 PyListObject *np;
386 if (!PyList_Check(bb)) {
Fred Drakeb6a9ada2000-06-01 03:12:13 +0000387 PyErr_Format(PyExc_TypeError,
Fred Drake914a2ed2000-06-01 14:31:03 +0000388 "can only concatenate list (not \"%.200s\") to list",
389 bb->ob_type->tp_name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000390 return NULL;
391 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000392#define b ((PyListObject *)bb)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000393 size = a->ob_size + b->ob_size;
Guido van Rossuma5c0e6d2002-10-11 21:05:56 +0000394 if (size < 0)
395 return PyErr_NoMemory();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000396 np = (PyListObject *) PyList_New(size);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000397 if (np == NULL) {
Guido van Rossum90933611991-06-07 16:10:43 +0000398 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000399 }
400 for (i = 0; i < a->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000401 PyObject *v = a->ob_item[i];
402 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000403 np->ob_item[i] = v;
404 }
405 for (i = 0; i < b->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000406 PyObject *v = b->ob_item[i];
407 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000408 np->ob_item[i + a->ob_size] = v;
409 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000410 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000411#undef b
412}
413
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000414static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000415list_repeat(PyListObject *a, int n)
Guido van Rossumed98d481991-03-06 13:07:53 +0000416{
417 int i, j;
418 int size;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000419 PyListObject *np;
420 PyObject **p;
Guido van Rossumed98d481991-03-06 13:07:53 +0000421 if (n < 0)
422 n = 0;
423 size = a->ob_size * n;
Guido van Rossumbfa5a142002-10-11 23:39:35 +0000424 if (n && size/n != a->ob_size)
Guido van Rossuma5c0e6d2002-10-11 21:05:56 +0000425 return PyErr_NoMemory();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000426 np = (PyListObject *) PyList_New(size);
Guido van Rossumed98d481991-03-06 13:07:53 +0000427 if (np == NULL)
428 return NULL;
429 p = np->ob_item;
430 for (i = 0; i < n; i++) {
431 for (j = 0; j < a->ob_size; j++) {
432 *p = a->ob_item[j];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000433 Py_INCREF(*p);
Guido van Rossumed98d481991-03-06 13:07:53 +0000434 p++;
435 }
436 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000437 return (PyObject *) np;
Guido van Rossumed98d481991-03-06 13:07:53 +0000438}
439
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000440static int
Fred Drakea2f55112000-07-09 15:16:51 +0000441list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000442{
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000443 /* Because [X]DECREF can recursively invoke list operations on
444 this list, we must postpone all [X]DECREF activity until
445 after the list is back in its canonical shape. Therefore
446 we must allocate an additional array, 'recycle', into which
447 we temporarily copy the items that are deleted from the
448 list. :-( */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000449 PyObject **recycle, **p;
450 PyObject **item;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000451 int n; /* Size of replacement list */
452 int d; /* Change in size */
453 int k; /* Loop index */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000454#define b ((PyListObject *)v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000455 if (v == NULL)
456 n = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000457 else if (PyList_Check(v)) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000458 n = b->ob_size;
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000459 if (a == b) {
460 /* Special case "a[i:j] = a" -- copy b first */
461 int ret;
462 v = list_slice(b, 0, n);
463 ret = list_ass_slice(a, ilow, ihigh, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000464 Py_DECREF(v);
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000465 return ret;
466 }
467 }
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000468 else {
Fred Drakeb6a9ada2000-06-01 03:12:13 +0000469 PyErr_Format(PyExc_TypeError,
470 "must assign list (not \"%.200s\") to slice",
471 v->ob_type->tp_name);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000472 return -1;
473 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000474 if (ilow < 0)
475 ilow = 0;
476 else if (ilow > a->ob_size)
477 ilow = a->ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000478 if (ihigh < ilow)
479 ihigh = ilow;
480 else if (ihigh > a->ob_size)
481 ihigh = a->ob_size;
482 item = a->ob_item;
483 d = n - (ihigh-ilow);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000484 if (ihigh > ilow)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000485 p = recycle = PyMem_NEW(PyObject *, (ihigh-ilow));
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000486 else
487 p = recycle = NULL;
488 if (d <= 0) { /* Delete -d items; recycle ihigh-ilow items */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000489 for (k = ilow; k < ihigh; k++)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000490 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000491 if (d < 0) {
492 for (/*k = ihigh*/; k < a->ob_size; k++)
493 item[k+d] = item[k];
494 a->ob_size += d;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000495 NRESIZE(item, PyObject *, a->ob_size); /* Can't fail */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000496 a->ob_item = item;
497 }
498 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000499 else { /* Insert d items; recycle ihigh-ilow items */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000500 NRESIZE(item, PyObject *, a->ob_size + d);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000501 if (item == NULL) {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000502 if (recycle != NULL)
503 PyMem_DEL(recycle);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000504 PyErr_NoMemory();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000505 return -1;
506 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000507 for (k = a->ob_size; --k >= ihigh; )
508 item[k+d] = item[k];
509 for (/*k = ihigh-1*/; k >= ilow; --k)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000510 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000511 a->ob_item = item;
512 a->ob_size += d;
513 }
514 for (k = 0; k < n; k++, ilow++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000515 PyObject *w = b->ob_item[k];
516 Py_XINCREF(w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000517 item[ilow] = w;
518 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000519 if (recycle) {
520 while (--p >= recycle)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000521 Py_XDECREF(*p);
522 PyMem_DEL(recycle);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000523 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000524 if (a->ob_size == 0 && a->ob_item != NULL) {
525 PyMem_FREE(a->ob_item);
526 a->ob_item = NULL;
527 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000528 return 0;
529#undef b
530}
531
Guido van Rossum234f9421993-06-17 12:35:49 +0000532int
Fred Drakea2f55112000-07-09 15:16:51 +0000533PyList_SetSlice(PyObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum234f9421993-06-17 12:35:49 +0000534{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000535 if (!PyList_Check(a)) {
536 PyErr_BadInternalCall();
Guido van Rossum1fc238a1993-07-29 08:25:09 +0000537 return -1;
Guido van Rossum234f9421993-06-17 12:35:49 +0000538 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000539 return list_ass_slice((PyListObject *)a, ilow, ihigh, v);
Guido van Rossum234f9421993-06-17 12:35:49 +0000540}
541
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000542static PyObject *
543list_inplace_repeat(PyListObject *self, int n)
544{
545 PyObject **items;
546 int size, i, j;
547
548
549 size = PyList_GET_SIZE(self);
550 if (size == 0) {
551 Py_INCREF(self);
552 return (PyObject *)self;
553 }
554
555 items = self->ob_item;
556
557 if (n < 1) {
558 self->ob_item = NULL;
559 self->ob_size = 0;
560 for (i = 0; i < size; i++)
561 Py_XDECREF(items[i]);
562 PyMem_DEL(items);
563 Py_INCREF(self);
564 return (PyObject *)self;
565 }
566
567 NRESIZE(items, PyObject*, size*n);
568 if (items == NULL) {
569 PyErr_NoMemory();
570 goto finally;
571 }
572 self->ob_item = items;
573 for (i = 1; i < n; i++) { /* Start counting at 1, not 0 */
574 for (j = 0; j < size; j++) {
575 PyObject *o = PyList_GET_ITEM(self, j);
576 Py_INCREF(o);
577 PyList_SET_ITEM(self, self->ob_size++, o);
578 }
579 }
580 Py_INCREF(self);
581 return (PyObject *)self;
582 finally:
583 return NULL;
584}
585
Guido van Rossum4a450d01991-04-03 19:05:18 +0000586static int
Fred Drakea2f55112000-07-09 15:16:51 +0000587list_ass_item(PyListObject *a, int i, PyObject *v)
Guido van Rossum4a450d01991-04-03 19:05:18 +0000588{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000589 PyObject *old_value;
Guido van Rossum4a450d01991-04-03 19:05:18 +0000590 if (i < 0 || i >= a->ob_size) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000591 PyErr_SetString(PyExc_IndexError,
592 "list assignment index out of range");
Guido van Rossum4a450d01991-04-03 19:05:18 +0000593 return -1;
594 }
595 if (v == NULL)
596 return list_ass_slice(a, i, i+1, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000597 Py_INCREF(v);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000598 old_value = a->ob_item[i];
Guido van Rossum4a450d01991-04-03 19:05:18 +0000599 a->ob_item[i] = v;
Tim Peters3b01a122002-07-19 02:35:45 +0000600 Py_DECREF(old_value);
Guido van Rossum4a450d01991-04-03 19:05:18 +0000601 return 0;
602}
603
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000604static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000605ins(PyListObject *self, int where, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000606{
607 if (ins1(self, where, v) != 0)
608 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000609 Py_INCREF(Py_None);
610 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000611}
612
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000613static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000614listinsert(PyListObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000615{
616 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000617 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000618 if (!PyArg_ParseTuple(args, "iO:insert", &i, &v))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000619 return NULL;
Guido van Rossumbf80e541993-02-08 15:49:17 +0000620 return ins(self, i, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000621}
622
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000623static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000624listappend(PyListObject *self, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000625{
Guido van Rossumbf80e541993-02-08 15:49:17 +0000626 return ins(self, (int) self->ob_size, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000627}
628
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000629static int
630listextend_internal(PyListObject *self, PyObject *b)
Barry Warsawdedf6d61998-10-09 16:37:25 +0000631{
Barry Warsawdedf6d61998-10-09 16:37:25 +0000632 PyObject **items;
633 int selflen = PyList_GET_SIZE(self);
634 int blen;
635 register int i;
636
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000637 if (PyObject_Size(b) == 0) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000638 /* short circuit when b is empty */
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000639 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000640 return 0;
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000641 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000642
Barry Warsawdedf6d61998-10-09 16:37:25 +0000643 if (self == (PyListObject*)b) {
644 /* as in list_ass_slice() we must special case the
645 * situation: a.extend(a)
646 *
647 * XXX: I think this way ought to be faster than using
648 * list_slice() the way list_ass_slice() does.
649 */
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000650 Py_DECREF(b);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000651 b = PyList_New(selflen);
652 if (!b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000653 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000654 for (i = 0; i < selflen; i++) {
655 PyObject *o = PyList_GET_ITEM(self, i);
656 Py_INCREF(o);
657 PyList_SET_ITEM(b, i, o);
658 }
659 }
Barry Warsawdedf6d61998-10-09 16:37:25 +0000660
Jeremy Hylton03657cf2000-07-12 13:05:33 +0000661 blen = PyObject_Size(b);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000662
Barry Warsawdedf6d61998-10-09 16:37:25 +0000663 /* resize a using idiom */
664 items = self->ob_item;
665 NRESIZE(items, PyObject*, selflen + blen);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000666 if (items == NULL) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000667 PyErr_NoMemory();
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000668 Py_DECREF(b);
669 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000670 }
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000671
Barry Warsawdedf6d61998-10-09 16:37:25 +0000672 self->ob_item = items;
673
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000674 /* populate the end of self with b's items */
Barry Warsawdedf6d61998-10-09 16:37:25 +0000675 for (i = 0; i < blen; i++) {
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000676 PyObject *o = PySequence_Fast_GET_ITEM(b, i);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000677 Py_INCREF(o);
678 PyList_SET_ITEM(self, self->ob_size++, o);
679 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000680 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000681 return 0;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000682}
683
684
685static PyObject *
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000686list_inplace_concat(PyListObject *self, PyObject *other)
687{
Tim Peters1af03e92001-05-26 19:37:54 +0000688 other = PySequence_Fast(other, "argument to += must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000689 if (!other)
690 return NULL;
691
692 if (listextend_internal(self, other) < 0)
693 return NULL;
694
695 Py_INCREF(self);
696 return (PyObject *)self;
697}
698
699static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000700listextend(PyListObject *self, PyObject *b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000701{
702
Tim Peters1af03e92001-05-26 19:37:54 +0000703 b = PySequence_Fast(b, "list.extend() argument must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000704 if (!b)
705 return NULL;
706
707 if (listextend_internal(self, b) < 0)
708 return NULL;
709
710 Py_INCREF(Py_None);
711 return Py_None;
712}
713
714static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000715listpop(PyListObject *self, PyObject *args)
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000716{
717 int i = -1;
718 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000719 if (!PyArg_ParseTuple(args, "|i:pop", &i))
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000720 return NULL;
721 if (self->ob_size == 0) {
722 /* Special-case most common failure cause */
723 PyErr_SetString(PyExc_IndexError, "pop from empty list");
724 return NULL;
725 }
726 if (i < 0)
727 i += self->ob_size;
728 if (i < 0 || i >= self->ob_size) {
729 PyErr_SetString(PyExc_IndexError, "pop index out of range");
730 return NULL;
731 }
732 v = self->ob_item[i];
733 Py_INCREF(v);
734 if (list_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {
735 Py_DECREF(v);
736 return NULL;
737 }
738 return v;
739}
740
Tim Peters8e2e7ca2002-07-19 02:33:08 +0000741/* Reverse a slice of a list in place, from lo up to (exclusive) hi. */
742static void
743reverse_slice(PyObject **lo, PyObject **hi)
744{
745 assert(lo && hi);
746
747 --hi;
748 while (lo < hi) {
749 PyObject *t = *lo;
750 *lo = *hi;
751 *hi = t;
752 ++lo;
753 --hi;
754 }
755}
756
Tim Petersa64dc242002-08-01 02:13:36 +0000757/* Lots of code for an adaptive, stable, natural mergesort. There are many
758 * pieces to this algorithm; read listsort.txt for overviews and details.
759 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000760
Guido van Rossum3f236de1996-12-10 23:55:39 +0000761/* Comparison function. Takes care of calling a user-supplied
Tim Peters66860f62002-08-04 17:47:26 +0000762 * comparison function (any callable Python object), which must not be
763 * NULL (use the ISLT macro if you don't know, or call PyObject_RichCompareBool
764 * with Py_LT if you know it's NULL).
Tim Petersa64dc242002-08-01 02:13:36 +0000765 * Returns -1 on error, 1 if x < y, 0 if x >= y.
766 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000767static int
Tim Petersa8c974c2002-07-19 03:30:57 +0000768islt(PyObject *x, PyObject *y, PyObject *compare)
Guido van Rossum3f236de1996-12-10 23:55:39 +0000769{
Tim Petersf2a04732002-07-11 21:46:16 +0000770 PyObject *res;
771 PyObject *args;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000772 int i;
773
Tim Peters66860f62002-08-04 17:47:26 +0000774 assert(compare != NULL);
Tim Petersa8c974c2002-07-19 03:30:57 +0000775 /* Call the user's comparison function and translate the 3-way
776 * result into true or false (or error).
777 */
Tim Petersf2a04732002-07-11 21:46:16 +0000778 args = PyTuple_New(2);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000779 if (args == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000780 return -1;
Tim Petersf2a04732002-07-11 21:46:16 +0000781 Py_INCREF(x);
782 Py_INCREF(y);
783 PyTuple_SET_ITEM(args, 0, x);
784 PyTuple_SET_ITEM(args, 1, y);
Tim Peters58cf3612002-07-15 05:16:13 +0000785 res = PyObject_Call(compare, args, NULL);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000786 Py_DECREF(args);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000787 if (res == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000788 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000789 if (!PyInt_Check(res)) {
790 Py_DECREF(res);
791 PyErr_SetString(PyExc_TypeError,
Guido van Rossum9bcd1d71999-04-19 17:44:39 +0000792 "comparison function must return int");
Tim Petersa8c974c2002-07-19 03:30:57 +0000793 return -1;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000794 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000795 i = PyInt_AsLong(res);
796 Py_DECREF(res);
Tim Petersa8c974c2002-07-19 03:30:57 +0000797 return i < 0;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000798}
799
Tim Peters66860f62002-08-04 17:47:26 +0000800/* If COMPARE is NULL, calls PyObject_RichCompareBool with Py_LT, else calls
801 * islt. This avoids a layer of function call in the usual case, and
802 * sorting does many comparisons.
803 * Returns -1 on error, 1 if x < y, 0 if x >= y.
804 */
805#define ISLT(X, Y, COMPARE) ((COMPARE) == NULL ? \
806 PyObject_RichCompareBool(X, Y, Py_LT) : \
807 islt(X, Y, COMPARE))
808
809/* Compare X to Y via "<". Goto "fail" if the comparison raises an
Tim Petersa8c974c2002-07-19 03:30:57 +0000810 error. Else "k" is set to true iff X<Y, and an "if (k)" block is
811 started. It makes more sense in context <wink>. X and Y are PyObject*s.
812*/
Tim Peters66860f62002-08-04 17:47:26 +0000813#define IFLT(X, Y) if ((k = ISLT(X, Y, compare)) < 0) goto fail; \
Tim Petersa8c974c2002-07-19 03:30:57 +0000814 if (k)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000815
816/* binarysort is the best method for sorting small arrays: it does
817 few compares, but can do data movement quadratic in the number of
818 elements.
Guido van Rossum42812581998-06-17 14:15:44 +0000819 [lo, hi) is a contiguous slice of a list, and is sorted via
Tim Petersa8c974c2002-07-19 03:30:57 +0000820 binary insertion. This sort is stable.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000821 On entry, must have lo <= start <= hi, and that [lo, start) is already
822 sorted (pass start == lo if you don't know!).
Tim Petersa8c974c2002-07-19 03:30:57 +0000823 If islt() complains return -1, else 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000824 Even in case of error, the output slice will be some permutation of
825 the input (nothing is lost or duplicated).
826*/
Guido van Rossum3f236de1996-12-10 23:55:39 +0000827static int
Fred Drakea2f55112000-07-09 15:16:51 +0000828binarysort(PyObject **lo, PyObject **hi, PyObject **start, PyObject *compare)
829 /* compare -- comparison function object, or NULL for default */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000830{
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000831 register int k;
832 register PyObject **l, **p, **r;
833 register PyObject *pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000834
Tim Petersa8c974c2002-07-19 03:30:57 +0000835 assert(lo <= start && start <= hi);
836 /* assert [lo, start) is sorted */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000837 if (lo == start)
838 ++start;
839 for (; start < hi; ++start) {
840 /* set l to where *start belongs */
841 l = lo;
842 r = start;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000843 pivot = *r;
Tim Peters0fe977c2002-07-19 06:12:32 +0000844 /* Invariants:
845 * pivot >= all in [lo, l).
846 * pivot < all in [r, start).
847 * The second is vacuously true at the start.
848 */
849 assert(l < r);
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000850 do {
851 p = l + ((r - l) >> 1);
Tim Petersa8c974c2002-07-19 03:30:57 +0000852 IFLT(pivot, *p)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000853 r = p;
854 else
Tim Peters0fe977c2002-07-19 06:12:32 +0000855 l = p+1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000856 } while (l < r);
Tim Peters0fe977c2002-07-19 06:12:32 +0000857 assert(l == r);
858 /* The invariants still hold, so pivot >= all in [lo, l) and
859 pivot < all in [l, start), so pivot belongs at l. Note
860 that if there are elements equal to pivot, l points to the
861 first slot after them -- that's why this sort is stable.
862 Slide over to make room.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000863 Caution: using memmove is much slower under MSVC 5;
864 we're not usually moving many slots. */
865 for (p = start; p > l; --p)
866 *p = *(p-1);
867 *l = pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000868 }
Guido van Rossum3f236de1996-12-10 23:55:39 +0000869 return 0;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000870
871 fail:
872 return -1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000873}
874
Tim Petersa64dc242002-08-01 02:13:36 +0000875/*
876Return the length of the run beginning at lo, in the slice [lo, hi). lo < hi
877is required on entry. "A run" is the longest ascending sequence, with
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000878
Tim Petersa64dc242002-08-01 02:13:36 +0000879 lo[0] <= lo[1] <= lo[2] <= ...
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000880
Tim Petersa64dc242002-08-01 02:13:36 +0000881or the longest descending sequence, with
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000882
Tim Petersa64dc242002-08-01 02:13:36 +0000883 lo[0] > lo[1] > lo[2] > ...
Tim Peters3b01a122002-07-19 02:35:45 +0000884
Tim Petersa64dc242002-08-01 02:13:36 +0000885Boolean *descending is set to 0 in the former case, or to 1 in the latter.
886For its intended use in a stable mergesort, the strictness of the defn of
887"descending" is needed so that the caller can safely reverse a descending
888sequence without violating stability (strict > ensures there are no equal
889elements to get out of order).
890
891Returns -1 in case of error.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000892*/
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000893static int
Tim Petersa64dc242002-08-01 02:13:36 +0000894count_run(PyObject **lo, PyObject **hi, PyObject *compare, int *descending)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000895{
Tim Petersa64dc242002-08-01 02:13:36 +0000896 int k;
897 int n;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000898
Tim Petersa64dc242002-08-01 02:13:36 +0000899 assert(lo < hi);
900 *descending = 0;
901 ++lo;
902 if (lo == hi)
903 return 1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000904
Tim Petersa64dc242002-08-01 02:13:36 +0000905 n = 2;
906 IFLT(*lo, *(lo-1)) {
907 *descending = 1;
908 for (lo = lo+1; lo < hi; ++lo, ++n) {
909 IFLT(*lo, *(lo-1))
910 ;
911 else
912 break;
913 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000914 }
Tim Petersa64dc242002-08-01 02:13:36 +0000915 else {
916 for (lo = lo+1; lo < hi; ++lo, ++n) {
917 IFLT(*lo, *(lo-1))
918 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000919 }
920 }
921
Tim Petersa64dc242002-08-01 02:13:36 +0000922 return n;
923fail:
924 return -1;
925}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000926
Tim Petersa64dc242002-08-01 02:13:36 +0000927/*
928Locate the proper position of key in a sorted vector; if the vector contains
929an element equal to key, return the position immediately to the left of
930the leftmost equal element. [gallop_right() does the same except returns
931the position to the right of the rightmost equal element (if any).]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000932
Tim Petersa64dc242002-08-01 02:13:36 +0000933"a" is a sorted vector with n elements, starting at a[0]. n must be > 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000934
Tim Petersa64dc242002-08-01 02:13:36 +0000935"hint" is an index at which to begin the search, 0 <= hint < n. The closer
936hint is to the final result, the faster this runs.
937
938The return value is the int k in 0..n such that
939
940 a[k-1] < key <= a[k]
941
942pretending that *(a-1) is minus infinity and a[n] is plus infinity. IOW,
943key belongs at index k; or, IOW, the first k elements of a should precede
944key, and the last n-k should follow key.
945
946Returns -1 on error. See listsort.txt for info on the method.
947*/
948static int
949gallop_left(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
950{
951 int ofs;
952 int lastofs;
953 int k;
954
955 assert(key && a && n > 0 && hint >= 0 && hint < n);
956
957 a += hint;
958 lastofs = 0;
959 ofs = 1;
960 IFLT(*a, key) {
961 /* a[hint] < key -- gallop right, until
962 * a[hint + lastofs] < key <= a[hint + ofs]
963 */
964 const int maxofs = n - hint; /* &a[n-1] is highest */
965 while (ofs < maxofs) {
966 IFLT(a[ofs], key) {
967 lastofs = ofs;
968 ofs = (ofs << 1) + 1;
969 if (ofs <= 0) /* int overflow */
970 ofs = maxofs;
971 }
972 else /* key <= a[hint + ofs] */
973 break;
974 }
975 if (ofs > maxofs)
976 ofs = maxofs;
977 /* Translate back to offsets relative to &a[0]. */
978 lastofs += hint;
979 ofs += hint;
980 }
981 else {
982 /* key <= a[hint] -- gallop left, until
983 * a[hint - ofs] < key <= a[hint - lastofs]
984 */
985 const int maxofs = hint + 1; /* &a[0] is lowest */
986 while (ofs < maxofs) {
987 IFLT(*(a-ofs), key)
988 break;
989 /* key <= a[hint - ofs] */
990 lastofs = ofs;
991 ofs = (ofs << 1) + 1;
992 if (ofs <= 0) /* int overflow */
993 ofs = maxofs;
994 }
995 if (ofs > maxofs)
996 ofs = maxofs;
997 /* Translate back to positive offsets relative to &a[0]. */
998 k = lastofs;
999 lastofs = hint - ofs;
1000 ofs = hint - k;
1001 }
1002 a -= hint;
1003
1004 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1005 /* Now a[lastofs] < key <= a[ofs], so key belongs somewhere to the
1006 * right of lastofs but no farther right than ofs. Do a binary
1007 * search, with invariant a[lastofs-1] < key <= a[ofs].
1008 */
1009 ++lastofs;
1010 while (lastofs < ofs) {
1011 int m = lastofs + ((ofs - lastofs) >> 1);
1012
1013 IFLT(a[m], key)
1014 lastofs = m+1; /* a[m] < key */
1015 else
1016 ofs = m; /* key <= a[m] */
1017 }
1018 assert(lastofs == ofs); /* so a[ofs-1] < key <= a[ofs] */
1019 return ofs;
1020
1021fail:
1022 return -1;
1023}
1024
1025/*
1026Exactly like gallop_left(), except that if key already exists in a[0:n],
1027finds the position immediately to the right of the rightmost equal value.
1028
1029The return value is the int k in 0..n such that
1030
1031 a[k-1] <= key < a[k]
1032
1033or -1 if error.
1034
1035The code duplication is massive, but this is enough different given that
1036we're sticking to "<" comparisons that it's much harder to follow if
1037written as one routine with yet another "left or right?" flag.
1038*/
1039static int
1040gallop_right(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
1041{
1042 int ofs;
1043 int lastofs;
1044 int k;
1045
1046 assert(key && a && n > 0 && hint >= 0 && hint < n);
1047
1048 a += hint;
1049 lastofs = 0;
1050 ofs = 1;
1051 IFLT(key, *a) {
1052 /* key < a[hint] -- gallop left, until
1053 * a[hint - ofs] <= key < a[hint - lastofs]
1054 */
1055 const int maxofs = hint + 1; /* &a[0] is lowest */
1056 while (ofs < maxofs) {
1057 IFLT(key, *(a-ofs)) {
1058 lastofs = ofs;
1059 ofs = (ofs << 1) + 1;
1060 if (ofs <= 0) /* int overflow */
1061 ofs = maxofs;
1062 }
1063 else /* a[hint - ofs] <= key */
1064 break;
1065 }
1066 if (ofs > maxofs)
1067 ofs = maxofs;
1068 /* Translate back to positive offsets relative to &a[0]. */
1069 k = lastofs;
1070 lastofs = hint - ofs;
1071 ofs = hint - k;
1072 }
1073 else {
1074 /* a[hint] <= key -- gallop right, until
1075 * a[hint + lastofs] <= key < a[hint + ofs]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001076 */
Tim Petersa64dc242002-08-01 02:13:36 +00001077 const int maxofs = n - hint; /* &a[n-1] is highest */
1078 while (ofs < maxofs) {
1079 IFLT(key, a[ofs])
1080 break;
1081 /* a[hint + ofs] <= key */
1082 lastofs = ofs;
1083 ofs = (ofs << 1) + 1;
1084 if (ofs <= 0) /* int overflow */
1085 ofs = maxofs;
1086 }
1087 if (ofs > maxofs)
1088 ofs = maxofs;
1089 /* Translate back to offsets relative to &a[0]. */
1090 lastofs += hint;
1091 ofs += hint;
1092 }
1093 a -= hint;
1094
1095 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1096 /* Now a[lastofs] <= key < a[ofs], so key belongs somewhere to the
1097 * right of lastofs but no farther right than ofs. Do a binary
1098 * search, with invariant a[lastofs-1] <= key < a[ofs].
1099 */
1100 ++lastofs;
1101 while (lastofs < ofs) {
1102 int m = lastofs + ((ofs - lastofs) >> 1);
1103
1104 IFLT(key, a[m])
1105 ofs = m; /* key < a[m] */
1106 else
1107 lastofs = m+1; /* a[m] <= key */
1108 }
1109 assert(lastofs == ofs); /* so a[ofs-1] <= key < a[ofs] */
1110 return ofs;
1111
1112fail:
1113 return -1;
1114}
1115
1116/* The maximum number of entries in a MergeState's pending-runs stack.
1117 * This is enough to sort arrays of size up to about
1118 * 32 * phi ** MAX_MERGE_PENDING
1119 * where phi ~= 1.618. 85 is ridiculouslylarge enough, good for an array
1120 * with 2**64 elements.
1121 */
1122#define MAX_MERGE_PENDING 85
1123
Tim Peterse05f65a2002-08-10 05:21:15 +00001124/* When we get into galloping mode, we stay there until both runs win less
1125 * often than MIN_GALLOP consecutive times. See listsort.txt for more info.
Tim Petersa64dc242002-08-01 02:13:36 +00001126 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001127#define MIN_GALLOP 7
Tim Petersa64dc242002-08-01 02:13:36 +00001128
1129/* Avoid malloc for small temp arrays. */
1130#define MERGESTATE_TEMP_SIZE 256
1131
1132/* One MergeState exists on the stack per invocation of mergesort. It's just
1133 * a convenient way to pass state around among the helper functions.
1134 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001135struct s_slice {
1136 PyObject **base;
1137 int len;
1138};
1139
Tim Petersa64dc242002-08-01 02:13:36 +00001140typedef struct s_MergeState {
1141 /* The user-supplied comparison function. or NULL if none given. */
1142 PyObject *compare;
1143
Tim Peterse05f65a2002-08-10 05:21:15 +00001144 /* This controls when we get *into* galloping mode. It's initialized
1145 * to MIN_GALLOP. merge_lo and merge_hi tend to nudge it higher for
1146 * random data, and lower for highly structured data.
1147 */
1148 int min_gallop;
1149
Tim Petersa64dc242002-08-01 02:13:36 +00001150 /* 'a' is temp storage to help with merges. It contains room for
1151 * alloced entries.
1152 */
1153 PyObject **a; /* may point to temparray below */
1154 int alloced;
1155
1156 /* A stack of n pending runs yet to be merged. Run #i starts at
1157 * address base[i] and extends for len[i] elements. It's always
1158 * true (so long as the indices are in bounds) that
1159 *
Tim Peterse05f65a2002-08-10 05:21:15 +00001160 * pending[i].base + pending[i].len == pending[i+1].base
Tim Petersa64dc242002-08-01 02:13:36 +00001161 *
1162 * so we could cut the storage for this, but it's a minor amount,
1163 * and keeping all the info explicit simplifies the code.
1164 */
1165 int n;
Tim Peterse05f65a2002-08-10 05:21:15 +00001166 struct s_slice pending[MAX_MERGE_PENDING];
Tim Petersa64dc242002-08-01 02:13:36 +00001167
1168 /* 'a' points to this when possible, rather than muck with malloc. */
1169 PyObject *temparray[MERGESTATE_TEMP_SIZE];
1170} MergeState;
1171
1172/* Conceptually a MergeState's constructor. */
1173static void
1174merge_init(MergeState *ms, PyObject *compare)
1175{
1176 assert(ms != NULL);
1177 ms->compare = compare;
1178 ms->a = ms->temparray;
1179 ms->alloced = MERGESTATE_TEMP_SIZE;
1180 ms->n = 0;
Tim Peterse05f65a2002-08-10 05:21:15 +00001181 ms->min_gallop = MIN_GALLOP;
Tim Petersa64dc242002-08-01 02:13:36 +00001182}
1183
1184/* Free all the temp memory owned by the MergeState. This must be called
1185 * when you're done with a MergeState, and may be called before then if
1186 * you want to free the temp memory early.
1187 */
1188static void
1189merge_freemem(MergeState *ms)
1190{
1191 assert(ms != NULL);
1192 if (ms->a != ms->temparray)
1193 PyMem_Free(ms->a);
1194 ms->a = ms->temparray;
1195 ms->alloced = MERGESTATE_TEMP_SIZE;
1196}
1197
1198/* Ensure enough temp memory for 'need' array slots is available.
1199 * Returns 0 on success and -1 if the memory can't be gotten.
1200 */
1201static int
1202merge_getmem(MergeState *ms, int need)
1203{
1204 assert(ms != NULL);
1205 if (need <= ms->alloced)
1206 return 0;
1207 /* Don't realloc! That can cost cycles to copy the old data, but
1208 * we don't care what's in the block.
1209 */
1210 merge_freemem(ms);
1211 ms->a = (PyObject **)PyMem_Malloc(need * sizeof(PyObject*));
1212 if (ms->a) {
1213 ms->alloced = need;
1214 return 0;
1215 }
1216 PyErr_NoMemory();
1217 merge_freemem(ms); /* reset to sane state */
1218 return -1;
1219}
1220#define MERGE_GETMEM(MS, NEED) ((NEED) <= (MS)->alloced ? 0 : \
1221 merge_getmem(MS, NEED))
1222
1223/* Merge the na elements starting at pa with the nb elements starting at pb
1224 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1225 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1226 * merge, and should have na <= nb. See listsort.txt for more info.
1227 * Return 0 if successful, -1 if error.
1228 */
1229static int
1230merge_lo(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1231{
1232 int k;
1233 PyObject *compare;
1234 PyObject **dest;
1235 int result = -1; /* guilty until proved innocent */
Tim Peterse05f65a2002-08-10 05:21:15 +00001236 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001237
1238 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1239 if (MERGE_GETMEM(ms, na) < 0)
1240 return -1;
1241 memcpy(ms->a, pa, na * sizeof(PyObject*));
1242 dest = pa;
1243 pa = ms->a;
1244
1245 *dest++ = *pb++;
1246 --nb;
1247 if (nb == 0)
1248 goto Succeed;
1249 if (na == 1)
1250 goto CopyB;
1251
1252 compare = ms->compare;
1253 for (;;) {
1254 int acount = 0; /* # of times A won in a row */
1255 int bcount = 0; /* # of times B won in a row */
1256
1257 /* Do the straightforward thing until (if ever) one run
1258 * appears to win consistently.
1259 */
1260 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001261 assert(na > 1 && nb > 0);
Tim Peters66860f62002-08-04 17:47:26 +00001262 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001263 if (k) {
1264 if (k < 0)
1265 goto Fail;
1266 *dest++ = *pb++;
1267 ++bcount;
1268 acount = 0;
1269 --nb;
1270 if (nb == 0)
1271 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001272 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001273 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001274 }
1275 else {
Tim Petersa64dc242002-08-01 02:13:36 +00001276 *dest++ = *pa++;
1277 ++acount;
1278 bcount = 0;
1279 --na;
1280 if (na == 1)
1281 goto CopyB;
Tim Peterse05f65a2002-08-10 05:21:15 +00001282 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001283 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001284 }
Tim Petersa64dc242002-08-01 02:13:36 +00001285 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001286
Tim Petersa64dc242002-08-01 02:13:36 +00001287 /* One run is winning so consistently that galloping may
1288 * be a huge win. So try that, and continue galloping until
1289 * (if ever) neither run appears to be winning consistently
1290 * anymore.
1291 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001292 ++min_gallop;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001293 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001294 assert(na > 1 && nb > 0);
1295 min_gallop -= min_gallop > 1;
1296 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001297 k = gallop_right(*pb, pa, na, 0, compare);
1298 acount = k;
1299 if (k) {
1300 if (k < 0)
1301 goto Fail;
1302 memcpy(dest, pa, k * sizeof(PyObject *));
1303 dest += k;
1304 pa += k;
1305 na -= k;
1306 if (na == 1)
1307 goto CopyB;
1308 /* na==0 is impossible now if the comparison
1309 * function is consistent, but we can't assume
1310 * that it is.
1311 */
1312 if (na == 0)
1313 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001314 }
Tim Petersa64dc242002-08-01 02:13:36 +00001315 *dest++ = *pb++;
1316 --nb;
1317 if (nb == 0)
1318 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001319
Tim Petersa64dc242002-08-01 02:13:36 +00001320 k = gallop_left(*pa, pb, nb, 0, compare);
1321 bcount = k;
1322 if (k) {
1323 if (k < 0)
1324 goto Fail;
1325 memmove(dest, pb, k * sizeof(PyObject *));
1326 dest += k;
1327 pb += k;
1328 nb -= k;
1329 if (nb == 0)
1330 goto Succeed;
1331 }
1332 *dest++ = *pa++;
1333 --na;
1334 if (na == 1)
1335 goto CopyB;
1336 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001337 ++min_gallop; /* penalize it for leaving galloping mode */
1338 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001339 }
1340Succeed:
1341 result = 0;
1342Fail:
1343 if (na)
1344 memcpy(dest, pa, na * sizeof(PyObject*));
1345 return result;
1346CopyB:
1347 assert(na == 1 && nb > 0);
1348 /* The last element of pa belongs at the end of the merge. */
1349 memmove(dest, pb, nb * sizeof(PyObject *));
1350 dest[nb] = *pa;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001351 return 0;
Tim Petersa64dc242002-08-01 02:13:36 +00001352}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001353
Tim Petersa64dc242002-08-01 02:13:36 +00001354/* Merge the na elements starting at pa with the nb elements starting at pb
1355 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1356 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1357 * merge, and should have na >= nb. See listsort.txt for more info.
1358 * Return 0 if successful, -1 if error.
1359 */
1360static int
1361merge_hi(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1362{
1363 int k;
1364 PyObject *compare;
1365 PyObject **dest;
1366 int result = -1; /* guilty until proved innocent */
1367 PyObject **basea;
1368 PyObject **baseb;
Tim Peterse05f65a2002-08-10 05:21:15 +00001369 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001370
1371 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1372 if (MERGE_GETMEM(ms, nb) < 0)
1373 return -1;
1374 dest = pb + nb - 1;
1375 memcpy(ms->a, pb, nb * sizeof(PyObject*));
1376 basea = pa;
1377 baseb = ms->a;
1378 pb = ms->a + nb - 1;
1379 pa += na - 1;
1380
1381 *dest-- = *pa--;
1382 --na;
1383 if (na == 0)
1384 goto Succeed;
1385 if (nb == 1)
1386 goto CopyA;
1387
1388 compare = ms->compare;
1389 for (;;) {
1390 int acount = 0; /* # of times A won in a row */
1391 int bcount = 0; /* # of times B won in a row */
1392
1393 /* Do the straightforward thing until (if ever) one run
1394 * appears to win consistently.
1395 */
1396 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001397 assert(na > 0 && nb > 1);
Tim Peters66860f62002-08-04 17:47:26 +00001398 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001399 if (k) {
1400 if (k < 0)
1401 goto Fail;
1402 *dest-- = *pa--;
1403 ++acount;
1404 bcount = 0;
1405 --na;
1406 if (na == 0)
1407 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001408 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001409 break;
1410 }
1411 else {
1412 *dest-- = *pb--;
1413 ++bcount;
1414 acount = 0;
1415 --nb;
1416 if (nb == 1)
1417 goto CopyA;
Tim Peterse05f65a2002-08-10 05:21:15 +00001418 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001419 break;
1420 }
1421 }
1422
1423 /* One run is winning so consistently that galloping may
1424 * be a huge win. So try that, and continue galloping until
1425 * (if ever) neither run appears to be winning consistently
1426 * anymore.
1427 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001428 ++min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001429 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001430 assert(na > 0 && nb > 1);
1431 min_gallop -= min_gallop > 1;
1432 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001433 k = gallop_right(*pb, basea, na, na-1, compare);
1434 if (k < 0)
1435 goto Fail;
1436 k = na - k;
1437 acount = k;
1438 if (k) {
1439 dest -= k;
1440 pa -= k;
1441 memmove(dest+1, pa+1, k * sizeof(PyObject *));
1442 na -= k;
1443 if (na == 0)
1444 goto Succeed;
1445 }
1446 *dest-- = *pb--;
1447 --nb;
1448 if (nb == 1)
1449 goto CopyA;
1450
1451 k = gallop_left(*pa, baseb, nb, nb-1, compare);
1452 if (k < 0)
1453 goto Fail;
1454 k = nb - k;
1455 bcount = k;
1456 if (k) {
1457 dest -= k;
1458 pb -= k;
1459 memcpy(dest+1, pb+1, k * sizeof(PyObject *));
1460 nb -= k;
1461 if (nb == 1)
1462 goto CopyA;
1463 /* nb==0 is impossible now if the comparison
1464 * function is consistent, but we can't assume
1465 * that it is.
1466 */
1467 if (nb == 0)
1468 goto Succeed;
1469 }
1470 *dest-- = *pa--;
1471 --na;
1472 if (na == 0)
1473 goto Succeed;
1474 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001475 ++min_gallop; /* penalize it for leaving galloping mode */
1476 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001477 }
1478Succeed:
1479 result = 0;
1480Fail:
1481 if (nb)
1482 memcpy(dest-(nb-1), baseb, nb * sizeof(PyObject*));
1483 return result;
1484CopyA:
1485 assert(nb == 1 && na > 0);
1486 /* The first element of pb belongs at the front of the merge. */
1487 dest -= na;
1488 pa -= na;
1489 memmove(dest+1, pa+1, na * sizeof(PyObject *));
1490 *dest = *pb;
1491 return 0;
1492}
1493
1494/* Merge the two runs at stack indices i and i+1.
1495 * Returns 0 on success, -1 on error.
1496 */
1497static int
1498merge_at(MergeState *ms, int i)
1499{
1500 PyObject **pa, **pb;
1501 int na, nb;
1502 int k;
1503 PyObject *compare;
1504
1505 assert(ms != NULL);
1506 assert(ms->n >= 2);
1507 assert(i >= 0);
1508 assert(i == ms->n - 2 || i == ms->n - 3);
1509
Tim Peterse05f65a2002-08-10 05:21:15 +00001510 pa = ms->pending[i].base;
1511 na = ms->pending[i].len;
1512 pb = ms->pending[i+1].base;
1513 nb = ms->pending[i+1].len;
Tim Petersa64dc242002-08-01 02:13:36 +00001514 assert(na > 0 && nb > 0);
1515 assert(pa + na == pb);
1516
1517 /* Record the length of the combined runs; if i is the 3rd-last
1518 * run now, also slide over the last run (which isn't involved
1519 * in this merge). The current run i+1 goes away in any case.
1520 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001521 ms->pending[i].len = na + nb;
1522 if (i == ms->n - 3)
1523 ms->pending[i+1] = ms->pending[i+2];
Tim Petersa64dc242002-08-01 02:13:36 +00001524 --ms->n;
1525
1526 /* Where does b start in a? Elements in a before that can be
1527 * ignored (already in place).
1528 */
1529 compare = ms->compare;
1530 k = gallop_right(*pb, pa, na, 0, compare);
1531 if (k < 0)
1532 return -1;
1533 pa += k;
1534 na -= k;
1535 if (na == 0)
1536 return 0;
1537
1538 /* Where does a end in b? Elements in b after that can be
1539 * ignored (already in place).
1540 */
1541 nb = gallop_left(pa[na-1], pb, nb, nb-1, compare);
1542 if (nb <= 0)
1543 return nb;
1544
1545 /* Merge what remains of the runs, using a temp array with
1546 * min(na, nb) elements.
1547 */
1548 if (na <= nb)
1549 return merge_lo(ms, pa, na, pb, nb);
1550 else
1551 return merge_hi(ms, pa, na, pb, nb);
1552}
1553
1554/* Examine the stack of runs waiting to be merged, merging adjacent runs
1555 * until the stack invariants are re-established:
1556 *
1557 * 1. len[-3] > len[-2] + len[-1]
1558 * 2. len[-2] > len[-1]
1559 *
1560 * See listsort.txt for more info.
1561 *
1562 * Returns 0 on success, -1 on error.
1563 */
1564static int
1565merge_collapse(MergeState *ms)
1566{
Tim Peterse05f65a2002-08-10 05:21:15 +00001567 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001568
1569 assert(ms);
1570 while (ms->n > 1) {
1571 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001572 if (n > 0 && p[n-1].len <= p[n].len + p[n+1].len) {
1573 if (p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001574 --n;
1575 if (merge_at(ms, n) < 0)
1576 return -1;
1577 }
Tim Peterse05f65a2002-08-10 05:21:15 +00001578 else if (p[n].len <= p[n+1].len) {
Tim Petersa64dc242002-08-01 02:13:36 +00001579 if (merge_at(ms, n) < 0)
1580 return -1;
1581 }
1582 else
1583 break;
1584 }
1585 return 0;
1586}
1587
1588/* Regardless of invariants, merge all runs on the stack until only one
1589 * remains. This is used at the end of the mergesort.
1590 *
1591 * Returns 0 on success, -1 on error.
1592 */
1593static int
1594merge_force_collapse(MergeState *ms)
1595{
Tim Peterse05f65a2002-08-10 05:21:15 +00001596 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001597
1598 assert(ms);
1599 while (ms->n > 1) {
1600 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001601 if (n > 0 && p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001602 --n;
1603 if (merge_at(ms, n) < 0)
1604 return -1;
1605 }
1606 return 0;
1607}
1608
1609/* Compute a good value for the minimum run length; natural runs shorter
1610 * than this are boosted artificially via binary insertion.
1611 *
1612 * If n < 64, return n (it's too small to bother with fancy stuff).
1613 * Else if n is an exact power of 2, return 32.
1614 * Else return an int k, 32 <= k <= 64, such that n/k is close to, but
1615 * strictly less than, an exact power of 2.
1616 *
1617 * See listsort.txt for more info.
1618 */
1619static int
1620merge_compute_minrun(int n)
1621{
1622 int r = 0; /* becomes 1 if any 1 bits are shifted off */
1623
1624 assert(n >= 0);
1625 while (n >= 64) {
1626 r |= n & 1;
1627 n >>= 1;
1628 }
1629 return n + r;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001630}
Guido van Rossuma119c0d1998-05-29 17:56:32 +00001631
Jeremy Hylton938ace62002-07-17 16:30:39 +00001632static PyTypeObject immutable_list_type;
Guido van Rossum3f236de1996-12-10 23:55:39 +00001633
Tim Petersa64dc242002-08-01 02:13:36 +00001634/* An adaptive, stable, natural mergesort. See listsort.txt.
1635 * Returns Py_None on success, NULL on error. Even in case of error, the
1636 * list will be some permutation of its input state (nothing is lost or
1637 * duplicated).
1638 */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001639static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001640listsort(PyListObject *self, PyObject *args)
Guido van Rossum3f236de1996-12-10 23:55:39 +00001641{
Tim Petersa64dc242002-08-01 02:13:36 +00001642 MergeState ms;
1643 PyObject **lo, **hi;
1644 int nremaining;
1645 int minrun;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001646 PyTypeObject *savetype;
Tim Petersa64dc242002-08-01 02:13:36 +00001647 PyObject *compare = NULL;
1648 PyObject *result = NULL; /* guilty until proved innocent */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001649
Tim Petersa64dc242002-08-01 02:13:36 +00001650 assert(self != NULL);
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001651 if (args != NULL) {
Tim Peters6bdbc9e2002-08-03 02:28:24 +00001652 if (!PyArg_ParseTuple(args, "|O:sort", &compare))
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001653 return NULL;
1654 }
Tim Petersa64dc242002-08-01 02:13:36 +00001655 merge_init(&ms, compare);
Tim Peters330f9e92002-07-19 07:05:44 +00001656
Tim Peters6d6c1a32001-08-02 04:15:00 +00001657 savetype = self->ob_type;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001658 self->ob_type = &immutable_list_type;
Tim Peters330f9e92002-07-19 07:05:44 +00001659
Tim Petersa64dc242002-08-01 02:13:36 +00001660 nremaining = self->ob_size;
1661 if (nremaining < 2)
1662 goto succeed;
Tim Peters330f9e92002-07-19 07:05:44 +00001663
Tim Petersa64dc242002-08-01 02:13:36 +00001664 /* March over the array once, left to right, finding natural runs,
1665 * and extending short natural runs to minrun elements.
1666 */
1667 lo = self->ob_item;
1668 hi = lo + nremaining;
1669 minrun = merge_compute_minrun(nremaining);
1670 do {
1671 int descending;
1672 int n;
Tim Peters330f9e92002-07-19 07:05:44 +00001673
Tim Petersa64dc242002-08-01 02:13:36 +00001674 /* Identify next run. */
1675 n = count_run(lo, hi, compare, &descending);
1676 if (n < 0)
1677 goto fail;
1678 if (descending)
1679 reverse_slice(lo, lo + n);
1680 /* If short, extend to min(minrun, nremaining). */
1681 if (n < minrun) {
1682 const int force = nremaining <= minrun ?
1683 nremaining : minrun;
1684 if (binarysort(lo, lo + force, lo + n, compare) < 0)
1685 goto fail;
1686 n = force;
1687 }
1688 /* Push run onto pending-runs stack, and maybe merge. */
1689 assert(ms.n < MAX_MERGE_PENDING);
Tim Peterse05f65a2002-08-10 05:21:15 +00001690 ms.pending[ms.n].base = lo;
1691 ms.pending[ms.n].len = n;
Tim Petersa64dc242002-08-01 02:13:36 +00001692 ++ms.n;
1693 if (merge_collapse(&ms) < 0)
1694 goto fail;
1695 /* Advance to find next run. */
1696 lo += n;
1697 nremaining -= n;
1698 } while (nremaining);
1699 assert(lo == hi);
Tim Peters330f9e92002-07-19 07:05:44 +00001700
Tim Petersa64dc242002-08-01 02:13:36 +00001701 if (merge_force_collapse(&ms) < 0)
1702 goto fail;
1703 assert(ms.n == 1);
Tim Peterse05f65a2002-08-10 05:21:15 +00001704 assert(ms.pending[0].base == self->ob_item);
1705 assert(ms.pending[0].len == self->ob_size);
Tim Petersa64dc242002-08-01 02:13:36 +00001706
1707succeed:
1708 result = Py_None;
Tim Peters330f9e92002-07-19 07:05:44 +00001709fail:
1710 self->ob_type = savetype;
Tim Petersa64dc242002-08-01 02:13:36 +00001711 merge_freemem(&ms);
1712 Py_XINCREF(result);
1713 return result;
Guido van Rossum3f236de1996-12-10 23:55:39 +00001714}
Tim Peters330f9e92002-07-19 07:05:44 +00001715#undef IFLT
Tim Peters66860f62002-08-04 17:47:26 +00001716#undef ISLT
Tim Peters330f9e92002-07-19 07:05:44 +00001717
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001718int
Fred Drakea2f55112000-07-09 15:16:51 +00001719PyList_Sort(PyObject *v)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001720{
1721 if (v == NULL || !PyList_Check(v)) {
1722 PyErr_BadInternalCall();
1723 return -1;
1724 }
1725 v = listsort((PyListObject *)v, (PyObject *)NULL);
1726 if (v == NULL)
1727 return -1;
1728 Py_DECREF(v);
1729 return 0;
1730}
1731
Guido van Rossumb86c5492001-02-12 22:06:02 +00001732static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001733listreverse(PyListObject *self)
Guido van Rossumb86c5492001-02-12 22:06:02 +00001734{
Tim Peters326b4482002-07-19 04:04:16 +00001735 if (self->ob_size > 1)
1736 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001737 Py_INCREF(Py_None);
1738 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001739}
1740
Guido van Rossum84c76f51990-10-30 13:32:20 +00001741int
Fred Drakea2f55112000-07-09 15:16:51 +00001742PyList_Reverse(PyObject *v)
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001743{
Tim Peters6063e262002-08-08 01:06:39 +00001744 PyListObject *self = (PyListObject *)v;
1745
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001746 if (v == NULL || !PyList_Check(v)) {
1747 PyErr_BadInternalCall();
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001748 return -1;
1749 }
Tim Peters6063e262002-08-08 01:06:39 +00001750 if (self->ob_size > 1)
1751 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001752 return 0;
1753}
1754
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001755PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001756PyList_AsTuple(PyObject *v)
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001757{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001758 PyObject *w;
1759 PyObject **p;
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001760 int n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001761 if (v == NULL || !PyList_Check(v)) {
1762 PyErr_BadInternalCall();
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001763 return NULL;
1764 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001765 n = ((PyListObject *)v)->ob_size;
1766 w = PyTuple_New(n);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001767 if (w == NULL)
1768 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001769 p = ((PyTupleObject *)w)->ob_item;
Thomas Wouters334fb892000-07-25 12:56:38 +00001770 memcpy((void *)p,
1771 (void *)((PyListObject *)v)->ob_item,
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001772 n*sizeof(PyObject *));
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001773 while (--n >= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001774 Py_INCREF(*p);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001775 p++;
1776 }
1777 return w;
1778}
1779
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001780static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001781listindex(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001782{
1783 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001784
Guido van Rossumed98d481991-03-06 13:07:53 +00001785 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001786 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1787 if (cmp > 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001788 return PyInt_FromLong((long)i);
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001789 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001790 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001791 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001792 PyErr_SetString(PyExc_ValueError, "list.index(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001793 return NULL;
1794}
1795
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001796static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001797listcount(PyListObject *self, PyObject *v)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001798{
1799 int count = 0;
1800 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001801
Guido van Rossume6f7d181991-10-20 20:20:40 +00001802 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001803 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1804 if (cmp > 0)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001805 count++;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001806 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001807 return NULL;
Guido van Rossume6f7d181991-10-20 20:20:40 +00001808 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001809 return PyInt_FromLong((long)count);
Guido van Rossume6f7d181991-10-20 20:20:40 +00001810}
1811
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001812static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001813listremove(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001814{
1815 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001816
Guido van Rossumed98d481991-03-06 13:07:53 +00001817 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001818 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1819 if (cmp > 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001820 if (list_ass_slice(self, i, i+1,
1821 (PyObject *)NULL) != 0)
Guido van Rossumed98d481991-03-06 13:07:53 +00001822 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001823 Py_INCREF(Py_None);
1824 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001825 }
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001826 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001827 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001828 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001829 PyErr_SetString(PyExc_ValueError, "list.remove(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001830 return NULL;
1831}
1832
Jeremy Hylton8caad492000-06-23 14:18:11 +00001833static int
1834list_traverse(PyListObject *o, visitproc visit, void *arg)
1835{
1836 int i, err;
1837 PyObject *x;
1838
1839 for (i = o->ob_size; --i >= 0; ) {
1840 x = o->ob_item[i];
1841 if (x != NULL) {
1842 err = visit(x, arg);
1843 if (err)
1844 return err;
1845 }
1846 }
1847 return 0;
1848}
1849
1850static int
1851list_clear(PyListObject *lp)
1852{
1853 (void) PyList_SetSlice((PyObject *)lp, 0, lp->ob_size, 0);
1854 return 0;
1855}
1856
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001857static PyObject *
1858list_richcompare(PyObject *v, PyObject *w, int op)
1859{
1860 PyListObject *vl, *wl;
1861 int i;
1862
1863 if (!PyList_Check(v) || !PyList_Check(w)) {
1864 Py_INCREF(Py_NotImplemented);
1865 return Py_NotImplemented;
1866 }
1867
1868 vl = (PyListObject *)v;
1869 wl = (PyListObject *)w;
1870
1871 if (vl->ob_size != wl->ob_size && (op == Py_EQ || op == Py_NE)) {
1872 /* Shortcut: if the lengths differ, the lists differ */
1873 PyObject *res;
1874 if (op == Py_EQ)
1875 res = Py_False;
1876 else
1877 res = Py_True;
1878 Py_INCREF(res);
1879 return res;
1880 }
1881
1882 /* Search for the first index where items are different */
1883 for (i = 0; i < vl->ob_size && i < wl->ob_size; i++) {
1884 int k = PyObject_RichCompareBool(vl->ob_item[i],
1885 wl->ob_item[i], Py_EQ);
1886 if (k < 0)
1887 return NULL;
1888 if (!k)
1889 break;
1890 }
1891
1892 if (i >= vl->ob_size || i >= wl->ob_size) {
1893 /* No more items to compare -- compare sizes */
1894 int vs = vl->ob_size;
1895 int ws = wl->ob_size;
1896 int cmp;
1897 PyObject *res;
1898 switch (op) {
1899 case Py_LT: cmp = vs < ws; break;
Tim Peters6ee42342001-07-06 17:45:43 +00001900 case Py_LE: cmp = vs <= ws; break;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001901 case Py_EQ: cmp = vs == ws; break;
1902 case Py_NE: cmp = vs != ws; break;
1903 case Py_GT: cmp = vs > ws; break;
1904 case Py_GE: cmp = vs >= ws; break;
1905 default: return NULL; /* cannot happen */
1906 }
1907 if (cmp)
1908 res = Py_True;
1909 else
1910 res = Py_False;
1911 Py_INCREF(res);
1912 return res;
1913 }
1914
1915 /* We have an item that differs -- shortcuts for EQ/NE */
1916 if (op == Py_EQ) {
1917 Py_INCREF(Py_False);
1918 return Py_False;
1919 }
1920 if (op == Py_NE) {
1921 Py_INCREF(Py_True);
1922 return Py_True;
1923 }
1924
1925 /* Compare the final item again using the proper operator */
1926 return PyObject_RichCompare(vl->ob_item[i], wl->ob_item[i], op);
1927}
1928
Tim Peters6d6c1a32001-08-02 04:15:00 +00001929/* Adapted from newer code by Tim */
1930static int
1931list_fill(PyListObject *result, PyObject *v)
1932{
1933 PyObject *it; /* iter(v) */
1934 int n; /* guess for result list size */
1935 int i;
1936
1937 n = result->ob_size;
1938
1939 /* Special-case list(a_list), for speed. */
1940 if (PyList_Check(v)) {
1941 if (v == (PyObject *)result)
1942 return 0; /* source is destination, we're done */
1943 return list_ass_slice(result, 0, n, v);
1944 }
1945
1946 /* Empty previous contents */
1947 if (n != 0) {
1948 if (list_ass_slice(result, 0, n, (PyObject *)NULL) != 0)
1949 return -1;
1950 }
1951
1952 /* Get iterator. There may be some low-level efficiency to be gained
1953 * by caching the tp_iternext slot instead of using PyIter_Next()
1954 * later, but premature optimization is the root etc.
1955 */
1956 it = PyObject_GetIter(v);
1957 if (it == NULL)
1958 return -1;
1959
1960 /* Guess a result list size. */
1961 n = -1; /* unknown */
1962 if (PySequence_Check(v) &&
1963 v->ob_type->tp_as_sequence->sq_length) {
1964 n = PySequence_Size(v);
1965 if (n < 0)
1966 PyErr_Clear();
1967 }
1968 if (n < 0)
1969 n = 8; /* arbitrary */
1970 NRESIZE(result->ob_item, PyObject*, n);
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001971 if (result->ob_item == NULL) {
1972 PyErr_NoMemory();
Tim Peters6d6c1a32001-08-02 04:15:00 +00001973 goto error;
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001974 }
Neal Norwitz35fc7602002-06-13 21:11:11 +00001975 memset(result->ob_item, 0, sizeof(*result->ob_item) * n);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001976 result->ob_size = n;
1977
1978 /* Run iterator to exhaustion. */
1979 for (i = 0; ; i++) {
1980 PyObject *item = PyIter_Next(it);
1981 if (item == NULL) {
1982 if (PyErr_Occurred())
1983 goto error;
1984 break;
1985 }
1986 if (i < n)
1987 PyList_SET_ITEM(result, i, item); /* steals ref */
1988 else {
1989 int status = ins1(result, result->ob_size, item);
1990 Py_DECREF(item); /* append creates a new ref */
1991 if (status < 0)
1992 goto error;
1993 }
1994 }
1995
1996 /* Cut back result list if initial guess was too large. */
1997 if (i < n && result != NULL) {
1998 if (list_ass_slice(result, i, n, (PyObject *)NULL) != 0)
1999 goto error;
2000 }
2001 Py_DECREF(it);
2002 return 0;
2003
2004 error:
2005 Py_DECREF(it);
2006 return -1;
2007}
2008
2009static int
2010list_init(PyListObject *self, PyObject *args, PyObject *kw)
2011{
2012 PyObject *arg = NULL;
2013 static char *kwlist[] = {"sequence", 0};
2014
2015 if (!PyArg_ParseTupleAndKeywords(args, kw, "|O:list", kwlist, &arg))
2016 return -1;
2017 if (arg != NULL)
2018 return list_fill(self, arg);
2019 if (self->ob_size > 0)
2020 return list_ass_slice(self, 0, self->ob_size, (PyObject*)NULL);
2021 return 0;
2022}
2023
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002024static long
2025list_nohash(PyObject *self)
2026{
2027 PyErr_SetString(PyExc_TypeError, "list objects are unhashable");
2028 return -1;
2029}
2030
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002031PyDoc_STRVAR(append_doc,
2032"L.append(object) -- append object to end");
2033PyDoc_STRVAR(extend_doc,
Martin v. Löwis673c0a22002-07-28 16:35:57 +00002034"L.extend(sequence) -- extend list by appending sequence elements");
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002035PyDoc_STRVAR(insert_doc,
2036"L.insert(index, object) -- insert object before index");
2037PyDoc_STRVAR(pop_doc,
2038"L.pop([index]) -> item -- remove and return item at index (default last)");
2039PyDoc_STRVAR(remove_doc,
2040"L.remove(value) -- remove first occurrence of value");
2041PyDoc_STRVAR(index_doc,
2042"L.index(value) -> integer -- return index of first occurrence of value");
2043PyDoc_STRVAR(count_doc,
2044"L.count(value) -> integer -- return number of occurrences of value");
2045PyDoc_STRVAR(reverse_doc,
2046"L.reverse() -- reverse *IN PLACE*");
2047PyDoc_STRVAR(sort_doc,
Tim Petersa64dc242002-08-01 02:13:36 +00002048"L.sort([cmpfunc]) -- stable sort *IN PLACE*; cmpfunc(x, y) -> -1, 0, 1");
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +00002049
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002050static PyMethodDef list_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002051 {"append", (PyCFunction)listappend, METH_O, append_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002052 {"insert", (PyCFunction)listinsert, METH_VARARGS, insert_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002053 {"extend", (PyCFunction)listextend, METH_O, extend_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002054 {"pop", (PyCFunction)listpop, METH_VARARGS, pop_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002055 {"remove", (PyCFunction)listremove, METH_O, remove_doc},
2056 {"index", (PyCFunction)listindex, METH_O, index_doc},
2057 {"count", (PyCFunction)listcount, METH_O, count_doc},
2058 {"reverse", (PyCFunction)listreverse, METH_NOARGS, reverse_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002059 {"sort", (PyCFunction)listsort, METH_VARARGS, sort_doc},
Tim Petersa64dc242002-08-01 02:13:36 +00002060 {NULL, NULL} /* sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002061};
2062
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002063static PySequenceMethods list_as_sequence = {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002064 (inquiry)list_length, /* sq_length */
2065 (binaryfunc)list_concat, /* sq_concat */
2066 (intargfunc)list_repeat, /* sq_repeat */
2067 (intargfunc)list_item, /* sq_item */
2068 (intintargfunc)list_slice, /* sq_slice */
2069 (intobjargproc)list_ass_item, /* sq_ass_item */
2070 (intintobjargproc)list_ass_slice, /* sq_ass_slice */
2071 (objobjproc)list_contains, /* sq_contains */
2072 (binaryfunc)list_inplace_concat, /* sq_inplace_concat */
2073 (intargfunc)list_inplace_repeat, /* sq_inplace_repeat */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002074};
2075
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002076PyDoc_STRVAR(list_doc,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002077"list() -> new list\n"
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002078"list(sequence) -> new list initialized from sequence's items");
Tim Peters6d6c1a32001-08-02 04:15:00 +00002079
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002080static PyObject *list_iter(PyObject *seq);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002081
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002082static PyObject *
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002083list_subscript(PyListObject* self, PyObject* item)
2084{
2085 if (PyInt_Check(item)) {
2086 long i = PyInt_AS_LONG(item);
2087 if (i < 0)
2088 i += PyList_GET_SIZE(self);
2089 return list_item(self, i);
2090 }
2091 else if (PyLong_Check(item)) {
2092 long i = PyLong_AsLong(item);
2093 if (i == -1 && PyErr_Occurred())
2094 return NULL;
2095 if (i < 0)
2096 i += PyList_GET_SIZE(self);
2097 return list_item(self, i);
2098 }
2099 else if (PySlice_Check(item)) {
2100 int start, stop, step, slicelength, cur, i;
2101 PyObject* result;
2102 PyObject* it;
2103
2104 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2105 &start, &stop, &step, &slicelength) < 0) {
2106 return NULL;
2107 }
2108
2109 if (slicelength <= 0) {
2110 return PyList_New(0);
2111 }
2112 else {
2113 result = PyList_New(slicelength);
2114 if (!result) return NULL;
2115
Tim Peters3b01a122002-07-19 02:35:45 +00002116 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002117 cur += step, i++) {
2118 it = PyList_GET_ITEM(self, cur);
2119 Py_INCREF(it);
2120 PyList_SET_ITEM(result, i, it);
2121 }
Tim Peters3b01a122002-07-19 02:35:45 +00002122
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002123 return result;
2124 }
2125 }
2126 else {
2127 PyErr_SetString(PyExc_TypeError,
2128 "list indices must be integers");
2129 return NULL;
2130 }
2131}
2132
Tim Peters3b01a122002-07-19 02:35:45 +00002133static int
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002134list_ass_subscript(PyListObject* self, PyObject* item, PyObject* value)
2135{
2136 if (PyInt_Check(item)) {
2137 long i = PyInt_AS_LONG(item);
2138 if (i < 0)
2139 i += PyList_GET_SIZE(self);
2140 return list_ass_item(self, i, value);
2141 }
2142 else if (PyLong_Check(item)) {
2143 long i = PyLong_AsLong(item);
2144 if (i == -1 && PyErr_Occurred())
2145 return -1;
2146 if (i < 0)
2147 i += PyList_GET_SIZE(self);
2148 return list_ass_item(self, i, value);
2149 }
2150 else if (PySlice_Check(item)) {
2151 int start, stop, step, slicelength;
2152
2153 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2154 &start, &stop, &step, &slicelength) < 0) {
2155 return -1;
2156 }
2157
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002158 /* treat L[slice(a,b)] = v _exactly_ like L[a:b] = v */
2159 if (step == 1 && ((PySliceObject*)item)->step == Py_None)
2160 return list_ass_slice(self, start, stop, value);
2161
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002162 if (value == NULL) {
2163 /* delete slice */
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002164 PyObject **garbage, **it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002165 int cur, i, j;
Tim Peters3b01a122002-07-19 02:35:45 +00002166
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002167 if (slicelength <= 0)
2168 return 0;
2169
2170 if (step < 0) {
2171 stop = start + 1;
2172 start = stop + step*(slicelength - 1) - 1;
2173 step = -step;
2174 }
2175
2176 garbage = (PyObject**)
2177 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002178
2179 /* drawing pictures might help
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002180 understand these for loops */
Guido van Rossum75a20b12002-06-11 12:22:28 +00002181 for (cur = start, i = 0;
2182 cur < stop;
Michael W. Hudson56796f62002-07-29 14:35:04 +00002183 cur += step, i++) {
2184 int lim = step;
2185
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002186 garbage[i] = PyList_GET_ITEM(self, cur);
2187
Michael W. Hudson56796f62002-07-29 14:35:04 +00002188 if (cur + step >= self->ob_size) {
2189 lim = self->ob_size - cur - 1;
2190 }
2191
2192 for (j = 0; j < lim; j++) {
Tim Peters3b01a122002-07-19 02:35:45 +00002193 PyList_SET_ITEM(self, cur + j - i,
Guido van Rossum75a20b12002-06-11 12:22:28 +00002194 PyList_GET_ITEM(self,
2195 cur + j + 1));
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002196 }
2197 }
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002198 for (cur = start + slicelength*step + 1;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002199 cur < self->ob_size; cur++) {
2200 PyList_SET_ITEM(self, cur - slicelength,
2201 PyList_GET_ITEM(self, cur));
2202 }
2203 self->ob_size -= slicelength;
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002204 it = self->ob_item;
2205 NRESIZE(it, PyObject*, self->ob_size);
2206 self->ob_item = it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002207
2208 for (i = 0; i < slicelength; i++) {
2209 Py_DECREF(garbage[i]);
2210 }
2211 PyMem_FREE(garbage);
2212
2213 return 0;
2214 }
2215 else {
2216 /* assign slice */
2217 PyObject **garbage, *ins;
2218 int cur, i;
2219
2220 if (!PyList_Check(value)) {
2221 PyErr_Format(PyExc_TypeError,
2222 "must assign list (not \"%.200s\") to slice",
2223 value->ob_type->tp_name);
2224 return -1;
2225 }
2226
2227 if (PyList_GET_SIZE(value) != slicelength) {
2228 PyErr_Format(PyExc_ValueError,
2229 "attempt to assign list of size %d to extended slice of size %d",
2230 PyList_Size(value), slicelength);
2231 return -1;
2232 }
2233
2234 if (!slicelength)
2235 return 0;
2236
2237 /* protect against a[::-1] = a */
Tim Peters3b01a122002-07-19 02:35:45 +00002238 if (self == (PyListObject*)value) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002239 value = list_slice((PyListObject*)value, 0,
2240 PyList_GET_SIZE(value));
Tim Peters3b01a122002-07-19 02:35:45 +00002241 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002242 else {
2243 Py_INCREF(value);
2244 }
2245
2246 garbage = (PyObject**)
2247 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002248
2249 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002250 cur += step, i++) {
2251 garbage[i] = PyList_GET_ITEM(self, cur);
Tim Peters3b01a122002-07-19 02:35:45 +00002252
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002253 ins = PyList_GET_ITEM(value, i);
2254 Py_INCREF(ins);
2255 PyList_SET_ITEM(self, cur, ins);
2256 }
2257
2258 for (i = 0; i < slicelength; i++) {
2259 Py_DECREF(garbage[i]);
2260 }
Tim Peters3b01a122002-07-19 02:35:45 +00002261
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002262 PyMem_FREE(garbage);
2263 Py_DECREF(value);
Tim Peters3b01a122002-07-19 02:35:45 +00002264
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002265 return 0;
2266 }
Tim Peters3b01a122002-07-19 02:35:45 +00002267 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002268 else {
Tim Peters3b01a122002-07-19 02:35:45 +00002269 PyErr_SetString(PyExc_TypeError,
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002270 "list indices must be integers");
2271 return -1;
2272 }
2273}
2274
2275static PyMappingMethods list_as_mapping = {
2276 (inquiry)list_length,
2277 (binaryfunc)list_subscript,
2278 (objobjargproc)list_ass_subscript
2279};
2280
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002281PyTypeObject PyList_Type = {
2282 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002283 0,
2284 "list",
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002285 sizeof(PyListObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002286 0,
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002287 (destructor)list_dealloc, /* tp_dealloc */
2288 (printfunc)list_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002289 0, /* tp_getattr */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002290 0, /* tp_setattr */
2291 0, /* tp_compare */
2292 (reprfunc)list_repr, /* tp_repr */
2293 0, /* tp_as_number */
2294 &list_as_sequence, /* tp_as_sequence */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002295 &list_as_mapping, /* tp_as_mapping */
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002296 list_nohash, /* tp_hash */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002297 0, /* tp_call */
2298 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002299 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002300 0, /* tp_setattro */
2301 0, /* tp_as_buffer */
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002302 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
Tim Peters6d6c1a32001-08-02 04:15:00 +00002303 Py_TPFLAGS_BASETYPE, /* tp_flags */
2304 list_doc, /* tp_doc */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002305 (traverseproc)list_traverse, /* tp_traverse */
2306 (inquiry)list_clear, /* tp_clear */
2307 list_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002308 0, /* tp_weaklistoffset */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002309 list_iter, /* tp_iter */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002310 0, /* tp_iternext */
2311 list_methods, /* tp_methods */
2312 0, /* tp_members */
2313 0, /* tp_getset */
2314 0, /* tp_base */
2315 0, /* tp_dict */
2316 0, /* tp_descr_get */
2317 0, /* tp_descr_set */
2318 0, /* tp_dictoffset */
2319 (initproc)list_init, /* tp_init */
2320 PyType_GenericAlloc, /* tp_alloc */
2321 PyType_GenericNew, /* tp_new */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002322 PyObject_GC_Del, /* tp_free */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002323};
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002324
2325
2326/* During a sort, we really can't have anyone modifying the list; it could
2327 cause core dumps. Thus, we substitute a dummy type that raises an
2328 explanatory exception when a modifying operation is used. Caveat:
2329 comparisons may behave differently; but I guess it's a bad idea anyway to
2330 compare a list that's being sorted... */
2331
2332static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00002333immutable_list_op(void)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002334{
2335 PyErr_SetString(PyExc_TypeError,
2336 "a list cannot be modified while it is being sorted");
2337 return NULL;
2338}
2339
2340static PyMethodDef immutable_list_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002341 {"append", (PyCFunction)immutable_list_op, METH_VARARGS},
2342 {"insert", (PyCFunction)immutable_list_op, METH_VARARGS},
Tim Peters52e07172001-08-30 06:15:32 +00002343 {"extend", (PyCFunction)immutable_list_op, METH_O},
2344 {"pop", (PyCFunction)immutable_list_op, METH_VARARGS},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002345 {"remove", (PyCFunction)immutable_list_op, METH_VARARGS},
2346 {"index", (PyCFunction)listindex, METH_O},
2347 {"count", (PyCFunction)listcount, METH_O},
2348 {"reverse", (PyCFunction)immutable_list_op, METH_VARARGS},
2349 {"sort", (PyCFunction)immutable_list_op, METH_VARARGS},
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002350 {NULL, NULL} /* sentinel */
2351};
2352
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002353static int
Fred Drakea2f55112000-07-09 15:16:51 +00002354immutable_list_ass(void)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002355{
2356 immutable_list_op();
2357 return -1;
2358}
2359
2360static PySequenceMethods immutable_list_as_sequence = {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002361 (inquiry)list_length, /* sq_length */
2362 (binaryfunc)list_concat, /* sq_concat */
2363 (intargfunc)list_repeat, /* sq_repeat */
2364 (intargfunc)list_item, /* sq_item */
2365 (intintargfunc)list_slice, /* sq_slice */
2366 (intobjargproc)immutable_list_ass, /* sq_ass_item */
2367 (intintobjargproc)immutable_list_ass, /* sq_ass_slice */
2368 (objobjproc)list_contains, /* sq_contains */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002369};
2370
2371static PyTypeObject immutable_list_type = {
2372 PyObject_HEAD_INIT(&PyType_Type)
2373 0,
2374 "list (immutable, during sort)",
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002375 sizeof(PyListObject),
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002376 0,
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002377 0, /* Cannot happen */ /* tp_dealloc */
2378 (printfunc)list_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002379 0, /* tp_getattr */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002380 0, /* tp_setattr */
2381 0, /* Won't be called */ /* tp_compare */
2382 (reprfunc)list_repr, /* tp_repr */
2383 0, /* tp_as_number */
2384 &immutable_list_as_sequence, /* tp_as_sequence */
2385 0, /* tp_as_mapping */
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002386 list_nohash, /* tp_hash */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002387 0, /* tp_call */
2388 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002389 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002390 0, /* tp_setattro */
2391 0, /* tp_as_buffer */
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002392 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002393 list_doc, /* tp_doc */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002394 (traverseproc)list_traverse, /* tp_traverse */
2395 0, /* tp_clear */
2396 list_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002397 0, /* tp_weaklistoffset */
2398 0, /* tp_iter */
2399 0, /* tp_iternext */
2400 immutable_list_methods, /* tp_methods */
2401 0, /* tp_members */
2402 0, /* tp_getset */
2403 0, /* tp_base */
2404 0, /* tp_dict */
2405 0, /* tp_descr_get */
2406 0, /* tp_descr_set */
2407 0, /* tp_init */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002408 /* NOTE: This is *not* the standard list_type struct! */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002409};
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002410
2411
2412/*********************** List Iterator **************************/
2413
2414typedef struct {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002415 PyObject_HEAD
2416 long it_index;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002417 PyListObject *it_seq; /* Set to NULL when iterator is exhausted */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002418} listiterobject;
2419
2420PyTypeObject PyListIter_Type;
2421
Guido van Rossum5086e492002-07-16 15:56:52 +00002422static PyObject *
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002423list_iter(PyObject *seq)
2424{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002425 listiterobject *it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002426
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002427 if (!PyList_Check(seq)) {
2428 PyErr_BadInternalCall();
2429 return NULL;
2430 }
2431 it = PyObject_GC_New(listiterobject, &PyListIter_Type);
2432 if (it == NULL)
2433 return NULL;
2434 it->it_index = 0;
2435 Py_INCREF(seq);
2436 it->it_seq = (PyListObject *)seq;
2437 _PyObject_GC_TRACK(it);
2438 return (PyObject *)it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002439}
2440
2441static void
2442listiter_dealloc(listiterobject *it)
2443{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002444 _PyObject_GC_UNTRACK(it);
Guido van Rossum86103ae2002-07-16 20:07:32 +00002445 Py_XDECREF(it->it_seq);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002446 PyObject_GC_Del(it);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002447}
2448
2449static int
2450listiter_traverse(listiterobject *it, visitproc visit, void *arg)
2451{
Guido van Rossum86103ae2002-07-16 20:07:32 +00002452 if (it->it_seq == NULL)
2453 return 0;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002454 return visit((PyObject *)it->it_seq, arg);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002455}
2456
2457
2458static PyObject *
2459listiter_getiter(PyObject *it)
2460{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002461 Py_INCREF(it);
2462 return it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002463}
2464
2465static PyObject *
Tim Peters93b2cc42002-06-01 05:22:55 +00002466listiter_next(listiterobject *it)
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002467{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002468 PyListObject *seq;
2469 PyObject *item;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002470
Tim Peters93b2cc42002-06-01 05:22:55 +00002471 assert(it != NULL);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002472 seq = it->it_seq;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002473 if (seq == NULL)
2474 return NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002475 assert(PyList_Check(seq));
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002476
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002477 if (it->it_index < PyList_GET_SIZE(seq)) {
Tim Peters93b2cc42002-06-01 05:22:55 +00002478 item = PyList_GET_ITEM(seq, it->it_index);
2479 ++it->it_index;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002480 Py_INCREF(item);
2481 return item;
2482 }
Guido van Rossum86103ae2002-07-16 20:07:32 +00002483
2484 Py_DECREF(seq);
2485 it->it_seq = NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002486 return NULL;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002487}
2488
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002489PyTypeObject PyListIter_Type = {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002490 PyObject_HEAD_INIT(&PyType_Type)
2491 0, /* ob_size */
2492 "listiterator", /* tp_name */
2493 sizeof(listiterobject), /* tp_basicsize */
2494 0, /* tp_itemsize */
2495 /* methods */
2496 (destructor)listiter_dealloc, /* tp_dealloc */
2497 0, /* tp_print */
2498 0, /* tp_getattr */
2499 0, /* tp_setattr */
2500 0, /* tp_compare */
2501 0, /* tp_repr */
2502 0, /* tp_as_number */
2503 0, /* tp_as_sequence */
2504 0, /* tp_as_mapping */
2505 0, /* tp_hash */
2506 0, /* tp_call */
2507 0, /* tp_str */
2508 PyObject_GenericGetAttr, /* tp_getattro */
2509 0, /* tp_setattro */
2510 0, /* tp_as_buffer */
2511 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2512 0, /* tp_doc */
2513 (traverseproc)listiter_traverse, /* tp_traverse */
2514 0, /* tp_clear */
2515 0, /* tp_richcompare */
2516 0, /* tp_weaklistoffset */
2517 (getiterfunc)listiter_getiter, /* tp_iter */
2518 (iternextfunc)listiter_next, /* tp_iternext */
Guido van Rossum86103ae2002-07-16 20:07:32 +00002519 0, /* tp_methods */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002520 0, /* tp_members */
2521 0, /* tp_getset */
2522 0, /* tp_base */
2523 0, /* tp_dict */
2524 0, /* tp_descr_get */
2525 0, /* tp_descr_set */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002526};