cpython-lru_cache-weakref: Modules/itertoolsmodule.c annotate

annotate Modules/itertoolsmodule.c @ 87690:66c1c9f32567 default tip

Add use_weakref option to functools.lru_cache The use_weakref option makes lru_cache create weak references to the function arguments when creating the cache key, which prevents them from staying in the memory forever even after all other references to them are gone. This is a proof of concept for http://bugs.python.org/issue19859

author	Radomir Dopieralski <devel@sheep.art.pl>
date	Tue, 03 Dec 2013 16:02:48 +0100
parents	900bf633b7f4
children

rev	line source
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2 #include "Python.h"
33779 c66e6c9ffc2d some platforms still need offsetof() from structmember.h Fred Drake <fdrake@acm.org> parents: 33777 diff changeset	3 #include "structmember.h"
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	5 /* Itertools module written and maintained
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	6 by Raymond D. Hettinger <python@rcn.com>
85632 fa1fa88b685b Issue 18752: Make chain.from_iterable() more visible in the documentation. Raymond Hettinger <python@rcn.com> parents: 85511 diff changeset	7 Copyright (c) 2003-2013 Python Software Foundation.
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	8 All rights reserved.
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	9 */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	10
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	11
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	12 /* groupby object ***********************************************************/
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	13
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	14 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	15 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	16 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	17 PyObject *keyfunc;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	18 PyObject *tgtkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	19 PyObject *currkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	20 PyObject *currvalue;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	21 } groupbyobject;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	22
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	23 static PyTypeObject groupby_type;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	24 static PyObject _grouper_create(groupbyobject , PyObject *);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	25
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	26 static PyObject *
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	27 groupby_new(PyTypeObject type, PyObject args, PyObject *kwds)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	28 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	29 static char *kwargs[] = {"iterable", "key", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	30 groupbyobject *gbo;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	31 PyObject it, keyfunc = Py_None;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	32
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	33 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|O:groupby", kwargs,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	34 &it, &keyfunc))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	35 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	36
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	37 gbo = (groupbyobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	38 if (gbo == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	39 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	40 gbo->tgtkey = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	41 gbo->currkey = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	42 gbo->currvalue = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	43 gbo->keyfunc = keyfunc;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	44 Py_INCREF(keyfunc);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	45 gbo->it = PyObject_GetIter(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	46 if (gbo->it == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	47 Py_DECREF(gbo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	48 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	49 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	50 return (PyObject *)gbo;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	51 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	52
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	53 static void
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	54 groupby_dealloc(groupbyobject *gbo)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	55 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	56 PyObject_GC_UnTrack(gbo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	57 Py_XDECREF(gbo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	58 Py_XDECREF(gbo->keyfunc);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	59 Py_XDECREF(gbo->tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	60 Py_XDECREF(gbo->currkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	61 Py_XDECREF(gbo->currvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	62 Py_TYPE(gbo)->tp_free(gbo);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	63 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	64
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	65 static int
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	66 groupby_traverse(groupbyobject gbo, visitproc visit, void arg)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	67 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	68 Py_VISIT(gbo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	69 Py_VISIT(gbo->keyfunc);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	70 Py_VISIT(gbo->tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	71 Py_VISIT(gbo->currkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	72 Py_VISIT(gbo->currvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	73 return 0;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	74 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	75
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	76 static PyObject *
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	77 groupby_next(groupbyobject *gbo)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	78 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	79 PyObject newvalue, newkey, r, grouper, *tmp;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	80
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	81 /* skip to next iteration group */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	82 for (;;) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	83 if (gbo->currkey == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	84 /* pass */;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	85 else if (gbo->tgtkey == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	86 break;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	87 else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	88 int rcmp;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	89
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	90 rcmp = PyObject_RichCompareBool(gbo->tgtkey,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	91 gbo->currkey, Py_EQ);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	92 if (rcmp == -1)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	93 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	94 else if (rcmp == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	95 break;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	96 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	97
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	98 newvalue = PyIter_Next(gbo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	99 if (newvalue == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	100 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	101
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	102 if (gbo->keyfunc == Py_None) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	103 newkey = newvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	104 Py_INCREF(newvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	105 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	106 newkey = PyObject_CallFunctionObjArgs(gbo->keyfunc,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	107 newvalue, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	108 if (newkey == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	109 Py_DECREF(newvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	110 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	111 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	112 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	113
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	114 tmp = gbo->currkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	115 gbo->currkey = newkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	116 Py_XDECREF(tmp);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	117
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	118 tmp = gbo->currvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	119 gbo->currvalue = newvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	120 Py_XDECREF(tmp);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	121 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	122
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	123 Py_INCREF(gbo->currkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	124 tmp = gbo->tgtkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	125 gbo->tgtkey = gbo->currkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	126 Py_XDECREF(tmp);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	127
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	128 grouper = _grouper_create(gbo, gbo->tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	129 if (grouper == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	130 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	131
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	132 r = PyTuple_Pack(2, gbo->currkey, grouper);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	133 Py_DECREF(grouper);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	134 return r;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	135 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	136
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	137 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	138 groupby_reduce(groupbyobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	139 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	140 /* reduce as a 'new' call with an optional 'setstate' if groupby
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	141 * has started
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	142 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	143 PyObject *value;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	144 if (lz->tgtkey && lz->currkey && lz->currvalue)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	145 value = Py_BuildValue("O(OO)(OOO)", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	146 lz->it, lz->keyfunc, lz->currkey, lz->currvalue, lz->tgtkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	147 else
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	148 value = Py_BuildValue("O(OO)", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	149 lz->it, lz->keyfunc);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	150
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	151 return value;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	152 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	153
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	154 PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	155
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	156 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	157 groupby_setstate(groupbyobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	158 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	159 PyObject currkey, currvalue, *tgtkey;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	160 if (!PyArg_ParseTuple(state, "OOO", &currkey, &currvalue, &tgtkey))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	161 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	162 Py_CLEAR(lz->currkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	163 lz->currkey = currkey;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	164 Py_INCREF(lz->currkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	165 Py_CLEAR(lz->currvalue);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	166 lz->currvalue = currvalue;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	167 Py_INCREF(lz->currvalue);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	168 Py_CLEAR(lz->tgtkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	169 lz->tgtkey = tgtkey;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	170 Py_INCREF(lz->tgtkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	171 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	172 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	173
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	174 PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	175
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	176 static PyMethodDef groupby_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	177 {"__reduce__", (PyCFunction)groupby_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	178 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	179 {"__setstate__", (PyCFunction)groupby_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	180 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	181 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	182 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	183
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	184 PyDoc_STRVAR(groupby_doc,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	185 "groupby(iterable[, keyfunc]) -> create an iterator which returns\n\
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	186 (key, sub-iterator) grouped by each value of key(value).\n");
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	187
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	188 static PyTypeObject groupby_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	189 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	190 "itertools.groupby", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	191 sizeof(groupbyobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	192 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	193 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	194 (destructor)groupby_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	195 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	196 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	197 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	198 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	199 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	200 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	201 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	202 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	203 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	204 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	205 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	206 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	207 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	208 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	209 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	210 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	211 groupby_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	212 (traverseproc)groupby_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	213 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	214 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	215 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	216 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	217 (iternextfunc)groupby_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	218 groupby_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	219 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	220 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	221 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	222 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	223 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	224 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	225 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	226 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	227 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	228 groupby_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	229 PyObject_GC_Del, /* tp_free */
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	230 };
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	231
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	232
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	233 /* _grouper object (internal) ************************************************/
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	234
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	235 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	236 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	237 PyObject *parent;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	238 PyObject *tgtkey;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	239 } _grouperobject;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	240
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	241 static PyTypeObject _grouper_type;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	242
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	243 static PyObject *
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	244 _grouper_new(PyTypeObject type, PyObject args, PyObject *kwds)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	245 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	246 PyObject parent, tgtkey;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	247
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	248 if (!PyArg_ParseTuple(args, "O!O", &groupby_type, &parent, &tgtkey))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	249 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	250
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	251 return _grouper_create((groupbyobject*) parent, tgtkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	252 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	253
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	254 static PyObject *
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	255 _grouper_create(groupbyobject parent, PyObject tgtkey)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	256 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	257 _grouperobject *igo;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	258
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	259 igo = PyObject_GC_New(_grouperobject, &_grouper_type);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	260 if (igo == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	261 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	262 igo->parent = (PyObject *)parent;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	263 Py_INCREF(parent);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	264 igo->tgtkey = tgtkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	265 Py_INCREF(tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	266
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	267 PyObject_GC_Track(igo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	268 return (PyObject *)igo;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	269 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	270
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	271 static void
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	272 _grouper_dealloc(_grouperobject *igo)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	273 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	274 PyObject_GC_UnTrack(igo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	275 Py_DECREF(igo->parent);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	276 Py_DECREF(igo->tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	277 PyObject_GC_Del(igo);
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	278 }
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	279
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	280 static int
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	281 _grouper_traverse(_grouperobject igo, visitproc visit, void arg)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	282 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	283 Py_VISIT(igo->parent);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	284 Py_VISIT(igo->tgtkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	285 return 0;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	286 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	287
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	288 static PyObject *
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	289 _grouper_next(_grouperobject *igo)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	290 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	291 groupbyobject gbo = (groupbyobject )igo->parent;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	292 PyObject newvalue, newkey, *r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	293 int rcmp;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	294
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	295 if (gbo->currvalue == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	296 newvalue = PyIter_Next(gbo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	297 if (newvalue == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	298 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	299
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	300 if (gbo->keyfunc == Py_None) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	301 newkey = newvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	302 Py_INCREF(newvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	303 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	304 newkey = PyObject_CallFunctionObjArgs(gbo->keyfunc,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	305 newvalue, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	306 if (newkey == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	307 Py_DECREF(newvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	308 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	309 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	310 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	311
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	312 assert(gbo->currkey == NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	313 gbo->currkey = newkey;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	314 gbo->currvalue = newvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	315 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	316
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	317 assert(gbo->currkey != NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	318 rcmp = PyObject_RichCompareBool(igo->tgtkey, gbo->currkey, Py_EQ);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	319 if (rcmp <= 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	320 /* got any error or current group is end */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	321 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	322
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	323 r = gbo->currvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	324 gbo->currvalue = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	325 Py_CLEAR(gbo->currkey);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	326
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	327 return r;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	328 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	329
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	330 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	331 _grouper_reduce(_grouperobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	332 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	333 return Py_BuildValue("O(OO)", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	334 lz->parent, lz->tgtkey);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	335 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	336
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	337 static PyMethodDef _grouper_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	338 {"__reduce__", (PyCFunction)_grouper_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	339 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	340 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	341 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	342
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	343
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	344 static PyTypeObject _grouper_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	345 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	346 "itertools._grouper", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	347 sizeof(_grouperobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	348 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	349 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	350 (destructor)_grouper_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	351 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	352 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	353 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	354 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	355 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	356 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	357 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	358 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	359 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	360 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	361 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	362 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	363 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	364 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	365 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	366 0, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	367 (traverseproc)_grouper_traverse,/* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	368 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	369 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	370 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	371 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	372 (iternextfunc)_grouper_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	373 _grouper_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	374 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	375 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	376 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	377 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	378 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	379 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	380 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	381 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	382 0, /* tp_alloc */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	383 _grouper_new, /* tp_new */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	384 PyObject_GC_Del, /* tp_free */
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	385 };
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	386
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	387
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	388
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	389 /* tee object and with supporting function and objects ***************/
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	390
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	391 /* The teedataobject pre-allocates space for LINKCELLS number of objects.
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	392 To help the object fit neatly inside cache lines (space for 16 to 32
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	393 pointers), the value should be a multiple of 16 minus space for
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	394 the other structure members including PyHEAD overhead. The larger the
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	395 value, the less memory overhead per object and the less time spent
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	396 allocating/deallocating new links. The smaller the number, the less
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	397 wasted space and the more rapid freeing of older data.
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	398 */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	399 #define LINKCELLS 57
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	400
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	401 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	402 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	403 PyObject *it;
85761 900bf633b7f4 Add a comment making it explicit that itertools.tee() is already 64bit-safe (issue #19049) Antoine Pitrou <solipsis@pitrou.net> parents: 85632 diff changeset	404 int numread; /* 0 <= numread <= LINKCELLS */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	405 PyObject *nextlink;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	406 PyObject *(values[LINKCELLS]);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	407 } teedataobject;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	408
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	409 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	410 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	411 teedataobject *dataobj;
85761 900bf633b7f4 Add a comment making it explicit that itertools.tee() is already 64bit-safe (issue #19049) Antoine Pitrou <solipsis@pitrou.net> parents: 85632 diff changeset	412 int index; /* 0 <= index <= LINKCELLS */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	413 PyObject *weakreflist;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	414 } teeobject;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	415
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	416 static PyTypeObject teedataobject_type;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	417
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	418 static PyObject *
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	419 teedataobject_newinternal(PyObject *it)
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	420 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	421 teedataobject *tdo;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	422
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	423 tdo = PyObject_GC_New(teedataobject, &teedataobject_type);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	424 if (tdo == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	425 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	426
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	427 tdo->numread = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	428 tdo->nextlink = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	429 Py_INCREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	430 tdo->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	431 PyObject_GC_Track(tdo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	432 return (PyObject *)tdo;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	433 }
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	434
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	435 static PyObject *
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	436 teedataobject_jumplink(teedataobject *tdo)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	437 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	438 if (tdo->nextlink == NULL)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	439 tdo->nextlink = teedataobject_newinternal(tdo->it);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	440 Py_XINCREF(tdo->nextlink);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	441 return tdo->nextlink;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	442 }
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	443
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	444 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	445 teedataobject_getitem(teedataobject *tdo, int i)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	446 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	447 PyObject *value;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	448
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	449 assert(i < LINKCELLS);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	450 if (i < tdo->numread)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	451 value = tdo->values[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	452 else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	453 /* this is the lead iterator, so fetch more data */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	454 assert(i == tdo->numread);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	455 value = PyIter_Next(tdo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	456 if (value == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	457 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	458 tdo->numread++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	459 tdo->values[i] = value;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	460 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	461 Py_INCREF(value);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	462 return value;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	463 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	464
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	465 static int
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	466 teedataobject_traverse(teedataobject tdo, visitproc visit, void arg)
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	467 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	468 int i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	469 Py_VISIT(tdo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	470 for (i = 0; i < tdo->numread; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	471 Py_VISIT(tdo->values[i]);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	472 Py_VISIT(tdo->nextlink);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	473 return 0;
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	474 }
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	475
81700 f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	476 static void
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	477 teedataobject_safe_decref(PyObject *obj)
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	478 {
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	479 while (obj && Py_TYPE(obj) == &teedataobject_type &&
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	480 Py_REFCNT(obj) == 1) {
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	481 PyObject nextlink = ((teedataobject )obj)->nextlink;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	482 ((teedataobject *)obj)->nextlink = NULL;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	483 Py_DECREF(obj);
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	484 obj = nextlink;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	485 }
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	486 Py_XDECREF(obj);
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	487 }
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	488
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	489 static int
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	490 teedataobject_clear(teedataobject *tdo)
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	491 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	492 int i;
81700 f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	493 PyObject *tmp;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	494
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	495 Py_CLEAR(tdo->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	496 for (i=0 ; i<tdo->numread ; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	497 Py_CLEAR(tdo->values[i]);
81700 f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	498 tmp = tdo->nextlink;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	499 tdo->nextlink = NULL;
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	500 teedataobject_safe_decref(tmp);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	501 return 0;
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	502 }
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	503
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	504 static void
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	505 teedataobject_dealloc(teedataobject *tdo)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	506 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	507 PyObject_GC_UnTrack(tdo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	508 teedataobject_clear(tdo);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	509 PyObject_GC_Del(tdo);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	510 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	511
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	512 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	513 teedataobject_reduce(teedataobject *tdo)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	514 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	515 int i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	516 /* create a temporary list of already iterated values */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	517 PyObject *values = PyList_New(tdo->numread);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	518 if (!values)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	519 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	520 for (i=0 ; i<tdo->numread ; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	521 Py_INCREF(tdo->values[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	522 PyList_SET_ITEM(values, i, tdo->values[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	523 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	524 return Py_BuildValue("O(ONO)", Py_TYPE(tdo), tdo->it,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	525 values,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	526 tdo->nextlink ? tdo->nextlink : Py_None);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	527 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	528
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	529 static PyTypeObject teedataobject_type;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	530
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	531 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	532 teedataobject_new(PyTypeObject type, PyObject args, PyObject *kw)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	533 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	534 teedataobject *tdo;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	535 PyObject it, values, *next;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	536 Py_ssize_t i, len;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	537
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	538 assert(type == &teedataobject_type);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	539 if (!PyArg_ParseTuple(args, "OO!O", &it, &PyList_Type, &values, &next))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	540 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	541
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	542 tdo = (teedataobject *)teedataobject_newinternal(it);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	543 if (!tdo)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	544 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	545
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	546 len = PyList_GET_SIZE(values);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	547 if (len > LINKCELLS)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	548 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	549 for (i=0; i<len; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	550 tdo->values[i] = PyList_GET_ITEM(values, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	551 Py_INCREF(tdo->values[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	552 }
76946 d274fa027cf5 Add safecase to silence Win64 warning. Martin v. Löwis <martin@v.loewis.de> parents: 76091 diff changeset	553 /* len <= LINKCELLS < INT_MAX */
d274fa027cf5 Add safecase to silence Win64 warning. Martin v. Löwis <martin@v.loewis.de> parents: 76091 diff changeset	554 tdo->numread = Py_SAFE_DOWNCAST(len, Py_ssize_t, int);
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	555
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	556 if (len == LINKCELLS) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	557 if (next != Py_None) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	558 if (Py_TYPE(next) != &teedataobject_type)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	559 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	560 assert(tdo->nextlink == NULL);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	561 Py_INCREF(next);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	562 tdo->nextlink = next;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	563 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	564 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	565 if (next != Py_None)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	566 goto err; /* shouldn't have a next if we are not full */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	567 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	568 return (PyObject*)tdo;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	569
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	570 err:
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	571 Py_XDECREF(tdo);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	572 PyErr_SetString(PyExc_ValueError, "Invalid arguments");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	573 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	574 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	575
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	576 static PyMethodDef teedataobject_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	577 {"__reduce__", (PyCFunction)teedataobject_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	578 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	579 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	580 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	581
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	582 PyDoc_STRVAR(teedataobject_doc, "Data container common to multiple tee objects.");
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	583
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	584 static PyTypeObject teedataobject_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	585 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	586 "itertools._tee_dataobject", /* tp_name */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	587 sizeof(teedataobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	588 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	589 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	590 (destructor)teedataobject_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	591 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	592 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	593 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	594 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	595 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	596 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	597 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	598 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	599 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	600 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	601 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	602 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	603 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	604 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	605 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	606 teedataobject_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	607 (traverseproc)teedataobject_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	608 (inquiry)teedataobject_clear, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	609 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	610 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	611 0, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	612 0, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	613 teedataobject_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	614 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	615 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	616 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	617 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	618 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	619 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	620 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	621 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	622 0, /* tp_alloc */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	623 teedataobject_new, /* tp_new */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	624 PyObject_GC_Del, /* tp_free */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	625 };
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	626
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	627
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	628 static PyTypeObject tee_type;
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	629
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	630 static PyObject *
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	631 tee_next(teeobject *to)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	632 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	633 PyObject value, link;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	634
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	635 if (to->index >= LINKCELLS) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	636 link = teedataobject_jumplink(to->dataobj);
81700 f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	637 if (link == NULL)
f7e14a1af609 Issue #13454: Fix a crash when deleting an iterator created by itertools.tee() Serhiy Storchaka <storchaka@gmail.com> parents: 78604 diff changeset	638 return NULL;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	639 Py_DECREF(to->dataobj);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	640 to->dataobj = (teedataobject *)link;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	641 to->index = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	642 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	643 value = teedataobject_getitem(to->dataobj, to->index);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	644 if (value == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	645 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	646 to->index++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	647 return value;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	648 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	649
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	650 static int
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	651 tee_traverse(teeobject to, visitproc visit, void arg)
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	652 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	653 Py_VISIT((PyObject *)to->dataobj);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	654 return 0;
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	655 }
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	656
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	657 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	658 tee_copy(teeobject *to)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	659 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	660 teeobject *newto;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	661
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	662 newto = PyObject_GC_New(teeobject, &tee_type);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	663 if (newto == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	664 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	665 Py_INCREF(to->dataobj);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	666 newto->dataobj = to->dataobj;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	667 newto->index = to->index;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	668 newto->weakreflist = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	669 PyObject_GC_Track(newto);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	670 return (PyObject *)newto;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	671 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	672
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	673 PyDoc_STRVAR(teecopy_doc, "Returns an independent iterator.");
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	674
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	675 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	676 tee_fromiterable(PyObject *iterable)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	677 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	678 teeobject *to;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	679 PyObject *it = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	680
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	681 it = PyObject_GetIter(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	682 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	683 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	684 if (PyObject_TypeCheck(it, &tee_type)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	685 to = (teeobject )tee_copy((teeobject )it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	686 goto done;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	687 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	688
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	689 to = PyObject_GC_New(teeobject, &tee_type);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	690 if (to == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	691 goto done;
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	692 to->dataobj = (teedataobject *)teedataobject_newinternal(it);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	693 if (!to->dataobj) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	694 PyObject_GC_Del(to);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	695 to = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	696 goto done;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	697 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	698
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	699 to->index = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	700 to->weakreflist = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	701 PyObject_GC_Track(to);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	702 done:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	703 Py_XDECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	704 return (PyObject *)to;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	705 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	706
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	707 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	708 tee_new(PyTypeObject type, PyObject args, PyObject *kw)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	709 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	710 PyObject *iterable;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	711
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	712 if (!PyArg_UnpackTuple(args, "_tee", 1, 1, &iterable))
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	713 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	714 return tee_fromiterable(iterable);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	715 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	716
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	717 static int
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	718 tee_clear(teeobject *to)
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	719 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	720 if (to->weakreflist != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	721 PyObject_ClearWeakRefs((PyObject *) to);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	722 Py_CLEAR(to->dataobj);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	723 return 0;
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	724 }
f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	725
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	726 static void
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	727 tee_dealloc(teeobject *to)
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	728 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	729 PyObject_GC_UnTrack(to);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	730 tee_clear(to);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	731 PyObject_GC_Del(to);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	732 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	733
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	734 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	735 tee_reduce(teeobject *to)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	736 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	737 return Py_BuildValue("O(())(Oi)", Py_TYPE(to), to->dataobj, to->index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	738 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	739
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	740 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	741 tee_setstate(teeobject to, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	742 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	743 teedataobject *tdo;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	744 int index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	745 if (!PyArg_ParseTuple(state, "O!i", &teedataobject_type, &tdo, &index))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	746 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	747 if (index < 0 \|\| index > LINKCELLS) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	748 PyErr_SetString(PyExc_ValueError, "Index out of range");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	749 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	750 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	751 Py_CLEAR(to->dataobj);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	752 to->dataobj = tdo;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	753 Py_INCREF(to->dataobj);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	754 to->index = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	755 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	756 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	757
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	758 PyDoc_STRVAR(teeobject_doc,
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	759 "Iterator wrapped to make it copyable");
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	760
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	761 static PyMethodDef tee_methods[] = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	762 {"__copy__", (PyCFunction)tee_copy, METH_NOARGS, teecopy_doc},
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	763 {"__reduce__", (PyCFunction)tee_reduce, METH_NOARGS, reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	764 {"__setstate__", (PyCFunction)tee_setstate, METH_O, setstate_doc},
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	765 {NULL, NULL} /* sentinel */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	766 };
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	767
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	768 static PyTypeObject tee_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	769 PyVarObject_HEAD_INIT(NULL, 0)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	770 "itertools._tee", /* tp_name */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	771 sizeof(teeobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	772 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	773 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	774 (destructor)tee_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	775 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	776 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	777 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	778 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	779 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	780 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	781 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	782 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	783 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	784 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	785 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	786 0, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	787 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	788 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	789 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	790 teeobject_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	791 (traverseproc)tee_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	792 (inquiry)tee_clear, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	793 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	794 offsetof(teeobject, weakreflist), /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	795 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	796 (iternextfunc)tee_next, /* tp_iternext */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	797 tee_methods, /* tp_methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	798 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	799 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	800 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	801 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	802 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	803 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	804 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	805 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	806 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	807 tee_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	808 PyObject_GC_Del, /* tp_free */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	809 };
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	810
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	811 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	812 tee(PyObject self, PyObject args)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	813 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	814 Py_ssize_t i, n=2;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	815 PyObject it, iterable, copyable, result;
72926 db4aa878cea2 Port SetAttrString/HasAttrString to SetAttrId/GetAttrId. Martin v. Löwis <martin@v.loewis.de> parents: 72925 diff changeset	816 _Py_IDENTIFIER(__copy__);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	817
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	818 if (!PyArg_ParseTuple(args, "O\|n", &iterable, &n))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	819 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	820 if (n < 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	821 PyErr_SetString(PyExc_ValueError, "n must be >= 0");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	822 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	823 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	824 result = PyTuple_New(n);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	825 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	826 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	827 if (n == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	828 return result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	829 it = PyObject_GetIter(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	830 if (it == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	831 Py_DECREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	832 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	833 }
72926 db4aa878cea2 Port SetAttrString/HasAttrString to SetAttrId/GetAttrId. Martin v. Löwis <martin@v.loewis.de> parents: 72925 diff changeset	834 if (!_PyObject_HasAttrId(it, &PyId___copy__)) {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	835 copyable = tee_fromiterable(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	836 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	837 if (copyable == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	838 Py_DECREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	839 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	840 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	841 } else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	842 copyable = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	843 PyTuple_SET_ITEM(result, 0, copyable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	844 for (i=1 ; i<n ; i++) {
72835 5e33219492bc Add API for static strings, primarily good for identifiers. Martin v. Löwis <martin@v.loewis.de> parents: 72460 diff changeset	845
5e33219492bc Add API for static strings, primarily good for identifiers. Martin v. Löwis <martin@v.loewis.de> parents: 72460 diff changeset	846 copyable = _PyObject_CallMethodId(copyable, &PyId___copy__, NULL);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	847 if (copyable == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	848 Py_DECREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	849 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	850 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	851 PyTuple_SET_ITEM(result, i, copyable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	852 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	853 return result;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	854 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	855
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	856 PyDoc_STRVAR(tee_doc,
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	857 "tee(iterable, n=2) --> tuple of n independent iterators.");
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	858
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	859
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	860 /* cycle object **********************************************************/
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	861
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	862 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	863 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	864 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	865 PyObject *saved;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	866 int firstpass;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	867 } cycleobject;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	868
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	869 static PyTypeObject cycle_type;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	870
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	871 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	872 cycle_new(PyTypeObject type, PyObject args, PyObject *kwds)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	873 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	874 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	875 PyObject *iterable;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	876 PyObject *saved;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	877 cycleobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	878
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	879 if (type == &cycle_type && !_PyArg_NoKeywords("cycle()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	880 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	881
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	882 if (!PyArg_UnpackTuple(args, "cycle", 1, 1, &iterable))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	883 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	884
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	885 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	886 it = PyObject_GetIter(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	887 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	888 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	889
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	890 saved = PyList_New(0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	891 if (saved == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	892 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	893 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	894 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	895
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	896 /* create cycleobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	897 lz = (cycleobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	898 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	899 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	900 Py_DECREF(saved);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	901 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	902 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	903 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	904 lz->saved = saved;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	905 lz->firstpass = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	906
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	907 return (PyObject *)lz;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	908 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	909
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	910 static void
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	911 cycle_dealloc(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	912 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	913 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	914 Py_XDECREF(lz->saved);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	915 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	916 Py_TYPE(lz)->tp_free(lz);
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	917 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	918
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	919 static int
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	920 cycle_traverse(cycleobject lz, visitproc visit, void arg)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	921 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	922 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	923 Py_VISIT(lz->saved);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	924 return 0;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	925 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	926
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	927 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	928 cycle_next(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	929 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	930 PyObject *item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	931 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	932 PyObject *tmp;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	933
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	934 while (1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	935 item = PyIter_Next(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	936 if (item != NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	937 if (!lz->firstpass && PyList_Append(lz->saved, item)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	938 Py_DECREF(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	939 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	940 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	941 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	942 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	943 if (PyErr_Occurred()) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	944 if (PyErr_ExceptionMatches(PyExc_StopIteration))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	945 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	946 else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	947 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	948 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	949 if (PyList_Size(lz->saved) == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	950 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	951 it = PyObject_GetIter(lz->saved);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	952 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	953 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	954 tmp = lz->it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	955 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	956 lz->firstpass = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	957 Py_DECREF(tmp);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	958 }
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	959 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	960
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	961 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	962 cycle_reduce(cycleobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	963 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	964 /* Create a new cycle with the iterator tuple, then set
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	965 * the saved state on it.
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	966 */
81705 20c4d55f4b69 Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81701 81704 diff changeset	967 return Py_BuildValue("O(O)(Oi)", Py_TYPE(lz),
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	968 lz->it, lz->saved, lz->firstpass);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	969 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	970
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	971 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	972 cycle_setstate(cycleobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	973 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	974 PyObject *saved=NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	975 int firstpass;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	976 if (!PyArg_ParseTuple(state, "Oi", &saved, &firstpass))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	977 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	978 Py_CLEAR(lz->saved);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	979 lz->saved = saved;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	980 Py_XINCREF(lz->saved);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	981 lz->firstpass = firstpass != 0;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	982 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	983 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	984
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	985 static PyMethodDef cycle_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	986 {"__reduce__", (PyCFunction)cycle_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	987 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	988 {"__setstate__", (PyCFunction)cycle_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	989 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	990 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	991 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	992
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	993 PyDoc_STRVAR(cycle_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	994 "cycle(iterable) --> cycle object\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	995 \n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	996 Return elements from the iterable until it is exhausted.\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	997 Then repeat the sequence indefinitely.");
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	998
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	999 static PyTypeObject cycle_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1000 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1001 "itertools.cycle", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1002 sizeof(cycleobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1003 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1004 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1005 (destructor)cycle_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1006 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1007 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1008 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1009 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1010 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1011 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1012 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1013 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1014 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1015 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1016 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1017 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1018 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1019 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1020 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1021 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1022 cycle_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1023 (traverseproc)cycle_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1024 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1025 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1026 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1027 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1028 (iternextfunc)cycle_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1029 cycle_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1030 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1031 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1032 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1033 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1034 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1035 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1036 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1037 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1038 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1039 cycle_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1040 PyObject_GC_Del, /* tp_free */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1041 };
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1042
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1043
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1044 /* dropwhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1045
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1046 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1047 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1048 PyObject *func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1049 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1050 long start;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1051 } dropwhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1052
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1053 static PyTypeObject dropwhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1054
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1055 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1056 dropwhile_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1057 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1058 PyObject func, seq;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1059 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1060 dropwhileobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1061
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1062 if (type == &dropwhile_type && !_PyArg_NoKeywords("dropwhile()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1063 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1064
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1065 if (!PyArg_UnpackTuple(args, "dropwhile", 2, 2, &func, &seq))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1066 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1067
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1068 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1069 it = PyObject_GetIter(seq);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1070 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1071 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1072
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1073 /* create dropwhileobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1074 lz = (dropwhileobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1075 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1076 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1077 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1078 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1079 Py_INCREF(func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1080 lz->func = func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1081 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1082 lz->start = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1083
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1084 return (PyObject *)lz;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1085 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1086
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1087 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1088 dropwhile_dealloc(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1089 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1090 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1091 Py_XDECREF(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1092 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1093 Py_TYPE(lz)->tp_free(lz);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1094 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1095
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1096 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1097 dropwhile_traverse(dropwhileobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1098 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1099 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1100 Py_VISIT(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1101 return 0;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1102 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1103
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1104 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1105 dropwhile_next(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1106 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1107 PyObject item, good;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1108 PyObject *it = lz->it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1109 long ok;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1110 PyObject (iternext)(PyObject *);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1111
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1112 iternext = *Py_TYPE(it)->tp_iternext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1113 for (;;) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1114 item = iternext(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1115 if (item == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1116 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1117 if (lz->start == 1)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1118 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1119
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1120 good = PyObject_CallFunctionObjArgs(lz->func, item, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1121 if (good == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1122 Py_DECREF(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1123 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1124 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1125 ok = PyObject_IsTrue(good);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1126 Py_DECREF(good);
78604 56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	1127 if (ok == 0) {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1128 lz->start = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1129 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1130 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1131 Py_DECREF(item);
78604 56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	1132 if (ok < 0)
56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	1133 return NULL;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1134 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1135 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1136
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1137 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1138 dropwhile_reduce(dropwhileobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1139 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1140 return Py_BuildValue("O(OO)l", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1141 lz->func, lz->it, lz->start);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1142 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1143
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1144 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1145 dropwhile_setstate(dropwhileobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1146 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1147 int start = PyObject_IsTrue(state);
78605 b878df1d23b1 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 76946 78604 diff changeset	1148 if (start < 0)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1149 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1150 lz->start = start;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1151 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1152 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1153
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1154 static PyMethodDef dropwhile_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1155 {"__reduce__", (PyCFunction)dropwhile_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1156 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1157 {"__setstate__", (PyCFunction)dropwhile_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1158 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1159 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1160 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1161
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1162 PyDoc_STRVAR(dropwhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1163 "dropwhile(predicate, iterable) --> dropwhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1164 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1165 Drop items from the iterable while predicate(item) is true.\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1166 Afterwards, return every element until the iterable is exhausted.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1167
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1168 static PyTypeObject dropwhile_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1169 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1170 "itertools.dropwhile", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1171 sizeof(dropwhileobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1172 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1173 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1174 (destructor)dropwhile_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1175 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1176 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1177 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1178 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1179 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1180 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1181 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1182 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1183 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1184 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1185 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1186 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1187 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1188 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1189 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1190 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1191 dropwhile_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1192 (traverseproc)dropwhile_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1193 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1194 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1195 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1196 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1197 (iternextfunc)dropwhile_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1198 dropwhile_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1199 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1200 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1201 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1202 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1203 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1204 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1205 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1206 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1207 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1208 dropwhile_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1209 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1210 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1211
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1212
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1213 /* takewhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1214
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1215 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1216 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1217 PyObject *func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1218 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1219 long stop;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1220 } takewhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1221
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1222 static PyTypeObject takewhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1223
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1224 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1225 takewhile_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1226 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1227 PyObject func, seq;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1228 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1229 takewhileobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1230
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1231 if (type == &takewhile_type && !_PyArg_NoKeywords("takewhile()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1232 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1233
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1234 if (!PyArg_UnpackTuple(args, "takewhile", 2, 2, &func, &seq))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1235 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1236
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1237 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1238 it = PyObject_GetIter(seq);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1239 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1240 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1241
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1242 /* create takewhileobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1243 lz = (takewhileobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1244 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1245 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1246 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1247 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1248 Py_INCREF(func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1249 lz->func = func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1250 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1251 lz->stop = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1252
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1253 return (PyObject *)lz;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1254 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1255
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1256 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1257 takewhile_dealloc(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1258 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1259 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1260 Py_XDECREF(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1261 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1262 Py_TYPE(lz)->tp_free(lz);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1263 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1264
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1265 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1266 takewhile_traverse(takewhileobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1267 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1268 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1269 Py_VISIT(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1270 return 0;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1271 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1272
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1273 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1274 takewhile_next(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1275 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1276 PyObject item, good;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1277 PyObject *it = lz->it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1278 long ok;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1279
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1280 if (lz->stop == 1)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1281 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1282
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1283 item = (*Py_TYPE(it)->tp_iternext)(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1284 if (item == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1285 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1286
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1287 good = PyObject_CallFunctionObjArgs(lz->func, item, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1288 if (good == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1289 Py_DECREF(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1290 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1291 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1292 ok = PyObject_IsTrue(good);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1293 Py_DECREF(good);
78605 b878df1d23b1 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 76946 78604 diff changeset	1294 if (ok == 1)
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1295 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1296 Py_DECREF(item);
78604 56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	1297 if (ok == 0)
56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	1298 lz->stop = 1;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1299 return NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1300 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1301
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1302 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1303 takewhile_reduce(takewhileobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1304 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1305 return Py_BuildValue("O(OO)l", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1306 lz->func, lz->it, lz->stop);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1307 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1308
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1309 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1310 takewhile_reduce_setstate(takewhileobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1311 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1312 int stop = PyObject_IsTrue(state);
78605 b878df1d23b1 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 76946 78604 diff changeset	1313 if (stop < 0)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1314 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1315 lz->stop = stop;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1316 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1317 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1318
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1319 static PyMethodDef takewhile_reduce_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1320 {"__reduce__", (PyCFunction)takewhile_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1321 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1322 {"__setstate__", (PyCFunction)takewhile_reduce_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1323 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1324 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1325 };
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1326 PyDoc_STRVAR(takewhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1327 "takewhile(predicate, iterable) --> takewhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1328 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1329 Return successive entries from an iterable as long as the \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1330 predicate evaluates to true for each entry.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1331
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1332 static PyTypeObject takewhile_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1333 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1334 "itertools.takewhile", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1335 sizeof(takewhileobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1336 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1337 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1338 (destructor)takewhile_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1339 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1340 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1341 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1342 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1343 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1344 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1345 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1346 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1347 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1348 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1349 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1350 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1351 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1352 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1353 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1354 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1355 takewhile_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1356 (traverseproc)takewhile_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1357 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1358 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1359 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1360 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1361 (iternextfunc)takewhile_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1362 takewhile_reduce_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1363 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1364 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1365 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1366 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1367 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1368 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1369 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1370 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1371 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1372 takewhile_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1373 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1374 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1375
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1376
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1377 /* islice object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1378
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1379 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1380 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1381 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1382 Py_ssize_t next;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1383 Py_ssize_t stop;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1384 Py_ssize_t step;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1385 Py_ssize_t cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1386 } isliceobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1387
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1388 static PyTypeObject islice_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1389
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1390 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1391 islice_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1392 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1393 PyObject *seq;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1394 Py_ssize_t start=0, stop=-1, step=1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1395 PyObject it, a1=NULL, a2=NULL, a3=NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1396 Py_ssize_t numargs;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1397 isliceobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1398
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1399 if (type == &islice_type && !_PyArg_NoKeywords("islice()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1400 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1401
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1402 if (!PyArg_UnpackTuple(args, "islice", 2, 4, &seq, &a1, &a2, &a3))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1403 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1404
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1405 numargs = PyTuple_Size(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1406 if (numargs == 2) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1407 if (a1 != Py_None) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1408 stop = PyLong_AsSsize_t(a1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1409 if (stop == -1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1410 if (PyErr_Occurred())
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1411 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1412 PyErr_SetString(PyExc_ValueError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1413 "Stop argument for islice() must be None or an integer: 0 <= x <= sys.maxsize.");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1414 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1415 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1416 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1417 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1418 if (a1 != Py_None)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1419 start = PyLong_AsSsize_t(a1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1420 if (start == -1 && PyErr_Occurred())
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1421 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1422 if (a2 != Py_None) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1423 stop = PyLong_AsSsize_t(a2);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1424 if (stop == -1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1425 if (PyErr_Occurred())
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1426 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1427 PyErr_SetString(PyExc_ValueError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1428 "Stop argument for islice() must be None or an integer: 0 <= x <= sys.maxsize.");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1429 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1430 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1431 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1432 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1433 if (start<0 \|\| stop<-1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1434 PyErr_SetString(PyExc_ValueError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1435 "Indices for islice() must be None or an integer: 0 <= x <= sys.maxsize.");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1436 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1437 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1438
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1439 if (a3 != NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1440 if (a3 != Py_None)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1441 step = PyLong_AsSsize_t(a3);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1442 if (step == -1 && PyErr_Occurred())
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1443 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1444 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1445 if (step<1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1446 PyErr_SetString(PyExc_ValueError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1447 "Step for islice() must be a positive integer or None.");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1448 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1449 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1450
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1451 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1452 it = PyObject_GetIter(seq);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1453 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1454 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1455
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1456 /* create isliceobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1457 lz = (isliceobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1458 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1459 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1460 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1461 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1462 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1463 lz->next = start;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1464 lz->stop = stop;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1465 lz->step = step;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1466 lz->cnt = 0L;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1467
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1468 return (PyObject *)lz;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1469 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1470
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1471 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1472 islice_dealloc(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1473 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1474 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1475 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1476 Py_TYPE(lz)->tp_free(lz);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1477 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1478
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1479 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1480 islice_traverse(isliceobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1481 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1482 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1483 return 0;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1484 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1485
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1486 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1487 islice_next(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1488 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1489 PyObject *item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1490 PyObject *it = lz->it;
66546 6799d54db378 Issue #10323: Predictable final state for slice(). Raymond Hettinger <python@rcn.com> parents: 62239 diff changeset	1491 Py_ssize_t stop = lz->stop;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1492 Py_ssize_t oldnext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1493 PyObject (iternext)(PyObject *);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1494
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1495 iternext = *Py_TYPE(it)->tp_iternext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1496 while (lz->cnt < lz->next) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1497 item = iternext(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1498 if (item == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1499 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1500 Py_DECREF(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1501 lz->cnt++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1502 }
66546 6799d54db378 Issue #10323: Predictable final state for slice(). Raymond Hettinger <python@rcn.com> parents: 62239 diff changeset	1503 if (stop != -1 && lz->cnt >= stop)
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1504 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1505 item = iternext(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1506 if (item == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1507 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1508 lz->cnt++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1509 oldnext = lz->next;
72459 b378864d8ff3 Issue #12973: Fix itertools bug caused by signed integer overflow. Thanks Stefan Krah. Mark Dickinson <mdickinson@enthought.com> parents: 69063 diff changeset	1510 /* The (size_t) cast below avoids the danger of undefined
b378864d8ff3 Issue #12973: Fix itertools bug caused by signed integer overflow. Thanks Stefan Krah. Mark Dickinson <mdickinson@enthought.com> parents: 69063 diff changeset	1511 behaviour from signed integer overflow. */
b378864d8ff3 Issue #12973: Fix itertools bug caused by signed integer overflow. Thanks Stefan Krah. Mark Dickinson <mdickinson@enthought.com> parents: 69063 diff changeset	1512 lz->next += (size_t)lz->step;
66546 6799d54db378 Issue #10323: Predictable final state for slice(). Raymond Hettinger <python@rcn.com> parents: 62239 diff changeset	1513 if (lz->next < oldnext \|\| (stop != -1 && lz->next > stop))
6799d54db378 Issue #10323: Predictable final state for slice(). Raymond Hettinger <python@rcn.com> parents: 62239 diff changeset	1514 lz->next = stop;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1515 return item;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1516 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1517
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1518 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1519 islice_reduce(isliceobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1520 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1521 /* When unpickled, generate a new object with the same bounds,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1522 * then 'setstate' with the next and count
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1523 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1524 PyObject *stop;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1525 if (lz->stop == -1) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1526 stop = Py_None;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1527 Py_INCREF(stop);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1528 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1529 stop = PyLong_FromSsize_t(lz->stop);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1530 if (stop == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1531 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1532 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1533 return Py_BuildValue("O(OnNn)n", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1534 lz->it, lz->next, stop, lz->step,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1535 lz->cnt);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1536 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1537
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1538 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1539 islice_setstate(isliceobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1540 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1541 Py_ssize_t cnt = PyLong_AsSsize_t(state);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1542 if (cnt == -1 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1543 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1544 lz->cnt = cnt;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1545 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1546 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1547
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1548 static PyMethodDef islice_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1549 {"__reduce__", (PyCFunction)islice_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1550 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1551 {"__setstate__", (PyCFunction)islice_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1552 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1553 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1554 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1555
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1556 PyDoc_STRVAR(islice_doc,
84263 7ecca1a98220 Closes #18220: expand itertools.islice docstring to 2 lines Andrew Kuchling <amk@amk.ca> parents: 84262 diff changeset	1557 "islice(iterable, stop) --> islice object\n\
7ecca1a98220 Closes #18220: expand itertools.islice docstring to 2 lines Andrew Kuchling <amk@amk.ca> parents: 84262 diff changeset	1558 islice(iterable, start, stop[, step]) --> islice object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1559 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1560 Return an iterator whose next() method returns selected values from an\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1561 iterable. If start is specified, will skip all preceding elements;\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1562 otherwise, start defaults to zero. Step defaults to one. If\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1563 specified as another value, step determines how many values are \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1564 skipped between successive calls. Works like a slice() on a list\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1565 but returns an iterator.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1566
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1567 static PyTypeObject islice_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1568 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1569 "itertools.islice", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1570 sizeof(isliceobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1571 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1572 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1573 (destructor)islice_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1574 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1575 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1576 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1577 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1578 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1579 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1580 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1581 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1582 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1583 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1584 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1585 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1586 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1587 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1588 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1589 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1590 islice_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1591 (traverseproc)islice_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1592 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1593 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1594 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1595 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1596 (iternextfunc)islice_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1597 islice_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1598 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1599 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1600 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1601 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1602 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1603 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1604 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1605 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1606 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1607 islice_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1608 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1609 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1610
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1611
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1612 /* starmap object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1613
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1614 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1615 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1616 PyObject *func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1617 PyObject *it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1618 } starmapobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1619
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1620 static PyTypeObject starmap_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1621
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1622 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1623 starmap_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1624 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1625 PyObject func, seq;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1626 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1627 starmapobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1628
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1629 if (type == &starmap_type && !_PyArg_NoKeywords("starmap()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1630 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1631
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1632 if (!PyArg_UnpackTuple(args, "starmap", 2, 2, &func, &seq))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1633 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1634
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1635 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1636 it = PyObject_GetIter(seq);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1637 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1638 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1639
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1640 /* create starmapobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1641 lz = (starmapobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1642 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1643 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1644 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1645 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1646 Py_INCREF(func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1647 lz->func = func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1648 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1649
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1650 return (PyObject *)lz;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1651 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1652
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1653 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1654 starmap_dealloc(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1655 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1656 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1657 Py_XDECREF(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1658 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1659 Py_TYPE(lz)->tp_free(lz);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1660 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1661
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1662 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1663 starmap_traverse(starmapobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1664 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1665 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1666 Py_VISIT(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1667 return 0;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1668 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1669
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1670 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1671 starmap_next(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1672 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1673 PyObject *args;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1674 PyObject *result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1675 PyObject *it = lz->it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1676
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1677 args = (*Py_TYPE(it)->tp_iternext)(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1678 if (args == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1679 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1680 if (!PyTuple_CheckExact(args)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1681 PyObject *newargs = PySequence_Tuple(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1682 Py_DECREF(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1683 if (newargs == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1684 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1685 args = newargs;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1686 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1687 result = PyObject_Call(lz->func, args, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1688 Py_DECREF(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1689 return result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1690 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1691
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1692 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1693 starmap_reduce(starmapobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1694 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1695 /* Just pickle the iterator */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1696 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->func, lz->it);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1697 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1698
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1699 static PyMethodDef starmap_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1700 {"__reduce__", (PyCFunction)starmap_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1701 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1702 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1703 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1704
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1705 PyDoc_STRVAR(starmap_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1706 "starmap(function, sequence) --> starmap object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1707 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1708 Return an iterator whose values are returned from the function evaluated\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1709 with a argument tuple taken from the given sequence.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1710
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1711 static PyTypeObject starmap_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1712 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1713 "itertools.starmap", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1714 sizeof(starmapobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1715 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1716 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1717 (destructor)starmap_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1718 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1719 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1720 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1721 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1722 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1723 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1724 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1725 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1726 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1727 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1728 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1729 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1730 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1731 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1732 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1733 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1734 starmap_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1735 (traverseproc)starmap_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1736 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1737 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1738 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1739 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1740 (iternextfunc)starmap_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1741 starmap_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1742 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1743 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1744 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1745 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1746 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1747 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1748 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1749 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1750 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1751 starmap_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1752 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1753 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1754
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1755
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1756 /* chain object ************************************************************/
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1757
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1758 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1759 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1760 PyObject source; / Iterator over input iterables */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1761 PyObject active; / Currently running input iterator */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1762 } chainobject;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1763
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1764 static PyTypeObject chain_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1765
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1766 static PyObject *
45569 937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1767 chain_new_internal(PyTypeObject type, PyObject source)
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1768 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1769 chainobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1770
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1771 lz = (chainobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1772 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1773 Py_DECREF(source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1774 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1775 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1776
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1777 lz->source = source;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1778 lz->active = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1779 return (PyObject *)lz;
45569 937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1780 }
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1781
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1782 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1783 chain_new(PyTypeObject type, PyObject args, PyObject *kwds)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1784 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1785 PyObject *source;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1786
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1787 if (type == &chain_type && !_PyArg_NoKeywords("chain()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1788 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1789
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1790 source = PyObject_GetIter(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1791 if (source == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1792 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1793
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1794 return chain_new_internal(type, source);
45569 937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1795 }
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1796
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1797 static PyObject *
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1798 chain_new_from_iterable(PyTypeObject type, PyObject arg)
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1799 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1800 PyObject *source;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1801
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1802 source = PyObject_GetIter(arg);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1803 if (source == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1804 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1805
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1806 return chain_new_internal(type, source);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1807 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1808
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1809 static void
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1810 chain_dealloc(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1811 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1812 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1813 Py_XDECREF(lz->active);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1814 Py_XDECREF(lz->source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1815 Py_TYPE(lz)->tp_free(lz);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1816 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1817
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1818 static int
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1819 chain_traverse(chainobject lz, visitproc visit, void arg)
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1820 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1821 Py_VISIT(lz->source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1822 Py_VISIT(lz->active);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1823 return 0;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1824 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1825
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1826 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1827 chain_next(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1828 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1829 PyObject *item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1830
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1831 if (lz->source == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1832 return NULL; /* already stopped */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1833
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1834 if (lz->active == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1835 PyObject *iterable = PyIter_Next(lz->source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1836 if (iterable == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1837 Py_CLEAR(lz->source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1838 return NULL; /* no more input sources */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1839 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1840 lz->active = PyObject_GetIter(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1841 Py_DECREF(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1842 if (lz->active == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1843 Py_CLEAR(lz->source);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1844 return NULL; /* input not iterable */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1845 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1846 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1847 item = PyIter_Next(lz->active);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1848 if (item != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1849 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1850 if (PyErr_Occurred()) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1851 if (PyErr_ExceptionMatches(PyExc_StopIteration))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1852 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1853 else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1854 return NULL; /* input raised an exception */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1855 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1856 Py_CLEAR(lz->active);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1857 return chain_next(lz); /* recurse and use next active */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1858 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1859
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1860 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1861 chain_reduce(chainobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1862 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1863 if (lz->source) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1864 /* we can't pickle function objects (itertools.from_iterable) so
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1865 * we must use setstate to replace the iterable. One day we
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1866 * will fix pickling of functions
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1867 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1868 if (lz->active) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1869 return Py_BuildValue("O()(OO)", Py_TYPE(lz), lz->source, lz->active);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1870 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1871 return Py_BuildValue("O()(O)", Py_TYPE(lz), lz->source);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1872 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1873 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1874 return Py_BuildValue("O()", Py_TYPE(lz)); /* exhausted */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1875 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1876 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1877 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1878
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1879 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1880 chain_setstate(chainobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1881 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1882 PyObject source, active=NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1883 if (! PyArg_ParseTuple(state, "O\|O", &source, &active))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1884 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1885
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1886 Py_CLEAR(lz->source);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1887 lz->source = source;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1888 Py_INCREF(lz->source);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1889 Py_CLEAR(lz->active);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1890 lz->active = active;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1891 Py_XINCREF(lz->active);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1892 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1893 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1894
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1895 PyDoc_STRVAR(chain_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1896 "chain(*iterables) --> chain object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1897 \n\
41501 0c5019afc55b PEP 3114: rename .next() to .__next__() and add next() builtin. Georg Brandl <georg@python.org> parents: 40893 diff changeset	1898 Return a chain object whose .__next__() method returns elements from the\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1899 first iterable until it is exhausted, then elements from the next\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1900 iterable, until all of the iterables are exhausted.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1901
45569 937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1902 PyDoc_STRVAR(chain_from_iterable_doc,
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1903 "chain.from_iterable(iterable) --> chain object\n\
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1904 \n\
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1905 Alternate chain() contructor taking a single iterable argument\n\
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1906 that evaluates lazily.");
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1907
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1908 static PyMethodDef chain_methods[] = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1909 {"from_iterable", (PyCFunction) chain_new_from_iterable, METH_O \| METH_CLASS,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1910 chain_from_iterable_doc},
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1911 {"__reduce__", (PyCFunction)chain_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1912 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1913 {"__setstate__", (PyCFunction)chain_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	1914 setstate_doc},
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1915 {NULL, NULL} /* sentinel */
45569 937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1916 };
937a3c5af691 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61083,61085,61092-61097,61103-61104,61110-61112,61114-61115,61117-61125 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45548 diff changeset	1917
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1918 static PyTypeObject chain_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1919 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1920 "itertools.chain", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1921 sizeof(chainobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1922 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1923 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1924 (destructor)chain_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1925 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1926 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1927 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1928 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1929 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1930 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1931 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1932 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1933 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1934 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1935 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1936 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1937 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1938 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1939 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1940 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1941 chain_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1942 (traverseproc)chain_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1943 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1944 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1945 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1946 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1947 (iternextfunc)chain_next, /* tp_iternext */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1948 chain_methods, /* tp_methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1949 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1950 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1951 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1952 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1953 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1954 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1955 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1956 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1957 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1958 chain_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1959 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1960 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1961
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1962
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1963 /* product object ************************************************************/
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1964
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1965 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1966 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1967 PyObject pools; / tuple of pool tuples */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1968 Py_ssize_t indices; / one index per pool */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1969 PyObject result; / most recently returned result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1970 int stopped; /* set to 1 when the product iterator is exhausted */
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1971 } productobject;
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1972
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1973 static PyTypeObject product_type;
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1974
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1975 static PyObject *
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1976 product_new(PyTypeObject type, PyObject args, PyObject *kwds)
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	1977 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1978 productobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1979 Py_ssize_t nargs, npools, repeat=1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1980 PyObject *pools = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1981 Py_ssize_t *indices = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1982 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1983
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1984 if (kwds != NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1985 char *kwlist[] = {"repeat", 0};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1986 PyObject *tmpargs = PyTuple_New(0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1987 if (tmpargs == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1988 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1989 if (!PyArg_ParseTupleAndKeywords(tmpargs, kwds, "\|n:product", kwlist, &repeat)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1990 Py_DECREF(tmpargs);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1991 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1992 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1993 Py_DECREF(tmpargs);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1994 if (repeat < 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1995 PyErr_SetString(PyExc_ValueError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1996 "repeat argument cannot be negative");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1997 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1998 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	1999 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2000
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2001 assert(PyTuple_Check(args));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2002 nargs = (repeat == 0) ? 0 : PyTuple_GET_SIZE(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2003 npools = nargs * repeat;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2004
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2005 indices = PyMem_Malloc(npools * sizeof(Py_ssize_t));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2006 if (indices == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2007 PyErr_NoMemory();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2008 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2009 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2010
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2011 pools = PyTuple_New(npools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2012 if (pools == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2013 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2014
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2015 for (i=0; i < nargs ; ++i) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2016 PyObject *item = PyTuple_GET_ITEM(args, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2017 PyObject *pool = PySequence_Tuple(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2018 if (pool == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2019 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2020 PyTuple_SET_ITEM(pools, i, pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2021 indices[i] = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2022 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2023 for ( ; i < npools; ++i) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2024 PyObject *pool = PyTuple_GET_ITEM(pools, i - nargs);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2025 Py_INCREF(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2026 PyTuple_SET_ITEM(pools, i, pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2027 indices[i] = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2028 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2029
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2030 /* create productobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2031 lz = (productobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2032 if (lz == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2033 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2034
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2035 lz->pools = pools;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2036 lz->indices = indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2037 lz->result = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2038 lz->stopped = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2039
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2040 return (PyObject *)lz;
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2041
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2042 error:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2043 if (indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2044 PyMem_Free(indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2045 Py_XDECREF(pools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2046 return NULL;
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2047 }
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2048
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2049 static void
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2050 product_dealloc(productobject *lz)
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2051 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2052 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2053 Py_XDECREF(lz->pools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2054 Py_XDECREF(lz->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2055 if (lz->indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2056 PyMem_Free(lz->indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2057 Py_TYPE(lz)->tp_free(lz);
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2058 }
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2059
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2060 static int
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2061 product_traverse(productobject lz, visitproc visit, void arg)
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2062 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2063 Py_VISIT(lz->pools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2064 Py_VISIT(lz->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2065 return 0;
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2066 }
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2067
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2068 static PyObject *
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2069 product_next(productobject *lz)
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2070 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2071 PyObject *pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2072 PyObject *elem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2073 PyObject *oldelem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2074 PyObject *pools = lz->pools;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2075 PyObject *result = lz->result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2076 Py_ssize_t npools = PyTuple_GET_SIZE(pools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2077 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2078
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2079 if (lz->stopped)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2080 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2081
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2082 if (result == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2083 /* On the first pass, return an initial tuple filled with the
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2084 first element from each pool. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2085 result = PyTuple_New(npools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2086 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2087 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2088 lz->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2089 for (i=0; i < npools; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2090 pool = PyTuple_GET_ITEM(pools, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2091 if (PyTuple_GET_SIZE(pool) == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2092 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2093 elem = PyTuple_GET_ITEM(pool, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2094 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2095 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2096 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2097 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2098 Py_ssize_t *indices = lz->indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2099
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2100 /* Copy the previous result tuple or re-use it if available */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2101 if (Py_REFCNT(result) > 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2102 PyObject *old_result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2103 result = PyTuple_New(npools);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2104 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2105 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2106 lz->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2107 for (i=0; i < npools; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2108 elem = PyTuple_GET_ITEM(old_result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2109 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2110 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2111 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2112 Py_DECREF(old_result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2113 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2114 /* Now, we've got the only copy so we can update it in-place */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2115 assert (npools==0 \|\| Py_REFCNT(result) == 1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2116
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2117 /* Update the pool indices right-to-left. Only advance to the
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2118 next pool when the previous one rolls-over */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2119 for (i=npools-1 ; i >= 0 ; i--) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2120 pool = PyTuple_GET_ITEM(pools, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2121 indices[i]++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2122 if (indices[i] == PyTuple_GET_SIZE(pool)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2123 /* Roll-over and advance to next pool */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2124 indices[i] = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2125 elem = PyTuple_GET_ITEM(pool, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2126 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2127 oldelem = PyTuple_GET_ITEM(result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2128 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2129 Py_DECREF(oldelem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2130 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2131 /* No rollover. Just increment and stop here. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2132 elem = PyTuple_GET_ITEM(pool, indices[i]);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2133 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2134 oldelem = PyTuple_GET_ITEM(result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2135 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2136 Py_DECREF(oldelem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2137 break;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2138 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2139 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2140
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2141 /* If i is negative, then the indices have all rolled-over
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2142 and we're done. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2143 if (i < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2144 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2145 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2146
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2147 Py_INCREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2148 return result;
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2149
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2150 empty:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2151 lz->stopped = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2152 return NULL;
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2153 }
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2154
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2155 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2156 product_reduce(productobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2157 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2158 if (lz->stopped) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2159 return Py_BuildValue("O(())", Py_TYPE(lz));
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2160 } else if (lz->result == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2161 return Py_BuildValue("OO", Py_TYPE(lz), lz->pools);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2162 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2163 PyObject *indices;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2164 Py_ssize_t n, i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2165
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2166 /* we must pickle the indices use them for setstate, and
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2167 * additionally indicate that the iterator has started
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2168 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2169 n = PyTuple_GET_SIZE(lz->pools);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2170 indices = PyTuple_New(n);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2171 if (indices == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2172 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2173 for (i=0; i<n; i++){
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2174 PyObject* index = PyLong_FromSsize_t(lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2175 if (!index) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2176 Py_DECREF(indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2177 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2178 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2179 PyTuple_SET_ITEM(indices, i, index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2180 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2181 return Py_BuildValue("OON", Py_TYPE(lz), lz->pools, indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2182 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2183 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2184
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2185 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2186 product_setstate(productobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2187 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2188 PyObject *result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2189 Py_ssize_t n, i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2190
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2191 n = PyTuple_GET_SIZE(lz->pools);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2192 if (!PyTuple_Check(state) \|\| PyTuple_GET_SIZE(state) != n) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2193 PyErr_SetString(PyExc_ValueError, "invalid arguments");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2194 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2195 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2196 for (i=0; i<n; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2197 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2198 PyObject* indexObject = PyTuple_GET_ITEM(state, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2199 Py_ssize_t index = PyLong_AsSsize_t(indexObject);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2200 if (index < 0 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2201 return NULL; /* not an integer */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2202 /* clamp the index */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2203 if (index < 0)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2204 index = 0;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2205 else if (index > n-1)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2206 index = n-1;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2207 lz->indices[i] = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2208 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2209
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2210 result = PyTuple_New(n);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2211 if (!result)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2212 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2213 for (i=0; i<n; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2214 PyObject *pool = PyTuple_GET_ITEM(lz->pools, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2215 PyObject *element = PyTuple_GET_ITEM(pool, lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2216 Py_INCREF(element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2217 PyTuple_SET_ITEM(result, i, element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2218 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2219 Py_CLEAR(lz->result);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2220 lz->result = result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2221 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2222 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2223
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2224 static PyMethodDef product_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2225 {"__reduce__", (PyCFunction)product_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2226 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2227 {"__setstate__", (PyCFunction)product_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2228 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2229 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2230 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2231
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2232 PyDoc_STRVAR(product_doc,
84262 1fad7a709aae Close #18285: add 'repeat' parameter to docstring for product Andrew Kuchling <amk@amk.ca> parents: 84236 diff changeset	2233 "product(*iterables, repeat=1) --> product object\n\
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2234 \n\
45446 5e0772ed20d1 Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900,60902-60906,60908,60911-60917,60919-60920,60922,60926,60929-60931,60933-60935,60937,60939-60941,60943-60954,60959-60961,60963-60969,60971-60976 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45435 diff changeset	2235 Cartesian product of input iterables. Equivalent to nested for-loops.\n\n\
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2236 For example, product(A, B) returns the same as: ((x,y) for x in A for y in B).\n\
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2237 The leftmost iterators are in the outermost for-loop, so the output tuples\n\
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2238 cycle in a manner similar to an odometer (with the rightmost element changing\n\
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2239 on every iteration).\n\n\
51503 e759da93df64 Merged revisions 69466,69480 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 51270 diff changeset	2240 To compute the product of an iterable with itself, specify the number\n\
e759da93df64 Merged revisions 69466,69480 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 51270 diff changeset	2241 of repetitions with the optional repeat keyword argument. For example,\n\
e759da93df64 Merged revisions 69466,69480 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 51270 diff changeset	2242 product(A, repeat=4) means the same as product(A, A, A, A).\n\n\
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2243 product('ab', range(3)) --> ('a',0) ('a',1) ('a',2) ('b',0) ('b',1) ('b',2)\n\
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2244 product((0,1), (0,1), (0,1)) --> (0,0,0) (0,0,1) (0,1,0) (0,1,1) (1,0,0) ...");
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2245
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2246 static PyTypeObject product_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2247 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2248 "itertools.product", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2249 sizeof(productobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2250 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2251 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2252 (destructor)product_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2253 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2254 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2255 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2256 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2257 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2258 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2259 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2260 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2261 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2262 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2263 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2264 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2265 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2266 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2267 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2268 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2269 product_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2270 (traverseproc)product_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2271 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2272 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2273 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2274 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2275 (iternextfunc)product_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2276 product_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2277 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2278 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2279 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2280 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2281 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2282 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2283 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2284 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2285 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2286 product_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2287 PyObject_GC_Del, /* tp_free */
45435 42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2288 };
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2289
42b3b117391d Merged revisions 60481,60485,60489-60492,60494-60496,60498-60499,60501-60503,60505-60506,60508-60509,60523-60524,60532,60543,60545,60547-60548,60552,60554,60556-60559,60561-60562,60569,60571-60572,60574,60576-60583,60585-60586,60589,60591,60594-60595,60597-60598,60600-60601,60606-60612,60615,60617,60619-60621,60623-60625,60627-60629,60631,60633,60635,60647,60650,60652,60654,60656,60658-60659,60664-60666,60668-60670,60672,60676,60678,60680-60683,60685-60686,60688,60690,60692-60694,60697-60700,60705-60706,60708,60711,60714,60720,60724-60730,60732,60736,60742,60744,60746,60748,60750-60751,60753,60756-60757,60759-60761,60763-60764,60766,60769-60770,60774-60784,60787-60789,60793,60796,60799-60809,60812-60813,60815-60821,60823-60826,60828-60829,60831-60834,60836,60838-60839,60846-60849,60852-60854,60856-60859,60861-60870,60874-60875,60880-60881,60886,60888-60890,60892,60894-60898,60900-60931,60933-60958 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44881 diff changeset	2290
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2291 /* combinations object ************************************************************/
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2292
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2293 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2294 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2295 PyObject pool; / input converted to a tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2296 Py_ssize_t indices; / one index per result element */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2297 PyObject result; / most recently returned result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2298 Py_ssize_t r; /* size of result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2299 int stopped; /* set to 1 when the combinations iterator is exhausted */
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2300 } combinationsobject;
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2301
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2302 static PyTypeObject combinations_type;
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2303
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2304 static PyObject *
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2305 combinations_new(PyTypeObject type, PyObject args, PyObject *kwds)
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2306 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2307 combinationsobject *co;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2308 Py_ssize_t n;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2309 Py_ssize_t r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2310 PyObject *pool = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2311 PyObject *iterable = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2312 Py_ssize_t *indices = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2313 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2314 static char *kwargs[] = {"iterable", "r", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2315
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2316 if (!PyArg_ParseTupleAndKeywords(args, kwds, "On:combinations", kwargs,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2317 &iterable, &r))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2318 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2319
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2320 pool = PySequence_Tuple(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2321 if (pool == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2322 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2323 n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2324 if (r < 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2325 PyErr_SetString(PyExc_ValueError, "r must be non-negative");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2326 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2327 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2328
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2329 indices = PyMem_Malloc(r * sizeof(Py_ssize_t));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2330 if (indices == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2331 PyErr_NoMemory();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2332 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2333 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2334
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2335 for (i=0 ; i<r ; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2336 indices[i] = i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2337
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2338 /* create combinationsobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2339 co = (combinationsobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2340 if (co == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2341 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2342
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2343 co->pool = pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2344 co->indices = indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2345 co->result = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2346 co->r = r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2347 co->stopped = r > n ? 1 : 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2348
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2349 return (PyObject *)co;
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2350
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2351 error:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2352 if (indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2353 PyMem_Free(indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2354 Py_XDECREF(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2355 return NULL;
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2356 }
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2357
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2358 static void
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2359 combinations_dealloc(combinationsobject *co)
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2360 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2361 PyObject_GC_UnTrack(co);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2362 Py_XDECREF(co->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2363 Py_XDECREF(co->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2364 if (co->indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2365 PyMem_Free(co->indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2366 Py_TYPE(co)->tp_free(co);
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2367 }
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2368
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2369 static int
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2370 combinations_traverse(combinationsobject co, visitproc visit, void arg)
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2371 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2372 Py_VISIT(co->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2373 Py_VISIT(co->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2374 return 0;
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2375 }
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2376
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2377 static PyObject *
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2378 combinations_next(combinationsobject *co)
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2379 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2380 PyObject *elem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2381 PyObject *oldelem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2382 PyObject *pool = co->pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2383 Py_ssize_t *indices = co->indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2384 PyObject *result = co->result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2385 Py_ssize_t n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2386 Py_ssize_t r = co->r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2387 Py_ssize_t i, j, index;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2388
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2389 if (co->stopped)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2390 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2391
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2392 if (result == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2393 /* On the first pass, initialize result tuple using the indices */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2394 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2395 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2396 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2397 co->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2398 for (i=0; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2399 index = indices[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2400 elem = PyTuple_GET_ITEM(pool, index);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2401 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2402 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2403 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2404 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2405 /* Copy the previous result tuple or re-use it if available */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2406 if (Py_REFCNT(result) > 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2407 PyObject *old_result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2408 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2409 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2410 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2411 co->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2412 for (i=0; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2413 elem = PyTuple_GET_ITEM(old_result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2414 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2415 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2416 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2417 Py_DECREF(old_result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2418 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2419 /* Now, we've got the only copy so we can update it in-place
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2420 * CPython's empty tuple is a singleton and cached in
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2421 * PyTuple's freelist.
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2422 */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2423 assert(r == 0 \|\| Py_REFCNT(result) == 1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2424
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2425 /* Scan indices right-to-left until finding one that is not
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2426 at its maximum (i + n - r). */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2427 for (i=r-1 ; i >= 0 && indices[i] == i+n-r ; i--)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2428 ;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2429
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2430 /* If i is negative, then the indices are all at
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2431 their maximum value and we're done. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2432 if (i < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2433 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2434
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2435 /* Increment the current index which we know is not at its
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2436 maximum. Then move back to the right setting each index
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2437 to its lowest possible value (one higher than the index
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2438 to its left -- this maintains the sort order invariant). */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2439 indices[i]++;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2440 for (j=i+1 ; j<r ; j++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2441 indices[j] = indices[j-1] + 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2442
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2443 /* Update the result tuple for the new indices
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2444 starting with i, the leftmost index that changed */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2445 for ( ; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2446 index = indices[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2447 elem = PyTuple_GET_ITEM(pool, index);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2448 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2449 oldelem = PyTuple_GET_ITEM(result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2450 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2451 Py_DECREF(oldelem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2452 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2453 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2454
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2455 Py_INCREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2456 return result;
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2457
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2458 empty:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2459 co->stopped = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2460 return NULL;
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2461 }
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2462
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2463 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2464 combinations_reduce(combinationsobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2465 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2466 if (lz->result == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2467 return Py_BuildValue("O(On)", Py_TYPE(lz), lz->pool, lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2468 } else if (lz->stopped) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2469 return Py_BuildValue("O(()n)", Py_TYPE(lz), lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2470 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2471 PyObject *indices;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2472 Py_ssize_t i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2473
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2474 /* we must pickle the indices and use them for setstate */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2475 indices = PyTuple_New(lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2476 if (!indices)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2477 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2478 for (i=0; i<lz->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2479 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2480 PyObject* index = PyLong_FromSsize_t(lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2481 if (!index) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2482 Py_DECREF(indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2483 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2484 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2485 PyTuple_SET_ITEM(indices, i, index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2486 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2487
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2488 return Py_BuildValue("O(On)N", Py_TYPE(lz), lz->pool, lz->r, indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2489 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2490 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2491
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2492 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2493 combinations_setstate(combinationsobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2494 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2495 PyObject *result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2496 Py_ssize_t i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2497 Py_ssize_t n = PyTuple_GET_SIZE(lz->pool);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2498
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2499 if (!PyTuple_Check(state) \|\| PyTuple_GET_SIZE(state) != lz->r)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2500 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2501 PyErr_SetString(PyExc_ValueError, "invalid arguments");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2502 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2503 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2504
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2505 for (i=0; i<lz->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2506 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2507 Py_ssize_t max;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2508 PyObject* indexObject = PyTuple_GET_ITEM(state, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2509 Py_ssize_t index = PyLong_AsSsize_t(indexObject);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2510 if (index == -1 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2511 return NULL; /* not an integer */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2512 max = i + n - lz->r;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2513 /* clamp the index (beware of negative max) */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2514 if (index > max)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2515 index = max;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2516 if (index < 0)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2517 index = 0;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2518 lz->indices[i] = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2519 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2520
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2521 result = PyTuple_New(lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2522 if (result == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2523 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2524 for (i=0; i<lz->r; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2525 PyObject *element = PyTuple_GET_ITEM(lz->pool, lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2526 Py_INCREF(element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2527 PyTuple_SET_ITEM(result, i, element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2528 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2529
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2530 Py_CLEAR(lz->result);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2531 lz->result = result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2532 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2533 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2534
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2535 static PyMethodDef combinations_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2536 {"__reduce__", (PyCFunction)combinations_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2537 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2538 {"__setstate__", (PyCFunction)combinations_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2539 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2540 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2541 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2542
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2543 PyDoc_STRVAR(combinations_doc,
57344 a790f84fbafa Fix docstrings for itertools combinatoric functions. Raymond Hettinger <python@rcn.com> parents: 57042 diff changeset	2544 "combinations(iterable, r) --> combinations object\n\
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2545 \n\
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2546 Return successive r-length combinations of elements in the iterable.\n\n\
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2547 combinations(range(4), 3) --> (0,1,2), (0,1,3), (0,2,3), (1,2,3)");
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2549 static PyTypeObject combinations_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2550 PyVarObject_HEAD_INIT(NULL, 0)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2551 "itertools.combinations", /* tp_name */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2552 sizeof(combinationsobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2553 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2554 /* methods */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2555 (destructor)combinations_dealloc, /* tp_dealloc */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2556 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2557 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2558 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2559 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2560 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2561 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2562 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2563 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2564 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2565 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2566 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2567 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2568 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2569 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2570 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2571 Py_TPFLAGS_BASETYPE, /* tp_flags */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2572 combinations_doc, /* tp_doc */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2573 (traverseproc)combinations_traverse,/* tp_traverse */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2574 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2575 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2576 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2577 PyObject_SelfIter, /* tp_iter */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2578 (iternextfunc)combinations_next, /* tp_iternext */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2579 combinations_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2580 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2581 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2582 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2583 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2584 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2585 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2586 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2587 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2588 0, /* tp_alloc */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2589 combinations_new, /* tp_new */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2590 PyObject_GC_Del, /* tp_free */
45548 8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2591 };
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2592
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45534 diff changeset	2593
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2594 /* combinations with replacement object *******************************************/
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2595
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2596 /* Equivalent to:
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2597
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2598 def combinations_with_replacement(iterable, r):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2599 "combinations_with_replacement('ABC', 2) --> AA AB AC BB BC CC"
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2600 # number items returned: (n+r-1)! / r! / (n-1)!
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2601 pool = tuple(iterable)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2602 n = len(pool)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2603 indices = [0] * r
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2604 yield tuple(pool[i] for i in indices)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2605 while 1:
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2606 for i in reversed(range(r)):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2607 if indices[i] != n - 1:
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2608 break
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2609 else:
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2610 return
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2611 indices[i:] = [indices[i] + 1] * (r - i)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2612 yield tuple(pool[i] for i in indices)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2613
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2614 def combinations_with_replacement2(iterable, r):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2615 'Alternate version that filters from product()'
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2616 pool = tuple(iterable)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2617 n = len(pool)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2618 for indices in product(range(n), repeat=r):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2619 if sorted(indices) == list(indices):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2620 yield tuple(pool[i] for i in indices)
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2621 */
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2622 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2623 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2624 PyObject pool; / input converted to a tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2625 Py_ssize_t indices; / one index per result element */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2626 PyObject result; / most recently returned result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2627 Py_ssize_t r; /* size of result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2628 int stopped; /* set to 1 when the cwr iterator is exhausted */
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2629 } cwrobject;
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2630
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2631 static PyTypeObject cwr_type;
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2632
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2633 static PyObject *
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2634 cwr_new(PyTypeObject type, PyObject args, PyObject *kwds)
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2635 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2636 cwrobject *co;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2637 Py_ssize_t n;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2638 Py_ssize_t r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2639 PyObject *pool = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2640 PyObject *iterable = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2641 Py_ssize_t *indices = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2642 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2643 static char *kwargs[] = {"iterable", "r", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2644
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2645 if (!PyArg_ParseTupleAndKeywords(args, kwds, "On:combinations_with_replacement", kwargs,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2646 &iterable, &r))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2647 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2648
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2649 pool = PySequence_Tuple(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2650 if (pool == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2651 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2652 n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2653 if (r < 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2654 PyErr_SetString(PyExc_ValueError, "r must be non-negative");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2655 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2656 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2657
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2658 indices = PyMem_Malloc(r * sizeof(Py_ssize_t));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2659 if (indices == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2660 PyErr_NoMemory();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2661 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2662 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2663
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2664 for (i=0 ; i<r ; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2665 indices[i] = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2666
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2667 /* create cwrobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2668 co = (cwrobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2669 if (co == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2670 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2671
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2672 co->pool = pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2673 co->indices = indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2674 co->result = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2675 co->r = r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2676 co->stopped = !n && r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2677
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2678 return (PyObject *)co;
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2679
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2680 error:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2681 if (indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2682 PyMem_Free(indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2683 Py_XDECREF(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2684 return NULL;
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2685 }
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2686
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2687 static void
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2688 cwr_dealloc(cwrobject *co)
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2689 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2690 PyObject_GC_UnTrack(co);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2691 Py_XDECREF(co->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2692 Py_XDECREF(co->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2693 if (co->indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2694 PyMem_Free(co->indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2695 Py_TYPE(co)->tp_free(co);
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2696 }
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2697
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2698 static int
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2699 cwr_traverse(cwrobject co, visitproc visit, void arg)
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2700 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2701 Py_VISIT(co->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2702 Py_VISIT(co->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2703 return 0;
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2704 }
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2705
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2706 static PyObject *
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2707 cwr_next(cwrobject *co)
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2708 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2709 PyObject *elem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2710 PyObject *oldelem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2711 PyObject *pool = co->pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2712 Py_ssize_t *indices = co->indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2713 PyObject *result = co->result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2714 Py_ssize_t n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2715 Py_ssize_t r = co->r;
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2716 Py_ssize_t i, index;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2717
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2718 if (co->stopped)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2719 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2720
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2721 if (result == NULL) {
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2722 /* On the first pass, initialize result tuple with pool[0] */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2723 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2724 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2725 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2726 co->result = result;
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2727 elem = PyTuple_GET_ITEM(pool, 0);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2728 for (i=0; i<r ; i++) {
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2729 assert(indices[i] == 0);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2730 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2731 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2732 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2733 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2734 /* Copy the previous result tuple or re-use it if available */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2735 if (Py_REFCNT(result) > 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2736 PyObject *old_result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2737 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2738 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2739 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2740 co->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2741 for (i=0; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2742 elem = PyTuple_GET_ITEM(old_result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2743 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2744 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2745 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2746 Py_DECREF(old_result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2747 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2748 /* Now, we've got the only copy so we can update it in-place CPython's
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2749 empty tuple is a singleton and cached in PyTuple's freelist. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2750 assert(r == 0 \|\| Py_REFCNT(result) == 1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2751
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2752 /* Scan indices right-to-left until finding one that is not
4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2753 * at its maximum (n-1). */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2754 for (i=r-1 ; i >= 0 && indices[i] == n-1; i--)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2755 ;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2756
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2757 /* If i is negative, then the indices are all at
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2758 their maximum value and we're done. */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2759 if (i < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2760 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2761
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2762 /* Increment the current index which we know is not at its
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2763 maximum. Then set all to the right to the same value. */
4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2764 index = indices[i] + 1;
4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2765 assert(index < n);
4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2766 elem = PyTuple_GET_ITEM(pool, index);
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2767 for ( ; i<r ; i++) {
85511 4bbaa3d7bcbf cwr_next(): move invariants out of loops. Tim Peters <tim@python.org> parents: 85507 diff changeset	2768 indices[i] = index;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2769 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2770 oldelem = PyTuple_GET_ITEM(result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2771 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2772 Py_DECREF(oldelem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2773 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2774 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2775
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2776 Py_INCREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2777 return result;
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2778
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2779 empty:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2780 co->stopped = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2781 return NULL;
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2782 }
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2783
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2784 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2785 cwr_reduce(cwrobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2786 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2787 if (lz->result == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2788 return Py_BuildValue("O(On)", Py_TYPE(lz), lz->pool, lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2789 } else if (lz->stopped) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2790 return Py_BuildValue("O(()n)", Py_TYPE(lz), lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2791 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2792 PyObject *indices;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2793 Py_ssize_t i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2794
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2795 /* we must pickle the indices and use them for setstate */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2796 indices = PyTuple_New(lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2797 if (!indices)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2798 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2799 for (i=0; i<lz->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2800 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2801 PyObject* index = PyLong_FromSsize_t(lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2802 if (!index) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2803 Py_DECREF(indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2804 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2805 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2806 PyTuple_SET_ITEM(indices, i, index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2807 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2808
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2809 return Py_BuildValue("O(On)N", Py_TYPE(lz), lz->pool, lz->r, indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2810 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2811 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2812
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2813 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2814 cwr_setstate(cwrobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2815 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2816 PyObject *result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2817 Py_ssize_t n, i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2818
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2819 if (!PyTuple_Check(state) \|\| PyTuple_GET_SIZE(state) != lz->r)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2820 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2821 PyErr_SetString(PyExc_ValueError, "invalid arguments");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2822 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2823 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2824
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2825 n = PyTuple_GET_SIZE(lz->pool);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2826 for (i=0; i<lz->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2827 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2828 PyObject* indexObject = PyTuple_GET_ITEM(state, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2829 Py_ssize_t index = PyLong_AsSsize_t(indexObject);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2830 if (index < 0 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2831 return NULL; /* not an integer */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2832 /* clamp the index */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2833 if (index < 0)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2834 index = 0;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2835 else if (index > n-1)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2836 index = n-1;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2837 lz->indices[i] = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2838 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2839 result = PyTuple_New(lz->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2840 if (result == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2841 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2842 for (i=0; i<lz->r; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2843 PyObject *element = PyTuple_GET_ITEM(lz->pool, lz->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2844 Py_INCREF(element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2845 PyTuple_SET_ITEM(result, i, element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2846 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2847 Py_CLEAR(lz->result);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2848 lz->result = result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2849 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2850 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2851
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2852 static PyMethodDef cwr_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2853 {"__reduce__", (PyCFunction)cwr_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2854 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2855 {"__setstate__", (PyCFunction)cwr_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2856 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2857 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2858 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2859
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2860 PyDoc_STRVAR(cwr_doc,
57344 a790f84fbafa Fix docstrings for itertools combinatoric functions. Raymond Hettinger <python@rcn.com> parents: 57042 diff changeset	2861 "combinations_with_replacement(iterable, r) --> combinations_with_replacement object\n\
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2862 \n\
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2863 Return successive r-length combinations of elements in the iterable\n\
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2864 allowing individual elements to have successive repeats.\n\
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2865 combinations_with_replacement('ABC', 2) --> AA AB AC BB BC CC");
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2866
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2867 static PyTypeObject cwr_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2868 PyVarObject_HEAD_INIT(NULL, 0)
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2869 "itertools.combinations_with_replacement", /* tp_name */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2870 sizeof(cwrobject), /* tp_basicsize */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2871 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2872 /* methods */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2873 (destructor)cwr_dealloc, /* tp_dealloc */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2874 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2875 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2876 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2877 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2878 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2879 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2880 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2881 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2882 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2883 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2884 0, /* tp_str */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2885 PyObject_GenericGetAttr, /* tp_getattro */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2886 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2887 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2888 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2889 Py_TPFLAGS_BASETYPE, /* tp_flags */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2890 cwr_doc, /* tp_doc */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2891 (traverseproc)cwr_traverse, /* tp_traverse */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2892 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2893 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2894 0, /* tp_weaklistoffset */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2895 PyObject_SelfIter, /* tp_iter */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2896 (iternextfunc)cwr_next, /* tp_iternext */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2897 cwr_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2898 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2899 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2900 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2901 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2902 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2903 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2904 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2905 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2906 0, /* tp_alloc */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2907 cwr_new, /* tp_new */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	2908 PyObject_GC_Del, /* tp_free */
51100 4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2909 };
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2910
4005011d3a09 Forward port r69001: itertools.combinations_with_replacement(). Raymond Hettinger <python@rcn.com> parents: 51063 diff changeset	2911
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2912 /* permutations object ************************************************************
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2913
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2914 def permutations(iterable, r=None):
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2915 'permutations(range(3), 2) --> (0,1) (0,2) (1,0) (1,2) (2,0) (2,1)'
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2916 pool = tuple(iterable)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2917 n = len(pool)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2918 r = n if r is None else r
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2919 indices = range(n)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2920 cycles = range(n-r+1, n+1)[::-1]
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2921 yield tuple(pool[i] for i in indices[:r])
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2922 while n:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2923 for i in reversed(range(r)):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2924 cycles[i] -= 1
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2925 if cycles[i] == 0:
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2926 indices[i:] = indices[i+1:] + indices[i:i+1]
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2927 cycles[i] = n - i
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2928 else:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2929 j = cycles[i]
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2930 indices[i], indices[-j] = indices[-j], indices[i]
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2931 yield tuple(pool[i] for i in indices[:r])
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2932 break
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2933 else:
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2934 return
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2935 */
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2936
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2937 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2938 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2939 PyObject pool; / input converted to a tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2940 Py_ssize_t indices; / one index per element in the pool */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2941 Py_ssize_t cycles; / one rollover counter per element in the result */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2942 PyObject result; / most recently returned result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2943 Py_ssize_t r; /* size of result tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2944 int stopped; /* set to 1 when the permutations iterator is exhausted */
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2945 } permutationsobject;
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2946
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2947 static PyTypeObject permutations_type;
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2948
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2949 static PyObject *
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2950 permutations_new(PyTypeObject type, PyObject args, PyObject *kwds)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	2951 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2952 permutationsobject *po;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2953 Py_ssize_t n;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2954 Py_ssize_t r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2955 PyObject *robj = Py_None;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2956 PyObject *pool = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2957 PyObject *iterable = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2958 Py_ssize_t *indices = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2959 Py_ssize_t *cycles = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2960 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2961 static char *kwargs[] = {"iterable", "r", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2962
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2963 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|O:permutations", kwargs,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2964 &iterable, &robj))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2965 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2966
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2967 pool = PySequence_Tuple(iterable);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2968 if (pool == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2969 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2970 n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2971
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2972 r = n;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2973 if (robj != Py_None) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2974 if (!PyLong_Check(robj)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2975 PyErr_SetString(PyExc_TypeError, "Expected int as r");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2976 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2977 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2978 r = PyLong_AsSsize_t(robj);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2979 if (r == -1 && PyErr_Occurred())
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2980 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2981 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2982 if (r < 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2983 PyErr_SetString(PyExc_ValueError, "r must be non-negative");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2984 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2985 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2986
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2987 indices = PyMem_Malloc(n * sizeof(Py_ssize_t));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2988 cycles = PyMem_Malloc(r * sizeof(Py_ssize_t));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2989 if (indices == NULL \|\| cycles == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2990 PyErr_NoMemory();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2991 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2992 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2993
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2994 for (i=0 ; i<n ; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2995 indices[i] = i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2996 for (i=0 ; i<r ; i++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2997 cycles[i] = n - i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2998
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	2999 /* create permutationsobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3000 po = (permutationsobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3001 if (po == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3002 goto error;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3003
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3004 po->pool = pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3005 po->indices = indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3006 po->cycles = cycles;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3007 po->result = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3008 po->r = r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3009 po->stopped = r > n ? 1 : 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3010
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3011 return (PyObject *)po;
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3012
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3013 error:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3014 if (indices != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3015 PyMem_Free(indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3016 if (cycles != NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3017 PyMem_Free(cycles);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3018 Py_XDECREF(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3019 return NULL;
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3020 }
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3021
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3022 static void
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3023 permutations_dealloc(permutationsobject *po)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3024 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3025 PyObject_GC_UnTrack(po);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3026 Py_XDECREF(po->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3027 Py_XDECREF(po->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3028 PyMem_Free(po->indices);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3029 PyMem_Free(po->cycles);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3030 Py_TYPE(po)->tp_free(po);
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3031 }
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3032
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3033 static int
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3034 permutations_traverse(permutationsobject po, visitproc visit, void arg)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3035 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3036 Py_VISIT(po->pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3037 Py_VISIT(po->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3038 return 0;
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3039 }
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3040
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3041 static PyObject *
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3042 permutations_next(permutationsobject *po)
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3043 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3044 PyObject *elem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3045 PyObject *oldelem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3046 PyObject *pool = po->pool;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3047 Py_ssize_t *indices = po->indices;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3048 Py_ssize_t *cycles = po->cycles;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3049 PyObject *result = po->result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3050 Py_ssize_t n = PyTuple_GET_SIZE(pool);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3051 Py_ssize_t r = po->r;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3052 Py_ssize_t i, j, k, index;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3053
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3054 if (po->stopped)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3055 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3056
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3057 if (result == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3058 /* On the first pass, initialize result tuple using the indices */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3059 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3060 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3061 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3062 po->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3063 for (i=0; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3064 index = indices[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3065 elem = PyTuple_GET_ITEM(pool, index);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3066 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3067 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3068 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3069 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3070 if (n == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3071 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3072
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3073 /* Copy the previous result tuple or re-use it if available */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3074 if (Py_REFCNT(result) > 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3075 PyObject *old_result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3076 result = PyTuple_New(r);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3077 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3078 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3079 po->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3080 for (i=0; i<r ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3081 elem = PyTuple_GET_ITEM(old_result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3082 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3083 PyTuple_SET_ITEM(result, i, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3084 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3085 Py_DECREF(old_result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3086 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3087 /* Now, we've got the only copy so we can update it in-place */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3088 assert(r == 0 \|\| Py_REFCNT(result) == 1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3089
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3090 /* Decrement rightmost cycle, moving leftward upon zero rollover */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3091 for (i=r-1 ; i>=0 ; i--) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3092 cycles[i] -= 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3093 if (cycles[i] == 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3094 /* rotatation: indices[i:] = indices[i+1:] + indices[i:i+1] */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3095 index = indices[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3096 for (j=i ; j<n-1 ; j++)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3097 indices[j] = indices[j+1];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3098 indices[n-1] = index;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3099 cycles[i] = n - i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3100 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3101 j = cycles[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3102 index = indices[i];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3103 indices[i] = indices[n-j];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3104 indices[n-j] = index;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3105
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3106 for (k=i; k<r ; k++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3107 /* start with i, the leftmost element that changed */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3108 /* yield tuple(pool[k] for k in indices[:r]) */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3109 index = indices[k];
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3110 elem = PyTuple_GET_ITEM(pool, index);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3111 Py_INCREF(elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3112 oldelem = PyTuple_GET_ITEM(result, k);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3113 PyTuple_SET_ITEM(result, k, elem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3114 Py_DECREF(oldelem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3115 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3116 break;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3117 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3118 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3119 /* If i is negative, then the cycles have all
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3120 rolled-over and we're done. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3121 if (i < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3122 goto empty;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3123 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3124 Py_INCREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3125 return result;
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3126
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3127 empty:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3128 po->stopped = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3129 return NULL;
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3130 }
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3131
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3132 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3133 permutations_reduce(permutationsobject *po)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3134 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3135 if (po->result == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3136 return Py_BuildValue("O(On)", Py_TYPE(po), po->pool, po->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3137 } else if (po->stopped) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3138 return Py_BuildValue("O(()n)", Py_TYPE(po), po->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3139 } else {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3140 PyObject indices=NULL, cycles=NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3141 Py_ssize_t n, i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3142
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3143 /* we must pickle the indices and cycles and use them for setstate */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3144 n = PyTuple_GET_SIZE(po->pool);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3145 indices = PyTuple_New(n);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3146 if (indices == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3147 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3148 for (i=0; i<n; i++){
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3149 PyObject* index = PyLong_FromSsize_t(po->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3150 if (!index)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3151 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3152 PyTuple_SET_ITEM(indices, i, index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3153 }
81705 20c4d55f4b69 Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81701 81704 diff changeset	3154
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3155 cycles = PyTuple_New(po->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3156 if (cycles == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3157 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3158 for (i=0; i<po->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3159 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3160 PyObject* index = PyLong_FromSsize_t(po->cycles[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3161 if (!index)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3162 goto err;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3163 PyTuple_SET_ITEM(cycles, i, index);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3164 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3165 return Py_BuildValue("O(On)(NN)", Py_TYPE(po),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3166 po->pool, po->r,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3167 indices, cycles);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3168 err:
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3169 Py_XDECREF(indices);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3170 Py_XDECREF(cycles);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3171 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3172 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3173 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3174
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3175 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3176 permutations_setstate(permutationsobject po, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3177 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3178 PyObject indices, cycles, *result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3179 Py_ssize_t n, i;
81705 20c4d55f4b69 Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81701 81704 diff changeset	3180
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3181 if (!PyArg_ParseTuple(state, "O!O!",
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3182 &PyTuple_Type, &indices,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3183 &PyTuple_Type, &cycles))
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3184 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3185
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3186 n = PyTuple_GET_SIZE(po->pool);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3187 if (PyTuple_GET_SIZE(indices) != n \|\|
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3188 PyTuple_GET_SIZE(cycles) != po->r)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3189 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3190 PyErr_SetString(PyExc_ValueError, "invalid arguments");
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3191 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3192 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3193
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3194 for (i=0; i<n; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3195 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3196 PyObject* indexObject = PyTuple_GET_ITEM(indices, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3197 Py_ssize_t index = PyLong_AsSsize_t(indexObject);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3198 if (index < 0 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3199 return NULL; /* not an integer */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3200 /* clamp the index */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3201 if (index < 0)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3202 index = 0;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3203 else if (index > n-1)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3204 index = n-1;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3205 po->indices[i] = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3206 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3207
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3208 for (i=0; i<po->r; i++)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3209 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3210 PyObject* indexObject = PyTuple_GET_ITEM(cycles, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3211 Py_ssize_t index = PyLong_AsSsize_t(indexObject);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3212 if (index < 0 && PyErr_Occurred())
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3213 return NULL; /* not an integer */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3214 if (index < 1)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3215 index = 1;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3216 else if (index > n-i)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3217 index = n-i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3218 po->cycles[i] = index;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3219 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3220 result = PyTuple_New(po->r);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3221 if (result == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3222 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3223 for (i=0; i<po->r; i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3224 PyObject *element = PyTuple_GET_ITEM(po->pool, po->indices[i]);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3225 Py_INCREF(element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3226 PyTuple_SET_ITEM(result, i, element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3227 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3228 Py_CLEAR(po->result);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3229 po->result = result;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3230 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3231 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3232
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3233 static PyMethodDef permuations_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3234 {"__reduce__", (PyCFunction)permutations_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3235 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3236 {"__setstate__", (PyCFunction)permutations_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3237 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3238 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3239 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3240
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3241 PyDoc_STRVAR(permutations_doc,
47555 7d824fd131ad Merged revisions 63829-63831,63858,63865,63879,63882,63948,63970-63972,63976,63989,64014-64015,64021-64022,64063-64065,64067 via svnmerge from Georg Brandl <georg@python.org> parents: 46238 diff changeset	3242 "permutations(iterable[, r]) --> permutations object\n\
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3243 \n\
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3244 Return successive r-length permutations of elements in the iterable.\n\n\
47555 7d824fd131ad Merged revisions 63829-63831,63858,63865,63879,63882,63948,63970-63972,63976,63989,64014-64015,64021-64022,64063-64065,64067 via svnmerge from Georg Brandl <georg@python.org> parents: 46238 diff changeset	3245 permutations(range(3), 2) --> (0,1), (0,2), (1,0), (1,2), (2,0), (2,1)");
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3246
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3247 static PyTypeObject permutations_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3248 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3249 "itertools.permutations", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3250 sizeof(permutationsobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3251 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3252 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3253 (destructor)permutations_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3254 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3255 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3256 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3257 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3258 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3259 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3260 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3261 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3262 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3263 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3264 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3265 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3266 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3267 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3268 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3269 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3270 permutations_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3271 (traverseproc)permutations_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3272 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3273 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3274 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3275 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3276 (iternextfunc)permutations_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3277 permuations_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3278 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3279 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3280 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3281 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3282 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3283 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3284 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3285 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3286 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3287 permutations_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3288 PyObject_GC_Del, /* tp_free */
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3289 };
832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3290
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3291 /* accumulate object ************************************************************/
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3292
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3293 typedef struct {
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3294 PyObject_HEAD
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3295 PyObject *total;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3296 PyObject *it;
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3297 PyObject *binop;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3298 } accumulateobject;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3299
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3300 static PyTypeObject accumulate_type;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3301
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3302 static PyObject *
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3303 accumulate_new(PyTypeObject type, PyObject args, PyObject *kwds)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3304 {
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3305 static char *kwargs[] = {"iterable", "func", NULL};
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3306 PyObject *iterable;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3307 PyObject *it;
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3308 PyObject *binop = Py_None;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3309 accumulateobject *lz;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3310
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3311 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|O:accumulate",
79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3312 kwargs, &iterable, &binop))
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3313 return NULL;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3314
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3315 /* Get iterator. */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3316 it = PyObject_GetIter(iterable);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3317 if (it == NULL)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3318 return NULL;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3319
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3320 /* create accumulateobject structure */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3321 lz = (accumulateobject *)type->tp_alloc(type, 0);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3322 if (lz == NULL) {
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3323 Py_DECREF(it);
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3324 return NULL;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3325 }
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3326
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3327 if (binop != Py_None) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3328 Py_XINCREF(binop);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3329 lz->binop = binop;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3330 }
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3331 lz->total = NULL;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3332 lz->it = it;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3333 return (PyObject *)lz;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3334 }
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3335
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3336 static void
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3337 accumulate_dealloc(accumulateobject *lz)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3338 {
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3339 PyObject_GC_UnTrack(lz);
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3340 Py_XDECREF(lz->binop);
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3341 Py_XDECREF(lz->total);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3342 Py_XDECREF(lz->it);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3343 Py_TYPE(lz)->tp_free(lz);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3344 }
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3345
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3346 static int
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3347 accumulate_traverse(accumulateobject lz, visitproc visit, void arg)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3348 {
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3349 Py_VISIT(lz->binop);
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3350 Py_VISIT(lz->it);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3351 Py_VISIT(lz->total);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3352 return 0;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3353 }
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3354
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3355 static PyObject *
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3356 accumulate_next(accumulateobject *lz)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3357 {
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3358 PyObject val, oldtotal, *newtotal;
81704 68485ac1972a Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81700 diff changeset	3359
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3360 val = PyIter_Next(lz->it);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3361 if (val == NULL)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3362 return NULL;
81704 68485ac1972a Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81700 diff changeset	3363
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3364 if (lz->total == NULL) {
1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3365 Py_INCREF(val);
1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3366 lz->total = val;
1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3367 return lz->total;
1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3368 }
81704 68485ac1972a Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81700 diff changeset	3369
81705 20c4d55f4b69 Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81701 81704 diff changeset	3370 if (lz->binop == NULL)
69021 79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3371 newtotal = PyNumber_Add(lz->total, val);
79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3372 else
79ccefd30a37 Add optional func argument to itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66619 diff changeset	3373 newtotal = PyObject_CallFunctionObjArgs(lz->binop, lz->total, val, NULL);
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3374 Py_DECREF(val);
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3375 if (newtotal == NULL)
66619 1098fab85e47 Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max(). Raymond Hettinger <python@rcn.com> parents: 66589 diff changeset	3376 return NULL;
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3377
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3378 oldtotal = lz->total;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3379 lz->total = newtotal;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3380 Py_DECREF(oldtotal);
81704 68485ac1972a Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81700 diff changeset	3381
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3382 Py_INCREF(newtotal);
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3383 return newtotal;
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3384 }
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3385
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3386 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3387 accumulate_reduce(accumulateobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3388 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3389 return Py_BuildValue("O(OO)O", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3390 lz->it, lz->binop?lz->binop:Py_None,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3391 lz->total?lz->total:Py_None);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3392 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3393
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3394 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3395 accumulate_setstate(accumulateobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3396 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3397 Py_CLEAR(lz->total);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3398 lz->total = state;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3399 Py_INCREF(lz->total);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3400 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3401 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3402
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3403 static PyMethodDef accumulate_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3404 {"__reduce__", (PyCFunction)accumulate_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3405 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3406 {"__setstate__", (PyCFunction)accumulate_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3407 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3408 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3409 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3410
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3411 PyDoc_STRVAR(accumulate_doc,
69022 822c7c0d27d1 Update docstrings for itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 69021 diff changeset	3412 "accumulate(iterable[, func]) --> accumulate object\n\
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3413 \n\
69022 822c7c0d27d1 Update docstrings for itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 69021 diff changeset	3414 Return series of accumulated sums (or other binary function results).");
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3415
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3416 static PyTypeObject accumulate_type = {
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3417 PyVarObject_HEAD_INIT(NULL, 0)
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3418 "itertools.accumulate", /* tp_name */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3419 sizeof(accumulateobject), /* tp_basicsize */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3420 0, /* tp_itemsize */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3421 /* methods */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3422 (destructor)accumulate_dealloc, /* tp_dealloc */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3423 0, /* tp_print */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3424 0, /* tp_getattr */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3425 0, /* tp_setattr */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3426 0, /* tp_reserved */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3427 0, /* tp_repr */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3428 0, /* tp_as_number */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3429 0, /* tp_as_sequence */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3430 0, /* tp_as_mapping */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3431 0, /* tp_hash */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3432 0, /* tp_call */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3433 0, /* tp_str */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3434 PyObject_GenericGetAttr, /* tp_getattro */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3435 0, /* tp_setattro */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3436 0, /* tp_as_buffer */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3437 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3438 Py_TPFLAGS_BASETYPE, /* tp_flags */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3439 accumulate_doc, /* tp_doc */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3440 (traverseproc)accumulate_traverse, /* tp_traverse */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3441 0, /* tp_clear */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3442 0, /* tp_richcompare */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3443 0, /* tp_weaklistoffset */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3444 PyObject_SelfIter, /* tp_iter */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3445 (iternextfunc)accumulate_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3446 accumulate_methods, /* tp_methods */
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3447 0, /* tp_members */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3448 0, /* tp_getset */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3449 0, /* tp_base */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3450 0, /* tp_dict */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3451 0, /* tp_descr_get */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3452 0, /* tp_descr_set */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3453 0, /* tp_dictoffset */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3454 0, /* tp_init */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3455 0, /* tp_alloc */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3456 accumulate_new, /* tp_new */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3457 PyObject_GC_Del, /* tp_free */
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3458 };
5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	3459
45773 832c820e9d14 Merged revisions 61239-61249,61252-61257,61260-61264,61269-61275,61278-61279,61285-61286,61288-61290,61298,61303-61305,61312-61314,61317,61329,61332,61344,61350-61351,61363-61376,61378-61379,61382-61383,61387-61388,61392,61395-61396,61402-61403 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 45738 diff changeset	3460
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3461 /* compress object ************************************************************/
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3462
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3463 /* Equivalent to:
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3464
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3465 def compress(data, selectors):
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3466 "compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F"
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3467 return (d for d, s in zip(data, selectors) if s)
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3468 */
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3469
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3470 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3471 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3472 PyObject *data;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3473 PyObject *selectors;
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3474 } compressobject;
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3475
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3476 static PyTypeObject compress_type;
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3477
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3478 static PyObject *
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3479 compress_new(PyTypeObject type, PyObject args, PyObject *kwds)
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3480 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3481 PyObject seq1, seq2;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3482 PyObject data=NULL, selectors=NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3483 compressobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3484 static char *kwargs[] = {"data", "selectors", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3485
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3486 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OO:compress", kwargs, &seq1, &seq2))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3487 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3488
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3489 data = PyObject_GetIter(seq1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3490 if (data == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3491 goto fail;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3492 selectors = PyObject_GetIter(seq2);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3493 if (selectors == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3494 goto fail;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3495
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3496 /* create compressobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3497 lz = (compressobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3498 if (lz == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3499 goto fail;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3500 lz->data = data;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3501 lz->selectors = selectors;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3502 return (PyObject *)lz;
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3503
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3504 fail:
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3505 Py_XDECREF(data);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3506 Py_XDECREF(selectors);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3507 return NULL;
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3508 }
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3509
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3510 static void
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3511 compress_dealloc(compressobject *lz)
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3512 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3513 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3514 Py_XDECREF(lz->data);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3515 Py_XDECREF(lz->selectors);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3516 Py_TYPE(lz)->tp_free(lz);
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3517 }
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3518
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3519 static int
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3520 compress_traverse(compressobject lz, visitproc visit, void arg)
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3521 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3522 Py_VISIT(lz->data);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3523 Py_VISIT(lz->selectors);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3524 return 0;
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3525 }
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3526
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3527 static PyObject *
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3528 compress_next(compressobject *lz)
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3529 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3530 PyObject data = lz->data, selectors = lz->selectors;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3531 PyObject datum, selector;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3532 PyObject (datanext)(PyObject ) = Py_TYPE(data)->tp_iternext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3533 PyObject (selectornext)(PyObject ) = Py_TYPE(selectors)->tp_iternext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3534 int ok;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3535
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3536 while (1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3537 /* Steps: get datum, get selector, evaluate selector.
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3538 Order is important (to match the pure python version
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3539 in terms of which input gets a chance to raise an
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3540 exception first).
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3541 */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3542
83166 7b75f0bd9a5e Revert a premature patch for issue #14010 (changeset aaaf36026511). Serhiy Storchaka <storchaka@gmail.com> parents: 83153 diff changeset	3543 datum = datanext(data);
7b75f0bd9a5e Revert a premature patch for issue #14010 (changeset aaaf36026511). Serhiy Storchaka <storchaka@gmail.com> parents: 83153 diff changeset	3544 if (datum == NULL)
83153 aaaf36026511 Issue #14010: Fix a crash when iterating or deleting deeply nested filters Serhiy Storchaka <storchaka@gmail.com> parents: 81705 diff changeset	3545 return NULL;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3546
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3547 selector = selectornext(selectors);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3548 if (selector == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3549 Py_DECREF(datum);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3550 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3551 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3552
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3553 ok = PyObject_IsTrue(selector);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3554 Py_DECREF(selector);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3555 if (ok == 1)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3556 return datum;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3557 Py_DECREF(datum);
78605 b878df1d23b1 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 76946 78604 diff changeset	3558 if (ok < 0)
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3559 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3560 }
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3561 }
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3562
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3563 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3564 compress_reduce(compressobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3565 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3566 return Py_BuildValue("O(OO)", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3567 lz->data, lz->selectors);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3568 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3569
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3570 static PyMethodDef compress_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3571 {"__reduce__", (PyCFunction)compress_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3572 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3573 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3574 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3575
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3576 PyDoc_STRVAR(compress_doc,
51717 e260d6daf784 Add keyword arg support to itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 51666 diff changeset	3577 "compress(data, selectors) --> iterator over selected data\n\
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3578 \n\
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3579 Return data elements corresponding to true selector elements.\n\
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3580 Forms a shorter iterator from selected data elements using the\n\
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3581 selectors to choose the data elements.");
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3582
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3583 static PyTypeObject compress_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3584 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3585 "itertools.compress", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3586 sizeof(compressobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3587 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3588 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3589 (destructor)compress_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3590 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3591 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3592 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3593 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3594 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3595 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3596 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3597 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3598 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3599 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3600 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3601 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3602 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3603 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3604 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3605 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3606 compress_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3607 (traverseproc)compress_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3608 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3609 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3610 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3611 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3612 (iternextfunc)compress_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3613 compress_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3614 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3615 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3616 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3617 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3618 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3619 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3620 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3621 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3622 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3623 compress_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3624 PyObject_GC_Del, /* tp_free */
51063 ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3625 };
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3626
ad1db0ae98c6 Forward port r68941 adding itertools.compress(). Raymond Hettinger <python@rcn.com> parents: 50755 diff changeset	3627
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3628 /* filterfalse object ************************************************************/
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3629
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3630 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3631 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3632 PyObject *func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3633 PyObject *it;
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3634 } filterfalseobject;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3635
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3636 static PyTypeObject filterfalse_type;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3637
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3638 static PyObject *
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3639 filterfalse_new(PyTypeObject type, PyObject args, PyObject *kwds)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3640 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3641 PyObject func, seq;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3642 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3643 filterfalseobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3644
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3645 if (type == &filterfalse_type &&
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3646 !_PyArg_NoKeywords("filterfalse()", kwds))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3647 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3648
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3649 if (!PyArg_UnpackTuple(args, "filterfalse", 2, 2, &func, &seq))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3650 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3651
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3652 /* Get iterator. */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3653 it = PyObject_GetIter(seq);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3654 if (it == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3655 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3656
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3657 /* create filterfalseobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3658 lz = (filterfalseobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3659 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3660 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3661 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3662 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3663 Py_INCREF(func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3664 lz->func = func;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3665 lz->it = it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3666
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3667 return (PyObject *)lz;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3668 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3669
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3670 static void
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3671 filterfalse_dealloc(filterfalseobject *lz)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3672 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3673 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3674 Py_XDECREF(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3675 Py_XDECREF(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3676 Py_TYPE(lz)->tp_free(lz);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3677 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3678
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3679 static int
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3680 filterfalse_traverse(filterfalseobject lz, visitproc visit, void arg)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3681 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3682 Py_VISIT(lz->it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3683 Py_VISIT(lz->func);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3684 return 0;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3685 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3686
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3687 static PyObject *
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3688 filterfalse_next(filterfalseobject *lz)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3689 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3690 PyObject *item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3691 PyObject *it = lz->it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3692 long ok;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3693 PyObject (iternext)(PyObject *);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3694
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3695 iternext = *Py_TYPE(it)->tp_iternext;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3696 for (;;) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3697 item = iternext(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3698 if (item == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3699 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3700
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3701 if (lz->func == Py_None \|\| lz->func == (PyObject *)&PyBool_Type) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3702 ok = PyObject_IsTrue(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3703 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3704 PyObject *good;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3705 good = PyObject_CallFunctionObjArgs(lz->func,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3706 item, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3707 if (good == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3708 Py_DECREF(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3709 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3710 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3711 ok = PyObject_IsTrue(good);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3712 Py_DECREF(good);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3713 }
78604 56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	3714 if (ok == 0)
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3715 return item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3716 Py_DECREF(item);
78604 56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	3717 if (ok < 0)
56dc7b09f390 Issue #15604: Update uses of PyObject_IsTrue() to check for and handle errors correctly. Antoine Pitrou <solipsis@pitrou.net> parents: 72459 diff changeset	3718 return NULL;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3719 }
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3720 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3721
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3722 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3723 filterfalse_reduce(filterfalseobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3724 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3725 return Py_BuildValue("O(OO)", Py_TYPE(lz),
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3726 lz->func, lz->it);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3727 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3728
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3729 static PyMethodDef filterfalse_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3730 {"__reduce__", (PyCFunction)filterfalse_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3731 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3732 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3733 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3734
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3735 PyDoc_STRVAR(filterfalse_doc,
7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3736 "filterfalse(function or None, sequence) --> filterfalse object\n\
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3737 \n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3738 Return those items of sequence for which function(item) is false.\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3739 If function is None, return the items that are false.");
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3740
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	3741 static PyTypeObject filterfalse_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3742 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3743 "itertools.filterfalse", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3744 sizeof(filterfalseobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3745 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3746 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3747 (destructor)filterfalse_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3748 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3749 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3750 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3751 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3752 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3753 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3754 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3755 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3756 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3757 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3758 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3759 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3760 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3761 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3762 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3763 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3764 filterfalse_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3765 (traverseproc)filterfalse_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3766 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3767 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3768 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3769 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3770 (iternextfunc)filterfalse_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3771 filterfalse_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3772 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3773 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3774 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3775 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3776 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3777 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3778 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3779 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3780 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3781 filterfalse_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3782 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3783 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3784
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3785
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3786 /* count object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3787
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3788 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3789 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3790 Py_ssize_t cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3791 PyObject *long_cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3792 PyObject *long_step;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3793 } countobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3794
51542 b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3795 /* Counting logic and invariants:
b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3796
51778 aa29d0967105 Port r69837: Fix keyword arguments for itertools.count(). Step arg without a start arg was ignored. Raymond Hettinger <python@rcn.com> parents: 51719 diff changeset	3797 fast_mode: when cnt an integer < PY_SSIZE_T_MAX and no step is specified.
51542 b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3798
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3799 assert(cnt != PY_SSIZE_T_MAX && long_cnt == NULL && long_step==PyInt(1));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3800 Advances with: cnt += 1
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3801 When count hits Y_SSIZE_T_MAX, switch to slow_mode.
51778 aa29d0967105 Port r69837: Fix keyword arguments for itertools.count(). Step arg without a start arg was ignored. Raymond Hettinger <python@rcn.com> parents: 51719 diff changeset	3802
aa29d0967105 Port r69837: Fix keyword arguments for itertools.count(). Step arg without a start arg was ignored. Raymond Hettinger <python@rcn.com> parents: 51719 diff changeset	3803 slow_mode: when cnt == PY_SSIZE_T_MAX, step is not int(1), or cnt is a float.
51542 b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3804
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3805 assert(cnt == PY_SSIZE_T_MAX && long_cnt != NULL && long_step != NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3806 All counting is done with python objects (no overflows or underflows).
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3807 Advances with: long_cnt += long_step
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3808 Step may be zero -- effectively a slow version of repeat(cnt).
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3809 Either long_cnt or long_step may be a float, Fraction, or Decimal.
51542 b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3810 */
b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3811
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	3812 static PyTypeObject count_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3813
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3814 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3815 count_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3816 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3817 countobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3818 int slow_mode = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3819 Py_ssize_t cnt = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3820 PyObject *long_cnt = NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3821 PyObject *long_step = NULL;
69062 9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3822 long step;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3823 static char *kwlist[] = {"start", "step", 0};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3824
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3825 if (!PyArg_ParseTupleAndKeywords(args, kwds, "\|OO:count",
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3826 kwlist, &long_cnt, &long_step))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3827 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3828
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3829 if ((long_cnt != NULL && !PyNumber_Check(long_cnt)) \|\|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3830 (long_step != NULL && !PyNumber_Check(long_step))) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3831 PyErr_SetString(PyExc_TypeError, "a number is required");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3832 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3833 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3834
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3835 if (long_cnt != NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3836 cnt = PyLong_AsSsize_t(long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3837 if ((cnt == -1 && PyErr_Occurred()) \|\| !PyLong_Check(long_cnt)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3838 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3839 slow_mode = 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3840 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3841 Py_INCREF(long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3842 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3843 cnt = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3844 long_cnt = PyLong_FromLong(0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3845 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3846
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3847 /* If not specified, step defaults to 1 */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3848 if (long_step == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3849 long_step = PyLong_FromLong(1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3850 if (long_step == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3851 Py_DECREF(long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3852 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3853 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3854 } else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3855 Py_INCREF(long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3856
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3857 assert(long_cnt != NULL && long_step != NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3858
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3859 /* Fast mode only works when the step is 1 */
69062 9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3860 step = PyLong_AsLong(long_step);
9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3861 if (step != 1) {
9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3862 slow_mode = 1;
9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3863 if (step == -1 && PyErr_Occurred())
9a6da9bc257d Bugfix: Properly test for errors from PyLong_AsLong() in itertools.cycle. Kristjan Valur Jonsson <sweskman@gmail.com> parents: 66546 diff changeset	3864 PyErr_Clear();
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3865 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3866
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3867 if (slow_mode)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3868 cnt = PY_SSIZE_T_MAX;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3869 else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3870 Py_CLEAR(long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3871
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3872 assert((cnt != PY_SSIZE_T_MAX && long_cnt == NULL && !slow_mode) \|\|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3873 (cnt == PY_SSIZE_T_MAX && long_cnt != NULL && slow_mode));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3874 assert(slow_mode \|\|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3875 (PyLong_Check(long_step) && PyLong_AS_LONG(long_step) == 1));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3876
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3877 /* create countobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3878 lz = (countobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3879 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3880 Py_XDECREF(long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3881 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3882 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3883 lz->cnt = cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3884 lz->long_cnt = long_cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3885 lz->long_step = long_step;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3886
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3887 return (PyObject *)lz;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3888 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3889
43734 3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3890 static void
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3891 count_dealloc(countobject *lz)
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3892 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3893 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3894 Py_XDECREF(lz->long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3895 Py_XDECREF(lz->long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3896 Py_TYPE(lz)->tp_free(lz);
51661 003e00797e91 Add GC support to count() objects. Raymond Hettinger <python@rcn.com> parents: 51624 diff changeset	3897 }
003e00797e91 Add GC support to count() objects. Raymond Hettinger <python@rcn.com> parents: 51624 diff changeset	3898
51666 f308953c701f Merged revisions 69688,69690 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 51661 diff changeset	3899 static int
51661 003e00797e91 Add GC support to count() objects. Raymond Hettinger <python@rcn.com> parents: 51624 diff changeset	3900 count_traverse(countobject lz, visitproc visit, void arg)
003e00797e91 Add GC support to count() objects. Raymond Hettinger <python@rcn.com> parents: 51624 diff changeset	3901 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3902 Py_VISIT(lz->long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3903 Py_VISIT(lz->long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3904 return 0;
43734 3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3905 }
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3906
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3907 static PyObject *
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3908 count_nextlong(countobject *lz)
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3909 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3910 PyObject *long_cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3911 PyObject *stepped_up;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3912
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3913 long_cnt = lz->long_cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3914 if (long_cnt == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3915 /* Switch to slow_mode */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3916 long_cnt = PyLong_FromSsize_t(PY_SSIZE_T_MAX);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3917 if (long_cnt == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3918 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3919 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3920 assert(lz->cnt == PY_SSIZE_T_MAX && long_cnt != NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3921
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3922 stepped_up = PyNumber_Add(long_cnt, lz->long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3923 if (stepped_up == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3924 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3925 lz->long_cnt = stepped_up;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3926 return long_cnt;
43734 3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3927 }
3e5edf19d5f0 Merged revisions 58221-58741 via svnmerge from Guido van Rossum <guido@python.org> parents: 42334 diff changeset	3928
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3929 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3930 count_next(countobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3931 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3932 if (lz->cnt == PY_SSIZE_T_MAX)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3933 return count_nextlong(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3934 return PyLong_FromSsize_t(lz->cnt++);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3935 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3936
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3937 static PyObject *
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3938 count_repr(countobject *lz)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3939 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3940 if (lz->cnt != PY_SSIZE_T_MAX)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3941 return PyUnicode_FromFormat("count(%zd)", lz->cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3942
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3943 if (PyLong_Check(lz->long_step)) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3944 long step = PyLong_AsLong(lz->long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3945 if (step == -1 && PyErr_Occurred()) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3946 PyErr_Clear();
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3947 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3948 if (step == 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3949 /* Don't display step when it is an integer equal to 1 */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3950 return PyUnicode_FromFormat("count(%R)", lz->long_cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3951 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3952 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3953 return PyUnicode_FromFormat("count(%R, %R)",
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3954 lz->long_cnt, lz->long_step);
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3955 }
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3956
57525 0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3957 static PyObject *
0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3958 count_reduce(countobject *lz)
0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3959 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3960 if (lz->cnt == PY_SSIZE_T_MAX)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3961 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->long_cnt, lz->long_step);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3962 return Py_BuildValue("O(n)", Py_TYPE(lz), lz->cnt);
57525 0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3963 }
0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3964
0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3965 static PyMethodDef count_methods[] = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3966 {"__reduce__", (PyCFunction)count_reduce, METH_NOARGS,
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	3967 reduce_doc},
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3968 {NULL, NULL} /* sentinel */
57525 0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3969 };
0a6137944e61 Issue #7410: deepcopy of itertools.count was resetting the count. Raymond Hettinger <python@rcn.com> parents: 57344 diff changeset	3970
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3971 PyDoc_STRVAR(count_doc,
62239 8319ca0d5434 Merged revisions 81889 via svnmerge from Ezio Melotti <ezio.melotti@gmail.com> parents: 61466 diff changeset	3972 "count(start=0, step=1) --> count object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3973 \n\
51778 aa29d0967105 Port r69837: Fix keyword arguments for itertools.count(). Step arg without a start arg was ignored. Raymond Hettinger <python@rcn.com> parents: 51719 diff changeset	3974 Return a count object whose .__next__() method returns consecutive values.\n\
aa29d0967105 Port r69837: Fix keyword arguments for itertools.count(). Step arg without a start arg was ignored. Raymond Hettinger <python@rcn.com> parents: 51719 diff changeset	3975 Equivalent to:\n\n\
51542 b0db109089c9 Issue 5032: added a step argument to itertools.count() and allowed non-integer arguments. Raymond Hettinger <python@rcn.com> parents: 51503 diff changeset	3976 def count(firstval=0, step=1):\n\
85507 8e174ee0575a Issue #18912: Fix indentation in docstring Eli Bendersky <eliben@gmail.com> parents: 84263 diff changeset	3977 x = firstval\n\
8e174ee0575a Issue #18912: Fix indentation in docstring Eli Bendersky <eliben@gmail.com> parents: 84263 diff changeset	3978 while 1:\n\
8e174ee0575a Issue #18912: Fix indentation in docstring Eli Bendersky <eliben@gmail.com> parents: 84263 diff changeset	3979 yield x\n\
8e174ee0575a Issue #18912: Fix indentation in docstring Eli Bendersky <eliben@gmail.com> parents: 84263 diff changeset	3980 x += step\n");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3981
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	3982 static PyTypeObject count_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3983 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3984 "itertools.count", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3985 sizeof(countobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3986 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3987 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3988 (destructor)count_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3989 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3990 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3991 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3992 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3993 (reprfunc)count_repr, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3994 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3995 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3996 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3997 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3998 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	3999 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4000 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4001 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4002 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4003 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4004 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4005 count_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4006 (traverseproc)count_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4007 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4008 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4009 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4010 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4011 (iternextfunc)count_next, /* tp_iternext */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4012 count_methods, /* tp_methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4013 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4014 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4015 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4016 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4017 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4018 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4019 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4020 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4021 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4022 count_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4023 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4024 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4025
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4026
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4027 /* repeat object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4028
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4029 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4030 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4031 PyObject *element;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4032 Py_ssize_t cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4033 } repeatobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4034
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	4035 static PyTypeObject repeat_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4036
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4037 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4038 repeat_new(PyTypeObject type, PyObject args, PyObject *kwds)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4039 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4040 repeatobject *ro;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4041 PyObject *element;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4042 Py_ssize_t cnt = -1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4043 static char *kwargs[] = {"object", "times", NULL};
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4044
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4045 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|n:repeat", kwargs,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4046 &element, &cnt))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4047 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4048
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4049 if (PyTuple_Size(args) == 2 && cnt < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4050 cnt = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4051
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4052 ro = (repeatobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4053 if (ro == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4054 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4055 Py_INCREF(element);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4056 ro->element = element;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4057 ro->cnt = cnt;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4058 return (PyObject *)ro;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4059 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4060
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4061 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4062 repeat_dealloc(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4063 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4064 PyObject_GC_UnTrack(ro);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4065 Py_XDECREF(ro->element);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4066 Py_TYPE(ro)->tp_free(ro);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4067 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4068
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4069 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4070 repeat_traverse(repeatobject ro, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4071 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4072 Py_VISIT(ro->element);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4073 return 0;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4074 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4075
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4076 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4077 repeat_next(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4078 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4079 if (ro->cnt == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4080 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4081 if (ro->cnt > 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4082 ro->cnt--;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4083 Py_INCREF(ro->element);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4084 return ro->element;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4085 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4086
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	4087 static PyObject *
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	4088 repeat_repr(repeatobject *ro)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	4089 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4090 if (ro->cnt == -1)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4091 return PyUnicode_FromFormat("repeat(%R)", ro->element);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4092 else
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4093 return PyUnicode_FromFormat("repeat(%R, %zd)", ro->element, ro->cnt);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4094 }
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	4095
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	4096 static PyObject *
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4097 repeat_len(repeatobject *ro)
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4098 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4099 if (ro->cnt == -1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4100 PyErr_SetString(PyExc_TypeError, "len() of unsized object");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4101 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4102 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4103 return PyLong_FromSize_t(ro->cnt);
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4104 }
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4105
36484 20f75d5e79e9 Renamed _length_cue() to __length_hint__(). See: Armin Rigo <arigo@tunes.org> parents: 36324 diff changeset	4106 PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	4107
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4108 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4109 repeat_reduce(repeatobject *ro)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4110 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4111 /* unpickle this so that a new repeat iterator is constructed with an
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4112 * object, then call __setstate__ on it to set cnt
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4113 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4114 if (ro->cnt >= 0)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4115 return Py_BuildValue("O(On)", Py_TYPE(ro), ro->element, ro->cnt);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4116 else
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4117 return Py_BuildValue("O(O)", Py_TYPE(ro), ro->element);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4118 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4119
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	4120 static PyMethodDef repeat_methods[] = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4121 {"__length_hint__", (PyCFunction)repeat_len, METH_NOARGS, length_hint_doc},
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4122 {"__reduce__", (PyCFunction)repeat_reduce, METH_NOARGS, reduce_doc},
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4123 {NULL, NULL} /* sentinel */
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4124 };
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	4125
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4126 PyDoc_STRVAR(repeat_doc,
51719 3dbdbc5e6d85 Add keyword arg support to itertools.repeat(). Raymond Hettinger <python@rcn.com> parents: 51717 diff changeset	4127 "repeat(object [,times]) -> create an iterator which returns the object\n\
3dbdbc5e6d85 Add keyword arg support to itertools.repeat(). Raymond Hettinger <python@rcn.com> parents: 51717 diff changeset	4128 for the specified number of times. If not specified, returns the object\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	4129 endlessly.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4130
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	4131 static PyTypeObject repeat_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4132 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4133 "itertools.repeat", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4134 sizeof(repeatobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4135 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4136 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4137 (destructor)repeat_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4138 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4139 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4140 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4141 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4142 (reprfunc)repeat_repr, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4143 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4144 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4145 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4146 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4147 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4148 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4149 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4150 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4151 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4152 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4153 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4154 repeat_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4155 (traverseproc)repeat_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4156 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4157 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4158 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4159 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4160 (iternextfunc)repeat_next, /* tp_iternext */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4161 repeat_methods, /* tp_methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4162 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4163 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4164 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4165 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4166 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4167 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4168 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4169 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4170 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4171 repeat_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4172 PyObject_GC_Del, /* tp_free */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4173 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4174
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4175 /* ziplongest object ************************************************************/
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4176
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4177 #include "Python.h"
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4178
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4179 typedef struct {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4180 PyObject_HEAD
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4181 Py_ssize_t tuplesize;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4182 Py_ssize_t numactive;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4183 PyObject ittuple; / tuple of iterators */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4184 PyObject *result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4185 PyObject *fillvalue;
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4186 } ziplongestobject;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4187
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4188 static PyTypeObject ziplongest_type;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4189
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4190 static PyObject *
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	4191 zip_longest_new(PyTypeObject type, PyObject args, PyObject *kwds)
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4192 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4193 ziplongestobject *lz;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4194 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4195 PyObject ittuple; / tuple of iterators */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4196 PyObject *result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4197 PyObject *fillvalue = Py_None;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4198 Py_ssize_t tuplesize = PySequence_Length(args);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4199
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4200 if (kwds != NULL && PyDict_CheckExact(kwds) && PyDict_Size(kwds) > 0) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4201 fillvalue = PyDict_GetItemString(kwds, "fillvalue");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4202 if (fillvalue == NULL \|\| PyDict_Size(kwds) > 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4203 PyErr_SetString(PyExc_TypeError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4204 "zip_longest() got an unexpected keyword argument");
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4205 return NULL;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4206 }
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4207 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4208
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4209 /* args must be a tuple */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4210 assert(PyTuple_Check(args));
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4211
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4212 /* obtain iterators */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4213 ittuple = PyTuple_New(tuplesize);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4214 if (ittuple == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4215 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4216 for (i=0; i < tuplesize; ++i) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4217 PyObject *item = PyTuple_GET_ITEM(args, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4218 PyObject *it = PyObject_GetIter(item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4219 if (it == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4220 if (PyErr_ExceptionMatches(PyExc_TypeError))
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4221 PyErr_Format(PyExc_TypeError,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4222 "zip_longest argument #%zd must support iteration",
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4223 i+1);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4224 Py_DECREF(ittuple);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4225 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4226 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4227 PyTuple_SET_ITEM(ittuple, i, it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4228 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4229
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4230 /* create a result holder */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4231 result = PyTuple_New(tuplesize);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4232 if (result == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4233 Py_DECREF(ittuple);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4234 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4235 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4236 for (i=0 ; i < tuplesize ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4237 Py_INCREF(Py_None);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4238 PyTuple_SET_ITEM(result, i, Py_None);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4239 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4240
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4241 /* create ziplongestobject structure */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4242 lz = (ziplongestobject *)type->tp_alloc(type, 0);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4243 if (lz == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4244 Py_DECREF(ittuple);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4245 Py_DECREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4246 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4247 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4248 lz->ittuple = ittuple;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4249 lz->tuplesize = tuplesize;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4250 lz->numactive = tuplesize;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4251 lz->result = result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4252 Py_INCREF(fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4253 lz->fillvalue = fillvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4254 return (PyObject *)lz;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4255 }
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4256
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4257 static void
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4258 zip_longest_dealloc(ziplongestobject *lz)
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4259 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4260 PyObject_GC_UnTrack(lz);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4261 Py_XDECREF(lz->ittuple);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4262 Py_XDECREF(lz->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4263 Py_XDECREF(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4264 Py_TYPE(lz)->tp_free(lz);
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4265 }
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4266
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4267 static int
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4268 zip_longest_traverse(ziplongestobject lz, visitproc visit, void arg)
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4269 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4270 Py_VISIT(lz->ittuple);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4271 Py_VISIT(lz->result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4272 Py_VISIT(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4273 return 0;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4274 }
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4275
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4276 static PyObject *
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4277 zip_longest_next(ziplongestobject *lz)
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4278 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4279 Py_ssize_t i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4280 Py_ssize_t tuplesize = lz->tuplesize;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4281 PyObject *result = lz->result;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4282 PyObject *it;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4283 PyObject *item;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4284 PyObject *olditem;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4285
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4286 if (tuplesize == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4287 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4288 if (lz->numactive == 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4289 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4290 if (Py_REFCNT(result) == 1) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4291 Py_INCREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4292 for (i=0 ; i < tuplesize ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4293 it = PyTuple_GET_ITEM(lz->ittuple, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4294 if (it == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4295 Py_INCREF(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4296 item = lz->fillvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4297 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4298 item = PyIter_Next(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4299 if (item == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4300 lz->numactive -= 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4301 if (lz->numactive == 0 \|\| PyErr_Occurred()) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4302 lz->numactive = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4303 Py_DECREF(result);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4304 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4305 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4306 Py_INCREF(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4307 item = lz->fillvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4308 PyTuple_SET_ITEM(lz->ittuple, i, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4309 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4310 }
57042 658eb58d2e68 Fix exception handling in itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 54839 diff changeset	4311 }
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4312 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4313 olditem = PyTuple_GET_ITEM(result, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4314 PyTuple_SET_ITEM(result, i, item);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4315 Py_DECREF(olditem);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4316 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4317 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4318 result = PyTuple_New(tuplesize);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4319 if (result == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4320 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4321 for (i=0 ; i < tuplesize ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4322 it = PyTuple_GET_ITEM(lz->ittuple, i);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4323 if (it == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4324 Py_INCREF(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4325 item = lz->fillvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4326 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4327 item = PyIter_Next(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4328 if (item == NULL) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4329 lz->numactive -= 1;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4330 if (lz->numactive == 0 \|\| PyErr_Occurred()) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4331 lz->numactive = 0;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4332 Py_DECREF(result);
57042 658eb58d2e68 Fix exception handling in itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 54839 diff changeset	4333 return NULL;
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4334 } else {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4335 Py_INCREF(lz->fillvalue);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4336 item = lz->fillvalue;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4337 PyTuple_SET_ITEM(lz->ittuple, i, NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4338 Py_DECREF(it);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4339 }
57042 658eb58d2e68 Fix exception handling in itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 54839 diff changeset	4340 }
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4341 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4342 PyTuple_SET_ITEM(result, i, item);
57042 658eb58d2e68 Fix exception handling in itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 54839 diff changeset	4343 }
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4344 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4345 return result;
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4346 }
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4347
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4348 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4349 zip_longest_reduce(ziplongestobject *lz)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4350 {
81705 20c4d55f4b69 Clean trailing whitespace in itertoolsmodule.c. Serhiy Storchaka <storchaka@gmail.com> parents: 81701 81704 diff changeset	4351
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4352 /* Create a new tuple with empty sequences where appropriate to pickle.
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4353 * Then use setstate to set the fillvalue
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4354 */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4355 int i;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4356 PyObject *args = PyTuple_New(PyTuple_GET_SIZE(lz->ittuple));
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4357 if (args == NULL)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4358 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4359 for (i=0; i<PyTuple_GET_SIZE(lz->ittuple); i++) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4360 PyObject *elem = PyTuple_GET_ITEM(lz->ittuple, i);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4361 if (elem == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4362 elem = PyTuple_New(0);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4363 if (elem == NULL) {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4364 Py_DECREF(args);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4365 return NULL;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4366 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4367 } else
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4368 Py_INCREF(elem);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4369 PyTuple_SET_ITEM(args, i, elem);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4370 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4371 return Py_BuildValue("ONO", Py_TYPE(lz), args, lz->fillvalue);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4372 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4373
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4374 static PyObject *
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4375 zip_longest_setstate(ziplongestobject lz, PyObject state)
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4376 {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4377 Py_CLEAR(lz->fillvalue);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4378 lz->fillvalue = state;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4379 Py_INCREF(lz->fillvalue);
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4380 Py_RETURN_NONE;
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4381 }
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4382
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4383 static PyMethodDef zip_longest_methods[] = {
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4384 {"__reduce__", (PyCFunction)zip_longest_reduce, METH_NOARGS,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4385 reduce_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4386 {"__setstate__", (PyCFunction)zip_longest_setstate, METH_O,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4387 setstate_doc},
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4388 {NULL, NULL} /* sentinel */
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4389 };
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4390
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	4391 PyDoc_STRVAR(zip_longest_doc,
7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	4392 "zip_longest(iter1 [,iter2 [...]], [fillvalue=None]) --> zip_longest object\n\
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4393 \n\
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	4394 Return an zip_longest object whose .__next__() method returns a tuple where\n\
41501 0c5019afc55b PEP 3114: rename .next() to .__next__() and add next() builtin. Georg Brandl <georg@python.org> parents: 40893 diff changeset	4395 the i-th element comes from the i-th iterable argument. The .__next__()\n\
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4396 method continues until the longest iterable in the argument sequence\n\
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4397 is exhausted and then it raises StopIteration. When the shorter iterables\n\
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4398 are exhausted, the fillvalue is substituted in their place. The fillvalue\n\
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4399 defaults to None or can be specified by a keyword argument.\n\
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4400 ");
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4401
45738 740b5808c57a Move itertools izip() code to builtins as zip(). Complete the renaming. Raymond Hettinger <python@rcn.com> parents: 45736 diff changeset	4402 static PyTypeObject ziplongest_type = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4403 PyVarObject_HEAD_INIT(NULL, 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4404 "itertools.zip_longest", /* tp_name */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4405 sizeof(ziplongestobject), /* tp_basicsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4406 0, /* tp_itemsize */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4407 /* methods */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4408 (destructor)zip_longest_dealloc, /* tp_dealloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4409 0, /* tp_print */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4410 0, /* tp_getattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4411 0, /* tp_setattr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4412 0, /* tp_reserved */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4413 0, /* tp_repr */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4414 0, /* tp_as_number */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4415 0, /* tp_as_sequence */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4416 0, /* tp_as_mapping */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4417 0, /* tp_hash */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4418 0, /* tp_call */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4419 0, /* tp_str */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4420 PyObject_GenericGetAttr, /* tp_getattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4421 0, /* tp_setattro */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4422 0, /* tp_as_buffer */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4423 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4424 Py_TPFLAGS_BASETYPE, /* tp_flags */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4425 zip_longest_doc, /* tp_doc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4426 (traverseproc)zip_longest_traverse, /* tp_traverse */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4427 0, /* tp_clear */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4428 0, /* tp_richcompare */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4429 0, /* tp_weaklistoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4430 PyObject_SelfIter, /* tp_iter */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4431 (iternextfunc)zip_longest_next, /* tp_iternext */
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4432 zip_longest_methods, /* tp_methods */
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4433 0, /* tp_members */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4434 0, /* tp_getset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4435 0, /* tp_base */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4436 0, /* tp_dict */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4437 0, /* tp_descr_get */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4438 0, /* tp_descr_set */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4439 0, /* tp_dictoffset */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4440 0, /* tp_init */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4441 0, /* tp_alloc */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4442 zip_longest_new, /* tp_new */
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4443 PyObject_GC_Del, /* tp_free */
40893 32c4897b5d46 Merged revisions 53623-53858 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40737 diff changeset	4444 };
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4445
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4446 /* module level code ********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4447
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4448 PyDoc_STRVAR(module_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4449 "Functional tools for creating and using iterators.\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4450 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4451 Infinite iterators:\n\
84236 ad0b44cdae41 Closes #18239: correct description of count() in module docstring Andrew Kuchling <amk@amk.ca> parents: 83166 diff changeset	4452 count(start=0, step=1) --> start, start+step, start+2*step, ...\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	4453 cycle(p) --> p0, p1, ... plast, p0, p1, ...\n\
28601 58db5fbd9c28 Fix docstring typo Andrew M. Kuchling <amk@amk.ca> parents: 28320 diff changeset	4454 repeat(elem [,n]) --> elem, elem, elem, ... endlessly or up to n times\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4455 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4456 Iterators terminating on the shortest input sequence:\n\
69022 822c7c0d27d1 Update docstrings for itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 69021 diff changeset	4457 accumulate(p[, func]) --> p0, p0+p1, p0+p1+p2\n\
51163 4f607c3ae8e2 Update itertools.__doc__ to include all tools. Raymond Hettinger <python@rcn.com> parents: 51100 diff changeset	4458 chain(p, q, ...) --> p0, p1, ... plast, q0, q1, ... \n\
85632 fa1fa88b685b Issue 18752: Make chain.from_iterable() more visible in the documentation. Raymond Hettinger <python@rcn.com> parents: 85511 diff changeset	4459 chain.from_iterable([p, q, ...]) --> p0, p1, ... plast, q0, q1, ... \n\
51163 4f607c3ae8e2 Update itertools.__doc__ to include all tools. Raymond Hettinger <python@rcn.com> parents: 51100 diff changeset	4460 compress(data, selectors) --> (d[0] if s[0]), (d[1] if s[1]), ...\n\
4f607c3ae8e2 Update itertools.__doc__ to include all tools. Raymond Hettinger <python@rcn.com> parents: 51100 diff changeset	4461 dropwhile(pred, seq) --> seq[n], seq[n+1], starting when pred fails\n\
4f607c3ae8e2 Update itertools.__doc__ to include all tools. Raymond Hettinger <python@rcn.com> parents: 51100 diff changeset	4462 groupby(iterable[, keyfunc]) --> sub-iterators grouped by value of keyfunc(v)\n\
45736 7752a57f9c49 Rename ifilterfalse() to filterfalse() and izip_longest() to zip_longest(). Raymond Hettinger <python@rcn.com> parents: 45735 diff changeset	4463 filterfalse(pred, seq) --> elements of seq where pred(elem) is False\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4464 islice(seq, [start,] stop [, step]) --> elements from\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4465 seq[start:stop:step]\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4466 starmap(fun, seq) --> fun(seq[0]), fun(seq[1]), ...\n\
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	4467 tee(it, n=2) --> (it1, it2 , ... itn) splits one iterator into n\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4468 takewhile(pred, seq) --> seq[0], seq[1], until pred fails\n\
51163 4f607c3ae8e2 Update itertools.__doc__ to include all tools. Raymond Hettinger <python@rcn.com> parents: 51100 diff changeset	4469 zip_longest(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ... \n\
51165 4b8976ea6966 Fix typo. Raymond Hettinger <python@rcn.com> parents: 51163 diff changeset	4470 \n\
4b8976ea6966 Fix typo. Raymond Hettinger <python@rcn.com> parents: 51163 diff changeset	4471 Combinatoric generators:\n\
4b8976ea6966 Fix typo. Raymond Hettinger <python@rcn.com> parents: 51163 diff changeset	4472 product(p, q, ... [repeat=1]) --> cartesian product\n\
4b8976ea6966 Fix typo. Raymond Hettinger <python@rcn.com> parents: 51163 diff changeset	4473 permutations(p[, r])\n\
57344 a790f84fbafa Fix docstrings for itertools combinatoric functions. Raymond Hettinger <python@rcn.com> parents: 57042 diff changeset	4474 combinations(p, r)\n\
a790f84fbafa Fix docstrings for itertools combinatoric functions. Raymond Hettinger <python@rcn.com> parents: 57042 diff changeset	4475 combinations_with_replacement(p, r)\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4476 ");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4477
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4478
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	4479 static PyMethodDef module_methods[] = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4480 {"tee", (PyCFunction)tee, METH_VARARGS, tee_doc},
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4481 {NULL, NULL} /* sentinel */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	4482 };
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	4483
47580 94f64f11fdd0 Implement PEP 3121: new module initialization and finalization API. Martin v. Löwis <martin@v.loewis.de> parents: 47555 diff changeset	4484
94f64f11fdd0 Implement PEP 3121: new module initialization and finalization API. Martin v. Löwis <martin@v.loewis.de> parents: 47555 diff changeset	4485 static struct PyModuleDef itertoolsmodule = {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4486 PyModuleDef_HEAD_INIT,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4487 "itertools",
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4488 module_doc,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4489 -1,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4490 module_methods,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4491 NULL,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4492 NULL,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4493 NULL,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4494 NULL
47580 94f64f11fdd0 Implement PEP 3121: new module initialization and finalization API. Martin v. Löwis <martin@v.loewis.de> parents: 47555 diff changeset	4495 };
94f64f11fdd0 Implement PEP 3121: new module initialization and finalization API. Martin v. Löwis <martin@v.loewis.de> parents: 47555 diff changeset	4496
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4497 PyMODINIT_FUNC
47580 94f64f11fdd0 Implement PEP 3121: new module initialization and finalization API. Martin v. Löwis <martin@v.loewis.de> parents: 47555 diff changeset	4498 PyInit_itertools(void)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4499 {
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4500 int i;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4501 PyObject *m;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4502 char *name;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4503 PyTypeObject *typelist[] = {
66589 5da093b49c2a Add itertools.accumulate(). Raymond Hettinger <python@rcn.com> parents: 66546 diff changeset	4504 &accumulate_type,
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4505 &combinations_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4506 &cwr_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4507 &cycle_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4508 &dropwhile_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4509 &takewhile_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4510 &islice_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4511 &starmap_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4512 &chain_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4513 &compress_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4514 &filterfalse_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4515 &count_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4516 &ziplongest_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4517 &permutations_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4518 &product_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4519 &repeat_type,
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4520 &groupby_type,
76091 4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4521 &_grouper_type,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4522 &tee_type,
4ff234337e24 Issue #14288: Serialization support for builtin iterators. Kristján Valur Jónsson <kristjan@ccpgames.com> parents: 72926 diff changeset	4523 &teedataobject_type,
61466 cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4524 NULL
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4525 };
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4526
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4527 Py_TYPE(&teedataobject_type) = &PyType_Type;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4528 m = PyModule_Create(&itertoolsmodule);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4529 if (m == NULL)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4530 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4531
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4532 for (i=0 ; typelist[i] != NULL ; i++) {
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4533 if (PyType_Ready(typelist[i]) < 0)
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4534 return NULL;
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4535 name = strchr(typelist[i]->tp_name, '.');
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4536 assert (name != NULL);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4537 Py_INCREF(typelist[i]);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4538 PyModule_AddObject(m, name+1, (PyObject *)typelist[i]);
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4539 }
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4540
cab14be0ada1 Recorded merge of revisions 81029 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 57525 diff changeset	4541 return m;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4542 }

Mercurial > cpython-lru_cache-weakref

annotate Modules/itertoolsmodule.c @ 87690:66c1c9f32567 default tip