blob: 64de38bcb326eb65866d08c020aad692b9163be5 [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 Rossumc0b618a1997-05-02 03:12:38 +0000394 np = (PyListObject *) PyList_New(size);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000395 if (np == NULL) {
Guido van Rossum90933611991-06-07 16:10:43 +0000396 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000397 }
398 for (i = 0; i < a->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000399 PyObject *v = a->ob_item[i];
400 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000401 np->ob_item[i] = v;
402 }
403 for (i = 0; i < b->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000404 PyObject *v = b->ob_item[i];
405 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000406 np->ob_item[i + a->ob_size] = v;
407 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000408 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000409#undef b
410}
411
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000412static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000413list_repeat(PyListObject *a, int n)
Guido van Rossumed98d481991-03-06 13:07:53 +0000414{
415 int i, j;
416 int size;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000417 PyListObject *np;
418 PyObject **p;
Guido van Rossumed98d481991-03-06 13:07:53 +0000419 if (n < 0)
420 n = 0;
421 size = a->ob_size * n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000422 np = (PyListObject *) PyList_New(size);
Guido van Rossumed98d481991-03-06 13:07:53 +0000423 if (np == NULL)
424 return NULL;
425 p = np->ob_item;
426 for (i = 0; i < n; i++) {
427 for (j = 0; j < a->ob_size; j++) {
428 *p = a->ob_item[j];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000429 Py_INCREF(*p);
Guido van Rossumed98d481991-03-06 13:07:53 +0000430 p++;
431 }
432 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000433 return (PyObject *) np;
Guido van Rossumed98d481991-03-06 13:07:53 +0000434}
435
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000436static int
Fred Drakea2f55112000-07-09 15:16:51 +0000437list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000438{
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000439 /* Because [X]DECREF can recursively invoke list operations on
440 this list, we must postpone all [X]DECREF activity until
441 after the list is back in its canonical shape. Therefore
442 we must allocate an additional array, 'recycle', into which
443 we temporarily copy the items that are deleted from the
444 list. :-( */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000445 PyObject **recycle, **p;
446 PyObject **item;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000447 int n; /* Size of replacement list */
448 int d; /* Change in size */
449 int k; /* Loop index */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000450#define b ((PyListObject *)v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000451 if (v == NULL)
452 n = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000453 else if (PyList_Check(v)) {
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000454 n = b->ob_size;
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000455 if (a == b) {
456 /* Special case "a[i:j] = a" -- copy b first */
457 int ret;
458 v = list_slice(b, 0, n);
459 ret = list_ass_slice(a, ilow, ihigh, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000460 Py_DECREF(v);
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000461 return ret;
462 }
463 }
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000464 else {
Fred Drakeb6a9ada2000-06-01 03:12:13 +0000465 PyErr_Format(PyExc_TypeError,
466 "must assign list (not \"%.200s\") to slice",
467 v->ob_type->tp_name);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000468 return -1;
469 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000470 if (ilow < 0)
471 ilow = 0;
472 else if (ilow > a->ob_size)
473 ilow = a->ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000474 if (ihigh < ilow)
475 ihigh = ilow;
476 else if (ihigh > a->ob_size)
477 ihigh = a->ob_size;
478 item = a->ob_item;
479 d = n - (ihigh-ilow);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000480 if (ihigh > ilow)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000481 p = recycle = PyMem_NEW(PyObject *, (ihigh-ilow));
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000482 else
483 p = recycle = NULL;
484 if (d <= 0) { /* Delete -d items; recycle ihigh-ilow items */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000485 for (k = ilow; k < ihigh; k++)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000486 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000487 if (d < 0) {
488 for (/*k = ihigh*/; k < a->ob_size; k++)
489 item[k+d] = item[k];
490 a->ob_size += d;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000491 NRESIZE(item, PyObject *, a->ob_size); /* Can't fail */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000492 a->ob_item = item;
493 }
494 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000495 else { /* Insert d items; recycle ihigh-ilow items */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000496 NRESIZE(item, PyObject *, a->ob_size + d);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000497 if (item == NULL) {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000498 if (recycle != NULL)
499 PyMem_DEL(recycle);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000500 PyErr_NoMemory();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000501 return -1;
502 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000503 for (k = a->ob_size; --k >= ihigh; )
504 item[k+d] = item[k];
505 for (/*k = ihigh-1*/; k >= ilow; --k)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000506 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000507 a->ob_item = item;
508 a->ob_size += d;
509 }
510 for (k = 0; k < n; k++, ilow++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000511 PyObject *w = b->ob_item[k];
512 Py_XINCREF(w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000513 item[ilow] = w;
514 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000515 if (recycle) {
516 while (--p >= recycle)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000517 Py_XDECREF(*p);
518 PyMem_DEL(recycle);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000519 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000520 if (a->ob_size == 0 && a->ob_item != NULL) {
521 PyMem_FREE(a->ob_item);
522 a->ob_item = NULL;
523 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000524 return 0;
525#undef b
526}
527
Guido van Rossum234f9421993-06-17 12:35:49 +0000528int
Fred Drakea2f55112000-07-09 15:16:51 +0000529PyList_SetSlice(PyObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum234f9421993-06-17 12:35:49 +0000530{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000531 if (!PyList_Check(a)) {
532 PyErr_BadInternalCall();
Guido van Rossum1fc238a1993-07-29 08:25:09 +0000533 return -1;
Guido van Rossum234f9421993-06-17 12:35:49 +0000534 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000535 return list_ass_slice((PyListObject *)a, ilow, ihigh, v);
Guido van Rossum234f9421993-06-17 12:35:49 +0000536}
537
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000538static PyObject *
539list_inplace_repeat(PyListObject *self, int n)
540{
541 PyObject **items;
542 int size, i, j;
543
544
545 size = PyList_GET_SIZE(self);
546 if (size == 0) {
547 Py_INCREF(self);
548 return (PyObject *)self;
549 }
550
551 items = self->ob_item;
552
553 if (n < 1) {
554 self->ob_item = NULL;
555 self->ob_size = 0;
556 for (i = 0; i < size; i++)
557 Py_XDECREF(items[i]);
558 PyMem_DEL(items);
559 Py_INCREF(self);
560 return (PyObject *)self;
561 }
562
563 NRESIZE(items, PyObject*, size*n);
564 if (items == NULL) {
565 PyErr_NoMemory();
566 goto finally;
567 }
568 self->ob_item = items;
569 for (i = 1; i < n; i++) { /* Start counting at 1, not 0 */
570 for (j = 0; j < size; j++) {
571 PyObject *o = PyList_GET_ITEM(self, j);
572 Py_INCREF(o);
573 PyList_SET_ITEM(self, self->ob_size++, o);
574 }
575 }
576 Py_INCREF(self);
577 return (PyObject *)self;
578 finally:
579 return NULL;
580}
581
Guido van Rossum4a450d01991-04-03 19:05:18 +0000582static int
Fred Drakea2f55112000-07-09 15:16:51 +0000583list_ass_item(PyListObject *a, int i, PyObject *v)
Guido van Rossum4a450d01991-04-03 19:05:18 +0000584{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000585 PyObject *old_value;
Guido van Rossum4a450d01991-04-03 19:05:18 +0000586 if (i < 0 || i >= a->ob_size) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000587 PyErr_SetString(PyExc_IndexError,
588 "list assignment index out of range");
Guido van Rossum4a450d01991-04-03 19:05:18 +0000589 return -1;
590 }
591 if (v == NULL)
592 return list_ass_slice(a, i, i+1, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000593 Py_INCREF(v);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000594 old_value = a->ob_item[i];
Guido van Rossum4a450d01991-04-03 19:05:18 +0000595 a->ob_item[i] = v;
Tim Peters3b01a122002-07-19 02:35:45 +0000596 Py_DECREF(old_value);
Guido van Rossum4a450d01991-04-03 19:05:18 +0000597 return 0;
598}
599
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000600static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000601ins(PyListObject *self, int where, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000602{
603 if (ins1(self, where, v) != 0)
604 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000605 Py_INCREF(Py_None);
606 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000607}
608
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000609static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000610listinsert(PyListObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000611{
612 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000613 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000614 if (!PyArg_ParseTuple(args, "iO:insert", &i, &v))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000615 return NULL;
Guido van Rossumbf80e541993-02-08 15:49:17 +0000616 return ins(self, i, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000617}
618
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000619static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000620listappend(PyListObject *self, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000621{
Guido van Rossumbf80e541993-02-08 15:49:17 +0000622 return ins(self, (int) self->ob_size, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000623}
624
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000625static int
626listextend_internal(PyListObject *self, PyObject *b)
Barry Warsawdedf6d61998-10-09 16:37:25 +0000627{
Barry Warsawdedf6d61998-10-09 16:37:25 +0000628 PyObject **items;
629 int selflen = PyList_GET_SIZE(self);
630 int blen;
631 register int i;
632
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000633 if (PyObject_Size(b) == 0) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000634 /* short circuit when b is empty */
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000635 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000636 return 0;
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000637 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000638
Barry Warsawdedf6d61998-10-09 16:37:25 +0000639 if (self == (PyListObject*)b) {
640 /* as in list_ass_slice() we must special case the
641 * situation: a.extend(a)
642 *
643 * XXX: I think this way ought to be faster than using
644 * list_slice() the way list_ass_slice() does.
645 */
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000646 Py_DECREF(b);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000647 b = PyList_New(selflen);
648 if (!b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000649 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000650 for (i = 0; i < selflen; i++) {
651 PyObject *o = PyList_GET_ITEM(self, i);
652 Py_INCREF(o);
653 PyList_SET_ITEM(b, i, o);
654 }
655 }
Barry Warsawdedf6d61998-10-09 16:37:25 +0000656
Jeremy Hylton03657cf2000-07-12 13:05:33 +0000657 blen = PyObject_Size(b);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000658
Barry Warsawdedf6d61998-10-09 16:37:25 +0000659 /* resize a using idiom */
660 items = self->ob_item;
661 NRESIZE(items, PyObject*, selflen + blen);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000662 if (items == NULL) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000663 PyErr_NoMemory();
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000664 Py_DECREF(b);
665 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000666 }
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000667
Barry Warsawdedf6d61998-10-09 16:37:25 +0000668 self->ob_item = items;
669
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000670 /* populate the end of self with b's items */
Barry Warsawdedf6d61998-10-09 16:37:25 +0000671 for (i = 0; i < blen; i++) {
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000672 PyObject *o = PySequence_Fast_GET_ITEM(b, i);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000673 Py_INCREF(o);
674 PyList_SET_ITEM(self, self->ob_size++, o);
675 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000676 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000677 return 0;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000678}
679
680
681static PyObject *
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000682list_inplace_concat(PyListObject *self, PyObject *other)
683{
Tim Peters1af03e92001-05-26 19:37:54 +0000684 other = PySequence_Fast(other, "argument to += must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000685 if (!other)
686 return NULL;
687
688 if (listextend_internal(self, other) < 0)
689 return NULL;
690
691 Py_INCREF(self);
692 return (PyObject *)self;
693}
694
695static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000696listextend(PyListObject *self, PyObject *b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000697{
698
Tim Peters1af03e92001-05-26 19:37:54 +0000699 b = PySequence_Fast(b, "list.extend() argument must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000700 if (!b)
701 return NULL;
702
703 if (listextend_internal(self, b) < 0)
704 return NULL;
705
706 Py_INCREF(Py_None);
707 return Py_None;
708}
709
710static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000711listpop(PyListObject *self, PyObject *args)
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000712{
713 int i = -1;
714 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000715 if (!PyArg_ParseTuple(args, "|i:pop", &i))
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000716 return NULL;
717 if (self->ob_size == 0) {
718 /* Special-case most common failure cause */
719 PyErr_SetString(PyExc_IndexError, "pop from empty list");
720 return NULL;
721 }
722 if (i < 0)
723 i += self->ob_size;
724 if (i < 0 || i >= self->ob_size) {
725 PyErr_SetString(PyExc_IndexError, "pop index out of range");
726 return NULL;
727 }
728 v = self->ob_item[i];
729 Py_INCREF(v);
730 if (list_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {
731 Py_DECREF(v);
732 return NULL;
733 }
734 return v;
735}
736
Tim Peters8e2e7ca2002-07-19 02:33:08 +0000737/* Reverse a slice of a list in place, from lo up to (exclusive) hi. */
738static void
739reverse_slice(PyObject **lo, PyObject **hi)
740{
741 assert(lo && hi);
742
743 --hi;
744 while (lo < hi) {
745 PyObject *t = *lo;
746 *lo = *hi;
747 *hi = t;
748 ++lo;
749 --hi;
750 }
751}
752
Tim Petersa64dc242002-08-01 02:13:36 +0000753/* Lots of code for an adaptive, stable, natural mergesort. There are many
754 * pieces to this algorithm; read listsort.txt for overviews and details.
755 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000756
Guido van Rossum3f236de1996-12-10 23:55:39 +0000757/* Comparison function. Takes care of calling a user-supplied
Tim Peters66860f62002-08-04 17:47:26 +0000758 * comparison function (any callable Python object), which must not be
759 * NULL (use the ISLT macro if you don't know, or call PyObject_RichCompareBool
760 * with Py_LT if you know it's NULL).
Tim Petersa64dc242002-08-01 02:13:36 +0000761 * Returns -1 on error, 1 if x < y, 0 if x >= y.
762 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000763static int
Tim Petersa8c974c2002-07-19 03:30:57 +0000764islt(PyObject *x, PyObject *y, PyObject *compare)
Guido van Rossum3f236de1996-12-10 23:55:39 +0000765{
Tim Petersf2a04732002-07-11 21:46:16 +0000766 PyObject *res;
767 PyObject *args;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000768 int i;
769
Tim Peters66860f62002-08-04 17:47:26 +0000770 assert(compare != NULL);
Tim Petersa8c974c2002-07-19 03:30:57 +0000771 /* Call the user's comparison function and translate the 3-way
772 * result into true or false (or error).
773 */
Tim Petersf2a04732002-07-11 21:46:16 +0000774 args = PyTuple_New(2);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000775 if (args == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000776 return -1;
Tim Petersf2a04732002-07-11 21:46:16 +0000777 Py_INCREF(x);
778 Py_INCREF(y);
779 PyTuple_SET_ITEM(args, 0, x);
780 PyTuple_SET_ITEM(args, 1, y);
Tim Peters58cf3612002-07-15 05:16:13 +0000781 res = PyObject_Call(compare, args, NULL);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000782 Py_DECREF(args);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000783 if (res == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000784 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000785 if (!PyInt_Check(res)) {
786 Py_DECREF(res);
787 PyErr_SetString(PyExc_TypeError,
Guido van Rossum9bcd1d71999-04-19 17:44:39 +0000788 "comparison function must return int");
Tim Petersa8c974c2002-07-19 03:30:57 +0000789 return -1;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000790 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000791 i = PyInt_AsLong(res);
792 Py_DECREF(res);
Tim Petersa8c974c2002-07-19 03:30:57 +0000793 return i < 0;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000794}
795
Tim Peters66860f62002-08-04 17:47:26 +0000796/* If COMPARE is NULL, calls PyObject_RichCompareBool with Py_LT, else calls
797 * islt. This avoids a layer of function call in the usual case, and
798 * sorting does many comparisons.
799 * Returns -1 on error, 1 if x < y, 0 if x >= y.
800 */
801#define ISLT(X, Y, COMPARE) ((COMPARE) == NULL ? \
802 PyObject_RichCompareBool(X, Y, Py_LT) : \
803 islt(X, Y, COMPARE))
804
805/* Compare X to Y via "<". Goto "fail" if the comparison raises an
Tim Petersa8c974c2002-07-19 03:30:57 +0000806 error. Else "k" is set to true iff X<Y, and an "if (k)" block is
807 started. It makes more sense in context <wink>. X and Y are PyObject*s.
808*/
Tim Peters66860f62002-08-04 17:47:26 +0000809#define IFLT(X, Y) if ((k = ISLT(X, Y, compare)) < 0) goto fail; \
Tim Petersa8c974c2002-07-19 03:30:57 +0000810 if (k)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000811
812/* binarysort is the best method for sorting small arrays: it does
813 few compares, but can do data movement quadratic in the number of
814 elements.
Guido van Rossum42812581998-06-17 14:15:44 +0000815 [lo, hi) is a contiguous slice of a list, and is sorted via
Tim Petersa8c974c2002-07-19 03:30:57 +0000816 binary insertion. This sort is stable.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000817 On entry, must have lo <= start <= hi, and that [lo, start) is already
818 sorted (pass start == lo if you don't know!).
Tim Petersa8c974c2002-07-19 03:30:57 +0000819 If islt() complains return -1, else 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000820 Even in case of error, the output slice will be some permutation of
821 the input (nothing is lost or duplicated).
822*/
Guido van Rossum3f236de1996-12-10 23:55:39 +0000823static int
Fred Drakea2f55112000-07-09 15:16:51 +0000824binarysort(PyObject **lo, PyObject **hi, PyObject **start, PyObject *compare)
825 /* compare -- comparison function object, or NULL for default */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000826{
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000827 register int k;
828 register PyObject **l, **p, **r;
829 register PyObject *pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000830
Tim Petersa8c974c2002-07-19 03:30:57 +0000831 assert(lo <= start && start <= hi);
832 /* assert [lo, start) is sorted */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000833 if (lo == start)
834 ++start;
835 for (; start < hi; ++start) {
836 /* set l to where *start belongs */
837 l = lo;
838 r = start;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000839 pivot = *r;
Tim Peters0fe977c2002-07-19 06:12:32 +0000840 /* Invariants:
841 * pivot >= all in [lo, l).
842 * pivot < all in [r, start).
843 * The second is vacuously true at the start.
844 */
845 assert(l < r);
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000846 do {
847 p = l + ((r - l) >> 1);
Tim Petersa8c974c2002-07-19 03:30:57 +0000848 IFLT(pivot, *p)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000849 r = p;
850 else
Tim Peters0fe977c2002-07-19 06:12:32 +0000851 l = p+1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000852 } while (l < r);
Tim Peters0fe977c2002-07-19 06:12:32 +0000853 assert(l == r);
854 /* The invariants still hold, so pivot >= all in [lo, l) and
855 pivot < all in [l, start), so pivot belongs at l. Note
856 that if there are elements equal to pivot, l points to the
857 first slot after them -- that's why this sort is stable.
858 Slide over to make room.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000859 Caution: using memmove is much slower under MSVC 5;
860 we're not usually moving many slots. */
861 for (p = start; p > l; --p)
862 *p = *(p-1);
863 *l = pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000864 }
Guido van Rossum3f236de1996-12-10 23:55:39 +0000865 return 0;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000866
867 fail:
868 return -1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000869}
870
Tim Petersa64dc242002-08-01 02:13:36 +0000871/*
872Return the length of the run beginning at lo, in the slice [lo, hi). lo < hi
873is required on entry. "A run" is the longest ascending sequence, with
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000874
Tim Petersa64dc242002-08-01 02:13:36 +0000875 lo[0] <= lo[1] <= lo[2] <= ...
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000876
Tim Petersa64dc242002-08-01 02:13:36 +0000877or the longest descending 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] > ...
Tim Peters3b01a122002-07-19 02:35:45 +0000880
Tim Petersa64dc242002-08-01 02:13:36 +0000881Boolean *descending is set to 0 in the former case, or to 1 in the latter.
882For its intended use in a stable mergesort, the strictness of the defn of
883"descending" is needed so that the caller can safely reverse a descending
884sequence without violating stability (strict > ensures there are no equal
885elements to get out of order).
886
887Returns -1 in case of error.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000888*/
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000889static int
Tim Petersa64dc242002-08-01 02:13:36 +0000890count_run(PyObject **lo, PyObject **hi, PyObject *compare, int *descending)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000891{
Tim Petersa64dc242002-08-01 02:13:36 +0000892 int k;
893 int n;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000894
Tim Petersa64dc242002-08-01 02:13:36 +0000895 assert(lo < hi);
896 *descending = 0;
897 ++lo;
898 if (lo == hi)
899 return 1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000900
Tim Petersa64dc242002-08-01 02:13:36 +0000901 n = 2;
902 IFLT(*lo, *(lo-1)) {
903 *descending = 1;
904 for (lo = lo+1; lo < hi; ++lo, ++n) {
905 IFLT(*lo, *(lo-1))
906 ;
907 else
908 break;
909 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000910 }
Tim Petersa64dc242002-08-01 02:13:36 +0000911 else {
912 for (lo = lo+1; lo < hi; ++lo, ++n) {
913 IFLT(*lo, *(lo-1))
914 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000915 }
916 }
917
Tim Petersa64dc242002-08-01 02:13:36 +0000918 return n;
919fail:
920 return -1;
921}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000922
Tim Petersa64dc242002-08-01 02:13:36 +0000923/*
924Locate the proper position of key in a sorted vector; if the vector contains
925an element equal to key, return the position immediately to the left of
926the leftmost equal element. [gallop_right() does the same except returns
927the position to the right of the rightmost equal element (if any).]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000928
Tim Petersa64dc242002-08-01 02:13:36 +0000929"a" is a sorted vector with n elements, starting at a[0]. n must be > 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000930
Tim Petersa64dc242002-08-01 02:13:36 +0000931"hint" is an index at which to begin the search, 0 <= hint < n. The closer
932hint is to the final result, the faster this runs.
933
934The return value is the int k in 0..n such that
935
936 a[k-1] < key <= a[k]
937
938pretending that *(a-1) is minus infinity and a[n] is plus infinity. IOW,
939key belongs at index k; or, IOW, the first k elements of a should precede
940key, and the last n-k should follow key.
941
942Returns -1 on error. See listsort.txt for info on the method.
943*/
944static int
945gallop_left(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
946{
947 int ofs;
948 int lastofs;
949 int k;
950
951 assert(key && a && n > 0 && hint >= 0 && hint < n);
952
953 a += hint;
954 lastofs = 0;
955 ofs = 1;
956 IFLT(*a, key) {
957 /* a[hint] < key -- gallop right, until
958 * a[hint + lastofs] < key <= a[hint + ofs]
959 */
960 const int maxofs = n - hint; /* &a[n-1] is highest */
961 while (ofs < maxofs) {
962 IFLT(a[ofs], key) {
963 lastofs = ofs;
964 ofs = (ofs << 1) + 1;
965 if (ofs <= 0) /* int overflow */
966 ofs = maxofs;
967 }
968 else /* key <= a[hint + ofs] */
969 break;
970 }
971 if (ofs > maxofs)
972 ofs = maxofs;
973 /* Translate back to offsets relative to &a[0]. */
974 lastofs += hint;
975 ofs += hint;
976 }
977 else {
978 /* key <= a[hint] -- gallop left, until
979 * a[hint - ofs] < key <= a[hint - lastofs]
980 */
981 const int maxofs = hint + 1; /* &a[0] is lowest */
982 while (ofs < maxofs) {
983 IFLT(*(a-ofs), key)
984 break;
985 /* key <= a[hint - ofs] */
986 lastofs = ofs;
987 ofs = (ofs << 1) + 1;
988 if (ofs <= 0) /* int overflow */
989 ofs = maxofs;
990 }
991 if (ofs > maxofs)
992 ofs = maxofs;
993 /* Translate back to positive offsets relative to &a[0]. */
994 k = lastofs;
995 lastofs = hint - ofs;
996 ofs = hint - k;
997 }
998 a -= hint;
999
1000 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1001 /* Now a[lastofs] < key <= a[ofs], so key belongs somewhere to the
1002 * right of lastofs but no farther right than ofs. Do a binary
1003 * search, with invariant a[lastofs-1] < key <= a[ofs].
1004 */
1005 ++lastofs;
1006 while (lastofs < ofs) {
1007 int m = lastofs + ((ofs - lastofs) >> 1);
1008
1009 IFLT(a[m], key)
1010 lastofs = m+1; /* a[m] < key */
1011 else
1012 ofs = m; /* key <= a[m] */
1013 }
1014 assert(lastofs == ofs); /* so a[ofs-1] < key <= a[ofs] */
1015 return ofs;
1016
1017fail:
1018 return -1;
1019}
1020
1021/*
1022Exactly like gallop_left(), except that if key already exists in a[0:n],
1023finds the position immediately to the right of the rightmost equal value.
1024
1025The return value is the int k in 0..n such that
1026
1027 a[k-1] <= key < a[k]
1028
1029or -1 if error.
1030
1031The code duplication is massive, but this is enough different given that
1032we're sticking to "<" comparisons that it's much harder to follow if
1033written as one routine with yet another "left or right?" flag.
1034*/
1035static int
1036gallop_right(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
1037{
1038 int ofs;
1039 int lastofs;
1040 int k;
1041
1042 assert(key && a && n > 0 && hint >= 0 && hint < n);
1043
1044 a += hint;
1045 lastofs = 0;
1046 ofs = 1;
1047 IFLT(key, *a) {
1048 /* key < a[hint] -- gallop left, until
1049 * a[hint - ofs] <= key < a[hint - lastofs]
1050 */
1051 const int maxofs = hint + 1; /* &a[0] is lowest */
1052 while (ofs < maxofs) {
1053 IFLT(key, *(a-ofs)) {
1054 lastofs = ofs;
1055 ofs = (ofs << 1) + 1;
1056 if (ofs <= 0) /* int overflow */
1057 ofs = maxofs;
1058 }
1059 else /* a[hint - ofs] <= key */
1060 break;
1061 }
1062 if (ofs > maxofs)
1063 ofs = maxofs;
1064 /* Translate back to positive offsets relative to &a[0]. */
1065 k = lastofs;
1066 lastofs = hint - ofs;
1067 ofs = hint - k;
1068 }
1069 else {
1070 /* a[hint] <= key -- gallop right, until
1071 * a[hint + lastofs] <= key < a[hint + ofs]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001072 */
Tim Petersa64dc242002-08-01 02:13:36 +00001073 const int maxofs = n - hint; /* &a[n-1] is highest */
1074 while (ofs < maxofs) {
1075 IFLT(key, a[ofs])
1076 break;
1077 /* a[hint + ofs] <= key */
1078 lastofs = ofs;
1079 ofs = (ofs << 1) + 1;
1080 if (ofs <= 0) /* int overflow */
1081 ofs = maxofs;
1082 }
1083 if (ofs > maxofs)
1084 ofs = maxofs;
1085 /* Translate back to offsets relative to &a[0]. */
1086 lastofs += hint;
1087 ofs += hint;
1088 }
1089 a -= hint;
1090
1091 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1092 /* Now a[lastofs] <= key < a[ofs], so key belongs somewhere to the
1093 * right of lastofs but no farther right than ofs. Do a binary
1094 * search, with invariant a[lastofs-1] <= key < a[ofs].
1095 */
1096 ++lastofs;
1097 while (lastofs < ofs) {
1098 int m = lastofs + ((ofs - lastofs) >> 1);
1099
1100 IFLT(key, a[m])
1101 ofs = m; /* key < a[m] */
1102 else
1103 lastofs = m+1; /* a[m] <= key */
1104 }
1105 assert(lastofs == ofs); /* so a[ofs-1] <= key < a[ofs] */
1106 return ofs;
1107
1108fail:
1109 return -1;
1110}
1111
1112/* The maximum number of entries in a MergeState's pending-runs stack.
1113 * This is enough to sort arrays of size up to about
1114 * 32 * phi ** MAX_MERGE_PENDING
1115 * where phi ~= 1.618. 85 is ridiculouslylarge enough, good for an array
1116 * with 2**64 elements.
1117 */
1118#define MAX_MERGE_PENDING 85
1119
Tim Peterse05f65a2002-08-10 05:21:15 +00001120/* When we get into galloping mode, we stay there until both runs win less
1121 * often than MIN_GALLOP consecutive times. See listsort.txt for more info.
Tim Petersa64dc242002-08-01 02:13:36 +00001122 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001123#define MIN_GALLOP 7
Tim Petersa64dc242002-08-01 02:13:36 +00001124
1125/* Avoid malloc for small temp arrays. */
1126#define MERGESTATE_TEMP_SIZE 256
1127
1128/* One MergeState exists on the stack per invocation of mergesort. It's just
1129 * a convenient way to pass state around among the helper functions.
1130 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001131struct s_slice {
1132 PyObject **base;
1133 int len;
1134};
1135
Tim Petersa64dc242002-08-01 02:13:36 +00001136typedef struct s_MergeState {
1137 /* The user-supplied comparison function. or NULL if none given. */
1138 PyObject *compare;
1139
Tim Peterse05f65a2002-08-10 05:21:15 +00001140 /* This controls when we get *into* galloping mode. It's initialized
1141 * to MIN_GALLOP. merge_lo and merge_hi tend to nudge it higher for
1142 * random data, and lower for highly structured data.
1143 */
1144 int min_gallop;
1145
Tim Petersa64dc242002-08-01 02:13:36 +00001146 /* 'a' is temp storage to help with merges. It contains room for
1147 * alloced entries.
1148 */
1149 PyObject **a; /* may point to temparray below */
1150 int alloced;
1151
1152 /* A stack of n pending runs yet to be merged. Run #i starts at
1153 * address base[i] and extends for len[i] elements. It's always
1154 * true (so long as the indices are in bounds) that
1155 *
Tim Peterse05f65a2002-08-10 05:21:15 +00001156 * pending[i].base + pending[i].len == pending[i+1].base
Tim Petersa64dc242002-08-01 02:13:36 +00001157 *
1158 * so we could cut the storage for this, but it's a minor amount,
1159 * and keeping all the info explicit simplifies the code.
1160 */
1161 int n;
Tim Peterse05f65a2002-08-10 05:21:15 +00001162 struct s_slice pending[MAX_MERGE_PENDING];
Tim Petersa64dc242002-08-01 02:13:36 +00001163
1164 /* 'a' points to this when possible, rather than muck with malloc. */
1165 PyObject *temparray[MERGESTATE_TEMP_SIZE];
1166} MergeState;
1167
1168/* Conceptually a MergeState's constructor. */
1169static void
1170merge_init(MergeState *ms, PyObject *compare)
1171{
1172 assert(ms != NULL);
1173 ms->compare = compare;
1174 ms->a = ms->temparray;
1175 ms->alloced = MERGESTATE_TEMP_SIZE;
1176 ms->n = 0;
Tim Peterse05f65a2002-08-10 05:21:15 +00001177 ms->min_gallop = MIN_GALLOP;
Tim Petersa64dc242002-08-01 02:13:36 +00001178}
1179
1180/* Free all the temp memory owned by the MergeState. This must be called
1181 * when you're done with a MergeState, and may be called before then if
1182 * you want to free the temp memory early.
1183 */
1184static void
1185merge_freemem(MergeState *ms)
1186{
1187 assert(ms != NULL);
1188 if (ms->a != ms->temparray)
1189 PyMem_Free(ms->a);
1190 ms->a = ms->temparray;
1191 ms->alloced = MERGESTATE_TEMP_SIZE;
1192}
1193
1194/* Ensure enough temp memory for 'need' array slots is available.
1195 * Returns 0 on success and -1 if the memory can't be gotten.
1196 */
1197static int
1198merge_getmem(MergeState *ms, int need)
1199{
1200 assert(ms != NULL);
1201 if (need <= ms->alloced)
1202 return 0;
1203 /* Don't realloc! That can cost cycles to copy the old data, but
1204 * we don't care what's in the block.
1205 */
1206 merge_freemem(ms);
1207 ms->a = (PyObject **)PyMem_Malloc(need * sizeof(PyObject*));
1208 if (ms->a) {
1209 ms->alloced = need;
1210 return 0;
1211 }
1212 PyErr_NoMemory();
1213 merge_freemem(ms); /* reset to sane state */
1214 return -1;
1215}
1216#define MERGE_GETMEM(MS, NEED) ((NEED) <= (MS)->alloced ? 0 : \
1217 merge_getmem(MS, NEED))
1218
1219/* Merge the na elements starting at pa with the nb elements starting at pb
1220 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1221 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1222 * merge, and should have na <= nb. See listsort.txt for more info.
1223 * Return 0 if successful, -1 if error.
1224 */
1225static int
1226merge_lo(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1227{
1228 int k;
1229 PyObject *compare;
1230 PyObject **dest;
1231 int result = -1; /* guilty until proved innocent */
Tim Peterse05f65a2002-08-10 05:21:15 +00001232 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001233
1234 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1235 if (MERGE_GETMEM(ms, na) < 0)
1236 return -1;
1237 memcpy(ms->a, pa, na * sizeof(PyObject*));
1238 dest = pa;
1239 pa = ms->a;
1240
1241 *dest++ = *pb++;
1242 --nb;
1243 if (nb == 0)
1244 goto Succeed;
1245 if (na == 1)
1246 goto CopyB;
1247
1248 compare = ms->compare;
1249 for (;;) {
1250 int acount = 0; /* # of times A won in a row */
1251 int bcount = 0; /* # of times B won in a row */
1252
1253 /* Do the straightforward thing until (if ever) one run
1254 * appears to win consistently.
1255 */
1256 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001257 assert(na > 1 && nb > 0);
Tim Peters66860f62002-08-04 17:47:26 +00001258 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001259 if (k) {
1260 if (k < 0)
1261 goto Fail;
1262 *dest++ = *pb++;
1263 ++bcount;
1264 acount = 0;
1265 --nb;
1266 if (nb == 0)
1267 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001268 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001269 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001270 }
1271 else {
Tim Petersa64dc242002-08-01 02:13:36 +00001272 *dest++ = *pa++;
1273 ++acount;
1274 bcount = 0;
1275 --na;
1276 if (na == 1)
1277 goto CopyB;
Tim Peterse05f65a2002-08-10 05:21:15 +00001278 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001279 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001280 }
Tim Petersa64dc242002-08-01 02:13:36 +00001281 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001282
Tim Petersa64dc242002-08-01 02:13:36 +00001283 /* One run is winning so consistently that galloping may
1284 * be a huge win. So try that, and continue galloping until
1285 * (if ever) neither run appears to be winning consistently
1286 * anymore.
1287 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001288 ++min_gallop;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001289 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001290 assert(na > 1 && nb > 0);
1291 min_gallop -= min_gallop > 1;
1292 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001293 k = gallop_right(*pb, pa, na, 0, compare);
1294 acount = k;
1295 if (k) {
1296 if (k < 0)
1297 goto Fail;
1298 memcpy(dest, pa, k * sizeof(PyObject *));
1299 dest += k;
1300 pa += k;
1301 na -= k;
1302 if (na == 1)
1303 goto CopyB;
1304 /* na==0 is impossible now if the comparison
1305 * function is consistent, but we can't assume
1306 * that it is.
1307 */
1308 if (na == 0)
1309 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001310 }
Tim Petersa64dc242002-08-01 02:13:36 +00001311 *dest++ = *pb++;
1312 --nb;
1313 if (nb == 0)
1314 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001315
Tim Petersa64dc242002-08-01 02:13:36 +00001316 k = gallop_left(*pa, pb, nb, 0, compare);
1317 bcount = k;
1318 if (k) {
1319 if (k < 0)
1320 goto Fail;
1321 memmove(dest, pb, k * sizeof(PyObject *));
1322 dest += k;
1323 pb += k;
1324 nb -= k;
1325 if (nb == 0)
1326 goto Succeed;
1327 }
1328 *dest++ = *pa++;
1329 --na;
1330 if (na == 1)
1331 goto CopyB;
1332 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001333 ++min_gallop; /* penalize it for leaving galloping mode */
1334 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001335 }
1336Succeed:
1337 result = 0;
1338Fail:
1339 if (na)
1340 memcpy(dest, pa, na * sizeof(PyObject*));
1341 return result;
1342CopyB:
1343 assert(na == 1 && nb > 0);
1344 /* The last element of pa belongs at the end of the merge. */
1345 memmove(dest, pb, nb * sizeof(PyObject *));
1346 dest[nb] = *pa;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001347 return 0;
Tim Petersa64dc242002-08-01 02:13:36 +00001348}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001349
Tim Petersa64dc242002-08-01 02:13:36 +00001350/* Merge the na elements starting at pa with the nb elements starting at pb
1351 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1352 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1353 * merge, and should have na >= nb. See listsort.txt for more info.
1354 * Return 0 if successful, -1 if error.
1355 */
1356static int
1357merge_hi(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1358{
1359 int k;
1360 PyObject *compare;
1361 PyObject **dest;
1362 int result = -1; /* guilty until proved innocent */
1363 PyObject **basea;
1364 PyObject **baseb;
Tim Peterse05f65a2002-08-10 05:21:15 +00001365 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001366
1367 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1368 if (MERGE_GETMEM(ms, nb) < 0)
1369 return -1;
1370 dest = pb + nb - 1;
1371 memcpy(ms->a, pb, nb * sizeof(PyObject*));
1372 basea = pa;
1373 baseb = ms->a;
1374 pb = ms->a + nb - 1;
1375 pa += na - 1;
1376
1377 *dest-- = *pa--;
1378 --na;
1379 if (na == 0)
1380 goto Succeed;
1381 if (nb == 1)
1382 goto CopyA;
1383
1384 compare = ms->compare;
1385 for (;;) {
1386 int acount = 0; /* # of times A won in a row */
1387 int bcount = 0; /* # of times B won in a row */
1388
1389 /* Do the straightforward thing until (if ever) one run
1390 * appears to win consistently.
1391 */
1392 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001393 assert(na > 0 && nb > 1);
Tim Peters66860f62002-08-04 17:47:26 +00001394 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001395 if (k) {
1396 if (k < 0)
1397 goto Fail;
1398 *dest-- = *pa--;
1399 ++acount;
1400 bcount = 0;
1401 --na;
1402 if (na == 0)
1403 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001404 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001405 break;
1406 }
1407 else {
1408 *dest-- = *pb--;
1409 ++bcount;
1410 acount = 0;
1411 --nb;
1412 if (nb == 1)
1413 goto CopyA;
Tim Peterse05f65a2002-08-10 05:21:15 +00001414 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001415 break;
1416 }
1417 }
1418
1419 /* One run is winning so consistently that galloping may
1420 * be a huge win. So try that, and continue galloping until
1421 * (if ever) neither run appears to be winning consistently
1422 * anymore.
1423 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001424 ++min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001425 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001426 assert(na > 0 && nb > 1);
1427 min_gallop -= min_gallop > 1;
1428 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001429 k = gallop_right(*pb, basea, na, na-1, compare);
1430 if (k < 0)
1431 goto Fail;
1432 k = na - k;
1433 acount = k;
1434 if (k) {
1435 dest -= k;
1436 pa -= k;
1437 memmove(dest+1, pa+1, k * sizeof(PyObject *));
1438 na -= k;
1439 if (na == 0)
1440 goto Succeed;
1441 }
1442 *dest-- = *pb--;
1443 --nb;
1444 if (nb == 1)
1445 goto CopyA;
1446
1447 k = gallop_left(*pa, baseb, nb, nb-1, compare);
1448 if (k < 0)
1449 goto Fail;
1450 k = nb - k;
1451 bcount = k;
1452 if (k) {
1453 dest -= k;
1454 pb -= k;
1455 memcpy(dest+1, pb+1, k * sizeof(PyObject *));
1456 nb -= k;
1457 if (nb == 1)
1458 goto CopyA;
1459 /* nb==0 is impossible now if the comparison
1460 * function is consistent, but we can't assume
1461 * that it is.
1462 */
1463 if (nb == 0)
1464 goto Succeed;
1465 }
1466 *dest-- = *pa--;
1467 --na;
1468 if (na == 0)
1469 goto Succeed;
1470 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001471 ++min_gallop; /* penalize it for leaving galloping mode */
1472 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001473 }
1474Succeed:
1475 result = 0;
1476Fail:
1477 if (nb)
1478 memcpy(dest-(nb-1), baseb, nb * sizeof(PyObject*));
1479 return result;
1480CopyA:
1481 assert(nb == 1 && na > 0);
1482 /* The first element of pb belongs at the front of the merge. */
1483 dest -= na;
1484 pa -= na;
1485 memmove(dest+1, pa+1, na * sizeof(PyObject *));
1486 *dest = *pb;
1487 return 0;
1488}
1489
1490/* Merge the two runs at stack indices i and i+1.
1491 * Returns 0 on success, -1 on error.
1492 */
1493static int
1494merge_at(MergeState *ms, int i)
1495{
1496 PyObject **pa, **pb;
1497 int na, nb;
1498 int k;
1499 PyObject *compare;
1500
1501 assert(ms != NULL);
1502 assert(ms->n >= 2);
1503 assert(i >= 0);
1504 assert(i == ms->n - 2 || i == ms->n - 3);
1505
Tim Peterse05f65a2002-08-10 05:21:15 +00001506 pa = ms->pending[i].base;
1507 na = ms->pending[i].len;
1508 pb = ms->pending[i+1].base;
1509 nb = ms->pending[i+1].len;
Tim Petersa64dc242002-08-01 02:13:36 +00001510 assert(na > 0 && nb > 0);
1511 assert(pa + na == pb);
1512
1513 /* Record the length of the combined runs; if i is the 3rd-last
1514 * run now, also slide over the last run (which isn't involved
1515 * in this merge). The current run i+1 goes away in any case.
1516 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001517 ms->pending[i].len = na + nb;
1518 if (i == ms->n - 3)
1519 ms->pending[i+1] = ms->pending[i+2];
Tim Petersa64dc242002-08-01 02:13:36 +00001520 --ms->n;
1521
1522 /* Where does b start in a? Elements in a before that can be
1523 * ignored (already in place).
1524 */
1525 compare = ms->compare;
1526 k = gallop_right(*pb, pa, na, 0, compare);
1527 if (k < 0)
1528 return -1;
1529 pa += k;
1530 na -= k;
1531 if (na == 0)
1532 return 0;
1533
1534 /* Where does a end in b? Elements in b after that can be
1535 * ignored (already in place).
1536 */
1537 nb = gallop_left(pa[na-1], pb, nb, nb-1, compare);
1538 if (nb <= 0)
1539 return nb;
1540
1541 /* Merge what remains of the runs, using a temp array with
1542 * min(na, nb) elements.
1543 */
1544 if (na <= nb)
1545 return merge_lo(ms, pa, na, pb, nb);
1546 else
1547 return merge_hi(ms, pa, na, pb, nb);
1548}
1549
1550/* Examine the stack of runs waiting to be merged, merging adjacent runs
1551 * until the stack invariants are re-established:
1552 *
1553 * 1. len[-3] > len[-2] + len[-1]
1554 * 2. len[-2] > len[-1]
1555 *
1556 * See listsort.txt for more info.
1557 *
1558 * Returns 0 on success, -1 on error.
1559 */
1560static int
1561merge_collapse(MergeState *ms)
1562{
Tim Peterse05f65a2002-08-10 05:21:15 +00001563 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001564
1565 assert(ms);
1566 while (ms->n > 1) {
1567 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001568 if (n > 0 && p[n-1].len <= p[n].len + p[n+1].len) {
1569 if (p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001570 --n;
1571 if (merge_at(ms, n) < 0)
1572 return -1;
1573 }
Tim Peterse05f65a2002-08-10 05:21:15 +00001574 else if (p[n].len <= p[n+1].len) {
Tim Petersa64dc242002-08-01 02:13:36 +00001575 if (merge_at(ms, n) < 0)
1576 return -1;
1577 }
1578 else
1579 break;
1580 }
1581 return 0;
1582}
1583
1584/* Regardless of invariants, merge all runs on the stack until only one
1585 * remains. This is used at the end of the mergesort.
1586 *
1587 * Returns 0 on success, -1 on error.
1588 */
1589static int
1590merge_force_collapse(MergeState *ms)
1591{
Tim Peterse05f65a2002-08-10 05:21:15 +00001592 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001593
1594 assert(ms);
1595 while (ms->n > 1) {
1596 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001597 if (n > 0 && p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001598 --n;
1599 if (merge_at(ms, n) < 0)
1600 return -1;
1601 }
1602 return 0;
1603}
1604
1605/* Compute a good value for the minimum run length; natural runs shorter
1606 * than this are boosted artificially via binary insertion.
1607 *
1608 * If n < 64, return n (it's too small to bother with fancy stuff).
1609 * Else if n is an exact power of 2, return 32.
1610 * Else return an int k, 32 <= k <= 64, such that n/k is close to, but
1611 * strictly less than, an exact power of 2.
1612 *
1613 * See listsort.txt for more info.
1614 */
1615static int
1616merge_compute_minrun(int n)
1617{
1618 int r = 0; /* becomes 1 if any 1 bits are shifted off */
1619
1620 assert(n >= 0);
1621 while (n >= 64) {
1622 r |= n & 1;
1623 n >>= 1;
1624 }
1625 return n + r;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001626}
Guido van Rossuma119c0d1998-05-29 17:56:32 +00001627
Jeremy Hylton938ace62002-07-17 16:30:39 +00001628static PyTypeObject immutable_list_type;
Guido van Rossum3f236de1996-12-10 23:55:39 +00001629
Tim Petersa64dc242002-08-01 02:13:36 +00001630/* An adaptive, stable, natural mergesort. See listsort.txt.
1631 * Returns Py_None on success, NULL on error. Even in case of error, the
1632 * list will be some permutation of its input state (nothing is lost or
1633 * duplicated).
1634 */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001635static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001636listsort(PyListObject *self, PyObject *args)
Guido van Rossum3f236de1996-12-10 23:55:39 +00001637{
Tim Petersa64dc242002-08-01 02:13:36 +00001638 MergeState ms;
1639 PyObject **lo, **hi;
1640 int nremaining;
1641 int minrun;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001642 PyTypeObject *savetype;
Tim Petersa64dc242002-08-01 02:13:36 +00001643 PyObject *compare = NULL;
1644 PyObject *result = NULL; /* guilty until proved innocent */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001645
Tim Petersa64dc242002-08-01 02:13:36 +00001646 assert(self != NULL);
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001647 if (args != NULL) {
Tim Peters6bdbc9e2002-08-03 02:28:24 +00001648 if (!PyArg_ParseTuple(args, "|O:sort", &compare))
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001649 return NULL;
1650 }
Tim Petersa64dc242002-08-01 02:13:36 +00001651 merge_init(&ms, compare);
Tim Peters330f9e92002-07-19 07:05:44 +00001652
Tim Peters6d6c1a32001-08-02 04:15:00 +00001653 savetype = self->ob_type;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001654 self->ob_type = &immutable_list_type;
Tim Peters330f9e92002-07-19 07:05:44 +00001655
Tim Petersa64dc242002-08-01 02:13:36 +00001656 nremaining = self->ob_size;
1657 if (nremaining < 2)
1658 goto succeed;
Tim Peters330f9e92002-07-19 07:05:44 +00001659
Tim Petersa64dc242002-08-01 02:13:36 +00001660 /* March over the array once, left to right, finding natural runs,
1661 * and extending short natural runs to minrun elements.
1662 */
1663 lo = self->ob_item;
1664 hi = lo + nremaining;
1665 minrun = merge_compute_minrun(nremaining);
1666 do {
1667 int descending;
1668 int n;
Tim Peters330f9e92002-07-19 07:05:44 +00001669
Tim Petersa64dc242002-08-01 02:13:36 +00001670 /* Identify next run. */
1671 n = count_run(lo, hi, compare, &descending);
1672 if (n < 0)
1673 goto fail;
1674 if (descending)
1675 reverse_slice(lo, lo + n);
1676 /* If short, extend to min(minrun, nremaining). */
1677 if (n < minrun) {
1678 const int force = nremaining <= minrun ?
1679 nremaining : minrun;
1680 if (binarysort(lo, lo + force, lo + n, compare) < 0)
1681 goto fail;
1682 n = force;
1683 }
1684 /* Push run onto pending-runs stack, and maybe merge. */
1685 assert(ms.n < MAX_MERGE_PENDING);
Tim Peterse05f65a2002-08-10 05:21:15 +00001686 ms.pending[ms.n].base = lo;
1687 ms.pending[ms.n].len = n;
Tim Petersa64dc242002-08-01 02:13:36 +00001688 ++ms.n;
1689 if (merge_collapse(&ms) < 0)
1690 goto fail;
1691 /* Advance to find next run. */
1692 lo += n;
1693 nremaining -= n;
1694 } while (nremaining);
1695 assert(lo == hi);
Tim Peters330f9e92002-07-19 07:05:44 +00001696
Tim Petersa64dc242002-08-01 02:13:36 +00001697 if (merge_force_collapse(&ms) < 0)
1698 goto fail;
1699 assert(ms.n == 1);
Tim Peterse05f65a2002-08-10 05:21:15 +00001700 assert(ms.pending[0].base == self->ob_item);
1701 assert(ms.pending[0].len == self->ob_size);
Tim Petersa64dc242002-08-01 02:13:36 +00001702
1703succeed:
1704 result = Py_None;
Tim Peters330f9e92002-07-19 07:05:44 +00001705fail:
1706 self->ob_type = savetype;
Tim Petersa64dc242002-08-01 02:13:36 +00001707 merge_freemem(&ms);
1708 Py_XINCREF(result);
1709 return result;
Guido van Rossum3f236de1996-12-10 23:55:39 +00001710}
Tim Peters330f9e92002-07-19 07:05:44 +00001711#undef IFLT
Tim Peters66860f62002-08-04 17:47:26 +00001712#undef ISLT
Tim Peters330f9e92002-07-19 07:05:44 +00001713
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001714int
Fred Drakea2f55112000-07-09 15:16:51 +00001715PyList_Sort(PyObject *v)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001716{
1717 if (v == NULL || !PyList_Check(v)) {
1718 PyErr_BadInternalCall();
1719 return -1;
1720 }
1721 v = listsort((PyListObject *)v, (PyObject *)NULL);
1722 if (v == NULL)
1723 return -1;
1724 Py_DECREF(v);
1725 return 0;
1726}
1727
Guido van Rossumb86c5492001-02-12 22:06:02 +00001728static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001729listreverse(PyListObject *self)
Guido van Rossumb86c5492001-02-12 22:06:02 +00001730{
Tim Peters326b4482002-07-19 04:04:16 +00001731 if (self->ob_size > 1)
1732 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001733 Py_INCREF(Py_None);
1734 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001735}
1736
Guido van Rossum84c76f51990-10-30 13:32:20 +00001737int
Fred Drakea2f55112000-07-09 15:16:51 +00001738PyList_Reverse(PyObject *v)
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001739{
Tim Peters6063e262002-08-08 01:06:39 +00001740 PyListObject *self = (PyListObject *)v;
1741
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001742 if (v == NULL || !PyList_Check(v)) {
1743 PyErr_BadInternalCall();
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001744 return -1;
1745 }
Tim Peters6063e262002-08-08 01:06:39 +00001746 if (self->ob_size > 1)
1747 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001748 return 0;
1749}
1750
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001751PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001752PyList_AsTuple(PyObject *v)
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001753{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001754 PyObject *w;
1755 PyObject **p;
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001756 int n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001757 if (v == NULL || !PyList_Check(v)) {
1758 PyErr_BadInternalCall();
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001759 return NULL;
1760 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001761 n = ((PyListObject *)v)->ob_size;
1762 w = PyTuple_New(n);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001763 if (w == NULL)
1764 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001765 p = ((PyTupleObject *)w)->ob_item;
Thomas Wouters334fb892000-07-25 12:56:38 +00001766 memcpy((void *)p,
1767 (void *)((PyListObject *)v)->ob_item,
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001768 n*sizeof(PyObject *));
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001769 while (--n >= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001770 Py_INCREF(*p);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001771 p++;
1772 }
1773 return w;
1774}
1775
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001776static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001777listindex(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001778{
1779 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001780
Guido van Rossumed98d481991-03-06 13:07:53 +00001781 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001782 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1783 if (cmp > 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001784 return PyInt_FromLong((long)i);
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001785 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001786 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001787 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001788 PyErr_SetString(PyExc_ValueError, "list.index(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001789 return NULL;
1790}
1791
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001792static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001793listcount(PyListObject *self, PyObject *v)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001794{
1795 int count = 0;
1796 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001797
Guido van Rossume6f7d181991-10-20 20:20:40 +00001798 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001799 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1800 if (cmp > 0)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001801 count++;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001802 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001803 return NULL;
Guido van Rossume6f7d181991-10-20 20:20:40 +00001804 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001805 return PyInt_FromLong((long)count);
Guido van Rossume6f7d181991-10-20 20:20:40 +00001806}
1807
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001808static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001809listremove(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001810{
1811 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001812
Guido van Rossumed98d481991-03-06 13:07:53 +00001813 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001814 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1815 if (cmp > 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001816 if (list_ass_slice(self, i, i+1,
1817 (PyObject *)NULL) != 0)
Guido van Rossumed98d481991-03-06 13:07:53 +00001818 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001819 Py_INCREF(Py_None);
1820 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001821 }
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001822 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001823 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001824 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001825 PyErr_SetString(PyExc_ValueError, "list.remove(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001826 return NULL;
1827}
1828
Jeremy Hylton8caad492000-06-23 14:18:11 +00001829static int
1830list_traverse(PyListObject *o, visitproc visit, void *arg)
1831{
1832 int i, err;
1833 PyObject *x;
1834
1835 for (i = o->ob_size; --i >= 0; ) {
1836 x = o->ob_item[i];
1837 if (x != NULL) {
1838 err = visit(x, arg);
1839 if (err)
1840 return err;
1841 }
1842 }
1843 return 0;
1844}
1845
1846static int
1847list_clear(PyListObject *lp)
1848{
1849 (void) PyList_SetSlice((PyObject *)lp, 0, lp->ob_size, 0);
1850 return 0;
1851}
1852
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001853static PyObject *
1854list_richcompare(PyObject *v, PyObject *w, int op)
1855{
1856 PyListObject *vl, *wl;
1857 int i;
1858
1859 if (!PyList_Check(v) || !PyList_Check(w)) {
1860 Py_INCREF(Py_NotImplemented);
1861 return Py_NotImplemented;
1862 }
1863
1864 vl = (PyListObject *)v;
1865 wl = (PyListObject *)w;
1866
1867 if (vl->ob_size != wl->ob_size && (op == Py_EQ || op == Py_NE)) {
1868 /* Shortcut: if the lengths differ, the lists differ */
1869 PyObject *res;
1870 if (op == Py_EQ)
1871 res = Py_False;
1872 else
1873 res = Py_True;
1874 Py_INCREF(res);
1875 return res;
1876 }
1877
1878 /* Search for the first index where items are different */
1879 for (i = 0; i < vl->ob_size && i < wl->ob_size; i++) {
1880 int k = PyObject_RichCompareBool(vl->ob_item[i],
1881 wl->ob_item[i], Py_EQ);
1882 if (k < 0)
1883 return NULL;
1884 if (!k)
1885 break;
1886 }
1887
1888 if (i >= vl->ob_size || i >= wl->ob_size) {
1889 /* No more items to compare -- compare sizes */
1890 int vs = vl->ob_size;
1891 int ws = wl->ob_size;
1892 int cmp;
1893 PyObject *res;
1894 switch (op) {
1895 case Py_LT: cmp = vs < ws; break;
Tim Peters6ee42342001-07-06 17:45:43 +00001896 case Py_LE: cmp = vs <= ws; break;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001897 case Py_EQ: cmp = vs == ws; break;
1898 case Py_NE: cmp = vs != ws; break;
1899 case Py_GT: cmp = vs > ws; break;
1900 case Py_GE: cmp = vs >= ws; break;
1901 default: return NULL; /* cannot happen */
1902 }
1903 if (cmp)
1904 res = Py_True;
1905 else
1906 res = Py_False;
1907 Py_INCREF(res);
1908 return res;
1909 }
1910
1911 /* We have an item that differs -- shortcuts for EQ/NE */
1912 if (op == Py_EQ) {
1913 Py_INCREF(Py_False);
1914 return Py_False;
1915 }
1916 if (op == Py_NE) {
1917 Py_INCREF(Py_True);
1918 return Py_True;
1919 }
1920
1921 /* Compare the final item again using the proper operator */
1922 return PyObject_RichCompare(vl->ob_item[i], wl->ob_item[i], op);
1923}
1924
Tim Peters6d6c1a32001-08-02 04:15:00 +00001925/* Adapted from newer code by Tim */
1926static int
1927list_fill(PyListObject *result, PyObject *v)
1928{
1929 PyObject *it; /* iter(v) */
1930 int n; /* guess for result list size */
1931 int i;
1932
1933 n = result->ob_size;
1934
1935 /* Special-case list(a_list), for speed. */
1936 if (PyList_Check(v)) {
1937 if (v == (PyObject *)result)
1938 return 0; /* source is destination, we're done */
1939 return list_ass_slice(result, 0, n, v);
1940 }
1941
1942 /* Empty previous contents */
1943 if (n != 0) {
1944 if (list_ass_slice(result, 0, n, (PyObject *)NULL) != 0)
1945 return -1;
1946 }
1947
1948 /* Get iterator. There may be some low-level efficiency to be gained
1949 * by caching the tp_iternext slot instead of using PyIter_Next()
1950 * later, but premature optimization is the root etc.
1951 */
1952 it = PyObject_GetIter(v);
1953 if (it == NULL)
1954 return -1;
1955
1956 /* Guess a result list size. */
1957 n = -1; /* unknown */
1958 if (PySequence_Check(v) &&
1959 v->ob_type->tp_as_sequence->sq_length) {
1960 n = PySequence_Size(v);
1961 if (n < 0)
1962 PyErr_Clear();
1963 }
1964 if (n < 0)
1965 n = 8; /* arbitrary */
1966 NRESIZE(result->ob_item, PyObject*, n);
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001967 if (result->ob_item == NULL) {
1968 PyErr_NoMemory();
Tim Peters6d6c1a32001-08-02 04:15:00 +00001969 goto error;
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001970 }
Neal Norwitz35fc7602002-06-13 21:11:11 +00001971 memset(result->ob_item, 0, sizeof(*result->ob_item) * n);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001972 result->ob_size = n;
1973
1974 /* Run iterator to exhaustion. */
1975 for (i = 0; ; i++) {
1976 PyObject *item = PyIter_Next(it);
1977 if (item == NULL) {
1978 if (PyErr_Occurred())
1979 goto error;
1980 break;
1981 }
1982 if (i < n)
1983 PyList_SET_ITEM(result, i, item); /* steals ref */
1984 else {
1985 int status = ins1(result, result->ob_size, item);
1986 Py_DECREF(item); /* append creates a new ref */
1987 if (status < 0)
1988 goto error;
1989 }
1990 }
1991
1992 /* Cut back result list if initial guess was too large. */
1993 if (i < n && result != NULL) {
1994 if (list_ass_slice(result, i, n, (PyObject *)NULL) != 0)
1995 goto error;
1996 }
1997 Py_DECREF(it);
1998 return 0;
1999
2000 error:
2001 Py_DECREF(it);
2002 return -1;
2003}
2004
2005static int
2006list_init(PyListObject *self, PyObject *args, PyObject *kw)
2007{
2008 PyObject *arg = NULL;
2009 static char *kwlist[] = {"sequence", 0};
2010
2011 if (!PyArg_ParseTupleAndKeywords(args, kw, "|O:list", kwlist, &arg))
2012 return -1;
2013 if (arg != NULL)
2014 return list_fill(self, arg);
2015 if (self->ob_size > 0)
2016 return list_ass_slice(self, 0, self->ob_size, (PyObject*)NULL);
2017 return 0;
2018}
2019
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002020static long
2021list_nohash(PyObject *self)
2022{
2023 PyErr_SetString(PyExc_TypeError, "list objects are unhashable");
2024 return -1;
2025}
2026
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002027PyDoc_STRVAR(append_doc,
2028"L.append(object) -- append object to end");
2029PyDoc_STRVAR(extend_doc,
Martin v. Löwis673c0a22002-07-28 16:35:57 +00002030"L.extend(sequence) -- extend list by appending sequence elements");
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002031PyDoc_STRVAR(insert_doc,
2032"L.insert(index, object) -- insert object before index");
2033PyDoc_STRVAR(pop_doc,
2034"L.pop([index]) -> item -- remove and return item at index (default last)");
2035PyDoc_STRVAR(remove_doc,
2036"L.remove(value) -- remove first occurrence of value");
2037PyDoc_STRVAR(index_doc,
2038"L.index(value) -> integer -- return index of first occurrence of value");
2039PyDoc_STRVAR(count_doc,
2040"L.count(value) -> integer -- return number of occurrences of value");
2041PyDoc_STRVAR(reverse_doc,
2042"L.reverse() -- reverse *IN PLACE*");
2043PyDoc_STRVAR(sort_doc,
Tim Petersa64dc242002-08-01 02:13:36 +00002044"L.sort([cmpfunc]) -- stable sort *IN PLACE*; cmpfunc(x, y) -> -1, 0, 1");
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +00002045
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002046static PyMethodDef list_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002047 {"append", (PyCFunction)listappend, METH_O, append_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002048 {"insert", (PyCFunction)listinsert, METH_VARARGS, insert_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002049 {"extend", (PyCFunction)listextend, METH_O, extend_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002050 {"pop", (PyCFunction)listpop, METH_VARARGS, pop_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002051 {"remove", (PyCFunction)listremove, METH_O, remove_doc},
2052 {"index", (PyCFunction)listindex, METH_O, index_doc},
2053 {"count", (PyCFunction)listcount, METH_O, count_doc},
2054 {"reverse", (PyCFunction)listreverse, METH_NOARGS, reverse_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002055 {"sort", (PyCFunction)listsort, METH_VARARGS, sort_doc},
Tim Petersa64dc242002-08-01 02:13:36 +00002056 {NULL, NULL} /* sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002057};
2058
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002059static PySequenceMethods list_as_sequence = {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002060 (inquiry)list_length, /* sq_length */
2061 (binaryfunc)list_concat, /* sq_concat */
2062 (intargfunc)list_repeat, /* sq_repeat */
2063 (intargfunc)list_item, /* sq_item */
2064 (intintargfunc)list_slice, /* sq_slice */
2065 (intobjargproc)list_ass_item, /* sq_ass_item */
2066 (intintobjargproc)list_ass_slice, /* sq_ass_slice */
2067 (objobjproc)list_contains, /* sq_contains */
2068 (binaryfunc)list_inplace_concat, /* sq_inplace_concat */
2069 (intargfunc)list_inplace_repeat, /* sq_inplace_repeat */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002070};
2071
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002072PyDoc_STRVAR(list_doc,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002073"list() -> new list\n"
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002074"list(sequence) -> new list initialized from sequence's items");
Tim Peters6d6c1a32001-08-02 04:15:00 +00002075
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002076static PyObject *list_iter(PyObject *seq);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002077
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002078static PyObject *
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002079list_subscript(PyListObject* self, PyObject* item)
2080{
2081 if (PyInt_Check(item)) {
2082 long i = PyInt_AS_LONG(item);
2083 if (i < 0)
2084 i += PyList_GET_SIZE(self);
2085 return list_item(self, i);
2086 }
2087 else if (PyLong_Check(item)) {
2088 long i = PyLong_AsLong(item);
2089 if (i == -1 && PyErr_Occurred())
2090 return NULL;
2091 if (i < 0)
2092 i += PyList_GET_SIZE(self);
2093 return list_item(self, i);
2094 }
2095 else if (PySlice_Check(item)) {
2096 int start, stop, step, slicelength, cur, i;
2097 PyObject* result;
2098 PyObject* it;
2099
2100 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2101 &start, &stop, &step, &slicelength) < 0) {
2102 return NULL;
2103 }
2104
2105 if (slicelength <= 0) {
2106 return PyList_New(0);
2107 }
2108 else {
2109 result = PyList_New(slicelength);
2110 if (!result) return NULL;
2111
Tim Peters3b01a122002-07-19 02:35:45 +00002112 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002113 cur += step, i++) {
2114 it = PyList_GET_ITEM(self, cur);
2115 Py_INCREF(it);
2116 PyList_SET_ITEM(result, i, it);
2117 }
Tim Peters3b01a122002-07-19 02:35:45 +00002118
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002119 return result;
2120 }
2121 }
2122 else {
2123 PyErr_SetString(PyExc_TypeError,
2124 "list indices must be integers");
2125 return NULL;
2126 }
2127}
2128
Tim Peters3b01a122002-07-19 02:35:45 +00002129static int
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002130list_ass_subscript(PyListObject* self, PyObject* item, PyObject* value)
2131{
2132 if (PyInt_Check(item)) {
2133 long i = PyInt_AS_LONG(item);
2134 if (i < 0)
2135 i += PyList_GET_SIZE(self);
2136 return list_ass_item(self, i, value);
2137 }
2138 else if (PyLong_Check(item)) {
2139 long i = PyLong_AsLong(item);
2140 if (i == -1 && PyErr_Occurred())
2141 return -1;
2142 if (i < 0)
2143 i += PyList_GET_SIZE(self);
2144 return list_ass_item(self, i, value);
2145 }
2146 else if (PySlice_Check(item)) {
2147 int start, stop, step, slicelength;
2148
2149 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2150 &start, &stop, &step, &slicelength) < 0) {
2151 return -1;
2152 }
2153
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002154 /* treat L[slice(a,b)] = v _exactly_ like L[a:b] = v */
2155 if (step == 1 && ((PySliceObject*)item)->step == Py_None)
2156 return list_ass_slice(self, start, stop, value);
2157
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002158 if (value == NULL) {
2159 /* delete slice */
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002160 PyObject **garbage, **it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002161 int cur, i, j;
Tim Peters3b01a122002-07-19 02:35:45 +00002162
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002163 if (slicelength <= 0)
2164 return 0;
2165
2166 if (step < 0) {
2167 stop = start + 1;
2168 start = stop + step*(slicelength - 1) - 1;
2169 step = -step;
2170 }
2171
2172 garbage = (PyObject**)
2173 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002174
2175 /* drawing pictures might help
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002176 understand these for loops */
Guido van Rossum75a20b12002-06-11 12:22:28 +00002177 for (cur = start, i = 0;
2178 cur < stop;
Michael W. Hudson56796f62002-07-29 14:35:04 +00002179 cur += step, i++) {
2180 int lim = step;
2181
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002182 garbage[i] = PyList_GET_ITEM(self, cur);
2183
Michael W. Hudson56796f62002-07-29 14:35:04 +00002184 if (cur + step >= self->ob_size) {
2185 lim = self->ob_size - cur - 1;
2186 }
2187
2188 for (j = 0; j < lim; j++) {
Tim Peters3b01a122002-07-19 02:35:45 +00002189 PyList_SET_ITEM(self, cur + j - i,
Guido van Rossum75a20b12002-06-11 12:22:28 +00002190 PyList_GET_ITEM(self,
2191 cur + j + 1));
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002192 }
2193 }
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002194 for (cur = start + slicelength*step + 1;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002195 cur < self->ob_size; cur++) {
2196 PyList_SET_ITEM(self, cur - slicelength,
2197 PyList_GET_ITEM(self, cur));
2198 }
2199 self->ob_size -= slicelength;
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002200 it = self->ob_item;
2201 NRESIZE(it, PyObject*, self->ob_size);
2202 self->ob_item = it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002203
2204 for (i = 0; i < slicelength; i++) {
2205 Py_DECREF(garbage[i]);
2206 }
2207 PyMem_FREE(garbage);
2208
2209 return 0;
2210 }
2211 else {
2212 /* assign slice */
2213 PyObject **garbage, *ins;
2214 int cur, i;
2215
2216 if (!PyList_Check(value)) {
2217 PyErr_Format(PyExc_TypeError,
2218 "must assign list (not \"%.200s\") to slice",
2219 value->ob_type->tp_name);
2220 return -1;
2221 }
2222
2223 if (PyList_GET_SIZE(value) != slicelength) {
2224 PyErr_Format(PyExc_ValueError,
2225 "attempt to assign list of size %d to extended slice of size %d",
2226 PyList_Size(value), slicelength);
2227 return -1;
2228 }
2229
2230 if (!slicelength)
2231 return 0;
2232
2233 /* protect against a[::-1] = a */
Tim Peters3b01a122002-07-19 02:35:45 +00002234 if (self == (PyListObject*)value) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002235 value = list_slice((PyListObject*)value, 0,
2236 PyList_GET_SIZE(value));
Tim Peters3b01a122002-07-19 02:35:45 +00002237 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002238 else {
2239 Py_INCREF(value);
2240 }
2241
2242 garbage = (PyObject**)
2243 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002244
2245 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002246 cur += step, i++) {
2247 garbage[i] = PyList_GET_ITEM(self, cur);
Tim Peters3b01a122002-07-19 02:35:45 +00002248
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002249 ins = PyList_GET_ITEM(value, i);
2250 Py_INCREF(ins);
2251 PyList_SET_ITEM(self, cur, ins);
2252 }
2253
2254 for (i = 0; i < slicelength; i++) {
2255 Py_DECREF(garbage[i]);
2256 }
Tim Peters3b01a122002-07-19 02:35:45 +00002257
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002258 PyMem_FREE(garbage);
2259 Py_DECREF(value);
Tim Peters3b01a122002-07-19 02:35:45 +00002260
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002261 return 0;
2262 }
Tim Peters3b01a122002-07-19 02:35:45 +00002263 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002264 else {
Tim Peters3b01a122002-07-19 02:35:45 +00002265 PyErr_SetString(PyExc_TypeError,
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002266 "list indices must be integers");
2267 return -1;
2268 }
2269}
2270
2271static PyMappingMethods list_as_mapping = {
2272 (inquiry)list_length,
2273 (binaryfunc)list_subscript,
2274 (objobjargproc)list_ass_subscript
2275};
2276
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002277PyTypeObject PyList_Type = {
2278 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002279 0,
2280 "list",
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002281 sizeof(PyListObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002282 0,
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002283 (destructor)list_dealloc, /* tp_dealloc */
2284 (printfunc)list_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002285 0, /* tp_getattr */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002286 0, /* tp_setattr */
2287 0, /* tp_compare */
2288 (reprfunc)list_repr, /* tp_repr */
2289 0, /* tp_as_number */
2290 &list_as_sequence, /* tp_as_sequence */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002291 &list_as_mapping, /* tp_as_mapping */
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002292 list_nohash, /* tp_hash */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002293 0, /* tp_call */
2294 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002295 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002296 0, /* tp_setattro */
2297 0, /* tp_as_buffer */
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002298 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
Tim Peters6d6c1a32001-08-02 04:15:00 +00002299 Py_TPFLAGS_BASETYPE, /* tp_flags */
2300 list_doc, /* tp_doc */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002301 (traverseproc)list_traverse, /* tp_traverse */
2302 (inquiry)list_clear, /* tp_clear */
2303 list_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002304 0, /* tp_weaklistoffset */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002305 list_iter, /* tp_iter */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002306 0, /* tp_iternext */
2307 list_methods, /* tp_methods */
2308 0, /* tp_members */
2309 0, /* tp_getset */
2310 0, /* tp_base */
2311 0, /* tp_dict */
2312 0, /* tp_descr_get */
2313 0, /* tp_descr_set */
2314 0, /* tp_dictoffset */
2315 (initproc)list_init, /* tp_init */
2316 PyType_GenericAlloc, /* tp_alloc */
2317 PyType_GenericNew, /* tp_new */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002318 PyObject_GC_Del, /* tp_free */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002319};
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002320
2321
2322/* During a sort, we really can't have anyone modifying the list; it could
2323 cause core dumps. Thus, we substitute a dummy type that raises an
2324 explanatory exception when a modifying operation is used. Caveat:
2325 comparisons may behave differently; but I guess it's a bad idea anyway to
2326 compare a list that's being sorted... */
2327
2328static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00002329immutable_list_op(void)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002330{
2331 PyErr_SetString(PyExc_TypeError,
2332 "a list cannot be modified while it is being sorted");
2333 return NULL;
2334}
2335
2336static PyMethodDef immutable_list_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002337 {"append", (PyCFunction)immutable_list_op, METH_VARARGS},
2338 {"insert", (PyCFunction)immutable_list_op, METH_VARARGS},
Tim Peters52e07172001-08-30 06:15:32 +00002339 {"extend", (PyCFunction)immutable_list_op, METH_O},
2340 {"pop", (PyCFunction)immutable_list_op, METH_VARARGS},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002341 {"remove", (PyCFunction)immutable_list_op, METH_VARARGS},
2342 {"index", (PyCFunction)listindex, METH_O},
2343 {"count", (PyCFunction)listcount, METH_O},
2344 {"reverse", (PyCFunction)immutable_list_op, METH_VARARGS},
2345 {"sort", (PyCFunction)immutable_list_op, METH_VARARGS},
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002346 {NULL, NULL} /* sentinel */
2347};
2348
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002349static int
Fred Drakea2f55112000-07-09 15:16:51 +00002350immutable_list_ass(void)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002351{
2352 immutable_list_op();
2353 return -1;
2354}
2355
2356static PySequenceMethods immutable_list_as_sequence = {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002357 (inquiry)list_length, /* sq_length */
2358 (binaryfunc)list_concat, /* sq_concat */
2359 (intargfunc)list_repeat, /* sq_repeat */
2360 (intargfunc)list_item, /* sq_item */
2361 (intintargfunc)list_slice, /* sq_slice */
2362 (intobjargproc)immutable_list_ass, /* sq_ass_item */
2363 (intintobjargproc)immutable_list_ass, /* sq_ass_slice */
2364 (objobjproc)list_contains, /* sq_contains */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002365};
2366
2367static PyTypeObject immutable_list_type = {
2368 PyObject_HEAD_INIT(&PyType_Type)
2369 0,
2370 "list (immutable, during sort)",
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002371 sizeof(PyListObject),
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002372 0,
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002373 0, /* Cannot happen */ /* tp_dealloc */
2374 (printfunc)list_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002375 0, /* tp_getattr */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002376 0, /* tp_setattr */
2377 0, /* Won't be called */ /* tp_compare */
2378 (reprfunc)list_repr, /* tp_repr */
2379 0, /* tp_as_number */
2380 &immutable_list_as_sequence, /* tp_as_sequence */
2381 0, /* tp_as_mapping */
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002382 list_nohash, /* tp_hash */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002383 0, /* tp_call */
2384 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002385 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002386 0, /* tp_setattro */
2387 0, /* tp_as_buffer */
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002388 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002389 list_doc, /* tp_doc */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002390 (traverseproc)list_traverse, /* tp_traverse */
2391 0, /* tp_clear */
2392 list_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002393 0, /* tp_weaklistoffset */
2394 0, /* tp_iter */
2395 0, /* tp_iternext */
2396 immutable_list_methods, /* tp_methods */
2397 0, /* tp_members */
2398 0, /* tp_getset */
2399 0, /* tp_base */
2400 0, /* tp_dict */
2401 0, /* tp_descr_get */
2402 0, /* tp_descr_set */
2403 0, /* tp_init */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002404 /* NOTE: This is *not* the standard list_type struct! */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002405};
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002406
2407
2408/*********************** List Iterator **************************/
2409
2410typedef struct {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002411 PyObject_HEAD
2412 long it_index;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002413 PyListObject *it_seq; /* Set to NULL when iterator is exhausted */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002414} listiterobject;
2415
2416PyTypeObject PyListIter_Type;
2417
Guido van Rossum5086e492002-07-16 15:56:52 +00002418static PyObject *
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002419list_iter(PyObject *seq)
2420{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002421 listiterobject *it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002422
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002423 if (!PyList_Check(seq)) {
2424 PyErr_BadInternalCall();
2425 return NULL;
2426 }
2427 it = PyObject_GC_New(listiterobject, &PyListIter_Type);
2428 if (it == NULL)
2429 return NULL;
2430 it->it_index = 0;
2431 Py_INCREF(seq);
2432 it->it_seq = (PyListObject *)seq;
2433 _PyObject_GC_TRACK(it);
2434 return (PyObject *)it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002435}
2436
2437static void
2438listiter_dealloc(listiterobject *it)
2439{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002440 _PyObject_GC_UNTRACK(it);
Guido van Rossum86103ae2002-07-16 20:07:32 +00002441 Py_XDECREF(it->it_seq);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002442 PyObject_GC_Del(it);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002443}
2444
2445static int
2446listiter_traverse(listiterobject *it, visitproc visit, void *arg)
2447{
Guido van Rossum86103ae2002-07-16 20:07:32 +00002448 if (it->it_seq == NULL)
2449 return 0;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002450 return visit((PyObject *)it->it_seq, arg);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002451}
2452
2453
2454static PyObject *
2455listiter_getiter(PyObject *it)
2456{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002457 Py_INCREF(it);
2458 return it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002459}
2460
2461static PyObject *
Tim Peters93b2cc42002-06-01 05:22:55 +00002462listiter_next(listiterobject *it)
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002463{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002464 PyListObject *seq;
2465 PyObject *item;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002466
Tim Peters93b2cc42002-06-01 05:22:55 +00002467 assert(it != NULL);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002468 seq = it->it_seq;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002469 if (seq == NULL)
2470 return NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002471 assert(PyList_Check(seq));
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002472
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002473 if (it->it_index < PyList_GET_SIZE(seq)) {
Tim Peters93b2cc42002-06-01 05:22:55 +00002474 item = PyList_GET_ITEM(seq, it->it_index);
2475 ++it->it_index;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002476 Py_INCREF(item);
2477 return item;
2478 }
Guido van Rossum86103ae2002-07-16 20:07:32 +00002479
2480 Py_DECREF(seq);
2481 it->it_seq = NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002482 return NULL;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002483}
2484
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002485PyTypeObject PyListIter_Type = {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002486 PyObject_HEAD_INIT(&PyType_Type)
2487 0, /* ob_size */
2488 "listiterator", /* tp_name */
2489 sizeof(listiterobject), /* tp_basicsize */
2490 0, /* tp_itemsize */
2491 /* methods */
2492 (destructor)listiter_dealloc, /* tp_dealloc */
2493 0, /* tp_print */
2494 0, /* tp_getattr */
2495 0, /* tp_setattr */
2496 0, /* tp_compare */
2497 0, /* tp_repr */
2498 0, /* tp_as_number */
2499 0, /* tp_as_sequence */
2500 0, /* tp_as_mapping */
2501 0, /* tp_hash */
2502 0, /* tp_call */
2503 0, /* tp_str */
2504 PyObject_GenericGetAttr, /* tp_getattro */
2505 0, /* tp_setattro */
2506 0, /* tp_as_buffer */
2507 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2508 0, /* tp_doc */
2509 (traverseproc)listiter_traverse, /* tp_traverse */
2510 0, /* tp_clear */
2511 0, /* tp_richcompare */
2512 0, /* tp_weaklistoffset */
2513 (getiterfunc)listiter_getiter, /* tp_iter */
2514 (iternextfunc)listiter_next, /* tp_iternext */
Guido van Rossum86103ae2002-07-16 20:07:32 +00002515 0, /* tp_methods */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002516 0, /* tp_members */
2517 0, /* tp_getset */
2518 0, /* tp_base */
2519 0, /* tp_dict */
2520 0, /* tp_descr_get */
2521 0, /* tp_descr_set */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002522};