The _lsprof module could crash the interpreter if it was given an external
[python.git] / Objects / listobject.c
blob10f2c5da259b83a00cca77a3f26819643ca4be2c
1 /* List object implementation */
3 #include "Python.h"
5 #ifdef STDC_HEADERS
6 #include <stddef.h>
7 #else
8 #include <sys/types.h> /* For size_t */
9 #endif
11 /* Ensure ob_item has room for at least newsize elements, and set
12 * ob_size to newsize. If newsize > ob_size on entry, the content
13 * of the new slots at exit is undefined heap trash; it's the caller's
14 * responsiblity to overwrite them with sane values.
15 * The number of allocated elements may grow, shrink, or stay the same.
16 * Failure is impossible if newsize <= self.allocated on entry, although
17 * that partly relies on an assumption that the system realloc() never
18 * fails when passed a number of bytes <= the number of bytes last
19 * allocated (the C standard doesn't guarantee this, but it's hard to
20 * imagine a realloc implementation where it wouldn't be true).
21 * Note that self->ob_item may change, and even if newsize is less
22 * than ob_size on entry.
24 static int
25 list_resize(PyListObject *self, Py_ssize_t newsize)
27 PyObject **items;
28 size_t new_allocated;
29 Py_ssize_t allocated = self->allocated;
31 /* Bypass realloc() when a previous overallocation is large enough
32 to accommodate the newsize. If the newsize falls lower than half
33 the allocated size, then proceed with the realloc() to shrink the list.
35 if (allocated >= newsize && newsize >= (allocated >> 1)) {
36 assert(self->ob_item != NULL || newsize == 0);
37 Py_SIZE(self) = newsize;
38 return 0;
41 /* This over-allocates proportional to the list size, making room
42 * for additional growth. The over-allocation is mild, but is
43 * enough to give linear-time amortized behavior over a long
44 * sequence of appends() in the presence of a poorly-performing
45 * system realloc().
46 * The growth pattern is: 0, 4, 8, 16, 25, 35, 46, 58, 72, 88, ...
48 new_allocated = (newsize >> 3) + (newsize < 9 ? 3 : 6);
50 /* check for integer overflow */
51 if (new_allocated > PY_SIZE_MAX - newsize) {
52 PyErr_NoMemory();
53 return -1;
54 } else {
55 new_allocated += newsize;
58 if (newsize == 0)
59 new_allocated = 0;
60 items = self->ob_item;
61 if (new_allocated <= ((~(size_t)0) / sizeof(PyObject *)))
62 PyMem_RESIZE(items, PyObject *, new_allocated);
63 else
64 items = NULL;
65 if (items == NULL) {
66 PyErr_NoMemory();
67 return -1;
69 self->ob_item = items;
70 Py_SIZE(self) = newsize;
71 self->allocated = new_allocated;
72 return 0;
75 /* Debug statistic to compare allocations with reuse through the free list */
76 #undef SHOW_ALLOC_COUNT
77 #ifdef SHOW_ALLOC_COUNT
78 static size_t count_alloc = 0;
79 static size_t count_reuse = 0;
81 static void
82 show_alloc(void)
84 fprintf(stderr, "List allocations: %" PY_FORMAT_SIZE_T "d\n",
85 count_alloc);
86 fprintf(stderr, "List reuse through freelist: %" PY_FORMAT_SIZE_T
87 "d\n", count_reuse);
88 fprintf(stderr, "%.2f%% reuse rate\n\n",
89 (100.0*count_reuse/(count_alloc+count_reuse)));
91 #endif
93 /* Empty list reuse scheme to save calls to malloc and free */
94 #ifndef PyList_MAXFREELIST
95 #define PyList_MAXFREELIST 80
96 #endif
97 static PyListObject *free_list[PyList_MAXFREELIST];
98 static int numfree = 0;
100 void
101 PyList_Fini(void)
103 PyListObject *op;
105 while (numfree) {
106 op = free_list[--numfree];
107 assert(PyList_CheckExact(op));
108 PyObject_GC_Del(op);
112 PyObject *
113 PyList_New(Py_ssize_t size)
115 PyListObject *op;
116 size_t nbytes;
117 #ifdef SHOW_ALLOC_COUNT
118 static int initialized = 0;
119 if (!initialized) {
120 Py_AtExit(show_alloc);
121 initialized = 1;
123 #endif
125 if (size < 0) {
126 PyErr_BadInternalCall();
127 return NULL;
129 nbytes = size * sizeof(PyObject *);
130 /* Check for overflow without an actual overflow,
131 * which can cause compiler to optimise out */
132 if (size > PY_SIZE_MAX / sizeof(PyObject *))
133 return PyErr_NoMemory();
134 if (numfree) {
135 numfree--;
136 op = free_list[numfree];
137 _Py_NewReference((PyObject *)op);
138 #ifdef SHOW_ALLOC_COUNT
139 count_reuse++;
140 #endif
141 } else {
142 op = PyObject_GC_New(PyListObject, &PyList_Type);
143 if (op == NULL)
144 return NULL;
145 #ifdef SHOW_ALLOC_COUNT
146 count_alloc++;
147 #endif
149 if (size <= 0)
150 op->ob_item = NULL;
151 else {
152 op->ob_item = (PyObject **) PyMem_MALLOC(nbytes);
153 if (op->ob_item == NULL) {
154 Py_DECREF(op);
155 return PyErr_NoMemory();
157 memset(op->ob_item, 0, nbytes);
159 Py_SIZE(op) = size;
160 op->allocated = size;
161 _PyObject_GC_TRACK(op);
162 return (PyObject *) op;
165 Py_ssize_t
166 PyList_Size(PyObject *op)
168 if (!PyList_Check(op)) {
169 PyErr_BadInternalCall();
170 return -1;
172 else
173 return Py_SIZE(op);
176 static PyObject *indexerr = NULL;
178 PyObject *
179 PyList_GetItem(PyObject *op, Py_ssize_t i)
181 if (!PyList_Check(op)) {
182 PyErr_BadInternalCall();
183 return NULL;
185 if (i < 0 || i >= Py_SIZE(op)) {
186 if (indexerr == NULL)
187 indexerr = PyString_FromString(
188 "list index out of range");
189 PyErr_SetObject(PyExc_IndexError, indexerr);
190 return NULL;
192 return ((PyListObject *)op) -> ob_item[i];
196 PyList_SetItem(register PyObject *op, register Py_ssize_t i,
197 register PyObject *newitem)
199 register PyObject *olditem;
200 register PyObject **p;
201 if (!PyList_Check(op)) {
202 Py_XDECREF(newitem);
203 PyErr_BadInternalCall();
204 return -1;
206 if (i < 0 || i >= Py_SIZE(op)) {
207 Py_XDECREF(newitem);
208 PyErr_SetString(PyExc_IndexError,
209 "list assignment index out of range");
210 return -1;
212 p = ((PyListObject *)op) -> ob_item + i;
213 olditem = *p;
214 *p = newitem;
215 Py_XDECREF(olditem);
216 return 0;
219 static int
220 ins1(PyListObject *self, Py_ssize_t where, PyObject *v)
222 Py_ssize_t i, n = Py_SIZE(self);
223 PyObject **items;
224 if (v == NULL) {
225 PyErr_BadInternalCall();
226 return -1;
228 if (n == PY_SSIZE_T_MAX) {
229 PyErr_SetString(PyExc_OverflowError,
230 "cannot add more objects to list");
231 return -1;
234 if (list_resize(self, n+1) == -1)
235 return -1;
237 if (where < 0) {
238 where += n;
239 if (where < 0)
240 where = 0;
242 if (where > n)
243 where = n;
244 items = self->ob_item;
245 for (i = n; --i >= where; )
246 items[i+1] = items[i];
247 Py_INCREF(v);
248 items[where] = v;
249 return 0;
253 PyList_Insert(PyObject *op, Py_ssize_t where, PyObject *newitem)
255 if (!PyList_Check(op)) {
256 PyErr_BadInternalCall();
257 return -1;
259 return ins1((PyListObject *)op, where, newitem);
262 static int
263 app1(PyListObject *self, PyObject *v)
265 Py_ssize_t n = PyList_GET_SIZE(self);
267 assert (v != NULL);
268 if (n == PY_SSIZE_T_MAX) {
269 PyErr_SetString(PyExc_OverflowError,
270 "cannot add more objects to list");
271 return -1;
274 if (list_resize(self, n+1) == -1)
275 return -1;
277 Py_INCREF(v);
278 PyList_SET_ITEM(self, n, v);
279 return 0;
283 PyList_Append(PyObject *op, PyObject *newitem)
285 if (PyList_Check(op) && (newitem != NULL))
286 return app1((PyListObject *)op, newitem);
287 PyErr_BadInternalCall();
288 return -1;
291 /* Methods */
293 static void
294 list_dealloc(PyListObject *op)
296 Py_ssize_t i;
297 PyObject_GC_UnTrack(op);
298 Py_TRASHCAN_SAFE_BEGIN(op)
299 if (op->ob_item != NULL) {
300 /* Do it backwards, for Christian Tismer.
301 There's a simple test case where somehow this reduces
302 thrashing when a *very* large list is created and
303 immediately deleted. */
304 i = Py_SIZE(op);
305 while (--i >= 0) {
306 Py_XDECREF(op->ob_item[i]);
308 PyMem_FREE(op->ob_item);
310 if (numfree < PyList_MAXFREELIST && PyList_CheckExact(op))
311 free_list[numfree++] = op;
312 else
313 Py_TYPE(op)->tp_free((PyObject *)op);
314 Py_TRASHCAN_SAFE_END(op)
317 static int
318 list_print(PyListObject *op, FILE *fp, int flags)
320 int rc;
321 Py_ssize_t i;
323 rc = Py_ReprEnter((PyObject*)op);
324 if (rc != 0) {
325 if (rc < 0)
326 return rc;
327 Py_BEGIN_ALLOW_THREADS
328 fprintf(fp, "[...]");
329 Py_END_ALLOW_THREADS
330 return 0;
332 Py_BEGIN_ALLOW_THREADS
333 fprintf(fp, "[");
334 Py_END_ALLOW_THREADS
335 for (i = 0; i < Py_SIZE(op); i++) {
336 if (i > 0) {
337 Py_BEGIN_ALLOW_THREADS
338 fprintf(fp, ", ");
339 Py_END_ALLOW_THREADS
341 if (PyObject_Print(op->ob_item[i], fp, 0) != 0) {
342 Py_ReprLeave((PyObject *)op);
343 return -1;
346 Py_BEGIN_ALLOW_THREADS
347 fprintf(fp, "]");
348 Py_END_ALLOW_THREADS
349 Py_ReprLeave((PyObject *)op);
350 return 0;
353 static PyObject *
354 list_repr(PyListObject *v)
356 Py_ssize_t i;
357 PyObject *s, *temp;
358 PyObject *pieces = NULL, *result = NULL;
360 i = Py_ReprEnter((PyObject*)v);
361 if (i != 0) {
362 return i > 0 ? PyString_FromString("[...]") : NULL;
365 if (Py_SIZE(v) == 0) {
366 result = PyString_FromString("[]");
367 goto Done;
370 pieces = PyList_New(0);
371 if (pieces == NULL)
372 goto Done;
374 /* Do repr() on each element. Note that this may mutate the list,
375 so must refetch the list size on each iteration. */
376 for (i = 0; i < Py_SIZE(v); ++i) {
377 int status;
378 if (Py_EnterRecursiveCall(" while getting the repr of a list"))
379 goto Done;
380 s = PyObject_Repr(v->ob_item[i]);
381 Py_LeaveRecursiveCall();
382 if (s == NULL)
383 goto Done;
384 status = PyList_Append(pieces, s);
385 Py_DECREF(s); /* append created a new ref */
386 if (status < 0)
387 goto Done;
390 /* Add "[]" decorations to the first and last items. */
391 assert(PyList_GET_SIZE(pieces) > 0);
392 s = PyString_FromString("[");
393 if (s == NULL)
394 goto Done;
395 temp = PyList_GET_ITEM(pieces, 0);
396 PyString_ConcatAndDel(&s, temp);
397 PyList_SET_ITEM(pieces, 0, s);
398 if (s == NULL)
399 goto Done;
401 s = PyString_FromString("]");
402 if (s == NULL)
403 goto Done;
404 temp = PyList_GET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1);
405 PyString_ConcatAndDel(&temp, s);
406 PyList_SET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1, temp);
407 if (temp == NULL)
408 goto Done;
410 /* Paste them all together with ", " between. */
411 s = PyString_FromString(", ");
412 if (s == NULL)
413 goto Done;
414 result = _PyString_Join(s, pieces);
415 Py_DECREF(s);
417 Done:
418 Py_XDECREF(pieces);
419 Py_ReprLeave((PyObject *)v);
420 return result;
423 static Py_ssize_t
424 list_length(PyListObject *a)
426 return Py_SIZE(a);
429 static int
430 list_contains(PyListObject *a, PyObject *el)
432 Py_ssize_t i;
433 int cmp;
435 for (i = 0, cmp = 0 ; cmp == 0 && i < Py_SIZE(a); ++i)
436 cmp = PyObject_RichCompareBool(el, PyList_GET_ITEM(a, i),
437 Py_EQ);
438 return cmp;
441 static PyObject *
442 list_item(PyListObject *a, Py_ssize_t i)
444 if (i < 0 || i >= Py_SIZE(a)) {
445 if (indexerr == NULL)
446 indexerr = PyString_FromString(
447 "list index out of range");
448 PyErr_SetObject(PyExc_IndexError, indexerr);
449 return NULL;
451 Py_INCREF(a->ob_item[i]);
452 return a->ob_item[i];
455 static PyObject *
456 list_slice(PyListObject *a, Py_ssize_t ilow, Py_ssize_t ihigh)
458 PyListObject *np;
459 PyObject **src, **dest;
460 Py_ssize_t i, len;
461 if (ilow < 0)
462 ilow = 0;
463 else if (ilow > Py_SIZE(a))
464 ilow = Py_SIZE(a);
465 if (ihigh < ilow)
466 ihigh = ilow;
467 else if (ihigh > Py_SIZE(a))
468 ihigh = Py_SIZE(a);
469 len = ihigh - ilow;
470 np = (PyListObject *) PyList_New(len);
471 if (np == NULL)
472 return NULL;
474 src = a->ob_item + ilow;
475 dest = np->ob_item;
476 for (i = 0; i < len; i++) {
477 PyObject *v = src[i];
478 Py_INCREF(v);
479 dest[i] = v;
481 return (PyObject *)np;
484 PyObject *
485 PyList_GetSlice(PyObject *a, Py_ssize_t ilow, Py_ssize_t ihigh)
487 if (!PyList_Check(a)) {
488 PyErr_BadInternalCall();
489 return NULL;
491 return list_slice((PyListObject *)a, ilow, ihigh);
494 static PyObject *
495 list_concat(PyListObject *a, PyObject *bb)
497 Py_ssize_t size;
498 Py_ssize_t i;
499 PyObject **src, **dest;
500 PyListObject *np;
501 if (!PyList_Check(bb)) {
502 PyErr_Format(PyExc_TypeError,
503 "can only concatenate list (not \"%.200s\") to list",
504 bb->ob_type->tp_name);
505 return NULL;
507 #define b ((PyListObject *)bb)
508 size = Py_SIZE(a) + Py_SIZE(b);
509 if (size < 0)
510 return PyErr_NoMemory();
511 np = (PyListObject *) PyList_New(size);
512 if (np == NULL) {
513 return NULL;
515 src = a->ob_item;
516 dest = np->ob_item;
517 for (i = 0; i < Py_SIZE(a); i++) {
518 PyObject *v = src[i];
519 Py_INCREF(v);
520 dest[i] = v;
522 src = b->ob_item;
523 dest = np->ob_item + Py_SIZE(a);
524 for (i = 0; i < Py_SIZE(b); i++) {
525 PyObject *v = src[i];
526 Py_INCREF(v);
527 dest[i] = v;
529 return (PyObject *)np;
530 #undef b
533 static PyObject *
534 list_repeat(PyListObject *a, Py_ssize_t n)
536 Py_ssize_t i, j;
537 Py_ssize_t size;
538 PyListObject *np;
539 PyObject **p, **items;
540 PyObject *elem;
541 if (n < 0)
542 n = 0;
543 size = Py_SIZE(a) * n;
544 if (n && size/n != Py_SIZE(a))
545 return PyErr_NoMemory();
546 if (size == 0)
547 return PyList_New(0);
548 np = (PyListObject *) PyList_New(size);
549 if (np == NULL)
550 return NULL;
552 items = np->ob_item;
553 if (Py_SIZE(a) == 1) {
554 elem = a->ob_item[0];
555 for (i = 0; i < n; i++) {
556 items[i] = elem;
557 Py_INCREF(elem);
559 return (PyObject *) np;
561 p = np->ob_item;
562 items = a->ob_item;
563 for (i = 0; i < n; i++) {
564 for (j = 0; j < Py_SIZE(a); j++) {
565 *p = items[j];
566 Py_INCREF(*p);
567 p++;
570 return (PyObject *) np;
573 static int
574 list_clear(PyListObject *a)
576 Py_ssize_t i;
577 PyObject **item = a->ob_item;
578 if (item != NULL) {
579 /* Because XDECREF can recursively invoke operations on
580 this list, we make it empty first. */
581 i = Py_SIZE(a);
582 Py_SIZE(a) = 0;
583 a->ob_item = NULL;
584 a->allocated = 0;
585 while (--i >= 0) {
586 Py_XDECREF(item[i]);
588 PyMem_FREE(item);
590 /* Never fails; the return value can be ignored.
591 Note that there is no guarantee that the list is actually empty
592 at this point, because XDECREF may have populated it again! */
593 return 0;
596 /* a[ilow:ihigh] = v if v != NULL.
597 * del a[ilow:ihigh] if v == NULL.
599 * Special speed gimmick: when v is NULL and ihigh - ilow <= 8, it's
600 * guaranteed the call cannot fail.
602 static int
603 list_ass_slice(PyListObject *a, Py_ssize_t ilow, Py_ssize_t ihigh, PyObject *v)
605 /* Because [X]DECREF can recursively invoke list operations on
606 this list, we must postpone all [X]DECREF activity until
607 after the list is back in its canonical shape. Therefore
608 we must allocate an additional array, 'recycle', into which
609 we temporarily copy the items that are deleted from the
610 list. :-( */
611 PyObject *recycle_on_stack[8];
612 PyObject **recycle = recycle_on_stack; /* will allocate more if needed */
613 PyObject **item;
614 PyObject **vitem = NULL;
615 PyObject *v_as_SF = NULL; /* PySequence_Fast(v) */
616 Py_ssize_t n; /* # of elements in replacement list */
617 Py_ssize_t norig; /* # of elements in list getting replaced */
618 Py_ssize_t d; /* Change in size */
619 Py_ssize_t k;
620 size_t s;
621 int result = -1; /* guilty until proved innocent */
622 #define b ((PyListObject *)v)
623 if (v == NULL)
624 n = 0;
625 else {
626 if (a == b) {
627 /* Special case "a[i:j] = a" -- copy b first */
628 v = list_slice(b, 0, Py_SIZE(b));
629 if (v == NULL)
630 return result;
631 result = list_ass_slice(a, ilow, ihigh, v);
632 Py_DECREF(v);
633 return result;
635 v_as_SF = PySequence_Fast(v, "can only assign an iterable");
636 if(v_as_SF == NULL)
637 goto Error;
638 n = PySequence_Fast_GET_SIZE(v_as_SF);
639 vitem = PySequence_Fast_ITEMS(v_as_SF);
641 if (ilow < 0)
642 ilow = 0;
643 else if (ilow > Py_SIZE(a))
644 ilow = Py_SIZE(a);
646 if (ihigh < ilow)
647 ihigh = ilow;
648 else if (ihigh > Py_SIZE(a))
649 ihigh = Py_SIZE(a);
651 norig = ihigh - ilow;
652 assert(norig >= 0);
653 d = n - norig;
654 if (Py_SIZE(a) + d == 0) {
655 Py_XDECREF(v_as_SF);
656 return list_clear(a);
658 item = a->ob_item;
659 /* recycle the items that we are about to remove */
660 s = norig * sizeof(PyObject *);
661 if (s > sizeof(recycle_on_stack)) {
662 recycle = (PyObject **)PyMem_MALLOC(s);
663 if (recycle == NULL) {
664 PyErr_NoMemory();
665 goto Error;
668 memcpy(recycle, &item[ilow], s);
670 if (d < 0) { /* Delete -d items */
671 memmove(&item[ihigh+d], &item[ihigh],
672 (Py_SIZE(a) - ihigh)*sizeof(PyObject *));
673 list_resize(a, Py_SIZE(a) + d);
674 item = a->ob_item;
676 else if (d > 0) { /* Insert d items */
677 k = Py_SIZE(a);
678 if (list_resize(a, k+d) < 0)
679 goto Error;
680 item = a->ob_item;
681 memmove(&item[ihigh+d], &item[ihigh],
682 (k - ihigh)*sizeof(PyObject *));
684 for (k = 0; k < n; k++, ilow++) {
685 PyObject *w = vitem[k];
686 Py_XINCREF(w);
687 item[ilow] = w;
689 for (k = norig - 1; k >= 0; --k)
690 Py_XDECREF(recycle[k]);
691 result = 0;
692 Error:
693 if (recycle != recycle_on_stack)
694 PyMem_FREE(recycle);
695 Py_XDECREF(v_as_SF);
696 return result;
697 #undef b
701 PyList_SetSlice(PyObject *a, Py_ssize_t ilow, Py_ssize_t ihigh, PyObject *v)
703 if (!PyList_Check(a)) {
704 PyErr_BadInternalCall();
705 return -1;
707 return list_ass_slice((PyListObject *)a, ilow, ihigh, v);
710 static PyObject *
711 list_inplace_repeat(PyListObject *self, Py_ssize_t n)
713 PyObject **items;
714 Py_ssize_t size, i, j, p;
717 size = PyList_GET_SIZE(self);
718 if (size == 0 || n == 1) {
719 Py_INCREF(self);
720 return (PyObject *)self;
723 if (n < 1) {
724 (void)list_clear(self);
725 Py_INCREF(self);
726 return (PyObject *)self;
729 if (size > PY_SSIZE_T_MAX / n) {
730 return PyErr_NoMemory();
733 if (list_resize(self, size*n) == -1)
734 return NULL;
736 p = size;
737 items = self->ob_item;
738 for (i = 1; i < n; i++) { /* Start counting at 1, not 0 */
739 for (j = 0; j < size; j++) {
740 PyObject *o = items[j];
741 Py_INCREF(o);
742 items[p++] = o;
745 Py_INCREF(self);
746 return (PyObject *)self;
749 static int
750 list_ass_item(PyListObject *a, Py_ssize_t i, PyObject *v)
752 PyObject *old_value;
753 if (i < 0 || i >= Py_SIZE(a)) {
754 PyErr_SetString(PyExc_IndexError,
755 "list assignment index out of range");
756 return -1;
758 if (v == NULL)
759 return list_ass_slice(a, i, i+1, v);
760 Py_INCREF(v);
761 old_value = a->ob_item[i];
762 a->ob_item[i] = v;
763 Py_DECREF(old_value);
764 return 0;
767 static PyObject *
768 listinsert(PyListObject *self, PyObject *args)
770 Py_ssize_t i;
771 PyObject *v;
772 if (!PyArg_ParseTuple(args, "nO:insert", &i, &v))
773 return NULL;
774 if (ins1(self, i, v) == 0)
775 Py_RETURN_NONE;
776 return NULL;
779 static PyObject *
780 listappend(PyListObject *self, PyObject *v)
782 if (app1(self, v) == 0)
783 Py_RETURN_NONE;
784 return NULL;
787 static PyObject *
788 listextend(PyListObject *self, PyObject *b)
790 PyObject *it; /* iter(v) */
791 Py_ssize_t m; /* size of self */
792 Py_ssize_t n; /* guess for size of b */
793 Py_ssize_t mn; /* m + n */
794 Py_ssize_t i;
795 PyObject *(*iternext)(PyObject *);
797 /* Special cases:
798 1) lists and tuples which can use PySequence_Fast ops
799 2) extending self to self requires making a copy first
801 if (PyList_CheckExact(b) || PyTuple_CheckExact(b) || (PyObject *)self == b) {
802 PyObject **src, **dest;
803 b = PySequence_Fast(b, "argument must be iterable");
804 if (!b)
805 return NULL;
806 n = PySequence_Fast_GET_SIZE(b);
807 if (n == 0) {
808 /* short circuit when b is empty */
809 Py_DECREF(b);
810 Py_RETURN_NONE;
812 m = Py_SIZE(self);
813 if (list_resize(self, m + n) == -1) {
814 Py_DECREF(b);
815 return NULL;
817 /* note that we may still have self == b here for the
818 * situation a.extend(a), but the following code works
819 * in that case too. Just make sure to resize self
820 * before calling PySequence_Fast_ITEMS.
822 /* populate the end of self with b's items */
823 src = PySequence_Fast_ITEMS(b);
824 dest = self->ob_item + m;
825 for (i = 0; i < n; i++) {
826 PyObject *o = src[i];
827 Py_INCREF(o);
828 dest[i] = o;
830 Py_DECREF(b);
831 Py_RETURN_NONE;
834 it = PyObject_GetIter(b);
835 if (it == NULL)
836 return NULL;
837 iternext = *it->ob_type->tp_iternext;
839 /* Guess a result list size. */
840 n = _PyObject_LengthHint(b, 8);
841 m = Py_SIZE(self);
842 mn = m + n;
843 if (mn >= m) {
844 /* Make room. */
845 if (list_resize(self, mn) == -1)
846 goto error;
847 /* Make the list sane again. */
848 Py_SIZE(self) = m;
850 /* Else m + n overflowed; on the chance that n lied, and there really
851 * is enough room, ignore it. If n was telling the truth, we'll
852 * eventually run out of memory during the loop.
855 /* Run iterator to exhaustion. */
856 for (;;) {
857 PyObject *item = iternext(it);
858 if (item == NULL) {
859 if (PyErr_Occurred()) {
860 if (PyErr_ExceptionMatches(PyExc_StopIteration))
861 PyErr_Clear();
862 else
863 goto error;
865 break;
867 if (Py_SIZE(self) < self->allocated) {
868 /* steals ref */
869 PyList_SET_ITEM(self, Py_SIZE(self), item);
870 ++Py_SIZE(self);
872 else {
873 int status = app1(self, item);
874 Py_DECREF(item); /* append creates a new ref */
875 if (status < 0)
876 goto error;
880 /* Cut back result list if initial guess was too large. */
881 if (Py_SIZE(self) < self->allocated)
882 list_resize(self, Py_SIZE(self)); /* shrinking can't fail */
884 Py_DECREF(it);
885 Py_RETURN_NONE;
887 error:
888 Py_DECREF(it);
889 return NULL;
892 PyObject *
893 _PyList_Extend(PyListObject *self, PyObject *b)
895 return listextend(self, b);
898 static PyObject *
899 list_inplace_concat(PyListObject *self, PyObject *other)
901 PyObject *result;
903 result = listextend(self, other);
904 if (result == NULL)
905 return result;
906 Py_DECREF(result);
907 Py_INCREF(self);
908 return (PyObject *)self;
911 static PyObject *
912 listpop(PyListObject *self, PyObject *args)
914 Py_ssize_t i = -1;
915 PyObject *v;
916 int status;
918 if (!PyArg_ParseTuple(args, "|n:pop", &i))
919 return NULL;
921 if (Py_SIZE(self) == 0) {
922 /* Special-case most common failure cause */
923 PyErr_SetString(PyExc_IndexError, "pop from empty list");
924 return NULL;
926 if (i < 0)
927 i += Py_SIZE(self);
928 if (i < 0 || i >= Py_SIZE(self)) {
929 PyErr_SetString(PyExc_IndexError, "pop index out of range");
930 return NULL;
932 v = self->ob_item[i];
933 if (i == Py_SIZE(self) - 1) {
934 status = list_resize(self, Py_SIZE(self) - 1);
935 assert(status >= 0);
936 return v; /* and v now owns the reference the list had */
938 Py_INCREF(v);
939 status = list_ass_slice(self, i, i+1, (PyObject *)NULL);
940 assert(status >= 0);
941 /* Use status, so that in a release build compilers don't
942 * complain about the unused name.
944 (void) status;
946 return v;
949 /* Reverse a slice of a list in place, from lo up to (exclusive) hi. */
950 static void
951 reverse_slice(PyObject **lo, PyObject **hi)
953 assert(lo && hi);
955 --hi;
956 while (lo < hi) {
957 PyObject *t = *lo;
958 *lo = *hi;
959 *hi = t;
960 ++lo;
961 --hi;
965 /* Lots of code for an adaptive, stable, natural mergesort. There are many
966 * pieces to this algorithm; read listsort.txt for overviews and details.
969 /* Comparison function. Takes care of calling a user-supplied
970 * comparison function (any callable Python object), which must not be
971 * NULL (use the ISLT macro if you don't know, or call PyObject_RichCompareBool
972 * with Py_LT if you know it's NULL).
973 * Returns -1 on error, 1 if x < y, 0 if x >= y.
975 static int
976 islt(PyObject *x, PyObject *y, PyObject *compare)
978 PyObject *res;
979 PyObject *args;
980 Py_ssize_t i;
982 assert(compare != NULL);
983 /* Call the user's comparison function and translate the 3-way
984 * result into true or false (or error).
986 args = PyTuple_New(2);
987 if (args == NULL)
988 return -1;
989 Py_INCREF(x);
990 Py_INCREF(y);
991 PyTuple_SET_ITEM(args, 0, x);
992 PyTuple_SET_ITEM(args, 1, y);
993 res = PyObject_Call(compare, args, NULL);
994 Py_DECREF(args);
995 if (res == NULL)
996 return -1;
997 if (!PyInt_Check(res)) {
998 PyErr_Format(PyExc_TypeError,
999 "comparison function must return int, not %.200s",
1000 res->ob_type->tp_name);
1001 Py_DECREF(res);
1002 return -1;
1004 i = PyInt_AsLong(res);
1005 Py_DECREF(res);
1006 return i < 0;
1009 /* If COMPARE is NULL, calls PyObject_RichCompareBool with Py_LT, else calls
1010 * islt. This avoids a layer of function call in the usual case, and
1011 * sorting does many comparisons.
1012 * Returns -1 on error, 1 if x < y, 0 if x >= y.
1014 #define ISLT(X, Y, COMPARE) ((COMPARE) == NULL ? \
1015 PyObject_RichCompareBool(X, Y, Py_LT) : \
1016 islt(X, Y, COMPARE))
1018 /* Compare X to Y via "<". Goto "fail" if the comparison raises an
1019 error. Else "k" is set to true iff X<Y, and an "if (k)" block is
1020 started. It makes more sense in context <wink>. X and Y are PyObject*s.
1022 #define IFLT(X, Y) if ((k = ISLT(X, Y, compare)) < 0) goto fail; \
1023 if (k)
1025 /* binarysort is the best method for sorting small arrays: it does
1026 few compares, but can do data movement quadratic in the number of
1027 elements.
1028 [lo, hi) is a contiguous slice of a list, and is sorted via
1029 binary insertion. This sort is stable.
1030 On entry, must have lo <= start <= hi, and that [lo, start) is already
1031 sorted (pass start == lo if you don't know!).
1032 If islt() complains return -1, else 0.
1033 Even in case of error, the output slice will be some permutation of
1034 the input (nothing is lost or duplicated).
1036 static int
1037 binarysort(PyObject **lo, PyObject **hi, PyObject **start, PyObject *compare)
1038 /* compare -- comparison function object, or NULL for default */
1040 register Py_ssize_t k;
1041 register PyObject **l, **p, **r;
1042 register PyObject *pivot;
1044 assert(lo <= start && start <= hi);
1045 /* assert [lo, start) is sorted */
1046 if (lo == start)
1047 ++start;
1048 for (; start < hi; ++start) {
1049 /* set l to where *start belongs */
1050 l = lo;
1051 r = start;
1052 pivot = *r;
1053 /* Invariants:
1054 * pivot >= all in [lo, l).
1055 * pivot < all in [r, start).
1056 * The second is vacuously true at the start.
1058 assert(l < r);
1059 do {
1060 p = l + ((r - l) >> 1);
1061 IFLT(pivot, *p)
1062 r = p;
1063 else
1064 l = p+1;
1065 } while (l < r);
1066 assert(l == r);
1067 /* The invariants still hold, so pivot >= all in [lo, l) and
1068 pivot < all in [l, start), so pivot belongs at l. Note
1069 that if there are elements equal to pivot, l points to the
1070 first slot after them -- that's why this sort is stable.
1071 Slide over to make room.
1072 Caution: using memmove is much slower under MSVC 5;
1073 we're not usually moving many slots. */
1074 for (p = start; p > l; --p)
1075 *p = *(p-1);
1076 *l = pivot;
1078 return 0;
1080 fail:
1081 return -1;
1085 Return the length of the run beginning at lo, in the slice [lo, hi). lo < hi
1086 is required on entry. "A run" is the longest ascending sequence, with
1088 lo[0] <= lo[1] <= lo[2] <= ...
1090 or the longest descending sequence, with
1092 lo[0] > lo[1] > lo[2] > ...
1094 Boolean *descending is set to 0 in the former case, or to 1 in the latter.
1095 For its intended use in a stable mergesort, the strictness of the defn of
1096 "descending" is needed so that the caller can safely reverse a descending
1097 sequence without violating stability (strict > ensures there are no equal
1098 elements to get out of order).
1100 Returns -1 in case of error.
1102 static Py_ssize_t
1103 count_run(PyObject **lo, PyObject **hi, PyObject *compare, int *descending)
1105 Py_ssize_t k;
1106 Py_ssize_t n;
1108 assert(lo < hi);
1109 *descending = 0;
1110 ++lo;
1111 if (lo == hi)
1112 return 1;
1114 n = 2;
1115 IFLT(*lo, *(lo-1)) {
1116 *descending = 1;
1117 for (lo = lo+1; lo < hi; ++lo, ++n) {
1118 IFLT(*lo, *(lo-1))
1120 else
1121 break;
1124 else {
1125 for (lo = lo+1; lo < hi; ++lo, ++n) {
1126 IFLT(*lo, *(lo-1))
1127 break;
1131 return n;
1132 fail:
1133 return -1;
1137 Locate the proper position of key in a sorted vector; if the vector contains
1138 an element equal to key, return the position immediately to the left of
1139 the leftmost equal element. [gallop_right() does the same except returns
1140 the position to the right of the rightmost equal element (if any).]
1142 "a" is a sorted vector with n elements, starting at a[0]. n must be > 0.
1144 "hint" is an index at which to begin the search, 0 <= hint < n. The closer
1145 hint is to the final result, the faster this runs.
1147 The return value is the int k in 0..n such that
1149 a[k-1] < key <= a[k]
1151 pretending that *(a-1) is minus infinity and a[n] is plus infinity. IOW,
1152 key belongs at index k; or, IOW, the first k elements of a should precede
1153 key, and the last n-k should follow key.
1155 Returns -1 on error. See listsort.txt for info on the method.
1157 static Py_ssize_t
1158 gallop_left(PyObject *key, PyObject **a, Py_ssize_t n, Py_ssize_t hint, PyObject *compare)
1160 Py_ssize_t ofs;
1161 Py_ssize_t lastofs;
1162 Py_ssize_t k;
1164 assert(key && a && n > 0 && hint >= 0 && hint < n);
1166 a += hint;
1167 lastofs = 0;
1168 ofs = 1;
1169 IFLT(*a, key) {
1170 /* a[hint] < key -- gallop right, until
1171 * a[hint + lastofs] < key <= a[hint + ofs]
1173 const Py_ssize_t maxofs = n - hint; /* &a[n-1] is highest */
1174 while (ofs < maxofs) {
1175 IFLT(a[ofs], key) {
1176 lastofs = ofs;
1177 ofs = (ofs << 1) + 1;
1178 if (ofs <= 0) /* int overflow */
1179 ofs = maxofs;
1181 else /* key <= a[hint + ofs] */
1182 break;
1184 if (ofs > maxofs)
1185 ofs = maxofs;
1186 /* Translate back to offsets relative to &a[0]. */
1187 lastofs += hint;
1188 ofs += hint;
1190 else {
1191 /* key <= a[hint] -- gallop left, until
1192 * a[hint - ofs] < key <= a[hint - lastofs]
1194 const Py_ssize_t maxofs = hint + 1; /* &a[0] is lowest */
1195 while (ofs < maxofs) {
1196 IFLT(*(a-ofs), key)
1197 break;
1198 /* key <= a[hint - ofs] */
1199 lastofs = ofs;
1200 ofs = (ofs << 1) + 1;
1201 if (ofs <= 0) /* int overflow */
1202 ofs = maxofs;
1204 if (ofs > maxofs)
1205 ofs = maxofs;
1206 /* Translate back to positive offsets relative to &a[0]. */
1207 k = lastofs;
1208 lastofs = hint - ofs;
1209 ofs = hint - k;
1211 a -= hint;
1213 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1214 /* Now a[lastofs] < key <= a[ofs], so key belongs somewhere to the
1215 * right of lastofs but no farther right than ofs. Do a binary
1216 * search, with invariant a[lastofs-1] < key <= a[ofs].
1218 ++lastofs;
1219 while (lastofs < ofs) {
1220 Py_ssize_t m = lastofs + ((ofs - lastofs) >> 1);
1222 IFLT(a[m], key)
1223 lastofs = m+1; /* a[m] < key */
1224 else
1225 ofs = m; /* key <= a[m] */
1227 assert(lastofs == ofs); /* so a[ofs-1] < key <= a[ofs] */
1228 return ofs;
1230 fail:
1231 return -1;
1235 Exactly like gallop_left(), except that if key already exists in a[0:n],
1236 finds the position immediately to the right of the rightmost equal value.
1238 The return value is the int k in 0..n such that
1240 a[k-1] <= key < a[k]
1242 or -1 if error.
1244 The code duplication is massive, but this is enough different given that
1245 we're sticking to "<" comparisons that it's much harder to follow if
1246 written as one routine with yet another "left or right?" flag.
1248 static Py_ssize_t
1249 gallop_right(PyObject *key, PyObject **a, Py_ssize_t n, Py_ssize_t hint, PyObject *compare)
1251 Py_ssize_t ofs;
1252 Py_ssize_t lastofs;
1253 Py_ssize_t k;
1255 assert(key && a && n > 0 && hint >= 0 && hint < n);
1257 a += hint;
1258 lastofs = 0;
1259 ofs = 1;
1260 IFLT(key, *a) {
1261 /* key < a[hint] -- gallop left, until
1262 * a[hint - ofs] <= key < a[hint - lastofs]
1264 const Py_ssize_t maxofs = hint + 1; /* &a[0] is lowest */
1265 while (ofs < maxofs) {
1266 IFLT(key, *(a-ofs)) {
1267 lastofs = ofs;
1268 ofs = (ofs << 1) + 1;
1269 if (ofs <= 0) /* int overflow */
1270 ofs = maxofs;
1272 else /* a[hint - ofs] <= key */
1273 break;
1275 if (ofs > maxofs)
1276 ofs = maxofs;
1277 /* Translate back to positive offsets relative to &a[0]. */
1278 k = lastofs;
1279 lastofs = hint - ofs;
1280 ofs = hint - k;
1282 else {
1283 /* a[hint] <= key -- gallop right, until
1284 * a[hint + lastofs] <= key < a[hint + ofs]
1286 const Py_ssize_t maxofs = n - hint; /* &a[n-1] is highest */
1287 while (ofs < maxofs) {
1288 IFLT(key, a[ofs])
1289 break;
1290 /* a[hint + ofs] <= key */
1291 lastofs = ofs;
1292 ofs = (ofs << 1) + 1;
1293 if (ofs <= 0) /* int overflow */
1294 ofs = maxofs;
1296 if (ofs > maxofs)
1297 ofs = maxofs;
1298 /* Translate back to offsets relative to &a[0]. */
1299 lastofs += hint;
1300 ofs += hint;
1302 a -= hint;
1304 assert(-1 <= lastofs && lastofs < ofs && ofs <= n);
1305 /* Now a[lastofs] <= key < a[ofs], so key belongs somewhere to the
1306 * right of lastofs but no farther right than ofs. Do a binary
1307 * search, with invariant a[lastofs-1] <= key < a[ofs].
1309 ++lastofs;
1310 while (lastofs < ofs) {
1311 Py_ssize_t m = lastofs + ((ofs - lastofs) >> 1);
1313 IFLT(key, a[m])
1314 ofs = m; /* key < a[m] */
1315 else
1316 lastofs = m+1; /* a[m] <= key */
1318 assert(lastofs == ofs); /* so a[ofs-1] <= key < a[ofs] */
1319 return ofs;
1321 fail:
1322 return -1;
1325 /* The maximum number of entries in a MergeState's pending-runs stack.
1326 * This is enough to sort arrays of size up to about
1327 * 32 * phi ** MAX_MERGE_PENDING
1328 * where phi ~= 1.618. 85 is ridiculouslylarge enough, good for an array
1329 * with 2**64 elements.
1331 #define MAX_MERGE_PENDING 85
1333 /* When we get into galloping mode, we stay there until both runs win less
1334 * often than MIN_GALLOP consecutive times. See listsort.txt for more info.
1336 #define MIN_GALLOP 7
1338 /* Avoid malloc for small temp arrays. */
1339 #define MERGESTATE_TEMP_SIZE 256
1341 /* One MergeState exists on the stack per invocation of mergesort. It's just
1342 * a convenient way to pass state around among the helper functions.
1344 struct s_slice {
1345 PyObject **base;
1346 Py_ssize_t len;
1349 typedef struct s_MergeState {
1350 /* The user-supplied comparison function. or NULL if none given. */
1351 PyObject *compare;
1353 /* This controls when we get *into* galloping mode. It's initialized
1354 * to MIN_GALLOP. merge_lo and merge_hi tend to nudge it higher for
1355 * random data, and lower for highly structured data.
1357 Py_ssize_t min_gallop;
1359 /* 'a' is temp storage to help with merges. It contains room for
1360 * alloced entries.
1362 PyObject **a; /* may point to temparray below */
1363 Py_ssize_t alloced;
1365 /* A stack of n pending runs yet to be merged. Run #i starts at
1366 * address base[i] and extends for len[i] elements. It's always
1367 * true (so long as the indices are in bounds) that
1369 * pending[i].base + pending[i].len == pending[i+1].base
1371 * so we could cut the storage for this, but it's a minor amount,
1372 * and keeping all the info explicit simplifies the code.
1374 int n;
1375 struct s_slice pending[MAX_MERGE_PENDING];
1377 /* 'a' points to this when possible, rather than muck with malloc. */
1378 PyObject *temparray[MERGESTATE_TEMP_SIZE];
1379 } MergeState;
1381 /* Conceptually a MergeState's constructor. */
1382 static void
1383 merge_init(MergeState *ms, PyObject *compare)
1385 assert(ms != NULL);
1386 ms->compare = compare;
1387 ms->a = ms->temparray;
1388 ms->alloced = MERGESTATE_TEMP_SIZE;
1389 ms->n = 0;
1390 ms->min_gallop = MIN_GALLOP;
1393 /* Free all the temp memory owned by the MergeState. This must be called
1394 * when you're done with a MergeState, and may be called before then if
1395 * you want to free the temp memory early.
1397 static void
1398 merge_freemem(MergeState *ms)
1400 assert(ms != NULL);
1401 if (ms->a != ms->temparray)
1402 PyMem_Free(ms->a);
1403 ms->a = ms->temparray;
1404 ms->alloced = MERGESTATE_TEMP_SIZE;
1407 /* Ensure enough temp memory for 'need' array slots is available.
1408 * Returns 0 on success and -1 if the memory can't be gotten.
1410 static int
1411 merge_getmem(MergeState *ms, Py_ssize_t need)
1413 assert(ms != NULL);
1414 if (need <= ms->alloced)
1415 return 0;
1416 /* Don't realloc! That can cost cycles to copy the old data, but
1417 * we don't care what's in the block.
1419 merge_freemem(ms);
1420 if (need > PY_SSIZE_T_MAX / sizeof(PyObject*)) {
1421 PyErr_NoMemory();
1422 return -1;
1424 ms->a = (PyObject **)PyMem_Malloc(need * sizeof(PyObject*));
1425 if (ms->a) {
1426 ms->alloced = need;
1427 return 0;
1429 PyErr_NoMemory();
1430 merge_freemem(ms); /* reset to sane state */
1431 return -1;
1433 #define MERGE_GETMEM(MS, NEED) ((NEED) <= (MS)->alloced ? 0 : \
1434 merge_getmem(MS, NEED))
1436 /* Merge the na elements starting at pa with the nb elements starting at pb
1437 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1438 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1439 * merge, and should have na <= nb. See listsort.txt for more info.
1440 * Return 0 if successful, -1 if error.
1442 static Py_ssize_t
1443 merge_lo(MergeState *ms, PyObject **pa, Py_ssize_t na,
1444 PyObject **pb, Py_ssize_t nb)
1446 Py_ssize_t k;
1447 PyObject *compare;
1448 PyObject **dest;
1449 int result = -1; /* guilty until proved innocent */
1450 Py_ssize_t min_gallop;
1452 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1453 if (MERGE_GETMEM(ms, na) < 0)
1454 return -1;
1455 memcpy(ms->a, pa, na * sizeof(PyObject*));
1456 dest = pa;
1457 pa = ms->a;
1459 *dest++ = *pb++;
1460 --nb;
1461 if (nb == 0)
1462 goto Succeed;
1463 if (na == 1)
1464 goto CopyB;
1466 min_gallop = ms->min_gallop;
1467 compare = ms->compare;
1468 for (;;) {
1469 Py_ssize_t acount = 0; /* # of times A won in a row */
1470 Py_ssize_t bcount = 0; /* # of times B won in a row */
1472 /* Do the straightforward thing until (if ever) one run
1473 * appears to win consistently.
1475 for (;;) {
1476 assert(na > 1 && nb > 0);
1477 k = ISLT(*pb, *pa, compare);
1478 if (k) {
1479 if (k < 0)
1480 goto Fail;
1481 *dest++ = *pb++;
1482 ++bcount;
1483 acount = 0;
1484 --nb;
1485 if (nb == 0)
1486 goto Succeed;
1487 if (bcount >= min_gallop)
1488 break;
1490 else {
1491 *dest++ = *pa++;
1492 ++acount;
1493 bcount = 0;
1494 --na;
1495 if (na == 1)
1496 goto CopyB;
1497 if (acount >= min_gallop)
1498 break;
1502 /* One run is winning so consistently that galloping may
1503 * be a huge win. So try that, and continue galloping until
1504 * (if ever) neither run appears to be winning consistently
1505 * anymore.
1507 ++min_gallop;
1508 do {
1509 assert(na > 1 && nb > 0);
1510 min_gallop -= min_gallop > 1;
1511 ms->min_gallop = min_gallop;
1512 k = gallop_right(*pb, pa, na, 0, compare);
1513 acount = k;
1514 if (k) {
1515 if (k < 0)
1516 goto Fail;
1517 memcpy(dest, pa, k * sizeof(PyObject *));
1518 dest += k;
1519 pa += k;
1520 na -= k;
1521 if (na == 1)
1522 goto CopyB;
1523 /* na==0 is impossible now if the comparison
1524 * function is consistent, but we can't assume
1525 * that it is.
1527 if (na == 0)
1528 goto Succeed;
1530 *dest++ = *pb++;
1531 --nb;
1532 if (nb == 0)
1533 goto Succeed;
1535 k = gallop_left(*pa, pb, nb, 0, compare);
1536 bcount = k;
1537 if (k) {
1538 if (k < 0)
1539 goto Fail;
1540 memmove(dest, pb, k * sizeof(PyObject *));
1541 dest += k;
1542 pb += k;
1543 nb -= k;
1544 if (nb == 0)
1545 goto Succeed;
1547 *dest++ = *pa++;
1548 --na;
1549 if (na == 1)
1550 goto CopyB;
1551 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
1552 ++min_gallop; /* penalize it for leaving galloping mode */
1553 ms->min_gallop = min_gallop;
1555 Succeed:
1556 result = 0;
1557 Fail:
1558 if (na)
1559 memcpy(dest, pa, na * sizeof(PyObject*));
1560 return result;
1561 CopyB:
1562 assert(na == 1 && nb > 0);
1563 /* The last element of pa belongs at the end of the merge. */
1564 memmove(dest, pb, nb * sizeof(PyObject *));
1565 dest[nb] = *pa;
1566 return 0;
1569 /* Merge the na elements starting at pa with the nb elements starting at pb
1570 * in a stable way, in-place. na and nb must be > 0, and pa + na == pb.
1571 * Must also have that *pb < *pa, that pa[na-1] belongs at the end of the
1572 * merge, and should have na >= nb. See listsort.txt for more info.
1573 * Return 0 if successful, -1 if error.
1575 static Py_ssize_t
1576 merge_hi(MergeState *ms, PyObject **pa, Py_ssize_t na, PyObject **pb, Py_ssize_t nb)
1578 Py_ssize_t k;
1579 PyObject *compare;
1580 PyObject **dest;
1581 int result = -1; /* guilty until proved innocent */
1582 PyObject **basea;
1583 PyObject **baseb;
1584 Py_ssize_t min_gallop;
1586 assert(ms && pa && pb && na > 0 && nb > 0 && pa + na == pb);
1587 if (MERGE_GETMEM(ms, nb) < 0)
1588 return -1;
1589 dest = pb + nb - 1;
1590 memcpy(ms->a, pb, nb * sizeof(PyObject*));
1591 basea = pa;
1592 baseb = ms->a;
1593 pb = ms->a + nb - 1;
1594 pa += na - 1;
1596 *dest-- = *pa--;
1597 --na;
1598 if (na == 0)
1599 goto Succeed;
1600 if (nb == 1)
1601 goto CopyA;
1603 min_gallop = ms->min_gallop;
1604 compare = ms->compare;
1605 for (;;) {
1606 Py_ssize_t acount = 0; /* # of times A won in a row */
1607 Py_ssize_t bcount = 0; /* # of times B won in a row */
1609 /* Do the straightforward thing until (if ever) one run
1610 * appears to win consistently.
1612 for (;;) {
1613 assert(na > 0 && nb > 1);
1614 k = ISLT(*pb, *pa, compare);
1615 if (k) {
1616 if (k < 0)
1617 goto Fail;
1618 *dest-- = *pa--;
1619 ++acount;
1620 bcount = 0;
1621 --na;
1622 if (na == 0)
1623 goto Succeed;
1624 if (acount >= min_gallop)
1625 break;
1627 else {
1628 *dest-- = *pb--;
1629 ++bcount;
1630 acount = 0;
1631 --nb;
1632 if (nb == 1)
1633 goto CopyA;
1634 if (bcount >= min_gallop)
1635 break;
1639 /* One run is winning so consistently that galloping may
1640 * be a huge win. So try that, and continue galloping until
1641 * (if ever) neither run appears to be winning consistently
1642 * anymore.
1644 ++min_gallop;
1645 do {
1646 assert(na > 0 && nb > 1);
1647 min_gallop -= min_gallop > 1;
1648 ms->min_gallop = min_gallop;
1649 k = gallop_right(*pb, basea, na, na-1, compare);
1650 if (k < 0)
1651 goto Fail;
1652 k = na - k;
1653 acount = k;
1654 if (k) {
1655 dest -= k;
1656 pa -= k;
1657 memmove(dest+1, pa+1, k * sizeof(PyObject *));
1658 na -= k;
1659 if (na == 0)
1660 goto Succeed;
1662 *dest-- = *pb--;
1663 --nb;
1664 if (nb == 1)
1665 goto CopyA;
1667 k = gallop_left(*pa, baseb, nb, nb-1, compare);
1668 if (k < 0)
1669 goto Fail;
1670 k = nb - k;
1671 bcount = k;
1672 if (k) {
1673 dest -= k;
1674 pb -= k;
1675 memcpy(dest+1, pb+1, k * sizeof(PyObject *));
1676 nb -= k;
1677 if (nb == 1)
1678 goto CopyA;
1679 /* nb==0 is impossible now if the comparison
1680 * function is consistent, but we can't assume
1681 * that it is.
1683 if (nb == 0)
1684 goto Succeed;
1686 *dest-- = *pa--;
1687 --na;
1688 if (na == 0)
1689 goto Succeed;
1690 } while (acount >= MIN_GALLOP || bcount >= MIN_GALLOP);
1691 ++min_gallop; /* penalize it for leaving galloping mode */
1692 ms->min_gallop = min_gallop;
1694 Succeed:
1695 result = 0;
1696 Fail:
1697 if (nb)
1698 memcpy(dest-(nb-1), baseb, nb * sizeof(PyObject*));
1699 return result;
1700 CopyA:
1701 assert(nb == 1 && na > 0);
1702 /* The first element of pb belongs at the front of the merge. */
1703 dest -= na;
1704 pa -= na;
1705 memmove(dest+1, pa+1, na * sizeof(PyObject *));
1706 *dest = *pb;
1707 return 0;
1710 /* Merge the two runs at stack indices i and i+1.
1711 * Returns 0 on success, -1 on error.
1713 static Py_ssize_t
1714 merge_at(MergeState *ms, Py_ssize_t i)
1716 PyObject **pa, **pb;
1717 Py_ssize_t na, nb;
1718 Py_ssize_t k;
1719 PyObject *compare;
1721 assert(ms != NULL);
1722 assert(ms->n >= 2);
1723 assert(i >= 0);
1724 assert(i == ms->n - 2 || i == ms->n - 3);
1726 pa = ms->pending[i].base;
1727 na = ms->pending[i].len;
1728 pb = ms->pending[i+1].base;
1729 nb = ms->pending[i+1].len;
1730 assert(na > 0 && nb > 0);
1731 assert(pa + na == pb);
1733 /* Record the length of the combined runs; if i is the 3rd-last
1734 * run now, also slide over the last run (which isn't involved
1735 * in this merge). The current run i+1 goes away in any case.
1737 ms->pending[i].len = na + nb;
1738 if (i == ms->n - 3)
1739 ms->pending[i+1] = ms->pending[i+2];
1740 --ms->n;
1742 /* Where does b start in a? Elements in a before that can be
1743 * ignored (already in place).
1745 compare = ms->compare;
1746 k = gallop_right(*pb, pa, na, 0, compare);
1747 if (k < 0)
1748 return -1;
1749 pa += k;
1750 na -= k;
1751 if (na == 0)
1752 return 0;
1754 /* Where does a end in b? Elements in b after that can be
1755 * ignored (already in place).
1757 nb = gallop_left(pa[na-1], pb, nb, nb-1, compare);
1758 if (nb <= 0)
1759 return nb;
1761 /* Merge what remains of the runs, using a temp array with
1762 * min(na, nb) elements.
1764 if (na <= nb)
1765 return merge_lo(ms, pa, na, pb, nb);
1766 else
1767 return merge_hi(ms, pa, na, pb, nb);
1770 /* Examine the stack of runs waiting to be merged, merging adjacent runs
1771 * until the stack invariants are re-established:
1773 * 1. len[-3] > len[-2] + len[-1]
1774 * 2. len[-2] > len[-1]
1776 * See listsort.txt for more info.
1778 * Returns 0 on success, -1 on error.
1780 static int
1781 merge_collapse(MergeState *ms)
1783 struct s_slice *p = ms->pending;
1785 assert(ms);
1786 while (ms->n > 1) {
1787 Py_ssize_t n = ms->n - 2;
1788 if (n > 0 && p[n-1].len <= p[n].len + p[n+1].len) {
1789 if (p[n-1].len < p[n+1].len)
1790 --n;
1791 if (merge_at(ms, n) < 0)
1792 return -1;
1794 else if (p[n].len <= p[n+1].len) {
1795 if (merge_at(ms, n) < 0)
1796 return -1;
1798 else
1799 break;
1801 return 0;
1804 /* Regardless of invariants, merge all runs on the stack until only one
1805 * remains. This is used at the end of the mergesort.
1807 * Returns 0 on success, -1 on error.
1809 static int
1810 merge_force_collapse(MergeState *ms)
1812 struct s_slice *p = ms->pending;
1814 assert(ms);
1815 while (ms->n > 1) {
1816 Py_ssize_t n = ms->n - 2;
1817 if (n > 0 && p[n-1].len < p[n+1].len)
1818 --n;
1819 if (merge_at(ms, n) < 0)
1820 return -1;
1822 return 0;
1825 /* Compute a good value for the minimum run length; natural runs shorter
1826 * than this are boosted artificially via binary insertion.
1828 * If n < 64, return n (it's too small to bother with fancy stuff).
1829 * Else if n is an exact power of 2, return 32.
1830 * Else return an int k, 32 <= k <= 64, such that n/k is close to, but
1831 * strictly less than, an exact power of 2.
1833 * See listsort.txt for more info.
1835 static Py_ssize_t
1836 merge_compute_minrun(Py_ssize_t n)
1838 Py_ssize_t r = 0; /* becomes 1 if any 1 bits are shifted off */
1840 assert(n >= 0);
1841 while (n >= 64) {
1842 r |= n & 1;
1843 n >>= 1;
1845 return n + r;
1848 /* Special wrapper to support stable sorting using the decorate-sort-undecorate
1849 pattern. Holds a key which is used for comparisons and the original record
1850 which is returned during the undecorate phase. By exposing only the key
1851 during comparisons, the underlying sort stability characteristics are left
1852 unchanged. Also, if a custom comparison function is used, it will only see
1853 the key instead of a full record. */
1855 typedef struct {
1856 PyObject_HEAD
1857 PyObject *key;
1858 PyObject *value;
1859 } sortwrapperobject;
1861 PyDoc_STRVAR(sortwrapper_doc, "Object wrapper with a custom sort key.");
1862 static PyObject *
1863 sortwrapper_richcompare(sortwrapperobject *, sortwrapperobject *, int);
1864 static void
1865 sortwrapper_dealloc(sortwrapperobject *);
1867 static PyTypeObject sortwrapper_type = {
1868 PyVarObject_HEAD_INIT(&PyType_Type, 0)
1869 "sortwrapper", /* tp_name */
1870 sizeof(sortwrapperobject), /* tp_basicsize */
1871 0, /* tp_itemsize */
1872 /* methods */
1873 (destructor)sortwrapper_dealloc, /* tp_dealloc */
1874 0, /* tp_print */
1875 0, /* tp_getattr */
1876 0, /* tp_setattr */
1877 0, /* tp_compare */
1878 0, /* tp_repr */
1879 0, /* tp_as_number */
1880 0, /* tp_as_sequence */
1881 0, /* tp_as_mapping */
1882 0, /* tp_hash */
1883 0, /* tp_call */
1884 0, /* tp_str */
1885 PyObject_GenericGetAttr, /* tp_getattro */
1886 0, /* tp_setattro */
1887 0, /* tp_as_buffer */
1888 Py_TPFLAGS_DEFAULT |
1889 Py_TPFLAGS_HAVE_RICHCOMPARE, /* tp_flags */
1890 sortwrapper_doc, /* tp_doc */
1891 0, /* tp_traverse */
1892 0, /* tp_clear */
1893 (richcmpfunc)sortwrapper_richcompare, /* tp_richcompare */
1897 static PyObject *
1898 sortwrapper_richcompare(sortwrapperobject *a, sortwrapperobject *b, int op)
1900 if (!PyObject_TypeCheck(b, &sortwrapper_type)) {
1901 PyErr_SetString(PyExc_TypeError,
1902 "expected a sortwrapperobject");
1903 return NULL;
1905 return PyObject_RichCompare(a->key, b->key, op);
1908 static void
1909 sortwrapper_dealloc(sortwrapperobject *so)
1911 Py_XDECREF(so->key);
1912 Py_XDECREF(so->value);
1913 PyObject_Del(so);
1916 /* Returns a new reference to a sortwrapper.
1917 Consumes the references to the two underlying objects. */
1919 static PyObject *
1920 build_sortwrapper(PyObject *key, PyObject *value)
1922 sortwrapperobject *so;
1924 so = PyObject_New(sortwrapperobject, &sortwrapper_type);
1925 if (so == NULL)
1926 return NULL;
1927 so->key = key;
1928 so->value = value;
1929 return (PyObject *)so;
1932 /* Returns a new reference to the value underlying the wrapper. */
1933 static PyObject *
1934 sortwrapper_getvalue(PyObject *so)
1936 PyObject *value;
1938 if (!PyObject_TypeCheck(so, &sortwrapper_type)) {
1939 PyErr_SetString(PyExc_TypeError,
1940 "expected a sortwrapperobject");
1941 return NULL;
1943 value = ((sortwrapperobject *)so)->value;
1944 Py_INCREF(value);
1945 return value;
1948 /* Wrapper for user specified cmp functions in combination with a
1949 specified key function. Makes sure the cmp function is presented
1950 with the actual key instead of the sortwrapper */
1952 typedef struct {
1953 PyObject_HEAD
1954 PyObject *func;
1955 } cmpwrapperobject;
1957 static void
1958 cmpwrapper_dealloc(cmpwrapperobject *co)
1960 Py_XDECREF(co->func);
1961 PyObject_Del(co);
1964 static PyObject *
1965 cmpwrapper_call(cmpwrapperobject *co, PyObject *args, PyObject *kwds)
1967 PyObject *x, *y, *xx, *yy;
1969 if (!PyArg_UnpackTuple(args, "", 2, 2, &x, &y))
1970 return NULL;
1971 if (!PyObject_TypeCheck(x, &sortwrapper_type) ||
1972 !PyObject_TypeCheck(y, &sortwrapper_type)) {
1973 PyErr_SetString(PyExc_TypeError,
1974 "expected a sortwrapperobject");
1975 return NULL;
1977 xx = ((sortwrapperobject *)x)->key;
1978 yy = ((sortwrapperobject *)y)->key;
1979 return PyObject_CallFunctionObjArgs(co->func, xx, yy, NULL);
1982 PyDoc_STRVAR(cmpwrapper_doc, "cmp() wrapper for sort with custom keys.");
1984 static PyTypeObject cmpwrapper_type = {
1985 PyVarObject_HEAD_INIT(&PyType_Type, 0)
1986 "cmpwrapper", /* tp_name */
1987 sizeof(cmpwrapperobject), /* tp_basicsize */
1988 0, /* tp_itemsize */
1989 /* methods */
1990 (destructor)cmpwrapper_dealloc, /* tp_dealloc */
1991 0, /* tp_print */
1992 0, /* tp_getattr */
1993 0, /* tp_setattr */
1994 0, /* tp_compare */
1995 0, /* tp_repr */
1996 0, /* tp_as_number */
1997 0, /* tp_as_sequence */
1998 0, /* tp_as_mapping */
1999 0, /* tp_hash */
2000 (ternaryfunc)cmpwrapper_call, /* tp_call */
2001 0, /* tp_str */
2002 PyObject_GenericGetAttr, /* tp_getattro */
2003 0, /* tp_setattro */
2004 0, /* tp_as_buffer */
2005 Py_TPFLAGS_DEFAULT, /* tp_flags */
2006 cmpwrapper_doc, /* tp_doc */
2009 static PyObject *
2010 build_cmpwrapper(PyObject *cmpfunc)
2012 cmpwrapperobject *co;
2014 co = PyObject_New(cmpwrapperobject, &cmpwrapper_type);
2015 if (co == NULL)
2016 return NULL;
2017 Py_INCREF(cmpfunc);
2018 co->func = cmpfunc;
2019 return (PyObject *)co;
2022 /* An adaptive, stable, natural mergesort. See listsort.txt.
2023 * Returns Py_None on success, NULL on error. Even in case of error, the
2024 * list will be some permutation of its input state (nothing is lost or
2025 * duplicated).
2027 static PyObject *
2028 listsort(PyListObject *self, PyObject *args, PyObject *kwds)
2030 MergeState ms;
2031 PyObject **lo, **hi;
2032 Py_ssize_t nremaining;
2033 Py_ssize_t minrun;
2034 Py_ssize_t saved_ob_size, saved_allocated;
2035 PyObject **saved_ob_item;
2036 PyObject **final_ob_item;
2037 PyObject *compare = NULL;
2038 PyObject *result = NULL; /* guilty until proved innocent */
2039 int reverse = 0;
2040 PyObject *keyfunc = NULL;
2041 Py_ssize_t i;
2042 PyObject *key, *value, *kvpair;
2043 static char *kwlist[] = {"cmp", "key", "reverse", 0};
2045 assert(self != NULL);
2046 assert (PyList_Check(self));
2047 if (args != NULL) {
2048 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOi:sort",
2049 kwlist, &compare, &keyfunc, &reverse))
2050 return NULL;
2052 if (compare == Py_None)
2053 compare = NULL;
2054 if (compare != NULL &&
2055 PyErr_WarnPy3k("the cmp argument is not supported in 3.x", 1) < 0)
2056 return NULL;
2057 if (keyfunc == Py_None)
2058 keyfunc = NULL;
2059 if (compare != NULL && keyfunc != NULL) {
2060 compare = build_cmpwrapper(compare);
2061 if (compare == NULL)
2062 return NULL;
2063 } else
2064 Py_XINCREF(compare);
2066 /* The list is temporarily made empty, so that mutations performed
2067 * by comparison functions can't affect the slice of memory we're
2068 * sorting (allowing mutations during sorting is a core-dump
2069 * factory, since ob_item may change).
2071 saved_ob_size = Py_SIZE(self);
2072 saved_ob_item = self->ob_item;
2073 saved_allocated = self->allocated;
2074 Py_SIZE(self) = 0;
2075 self->ob_item = NULL;
2076 self->allocated = -1; /* any operation will reset it to >= 0 */
2078 if (keyfunc != NULL) {
2079 for (i=0 ; i < saved_ob_size ; i++) {
2080 value = saved_ob_item[i];
2081 key = PyObject_CallFunctionObjArgs(keyfunc, value,
2082 NULL);
2083 if (key == NULL) {
2084 for (i=i-1 ; i>=0 ; i--) {
2085 kvpair = saved_ob_item[i];
2086 value = sortwrapper_getvalue(kvpair);
2087 saved_ob_item[i] = value;
2088 Py_DECREF(kvpair);
2090 goto dsu_fail;
2092 kvpair = build_sortwrapper(key, value);
2093 if (kvpair == NULL)
2094 goto dsu_fail;
2095 saved_ob_item[i] = kvpair;
2099 /* Reverse sort stability achieved by initially reversing the list,
2100 applying a stable forward sort, then reversing the final result. */
2101 if (reverse && saved_ob_size > 1)
2102 reverse_slice(saved_ob_item, saved_ob_item + saved_ob_size);
2104 merge_init(&ms, compare);
2106 nremaining = saved_ob_size;
2107 if (nremaining < 2)
2108 goto succeed;
2110 /* March over the array once, left to right, finding natural runs,
2111 * and extending short natural runs to minrun elements.
2113 lo = saved_ob_item;
2114 hi = lo + nremaining;
2115 minrun = merge_compute_minrun(nremaining);
2116 do {
2117 int descending;
2118 Py_ssize_t n;
2120 /* Identify next run. */
2121 n = count_run(lo, hi, compare, &descending);
2122 if (n < 0)
2123 goto fail;
2124 if (descending)
2125 reverse_slice(lo, lo + n);
2126 /* If short, extend to min(minrun, nremaining). */
2127 if (n < minrun) {
2128 const Py_ssize_t force = nremaining <= minrun ?
2129 nremaining : minrun;
2130 if (binarysort(lo, lo + force, lo + n, compare) < 0)
2131 goto fail;
2132 n = force;
2134 /* Push run onto pending-runs stack, and maybe merge. */
2135 assert(ms.n < MAX_MERGE_PENDING);
2136 ms.pending[ms.n].base = lo;
2137 ms.pending[ms.n].len = n;
2138 ++ms.n;
2139 if (merge_collapse(&ms) < 0)
2140 goto fail;
2141 /* Advance to find next run. */
2142 lo += n;
2143 nremaining -= n;
2144 } while (nremaining);
2145 assert(lo == hi);
2147 if (merge_force_collapse(&ms) < 0)
2148 goto fail;
2149 assert(ms.n == 1);
2150 assert(ms.pending[0].base == saved_ob_item);
2151 assert(ms.pending[0].len == saved_ob_size);
2153 succeed:
2154 result = Py_None;
2155 fail:
2156 if (keyfunc != NULL) {
2157 for (i=0 ; i < saved_ob_size ; i++) {
2158 kvpair = saved_ob_item[i];
2159 value = sortwrapper_getvalue(kvpair);
2160 saved_ob_item[i] = value;
2161 Py_DECREF(kvpair);
2165 if (self->allocated != -1 && result != NULL) {
2166 /* The user mucked with the list during the sort,
2167 * and we don't already have another error to report.
2169 PyErr_SetString(PyExc_ValueError, "list modified during sort");
2170 result = NULL;
2173 if (reverse && saved_ob_size > 1)
2174 reverse_slice(saved_ob_item, saved_ob_item + saved_ob_size);
2176 merge_freemem(&ms);
2178 dsu_fail:
2179 final_ob_item = self->ob_item;
2180 i = Py_SIZE(self);
2181 Py_SIZE(self) = saved_ob_size;
2182 self->ob_item = saved_ob_item;
2183 self->allocated = saved_allocated;
2184 if (final_ob_item != NULL) {
2185 /* we cannot use list_clear() for this because it does not
2186 guarantee that the list is really empty when it returns */
2187 while (--i >= 0) {
2188 Py_XDECREF(final_ob_item[i]);
2190 PyMem_FREE(final_ob_item);
2192 Py_XDECREF(compare);
2193 Py_XINCREF(result);
2194 return result;
2196 #undef IFLT
2197 #undef ISLT
2200 PyList_Sort(PyObject *v)
2202 if (v == NULL || !PyList_Check(v)) {
2203 PyErr_BadInternalCall();
2204 return -1;
2206 v = listsort((PyListObject *)v, (PyObject *)NULL, (PyObject *)NULL);
2207 if (v == NULL)
2208 return -1;
2209 Py_DECREF(v);
2210 return 0;
2213 static PyObject *
2214 listreverse(PyListObject *self)
2216 if (Py_SIZE(self) > 1)
2217 reverse_slice(self->ob_item, self->ob_item + Py_SIZE(self));
2218 Py_RETURN_NONE;
2222 PyList_Reverse(PyObject *v)
2224 PyListObject *self = (PyListObject *)v;
2226 if (v == NULL || !PyList_Check(v)) {
2227 PyErr_BadInternalCall();
2228 return -1;
2230 if (Py_SIZE(self) > 1)
2231 reverse_slice(self->ob_item, self->ob_item + Py_SIZE(self));
2232 return 0;
2235 PyObject *
2236 PyList_AsTuple(PyObject *v)
2238 PyObject *w;
2239 PyObject **p, **q;
2240 Py_ssize_t n;
2241 if (v == NULL || !PyList_Check(v)) {
2242 PyErr_BadInternalCall();
2243 return NULL;
2245 n = Py_SIZE(v);
2246 w = PyTuple_New(n);
2247 if (w == NULL)
2248 return NULL;
2249 p = ((PyTupleObject *)w)->ob_item;
2250 q = ((PyListObject *)v)->ob_item;
2251 while (--n >= 0) {
2252 Py_INCREF(*q);
2253 *p = *q;
2254 p++;
2255 q++;
2257 return w;
2260 static PyObject *
2261 listindex(PyListObject *self, PyObject *args)
2263 Py_ssize_t i, start=0, stop=Py_SIZE(self);
2264 PyObject *v;
2266 if (!PyArg_ParseTuple(args, "O|O&O&:index", &v,
2267 _PyEval_SliceIndex, &start,
2268 _PyEval_SliceIndex, &stop))
2269 return NULL;
2270 if (start < 0) {
2271 start += Py_SIZE(self);
2272 if (start < 0)
2273 start = 0;
2275 if (stop < 0) {
2276 stop += Py_SIZE(self);
2277 if (stop < 0)
2278 stop = 0;
2280 for (i = start; i < stop && i < Py_SIZE(self); i++) {
2281 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
2282 if (cmp > 0)
2283 return PyInt_FromSsize_t(i);
2284 else if (cmp < 0)
2285 return NULL;
2287 PyErr_SetString(PyExc_ValueError, "list.index(x): x not in list");
2288 return NULL;
2291 static PyObject *
2292 listcount(PyListObject *self, PyObject *v)
2294 Py_ssize_t count = 0;
2295 Py_ssize_t i;
2297 for (i = 0; i < Py_SIZE(self); i++) {
2298 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
2299 if (cmp > 0)
2300 count++;
2301 else if (cmp < 0)
2302 return NULL;
2304 return PyInt_FromSsize_t(count);
2307 static PyObject *
2308 listremove(PyListObject *self, PyObject *v)
2310 Py_ssize_t i;
2312 for (i = 0; i < Py_SIZE(self); i++) {
2313 int cmp = PyObject_RichCompareBool(self->ob_item[i], v, Py_EQ);
2314 if (cmp > 0) {
2315 if (list_ass_slice(self, i, i+1,
2316 (PyObject *)NULL) == 0)
2317 Py_RETURN_NONE;
2318 return NULL;
2320 else if (cmp < 0)
2321 return NULL;
2323 PyErr_SetString(PyExc_ValueError, "list.remove(x): x not in list");
2324 return NULL;
2327 static int
2328 list_traverse(PyListObject *o, visitproc visit, void *arg)
2330 Py_ssize_t i;
2332 for (i = Py_SIZE(o); --i >= 0; )
2333 Py_VISIT(o->ob_item[i]);
2334 return 0;
2337 static PyObject *
2338 list_richcompare(PyObject *v, PyObject *w, int op)
2340 PyListObject *vl, *wl;
2341 Py_ssize_t i;
2343 if (!PyList_Check(v) || !PyList_Check(w)) {
2344 Py_INCREF(Py_NotImplemented);
2345 return Py_NotImplemented;
2348 vl = (PyListObject *)v;
2349 wl = (PyListObject *)w;
2351 if (Py_SIZE(vl) != Py_SIZE(wl) && (op == Py_EQ || op == Py_NE)) {
2352 /* Shortcut: if the lengths differ, the lists differ */
2353 PyObject *res;
2354 if (op == Py_EQ)
2355 res = Py_False;
2356 else
2357 res = Py_True;
2358 Py_INCREF(res);
2359 return res;
2362 /* Search for the first index where items are different */
2363 for (i = 0; i < Py_SIZE(vl) && i < Py_SIZE(wl); i++) {
2364 int k = PyObject_RichCompareBool(vl->ob_item[i],
2365 wl->ob_item[i], Py_EQ);
2366 if (k < 0)
2367 return NULL;
2368 if (!k)
2369 break;
2372 if (i >= Py_SIZE(vl) || i >= Py_SIZE(wl)) {
2373 /* No more items to compare -- compare sizes */
2374 Py_ssize_t vs = Py_SIZE(vl);
2375 Py_ssize_t ws = Py_SIZE(wl);
2376 int cmp;
2377 PyObject *res;
2378 switch (op) {
2379 case Py_LT: cmp = vs < ws; break;
2380 case Py_LE: cmp = vs <= ws; break;
2381 case Py_EQ: cmp = vs == ws; break;
2382 case Py_NE: cmp = vs != ws; break;
2383 case Py_GT: cmp = vs > ws; break;
2384 case Py_GE: cmp = vs >= ws; break;
2385 default: return NULL; /* cannot happen */
2387 if (cmp)
2388 res = Py_True;
2389 else
2390 res = Py_False;
2391 Py_INCREF(res);
2392 return res;
2395 /* We have an item that differs -- shortcuts for EQ/NE */
2396 if (op == Py_EQ) {
2397 Py_INCREF(Py_False);
2398 return Py_False;
2400 if (op == Py_NE) {
2401 Py_INCREF(Py_True);
2402 return Py_True;
2405 /* Compare the final item again using the proper operator */
2406 return PyObject_RichCompare(vl->ob_item[i], wl->ob_item[i], op);
2409 static int
2410 list_init(PyListObject *self, PyObject *args, PyObject *kw)
2412 PyObject *arg = NULL;
2413 static char *kwlist[] = {"sequence", 0};
2415 if (!PyArg_ParseTupleAndKeywords(args, kw, "|O:list", kwlist, &arg))
2416 return -1;
2418 /* Verify list invariants established by PyType_GenericAlloc() */
2419 assert(0 <= Py_SIZE(self));
2420 assert(Py_SIZE(self) <= self->allocated || self->allocated == -1);
2421 assert(self->ob_item != NULL ||
2422 self->allocated == 0 || self->allocated == -1);
2424 /* Empty previous contents */
2425 if (self->ob_item != NULL) {
2426 (void)list_clear(self);
2428 if (arg != NULL) {
2429 PyObject *rv = listextend(self, arg);
2430 if (rv == NULL)
2431 return -1;
2432 Py_DECREF(rv);
2434 return 0;
2437 static PyObject *
2438 list_sizeof(PyListObject *self)
2440 Py_ssize_t res;
2442 res = sizeof(PyListObject) + self->allocated * sizeof(void*);
2443 return PyInt_FromSsize_t(res);
2446 static PyObject *list_iter(PyObject *seq);
2447 static PyObject *list_reversed(PyListObject* seq, PyObject* unused);
2449 PyDoc_STRVAR(getitem_doc,
2450 "x.__getitem__(y) <==> x[y]");
2451 PyDoc_STRVAR(reversed_doc,
2452 "L.__reversed__() -- return a reverse iterator over the list");
2453 PyDoc_STRVAR(sizeof_doc,
2454 "L.__sizeof__() -- size of L in memory, in bytes");
2455 PyDoc_STRVAR(append_doc,
2456 "L.append(object) -- append object to end");
2457 PyDoc_STRVAR(extend_doc,
2458 "L.extend(iterable) -- extend list by appending elements from the iterable");
2459 PyDoc_STRVAR(insert_doc,
2460 "L.insert(index, object) -- insert object before index");
2461 PyDoc_STRVAR(pop_doc,
2462 "L.pop([index]) -> item -- remove and return item at index (default last)");
2463 PyDoc_STRVAR(remove_doc,
2464 "L.remove(value) -- remove first occurrence of value");
2465 PyDoc_STRVAR(index_doc,
2466 "L.index(value, [start, [stop]]) -> integer -- return first index of value");
2467 PyDoc_STRVAR(count_doc,
2468 "L.count(value) -> integer -- return number of occurrences of value");
2469 PyDoc_STRVAR(reverse_doc,
2470 "L.reverse() -- reverse *IN PLACE*");
2471 PyDoc_STRVAR(sort_doc,
2472 "L.sort(cmp=None, key=None, reverse=False) -- stable sort *IN PLACE*;\n\
2473 cmp(x, y) -> -1, 0, 1");
2475 static PyObject *list_subscript(PyListObject*, PyObject*);
2477 static PyMethodDef list_methods[] = {
2478 {"__getitem__", (PyCFunction)list_subscript, METH_O|METH_COEXIST, getitem_doc},
2479 {"__reversed__",(PyCFunction)list_reversed, METH_NOARGS, reversed_doc},
2480 {"__sizeof__", (PyCFunction)list_sizeof, METH_NOARGS, sizeof_doc},
2481 {"append", (PyCFunction)listappend, METH_O, append_doc},
2482 {"insert", (PyCFunction)listinsert, METH_VARARGS, insert_doc},
2483 {"extend", (PyCFunction)listextend, METH_O, extend_doc},
2484 {"pop", (PyCFunction)listpop, METH_VARARGS, pop_doc},
2485 {"remove", (PyCFunction)listremove, METH_O, remove_doc},
2486 {"index", (PyCFunction)listindex, METH_VARARGS, index_doc},
2487 {"count", (PyCFunction)listcount, METH_O, count_doc},
2488 {"reverse", (PyCFunction)listreverse, METH_NOARGS, reverse_doc},
2489 {"sort", (PyCFunction)listsort, METH_VARARGS | METH_KEYWORDS, sort_doc},
2490 {NULL, NULL} /* sentinel */
2493 static PySequenceMethods list_as_sequence = {
2494 (lenfunc)list_length, /* sq_length */
2495 (binaryfunc)list_concat, /* sq_concat */
2496 (ssizeargfunc)list_repeat, /* sq_repeat */
2497 (ssizeargfunc)list_item, /* sq_item */
2498 (ssizessizeargfunc)list_slice, /* sq_slice */
2499 (ssizeobjargproc)list_ass_item, /* sq_ass_item */
2500 (ssizessizeobjargproc)list_ass_slice, /* sq_ass_slice */
2501 (objobjproc)list_contains, /* sq_contains */
2502 (binaryfunc)list_inplace_concat, /* sq_inplace_concat */
2503 (ssizeargfunc)list_inplace_repeat, /* sq_inplace_repeat */
2506 PyDoc_STRVAR(list_doc,
2507 "list() -> new list\n"
2508 "list(sequence) -> new list initialized from sequence's items");
2511 static PyObject *
2512 list_subscript(PyListObject* self, PyObject* item)
2514 if (PyIndex_Check(item)) {
2515 Py_ssize_t i;
2516 i = PyNumber_AsSsize_t(item, PyExc_IndexError);
2517 if (i == -1 && PyErr_Occurred())
2518 return NULL;
2519 if (i < 0)
2520 i += PyList_GET_SIZE(self);
2521 return list_item(self, i);
2523 else if (PySlice_Check(item)) {
2524 Py_ssize_t start, stop, step, slicelength, cur, i;
2525 PyObject* result;
2526 PyObject* it;
2527 PyObject **src, **dest;
2529 if (PySlice_GetIndicesEx((PySliceObject*)item, Py_SIZE(self),
2530 &start, &stop, &step, &slicelength) < 0) {
2531 return NULL;
2534 if (slicelength <= 0) {
2535 return PyList_New(0);
2537 else if (step == 1) {
2538 return list_slice(self, start, stop);
2540 else {
2541 result = PyList_New(slicelength);
2542 if (!result) return NULL;
2544 src = self->ob_item;
2545 dest = ((PyListObject *)result)->ob_item;
2546 for (cur = start, i = 0; i < slicelength;
2547 cur += step, i++) {
2548 it = src[cur];
2549 Py_INCREF(it);
2550 dest[i] = it;
2553 return result;
2556 else {
2557 PyErr_Format(PyExc_TypeError,
2558 "list indices must be integers, not %.200s",
2559 item->ob_type->tp_name);
2560 return NULL;
2564 static int
2565 list_ass_subscript(PyListObject* self, PyObject* item, PyObject* value)
2567 if (PyIndex_Check(item)) {
2568 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
2569 if (i == -1 && PyErr_Occurred())
2570 return -1;
2571 if (i < 0)
2572 i += PyList_GET_SIZE(self);
2573 return list_ass_item(self, i, value);
2575 else if (PySlice_Check(item)) {
2576 Py_ssize_t start, stop, step, slicelength;
2578 if (PySlice_GetIndicesEx((PySliceObject*)item, Py_SIZE(self),
2579 &start, &stop, &step, &slicelength) < 0) {
2580 return -1;
2583 if (step == 1)
2584 return list_ass_slice(self, start, stop, value);
2586 /* Make sure s[5:2] = [..] inserts at the right place:
2587 before 5, not before 2. */
2588 if ((step < 0 && start < stop) ||
2589 (step > 0 && start > stop))
2590 stop = start;
2592 if (value == NULL) {
2593 /* delete slice */
2594 PyObject **garbage;
2595 Py_ssize_t cur, i;
2597 if (slicelength <= 0)
2598 return 0;
2600 if (step < 0) {
2601 stop = start + 1;
2602 start = stop + step*(slicelength - 1) - 1;
2603 step = -step;
2606 assert(slicelength <= PY_SIZE_MAX / sizeof(PyObject*));
2608 garbage = (PyObject**)
2609 PyMem_MALLOC(slicelength*sizeof(PyObject*));
2610 if (!garbage) {
2611 PyErr_NoMemory();
2612 return -1;
2615 /* drawing pictures might help understand these for
2616 loops. Basically, we memmove the parts of the
2617 list that are *not* part of the slice: step-1
2618 items for each item that is part of the slice,
2619 and then tail end of the list that was not
2620 covered by the slice */
2621 for (cur = start, i = 0;
2622 cur < stop;
2623 cur += step, i++) {
2624 Py_ssize_t lim = step - 1;
2626 garbage[i] = PyList_GET_ITEM(self, cur);
2628 if (cur + step >= Py_SIZE(self)) {
2629 lim = Py_SIZE(self) - cur - 1;
2632 memmove(self->ob_item + cur - i,
2633 self->ob_item + cur + 1,
2634 lim * sizeof(PyObject *));
2636 cur = start + slicelength*step;
2637 if (cur < Py_SIZE(self)) {
2638 memmove(self->ob_item + cur - slicelength,
2639 self->ob_item + cur,
2640 (Py_SIZE(self) - cur) *
2641 sizeof(PyObject *));
2644 Py_SIZE(self) -= slicelength;
2645 list_resize(self, Py_SIZE(self));
2647 for (i = 0; i < slicelength; i++) {
2648 Py_DECREF(garbage[i]);
2650 PyMem_FREE(garbage);
2652 return 0;
2654 else {
2655 /* assign slice */
2656 PyObject *ins, *seq;
2657 PyObject **garbage, **seqitems, **selfitems;
2658 Py_ssize_t cur, i;
2660 /* protect against a[::-1] = a */
2661 if (self == (PyListObject*)value) {
2662 seq = list_slice((PyListObject*)value, 0,
2663 PyList_GET_SIZE(value));
2665 else {
2666 seq = PySequence_Fast(value,
2667 "must assign iterable "
2668 "to extended slice");
2670 if (!seq)
2671 return -1;
2673 if (PySequence_Fast_GET_SIZE(seq) != slicelength) {
2674 PyErr_Format(PyExc_ValueError,
2675 "attempt to assign sequence of "
2676 "size %zd to extended slice of "
2677 "size %zd",
2678 PySequence_Fast_GET_SIZE(seq),
2679 slicelength);
2680 Py_DECREF(seq);
2681 return -1;
2684 if (!slicelength) {
2685 Py_DECREF(seq);
2686 return 0;
2689 garbage = (PyObject**)
2690 PyMem_MALLOC(slicelength*sizeof(PyObject*));
2691 if (!garbage) {
2692 Py_DECREF(seq);
2693 PyErr_NoMemory();
2694 return -1;
2697 selfitems = self->ob_item;
2698 seqitems = PySequence_Fast_ITEMS(seq);
2699 for (cur = start, i = 0; i < slicelength;
2700 cur += step, i++) {
2701 garbage[i] = selfitems[cur];
2702 ins = seqitems[i];
2703 Py_INCREF(ins);
2704 selfitems[cur] = ins;
2707 for (i = 0; i < slicelength; i++) {
2708 Py_DECREF(garbage[i]);
2711 PyMem_FREE(garbage);
2712 Py_DECREF(seq);
2714 return 0;
2717 else {
2718 PyErr_Format(PyExc_TypeError,
2719 "list indices must be integers, not %.200s",
2720 item->ob_type->tp_name);
2721 return -1;
2725 static PyMappingMethods list_as_mapping = {
2726 (lenfunc)list_length,
2727 (binaryfunc)list_subscript,
2728 (objobjargproc)list_ass_subscript
2731 PyTypeObject PyList_Type = {
2732 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2733 "list",
2734 sizeof(PyListObject),
2736 (destructor)list_dealloc, /* tp_dealloc */
2737 (printfunc)list_print, /* tp_print */
2738 0, /* tp_getattr */
2739 0, /* tp_setattr */
2740 0, /* tp_compare */
2741 (reprfunc)list_repr, /* tp_repr */
2742 0, /* tp_as_number */
2743 &list_as_sequence, /* tp_as_sequence */
2744 &list_as_mapping, /* tp_as_mapping */
2745 (hashfunc)PyObject_HashNotImplemented, /* tp_hash */
2746 0, /* tp_call */
2747 0, /* tp_str */
2748 PyObject_GenericGetAttr, /* tp_getattro */
2749 0, /* tp_setattro */
2750 0, /* tp_as_buffer */
2751 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
2752 Py_TPFLAGS_BASETYPE | Py_TPFLAGS_LIST_SUBCLASS, /* tp_flags */
2753 list_doc, /* tp_doc */
2754 (traverseproc)list_traverse, /* tp_traverse */
2755 (inquiry)list_clear, /* tp_clear */
2756 list_richcompare, /* tp_richcompare */
2757 0, /* tp_weaklistoffset */
2758 list_iter, /* tp_iter */
2759 0, /* tp_iternext */
2760 list_methods, /* tp_methods */
2761 0, /* tp_members */
2762 0, /* tp_getset */
2763 0, /* tp_base */
2764 0, /* tp_dict */
2765 0, /* tp_descr_get */
2766 0, /* tp_descr_set */
2767 0, /* tp_dictoffset */
2768 (initproc)list_init, /* tp_init */
2769 PyType_GenericAlloc, /* tp_alloc */
2770 PyType_GenericNew, /* tp_new */
2771 PyObject_GC_Del, /* tp_free */
2775 /*********************** List Iterator **************************/
2777 typedef struct {
2778 PyObject_HEAD
2779 long it_index;
2780 PyListObject *it_seq; /* Set to NULL when iterator is exhausted */
2781 } listiterobject;
2783 static PyObject *list_iter(PyObject *);
2784 static void listiter_dealloc(listiterobject *);
2785 static int listiter_traverse(listiterobject *, visitproc, void *);
2786 static PyObject *listiter_next(listiterobject *);
2787 static PyObject *listiter_len(listiterobject *);
2789 PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
2791 static PyMethodDef listiter_methods[] = {
2792 {"__length_hint__", (PyCFunction)listiter_len, METH_NOARGS, length_hint_doc},
2793 {NULL, NULL} /* sentinel */
2796 PyTypeObject PyListIter_Type = {
2797 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2798 "listiterator", /* tp_name */
2799 sizeof(listiterobject), /* tp_basicsize */
2800 0, /* tp_itemsize */
2801 /* methods */
2802 (destructor)listiter_dealloc, /* tp_dealloc */
2803 0, /* tp_print */
2804 0, /* tp_getattr */
2805 0, /* tp_setattr */
2806 0, /* tp_compare */
2807 0, /* tp_repr */
2808 0, /* tp_as_number */
2809 0, /* tp_as_sequence */
2810 0, /* tp_as_mapping */
2811 0, /* tp_hash */
2812 0, /* tp_call */
2813 0, /* tp_str */
2814 PyObject_GenericGetAttr, /* tp_getattro */
2815 0, /* tp_setattro */
2816 0, /* tp_as_buffer */
2817 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2818 0, /* tp_doc */
2819 (traverseproc)listiter_traverse, /* tp_traverse */
2820 0, /* tp_clear */
2821 0, /* tp_richcompare */
2822 0, /* tp_weaklistoffset */
2823 PyObject_SelfIter, /* tp_iter */
2824 (iternextfunc)listiter_next, /* tp_iternext */
2825 listiter_methods, /* tp_methods */
2826 0, /* tp_members */
2830 static PyObject *
2831 list_iter(PyObject *seq)
2833 listiterobject *it;
2835 if (!PyList_Check(seq)) {
2836 PyErr_BadInternalCall();
2837 return NULL;
2839 it = PyObject_GC_New(listiterobject, &PyListIter_Type);
2840 if (it == NULL)
2841 return NULL;
2842 it->it_index = 0;
2843 Py_INCREF(seq);
2844 it->it_seq = (PyListObject *)seq;
2845 _PyObject_GC_TRACK(it);
2846 return (PyObject *)it;
2849 static void
2850 listiter_dealloc(listiterobject *it)
2852 _PyObject_GC_UNTRACK(it);
2853 Py_XDECREF(it->it_seq);
2854 PyObject_GC_Del(it);
2857 static int
2858 listiter_traverse(listiterobject *it, visitproc visit, void *arg)
2860 Py_VISIT(it->it_seq);
2861 return 0;
2864 static PyObject *
2865 listiter_next(listiterobject *it)
2867 PyListObject *seq;
2868 PyObject *item;
2870 assert(it != NULL);
2871 seq = it->it_seq;
2872 if (seq == NULL)
2873 return NULL;
2874 assert(PyList_Check(seq));
2876 if (it->it_index < PyList_GET_SIZE(seq)) {
2877 item = PyList_GET_ITEM(seq, it->it_index);
2878 ++it->it_index;
2879 Py_INCREF(item);
2880 return item;
2883 Py_DECREF(seq);
2884 it->it_seq = NULL;
2885 return NULL;
2888 static PyObject *
2889 listiter_len(listiterobject *it)
2891 Py_ssize_t len;
2892 if (it->it_seq) {
2893 len = PyList_GET_SIZE(it->it_seq) - it->it_index;
2894 if (len >= 0)
2895 return PyInt_FromSsize_t(len);
2897 return PyInt_FromLong(0);
2899 /*********************** List Reverse Iterator **************************/
2901 typedef struct {
2902 PyObject_HEAD
2903 Py_ssize_t it_index;
2904 PyListObject *it_seq; /* Set to NULL when iterator is exhausted */
2905 } listreviterobject;
2907 static PyObject *list_reversed(PyListObject *, PyObject *);
2908 static void listreviter_dealloc(listreviterobject *);
2909 static int listreviter_traverse(listreviterobject *, visitproc, void *);
2910 static PyObject *listreviter_next(listreviterobject *);
2911 static Py_ssize_t listreviter_len(listreviterobject *);
2913 static PySequenceMethods listreviter_as_sequence = {
2914 (lenfunc)listreviter_len, /* sq_length */
2915 0, /* sq_concat */
2918 PyTypeObject PyListRevIter_Type = {
2919 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2920 "listreverseiterator", /* tp_name */
2921 sizeof(listreviterobject), /* tp_basicsize */
2922 0, /* tp_itemsize */
2923 /* methods */
2924 (destructor)listreviter_dealloc, /* tp_dealloc */
2925 0, /* tp_print */
2926 0, /* tp_getattr */
2927 0, /* tp_setattr */
2928 0, /* tp_compare */
2929 0, /* tp_repr */
2930 0, /* tp_as_number */
2931 &listreviter_as_sequence, /* tp_as_sequence */
2932 0, /* tp_as_mapping */
2933 0, /* tp_hash */
2934 0, /* tp_call */
2935 0, /* tp_str */
2936 PyObject_GenericGetAttr, /* tp_getattro */
2937 0, /* tp_setattro */
2938 0, /* tp_as_buffer */
2939 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2940 0, /* tp_doc */
2941 (traverseproc)listreviter_traverse, /* tp_traverse */
2942 0, /* tp_clear */
2943 0, /* tp_richcompare */
2944 0, /* tp_weaklistoffset */
2945 PyObject_SelfIter, /* tp_iter */
2946 (iternextfunc)listreviter_next, /* tp_iternext */
2950 static PyObject *
2951 list_reversed(PyListObject *seq, PyObject *unused)
2953 listreviterobject *it;
2955 it = PyObject_GC_New(listreviterobject, &PyListRevIter_Type);
2956 if (it == NULL)
2957 return NULL;
2958 assert(PyList_Check(seq));
2959 it->it_index = PyList_GET_SIZE(seq) - 1;
2960 Py_INCREF(seq);
2961 it->it_seq = seq;
2962 PyObject_GC_Track(it);
2963 return (PyObject *)it;
2966 static void
2967 listreviter_dealloc(listreviterobject *it)
2969 PyObject_GC_UnTrack(it);
2970 Py_XDECREF(it->it_seq);
2971 PyObject_GC_Del(it);
2974 static int
2975 listreviter_traverse(listreviterobject *it, visitproc visit, void *arg)
2977 Py_VISIT(it->it_seq);
2978 return 0;
2981 static PyObject *
2982 listreviter_next(listreviterobject *it)
2984 PyObject *item;
2985 Py_ssize_t index = it->it_index;
2986 PyListObject *seq = it->it_seq;
2988 if (index>=0 && index < PyList_GET_SIZE(seq)) {
2989 item = PyList_GET_ITEM(seq, index);
2990 it->it_index--;
2991 Py_INCREF(item);
2992 return item;
2994 it->it_index = -1;
2995 if (seq != NULL) {
2996 it->it_seq = NULL;
2997 Py_DECREF(seq);
2999 return NULL;
3002 static Py_ssize_t
3003 listreviter_len(listreviterobject *it)
3005 Py_ssize_t len = it->it_index + 1;
3006 if (it->it_seq == NULL || PyList_GET_SIZE(it->it_seq) < len)
3007 return 0;
3008 return len;