blob: 3e3b4d7c7b7764dcace56c6eae3445e340a796fc [file] [log] [blame]
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001/* List object implementation */
2
Guido van Rossumc0b618a1997-05-02 03:12:38 +00003#include "Python.h"
4
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00005#ifdef STDC_HEADERS
6#include <stddef.h>
7#else
8#include <sys/types.h> /* For size_t */
9#endif
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000010
Guido van Rossuma46d51d1995-01-26 22:59:43 +000011static int
Fred Drakea2f55112000-07-09 15:16:51 +000012roundupsize(int n)
Guido van Rossuma46d51d1995-01-26 22:59:43 +000013{
Tim Peters65b8b842001-05-26 05:28:40 +000014 unsigned int nbits = 0;
15 unsigned int n2 = (unsigned int)n >> 5;
16
Tim Peters3b01a122002-07-19 02:35:45 +000017 /* Round up:
Tim Peters65b8b842001-05-26 05:28:40 +000018 * If n < 256, to a multiple of 8.
19 * If n < 2048, to a multiple of 64.
20 * If n < 16384, to a multiple of 512.
21 * If n < 131072, to a multiple of 4096.
22 * If n < 1048576, to a multiple of 32768.
23 * If n < 8388608, to a multiple of 262144.
24 * If n < 67108864, to a multiple of 2097152.
25 * If n < 536870912, to a multiple of 16777216.
26 * ...
27 * If n < 2**(5+3*i), to a multiple of 2**(3*i).
28 *
29 * This over-allocates proportional to the list size, making room
30 * for additional growth. The over-allocation is mild, but is
31 * enough to give linear-time amortized behavior over a long
32 * sequence of appends() in the presence of a poorly-performing
33 * system realloc() (which is a reality, e.g., across all flavors
34 * of Windows, with Win9x behavior being particularly bad -- and
35 * we've still got address space fragmentation problems on Win9x
36 * even with this scheme, although it requires much longer lists to
37 * provoke them than it used to).
38 */
39 do {
40 n2 >>= 3;
41 nbits += 3;
42 } while (n2);
43 return ((n >> nbits) + 1) << nbits;
44 }
Guido van Rossuma46d51d1995-01-26 22:59:43 +000045
Neal Norwitzd4e5be52002-05-22 23:19:17 +000046#define NRESIZE(var, type, nitems) \
47do { \
48 size_t _new_size = roundupsize(nitems); \
49 if (_new_size <= ((~(size_t)0) / sizeof(type))) \
50 PyMem_RESIZE(var, type, _new_size); \
51 else \
52 var = NULL; \
53} while (0)
Guido van Rossuma46d51d1995-01-26 22:59:43 +000054
Guido van Rossumc0b618a1997-05-02 03:12:38 +000055PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +000056PyList_New(int size)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000057{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000058 PyListObject *op;
Guido van Rossum6cd2fe01994-08-29 12:45:32 +000059 size_t nbytes;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000060 if (size < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000061 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000062 return NULL;
63 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000064 nbytes = size * sizeof(PyObject *);
Guido van Rossum1e28e5e1992-08-19 16:46:30 +000065 /* Check for overflow */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000066 if (nbytes / sizeof(PyObject *) != (size_t)size) {
67 return PyErr_NoMemory();
Guido van Rossum1e28e5e1992-08-19 16:46:30 +000068 }
Neil Schemenauere83c00e2001-08-29 23:54:21 +000069 op = PyObject_GC_New(PyListObject, &PyList_Type);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000070 if (op == NULL) {
Neil Schemenauere83c00e2001-08-29 23:54:21 +000071 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000072 }
73 if (size <= 0) {
74 op->ob_item = NULL;
75 }
76 else {
Guido van Rossumb18618d2000-05-03 23:44:39 +000077 op->ob_item = (PyObject **) PyMem_MALLOC(nbytes);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000078 if (op->ob_item == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000079 return PyErr_NoMemory();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000080 }
Neal Norwitz35fc7602002-06-13 21:11:11 +000081 memset(op->ob_item, 0, sizeof(*op->ob_item) * size);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000082 }
Neil Schemenauere83c00e2001-08-29 23:54:21 +000083 op->ob_size = size;
Neil Schemenauere83c00e2001-08-29 23:54:21 +000084 _PyObject_GC_TRACK(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000086}
87
88int
Fred Drakea2f55112000-07-09 15:16:51 +000089PyList_Size(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000090{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000091 if (!PyList_Check(op)) {
92 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000093 return -1;
94 }
95 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +000096 return ((PyListObject *)op) -> ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000097}
98
Guido van Rossumc0b618a1997-05-02 03:12:38 +000099static PyObject *indexerr;
Guido van Rossum929f1b81996-08-09 20:51:27 +0000100
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000101PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000102PyList_GetItem(PyObject *op, int i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000103{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000104 if (!PyList_Check(op)) {
105 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000106 return NULL;
107 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000108 if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
Guido van Rossum929f1b81996-08-09 20:51:27 +0000109 if (indexerr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000110 indexerr = PyString_FromString(
111 "list index out of range");
112 PyErr_SetObject(PyExc_IndexError, indexerr);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000113 return NULL;
114 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000115 return ((PyListObject *)op) -> ob_item[i];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000116}
117
118int
Fred Drakea2f55112000-07-09 15:16:51 +0000119PyList_SetItem(register PyObject *op, register int i,
120 register PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000121{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000122 register PyObject *olditem;
123 register PyObject **p;
124 if (!PyList_Check(op)) {
125 Py_XDECREF(newitem);
126 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000127 return -1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000128 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000129 if (i < 0 || i >= ((PyListObject *)op) -> ob_size) {
130 Py_XDECREF(newitem);
131 PyErr_SetString(PyExc_IndexError,
132 "list assignment index out of range");
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000133 return -1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000134 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000135 p = ((PyListObject *)op) -> ob_item + i;
Guido van Rossum5fe60581995-03-09 12:12:50 +0000136 olditem = *p;
137 *p = newitem;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000138 Py_XDECREF(olditem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000139 return 0;
140}
141
142static int
Fred Drakea2f55112000-07-09 15:16:51 +0000143ins1(PyListObject *self, int where, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000144{
145 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000146 PyObject **items;
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000147 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000148 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000149 return -1;
150 }
Trent Micka5846642000-08-13 22:47:45 +0000151 if (self->ob_size == INT_MAX) {
152 PyErr_SetString(PyExc_OverflowError,
153 "cannot add more objects to list");
154 return -1;
155 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000156 items = self->ob_item;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000157 NRESIZE(items, PyObject *, self->ob_size+1);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000158 if (items == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000159 PyErr_NoMemory();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000160 return -1;
161 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000162 if (where < 0)
163 where = 0;
164 if (where > self->ob_size)
165 where = self->ob_size;
166 for (i = self->ob_size; --i >= where; )
167 items[i+1] = items[i];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000168 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000169 items[where] = v;
170 self->ob_item = items;
171 self->ob_size++;
172 return 0;
173}
174
175int
Fred Drakea2f55112000-07-09 15:16:51 +0000176PyList_Insert(PyObject *op, int where, PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000177{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000178 if (!PyList_Check(op)) {
179 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000180 return -1;
181 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000182 return ins1((PyListObject *)op, where, newitem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000183}
184
185int
Fred Drakea2f55112000-07-09 15:16:51 +0000186PyList_Append(PyObject *op, PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000187{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000188 if (!PyList_Check(op)) {
189 PyErr_BadInternalCall();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000190 return -1;
191 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000192 return ins1((PyListObject *)op,
193 (int) ((PyListObject *)op)->ob_size, newitem);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000194}
195
196/* Methods */
197
198static void
Fred Drakea2f55112000-07-09 15:16:51 +0000199list_dealloc(PyListObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000200{
201 int i;
Guido van Rossumff413af2002-03-28 20:34:59 +0000202 PyObject_GC_UnTrack(op);
Guido van Rossumd724b232000-03-13 16:01:29 +0000203 Py_TRASHCAN_SAFE_BEGIN(op)
Jack Jansen7874d1f1995-01-19 12:09:27 +0000204 if (op->ob_item != NULL) {
Guido van Rossumfa717011999-06-09 15:19:34 +0000205 /* Do it backwards, for Christian Tismer.
206 There's a simple test case where somehow this reduces
207 thrashing when a *very* large list is created and
208 immediately deleted. */
209 i = op->ob_size;
210 while (--i >= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000211 Py_XDECREF(op->ob_item[i]);
Jack Jansen7874d1f1995-01-19 12:09:27 +0000212 }
Guido van Rossumb18618d2000-05-03 23:44:39 +0000213 PyMem_FREE(op->ob_item);
Jack Jansen7874d1f1995-01-19 12:09:27 +0000214 }
Guido van Rossum9475a232001-10-05 20:51:39 +0000215 op->ob_type->tp_free((PyObject *)op);
Guido van Rossumd724b232000-03-13 16:01:29 +0000216 Py_TRASHCAN_SAFE_END(op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000217}
218
Guido van Rossum90933611991-06-07 16:10:43 +0000219static int
Fred Drakea2f55112000-07-09 15:16:51 +0000220list_print(PyListObject *op, FILE *fp, int flags)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000221{
222 int i;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000223
224 i = Py_ReprEnter((PyObject*)op);
225 if (i != 0) {
226 if (i < 0)
227 return i;
228 fprintf(fp, "[...]");
229 return 0;
230 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000231 fprintf(fp, "[");
Guido van Rossum90933611991-06-07 16:10:43 +0000232 for (i = 0; i < op->ob_size; i++) {
233 if (i > 0)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000234 fprintf(fp, ", ");
Guido van Rossumfb376de1998-04-10 22:47:27 +0000235 if (PyObject_Print(op->ob_item[i], fp, 0) != 0) {
236 Py_ReprLeave((PyObject *)op);
Guido van Rossum90933611991-06-07 16:10:43 +0000237 return -1;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000238 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000239 }
240 fprintf(fp, "]");
Guido van Rossumfb376de1998-04-10 22:47:27 +0000241 Py_ReprLeave((PyObject *)op);
Guido van Rossum90933611991-06-07 16:10:43 +0000242 return 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000243}
244
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000245static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000246list_repr(PyListObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000247{
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000248 int i;
Tim Petersa7259592001-06-16 05:11:17 +0000249 PyObject *s, *temp;
250 PyObject *pieces = NULL, *result = NULL;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000251
252 i = Py_ReprEnter((PyObject*)v);
253 if (i != 0) {
Tim Petersa7259592001-06-16 05:11:17 +0000254 return i > 0 ? PyString_FromString("[...]") : NULL;
Guido van Rossumfb376de1998-04-10 22:47:27 +0000255 }
Tim Petersa7259592001-06-16 05:11:17 +0000256
257 if (v->ob_size == 0) {
258 result = PyString_FromString("[]");
259 goto Done;
260 }
261
262 pieces = PyList_New(0);
263 if (pieces == NULL)
264 goto Done;
265
266 /* Do repr() on each element. Note that this may mutate the list,
267 so must refetch the list size on each iteration. */
268 for (i = 0; i < v->ob_size; ++i) {
269 int status;
270 s = PyObject_Repr(v->ob_item[i]);
271 if (s == NULL)
272 goto Done;
273 status = PyList_Append(pieces, s);
274 Py_DECREF(s); /* append created a new ref */
275 if (status < 0)
276 goto Done;
277 }
278
279 /* Add "[]" decorations to the first and last items. */
280 assert(PyList_GET_SIZE(pieces) > 0);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000281 s = PyString_FromString("[");
Tim Petersa7259592001-06-16 05:11:17 +0000282 if (s == NULL)
283 goto Done;
284 temp = PyList_GET_ITEM(pieces, 0);
285 PyString_ConcatAndDel(&s, temp);
286 PyList_SET_ITEM(pieces, 0, s);
287 if (s == NULL)
288 goto Done;
289
290 s = PyString_FromString("]");
291 if (s == NULL)
292 goto Done;
293 temp = PyList_GET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1);
294 PyString_ConcatAndDel(&temp, s);
295 PyList_SET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1, temp);
296 if (temp == NULL)
297 goto Done;
298
299 /* Paste them all together with ", " between. */
300 s = PyString_FromString(", ");
301 if (s == NULL)
302 goto Done;
303 result = _PyString_Join(s, pieces);
Tim Peters3b01a122002-07-19 02:35:45 +0000304 Py_DECREF(s);
Tim Petersa7259592001-06-16 05:11:17 +0000305
306Done:
307 Py_XDECREF(pieces);
Guido van Rossumfb376de1998-04-10 22:47:27 +0000308 Py_ReprLeave((PyObject *)v);
Tim Petersa7259592001-06-16 05:11:17 +0000309 return result;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000310}
311
312static int
Fred Drakea2f55112000-07-09 15:16:51 +0000313list_length(PyListObject *a)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000314{
315 return a->ob_size;
316}
317
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000318
319
320static int
Fred Drakea2f55112000-07-09 15:16:51 +0000321list_contains(PyListObject *a, PyObject *el)
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000322{
Raymond Hettingeraae59992002-09-05 14:23:49 +0000323 int i, cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000324
Raymond Hettingeraae59992002-09-05 14:23:49 +0000325 for (i = 0, cmp = 0 ; cmp == 0 && i < a->ob_size; ++i)
326 cmp = PyObject_RichCompareBool(el, PyList_GET_ITEM(a, i),
Guido van Rossum65e1cea2001-01-17 22:11:59 +0000327 Py_EQ);
Neal Norwitzbb9c5f52002-09-05 21:32:55 +0000328 return cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000329}
330
331
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000332static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000333list_item(PyListObject *a, int i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000334{
335 if (i < 0 || i >= a->ob_size) {
Guido van Rossum929f1b81996-08-09 20:51:27 +0000336 if (indexerr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000337 indexerr = PyString_FromString(
338 "list index out of range");
339 PyErr_SetObject(PyExc_IndexError, indexerr);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000340 return NULL;
341 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000342 Py_INCREF(a->ob_item[i]);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000343 return a->ob_item[i];
344}
345
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000346static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000347list_slice(PyListObject *a, int ilow, int ihigh)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000348{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000349 PyListObject *np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000350 int i;
351 if (ilow < 0)
352 ilow = 0;
353 else if (ilow > a->ob_size)
354 ilow = a->ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000355 if (ihigh < ilow)
356 ihigh = ilow;
357 else if (ihigh > a->ob_size)
358 ihigh = a->ob_size;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000359 np = (PyListObject *) PyList_New(ihigh - ilow);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000360 if (np == NULL)
361 return NULL;
362 for (i = ilow; i < ihigh; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000363 PyObject *v = a->ob_item[i];
364 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000365 np->ob_item[i - ilow] = v;
366 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000367 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000368}
369
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000370PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000371PyList_GetSlice(PyObject *a, int ilow, int ihigh)
Guido van Rossum234f9421993-06-17 12:35:49 +0000372{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000373 if (!PyList_Check(a)) {
374 PyErr_BadInternalCall();
Guido van Rossum234f9421993-06-17 12:35:49 +0000375 return NULL;
376 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000377 return list_slice((PyListObject *)a, ilow, ihigh);
Guido van Rossum234f9421993-06-17 12:35:49 +0000378}
379
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000380static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000381list_concat(PyListObject *a, PyObject *bb)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000382{
383 int size;
384 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000385 PyListObject *np;
386 if (!PyList_Check(bb)) {
Fred Drakeb6a9ada2000-06-01 03:12:13 +0000387 PyErr_Format(PyExc_TypeError,
Fred Drake914a2ed2000-06-01 14:31:03 +0000388 "can only concatenate list (not \"%.200s\") to list",
389 bb->ob_type->tp_name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000390 return NULL;
391 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000392#define b ((PyListObject *)bb)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000393 size = a->ob_size + b->ob_size;
Guido van Rossuma5c0e6d2002-10-11 21:05:56 +0000394 if (size < 0)
395 return PyErr_NoMemory();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000396 np = (PyListObject *) PyList_New(size);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000397 if (np == NULL) {
Guido van Rossum90933611991-06-07 16:10:43 +0000398 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000399 }
400 for (i = 0; i < a->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000401 PyObject *v = a->ob_item[i];
402 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000403 np->ob_item[i] = v;
404 }
405 for (i = 0; i < b->ob_size; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000406 PyObject *v = b->ob_item[i];
407 Py_INCREF(v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000408 np->ob_item[i + a->ob_size] = v;
409 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000410 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000411#undef b
412}
413
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000414static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000415list_repeat(PyListObject *a, int n)
Guido van Rossumed98d481991-03-06 13:07:53 +0000416{
417 int i, j;
418 int size;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000419 PyListObject *np;
420 PyObject **p;
Guido van Rossumed98d481991-03-06 13:07:53 +0000421 if (n < 0)
422 n = 0;
423 size = a->ob_size * n;
Guido van Rossumbfa5a142002-10-11 23:39:35 +0000424 if (n && size/n != a->ob_size)
Guido van Rossuma5c0e6d2002-10-11 21:05:56 +0000425 return PyErr_NoMemory();
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000426 np = (PyListObject *) PyList_New(size);
Guido van Rossumed98d481991-03-06 13:07:53 +0000427 if (np == NULL)
428 return NULL;
429 p = np->ob_item;
430 for (i = 0; i < n; i++) {
431 for (j = 0; j < a->ob_size; j++) {
432 *p = a->ob_item[j];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000433 Py_INCREF(*p);
Guido van Rossumed98d481991-03-06 13:07:53 +0000434 p++;
435 }
436 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000437 return (PyObject *) np;
Guido van Rossumed98d481991-03-06 13:07:53 +0000438}
439
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000440static int
Fred Drakea2f55112000-07-09 15:16:51 +0000441list_ass_slice(PyListObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000442{
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000443 /* Because [X]DECREF can recursively invoke list operations on
444 this list, we must postpone all [X]DECREF activity until
445 after the list is back in its canonical shape. Therefore
446 we must allocate an additional array, 'recycle', into which
447 we temporarily copy the items that are deleted from the
448 list. :-( */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000449 PyObject **recycle, **p;
450 PyObject **item;
Michael W. Hudson5da854f2002-11-05 17:38:05 +0000451 PyObject *v_as_SF = NULL; /* PySequence_Fast(v) */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000452 int n; /* Size of replacement list */
453 int d; /* Change in size */
454 int k; /* Loop index */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000455#define b ((PyListObject *)v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000456 if (v == NULL)
457 n = 0;
Michael W. Hudson5da854f2002-11-05 17:38:05 +0000458 else {
459 char msg[256];
Neal Norwitz03b109a2002-11-05 22:41:37 +0000460 PyOS_snprintf(msg, sizeof(msg),
461 "must assign sequence"
462 " (not \"%.200s\") to slice",
463 v->ob_type->tp_name);
Michael W. Hudson5da854f2002-11-05 17:38:05 +0000464 v_as_SF = PySequence_Fast(v, msg);
465 if(v_as_SF == NULL)
466 return -1;
467 n = PySequence_Fast_GET_SIZE(v_as_SF);
468
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000469 if (a == b) {
470 /* Special case "a[i:j] = a" -- copy b first */
471 int ret;
472 v = list_slice(b, 0, n);
473 ret = list_ass_slice(a, ilow, ihigh, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000474 Py_DECREF(v);
Guido van Rossum32dffaa1991-12-24 13:27:34 +0000475 return ret;
476 }
477 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000478 if (ilow < 0)
479 ilow = 0;
480 else if (ilow > a->ob_size)
481 ilow = a->ob_size;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000482 if (ihigh < ilow)
483 ihigh = ilow;
484 else if (ihigh > a->ob_size)
485 ihigh = a->ob_size;
486 item = a->ob_item;
487 d = n - (ihigh-ilow);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000488 if (ihigh > ilow)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000489 p = recycle = PyMem_NEW(PyObject *, (ihigh-ilow));
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000490 else
491 p = recycle = NULL;
492 if (d <= 0) { /* Delete -d items; recycle ihigh-ilow items */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000493 for (k = ilow; k < ihigh; k++)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000494 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000495 if (d < 0) {
496 for (/*k = ihigh*/; k < a->ob_size; k++)
497 item[k+d] = item[k];
498 a->ob_size += d;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000499 NRESIZE(item, PyObject *, a->ob_size); /* Can't fail */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000500 a->ob_item = item;
501 }
502 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000503 else { /* Insert d items; recycle ihigh-ilow items */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000504 NRESIZE(item, PyObject *, a->ob_size + d);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000505 if (item == NULL) {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000506 if (recycle != NULL)
507 PyMem_DEL(recycle);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000508 PyErr_NoMemory();
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000509 return -1;
510 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000511 for (k = a->ob_size; --k >= ihigh; )
512 item[k+d] = item[k];
513 for (/*k = ihigh-1*/; k >= ilow; --k)
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000514 *p++ = item[k];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000515 a->ob_item = item;
516 a->ob_size += d;
517 }
518 for (k = 0; k < n; k++, ilow++) {
Michael W. Hudson5da854f2002-11-05 17:38:05 +0000519 PyObject *w = PySequence_Fast_GET_ITEM(v_as_SF, k);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000520 Py_XINCREF(w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000521 item[ilow] = w;
522 }
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000523 if (recycle) {
524 while (--p >= recycle)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000525 Py_XDECREF(*p);
526 PyMem_DEL(recycle);
Guido van Rossumae7bf1a1995-01-17 10:21:11 +0000527 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000528 if (a->ob_size == 0 && a->ob_item != NULL) {
529 PyMem_FREE(a->ob_item);
530 a->ob_item = NULL;
531 }
Michael W. Hudson5da854f2002-11-05 17:38:05 +0000532 Py_XDECREF(v_as_SF);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000533 return 0;
534#undef b
535}
536
Guido van Rossum234f9421993-06-17 12:35:49 +0000537int
Fred Drakea2f55112000-07-09 15:16:51 +0000538PyList_SetSlice(PyObject *a, int ilow, int ihigh, PyObject *v)
Guido van Rossum234f9421993-06-17 12:35:49 +0000539{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000540 if (!PyList_Check(a)) {
541 PyErr_BadInternalCall();
Guido van Rossum1fc238a1993-07-29 08:25:09 +0000542 return -1;
Guido van Rossum234f9421993-06-17 12:35:49 +0000543 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000544 return list_ass_slice((PyListObject *)a, ilow, ihigh, v);
Guido van Rossum234f9421993-06-17 12:35:49 +0000545}
546
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000547static PyObject *
548list_inplace_repeat(PyListObject *self, int n)
549{
550 PyObject **items;
551 int size, i, j;
552
553
554 size = PyList_GET_SIZE(self);
555 if (size == 0) {
556 Py_INCREF(self);
557 return (PyObject *)self;
558 }
559
560 items = self->ob_item;
561
562 if (n < 1) {
563 self->ob_item = NULL;
564 self->ob_size = 0;
565 for (i = 0; i < size; i++)
566 Py_XDECREF(items[i]);
567 PyMem_DEL(items);
568 Py_INCREF(self);
569 return (PyObject *)self;
570 }
571
572 NRESIZE(items, PyObject*, size*n);
573 if (items == NULL) {
574 PyErr_NoMemory();
575 goto finally;
576 }
577 self->ob_item = items;
578 for (i = 1; i < n; i++) { /* Start counting at 1, not 0 */
579 for (j = 0; j < size; j++) {
580 PyObject *o = PyList_GET_ITEM(self, j);
581 Py_INCREF(o);
582 PyList_SET_ITEM(self, self->ob_size++, o);
583 }
584 }
585 Py_INCREF(self);
586 return (PyObject *)self;
587 finally:
588 return NULL;
589}
590
Guido van Rossum4a450d01991-04-03 19:05:18 +0000591static int
Fred Drakea2f55112000-07-09 15:16:51 +0000592list_ass_item(PyListObject *a, int i, PyObject *v)
Guido van Rossum4a450d01991-04-03 19:05:18 +0000593{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000594 PyObject *old_value;
Guido van Rossum4a450d01991-04-03 19:05:18 +0000595 if (i < 0 || i >= a->ob_size) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000596 PyErr_SetString(PyExc_IndexError,
597 "list assignment index out of range");
Guido van Rossum4a450d01991-04-03 19:05:18 +0000598 return -1;
599 }
600 if (v == NULL)
601 return list_ass_slice(a, i, i+1, v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000602 Py_INCREF(v);
Guido van Rossumd7047b31995-01-02 19:07:15 +0000603 old_value = a->ob_item[i];
Guido van Rossum4a450d01991-04-03 19:05:18 +0000604 a->ob_item[i] = v;
Tim Peters3b01a122002-07-19 02:35:45 +0000605 Py_DECREF(old_value);
Guido van Rossum4a450d01991-04-03 19:05:18 +0000606 return 0;
607}
608
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000609static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000610ins(PyListObject *self, int where, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000611{
612 if (ins1(self, where, v) != 0)
613 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000614 Py_INCREF(Py_None);
615 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000616}
617
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000618static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000619listinsert(PyListObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000620{
621 int i;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000622 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000623 if (!PyArg_ParseTuple(args, "iO:insert", &i, &v))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000624 return NULL;
Guido van Rossumbf80e541993-02-08 15:49:17 +0000625 return ins(self, i, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000626}
627
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000628static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000629listappend(PyListObject *self, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000630{
Guido van Rossumbf80e541993-02-08 15:49:17 +0000631 return ins(self, (int) self->ob_size, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000632}
633
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000634static int
635listextend_internal(PyListObject *self, PyObject *b)
Barry Warsawdedf6d61998-10-09 16:37:25 +0000636{
Barry Warsawdedf6d61998-10-09 16:37:25 +0000637 PyObject **items;
638 int selflen = PyList_GET_SIZE(self);
639 int blen;
640 register int i;
641
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000642 if (PyObject_Size(b) == 0) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000643 /* short circuit when b is empty */
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000644 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000645 return 0;
Jeremy Hyltondb60bb52001-01-03 22:32:16 +0000646 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000647
Barry Warsawdedf6d61998-10-09 16:37:25 +0000648 if (self == (PyListObject*)b) {
649 /* as in list_ass_slice() we must special case the
650 * situation: a.extend(a)
651 *
652 * XXX: I think this way ought to be faster than using
653 * list_slice() the way list_ass_slice() does.
654 */
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000655 Py_DECREF(b);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000656 b = PyList_New(selflen);
657 if (!b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000658 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000659 for (i = 0; i < selflen; i++) {
660 PyObject *o = PyList_GET_ITEM(self, i);
661 Py_INCREF(o);
662 PyList_SET_ITEM(b, i, o);
663 }
664 }
Barry Warsawdedf6d61998-10-09 16:37:25 +0000665
Jeremy Hylton03657cf2000-07-12 13:05:33 +0000666 blen = PyObject_Size(b);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000667
Barry Warsawdedf6d61998-10-09 16:37:25 +0000668 /* resize a using idiom */
669 items = self->ob_item;
670 NRESIZE(items, PyObject*, selflen + blen);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000671 if (items == NULL) {
Barry Warsawdedf6d61998-10-09 16:37:25 +0000672 PyErr_NoMemory();
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000673 Py_DECREF(b);
674 return -1;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000675 }
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000676
Barry Warsawdedf6d61998-10-09 16:37:25 +0000677 self->ob_item = items;
678
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000679 /* populate the end of self with b's items */
Barry Warsawdedf6d61998-10-09 16:37:25 +0000680 for (i = 0; i < blen; i++) {
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000681 PyObject *o = PySequence_Fast_GET_ITEM(b, i);
Barry Warsawdedf6d61998-10-09 16:37:25 +0000682 Py_INCREF(o);
683 PyList_SET_ITEM(self, self->ob_size++, o);
684 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +0000685 Py_DECREF(b);
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000686 return 0;
Barry Warsawdedf6d61998-10-09 16:37:25 +0000687}
688
689
690static PyObject *
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000691list_inplace_concat(PyListObject *self, PyObject *other)
692{
Tim Peters1af03e92001-05-26 19:37:54 +0000693 other = PySequence_Fast(other, "argument to += must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000694 if (!other)
695 return NULL;
696
697 if (listextend_internal(self, other) < 0)
698 return NULL;
699
700 Py_INCREF(self);
701 return (PyObject *)self;
702}
703
704static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000705listextend(PyListObject *self, PyObject *b)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000706{
707
Tim Peters1af03e92001-05-26 19:37:54 +0000708 b = PySequence_Fast(b, "list.extend() argument must be iterable");
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000709 if (!b)
710 return NULL;
711
712 if (listextend_internal(self, b) < 0)
713 return NULL;
714
715 Py_INCREF(Py_None);
716 return Py_None;
717}
718
719static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000720listpop(PyListObject *self, PyObject *args)
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000721{
722 int i = -1;
723 PyObject *v;
Guido van Rossumc00a9382000-02-24 21:48:29 +0000724 if (!PyArg_ParseTuple(args, "|i:pop", &i))
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +0000725 return NULL;
726 if (self->ob_size == 0) {
727 /* Special-case most common failure cause */
728 PyErr_SetString(PyExc_IndexError, "pop from empty list");
729 return NULL;
730 }
731 if (i < 0)
732 i += self->ob_size;
733 if (i < 0 || i >= self->ob_size) {
734 PyErr_SetString(PyExc_IndexError, "pop index out of range");
735 return NULL;
736 }
737 v = self->ob_item[i];
738 Py_INCREF(v);
739 if (list_ass_slice(self, i, i+1, (PyObject *)NULL) != 0) {
740 Py_DECREF(v);
741 return NULL;
742 }
743 return v;
744}
745
Tim Peters8e2e7ca2002-07-19 02:33:08 +0000746/* Reverse a slice of a list in place, from lo up to (exclusive) hi. */
747static void
748reverse_slice(PyObject **lo, PyObject **hi)
749{
750 assert(lo && hi);
751
752 --hi;
753 while (lo < hi) {
754 PyObject *t = *lo;
755 *lo = *hi;
756 *hi = t;
757 ++lo;
758 --hi;
759 }
760}
761
Tim Petersa64dc242002-08-01 02:13:36 +0000762/* Lots of code for an adaptive, stable, natural mergesort. There are many
763 * pieces to this algorithm; read listsort.txt for overviews and details.
764 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000765
Guido van Rossum3f236de1996-12-10 23:55:39 +0000766/* Comparison function. Takes care of calling a user-supplied
Tim Peters66860f62002-08-04 17:47:26 +0000767 * comparison function (any callable Python object), which must not be
768 * NULL (use the ISLT macro if you don't know, or call PyObject_RichCompareBool
769 * with Py_LT if you know it's NULL).
Tim Petersa64dc242002-08-01 02:13:36 +0000770 * Returns -1 on error, 1 if x < y, 0 if x >= y.
771 */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000772static int
Tim Petersa8c974c2002-07-19 03:30:57 +0000773islt(PyObject *x, PyObject *y, PyObject *compare)
Guido van Rossum3f236de1996-12-10 23:55:39 +0000774{
Tim Petersf2a04732002-07-11 21:46:16 +0000775 PyObject *res;
776 PyObject *args;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000777 int i;
778
Tim Peters66860f62002-08-04 17:47:26 +0000779 assert(compare != NULL);
Tim Petersa8c974c2002-07-19 03:30:57 +0000780 /* Call the user's comparison function and translate the 3-way
781 * result into true or false (or error).
782 */
Tim Petersf2a04732002-07-11 21:46:16 +0000783 args = PyTuple_New(2);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000784 if (args == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000785 return -1;
Tim Petersf2a04732002-07-11 21:46:16 +0000786 Py_INCREF(x);
787 Py_INCREF(y);
788 PyTuple_SET_ITEM(args, 0, x);
789 PyTuple_SET_ITEM(args, 1, y);
Tim Peters58cf3612002-07-15 05:16:13 +0000790 res = PyObject_Call(compare, args, NULL);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000791 Py_DECREF(args);
Guido van Rossum3f236de1996-12-10 23:55:39 +0000792 if (res == NULL)
Tim Petersa8c974c2002-07-19 03:30:57 +0000793 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000794 if (!PyInt_Check(res)) {
795 Py_DECREF(res);
796 PyErr_SetString(PyExc_TypeError,
Guido van Rossum9bcd1d71999-04-19 17:44:39 +0000797 "comparison function must return int");
Tim Petersa8c974c2002-07-19 03:30:57 +0000798 return -1;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000799 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000800 i = PyInt_AsLong(res);
801 Py_DECREF(res);
Tim Petersa8c974c2002-07-19 03:30:57 +0000802 return i < 0;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000803}
804
Tim Peters66860f62002-08-04 17:47:26 +0000805/* If COMPARE is NULL, calls PyObject_RichCompareBool with Py_LT, else calls
806 * islt. This avoids a layer of function call in the usual case, and
807 * sorting does many comparisons.
808 * Returns -1 on error, 1 if x < y, 0 if x >= y.
809 */
810#define ISLT(X, Y, COMPARE) ((COMPARE) == NULL ? \
811 PyObject_RichCompareBool(X, Y, Py_LT) : \
812 islt(X, Y, COMPARE))
813
814/* Compare X to Y via "<". Goto "fail" if the comparison raises an
Tim Petersa8c974c2002-07-19 03:30:57 +0000815 error. Else "k" is set to true iff X<Y, and an "if (k)" block is
816 started. It makes more sense in context <wink>. X and Y are PyObject*s.
817*/
Tim Peters66860f62002-08-04 17:47:26 +0000818#define IFLT(X, Y) if ((k = ISLT(X, Y, compare)) < 0) goto fail; \
Tim Petersa8c974c2002-07-19 03:30:57 +0000819 if (k)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000820
821/* binarysort is the best method for sorting small arrays: it does
822 few compares, but can do data movement quadratic in the number of
823 elements.
Guido van Rossum42812581998-06-17 14:15:44 +0000824 [lo, hi) is a contiguous slice of a list, and is sorted via
Tim Petersa8c974c2002-07-19 03:30:57 +0000825 binary insertion. This sort is stable.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000826 On entry, must have lo <= start <= hi, and that [lo, start) is already
827 sorted (pass start == lo if you don't know!).
Tim Petersa8c974c2002-07-19 03:30:57 +0000828 If islt() complains return -1, else 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000829 Even in case of error, the output slice will be some permutation of
830 the input (nothing is lost or duplicated).
831*/
Guido van Rossum3f236de1996-12-10 23:55:39 +0000832static int
Fred Drakea2f55112000-07-09 15:16:51 +0000833binarysort(PyObject **lo, PyObject **hi, PyObject **start, PyObject *compare)
834 /* compare -- comparison function object, or NULL for default */
Guido van Rossum3f236de1996-12-10 23:55:39 +0000835{
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000836 register int k;
837 register PyObject **l, **p, **r;
838 register PyObject *pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000839
Tim Petersa8c974c2002-07-19 03:30:57 +0000840 assert(lo <= start && start <= hi);
841 /* assert [lo, start) is sorted */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000842 if (lo == start)
843 ++start;
844 for (; start < hi; ++start) {
845 /* set l to where *start belongs */
846 l = lo;
847 r = start;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000848 pivot = *r;
Tim Peters0fe977c2002-07-19 06:12:32 +0000849 /* Invariants:
850 * pivot >= all in [lo, l).
851 * pivot < all in [r, start).
852 * The second is vacuously true at the start.
853 */
854 assert(l < r);
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000855 do {
856 p = l + ((r - l) >> 1);
Tim Petersa8c974c2002-07-19 03:30:57 +0000857 IFLT(pivot, *p)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000858 r = p;
859 else
Tim Peters0fe977c2002-07-19 06:12:32 +0000860 l = p+1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000861 } while (l < r);
Tim Peters0fe977c2002-07-19 06:12:32 +0000862 assert(l == r);
863 /* The invariants still hold, so pivot >= all in [lo, l) and
864 pivot < all in [l, start), so pivot belongs at l. Note
865 that if there are elements equal to pivot, l points to the
866 first slot after them -- that's why this sort is stable.
867 Slide over to make room.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000868 Caution: using memmove is much slower under MSVC 5;
869 we're not usually moving many slots. */
870 for (p = start; p > l; --p)
871 *p = *(p-1);
872 *l = pivot;
Guido van Rossum3f236de1996-12-10 23:55:39 +0000873 }
Guido van Rossum3f236de1996-12-10 23:55:39 +0000874 return 0;
Guido van Rossuma119c0d1998-05-29 17:56:32 +0000875
876 fail:
877 return -1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000878}
879
Tim Petersa64dc242002-08-01 02:13:36 +0000880/*
881Return the length of the run beginning at lo, in the slice [lo, hi). lo < hi
882is required on entry. "A run" is the longest ascending sequence, with
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000883
Tim Petersa64dc242002-08-01 02:13:36 +0000884 lo[0] <= lo[1] <= lo[2] <= ...
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000885
Tim Petersa64dc242002-08-01 02:13:36 +0000886or the longest descending sequence, with
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000887
Tim Petersa64dc242002-08-01 02:13:36 +0000888 lo[0] > lo[1] > lo[2] > ...
Tim Peters3b01a122002-07-19 02:35:45 +0000889
Tim Petersa64dc242002-08-01 02:13:36 +0000890Boolean *descending is set to 0 in the former case, or to 1 in the latter.
891For its intended use in a stable mergesort, the strictness of the defn of
892"descending" is needed so that the caller can safely reverse a descending
893sequence without violating stability (strict > ensures there are no equal
894elements to get out of order).
895
896Returns -1 in case of error.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000897*/
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000898static int
Tim Petersa64dc242002-08-01 02:13:36 +0000899count_run(PyObject **lo, PyObject **hi, PyObject *compare, int *descending)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000900{
Tim Petersa64dc242002-08-01 02:13:36 +0000901 int k;
902 int n;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000903
Tim Petersa64dc242002-08-01 02:13:36 +0000904 assert(lo < hi);
905 *descending = 0;
906 ++lo;
907 if (lo == hi)
908 return 1;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000909
Tim Petersa64dc242002-08-01 02:13:36 +0000910 n = 2;
911 IFLT(*lo, *(lo-1)) {
912 *descending = 1;
913 for (lo = lo+1; lo < hi; ++lo, ++n) {
914 IFLT(*lo, *(lo-1))
915 ;
916 else
917 break;
918 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000919 }
Tim Petersa64dc242002-08-01 02:13:36 +0000920 else {
921 for (lo = lo+1; lo < hi; ++lo, ++n) {
922 IFLT(*lo, *(lo-1))
923 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000924 }
925 }
926
Tim Petersa64dc242002-08-01 02:13:36 +0000927 return n;
928fail:
929 return -1;
930}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000931
Tim Petersa64dc242002-08-01 02:13:36 +0000932/*
933Locate the proper position of key in a sorted vector; if the vector contains
934an element equal to key, return the position immediately to the left of
935the leftmost equal element. [gallop_right() does the same except returns
936the position to the right of the rightmost equal element (if any).]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000937
Tim Petersa64dc242002-08-01 02:13:36 +0000938"a" is a sorted vector with n elements, starting at a[0]. n must be > 0.
Guido van Rossum4c4e7df1998-06-16 15:18:28 +0000939
Tim Petersa64dc242002-08-01 02:13:36 +0000940"hint" is an index at which to begin the search, 0 <= hint < n. The closer
941hint is to the final result, the faster this runs.
942
943The return value is the int k in 0..n such that
944
945 a[k-1] < key <= a[k]
946
947pretending that *(a-1) is minus infinity and a[n] is plus infinity. IOW,
948key belongs at index k; or, IOW, the first k elements of a should precede
949key, and the last n-k should follow key.
950
951Returns -1 on error. See listsort.txt for info on the method.
952*/
953static int
954gallop_left(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
955{
956 int ofs;
957 int lastofs;
958 int k;
959
960 assert(key && a && n > 0 && hint >= 0 && hint < n);
961
962 a += hint;
963 lastofs = 0;
964 ofs = 1;
965 IFLT(*a, key) {
966 /* a[hint] < key -- gallop right, until
967 * a[hint + lastofs] < key <= a[hint + ofs]
968 */
969 const int maxofs = n - hint; /* &a[n-1] is highest */
970 while (ofs < maxofs) {
971 IFLT(a[ofs], key) {
972 lastofs = ofs;
973 ofs = (ofs << 1) + 1;
974 if (ofs <= 0) /* int overflow */
975 ofs = maxofs;
976 }
977 else /* key <= a[hint + ofs] */
978 break;
979 }
980 if (ofs > maxofs)
981 ofs = maxofs;
982 /* Translate back to offsets relative to &a[0]. */
983 lastofs += hint;
984 ofs += hint;
985 }
986 else {
987 /* key <= a[hint] -- gallop left, until
988 * a[hint - ofs] < key <= a[hint - lastofs]
989 */
990 const int maxofs = hint + 1; /* &a[0] is lowest */
991 while (ofs < maxofs) {
992 IFLT(*(a-ofs), key)
993 break;
994 /* key <= a[hint - ofs] */
995 lastofs = ofs;
996 ofs = (ofs << 1) + 1;
997 if (ofs <= 0) /* int overflow */
998 ofs = maxofs;
999 }
1000 if (ofs > maxofs)
1001 ofs = maxofs;
1002 /* Translate back to positive offsets relative to &a[0]. */
1003 k = lastofs;
1004 lastofs = hint - ofs;
1005 ofs = hint - k;
1006 }
1007 a -= hint;
1008
1009 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1010 /* Now a[lastofs] < key <= a[ofs], so key belongs somewhere to the
1011 * right of lastofs but no farther right than ofs. Do a binary
1012 * search, with invariant a[lastofs-1] < key <= a[ofs].
1013 */
1014 ++lastofs;
1015 while (lastofs < ofs) {
1016 int m = lastofs + ((ofs - lastofs) >> 1);
1017
1018 IFLT(a[m], key)
1019 lastofs = m+1; /* a[m] < key */
1020 else
1021 ofs = m; /* key <= a[m] */
1022 }
1023 assert(lastofs == ofs); /* so a[ofs-1] < key <= a[ofs] */
1024 return ofs;
1025
1026fail:
1027 return -1;
1028}
1029
1030/*
1031Exactly like gallop_left(), except that if key already exists in a[0:n],
1032finds the position immediately to the right of the rightmost equal value.
1033
1034The return value is the int k in 0..n such that
1035
1036 a[k-1] <= key < a[k]
1037
1038or -1 if error.
1039
1040The code duplication is massive, but this is enough different given that
1041we're sticking to "<" comparisons that it's much harder to follow if
1042written as one routine with yet another "left or right?" flag.
1043*/
1044static int
1045gallop_right(PyObject *key, PyObject **a, int n, int hint, PyObject *compare)
1046{
1047 int ofs;
1048 int lastofs;
1049 int k;
1050
1051 assert(key && a && n > 0 && hint >= 0 && hint < n);
1052
1053 a += hint;
1054 lastofs = 0;
1055 ofs = 1;
1056 IFLT(key, *a) {
1057 /* key < a[hint] -- gallop left, until
1058 * a[hint - ofs] <= key < a[hint - lastofs]
1059 */
1060 const int maxofs = hint + 1; /* &a[0] is lowest */
1061 while (ofs < maxofs) {
1062 IFLT(key, *(a-ofs)) {
1063 lastofs = ofs;
1064 ofs = (ofs << 1) + 1;
1065 if (ofs <= 0) /* int overflow */
1066 ofs = maxofs;
1067 }
1068 else /* a[hint - ofs] <= key */
1069 break;
1070 }
1071 if (ofs > maxofs)
1072 ofs = maxofs;
1073 /* Translate back to positive offsets relative to &a[0]. */
1074 k = lastofs;
1075 lastofs = hint - ofs;
1076 ofs = hint - k;
1077 }
1078 else {
1079 /* a[hint] <= key -- gallop right, until
1080 * a[hint + lastofs] <= key < a[hint + ofs]
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001081 */
Tim Petersa64dc242002-08-01 02:13:36 +00001082 const int maxofs = n - hint; /* &a[n-1] is highest */
1083 while (ofs < maxofs) {
1084 IFLT(key, a[ofs])
1085 break;
1086 /* a[hint + ofs] <= key */
1087 lastofs = ofs;
1088 ofs = (ofs << 1) + 1;
1089 if (ofs <= 0) /* int overflow */
1090 ofs = maxofs;
1091 }
1092 if (ofs > maxofs)
1093 ofs = maxofs;
1094 /* Translate back to offsets relative to &a[0]. */
1095 lastofs += hint;
1096 ofs += hint;
1097 }
1098 a -= hint;
1099
1100 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1101 /* Now a[lastofs] <= key < a[ofs], so key belongs somewhere to the
1102 * right of lastofs but no farther right than ofs. Do a binary
1103 * search, with invariant a[lastofs-1] <= key < a[ofs].
1104 */
1105 ++lastofs;
1106 while (lastofs < ofs) {
1107 int m = lastofs + ((ofs - lastofs) >> 1);
1108
1109 IFLT(key, a[m])
1110 ofs = m; /* key < a[m] */
1111 else
1112 lastofs = m+1; /* a[m] <= key */
1113 }
1114 assert(lastofs == ofs); /* so a[ofs-1] <= key < a[ofs] */
1115 return ofs;
1116
1117fail:
1118 return -1;
1119}
1120
1121/* The maximum number of entries in a MergeState's pending-runs stack.
1122 * This is enough to sort arrays of size up to about
1123 * 32 * phi ** MAX_MERGE_PENDING
1124 * where phi ~= 1.618. 85 is ridiculouslylarge enough, good for an array
1125 * with 2**64 elements.
1126 */
1127#define MAX_MERGE_PENDING 85
1128
Tim Peterse05f65a2002-08-10 05:21:15 +00001129/* When we get into galloping mode, we stay there until both runs win less
1130 * often than MIN_GALLOP consecutive times. See listsort.txt for more info.
Tim Petersa64dc242002-08-01 02:13:36 +00001131 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001132#define MIN_GALLOP 7
Tim Petersa64dc242002-08-01 02:13:36 +00001133
1134/* Avoid malloc for small temp arrays. */
1135#define MERGESTATE_TEMP_SIZE 256
1136
1137/* One MergeState exists on the stack per invocation of mergesort. It's just
1138 * a convenient way to pass state around among the helper functions.
1139 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001140struct s_slice {
1141 PyObject **base;
1142 int len;
1143};
1144
Tim Petersa64dc242002-08-01 02:13:36 +00001145typedef struct s_MergeState {
1146 /* The user-supplied comparison function. or NULL if none given. */
1147 PyObject *compare;
1148
Tim Peterse05f65a2002-08-10 05:21:15 +00001149 /* This controls when we get *into* galloping mode. It's initialized
1150 * to MIN_GALLOP. merge_lo and merge_hi tend to nudge it higher for
1151 * random data, and lower for highly structured data.
1152 */
1153 int min_gallop;
1154
Tim Petersa64dc242002-08-01 02:13:36 +00001155 /* 'a' is temp storage to help with merges. It contains room for
1156 * alloced entries.
1157 */
1158 PyObject **a; /* may point to temparray below */
1159 int alloced;
1160
1161 /* A stack of n pending runs yet to be merged. Run #i starts at
1162 * address base[i] and extends for len[i] elements. It's always
1163 * true (so long as the indices are in bounds) that
1164 *
Tim Peterse05f65a2002-08-10 05:21:15 +00001165 * pending[i].base + pending[i].len == pending[i+1].base
Tim Petersa64dc242002-08-01 02:13:36 +00001166 *
1167 * so we could cut the storage for this, but it's a minor amount,
1168 * and keeping all the info explicit simplifies the code.
1169 */
1170 int n;
Tim Peterse05f65a2002-08-10 05:21:15 +00001171 struct s_slice pending[MAX_MERGE_PENDING];
Tim Petersa64dc242002-08-01 02:13:36 +00001172
1173 /* 'a' points to this when possible, rather than muck with malloc. */
1174 PyObject *temparray[MERGESTATE_TEMP_SIZE];
1175} MergeState;
1176
1177/* Conceptually a MergeState's constructor. */
1178static void
1179merge_init(MergeState *ms, PyObject *compare)
1180{
1181 assert(ms != NULL);
1182 ms->compare = compare;
1183 ms->a = ms->temparray;
1184 ms->alloced = MERGESTATE_TEMP_SIZE;
1185 ms->n = 0;
Tim Peterse05f65a2002-08-10 05:21:15 +00001186 ms->min_gallop = MIN_GALLOP;
Tim Petersa64dc242002-08-01 02:13:36 +00001187}
1188
1189/* Free all the temp memory owned by the MergeState. This must be called
1190 * when you're done with a MergeState, and may be called before then if
1191 * you want to free the temp memory early.
1192 */
1193static void
1194merge_freemem(MergeState *ms)
1195{
1196 assert(ms != NULL);
1197 if (ms->a != ms->temparray)
1198 PyMem_Free(ms->a);
1199 ms->a = ms->temparray;
1200 ms->alloced = MERGESTATE_TEMP_SIZE;
1201}
1202
1203/* Ensure enough temp memory for 'need' array slots is available.
1204 * Returns 0 on success and -1 if the memory can't be gotten.
1205 */
1206static int
1207merge_getmem(MergeState *ms, int need)
1208{
1209 assert(ms != NULL);
1210 if (need <= ms->alloced)
1211 return 0;
1212 /* Don't realloc! That can cost cycles to copy the old data, but
1213 * we don't care what's in the block.
1214 */
1215 merge_freemem(ms);
1216 ms->a = (PyObject **)PyMem_Malloc(need * sizeof(PyObject*));
1217 if (ms->a) {
1218 ms->alloced = need;
1219 return 0;
1220 }
1221 PyErr_NoMemory();
1222 merge_freemem(ms); /* reset to sane state */
1223 return -1;
1224}
1225#define MERGE_GETMEM(MS, NEED) ((NEED) <= (MS)->alloced ? 0 : \
1226 merge_getmem(MS, NEED))
1227
1228/* Merge the na elements starting at pa with the nb elements starting at pb
1229 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1230 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1231 * merge, and should have na <= nb. See listsort.txt for more info.
1232 * Return 0 if successful, -1 if error.
1233 */
1234static int
1235merge_lo(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1236{
1237 int k;
1238 PyObject *compare;
1239 PyObject **dest;
1240 int result = -1; /* guilty until proved innocent */
Tim Peterse05f65a2002-08-10 05:21:15 +00001241 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001242
1243 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1244 if (MERGE_GETMEM(ms, na) < 0)
1245 return -1;
1246 memcpy(ms->a, pa, na * sizeof(PyObject*));
1247 dest = pa;
1248 pa = ms->a;
1249
1250 *dest++ = *pb++;
1251 --nb;
1252 if (nb == 0)
1253 goto Succeed;
1254 if (na == 1)
1255 goto CopyB;
1256
1257 compare = ms->compare;
1258 for (;;) {
1259 int acount = 0; /* # of times A won in a row */
1260 int bcount = 0; /* # of times B won in a row */
1261
1262 /* Do the straightforward thing until (if ever) one run
1263 * appears to win consistently.
1264 */
1265 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001266 assert(na > 1 && nb > 0);
Tim Peters66860f62002-08-04 17:47:26 +00001267 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001268 if (k) {
1269 if (k < 0)
1270 goto Fail;
1271 *dest++ = *pb++;
1272 ++bcount;
1273 acount = 0;
1274 --nb;
1275 if (nb == 0)
1276 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001277 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001278 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001279 }
1280 else {
Tim Petersa64dc242002-08-01 02:13:36 +00001281 *dest++ = *pa++;
1282 ++acount;
1283 bcount = 0;
1284 --na;
1285 if (na == 1)
1286 goto CopyB;
Tim Peterse05f65a2002-08-10 05:21:15 +00001287 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001288 break;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001289 }
Tim Petersa64dc242002-08-01 02:13:36 +00001290 }
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001291
Tim Petersa64dc242002-08-01 02:13:36 +00001292 /* One run is winning so consistently that galloping may
1293 * be a huge win. So try that, and continue galloping until
1294 * (if ever) neither run appears to be winning consistently
1295 * anymore.
1296 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001297 ++min_gallop;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001298 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001299 assert(na > 1 && nb > 0);
1300 min_gallop -= min_gallop > 1;
1301 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001302 k = gallop_right(*pb, pa, na, 0, compare);
1303 acount = k;
1304 if (k) {
1305 if (k < 0)
1306 goto Fail;
1307 memcpy(dest, pa, k * sizeof(PyObject *));
1308 dest += k;
1309 pa += k;
1310 na -= k;
1311 if (na == 1)
1312 goto CopyB;
1313 /* na==0 is impossible now if the comparison
1314 * function is consistent, but we can't assume
1315 * that it is.
1316 */
1317 if (na == 0)
1318 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001319 }
Tim Petersa64dc242002-08-01 02:13:36 +00001320 *dest++ = *pb++;
1321 --nb;
1322 if (nb == 0)
1323 goto Succeed;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001324
Tim Petersa64dc242002-08-01 02:13:36 +00001325 k = gallop_left(*pa, pb, nb, 0, compare);
1326 bcount = k;
1327 if (k) {
1328 if (k < 0)
1329 goto Fail;
1330 memmove(dest, pb, k * sizeof(PyObject *));
1331 dest += k;
1332 pb += k;
1333 nb -= k;
1334 if (nb == 0)
1335 goto Succeed;
1336 }
1337 *dest++ = *pa++;
1338 --na;
1339 if (na == 1)
1340 goto CopyB;
1341 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001342 ++min_gallop; /* penalize it for leaving galloping mode */
1343 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001344 }
1345Succeed:
1346 result = 0;
1347Fail:
1348 if (na)
1349 memcpy(dest, pa, na * sizeof(PyObject*));
1350 return result;
1351CopyB:
1352 assert(na == 1 && nb > 0);
1353 /* The last element of pa belongs at the end of the merge. */
1354 memmove(dest, pb, nb * sizeof(PyObject *));
1355 dest[nb] = *pa;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001356 return 0;
Tim Petersa64dc242002-08-01 02:13:36 +00001357}
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001358
Tim Petersa64dc242002-08-01 02:13:36 +00001359/* Merge the na elements starting at pa with the nb elements starting at pb
1360 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1361 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1362 * merge, and should have na >= nb. See listsort.txt for more info.
1363 * Return 0 if successful, -1 if error.
1364 */
1365static int
1366merge_hi(MergeState *ms, PyObject **pa, int na, PyObject **pb, int nb)
1367{
1368 int k;
1369 PyObject *compare;
1370 PyObject **dest;
1371 int result = -1; /* guilty until proved innocent */
1372 PyObject **basea;
1373 PyObject **baseb;
Tim Peterse05f65a2002-08-10 05:21:15 +00001374 int min_gallop = ms->min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001375
1376 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1377 if (MERGE_GETMEM(ms, nb) < 0)
1378 return -1;
1379 dest = pb + nb - 1;
1380 memcpy(ms->a, pb, nb * sizeof(PyObject*));
1381 basea = pa;
1382 baseb = ms->a;
1383 pb = ms->a + nb - 1;
1384 pa += na - 1;
1385
1386 *dest-- = *pa--;
1387 --na;
1388 if (na == 0)
1389 goto Succeed;
1390 if (nb == 1)
1391 goto CopyA;
1392
1393 compare = ms->compare;
1394 for (;;) {
1395 int acount = 0; /* # of times A won in a row */
1396 int bcount = 0; /* # of times B won in a row */
1397
1398 /* Do the straightforward thing until (if ever) one run
1399 * appears to win consistently.
1400 */
1401 for (;;) {
Tim Peterse05f65a2002-08-10 05:21:15 +00001402 assert(na > 0 && nb > 1);
Tim Peters66860f62002-08-04 17:47:26 +00001403 k = ISLT(*pb, *pa, compare);
Tim Petersa64dc242002-08-01 02:13:36 +00001404 if (k) {
1405 if (k < 0)
1406 goto Fail;
1407 *dest-- = *pa--;
1408 ++acount;
1409 bcount = 0;
1410 --na;
1411 if (na == 0)
1412 goto Succeed;
Tim Peterse05f65a2002-08-10 05:21:15 +00001413 if (acount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001414 break;
1415 }
1416 else {
1417 *dest-- = *pb--;
1418 ++bcount;
1419 acount = 0;
1420 --nb;
1421 if (nb == 1)
1422 goto CopyA;
Tim Peterse05f65a2002-08-10 05:21:15 +00001423 if (bcount >= min_gallop)
Tim Petersa64dc242002-08-01 02:13:36 +00001424 break;
1425 }
1426 }
1427
1428 /* One run is winning so consistently that galloping may
1429 * be a huge win. So try that, and continue galloping until
1430 * (if ever) neither run appears to be winning consistently
1431 * anymore.
1432 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001433 ++min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001434 do {
Tim Peterse05f65a2002-08-10 05:21:15 +00001435 assert(na > 0 && nb > 1);
1436 min_gallop -= min_gallop > 1;
1437 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001438 k = gallop_right(*pb, basea, na, na-1, compare);
1439 if (k < 0)
1440 goto Fail;
1441 k = na - k;
1442 acount = k;
1443 if (k) {
1444 dest -= k;
1445 pa -= k;
1446 memmove(dest+1, pa+1, k * sizeof(PyObject *));
1447 na -= k;
1448 if (na == 0)
1449 goto Succeed;
1450 }
1451 *dest-- = *pb--;
1452 --nb;
1453 if (nb == 1)
1454 goto CopyA;
1455
1456 k = gallop_left(*pa, baseb, nb, nb-1, compare);
1457 if (k < 0)
1458 goto Fail;
1459 k = nb - k;
1460 bcount = k;
1461 if (k) {
1462 dest -= k;
1463 pb -= k;
1464 memcpy(dest+1, pb+1, k * sizeof(PyObject *));
1465 nb -= k;
1466 if (nb == 1)
1467 goto CopyA;
1468 /* nb==0 is impossible now if the comparison
1469 * function is consistent, but we can't assume
1470 * that it is.
1471 */
1472 if (nb == 0)
1473 goto Succeed;
1474 }
1475 *dest-- = *pa--;
1476 --na;
1477 if (na == 0)
1478 goto Succeed;
1479 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
Tim Peterse05f65a2002-08-10 05:21:15 +00001480 ++min_gallop; /* penalize it for leaving galloping mode */
1481 ms->min_gallop = min_gallop;
Tim Petersa64dc242002-08-01 02:13:36 +00001482 }
1483Succeed:
1484 result = 0;
1485Fail:
1486 if (nb)
1487 memcpy(dest-(nb-1), baseb, nb * sizeof(PyObject*));
1488 return result;
1489CopyA:
1490 assert(nb == 1 && na > 0);
1491 /* The first element of pb belongs at the front of the merge. */
1492 dest -= na;
1493 pa -= na;
1494 memmove(dest+1, pa+1, na * sizeof(PyObject *));
1495 *dest = *pb;
1496 return 0;
1497}
1498
1499/* Merge the two runs at stack indices i and i+1.
1500 * Returns 0 on success, -1 on error.
1501 */
1502static int
1503merge_at(MergeState *ms, int i)
1504{
1505 PyObject **pa, **pb;
1506 int na, nb;
1507 int k;
1508 PyObject *compare;
1509
1510 assert(ms != NULL);
1511 assert(ms->n >= 2);
1512 assert(i >= 0);
1513 assert(i == ms->n - 2 || i == ms->n - 3);
1514
Tim Peterse05f65a2002-08-10 05:21:15 +00001515 pa = ms->pending[i].base;
1516 na = ms->pending[i].len;
1517 pb = ms->pending[i+1].base;
1518 nb = ms->pending[i+1].len;
Tim Petersa64dc242002-08-01 02:13:36 +00001519 assert(na > 0 && nb > 0);
1520 assert(pa + na == pb);
1521
1522 /* Record the length of the combined runs; if i is the 3rd-last
1523 * run now, also slide over the last run (which isn't involved
1524 * in this merge). The current run i+1 goes away in any case.
1525 */
Tim Peterse05f65a2002-08-10 05:21:15 +00001526 ms->pending[i].len = na + nb;
1527 if (i == ms->n - 3)
1528 ms->pending[i+1] = ms->pending[i+2];
Tim Petersa64dc242002-08-01 02:13:36 +00001529 --ms->n;
1530
1531 /* Where does b start in a? Elements in a before that can be
1532 * ignored (already in place).
1533 */
1534 compare = ms->compare;
1535 k = gallop_right(*pb, pa, na, 0, compare);
1536 if (k < 0)
1537 return -1;
1538 pa += k;
1539 na -= k;
1540 if (na == 0)
1541 return 0;
1542
1543 /* Where does a end in b? Elements in b after that can be
1544 * ignored (already in place).
1545 */
1546 nb = gallop_left(pa[na-1], pb, nb, nb-1, compare);
1547 if (nb <= 0)
1548 return nb;
1549
1550 /* Merge what remains of the runs, using a temp array with
1551 * min(na, nb) elements.
1552 */
1553 if (na <= nb)
1554 return merge_lo(ms, pa, na, pb, nb);
1555 else
1556 return merge_hi(ms, pa, na, pb, nb);
1557}
1558
1559/* Examine the stack of runs waiting to be merged, merging adjacent runs
1560 * until the stack invariants are re-established:
1561 *
1562 * 1. len[-3] > len[-2] + len[-1]
1563 * 2. len[-2] > len[-1]
1564 *
1565 * See listsort.txt for more info.
1566 *
1567 * Returns 0 on success, -1 on error.
1568 */
1569static int
1570merge_collapse(MergeState *ms)
1571{
Tim Peterse05f65a2002-08-10 05:21:15 +00001572 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001573
1574 assert(ms);
1575 while (ms->n > 1) {
1576 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001577 if (n > 0 && p[n-1].len <= p[n].len + p[n+1].len) {
1578 if (p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001579 --n;
1580 if (merge_at(ms, n) < 0)
1581 return -1;
1582 }
Tim Peterse05f65a2002-08-10 05:21:15 +00001583 else if (p[n].len <= p[n+1].len) {
Tim Petersa64dc242002-08-01 02:13:36 +00001584 if (merge_at(ms, n) < 0)
1585 return -1;
1586 }
1587 else
1588 break;
1589 }
1590 return 0;
1591}
1592
1593/* Regardless of invariants, merge all runs on the stack until only one
1594 * remains. This is used at the end of the mergesort.
1595 *
1596 * Returns 0 on success, -1 on error.
1597 */
1598static int
1599merge_force_collapse(MergeState *ms)
1600{
Tim Peterse05f65a2002-08-10 05:21:15 +00001601 struct s_slice *p = ms->pending;
Tim Petersa64dc242002-08-01 02:13:36 +00001602
1603 assert(ms);
1604 while (ms->n > 1) {
1605 int n = ms->n - 2;
Tim Peterse05f65a2002-08-10 05:21:15 +00001606 if (n > 0 && p[n-1].len < p[n+1].len)
Tim Petersa64dc242002-08-01 02:13:36 +00001607 --n;
1608 if (merge_at(ms, n) < 0)
1609 return -1;
1610 }
1611 return 0;
1612}
1613
1614/* Compute a good value for the minimum run length; natural runs shorter
1615 * than this are boosted artificially via binary insertion.
1616 *
1617 * If n < 64, return n (it's too small to bother with fancy stuff).
1618 * Else if n is an exact power of 2, return 32.
1619 * Else return an int k, 32 <= k <= 64, such that n/k is close to, but
1620 * strictly less than, an exact power of 2.
1621 *
1622 * See listsort.txt for more info.
1623 */
1624static int
1625merge_compute_minrun(int n)
1626{
1627 int r = 0; /* becomes 1 if any 1 bits are shifted off */
1628
1629 assert(n >= 0);
1630 while (n >= 64) {
1631 r |= n & 1;
1632 n >>= 1;
1633 }
1634 return n + r;
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001635}
Guido van Rossuma119c0d1998-05-29 17:56:32 +00001636
Tim Petersa64dc242002-08-01 02:13:36 +00001637/* An adaptive, stable, natural mergesort. See listsort.txt.
1638 * Returns Py_None on success, NULL on error. Even in case of error, the
1639 * list will be some permutation of its input state (nothing is lost or
1640 * duplicated).
1641 */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001642static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001643listsort(PyListObject *self, PyObject *args)
Guido van Rossum3f236de1996-12-10 23:55:39 +00001644{
Tim Petersa64dc242002-08-01 02:13:36 +00001645 MergeState ms;
1646 PyObject **lo, **hi;
1647 int nremaining;
1648 int minrun;
Tim Petersb9099c32002-11-12 22:08:10 +00001649 int saved_ob_size;
1650 PyObject **saved_ob_item;
1651 PyObject **empty_ob_item;
Tim Petersa64dc242002-08-01 02:13:36 +00001652 PyObject *compare = NULL;
1653 PyObject *result = NULL; /* guilty until proved innocent */
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001654
Tim Petersa64dc242002-08-01 02:13:36 +00001655 assert(self != NULL);
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001656 if (args != NULL) {
Tim Peters6bdbc9e2002-08-03 02:28:24 +00001657 if (!PyArg_ParseTuple(args, "|O:sort", &compare))
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001658 return NULL;
1659 }
Tim Petersa64dc242002-08-01 02:13:36 +00001660 merge_init(&ms, compare);
Tim Peters330f9e92002-07-19 07:05:44 +00001661
Tim Petersb9099c32002-11-12 22:08:10 +00001662 /* The list is temporarily made empty, so that mutations performed
1663 * by comparison functions can't affect the slice of memory we're
1664 * sorting (allowing mutations during sorting is a core-dump
1665 * factory, since ob_item may change).
1666 */
1667 saved_ob_size = self->ob_size;
1668 saved_ob_item = self->ob_item;
1669 self->ob_size = 0;
1670 self->ob_item = empty_ob_item = PyMem_NEW(PyObject *, 0);
Tim Peters330f9e92002-07-19 07:05:44 +00001671
Tim Petersb9099c32002-11-12 22:08:10 +00001672 nremaining = saved_ob_size;
Tim Petersa64dc242002-08-01 02:13:36 +00001673 if (nremaining < 2)
1674 goto succeed;
Tim Peters330f9e92002-07-19 07:05:44 +00001675
Tim Petersa64dc242002-08-01 02:13:36 +00001676 /* March over the array once, left to right, finding natural runs,
1677 * and extending short natural runs to minrun elements.
1678 */
Tim Petersb9099c32002-11-12 22:08:10 +00001679 lo = saved_ob_item;
Tim Petersa64dc242002-08-01 02:13:36 +00001680 hi = lo + nremaining;
1681 minrun = merge_compute_minrun(nremaining);
1682 do {
1683 int descending;
1684 int n;
Tim Peters330f9e92002-07-19 07:05:44 +00001685
Tim Petersa64dc242002-08-01 02:13:36 +00001686 /* Identify next run. */
1687 n = count_run(lo, hi, compare, &descending);
1688 if (n < 0)
1689 goto fail;
1690 if (descending)
1691 reverse_slice(lo, lo + n);
1692 /* If short, extend to min(minrun, nremaining). */
1693 if (n < minrun) {
1694 const int force = nremaining <= minrun ?
1695 nremaining : minrun;
1696 if (binarysort(lo, lo + force, lo + n, compare) < 0)
1697 goto fail;
1698 n = force;
1699 }
1700 /* Push run onto pending-runs stack, and maybe merge. */
1701 assert(ms.n < MAX_MERGE_PENDING);
Tim Peterse05f65a2002-08-10 05:21:15 +00001702 ms.pending[ms.n].base = lo;
1703 ms.pending[ms.n].len = n;
Tim Petersa64dc242002-08-01 02:13:36 +00001704 ++ms.n;
1705 if (merge_collapse(&ms) < 0)
1706 goto fail;
1707 /* Advance to find next run. */
1708 lo += n;
1709 nremaining -= n;
1710 } while (nremaining);
1711 assert(lo == hi);
Tim Peters330f9e92002-07-19 07:05:44 +00001712
Tim Petersa64dc242002-08-01 02:13:36 +00001713 if (merge_force_collapse(&ms) < 0)
1714 goto fail;
1715 assert(ms.n == 1);
Tim Petersb9099c32002-11-12 22:08:10 +00001716 assert(ms.pending[0].base == saved_ob_item);
1717 assert(ms.pending[0].len == saved_ob_size);
Tim Petersa64dc242002-08-01 02:13:36 +00001718
1719succeed:
1720 result = Py_None;
Tim Peters330f9e92002-07-19 07:05:44 +00001721fail:
Tim Petersb9099c32002-11-12 22:08:10 +00001722 if (self->ob_item != empty_ob_item || self->ob_size) {
1723 /* The user mucked with the list during the sort. */
1724 (void)list_ass_slice(self, 0, self->ob_size, (PyObject *)NULL);
1725 if (result != NULL) {
1726 PyErr_SetString(PyExc_ValueError,
1727 "list modified during sort");
1728 result = NULL;
1729 }
1730 }
1731 if (self->ob_item == empty_ob_item)
1732 PyMem_FREE(empty_ob_item);
1733 self->ob_size = saved_ob_size;
1734 self->ob_item = saved_ob_item;
Tim Petersa64dc242002-08-01 02:13:36 +00001735 merge_freemem(&ms);
1736 Py_XINCREF(result);
1737 return result;
Guido van Rossum3f236de1996-12-10 23:55:39 +00001738}
Tim Peters330f9e92002-07-19 07:05:44 +00001739#undef IFLT
Tim Peters66860f62002-08-04 17:47:26 +00001740#undef ISLT
Tim Peters330f9e92002-07-19 07:05:44 +00001741
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001742int
Fred Drakea2f55112000-07-09 15:16:51 +00001743PyList_Sort(PyObject *v)
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00001744{
1745 if (v == NULL || !PyList_Check(v)) {
1746 PyErr_BadInternalCall();
1747 return -1;
1748 }
1749 v = listsort((PyListObject *)v, (PyObject *)NULL);
1750 if (v == NULL)
1751 return -1;
1752 Py_DECREF(v);
1753 return 0;
1754}
1755
Guido van Rossumb86c5492001-02-12 22:06:02 +00001756static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001757listreverse(PyListObject *self)
Guido van Rossumb86c5492001-02-12 22:06:02 +00001758{
Tim Peters326b4482002-07-19 04:04:16 +00001759 if (self->ob_size > 1)
1760 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001761 Py_INCREF(Py_None);
1762 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001763}
1764
Guido van Rossum84c76f51990-10-30 13:32:20 +00001765int
Fred Drakea2f55112000-07-09 15:16:51 +00001766PyList_Reverse(PyObject *v)
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001767{
Tim Peters6063e262002-08-08 01:06:39 +00001768 PyListObject *self = (PyListObject *)v;
1769
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001770 if (v == NULL || !PyList_Check(v)) {
1771 PyErr_BadInternalCall();
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001772 return -1;
1773 }
Tim Peters6063e262002-08-08 01:06:39 +00001774 if (self->ob_size > 1)
1775 reverse_slice(self->ob_item, self->ob_item + self->ob_size);
Guido van Rossumb0fe3a91995-01-17 16:34:45 +00001776 return 0;
1777}
1778
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001779PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +00001780PyList_AsTuple(PyObject *v)
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001781{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001782 PyObject *w;
1783 PyObject **p;
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001784 int n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001785 if (v == NULL || !PyList_Check(v)) {
1786 PyErr_BadInternalCall();
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001787 return NULL;
1788 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001789 n = ((PyListObject *)v)->ob_size;
1790 w = PyTuple_New(n);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001791 if (w == NULL)
1792 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001793 p = ((PyTupleObject *)w)->ob_item;
Thomas Wouters334fb892000-07-25 12:56:38 +00001794 memcpy((void *)p,
1795 (void *)((PyListObject *)v)->ob_item,
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001796 n*sizeof(PyObject *));
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001797 while (--n >= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001798 Py_INCREF(*p);
Guido van Rossum6cd2fe01994-08-29 12:45:32 +00001799 p++;
1800 }
1801 return w;
1802}
1803
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001804static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001805listindex(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001806{
1807 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001808
Guido van Rossumed98d481991-03-06 13:07:53 +00001809 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001810 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1811 if (cmp > 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001812 return PyInt_FromLong((long)i);
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001813 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001814 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001815 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001816 PyErr_SetString(PyExc_ValueError, "list.index(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001817 return NULL;
1818}
1819
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001820static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001821listcount(PyListObject *self, PyObject *v)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001822{
1823 int count = 0;
1824 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001825
Guido van Rossume6f7d181991-10-20 20:20:40 +00001826 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001827 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1828 if (cmp > 0)
Guido van Rossume6f7d181991-10-20 20:20:40 +00001829 count++;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001830 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001831 return NULL;
Guido van Rossume6f7d181991-10-20 20:20:40 +00001832 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001833 return PyInt_FromLong((long)count);
Guido van Rossume6f7d181991-10-20 20:20:40 +00001834}
1835
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001836static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001837listremove(PyListObject *self, PyObject *v)
Guido van Rossumed98d481991-03-06 13:07:53 +00001838{
1839 int i;
Guido van Rossum4aa24f92000-02-24 15:23:03 +00001840
Guido van Rossumed98d481991-03-06 13:07:53 +00001841 for (i = 0; i < self->ob_size; i++) {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001842 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
1843 if (cmp > 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001844 if (list_ass_slice(self, i, i+1,
1845 (PyObject *)NULL) != 0)
Guido van Rossumed98d481991-03-06 13:07:53 +00001846 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001847 Py_INCREF(Py_None);
1848 return Py_None;
Guido van Rossumed98d481991-03-06 13:07:53 +00001849 }
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001850 else if (cmp < 0)
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001851 return NULL;
Guido van Rossumed98d481991-03-06 13:07:53 +00001852 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001853 PyErr_SetString(PyExc_ValueError, "list.remove(x): x not in list");
Guido van Rossumed98d481991-03-06 13:07:53 +00001854 return NULL;
1855}
1856
Jeremy Hylton8caad492000-06-23 14:18:11 +00001857static int
1858list_traverse(PyListObject *o, visitproc visit, void *arg)
1859{
1860 int i, err;
1861 PyObject *x;
1862
1863 for (i = o->ob_size; --i >= 0; ) {
1864 x = o->ob_item[i];
1865 if (x != NULL) {
1866 err = visit(x, arg);
1867 if (err)
1868 return err;
1869 }
1870 }
1871 return 0;
1872}
1873
1874static int
1875list_clear(PyListObject *lp)
1876{
1877 (void) PyList_SetSlice((PyObject *)lp, 0, lp->ob_size, 0);
1878 return 0;
1879}
1880
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001881static PyObject *
1882list_richcompare(PyObject *v, PyObject *w, int op)
1883{
1884 PyListObject *vl, *wl;
1885 int i;
1886
1887 if (!PyList_Check(v) || !PyList_Check(w)) {
1888 Py_INCREF(Py_NotImplemented);
1889 return Py_NotImplemented;
1890 }
1891
1892 vl = (PyListObject *)v;
1893 wl = (PyListObject *)w;
1894
1895 if (vl->ob_size != wl->ob_size && (op == Py_EQ || op == Py_NE)) {
1896 /* Shortcut: if the lengths differ, the lists differ */
1897 PyObject *res;
1898 if (op == Py_EQ)
1899 res = Py_False;
1900 else
1901 res = Py_True;
1902 Py_INCREF(res);
1903 return res;
1904 }
1905
1906 /* Search for the first index where items are different */
1907 for (i = 0; i < vl->ob_size && i < wl->ob_size; i++) {
1908 int k = PyObject_RichCompareBool(vl->ob_item[i],
1909 wl->ob_item[i], Py_EQ);
1910 if (k < 0)
1911 return NULL;
1912 if (!k)
1913 break;
1914 }
1915
1916 if (i >= vl->ob_size || i >= wl->ob_size) {
1917 /* No more items to compare -- compare sizes */
1918 int vs = vl->ob_size;
1919 int ws = wl->ob_size;
1920 int cmp;
1921 PyObject *res;
1922 switch (op) {
1923 case Py_LT: cmp = vs < ws; break;
Tim Peters6ee42342001-07-06 17:45:43 +00001924 case Py_LE: cmp = vs <= ws; break;
Guido van Rossum65e1cea2001-01-17 22:11:59 +00001925 case Py_EQ: cmp = vs == ws; break;
1926 case Py_NE: cmp = vs != ws; break;
1927 case Py_GT: cmp = vs > ws; break;
1928 case Py_GE: cmp = vs >= ws; break;
1929 default: return NULL; /* cannot happen */
1930 }
1931 if (cmp)
1932 res = Py_True;
1933 else
1934 res = Py_False;
1935 Py_INCREF(res);
1936 return res;
1937 }
1938
1939 /* We have an item that differs -- shortcuts for EQ/NE */
1940 if (op == Py_EQ) {
1941 Py_INCREF(Py_False);
1942 return Py_False;
1943 }
1944 if (op == Py_NE) {
1945 Py_INCREF(Py_True);
1946 return Py_True;
1947 }
1948
1949 /* Compare the final item again using the proper operator */
1950 return PyObject_RichCompare(vl->ob_item[i], wl->ob_item[i], op);
1951}
1952
Tim Peters6d6c1a32001-08-02 04:15:00 +00001953/* Adapted from newer code by Tim */
1954static int
1955list_fill(PyListObject *result, PyObject *v)
1956{
1957 PyObject *it; /* iter(v) */
1958 int n; /* guess for result list size */
1959 int i;
1960
1961 n = result->ob_size;
1962
1963 /* Special-case list(a_list), for speed. */
1964 if (PyList_Check(v)) {
1965 if (v == (PyObject *)result)
1966 return 0; /* source is destination, we're done */
1967 return list_ass_slice(result, 0, n, v);
1968 }
1969
1970 /* Empty previous contents */
1971 if (n != 0) {
1972 if (list_ass_slice(result, 0, n, (PyObject *)NULL) != 0)
1973 return -1;
1974 }
1975
1976 /* Get iterator. There may be some low-level efficiency to be gained
1977 * by caching the tp_iternext slot instead of using PyIter_Next()
1978 * later, but premature optimization is the root etc.
1979 */
1980 it = PyObject_GetIter(v);
1981 if (it == NULL)
1982 return -1;
1983
1984 /* Guess a result list size. */
1985 n = -1; /* unknown */
1986 if (PySequence_Check(v) &&
1987 v->ob_type->tp_as_sequence->sq_length) {
1988 n = PySequence_Size(v);
1989 if (n < 0)
1990 PyErr_Clear();
1991 }
1992 if (n < 0)
1993 n = 8; /* arbitrary */
1994 NRESIZE(result->ob_item, PyObject*, n);
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001995 if (result->ob_item == NULL) {
1996 PyErr_NoMemory();
Tim Peters6d6c1a32001-08-02 04:15:00 +00001997 goto error;
Neal Norwitzd4e5be52002-05-22 23:19:17 +00001998 }
Neal Norwitz35fc7602002-06-13 21:11:11 +00001999 memset(result->ob_item, 0, sizeof(*result->ob_item) * n);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002000 result->ob_size = n;
2001
2002 /* Run iterator to exhaustion. */
2003 for (i = 0; ; i++) {
2004 PyObject *item = PyIter_Next(it);
2005 if (item == NULL) {
2006 if (PyErr_Occurred())
2007 goto error;
2008 break;
2009 }
2010 if (i < n)
2011 PyList_SET_ITEM(result, i, item); /* steals ref */
2012 else {
2013 int status = ins1(result, result->ob_size, item);
2014 Py_DECREF(item); /* append creates a new ref */
2015 if (status < 0)
2016 goto error;
2017 }
2018 }
2019
2020 /* Cut back result list if initial guess was too large. */
2021 if (i < n && result != NULL) {
2022 if (list_ass_slice(result, i, n, (PyObject *)NULL) != 0)
2023 goto error;
2024 }
2025 Py_DECREF(it);
2026 return 0;
2027
2028 error:
2029 Py_DECREF(it);
2030 return -1;
2031}
2032
2033static int
2034list_init(PyListObject *self, PyObject *args, PyObject *kw)
2035{
2036 PyObject *arg = NULL;
2037 static char *kwlist[] = {"sequence", 0};
2038
2039 if (!PyArg_ParseTupleAndKeywords(args, kw, "|O:list", kwlist, &arg))
2040 return -1;
2041 if (arg != NULL)
2042 return list_fill(self, arg);
2043 if (self->ob_size > 0)
2044 return list_ass_slice(self, 0, self->ob_size, (PyObject*)NULL);
2045 return 0;
2046}
2047
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002048static long
2049list_nohash(PyObject *self)
2050{
2051 PyErr_SetString(PyExc_TypeError, "list objects are unhashable");
2052 return -1;
2053}
2054
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002055PyDoc_STRVAR(append_doc,
2056"L.append(object) -- append object to end");
2057PyDoc_STRVAR(extend_doc,
Martin v. Löwis673c0a22002-07-28 16:35:57 +00002058"L.extend(sequence) -- extend list by appending sequence elements");
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002059PyDoc_STRVAR(insert_doc,
2060"L.insert(index, object) -- insert object before index");
2061PyDoc_STRVAR(pop_doc,
2062"L.pop([index]) -> item -- remove and return item at index (default last)");
2063PyDoc_STRVAR(remove_doc,
2064"L.remove(value) -- remove first occurrence of value");
2065PyDoc_STRVAR(index_doc,
2066"L.index(value) -> integer -- return index of first occurrence of value");
2067PyDoc_STRVAR(count_doc,
2068"L.count(value) -> integer -- return number of occurrences of value");
2069PyDoc_STRVAR(reverse_doc,
2070"L.reverse() -- reverse *IN PLACE*");
2071PyDoc_STRVAR(sort_doc,
Tim Petersa64dc242002-08-01 02:13:36 +00002072"L.sort([cmpfunc]) -- stable sort *IN PLACE*; cmpfunc(x, y) -> -1, 0, 1");
Guido van Rossum3dd7f3f1998-06-30 15:36:32 +00002073
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002074static PyMethodDef list_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002075 {"append", (PyCFunction)listappend, METH_O, append_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002076 {"insert", (PyCFunction)listinsert, METH_VARARGS, insert_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002077 {"extend", (PyCFunction)listextend, METH_O, extend_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002078 {"pop", (PyCFunction)listpop, METH_VARARGS, pop_doc},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00002079 {"remove", (PyCFunction)listremove, METH_O, remove_doc},
2080 {"index", (PyCFunction)listindex, METH_O, index_doc},
2081 {"count", (PyCFunction)listcount, METH_O, count_doc},
2082 {"reverse", (PyCFunction)listreverse, METH_NOARGS, reverse_doc},
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002083 {"sort", (PyCFunction)listsort, METH_VARARGS, sort_doc},
Tim Petersa64dc242002-08-01 02:13:36 +00002084 {NULL, NULL} /* sentinel */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002085};
2086
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002087static PySequenceMethods list_as_sequence = {
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002088 (inquiry)list_length, /* sq_length */
2089 (binaryfunc)list_concat, /* sq_concat */
2090 (intargfunc)list_repeat, /* sq_repeat */
2091 (intargfunc)list_item, /* sq_item */
2092 (intintargfunc)list_slice, /* sq_slice */
2093 (intobjargproc)list_ass_item, /* sq_ass_item */
2094 (intintobjargproc)list_ass_slice, /* sq_ass_slice */
2095 (objobjproc)list_contains, /* sq_contains */
2096 (binaryfunc)list_inplace_concat, /* sq_inplace_concat */
2097 (intargfunc)list_inplace_repeat, /* sq_inplace_repeat */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002098};
2099
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002100PyDoc_STRVAR(list_doc,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002101"list() -> new list\n"
Neal Norwitz2c2e8272002-06-14 02:04:18 +00002102"list(sequence) -> new list initialized from sequence's items");
Tim Peters6d6c1a32001-08-02 04:15:00 +00002103
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002104static PyObject *list_iter(PyObject *seq);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002105
Jeremy Hyltona4b4c3b2002-07-13 03:51:17 +00002106static PyObject *
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002107list_subscript(PyListObject* self, PyObject* item)
2108{
2109 if (PyInt_Check(item)) {
2110 long i = PyInt_AS_LONG(item);
2111 if (i < 0)
2112 i += PyList_GET_SIZE(self);
2113 return list_item(self, i);
2114 }
2115 else if (PyLong_Check(item)) {
2116 long i = PyLong_AsLong(item);
2117 if (i == -1 && PyErr_Occurred())
2118 return NULL;
2119 if (i < 0)
2120 i += PyList_GET_SIZE(self);
2121 return list_item(self, i);
2122 }
2123 else if (PySlice_Check(item)) {
2124 int start, stop, step, slicelength, cur, i;
2125 PyObject* result;
2126 PyObject* it;
2127
2128 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2129 &start, &stop, &step, &slicelength) < 0) {
2130 return NULL;
2131 }
2132
2133 if (slicelength <= 0) {
2134 return PyList_New(0);
2135 }
2136 else {
2137 result = PyList_New(slicelength);
2138 if (!result) return NULL;
2139
Tim Peters3b01a122002-07-19 02:35:45 +00002140 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002141 cur += step, i++) {
2142 it = PyList_GET_ITEM(self, cur);
2143 Py_INCREF(it);
2144 PyList_SET_ITEM(result, i, it);
2145 }
Tim Peters3b01a122002-07-19 02:35:45 +00002146
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002147 return result;
2148 }
2149 }
2150 else {
2151 PyErr_SetString(PyExc_TypeError,
2152 "list indices must be integers");
2153 return NULL;
2154 }
2155}
2156
Tim Peters3b01a122002-07-19 02:35:45 +00002157static int
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002158list_ass_subscript(PyListObject* self, PyObject* item, PyObject* value)
2159{
2160 if (PyInt_Check(item)) {
2161 long i = PyInt_AS_LONG(item);
2162 if (i < 0)
2163 i += PyList_GET_SIZE(self);
2164 return list_ass_item(self, i, value);
2165 }
2166 else if (PyLong_Check(item)) {
2167 long i = PyLong_AsLong(item);
2168 if (i == -1 && PyErr_Occurred())
2169 return -1;
2170 if (i < 0)
2171 i += PyList_GET_SIZE(self);
2172 return list_ass_item(self, i, value);
2173 }
2174 else if (PySlice_Check(item)) {
2175 int start, stop, step, slicelength;
2176
2177 if (PySlice_GetIndicesEx((PySliceObject*)item, self->ob_size,
2178 &start, &stop, &step, &slicelength) < 0) {
2179 return -1;
2180 }
2181
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002182 /* treat L[slice(a,b)] = v _exactly_ like L[a:b] = v */
2183 if (step == 1 && ((PySliceObject*)item)->step == Py_None)
2184 return list_ass_slice(self, start, stop, value);
2185
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002186 if (value == NULL) {
2187 /* delete slice */
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002188 PyObject **garbage, **it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002189 int cur, i, j;
Tim Peters3b01a122002-07-19 02:35:45 +00002190
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002191 if (slicelength <= 0)
2192 return 0;
2193
2194 if (step < 0) {
2195 stop = start + 1;
2196 start = stop + step*(slicelength - 1) - 1;
2197 step = -step;
2198 }
2199
2200 garbage = (PyObject**)
2201 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002202
2203 /* drawing pictures might help
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002204 understand these for loops */
Guido van Rossum75a20b12002-06-11 12:22:28 +00002205 for (cur = start, i = 0;
2206 cur < stop;
Michael W. Hudson56796f62002-07-29 14:35:04 +00002207 cur += step, i++) {
2208 int lim = step;
2209
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002210 garbage[i] = PyList_GET_ITEM(self, cur);
2211
Michael W. Hudson56796f62002-07-29 14:35:04 +00002212 if (cur + step >= self->ob_size) {
2213 lim = self->ob_size - cur - 1;
2214 }
2215
2216 for (j = 0; j < lim; j++) {
Tim Peters3b01a122002-07-19 02:35:45 +00002217 PyList_SET_ITEM(self, cur + j - i,
Guido van Rossum75a20b12002-06-11 12:22:28 +00002218 PyList_GET_ITEM(self,
2219 cur + j + 1));
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002220 }
2221 }
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002222 for (cur = start + slicelength*step + 1;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002223 cur < self->ob_size; cur++) {
2224 PyList_SET_ITEM(self, cur - slicelength,
2225 PyList_GET_ITEM(self, cur));
2226 }
2227 self->ob_size -= slicelength;
Michael W. Hudson9c14bad2002-06-19 15:44:15 +00002228 it = self->ob_item;
2229 NRESIZE(it, PyObject*, self->ob_size);
2230 self->ob_item = it;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002231
2232 for (i = 0; i < slicelength; i++) {
2233 Py_DECREF(garbage[i]);
2234 }
2235 PyMem_FREE(garbage);
2236
2237 return 0;
2238 }
2239 else {
2240 /* assign slice */
2241 PyObject **garbage, *ins;
2242 int cur, i;
2243
2244 if (!PyList_Check(value)) {
2245 PyErr_Format(PyExc_TypeError,
2246 "must assign list (not \"%.200s\") to slice",
2247 value->ob_type->tp_name);
2248 return -1;
2249 }
2250
2251 if (PyList_GET_SIZE(value) != slicelength) {
2252 PyErr_Format(PyExc_ValueError,
2253 "attempt to assign list of size %d to extended slice of size %d",
2254 PyList_Size(value), slicelength);
2255 return -1;
2256 }
2257
2258 if (!slicelength)
2259 return 0;
2260
2261 /* protect against a[::-1] = a */
Tim Peters3b01a122002-07-19 02:35:45 +00002262 if (self == (PyListObject*)value) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002263 value = list_slice((PyListObject*)value, 0,
2264 PyList_GET_SIZE(value));
Tim Peters3b01a122002-07-19 02:35:45 +00002265 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002266 else {
2267 Py_INCREF(value);
2268 }
2269
2270 garbage = (PyObject**)
2271 PyMem_MALLOC(slicelength*sizeof(PyObject*));
Tim Peters3b01a122002-07-19 02:35:45 +00002272
2273 for (cur = start, i = 0; i < slicelength;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002274 cur += step, i++) {
2275 garbage[i] = PyList_GET_ITEM(self, cur);
Tim Peters3b01a122002-07-19 02:35:45 +00002276
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002277 ins = PyList_GET_ITEM(value, i);
2278 Py_INCREF(ins);
2279 PyList_SET_ITEM(self, cur, ins);
2280 }
2281
2282 for (i = 0; i < slicelength; i++) {
2283 Py_DECREF(garbage[i]);
2284 }
Tim Peters3b01a122002-07-19 02:35:45 +00002285
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002286 PyMem_FREE(garbage);
2287 Py_DECREF(value);
Tim Peters3b01a122002-07-19 02:35:45 +00002288
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002289 return 0;
2290 }
Tim Peters3b01a122002-07-19 02:35:45 +00002291 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002292 else {
Tim Peters3b01a122002-07-19 02:35:45 +00002293 PyErr_SetString(PyExc_TypeError,
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002294 "list indices must be integers");
2295 return -1;
2296 }
2297}
2298
2299static PyMappingMethods list_as_mapping = {
2300 (inquiry)list_length,
2301 (binaryfunc)list_subscript,
2302 (objobjargproc)list_ass_subscript
2303};
2304
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002305PyTypeObject PyList_Type = {
2306 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002307 0,
2308 "list",
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002309 sizeof(PyListObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002310 0,
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002311 (destructor)list_dealloc, /* tp_dealloc */
2312 (printfunc)list_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002313 0, /* tp_getattr */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002314 0, /* tp_setattr */
2315 0, /* tp_compare */
2316 (reprfunc)list_repr, /* tp_repr */
2317 0, /* tp_as_number */
2318 &list_as_sequence, /* tp_as_sequence */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00002319 &list_as_mapping, /* tp_as_mapping */
Guido van Rossumdbb53d92001-12-03 16:32:18 +00002320 list_nohash, /* tp_hash */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002321 0, /* tp_call */
2322 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002323 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002324 0, /* tp_setattro */
2325 0, /* tp_as_buffer */
Neil Schemenauere83c00e2001-08-29 23:54:21 +00002326 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
Tim Peters6d6c1a32001-08-02 04:15:00 +00002327 Py_TPFLAGS_BASETYPE, /* tp_flags */
2328 list_doc, /* tp_doc */
Guido van Rossum65e1cea2001-01-17 22:11:59 +00002329 (traverseproc)list_traverse, /* tp_traverse */
2330 (inquiry)list_clear, /* tp_clear */
2331 list_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002332 0, /* tp_weaklistoffset */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002333 list_iter, /* tp_iter */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002334 0, /* tp_iternext */
2335 list_methods, /* tp_methods */
2336 0, /* tp_members */
2337 0, /* tp_getset */
2338 0, /* tp_base */
2339 0, /* tp_dict */
2340 0, /* tp_descr_get */
2341 0, /* tp_descr_set */
2342 0, /* tp_dictoffset */
2343 (initproc)list_init, /* tp_init */
2344 PyType_GenericAlloc, /* tp_alloc */
2345 PyType_GenericNew, /* tp_new */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002346 PyObject_GC_Del, /* tp_free */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002347};
Guido van Rossum4c4e7df1998-06-16 15:18:28 +00002348
2349
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002350/*********************** List Iterator **************************/
2351
2352typedef struct {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002353 PyObject_HEAD
2354 long it_index;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002355 PyListObject *it_seq; /* Set to NULL when iterator is exhausted */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002356} listiterobject;
2357
2358PyTypeObject PyListIter_Type;
2359
Guido van Rossum5086e492002-07-16 15:56:52 +00002360static PyObject *
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002361list_iter(PyObject *seq)
2362{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002363 listiterobject *it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002364
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002365 if (!PyList_Check(seq)) {
2366 PyErr_BadInternalCall();
2367 return NULL;
2368 }
2369 it = PyObject_GC_New(listiterobject, &PyListIter_Type);
2370 if (it == NULL)
2371 return NULL;
2372 it->it_index = 0;
2373 Py_INCREF(seq);
2374 it->it_seq = (PyListObject *)seq;
2375 _PyObject_GC_TRACK(it);
2376 return (PyObject *)it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002377}
2378
2379static void
2380listiter_dealloc(listiterobject *it)
2381{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002382 _PyObject_GC_UNTRACK(it);
Guido van Rossum86103ae2002-07-16 20:07:32 +00002383 Py_XDECREF(it->it_seq);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002384 PyObject_GC_Del(it);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002385}
2386
2387static int
2388listiter_traverse(listiterobject *it, visitproc visit, void *arg)
2389{
Guido van Rossum86103ae2002-07-16 20:07:32 +00002390 if (it->it_seq == NULL)
2391 return 0;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002392 return visit((PyObject *)it->it_seq, arg);
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002393}
2394
2395
2396static PyObject *
2397listiter_getiter(PyObject *it)
2398{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002399 Py_INCREF(it);
2400 return it;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002401}
2402
2403static PyObject *
Tim Peters93b2cc42002-06-01 05:22:55 +00002404listiter_next(listiterobject *it)
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002405{
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002406 PyListObject *seq;
2407 PyObject *item;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002408
Tim Peters93b2cc42002-06-01 05:22:55 +00002409 assert(it != NULL);
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002410 seq = it->it_seq;
Guido van Rossum86103ae2002-07-16 20:07:32 +00002411 if (seq == NULL)
2412 return NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002413 assert(PyList_Check(seq));
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002414
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002415 if (it->it_index < PyList_GET_SIZE(seq)) {
Tim Peters93b2cc42002-06-01 05:22:55 +00002416 item = PyList_GET_ITEM(seq, it->it_index);
2417 ++it->it_index;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002418 Py_INCREF(item);
2419 return item;
2420 }
Guido van Rossum86103ae2002-07-16 20:07:32 +00002421
2422 Py_DECREF(seq);
2423 it->it_seq = NULL;
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002424 return NULL;
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002425}
2426
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002427PyTypeObject PyListIter_Type = {
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002428 PyObject_HEAD_INIT(&PyType_Type)
2429 0, /* ob_size */
2430 "listiterator", /* tp_name */
2431 sizeof(listiterobject), /* tp_basicsize */
2432 0, /* tp_itemsize */
2433 /* methods */
2434 (destructor)listiter_dealloc, /* tp_dealloc */
2435 0, /* tp_print */
2436 0, /* tp_getattr */
2437 0, /* tp_setattr */
2438 0, /* tp_compare */
2439 0, /* tp_repr */
2440 0, /* tp_as_number */
2441 0, /* tp_as_sequence */
2442 0, /* tp_as_mapping */
2443 0, /* tp_hash */
2444 0, /* tp_call */
2445 0, /* tp_str */
2446 PyObject_GenericGetAttr, /* tp_getattro */
2447 0, /* tp_setattro */
2448 0, /* tp_as_buffer */
2449 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2450 0, /* tp_doc */
2451 (traverseproc)listiter_traverse, /* tp_traverse */
2452 0, /* tp_clear */
2453 0, /* tp_richcompare */
2454 0, /* tp_weaklistoffset */
2455 (getiterfunc)listiter_getiter, /* tp_iter */
2456 (iternextfunc)listiter_next, /* tp_iternext */
Guido van Rossum86103ae2002-07-16 20:07:32 +00002457 0, /* tp_methods */
Guido van Rossum6b6272c2002-07-16 20:10:23 +00002458 0, /* tp_members */
2459 0, /* tp_getset */
2460 0, /* tp_base */
2461 0, /* tp_dict */
2462 0, /* tp_descr_get */
2463 0, /* tp_descr_set */
Raymond Hettinger14bd6de2002-05-31 21:40:38 +00002464};