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

annotate Modules/itertoolsmodule.c @ 45837:c5241069fffd legacy-trunk

Speed up test_dict by about 10x by only checking selected dict literal sizes, instead of every integer from 0 to 400. Exhaustive testing wastes time without providing enough more assurance that the code is correct.

author	Jeffrey Yasskin <jyasskin@gmail.com>
date	Tue, 18 Mar 2008 05:12:41 +0000
parents	d72d527de711
children	4ca7119d1700

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
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	5 /* Itertools module written and maintained
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>
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	7 Copyright (c) 2003 Python Software Foundation.
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 {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	15 PyObject_HEAD
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	16 PyObject *it;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	17 PyObject *keyfunc;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	18 PyObject *tgtkey;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	19 PyObject *currkey;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	20 PyObject *currvalue;
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 {
36684 0bebeb393045 More unconsting. Martin v. Löwis <martin@v.loewis.de> parents: 36518 diff changeset	29 static char *kwargs[] = {"iterable", "key", NULL};
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	30 groupbyobject *gbo;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	31 PyObject it, keyfunc = Py_None;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	32
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	33 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|O:groupby", kwargs,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	34 &it, &keyfunc))
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	35 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	36
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	37 gbo = (groupbyobject *)type->tp_alloc(type, 0);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	38 if (gbo == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	39 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	40 gbo->tgtkey = NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	41 gbo->currkey = NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	42 gbo->currvalue = NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	43 gbo->keyfunc = keyfunc;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	44 Py_INCREF(keyfunc);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	45 gbo->it = PyObject_GetIter(it);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	46 if (gbo->it == NULL) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	47 Py_DECREF(gbo);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	48 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	49 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	50 return (PyObject *)gbo;
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 {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	56 PyObject_GC_UnTrack(gbo);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	57 Py_XDECREF(gbo->it);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	58 Py_XDECREF(gbo->keyfunc);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	59 Py_XDECREF(gbo->tgtkey);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	60 Py_XDECREF(gbo->currkey);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	61 Py_XDECREF(gbo->currvalue);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 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 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	68 Py_VISIT(gbo->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	69 Py_VISIT(gbo->keyfunc);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	70 Py_VISIT(gbo->tgtkey);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	71 Py_VISIT(gbo->currkey);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	72 Py_VISIT(gbo->currvalue);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	73 return 0;
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 {
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	79 PyObject newvalue, newkey, r, grouper, *tmp;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	80
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	81 /* skip to next iteration group */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	82 for (;;) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	83 if (gbo->currkey == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	84 /* pass */;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	85 else if (gbo->tgtkey == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	86 break;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	87 else {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	88 int rcmp;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	89
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	90 rcmp = PyObject_RichCompareBool(gbo->tgtkey,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	91 gbo->currkey, Py_EQ);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	92 if (rcmp == -1)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	93 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	94 else if (rcmp == 0)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	95 break;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	96 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	97
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	98 newvalue = PyIter_Next(gbo->it);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	99 if (newvalue == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	100 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	101
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	102 if (gbo->keyfunc == Py_None) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	103 newkey = newvalue;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	104 Py_INCREF(newvalue);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	105 } else {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	106 newkey = PyObject_CallFunctionObjArgs(gbo->keyfunc,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	107 newvalue, NULL);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	108 if (newkey == NULL) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	109 Py_DECREF(newvalue);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	110 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	111 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	112 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	113
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	114 tmp = gbo->currkey;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	115 gbo->currkey = newkey;
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	116 Py_XDECREF(tmp);
31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	117
31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	118 tmp = gbo->currvalue;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	119 gbo->currvalue = newvalue;
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	120 Py_XDECREF(tmp);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	121 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	122
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	123 Py_INCREF(gbo->currkey);
31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	124 tmp = gbo->tgtkey;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	125 gbo->tgtkey = gbo->currkey;
33616 31b08670ebdd * Increase test coverage. Raymond Hettinger <python@rcn.com> parents: 33386 diff changeset	126 Py_XDECREF(tmp);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	127
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	128 grouper = _grouper_create(gbo, gbo->tgtkey);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	129 if (grouper == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	130 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	131
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	132 r = PyTuple_Pack(2, gbo->currkey, grouper);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	133 Py_DECREF(grouper);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	134 return r;
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
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	137 PyDoc_STRVAR(groupby_doc,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	138 "groupby(iterable[, keyfunc]) -> create an iterator which returns\n\
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	139 (key, sub-iterator) grouped by each value of key(value).\n");
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	140
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	141 static PyTypeObject groupby_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	142 PyVarObject_HEAD_INIT(NULL, 0)
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	143 "itertools.groupby", /* tp_name */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	144 sizeof(groupbyobject), /* tp_basicsize */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	145 0, /* tp_itemsize */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	146 /* methods */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	147 (destructor)groupby_dealloc, /* tp_dealloc */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	148 0, /* tp_print */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	149 0, /* tp_getattr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	150 0, /* tp_setattr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	151 0, /* tp_compare */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	152 0, /* tp_repr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	153 0, /* tp_as_number */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	154 0, /* tp_as_sequence */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	155 0, /* tp_as_mapping */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	156 0, /* tp_hash */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	157 0, /* tp_call */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	158 0, /* tp_str */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	159 PyObject_GenericGetAttr, /* tp_getattro */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	160 0, /* tp_setattro */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	161 0, /* tp_as_buffer */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	162 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	163 Py_TPFLAGS_BASETYPE, /* tp_flags */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	164 groupby_doc, /* tp_doc */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	165 (traverseproc)groupby_traverse, /* tp_traverse */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	166 0, /* tp_clear */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	167 0, /* tp_richcompare */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	168 0, /* tp_weaklistoffset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	169 PyObject_SelfIter, /* tp_iter */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	170 (iternextfunc)groupby_next, /* tp_iternext */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	171 0, /* tp_methods */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	172 0, /* tp_members */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	173 0, /* tp_getset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	174 0, /* tp_base */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	175 0, /* tp_dict */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	176 0, /* tp_descr_get */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	177 0, /* tp_descr_set */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	178 0, /* tp_dictoffset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	179 0, /* tp_init */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	180 0, /* tp_alloc */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	181 groupby_new, /* tp_new */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	182 PyObject_GC_Del, /* tp_free */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	183 };
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	184
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	185
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	186 /* _grouper object (internal) ************************************************/
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 typedef struct {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	189 PyObject_HEAD
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	190 PyObject *parent;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	191 PyObject *tgtkey;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	192 } _grouperobject;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	193
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	194 static PyTypeObject _grouper_type;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	195
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	196 static PyObject *
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	197 _grouper_create(groupbyobject parent, PyObject tgtkey)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	198 {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	199 _grouperobject *igo;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	200
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	201 igo = PyObject_GC_New(_grouperobject, &_grouper_type);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	202 if (igo == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	203 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	204 igo->parent = (PyObject *)parent;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	205 Py_INCREF(parent);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	206 igo->tgtkey = tgtkey;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	207 Py_INCREF(tgtkey);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	208
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	209 PyObject_GC_Track(igo);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	210 return (PyObject *)igo;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	211 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	212
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	213 static void
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	214 _grouper_dealloc(_grouperobject *igo)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	215 {
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	216 PyObject_GC_UnTrack(igo);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	217 Py_DECREF(igo->parent);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	218 Py_DECREF(igo->tgtkey);
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	219 PyObject_GC_Del(igo);
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	220 }
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	221
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	222 static int
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	223 _grouper_traverse(_grouperobject igo, visitproc visit, void arg)
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	224 {
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	225 Py_VISIT(igo->parent);
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	226 Py_VISIT(igo->tgtkey);
0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	227 return 0;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	228 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	229
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	230 static PyObject *
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	231 _grouper_next(_grouperobject *igo)
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 groupbyobject gbo = (groupbyobject )igo->parent;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	234 PyObject newvalue, newkey, *r;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	235 int rcmp;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	236
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	237 if (gbo->currvalue == NULL) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	238 newvalue = PyIter_Next(gbo->it);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	239 if (newvalue == NULL)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	240 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	241
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	242 if (gbo->keyfunc == Py_None) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	243 newkey = newvalue;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	244 Py_INCREF(newvalue);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	245 } else {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	246 newkey = PyObject_CallFunctionObjArgs(gbo->keyfunc,
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	247 newvalue, NULL);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	248 if (newkey == NULL) {
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	249 Py_DECREF(newvalue);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	250 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	251 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	252 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	253
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	254 assert(gbo->currkey == NULL);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	255 gbo->currkey = newkey;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	256 gbo->currvalue = newvalue;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	257 }
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	258
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	259 assert(gbo->currkey != NULL);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	260 rcmp = PyObject_RichCompareBool(igo->tgtkey, gbo->currkey, Py_EQ);
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	261 if (rcmp <= 0)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	262 /* got any error or current group is end */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	263 return NULL;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	264
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	265 r = gbo->currvalue;
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	266 gbo->currvalue = NULL;
33386 c2c49ee727d0 SF patch #1020188: Use Py_CLEAR where necessary to avoid crashes Raymond Hettinger <python@rcn.com> parents: 32682 diff changeset	267 Py_CLEAR(gbo->currkey);
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	268
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	269 return r;
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
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	272 static PyTypeObject _grouper_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	273 PyVarObject_HEAD_INIT(NULL, 0)
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	274 "itertools._grouper", /* tp_name */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	275 sizeof(_grouperobject), /* tp_basicsize */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	276 0, /* tp_itemsize */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	277 /* methods */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	278 (destructor)_grouper_dealloc, /* tp_dealloc */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	279 0, /* tp_print */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	280 0, /* tp_getattr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	281 0, /* tp_setattr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	282 0, /* tp_compare */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	283 0, /* tp_repr */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	284 0, /* tp_as_number */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	285 0, /* tp_as_sequence */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	286 0, /* tp_as_mapping */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	287 0, /* tp_hash */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	288 0, /* tp_call */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	289 0, /* tp_str */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	290 PyObject_GenericGetAttr, /* tp_getattro */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	291 0, /* tp_setattro */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	292 0, /* tp_as_buffer */
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	293 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	294 0, /* tp_doc */
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	295 (traverseproc)_grouper_traverse,/* tp_traverse */
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	296 0, /* tp_clear */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	297 0, /* tp_richcompare */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	298 0, /* tp_weaklistoffset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	299 PyObject_SelfIter, /* tp_iter */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	300 (iternextfunc)_grouper_next, /* tp_iternext */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	301 0, /* tp_methods */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	302 0, /* tp_members */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	303 0, /* tp_getset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	304 0, /* tp_base */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	305 0, /* tp_dict */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	306 0, /* tp_descr_get */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	307 0, /* tp_descr_set */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	308 0, /* tp_dictoffset */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	309 0, /* tp_init */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	310 0, /* tp_alloc */
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	311 0, /* tp_new */
45699 0be9b1cf1b16 Issue 2246: itertools grouper object did not participate in GC (should be backported). Raymond Hettinger <python@rcn.com> parents: 45676 diff changeset	312 PyObject_GC_Del, /* tp_free */
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	313 };
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	314
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	315
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	316
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	317 /* tee object and with supporting function and objects ***************/
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	318
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	319 /* 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	320 To help the object fit neatly inside cache lines (space for 16 to 32
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	321 pointers), the value should be a multiple of 16 minus space for
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	322 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	323 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	324 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	325 wasted space and the more rapid freeing of older data.
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	326 */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	327 #define LINKCELLS 57
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	328
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	329 typedef struct {
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	330 PyObject_HEAD
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	331 PyObject *it;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	332 int numread;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	333 PyObject *nextlink;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	334 PyObject *(values[LINKCELLS]);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	335 } teedataobject;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	336
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	337 typedef struct {
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	338 PyObject_HEAD
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	339 teedataobject *dataobj;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	340 int index;
33777 e4b7c923ed5f Fix and test weak referencing of itertools.tee objects. Raymond Hettinger <python@rcn.com> parents: 33646 diff changeset	341 PyObject *weakreflist;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	342 } teeobject;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	343
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	344 static PyTypeObject teedataobject_type;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	345
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	346 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	347 teedataobject_new(PyObject *it)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	348 {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	349 teedataobject *tdo;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	350
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	351 tdo = PyObject_GC_New(teedataobject, &teedataobject_type);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	352 if (tdo == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	353 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	354
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	355 tdo->numread = 0;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	356 tdo->nextlink = NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	357 Py_INCREF(it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	358 tdo->it = it;
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	359 PyObject_GC_Track(tdo);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	360 return (PyObject *)tdo;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	361 }
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	362
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	363 static PyObject *
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	364 teedataobject_jumplink(teedataobject *tdo)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	365 {
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	366 if (tdo->nextlink == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	367 tdo->nextlink = teedataobject_new(tdo->it);
39144 e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	368 Py_XINCREF(tdo->nextlink);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	369 return tdo->nextlink;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	370 }
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	371
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	372 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	373 teedataobject_getitem(teedataobject *tdo, int i)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	374 {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	375 PyObject *value;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	376
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	377 assert(i < LINKCELLS);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	378 if (i < tdo->numread)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	379 value = tdo->values[i];
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	380 else {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	381 /* this is the lead iterator, so fetch more data */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	382 assert(i == tdo->numread);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	383 value = PyIter_Next(tdo->it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	384 if (value == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	385 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	386 tdo->numread++;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	387 tdo->values[i] = value;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	388 }
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	389 Py_INCREF(value);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	390 return value;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	391 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	392
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	393 static int
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	394 teedataobject_traverse(teedataobject tdo, visitproc visit, void arg)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	395 {
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	396 int i;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	397 Py_VISIT(tdo->it);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	398 for (i = 0; i < tdo->numread; i++)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	399 Py_VISIT(tdo->values[i]);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	400 Py_VISIT(tdo->nextlink);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	401 return 0;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	402 }
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	403
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	404 static int
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	405 teedataobject_clear(teedataobject *tdo)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	406 {
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	407 int i;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	408 Py_CLEAR(tdo->it);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	409 for (i=0 ; i<tdo->numread ; i++)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	410 Py_CLEAR(tdo->values[i]);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	411 Py_CLEAR(tdo->nextlink);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	412 return 0;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	413 }
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	414
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	415 static void
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	416 teedataobject_dealloc(teedataobject *tdo)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	417 {
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	418 PyObject_GC_UnTrack(tdo);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	419 teedataobject_clear(tdo);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	420 PyObject_GC_Del(tdo);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	421 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	422
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	423 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	424
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	425 static PyTypeObject teedataobject_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	426 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	427 "itertools.tee_dataobject", /* tp_name */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	428 sizeof(teedataobject), /* tp_basicsize */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	429 0, /* tp_itemsize */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	430 /* methods */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	431 (destructor)teedataobject_dealloc, /* tp_dealloc */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	432 0, /* tp_print */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	433 0, /* tp_getattr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	434 0, /* tp_setattr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	435 0, /* tp_compare */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	436 0, /* tp_repr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	437 0, /* tp_as_number */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	438 0, /* tp_as_sequence */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	439 0, /* tp_as_mapping */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	440 0, /* tp_hash */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	441 0, /* tp_call */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	442 0, /* tp_str */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	443 PyObject_GenericGetAttr, /* tp_getattro */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	444 0, /* tp_setattro */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	445 0, /* tp_as_buffer */
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	446 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	447 teedataobject_doc, /* tp_doc */
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	448 (traverseproc)teedataobject_traverse, /* tp_traverse */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	449 (inquiry)teedataobject_clear, /* tp_clear */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	450 0, /* tp_richcompare */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	451 0, /* tp_weaklistoffset */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	452 0, /* tp_iter */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	453 0, /* tp_iternext */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	454 0, /* tp_methods */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	455 0, /* tp_members */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	456 0, /* tp_getset */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	457 0, /* tp_base */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	458 0, /* tp_dict */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	459 0, /* tp_descr_get */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	460 0, /* tp_descr_set */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	461 0, /* tp_dictoffset */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	462 0, /* tp_init */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	463 0, /* tp_alloc */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	464 0, /* tp_new */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	465 PyObject_GC_Del, /* tp_free */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	466 };
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	467
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	468
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	469 static PyTypeObject tee_type;
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	470
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	471 static PyObject *
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	472 tee_next(teeobject *to)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	473 {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	474 PyObject value, link;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	475
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	476 if (to->index >= LINKCELLS) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	477 link = teedataobject_jumplink(to->dataobj);
39144 e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	478 Py_DECREF(to->dataobj);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	479 to->dataobj = (teedataobject *)link;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	480 to->index = 0;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	481 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	482 value = teedataobject_getitem(to->dataobj, to->index);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	483 if (value == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	484 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	485 to->index++;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	486 return value;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	487 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	488
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	489 static int
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	490 tee_traverse(teeobject to, visitproc visit, void arg)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	491 {
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	492 Py_VISIT((PyObject *)to->dataobj);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	493 return 0;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	494 }
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	495
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	496 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	497 tee_copy(teeobject *to)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	498 {
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	499 teeobject *newto;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	500
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	501 newto = PyObject_GC_New(teeobject, &tee_type);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	502 if (newto == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	503 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	504 Py_INCREF(to->dataobj);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	505 newto->dataobj = to->dataobj;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	506 newto->index = to->index;
33777 e4b7c923ed5f Fix and test weak referencing of itertools.tee objects. Raymond Hettinger <python@rcn.com> parents: 33646 diff changeset	507 newto->weakreflist = NULL;
37757 e1388aba9896 Add missing PyObject_GC_Track call, causing some itertools.tee objects to Thomas Wouters <thomas@python.org> parents: 37248 diff changeset	508 PyObject_GC_Track(newto);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	509 return (PyObject *)newto;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	510 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	511
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	512 PyDoc_STRVAR(teecopy_doc, "Returns an independent iterator.");
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	513
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	514 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	515 tee_fromiterable(PyObject *iterable)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	516 {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	517 teeobject *to;
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	518 PyObject *it = NULL;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	519
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	520 it = PyObject_GetIter(iterable);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	521 if (it == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	522 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	523 if (PyObject_TypeCheck(it, &tee_type)) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	524 to = (teeobject )tee_copy((teeobject )it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	525 goto done;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	526 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	527
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	528 to = PyObject_GC_New(teeobject, &tee_type);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	529 if (to == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	530 goto done;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	531 to->dataobj = (teedataobject *)teedataobject_new(it);
39144 e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	532 if (!to->dataobj) {
e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	533 PyObject_GC_Del(to);
e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	534 to = NULL;
e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	535 goto done;
e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	536 }
e9773fc27e9b nextlink can be NULL if teedataobject_new fails, so use XINCREF. Neal Norwitz <nnorwitz@gmail.com> parents: 38420 diff changeset	537
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	538 to->index = 0;
33777 e4b7c923ed5f Fix and test weak referencing of itertools.tee objects. Raymond Hettinger <python@rcn.com> parents: 33646 diff changeset	539 to->weakreflist = NULL;
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	540 PyObject_GC_Track(to);
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	541 done:
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	542 Py_XDECREF(it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	543 return (PyObject *)to;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	544 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	545
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	546 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	547 tee_new(PyTypeObject type, PyObject args, PyObject *kw)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	548 {
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	549 PyObject *iterable;
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	550
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	551 if (!PyArg_UnpackTuple(args, "tee", 1, 1, &iterable))
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	552 return NULL;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	553 return tee_fromiterable(iterable);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	554 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	555
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	556 static int
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	557 tee_clear(teeobject *to)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	558 {
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	559 if (to->weakreflist != NULL)
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	560 PyObject_ClearWeakRefs((PyObject *) to);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	561 Py_CLEAR(to->dataobj);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	562 return 0;
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	563 }
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	564
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	565 static void
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	566 tee_dealloc(teeobject *to)
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	567 {
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	568 PyObject_GC_UnTrack(to);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	569 tee_clear(to);
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	570 PyObject_GC_Del(to);
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	571 }
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	572
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	573 PyDoc_STRVAR(teeobject_doc,
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	574 "Iterator wrapped to make it copyable");
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	575
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	576 static PyMethodDef tee_methods[] = {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	577 {"__copy__", (PyCFunction)tee_copy, METH_NOARGS, teecopy_doc},
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	578 {NULL, NULL} /* sentinel */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	579 };
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	580
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	581 static PyTypeObject tee_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	582 PyVarObject_HEAD_INIT(NULL, 0)
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	583 "itertools.tee", /* tp_name */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	584 sizeof(teeobject), /* tp_basicsize */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	585 0, /* tp_itemsize */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	586 /* methods */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	587 (destructor)tee_dealloc, /* tp_dealloc */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	588 0, /* tp_print */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	589 0, /* tp_getattr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	590 0, /* tp_setattr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	591 0, /* tp_compare */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	592 0, /* tp_repr */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	593 0, /* tp_as_number */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	594 0, /* tp_as_sequence */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	595 0, /* tp_as_mapping */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	596 0, /* tp_hash */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	597 0, /* tp_call */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	598 0, /* tp_str */
30786 225d848a5f96 Minor improvements to itertools.tee(): Raymond Hettinger <python@rcn.com> parents: 30773 diff changeset	599 0, /* tp_getattro */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	600 0, /* tp_setattro */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	601 0, /* tp_as_buffer */
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	602 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC, /* tp_flags */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	603 teeobject_doc, /* tp_doc */
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	604 (traverseproc)tee_traverse, /* tp_traverse */
00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	605 (inquiry)tee_clear, /* tp_clear */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	606 0, /* tp_richcompare */
33777 e4b7c923ed5f Fix and test weak referencing of itertools.tee objects. Raymond Hettinger <python@rcn.com> parents: 33646 diff changeset	607 offsetof(teeobject, weakreflist), /* tp_weaklistoffset */
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	608 PyObject_SelfIter, /* tp_iter */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	609 (iternextfunc)tee_next, /* tp_iternext */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	610 tee_methods, /* tp_methods */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	611 0, /* tp_members */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	612 0, /* tp_getset */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	613 0, /* tp_base */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	614 0, /* tp_dict */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	615 0, /* tp_descr_get */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	616 0, /* tp_descr_set */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	617 0, /* tp_dictoffset */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	618 0, /* tp_init */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	619 0, /* tp_alloc */
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	620 tee_new, /* tp_new */
37248 00e9d59c23b4 Make itertools.tee and its internal teedataobject participate in GC. This Thomas Wouters <thomas@python.org> parents: 36684 diff changeset	621 PyObject_GC_Del, /* tp_free */
30764 b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	622 };
b948e775a79e Added itertools.tee() Raymond Hettinger <python@rcn.com> parents: 30240 diff changeset	623
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	624 static PyObject *
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	625 tee(PyObject self, PyObject args)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	626 {
39720 1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	627 Py_ssize_t i, n=2;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	628 PyObject it, iterable, copyable, result;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	629
39720 1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	630 if (!PyArg_ParseTuple(args, "O\|n", &iterable, &n))
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	631 return NULL;
39720 1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	632 if (n < 0) {
1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	633 PyErr_SetString(PyExc_ValueError, "n must be >= 0");
1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	634 return NULL;
1352ad9afcc0 Bug #1550714: fix SystemError from itertools.tee on negative value for n. Neal Norwitz <nnorwitz@gmail.com> parents: 39144 diff changeset	635 }
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	636 result = PyTuple_New(n);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	637 if (result == NULL)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	638 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	639 if (n == 0)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	640 return result;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	641 it = PyObject_GetIter(iterable);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	642 if (it == NULL) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	643 Py_DECREF(result);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	644 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	645 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	646 if (!PyObject_HasAttrString(it, "__copy__")) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	647 copyable = tee_fromiterable(it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	648 Py_DECREF(it);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	649 if (copyable == NULL) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	650 Py_DECREF(result);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	651 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	652 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	653 } else
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	654 copyable = it;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	655 PyTuple_SET_ITEM(result, 0, copyable);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	656 for (i=1 ; i<n ; i++) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	657 copyable = PyObject_CallMethod(copyable, "__copy__", NULL);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	658 if (copyable == NULL) {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	659 Py_DECREF(result);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	660 return NULL;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	661 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	662 PyTuple_SET_ITEM(result, i, copyable);
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	663 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	664 return result;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	665 }
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	666
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	667 PyDoc_STRVAR(tee_doc,
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	668 "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	669
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	670
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	671 /* cycle object **********************************************************/
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	672
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	673 typedef struct {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	674 PyObject_HEAD
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	675 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	676 PyObject *saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	677 int firstpass;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	678 } cycleobject;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	679
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	680 static PyTypeObject cycle_type;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	681
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	682 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	683 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	684 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	685 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	686 PyObject *iterable;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	687 PyObject *saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	688 cycleobject *lz;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	689
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	690 if (type == &cycle_type && !_PyArg_NoKeywords("cycle()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	691 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	692
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	693 if (!PyArg_UnpackTuple(args, "cycle", 1, 1, &iterable))
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	694 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	695
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	696 /* Get iterator. */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	697 it = PyObject_GetIter(iterable);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	698 if (it == NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	699 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	700
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	701 saved = PyList_New(0);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	702 if (saved == NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	703 Py_DECREF(it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	704 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	705 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	706
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	707 /* create cycleobject structure */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	708 lz = (cycleobject *)type->tp_alloc(type, 0);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	709 if (lz == NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	710 Py_DECREF(it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	711 Py_DECREF(saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	712 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	713 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	714 lz->it = it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	715 lz->saved = saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	716 lz->firstpass = 0;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	717
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	718 return (PyObject *)lz;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	719 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	720
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	721 static void
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	722 cycle_dealloc(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	723 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	724 PyObject_GC_UnTrack(lz);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	725 Py_XDECREF(lz->saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	726 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	727 Py_TYPE(lz)->tp_free(lz);
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	728 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	729
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	730 static int
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	731 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	732 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	733 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	734 Py_VISIT(lz->saved);
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	735 return 0;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	736 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	737
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	738 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	739 cycle_next(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	740 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	741 PyObject *item;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	742 PyObject *it;
33646 3d8d78217599 Replace structure member before decreffing. Raymond Hettinger <python@rcn.com> parents: 33616 diff changeset	743 PyObject *tmp;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	744
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	745 while (1) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	746 item = PyIter_Next(lz->it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	747 if (item != NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	748 if (!lz->firstpass)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	749 PyList_Append(lz->saved, item);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	750 return item;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	751 }
32010 544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	752 if (PyErr_Occurred()) {
544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	753 if (PyErr_ExceptionMatches(PyExc_StopIteration))
544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	754 PyErr_Clear();
544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	755 else
544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	756 return NULL;
544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	757 }
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	758 if (PyList_Size(lz->saved) == 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	759 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	760 it = PyObject_GetIter(lz->saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	761 if (it == NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	762 return NULL;
33646 3d8d78217599 Replace structure member before decreffing. Raymond Hettinger <python@rcn.com> parents: 33616 diff changeset	763 tmp = lz->it;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	764 lz->it = it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	765 lz->firstpass = 1;
33646 3d8d78217599 Replace structure member before decreffing. Raymond Hettinger <python@rcn.com> parents: 33616 diff changeset	766 Py_DECREF(tmp);
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	767 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	768 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	769
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	770 PyDoc_STRVAR(cycle_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	771 "cycle(iterable) --> cycle object\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	772 \n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	773 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	774 Then repeat the sequence indefinitely.");
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	775
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	776 static PyTypeObject cycle_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	777 PyVarObject_HEAD_INIT(NULL, 0)
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	778 "itertools.cycle", /* tp_name */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	779 sizeof(cycleobject), /* tp_basicsize */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	780 0, /* tp_itemsize */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	781 /* methods */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	782 (destructor)cycle_dealloc, /* tp_dealloc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	783 0, /* tp_print */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	784 0, /* tp_getattr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	785 0, /* tp_setattr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	786 0, /* tp_compare */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	787 0, /* tp_repr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	788 0, /* tp_as_number */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	789 0, /* tp_as_sequence */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	790 0, /* tp_as_mapping */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	791 0, /* tp_hash */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	792 0, /* tp_call */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	793 0, /* tp_str */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	794 PyObject_GenericGetAttr, /* tp_getattro */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	795 0, /* tp_setattro */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	796 0, /* tp_as_buffer */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	797 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	798 Py_TPFLAGS_BASETYPE, /* tp_flags */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	799 cycle_doc, /* tp_doc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	800 (traverseproc)cycle_traverse, /* tp_traverse */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	801 0, /* tp_clear */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	802 0, /* tp_richcompare */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	803 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	804 PyObject_SelfIter, /* tp_iter */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	805 (iternextfunc)cycle_next, /* tp_iternext */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	806 0, /* tp_methods */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	807 0, /* tp_members */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	808 0, /* tp_getset */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	809 0, /* tp_base */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	810 0, /* tp_dict */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	811 0, /* tp_descr_get */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	812 0, /* tp_descr_set */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	813 0, /* tp_dictoffset */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	814 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	815 0, /* tp_alloc */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	816 cycle_new, /* tp_new */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	817 PyObject_GC_Del, /* tp_free */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	818 };
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	819
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	820
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	821 /* dropwhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	822
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	823 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	824 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	825 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	826 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	827 long start;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	828 } dropwhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	829
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	830 static PyTypeObject dropwhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	831
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	832 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	833 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	834 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	835 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	836 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	837 dropwhileobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	838
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	839 if (type == &dropwhile_type && !_PyArg_NoKeywords("dropwhile()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	840 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	841
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	842 if (!PyArg_UnpackTuple(args, "dropwhile", 2, 2, &func, &seq))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	843 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	844
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	845 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	846 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	847 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	848 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	849
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	850 /* create dropwhileobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	851 lz = (dropwhileobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	852 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	853 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	854 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	855 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	856 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	857 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	858 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	859 lz->start = 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	860
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	861 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	862 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	863
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	864 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	865 dropwhile_dealloc(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	866 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	867 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	868 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	869 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	870 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	871 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	872
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	873 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	874 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	875 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	876 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	877 Py_VISIT(lz->func);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	878 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	879 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	880
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	881 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	882 dropwhile_next(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	883 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	884 PyObject item, good;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	885 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	886 long ok;
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	887 PyObject (iternext)(PyObject *);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	888
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	889 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	890 iternext = *Py_TYPE(it)->tp_iternext;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	891 for (;;) {
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	892 item = iternext(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	893 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	894 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	895 if (lz->start == 1)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	896 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	897
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	898 good = PyObject_CallFunctionObjArgs(lz->func, item, NULL);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	899 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	900 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	901 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	902 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	903 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	904 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	905 if (!ok) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	906 lz->start = 1;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	907 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	908 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	909 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	910 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	911 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	912
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	913 PyDoc_STRVAR(dropwhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	914 "dropwhile(predicate, iterable) --> dropwhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	915 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	916 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	917 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	918
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	919 static PyTypeObject dropwhile_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	920 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	921 "itertools.dropwhile", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	922 sizeof(dropwhileobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	923 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	924 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	925 (destructor)dropwhile_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	926 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	927 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	928 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	929 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	930 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	931 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	932 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	933 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	934 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	935 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	936 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	937 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	938 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	939 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	940 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	941 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	942 dropwhile_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	943 (traverseproc)dropwhile_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	944 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	945 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	946 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	947 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	948 (iternextfunc)dropwhile_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	949 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	950 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	951 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	952 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	953 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	954 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	955 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	956 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	957 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	958 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	959 dropwhile_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	960 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	961 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	962
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	963
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	964 /* takewhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	965
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	966 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	967 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	968 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	969 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	970 long stop;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	971 } takewhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	972
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	973 static PyTypeObject takewhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	974
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	975 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	976 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	977 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	978 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	979 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	980 takewhileobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	981
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	982 if (type == &takewhile_type && !_PyArg_NoKeywords("takewhile()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	983 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	984
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	985 if (!PyArg_UnpackTuple(args, "takewhile", 2, 2, &func, &seq))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	986 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	987
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	988 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	989 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	990 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	991 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	992
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	993 /* create takewhileobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	994 lz = (takewhileobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	995 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	996 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	997 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	998 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	999 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1000 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1001 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1002 lz->stop = 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1003
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1004 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1005 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1006
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1007 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1008 takewhile_dealloc(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1009 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1010 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1011 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1012 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1013 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	1014 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1015
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1016 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1017 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	1018 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1019 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1020 Py_VISIT(lz->func);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1021 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1022 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1023
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1024 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1025 takewhile_next(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1026 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1027 PyObject item, good;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1028 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1029 long ok;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1030
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1031 if (lz->stop == 1)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1032 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1033
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1034 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1035 item = (*Py_TYPE(it)->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1036 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1037 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1038
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1039 good = PyObject_CallFunctionObjArgs(lz->func, item, NULL);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1040 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1041 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1042 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1043 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1044 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1045 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1046 if (ok)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1047 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1048 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1049 lz->stop = 1;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1050 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1051 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1052
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1053 PyDoc_STRVAR(takewhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1054 "takewhile(predicate, iterable) --> takewhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1055 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1056 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	1057 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	1058
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1059 static PyTypeObject takewhile_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	1060 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1061 "itertools.takewhile", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1062 sizeof(takewhileobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1063 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1064 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1065 (destructor)takewhile_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1066 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1067 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1068 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1069 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1070 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1071 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1072 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1073 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1074 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1075 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1076 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1077 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1078 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1079 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1080 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1081 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1082 takewhile_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1083 (traverseproc)takewhile_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1084 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1085 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1086 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1087 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1088 (iternextfunc)takewhile_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1089 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1090 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1091 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1092 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1093 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1094 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1095 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1096 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1097 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1098 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1099 takewhile_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1100 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1101 };
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 /* islice object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1105
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1106 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1107 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1108 PyObject *it;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1109 Py_ssize_t next;
a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1110 Py_ssize_t stop;
a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1111 Py_ssize_t step;
a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1112 Py_ssize_t cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1113 } isliceobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1114
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1115 static PyTypeObject islice_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1116
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1117 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1118 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	1119 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1120 PyObject *seq;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1121 Py_ssize_t start=0, stop=-1, step=1;
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1122 PyObject it, a1=NULL, a2=NULL, a3=NULL;
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	1123 Py_ssize_t numargs;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1124 isliceobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1125
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	1126 if (type == &islice_type && !_PyArg_NoKeywords("islice()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1127 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1128
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1129 if (!PyArg_UnpackTuple(args, "islice", 2, 4, &seq, &a1, &a2, &a3))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1130 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1131
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1132 numargs = PyTuple_Size(args);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1133 if (numargs == 2) {
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1134 if (a1 != Py_None) {
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1135 stop = PyInt_AsSsize_t(a1);
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1136 if (stop == -1) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1137 if (PyErr_Occurred())
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1138 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1139 PyErr_SetString(PyExc_ValueError,
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1140 "Stop argument for islice() must be a non-negative integer or None.");
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1141 return NULL;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1142 }
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1143 }
28930 18082b2d2ebd The previous made the stop argument optional. Raymond Hettinger <python@rcn.com> parents: 28929 diff changeset	1144 } else {
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1145 if (a1 != Py_None)
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1146 start = PyInt_AsSsize_t(a1);
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1147 if (start == -1 && PyErr_Occurred())
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1148 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1149 if (a2 != Py_None) {
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1150 stop = PyInt_AsSsize_t(a2);
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1151 if (stop == -1) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1152 if (PyErr_Occurred())
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1153 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1154 PyErr_SetString(PyExc_ValueError,
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1155 "Stop argument for islice() must be a non-negative integer or None.");
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1156 return NULL;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1157 }
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1158 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1159 }
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1160 if (start<0 \|\| stop<-1) {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1161 PyErr_SetString(PyExc_ValueError,
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1162 "Indices for islice() must be non-negative integers or None.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1163 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1164 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1165
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1166 if (a3 != NULL) {
1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1167 if (a3 != Py_None)
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1168 step = PyInt_AsSsize_t(a3);
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1169 if (step == -1 && PyErr_Occurred())
1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1170 PyErr_Clear();
1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1171 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1172 if (step<1) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1173 PyErr_SetString(PyExc_ValueError,
34108 1d1d6377acb1 Added optional None arguments to itertools.islice(). Raymond Hettinger <python@rcn.com> parents: 33779 diff changeset	1174 "Step for islice() must be a positive integer or None.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1175 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1176 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1177
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1178 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1179 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1180 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1181 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1182
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1183 /* create isliceobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1184 lz = (isliceobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1185 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1186 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1187 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1188 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1189 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1190 lz->next = start;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1191 lz->stop = stop;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1192 lz->step = step;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1193 lz->cnt = 0L;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1194
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1195 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1196 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1197
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1198 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1199 islice_dealloc(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1200 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1201 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1202 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1203 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	1204 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1205
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1206 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1207 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	1208 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1209 Py_VISIT(lz->it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1210 return 0;
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 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1214 islice_next(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1215 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1216 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1217 PyObject *it = lz->it;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	1218 Py_ssize_t oldnext;
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	1219 PyObject (iternext)(PyObject *);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1220
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	1221 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1222 iternext = *Py_TYPE(it)->tp_iternext;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1223 while (lz->cnt < lz->next) {
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	1224 item = iternext(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1225 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1226 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1227 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1228 lz->cnt++;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1229 }
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	1230 if (lz->stop != -1 && lz->cnt >= lz->stop)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1231 return NULL;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1232 assert(PyIter_Check(it));
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	1233 item = iternext(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1234 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1235 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1236 lz->cnt++;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1237 oldnext = lz->next;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1238 lz->next += lz->step;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1239 if (lz->next < oldnext) /* Check for overflow */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1240 lz->next = lz->stop;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1241 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1242 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1243
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1244 PyDoc_STRVAR(islice_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1245 "islice(iterable, [start,] stop [, step]) --> islice object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1246 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1247 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	1248 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	1249 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	1250 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	1251 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	1252 but returns an iterator.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1253
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1254 static PyTypeObject islice_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	1255 PyVarObject_HEAD_INIT(NULL, 0)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1256 "itertools.islice", /* tp_name */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1257 sizeof(isliceobject), /* tp_basicsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1258 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1259 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1260 (destructor)islice_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1261 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1262 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1263 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1264 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1265 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1266 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1267 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1268 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1269 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1270 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1271 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1272 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1273 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1274 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1275 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1276 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1277 islice_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1278 (traverseproc)islice_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1279 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1280 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1281 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1282 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1283 (iternextfunc)islice_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1284 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1285 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1286 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1287 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1288 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1289 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1290 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1291 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1292 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1293 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1294 islice_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1295 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1296 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1297
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1298
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1299 /* starmap object ************************************************************/
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 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1302 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1303 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1304 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1305 } starmapobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1306
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1307 static PyTypeObject starmap_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1308
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1309 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1310 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	1311 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1312 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1313 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1314 starmapobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1315
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	1316 if (type == &starmap_type && !_PyArg_NoKeywords("starmap()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1317 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1318
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1319 if (!PyArg_UnpackTuple(args, "starmap", 2, 2, &func, &seq))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1320 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1321
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1322 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1323 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1324 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1325 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1326
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1327 /* create starmapobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1328 lz = (starmapobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1329 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1330 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1331 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1332 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1333 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1334 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1335 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1336
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1337 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1338 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1339
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1340 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1341 starmap_dealloc(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1342 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1343 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1344 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1345 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1346 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	1347 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1348
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1349 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1350 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	1351 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1352 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1353 Py_VISIT(lz->func);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1354 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1355 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1356
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1357 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1358 starmap_next(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1359 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1360 PyObject *args;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1361 PyObject *result;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1362 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1363
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1364 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1365 args = (*Py_TYPE(it)->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1366 if (args == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1367 return NULL;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1368 if (!PyTuple_CheckExact(args)) {
44717 d0f7963cf8b0 Make starmap() match its pure python definition and accept any itertable input (not just tuples). Raymond Hettinger <python@rcn.com> parents: 44350 diff changeset	1369 PyObject *newargs = PySequence_Tuple(args);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1370 Py_DECREF(args);
44717 d0f7963cf8b0 Make starmap() match its pure python definition and accept any itertable input (not just tuples). Raymond Hettinger <python@rcn.com> parents: 44350 diff changeset	1371 if (newargs == NULL)
d0f7963cf8b0 Make starmap() match its pure python definition and accept any itertable input (not just tuples). Raymond Hettinger <python@rcn.com> parents: 44350 diff changeset	1372 return NULL;
d0f7963cf8b0 Make starmap() match its pure python definition and accept any itertable input (not just tuples). Raymond Hettinger <python@rcn.com> parents: 44350 diff changeset	1373 args = newargs;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1374 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1375 result = PyObject_Call(lz->func, args, NULL);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1376 Py_DECREF(args);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1377 return result;
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
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1380 PyDoc_STRVAR(starmap_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1381 "starmap(function, sequence) --> starmap object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1382 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1383 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	1384 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	1385
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1386 static PyTypeObject starmap_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	1387 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1388 "itertools.starmap", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1389 sizeof(starmapobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1390 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1391 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1392 (destructor)starmap_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1393 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1394 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1395 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1396 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1397 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1398 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1399 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1400 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1401 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1402 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1403 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1404 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1405 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1406 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1407 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1408 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1409 starmap_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1410 (traverseproc)starmap_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1411 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1412 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1413 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1414 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1415 (iternextfunc)starmap_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1416 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1417 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1418 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1419 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1420 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1421 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1422 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1423 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1424 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1425 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1426 starmap_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1427 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1428 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1429
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1430
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1431 /* imap object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1432
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1433 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1434 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1435 PyObject *iters;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1436 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1437 } imapobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1438
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1439 static PyTypeObject imap_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1440
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1441 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1442 imap_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	1443 {
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1444 PyObject it, iters, *func;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1445 imapobject *lz;
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	1446 Py_ssize_t numargs, i;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1447
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	1448 if (Py_Py3kWarningFlag &&
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	1449 PyErr_Warn(PyExc_DeprecationWarning,
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	1450 "In 3.x, itertools.imap() was moved to builtin map()") < 0)
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	1451 return NULL;
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	1452
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	1453 if (type == &imap_type && !_PyArg_NoKeywords("imap()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1454 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1455
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1456 numargs = PyTuple_Size(args);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1457 if (numargs < 2) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1458 PyErr_SetString(PyExc_TypeError,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1459 "imap() must have at least two arguments.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1460 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1461 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1462
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1463 iters = PyTuple_New(numargs-1);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1464 if (iters == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1465 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1466
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1467 for (i=1 ; i<numargs ; i++) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1468 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1469 it = PyObject_GetIter(PyTuple_GET_ITEM(args, i));
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1470 if (it == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1471 Py_DECREF(iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1472 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1473 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1474 PyTuple_SET_ITEM(iters, i-1, it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1475 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1476
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1477 /* create imapobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1478 lz = (imapobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1479 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1480 Py_DECREF(iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1481 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1482 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1483 lz->iters = iters;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1484 func = PyTuple_GET_ITEM(args, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1485 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1486 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1487
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1488 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1489 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1490
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1491 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1492 imap_dealloc(imapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1493 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1494 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1495 Py_XDECREF(lz->iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1496 Py_XDECREF(lz->func);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1497 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	1498 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1499
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1500 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1501 imap_traverse(imapobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1502 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1503 Py_VISIT(lz->iters);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	1504 Py_VISIT(lz->func);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1505 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1506 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1507
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1508 /*
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1509 imap() is an iterator version of __builtins__.map() except that it does
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1510 not have the None fill-in feature. That was intentionally left out for
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1511 the following reasons:
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1512
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1513 1) Itertools are designed to be easily combined and chained together.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1514 Having all tools stop with the shortest input is a unifying principle
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1515 that makes it easier to combine finite iterators (supplying data) with
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1516 infinite iterators like count() and repeat() (for supplying sequential
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1517 or constant arguments to a function).
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1518
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1519 2) In typical use cases for combining itertools, having one finite data
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1520 supplier run out before another is likely to be an error condition which
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1521 should not pass silently by automatically supplying None.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1522
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1523 3) The use cases for automatic None fill-in are rare -- not many functions
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1524 do something useful when a parameter suddenly switches type and becomes
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1525 None.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1526
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1527 4) If a need does arise, it can be met by __builtins__.map() or by
29535 c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1528 writing: chain(iterable, repeat(None)).
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1529
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1530 5) Similar toolsets in Haskell and SML do not have automatic None fill-in.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1531 */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1532
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1533 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1534 imap_next(imapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1535 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1536 PyObject *val;
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1537 PyObject *argtuple;
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1538 PyObject *result;
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	1539 Py_ssize_t numargs, i;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1540
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1541 numargs = PyTuple_Size(lz->iters);
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1542 argtuple = PyTuple_New(numargs);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1543 if (argtuple == NULL)
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1544 return NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1545
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1546 for (i=0 ; i<numargs ; i++) {
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1547 val = PyIter_Next(PyTuple_GET_ITEM(lz->iters, i));
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1548 if (val == NULL) {
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1549 Py_DECREF(argtuple);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1550 return NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1551 }
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1552 PyTuple_SET_ITEM(argtuple, i, val);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1553 }
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1554 if (lz->func == Py_None)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1555 return argtuple;
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1556 result = PyObject_Call(lz->func, argtuple, NULL);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1557 Py_DECREF(argtuple);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1558 return result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1559 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1560
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1561 PyDoc_STRVAR(imap_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1562 "imap(func, *iterables) --> imap object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1563 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1564 Make an iterator that computes the function using arguments from\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1565 each of the iterables. Like map() except that it returns\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1566 an iterator instead of a list and that it stops when the shortest\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1567 iterable is exhausted instead of filling in None for shorter\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1568 iterables.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1569
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1570 static PyTypeObject imap_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	1571 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1572 "itertools.imap", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1573 sizeof(imapobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1574 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1575 /* methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1576 (destructor)imap_dealloc, /* tp_dealloc */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1577 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1578 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1579 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1580 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1581 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1582 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1583 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1584 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1585 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1586 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1587 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1588 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1589 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1590 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1591 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1592 Py_TPFLAGS_BASETYPE, /* tp_flags */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1593 imap_doc, /* tp_doc */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1594 (traverseproc)imap_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1595 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1596 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1597 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1598 PyObject_SelfIter, /* tp_iter */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1599 (iternextfunc)imap_next, /* tp_iternext */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1600 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1601 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1602 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1603 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1604 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1605 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1606 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1607 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1608 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1609 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1610 imap_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1611 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1612 };
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
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1615 /* chain object ************************************************************/
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1616
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1617 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1618 PyObject_HEAD
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1619 PyObject source; / Iterator over input iterables */
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1620 PyObject active; / Currently running input iterator */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1621 } chainobject;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1622
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1623 static PyTypeObject chain_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1624
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1625 static PyObject *
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1626 chain_new_internal(PyTypeObject type, PyObject source)
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1627 {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1628 chainobject *lz;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1629
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1630 lz = (chainobject *)type->tp_alloc(type, 0);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1631 if (lz == NULL) {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1632 Py_DECREF(source);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1633 return NULL;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1634 }
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1635
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1636 lz->source = source;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1637 lz->active = NULL;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1638 return (PyObject *)lz;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1639 }
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1640
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1641 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1642 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	1643 {
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1644 PyObject *source;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1645
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	1646 if (type == &chain_type && !_PyArg_NoKeywords("chain()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	1647 return NULL;
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1648
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1649 source = PyObject_GetIter(args);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1650 if (source == NULL)
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1651 return NULL;
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1652
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1653 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	1654 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1655
45563 7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1656 static PyObject *
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1657 chain_new_from_iterable(PyTypeObject type, PyObject arg)
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1658 {
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1659 PyObject *source;
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1660
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1661 source = PyObject_GetIter(arg);
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1662 if (source == NULL)
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1663 return NULL;
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1664
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1665 return chain_new_internal(type, source);
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1666 }
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1667
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1668 static void
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1669 chain_dealloc(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1670 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1671 PyObject_GC_UnTrack(lz);
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1672 Py_XDECREF(lz->active);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1673 Py_XDECREF(lz->source);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	1674 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	1675 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1676
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1677 static int
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1678 chain_traverse(chainobject lz, visitproc visit, void arg)
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1679 {
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1680 Py_VISIT(lz->source);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1681 Py_VISIT(lz->active);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1682 return 0;
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1683 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1684
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1685 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1686 chain_next(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1687 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1688 PyObject *item;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1689
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1690 if (lz->source == NULL)
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1691 return NULL; /* already stopped */
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1692
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1693 if (lz->active == NULL) {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1694 PyObject *iterable = PyIter_Next(lz->source);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1695 if (iterable == NULL) {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1696 Py_CLEAR(lz->source);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1697 return NULL; /* no more input sources */
32010 544f2172363d SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 31909 diff changeset	1698 }
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1699 lz->active = PyObject_GetIter(iterable);
45658 8a5c75efae4c Fix refleak in chain(). Raymond Hettinger <python@rcn.com> parents: 45651 diff changeset	1700 Py_DECREF(iterable);
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1701 if (lz->active == NULL) {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1702 Py_CLEAR(lz->source);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1703 return NULL; /* input not iterable */
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1704 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1705 }
45562 43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1706 item = PyIter_Next(lz->active);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1707 if (item != NULL)
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1708 return item;
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1709 if (PyErr_Occurred()) {
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1710 if (PyErr_ExceptionMatches(PyExc_StopIteration))
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1711 PyErr_Clear();
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1712 else
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1713 return NULL; /* input raised an exception */
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1714 }
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1715 Py_CLEAR(lz->active);
43e084847e53 Have itertools.chain() consume its inputs lazily instead of building a tuple of iterators at the outset. Raymond Hettinger <python@rcn.com> parents: 45547 diff changeset	1716 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	1717 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1718
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1719 PyDoc_STRVAR(chain_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1720 "chain(*iterables) --> chain object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1721 \n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1722 Return a chain object whose .next() method returns elements from the\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1723 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	1724 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	1725
45563 7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1726 PyDoc_STRVAR(chain_from_iterable_doc,
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1727 "chain.from_iterable(iterable) --> chain object\n\
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1728 \n\
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1729 Alternate chain() contructor taking a single iterable argument\n\
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1730 that evaluates lazily.");
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1731
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1732 static PyMethodDef chain_methods[] = {
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1733 {"from_iterable", (PyCFunction) chain_new_from_iterable, METH_O \| METH_CLASS,
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1734 chain_from_iterable_doc},
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1735 {NULL, NULL} /* sentinel */
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1736 };
7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1737
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1738 static PyTypeObject chain_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	1739 PyVarObject_HEAD_INIT(NULL, 0)
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1740 "itertools.chain", /* tp_name */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1741 sizeof(chainobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1742 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1743 /* methods */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1744 (destructor)chain_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1745 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1746 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1747 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1748 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1749 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1750 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1751 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1752 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1753 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1754 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1755 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1756 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1757 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1758 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1759 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1760 Py_TPFLAGS_BASETYPE, /* tp_flags */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1761 chain_doc, /* tp_doc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1762 (traverseproc)chain_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1763 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1764 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1765 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1766 PyObject_SelfIter, /* tp_iter */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1767 (iternextfunc)chain_next, /* tp_iternext */
45563 7da36b49c9e5 Add alternate constructor for itertools.chain(). Raymond Hettinger <python@rcn.com> parents: 45562 diff changeset	1768 chain_methods, /* tp_methods */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1769 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1770 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1771 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1772 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1773 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1774 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1775 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1776 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1777 0, /* tp_alloc */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1778 chain_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1779 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1780 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1781
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1782
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1783 /* product object ************************************************************/
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1784
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1785 typedef struct {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1786 PyObject_HEAD
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1787 PyObject pools; / tuple of pool tuples */
45441 147b48b3bc23 Add more comments Raymond Hettinger <python@rcn.com> parents: 45438 diff changeset	1788 Py_ssize_t indices; / one index per pool */
147b48b3bc23 Add more comments Raymond Hettinger <python@rcn.com> parents: 45438 diff changeset	1789 PyObject result; / most recently returned result tuple */
147b48b3bc23 Add more comments Raymond Hettinger <python@rcn.com> parents: 45438 diff changeset	1790 int stopped; /* set to 1 when the product iterator is exhausted */
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1791 } productobject;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1792
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1793 static PyTypeObject product_type;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1794
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1795 static PyObject *
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1796 product_new(PyTypeObject type, PyObject args, PyObject *kwds)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1797 {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1798 productobject *lz;
45567 10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1799 Py_ssize_t nargs, npools, repeat=1;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1800 PyObject *pools = NULL;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1801 Py_ssize_t *indices = NULL;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1802 Py_ssize_t i;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1803
45567 10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1804 if (kwds != NULL) {
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1805 char *kwlist[] = {"repeat", 0};
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1806 PyObject *tmpargs = PyTuple_New(0);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1807 if (tmpargs == NULL)
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1808 return NULL;
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1809 if (!PyArg_ParseTupleAndKeywords(tmpargs, kwds, "\|n:product", kwlist, &repeat)) {
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1810 Py_DECREF(tmpargs);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1811 return NULL;
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1812 }
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1813 Py_DECREF(tmpargs);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1814 if (repeat < 0) {
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1815 PyErr_SetString(PyExc_ValueError,
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1816 "repeat argument cannot be negative");
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1817 return NULL;
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1818 }
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1819 }
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1820
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1821 assert(PyTuple_Check(args));
45567 10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1822 nargs = (repeat == 0) ? 0 : PyTuple_GET_SIZE(args);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1823 npools = nargs * repeat;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1824
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1825 indices = PyMem_Malloc(npools * sizeof(Py_ssize_t));
45606 f72b2de997b9 Simplify code for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45567 diff changeset	1826 if (indices == NULL) {
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1827 PyErr_NoMemory();
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1828 goto error;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1829 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1830
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1831 pools = PyTuple_New(npools);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1832 if (pools == NULL)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1833 goto error;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1834
45567 10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1835 for (i=0; i < nargs ; ++i) {
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1836 PyObject *item = PyTuple_GET_ITEM(args, i);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1837 PyObject *pool = PySequence_Tuple(item);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1838 if (pool == NULL)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1839 goto error;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1840 PyTuple_SET_ITEM(pools, i, pool);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1841 indices[i] = 0;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1842 }
45567 10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1843 for ( ; i < npools; ++i) {
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1844 PyObject *pool = PyTuple_GET_ITEM(pools, i - nargs);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1845 Py_INCREF(pool);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1846 PyTuple_SET_ITEM(pools, i, pool);
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1847 indices[i] = 0;
10962b3f77a9 Handle the repeat keyword argument for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45563 diff changeset	1848 }
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1849
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1850 /* create productobject structure */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1851 lz = (productobject *)type->tp_alloc(type, 0);
45540 e425dda4f630 One too many decrefs. Raymond Hettinger <python@rcn.com> parents: 45539 diff changeset	1852 if (lz == NULL)
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1853 goto error;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1854
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1855 lz->pools = pools;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1856 lz->indices = indices;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1857 lz->result = NULL;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1858 lz->stopped = 0;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1859
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1860 return (PyObject *)lz;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1861
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1862 error:
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1863 if (indices != NULL)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1864 PyMem_Free(indices);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1865 Py_XDECREF(pools);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1866 return NULL;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1867 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1868
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1869 static void
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1870 product_dealloc(productobject *lz)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1871 {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1872 PyObject_GC_UnTrack(lz);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1873 Py_XDECREF(lz->pools);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1874 Py_XDECREF(lz->result);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1875 PyMem_Free(lz->indices);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1876 Py_TYPE(lz)->tp_free(lz);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1877 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1878
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1879 static int
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1880 product_traverse(productobject lz, visitproc visit, void arg)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1881 {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1882 Py_VISIT(lz->pools);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1883 Py_VISIT(lz->result);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1884 return 0;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1885 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1886
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1887 static PyObject *
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1888 product_next(productobject *lz)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1889 {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1890 PyObject *pool;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1891 PyObject *elem;
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1892 PyObject *oldelem;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1893 PyObject *pools = lz->pools;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1894 PyObject *result = lz->result;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1895 Py_ssize_t npools = PyTuple_GET_SIZE(pools);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1896 Py_ssize_t i;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1897
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1898 if (lz->stopped)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1899 return NULL;
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1900
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1901 if (result == NULL) {
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1902 /* On the first pass, return an initial tuple filled with the
45651 e44c745bec0b Beef-up docs and tests for itertools. Fix-up end-case for product(). Raymond Hettinger <python@rcn.com> parents: 45607 diff changeset	1903 first element from each pool. */
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1904 result = PyTuple_New(npools);
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1905 if (result == NULL)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1906 goto empty;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1907 lz->result = result;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1908 for (i=0; i < npools; i++) {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1909 pool = PyTuple_GET_ITEM(pools, i);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1910 if (PyTuple_GET_SIZE(pool) == 0)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1911 goto empty;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1912 elem = PyTuple_GET_ITEM(pool, 0);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1913 Py_INCREF(elem);
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1914 PyTuple_SET_ITEM(result, i, elem);
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1915 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1916 } else {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1917 Py_ssize_t *indices = lz->indices;
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1918
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1919 /* Copy the previous result tuple or re-use it if available */
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1920 if (Py_REFCNT(result) > 1) {
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1921 PyObject *old_result = result;
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1922 result = PyTuple_New(npools);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1923 if (result == NULL)
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1924 goto empty;
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1925 lz->result = result;
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1926 for (i=0; i < npools; i++) {
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1927 elem = PyTuple_GET_ITEM(old_result, i);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1928 Py_INCREF(elem);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1929 PyTuple_SET_ITEM(result, i, elem);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1930 }
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1931 Py_DECREF(old_result);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1932 }
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1933 /* Now, we've got the only copy so we can update it in-place */
45607 96ca8c7cedc4 Handle 0-tuples which can be singletons. Raymond Hettinger <python@rcn.com> parents: 45606 diff changeset	1934 assert (npools==0 \|\| Py_REFCNT(result) == 1);
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1935
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1936 /* Update the pool indices right-to-left. Only advance to the
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1937 next pool when the previous one rolls-over */
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1938 for (i=npools-1 ; i >= 0 ; i--) {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1939 pool = PyTuple_GET_ITEM(pools, i);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1940 indices[i]++;
45606 f72b2de997b9 Simplify code for itertools.product(). Raymond Hettinger <python@rcn.com> parents: 45567 diff changeset	1941 if (indices[i] == PyTuple_GET_SIZE(pool)) {
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1942 /* Roll-over and advance to next pool */
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1943 indices[i] = 0;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1944 elem = PyTuple_GET_ITEM(pool, 0);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1945 Py_INCREF(elem);
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1946 oldelem = PyTuple_GET_ITEM(result, i);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1947 PyTuple_SET_ITEM(result, i, elem);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1948 Py_DECREF(oldelem);
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1949 } else {
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1950 /* No rollover. Just increment and stop here. */
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1951 elem = PyTuple_GET_ITEM(pool, indices[i]);
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1952 Py_INCREF(elem);
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1953 oldelem = PyTuple_GET_ITEM(result, i);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1954 PyTuple_SET_ITEM(result, i, elem);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1955 Py_DECREF(oldelem);
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1956 break;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1957 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1958 }
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1959
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1960 /* If i is negative, then the indices have all rolled-over
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1961 and we're done. */
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1962 if (i < 0)
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1963 goto empty;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1964 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1965
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1966 Py_INCREF(result);
11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1967 return result;
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1968
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1969 empty:
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1970 lz->stopped = 1;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1971 return NULL;
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1972 }
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1973
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1974 PyDoc_STRVAR(product_doc,
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1975 "product(*iterables) --> product object\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1976 \n\
45438 11a73fbbf86f Improve the implementation of itertools.product() Raymond Hettinger <python@rcn.com> parents: 45425 diff changeset	1977 Cartesian product of input iterables. Equivalent to nested for-loops.\n\n\
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1978 For example, product(A, B) returns the same as: ((x,y) for x in A for y in B).\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1979 The leftmost iterators are in the outermost for-loop, so the output tuples\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1980 cycle in a manner similar to an odometer (with the rightmost element changing\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1981 on every iteration).\n\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1982 product('ab', range(3)) --> ('a',0) ('a',1) ('a',2) ('b',0) ('b',1) ('b',2)\n\
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1983 product((0,1), (0,1), (0,1)) --> (0,0,0) (0,0,1) (0,1,0) (0,1,1) (1,0,0) ...");
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1984
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1985 static PyTypeObject product_type = {
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1986 PyVarObject_HEAD_INIT(NULL, 0)
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1987 "itertools.product", /* tp_name */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1988 sizeof(productobject), /* tp_basicsize */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1989 0, /* tp_itemsize */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1990 /* methods */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1991 (destructor)product_dealloc, /* tp_dealloc */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1992 0, /* tp_print */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1993 0, /* tp_getattr */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1994 0, /* tp_setattr */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1995 0, /* tp_compare */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1996 0, /* tp_repr */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1997 0, /* tp_as_number */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1998 0, /* tp_as_sequence */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	1999 0, /* tp_as_mapping */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2000 0, /* tp_hash */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2001 0, /* tp_call */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2002 0, /* tp_str */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2003 PyObject_GenericGetAttr, /* tp_getattro */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2004 0, /* tp_setattro */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2005 0, /* tp_as_buffer */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2006 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2007 Py_TPFLAGS_BASETYPE, /* tp_flags */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2008 product_doc, /* tp_doc */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2009 (traverseproc)product_traverse, /* tp_traverse */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2010 0, /* tp_clear */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2011 0, /* tp_richcompare */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2012 0, /* tp_weaklistoffset */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2013 PyObject_SelfIter, /* tp_iter */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2014 (iternextfunc)product_next, /* tp_iternext */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2015 0, /* tp_methods */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2016 0, /* tp_members */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2017 0, /* tp_getset */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2018 0, /* tp_base */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2019 0, /* tp_dict */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2020 0, /* tp_descr_get */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2021 0, /* tp_descr_set */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2022 0, /* tp_dictoffset */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2023 0, /* tp_init */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2024 0, /* tp_alloc */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2025 product_new, /* tp_new */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2026 PyObject_GC_Del, /* tp_free */
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2027 };
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2028
3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	2029
45539 ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2030 /* combinations object ************************************************************/
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2031
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2032 typedef struct {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2033 PyObject_HEAD
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2034 PyObject pool; / input converted to a tuple */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2035 Py_ssize_t indices; / one index per result element */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2036 PyObject result; / most recently returned result tuple */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2037 Py_ssize_t r; /* size of result tuple */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2038 int stopped; /* set to 1 when the combinations iterator is exhausted */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2039 } combinationsobject;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2040
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2041 static PyTypeObject combinations_type;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2042
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2043 static PyObject *
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2044 combinations_new(PyTypeObject type, PyObject args, PyObject *kwds)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2045 {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2046 combinationsobject *co;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2047 Py_ssize_t n;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2048 Py_ssize_t r;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2049 PyObject *pool = NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2050 PyObject *iterable = NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2051 Py_ssize_t *indices = NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2052 Py_ssize_t i;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2053 static char *kwargs[] = {"iterable", "r", NULL};
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2054
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2055 if (!PyArg_ParseTupleAndKeywords(args, kwds, "On:combinations", kwargs,
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2056 &iterable, &r))
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2057 return NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2058
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2059 pool = PySequence_Tuple(iterable);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2060 if (pool == NULL)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2061 goto error;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2062 n = PyTuple_GET_SIZE(pool);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2063 if (r < 0) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2064 PyErr_SetString(PyExc_ValueError, "r must be non-negative");
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2065 goto error;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2066 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2067 if (r > n) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2068 PyErr_SetString(PyExc_ValueError, "r cannot be bigger than the iterable");
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2069 goto error;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2070 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2071
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2072 indices = PyMem_Malloc(r * sizeof(Py_ssize_t));
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2073 if (indices == NULL) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2074 PyErr_NoMemory();
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2075 goto error;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2076 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2077
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2078 for (i=0 ; i<r ; i++)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2079 indices[i] = i;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2080
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2081 /* create combinationsobject structure */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2082 co = (combinationsobject *)type->tp_alloc(type, 0);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2083 if (co == NULL)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2084 goto error;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2085
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2086 co->pool = pool;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2087 co->indices = indices;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2088 co->result = NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2089 co->r = r;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2090 co->stopped = 0;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2091
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2092 return (PyObject *)co;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2093
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2094 error:
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2095 if (indices != NULL)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2096 PyMem_Free(indices);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2097 Py_XDECREF(pool);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2098 return NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2099 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2100
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2101 static void
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2102 combinations_dealloc(combinationsobject *co)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2103 {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2104 PyObject_GC_UnTrack(co);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2105 Py_XDECREF(co->pool);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2106 Py_XDECREF(co->result);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2107 PyMem_Free(co->indices);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2108 Py_TYPE(co)->tp_free(co);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2109 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2110
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2111 static int
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2112 combinations_traverse(combinationsobject co, visitproc visit, void arg)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2113 {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2114 Py_VISIT(co->pool);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2115 Py_VISIT(co->result);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2116 return 0;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2117 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2118
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2119 static PyObject *
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2120 combinations_next(combinationsobject *co)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2121 {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2122 PyObject *elem;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2123 PyObject *oldelem;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2124 PyObject *pool = co->pool;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2125 Py_ssize_t *indices = co->indices;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2126 PyObject *result = co->result;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2127 Py_ssize_t n = PyTuple_GET_SIZE(pool);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2128 Py_ssize_t r = co->r;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2129 Py_ssize_t i, j, index;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2130
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2131 if (co->stopped)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2132 return NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2133
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2134 if (result == NULL) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2135 /* On the first pass, initialize result tuple using the indices */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2136 result = PyTuple_New(r);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2137 if (result == NULL)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2138 goto empty;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2139 co->result = result;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2140 for (i=0; i<r ; i++) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2141 index = indices[i];
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2142 elem = PyTuple_GET_ITEM(pool, index);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2143 Py_INCREF(elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2144 PyTuple_SET_ITEM(result, i, elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2145 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2146 } else {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2147 /* Copy the previous result tuple or re-use it if available */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2148 if (Py_REFCNT(result) > 1) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2149 PyObject *old_result = result;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2150 result = PyTuple_New(r);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2151 if (result == NULL)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2152 goto empty;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2153 co->result = result;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2154 for (i=0; i<r ; i++) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2155 elem = PyTuple_GET_ITEM(old_result, i);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2156 Py_INCREF(elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2157 PyTuple_SET_ITEM(result, i, elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2158 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2159 Py_DECREF(old_result);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2160 }
45547 96d5a157316e The empty tuple is usually a singleton with a much higher refcnt than 1 Christian Heimes <christian@cheimes.de> parents: 45540 diff changeset	2161 /* Now, we've got the only copy so we can update it in-place
96d5a157316e The empty tuple is usually a singleton with a much higher refcnt than 1 Christian Heimes <christian@cheimes.de> parents: 45540 diff changeset	2162 * CPython's empty tuple is a singleton and cached in
96d5a157316e The empty tuple is usually a singleton with a much higher refcnt than 1 Christian Heimes <christian@cheimes.de> parents: 45540 diff changeset	2163 * PyTuple's freelist.
96d5a157316e The empty tuple is usually a singleton with a much higher refcnt than 1 Christian Heimes <christian@cheimes.de> parents: 45540 diff changeset	2164 */
96d5a157316e The empty tuple is usually a singleton with a much higher refcnt than 1 Christian Heimes <christian@cheimes.de> parents: 45540 diff changeset	2165 assert(r == 0 \|\| Py_REFCNT(result) == 1);
45539 ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2166
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2167 /* Scan indices right-to-left until finding one that is not
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2168 at its maximum (i + n - r). */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2169 for (i=r-1 ; i >= 0 && indices[i] == i+n-r ; i--)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2170 ;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2171
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2172 /* If i is negative, then the indices are all at
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2173 their maximum value and we're done. */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2174 if (i < 0)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2175 goto empty;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2176
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2177 /* Increment the current index which we know is not at its
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2178 maximum. Then move back to the right setting each index
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2179 to its lowest possible value (one higher than the index
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2180 to its left -- this maintains the sort order invariant). */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2181 indices[i]++;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2182 for (j=i+1 ; j<r ; j++)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2183 indices[j] = indices[j-1] + 1;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2184
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2185 /* Update the result tuple for the new indices
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2186 starting with i, the leftmost index that changed */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2187 for ( ; i<r ; i++) {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2188 index = indices[i];
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2189 elem = PyTuple_GET_ITEM(pool, index);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2190 Py_INCREF(elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2191 oldelem = PyTuple_GET_ITEM(result, i);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2192 PyTuple_SET_ITEM(result, i, elem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2193 Py_DECREF(oldelem);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2194 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2195 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2196
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2197 Py_INCREF(result);
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2198 return result;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2199
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2200 empty:
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2201 co->stopped = 1;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2202 return NULL;
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2203 }
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2204
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2205 PyDoc_STRVAR(combinations_doc,
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2206 "combinations(iterables) --> combinations object\n\
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2207 \n\
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2208 Return successive r-length combinations of elements in the iterable.\n\n\
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2209 combinations(range(4), 3) --> (0,1,2), (0,1,3), (0,2,3), (1,2,3)");
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2210
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2211 static PyTypeObject combinations_type = {
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2212 PyVarObject_HEAD_INIT(NULL, 0)
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2213 "itertools.combinations", /* tp_name */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2214 sizeof(combinationsobject), /* tp_basicsize */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2215 0, /* tp_itemsize */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2216 /* methods */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2217 (destructor)combinations_dealloc, /* tp_dealloc */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2218 0, /* tp_print */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2219 0, /* tp_getattr */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2220 0, /* tp_setattr */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2221 0, /* tp_compare */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2222 0, /* tp_repr */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2223 0, /* tp_as_number */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2224 0, /* tp_as_sequence */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2225 0, /* tp_as_mapping */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2226 0, /* tp_hash */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2227 0, /* tp_call */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2228 0, /* tp_str */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2229 PyObject_GenericGetAttr, /* tp_getattro */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2230 0, /* tp_setattro */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2231 0, /* tp_as_buffer */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2232 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2233 Py_TPFLAGS_BASETYPE, /* tp_flags */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2234 combinations_doc, /* tp_doc */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2235 (traverseproc)combinations_traverse, /* tp_traverse */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2236 0, /* tp_clear */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2237 0, /* tp_richcompare */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2238 0, /* tp_weaklistoffset */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2239 PyObject_SelfIter, /* tp_iter */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2240 (iternextfunc)combinations_next, /* tp_iternext */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2241 0, /* tp_methods */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2242 0, /* tp_members */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2243 0, /* tp_getset */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2244 0, /* tp_base */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2245 0, /* tp_dict */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2246 0, /* tp_descr_get */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2247 0, /* tp_descr_set */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2248 0, /* tp_dictoffset */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2249 0, /* tp_init */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2250 0, /* tp_alloc */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2251 combinations_new, /* tp_new */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2252 PyObject_GC_Del, /* tp_free */
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2253 };
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2254
ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	2255
45675 c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2256 /* permutations object ************************************************************
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2257
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2258 def permutations(iterable, r=None):
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2259 'permutations(range(3), 2) --> (0,1) (0,2) (1,0) (1,2) (2,0) (2,1)'
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2260 pool = tuple(iterable)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2261 n = len(pool)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2262 r = n if r is None else r
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2263 indices = range(n)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2264 cycles = range(n-r+1, n+1)[::-1]
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2265 yield tuple(pool[i] for i in indices[:r])
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2266 while n:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2267 for i in reversed(range(r)):
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2268 cycles[i] -= 1
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2269 if cycles[i] == 0:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2270 indices[i:] = indices[i+1:] + indices[i:i+1]
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2271 cycles[i] = n - i
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2272 else:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2273 j = cycles[i]
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2274 indices[i], indices[-j] = indices[-j], indices[i]
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2275 yield tuple(pool[i] for i in indices[:r])
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2276 break
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2277 else:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2278 return
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2279 */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2280
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2281 typedef struct {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2282 PyObject_HEAD
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2283 PyObject pool; / input converted to a tuple */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2284 Py_ssize_t indices; / one index per element in the pool */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2285 Py_ssize_t cycles; / one rollover counter per element in the result */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2286 PyObject result; / most recently returned result tuple */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2287 Py_ssize_t r; /* size of result tuple */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2288 int stopped; /* set to 1 when the permutations iterator is exhausted */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2289 } permutationsobject;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2290
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2291 static PyTypeObject permutations_type;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2292
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2293 static PyObject *
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2294 permutations_new(PyTypeObject type, PyObject args, PyObject *kwds)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2295 {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2296 permutationsobject *po;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2297 Py_ssize_t n;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2298 Py_ssize_t r;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2299 PyObject *robj = Py_None;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2300 PyObject *pool = NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2301 PyObject *iterable = NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2302 Py_ssize_t *indices = NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2303 Py_ssize_t *cycles = NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2304 Py_ssize_t i;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2305 static char *kwargs[] = {"iterable", "r", NULL};
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2306
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2307 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O\|O:permutations", kwargs,
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2308 &iterable, &robj))
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2309 return NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2310
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2311 pool = PySequence_Tuple(iterable);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2312 if (pool == NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2313 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2314 n = PyTuple_GET_SIZE(pool);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2315
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2316 r = n;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2317 if (robj != Py_None) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2318 r = PyInt_AsSsize_t(robj);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2319 if (r == -1 && PyErr_Occurred())
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2320 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2321 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2322 if (r < 0) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2323 PyErr_SetString(PyExc_ValueError, "r must be non-negative");
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2324 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2325 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2326 if (r > n) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2327 PyErr_SetString(PyExc_ValueError, "r cannot be bigger than the iterable");
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2328 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2329 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2330
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2331 indices = PyMem_Malloc(n * sizeof(Py_ssize_t));
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2332 cycles = PyMem_Malloc(r * sizeof(Py_ssize_t));
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2333 if (indices == NULL \|\| cycles == NULL) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2334 PyErr_NoMemory();
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2335 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2336 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2337
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2338 for (i=0 ; i<n ; i++)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2339 indices[i] = i;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2340 for (i=0 ; i<r ; i++)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2341 cycles[i] = n - i;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2342
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2343 /* create permutationsobject structure */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2344 po = (permutationsobject *)type->tp_alloc(type, 0);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2345 if (po == NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2346 goto error;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2347
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2348 po->pool = pool;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2349 po->indices = indices;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2350 po->cycles = cycles;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2351 po->result = NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2352 po->r = r;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2353 po->stopped = 0;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2354
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2355 return (PyObject *)po;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2356
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2357 error:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2358 if (indices != NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2359 PyMem_Free(indices);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2360 if (cycles != NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2361 PyMem_Free(cycles);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2362 Py_XDECREF(pool);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2363 return NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2364 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2365
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2366 static void
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2367 permutations_dealloc(permutationsobject *po)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2368 {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2369 PyObject_GC_UnTrack(po);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2370 Py_XDECREF(po->pool);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2371 Py_XDECREF(po->result);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2372 PyMem_Free(po->indices);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2373 PyMem_Free(po->cycles);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2374 Py_TYPE(po)->tp_free(po);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2375 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2376
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2377 static int
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2378 permutations_traverse(permutationsobject po, visitproc visit, void arg)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2379 {
45676 eefc604ad935 Small code cleanup. Raymond Hettinger <python@rcn.com> parents: 45675 diff changeset	2380 Py_VISIT(po->pool);
eefc604ad935 Small code cleanup. Raymond Hettinger <python@rcn.com> parents: 45675 diff changeset	2381 Py_VISIT(po->result);
45675 c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2382 return 0;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2383 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2384
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2385 static PyObject *
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2386 permutations_next(permutationsobject *po)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2387 {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2388 PyObject *elem;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2389 PyObject *oldelem;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2390 PyObject *pool = po->pool;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2391 Py_ssize_t *indices = po->indices;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2392 Py_ssize_t *cycles = po->cycles;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2393 PyObject *result = po->result;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2394 Py_ssize_t n = PyTuple_GET_SIZE(pool);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2395 Py_ssize_t r = po->r;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2396 Py_ssize_t i, j, k, index;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2397
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2398 if (po->stopped)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2399 return NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2400
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2401 if (result == NULL) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2402 /* On the first pass, initialize result tuple using the indices */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2403 result = PyTuple_New(r);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2404 if (result == NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2405 goto empty;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2406 po->result = result;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2407 for (i=0; i<r ; i++) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2408 index = indices[i];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2409 elem = PyTuple_GET_ITEM(pool, index);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2410 Py_INCREF(elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2411 PyTuple_SET_ITEM(result, i, elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2412 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2413 } else {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2414 if (n == 0)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2415 goto empty;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2416
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2417 /* Copy the previous result tuple or re-use it if available */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2418 if (Py_REFCNT(result) > 1) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2419 PyObject *old_result = result;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2420 result = PyTuple_New(r);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2421 if (result == NULL)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2422 goto empty;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2423 po->result = result;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2424 for (i=0; i<r ; i++) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2425 elem = PyTuple_GET_ITEM(old_result, i);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2426 Py_INCREF(elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2427 PyTuple_SET_ITEM(result, i, elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2428 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2429 Py_DECREF(old_result);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2430 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2431 /* Now, we've got the only copy so we can update it in-place */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2432 assert(r == 0 \|\| Py_REFCNT(result) == 1);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2433
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2434 /* Decrement rightmost cycle, moving leftward upon zero rollover */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2435 for (i=r-1 ; i>=0 ; i--) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2436 cycles[i] -= 1;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2437 if (cycles[i] == 0) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2438 /* rotatation: indices[i:] = indices[i+1:] + indices[i:i+1] */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2439 index = indices[i];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2440 for (j=i ; j<n-1 ; j++)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2441 indices[j] = indices[j+1];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2442 indices[n-1] = index;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2443 cycles[i] = n - i;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2444 } else {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2445 j = cycles[i];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2446 index = indices[i];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2447 indices[i] = indices[n-j];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2448 indices[n-j] = index;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2449
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2450 for (k=i; k<r ; k++) {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2451 /* start with i, the leftmost element that changed */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2452 /* yield tuple(pool[k] for k in indices[:r]) */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2453 index = indices[k];
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2454 elem = PyTuple_GET_ITEM(pool, index);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2455 Py_INCREF(elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2456 oldelem = PyTuple_GET_ITEM(result, k);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2457 PyTuple_SET_ITEM(result, k, elem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2458 Py_DECREF(oldelem);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2459 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2460 break;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2461 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2462 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2463 /* If i is negative, then the cycles have all
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2464 rolled-over and we're done. */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2465 if (i < 0)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2466 goto empty;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2467 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2468 Py_INCREF(result);
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2469 return result;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2470
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2471 empty:
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2472 po->stopped = 1;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2473 return NULL;
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2474 }
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2475
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2476 PyDoc_STRVAR(permutations_doc,
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2477 "permutations(iterables[, r]) --> permutations object\n\
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2478 \n\
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2479 Return successive r-length permutations of elements in the iterable.\n\n\
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2480 permutations(range(4), 3) --> (0,1,2), (0,1,3), (0,2,3), (1,2,3)");
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2481
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2482 static PyTypeObject permutations_type = {
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2483 PyVarObject_HEAD_INIT(NULL, 0)
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2484 "itertools.permutations", /* tp_name */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2485 sizeof(permutationsobject), /* tp_basicsize */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2486 0, /* tp_itemsize */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2487 /* methods */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2488 (destructor)permutations_dealloc, /* tp_dealloc */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2489 0, /* tp_print */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2490 0, /* tp_getattr */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2491 0, /* tp_setattr */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2492 0, /* tp_compare */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2493 0, /* tp_repr */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2494 0, /* tp_as_number */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2495 0, /* tp_as_sequence */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2496 0, /* tp_as_mapping */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2497 0, /* tp_hash */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2498 0, /* tp_call */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2499 0, /* tp_str */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2500 PyObject_GenericGetAttr, /* tp_getattro */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2501 0, /* tp_setattro */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2502 0, /* tp_as_buffer */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2503 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2504 Py_TPFLAGS_BASETYPE, /* tp_flags */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2505 permutations_doc, /* tp_doc */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2506 (traverseproc)permutations_traverse, /* tp_traverse */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2507 0, /* tp_clear */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2508 0, /* tp_richcompare */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2509 0, /* tp_weaklistoffset */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2510 PyObject_SelfIter, /* tp_iter */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2511 (iternextfunc)permutations_next, /* tp_iternext */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2512 0, /* tp_methods */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2513 0, /* tp_members */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2514 0, /* tp_getset */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2515 0, /* tp_base */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2516 0, /* tp_dict */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2517 0, /* tp_descr_get */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2518 0, /* tp_descr_set */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2519 0, /* tp_dictoffset */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2520 0, /* tp_init */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2521 0, /* tp_alloc */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2522 permutations_new, /* tp_new */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2523 PyObject_GC_Del, /* tp_free */
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2524 };
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2525
c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	2526
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2527 /* ifilter object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2528
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2529 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2530 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2531 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2532 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2533 } ifilterobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2534
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2535 static PyTypeObject ifilter_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2536
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2537 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2538 ifilter_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	2539 {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2540 PyObject func, seq;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2541 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2542 ifilterobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2543
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2544 if (Py_Py3kWarningFlag &&
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2545 PyErr_Warn(PyExc_DeprecationWarning,
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2546 "In 3.x, itertools.ifilter() was moved to builtin filter().") < 0)
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2547 return NULL;
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2548
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	2549 if (type == &ifilter_type && !_PyArg_NoKeywords("ifilter()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2550 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2551
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2552 if (!PyArg_UnpackTuple(args, "ifilter", 2, 2, &func, &seq))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2553 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2554
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2555 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2556 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2557 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2558 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2559
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2560 /* create ifilterobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2561 lz = (ifilterobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2562 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2563 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2564 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2565 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2566 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2567 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2568 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2569
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2570 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2571 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2572
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2573 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2574 ifilter_dealloc(ifilterobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2575 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2576 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2577 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2578 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	2579 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	2580 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2581
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2582 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2583 ifilter_traverse(ifilterobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2584 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	2585 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	2586 Py_VISIT(lz->func);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2587 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2588 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2589
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2590 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2591 ifilter_next(ifilterobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2592 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2593 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	2594 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2595 long ok;
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2596 PyObject (iternext)(PyObject *);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2597
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2598 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	2599 iternext = *Py_TYPE(it)->tp_iternext;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2600 for (;;) {
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2601 item = iternext(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2602 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2603 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2604
45526 9b381ed1db14 Coerced PyBool_Type to be able to compare it. Facundo Batista <facundobatista@gmail.com> parents: 45524 diff changeset	2605 if (lz->func == Py_None \|\| lz->func == (PyObject *)&PyBool_Type) {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2606 ok = PyObject_IsTrue(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2607 } else {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2608 PyObject *good;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2609 good = PyObject_CallFunctionObjArgs(lz->func,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2610 item, NULL);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2611 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2612 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2613 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2614 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2615 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2616 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2617 }
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2618 if (ok)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2619 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2620 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2621 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2622 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2623
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2624 PyDoc_STRVAR(ifilter_doc,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2625 "ifilter(function or None, sequence) --> ifilter object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2626 \n\
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2627 Return those items of sequence for which function(item) is true.\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2628 If function is None, return the items that are true.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2629
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2630 static PyTypeObject ifilter_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	2631 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2632 "itertools.ifilter", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2633 sizeof(ifilterobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2634 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2635 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2636 (destructor)ifilter_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2637 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2638 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2639 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2640 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2641 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2642 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2643 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2644 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2645 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2646 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2647 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2648 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2649 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2650 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2651 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2652 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2653 ifilter_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2654 (traverseproc)ifilter_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2655 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2656 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2657 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	2658 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2659 (iternextfunc)ifilter_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2660 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2661 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2662 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2663 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2664 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2665 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2666 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2667 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2668 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	2669 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2670 ifilter_new, /* tp_new */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2671 PyObject_GC_Del, /* tp_free */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2672 };
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2673
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2674
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2675 /* ifilterfalse object ************************************************************/
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2676
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2677 typedef struct {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2678 PyObject_HEAD
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2679 PyObject *func;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2680 PyObject *it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2681 } ifilterfalseobject;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2682
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2683 static PyTypeObject ifilterfalse_type;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2684
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2685 static PyObject *
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2686 ifilterfalse_new(PyTypeObject type, PyObject args, PyObject *kwds)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2687 {
27770 2c389a0448a4 Remove unused variable. Guido van Rossum <guido@python.org> parents: 27769 diff changeset	2688 PyObject func, seq;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2689 PyObject *it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2690 ifilterfalseobject *lz;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2691
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2692 if (Py_Py3kWarningFlag &&
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2693 PyErr_Warn(PyExc_DeprecationWarning,
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2694 "In 3.x, ifilterfalse() was renamed to filterfalse().") < 0)
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2695 return NULL;
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	2696
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	2697 if (type == &ifilterfalse_type &&
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	2698 !_PyArg_NoKeywords("ifilterfalse()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2699 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2700
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2701 if (!PyArg_UnpackTuple(args, "ifilterfalse", 2, 2, &func, &seq))
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2702 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2703
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2704 /* Get iterator. */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2705 it = PyObject_GetIter(seq);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2706 if (it == NULL)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2707 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2708
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2709 /* create ifilterfalseobject structure */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2710 lz = (ifilterfalseobject *)type->tp_alloc(type, 0);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2711 if (lz == NULL) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2712 Py_DECREF(it);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2713 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2714 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2715 Py_INCREF(func);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2716 lz->func = func;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2717 lz->it = it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2718
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2719 return (PyObject *)lz;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2720 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2721
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2722 static void
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2723 ifilterfalse_dealloc(ifilterfalseobject *lz)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2724 {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2725 PyObject_GC_UnTrack(lz);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2726 Py_XDECREF(lz->func);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2727 Py_XDECREF(lz->it);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	2728 Py_TYPE(lz)->tp_free(lz);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2729 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2730
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2731 static int
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2732 ifilterfalse_traverse(ifilterfalseobject lz, visitproc visit, void arg)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2733 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	2734 Py_VISIT(lz->it);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	2735 Py_VISIT(lz->func);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2736 return 0;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2737 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2738
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2739 static PyObject *
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2740 ifilterfalse_next(ifilterfalseobject *lz)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2741 {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2742 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	2743 PyObject *it = lz->it;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2744 long ok;
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2745 PyObject (iternext)(PyObject *);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2746
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2747 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	2748 iternext = *Py_TYPE(it)->tp_iternext;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2749 for (;;) {
31675 f7374e3b279a Speedup the inner loops for dropwhile(), islice(), ifilter(), and Raymond Hettinger <python@rcn.com> parents: 31508 diff changeset	2750 item = iternext(it);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2751 if (item == NULL)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2752 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2753
45526 9b381ed1db14 Coerced PyBool_Type to be able to compare it. Facundo Batista <facundobatista@gmail.com> parents: 45524 diff changeset	2754 if (lz->func == Py_None \|\| lz->func == (PyObject *)&PyBool_Type) {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2755 ok = PyObject_IsTrue(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2756 } else {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2757 PyObject *good;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2758 good = PyObject_CallFunctionObjArgs(lz->func,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2759 item, NULL);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2760 if (good == NULL) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2761 Py_DECREF(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2762 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2763 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2764 ok = PyObject_IsTrue(good);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2765 Py_DECREF(good);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2766 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2767 if (!ok)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2768 return item;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2769 Py_DECREF(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2770 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2771 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2772
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2773 PyDoc_STRVAR(ifilterfalse_doc,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2774 "ifilterfalse(function or None, sequence) --> ifilterfalse object\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2775 \n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2776 Return those items of sequence for which function(item) is false.\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2777 If function is None, return the items that are false.");
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2778
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2779 static PyTypeObject ifilterfalse_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	2780 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2781 "itertools.ifilterfalse", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2782 sizeof(ifilterfalseobject), /* tp_basicsize */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2783 0, /* tp_itemsize */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2784 /* methods */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2785 (destructor)ifilterfalse_dealloc, /* tp_dealloc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2786 0, /* tp_print */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2787 0, /* tp_getattr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2788 0, /* tp_setattr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2789 0, /* tp_compare */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2790 0, /* tp_repr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2791 0, /* tp_as_number */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2792 0, /* tp_as_sequence */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2793 0, /* tp_as_mapping */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2794 0, /* tp_hash */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2795 0, /* tp_call */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2796 0, /* tp_str */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2797 PyObject_GenericGetAttr, /* tp_getattro */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2798 0, /* tp_setattro */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2799 0, /* tp_as_buffer */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2800 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2801 Py_TPFLAGS_BASETYPE, /* tp_flags */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2802 ifilterfalse_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2803 (traverseproc)ifilterfalse_traverse, /* tp_traverse */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2804 0, /* tp_clear */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2805 0, /* tp_richcompare */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2806 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	2807 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2808 (iternextfunc)ifilterfalse_next, /* tp_iternext */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2809 0, /* tp_methods */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2810 0, /* tp_members */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2811 0, /* tp_getset */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2812 0, /* tp_base */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2813 0, /* tp_dict */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2814 0, /* tp_descr_get */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2815 0, /* tp_descr_set */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2816 0, /* tp_dictoffset */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2817 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	2818 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2819 ifilterfalse_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2820 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2821 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2822
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2823
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2824 /* count object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2825
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2826 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2827 PyObject_HEAD
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	2828 Py_ssize_t cnt;
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2829 PyObject long_cnt; / Arbitrarily large count when cnt >= PY_SSIZE_T_MAX */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2830 } countobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2831
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2832 static PyTypeObject count_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2833
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2834 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2835 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	2836 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2837 countobject *lz;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	2838 Py_ssize_t cnt = 0;
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2839 PyObject *cnt_arg = NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2840 PyObject *long_cnt = NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2841
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	2842 if (type == &count_type && !_PyArg_NoKeywords("count()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2843 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	2844
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2845 if (!PyArg_UnpackTuple(args, "count", 0, 1, &cnt_arg))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2846 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2847
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2848 if (cnt_arg != NULL) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2849 cnt = PyInt_AsSsize_t(cnt_arg);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2850 if (cnt == -1 && PyErr_Occurred()) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2851 PyErr_Clear();
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2852 if (!PyLong_Check(cnt_arg)) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2853 PyErr_SetString(PyExc_TypeError, "an integer is required");
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2854 return NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2855 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2856 long_cnt = cnt_arg;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2857 Py_INCREF(long_cnt);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2858 cnt = PY_SSIZE_T_MAX;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2859 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2860 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2861
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2862 /* create countobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2863 lz = (countobject *)PyObject_New(countobject, &count_type);
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2864 if (lz == NULL) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2865 Py_XDECREF(long_cnt);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2866 return NULL;
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2867 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2868 lz->cnt = cnt;
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2869 lz->long_cnt = long_cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2870
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2871 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2872 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2873
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2874 static void
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2875 count_dealloc(countobject *lz)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2876 {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2877 Py_XDECREF(lz->long_cnt);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2878 PyObject_Del(lz);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2879 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2880
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2881 static PyObject *
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2882 count_nextlong(countobject *lz)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2883 {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2884 static PyObject *one = NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2885 PyObject *cnt;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2886 PyObject *stepped_up;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2887
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2888 if (lz->long_cnt == NULL) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2889 lz->long_cnt = PyInt_FromSsize_t(PY_SSIZE_T_MAX);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2890 if (lz->long_cnt == NULL)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2891 return NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2892 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2893 if (one == NULL) {
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2894 one = PyInt_FromLong(1);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2895 if (one == NULL)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2896 return NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2897 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2898 cnt = lz->long_cnt;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2899 assert(cnt != NULL);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2900 stepped_up = PyNumber_Add(cnt, one);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2901 if (stepped_up == NULL)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2902 return NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2903 lz->long_cnt = stepped_up;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2904 return cnt;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2905 }
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2906
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2907 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2908 count_next(countobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2909 {
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2910 if (lz->cnt == PY_SSIZE_T_MAX)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2911 return count_nextlong(lz);
39866 69530aa21b8a * regression bug, count_next was coercing a Py_ssize_t to an unsigned Py_size_t Jack Diederich <jackdied@gmail.com> parents: 39720 diff changeset	2912 return PyInt_FromSsize_t(lz->cnt++);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2913 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2914
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2915 static PyObject *
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2916 count_repr(countobject *lz)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2917 {
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2918 PyObject *cnt_repr;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2919 PyObject *result;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2920
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2921 if (lz->cnt != PY_SSIZE_T_MAX)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2922 return PyString_FromFormat("count(%zd)", lz->cnt);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2923
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2924 cnt_repr = PyObject_Repr(lz->long_cnt);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2925 if (cnt_repr == NULL)
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2926 return NULL;
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2927 result = PyString_FromFormat("count(%s)", PyString_AS_STRING(cnt_repr));
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2928 Py_DECREF(cnt_repr);
333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2929 return result;
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2930 }
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2931
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2932 PyDoc_STRVAR(count_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2933 "count([firstval]) --> count object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2934 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2935 Return a count object whose .next() method returns consecutive\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2936 integers starting from zero or, if specified, from firstval.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2937
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2938 static PyTypeObject count_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	2939 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2940 "itertools.count", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2941 sizeof(countobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2942 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2943 /* methods */
43458 333b052e9da9 itertools.count() no longer limited to sys.maxint. Raymond Hettinger <python@rcn.com> parents: 42333 diff changeset	2944 (destructor)count_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2945 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2946 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2947 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2948 0, /* tp_compare */
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	2949 (reprfunc)count_repr, /* tp_repr */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2950 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2951 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2952 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2953 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2954 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2955 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2956 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2957 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2958 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2959 Py_TPFLAGS_DEFAULT, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2960 count_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2961 0, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2962 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2963 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2964 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	2965 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	2966 (iternextfunc)count_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2967 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2968 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2969 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2970 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2971 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2972 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2973 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2974 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2975 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	2976 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2977 count_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2978 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2979
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2980
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2981 /* izip object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2982
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2983 #include "Python.h"
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2984
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2985 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2986 PyObject_HEAD
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	2987 Py_ssize_t tuplesize;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2988 PyObject ittuple; / tuple of iterators */
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	2989 PyObject *result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2990 } izipobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2991
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	2992 static PyTypeObject izip_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2993
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2994 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2995 izip_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	2996 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2997 izipobject *lz;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	2998 Py_ssize_t i;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2999 PyObject ittuple; / tuple of iterators */
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3000 PyObject *result;
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	3001 Py_ssize_t tuplesize = PySequence_Length(args);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3002
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3003 if (Py_Py3kWarningFlag &&
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3004 PyErr_Warn(PyExc_DeprecationWarning,
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3005 "In 3.x, itertools.izip() was moved to builtin zip()") < 0)
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3006 return NULL;
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3007
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	3008 if (type == &izip_type && !_PyArg_NoKeywords("izip()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	3009 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	3010
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3011 /* args must be a tuple */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3012 assert(PyTuple_Check(args));
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3013
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3014 /* obtain iterators */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3015 ittuple = PyTuple_New(tuplesize);
38420 499cf7c2f777 Fix compiler warning (and whitespace) on Mac OS 10.4. (A lot of this code looked duplicated, I wonder if a utility function could help reduce the duplication here.) Neal Norwitz <nnorwitz@gmail.com> parents: 38341 diff changeset	3016 if (ittuple == NULL)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3017 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3018 for (i=0; i < tuplesize; ++i) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3019 PyObject *item = PyTuple_GET_ITEM(args, i);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3020 PyObject *it = PyObject_GetIter(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3021 if (it == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3022 if (PyErr_ExceptionMatches(PyExc_TypeError))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3023 PyErr_Format(PyExc_TypeError,
38420 499cf7c2f777 Fix compiler warning (and whitespace) on Mac OS 10.4. (A lot of this code looked duplicated, I wonder if a utility function could help reduce the duplication here.) Neal Norwitz <nnorwitz@gmail.com> parents: 38341 diff changeset	3024 "izip argument #%zd must support iteration",
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3025 i+1);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3026 Py_DECREF(ittuple);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3027 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3028 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3029 PyTuple_SET_ITEM(ittuple, i, it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3030 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3031
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3032 /* create a result holder */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3033 result = PyTuple_New(tuplesize);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3034 if (result == NULL) {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3035 Py_DECREF(ittuple);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3036 return NULL;
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3037 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3038 for (i=0 ; i < tuplesize ; i++) {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3039 Py_INCREF(Py_None);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3040 PyTuple_SET_ITEM(result, i, Py_None);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3041 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3042
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3043 /* create izipobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3044 lz = (izipobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3045 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3046 Py_DECREF(ittuple);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3047 Py_DECREF(result);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3048 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3049 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3050 lz->ittuple = ittuple;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3051 lz->tuplesize = tuplesize;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3052 lz->result = result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3053
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3054 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3055 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3056
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3057 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3058 izip_dealloc(izipobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3059 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3060 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3061 Py_XDECREF(lz->ittuple);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3062 Py_XDECREF(lz->result);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3063 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	3064 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3065
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3066 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3067 izip_traverse(izipobject lz, visitproc visit, void arg)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3068 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	3069 Py_VISIT(lz->ittuple);
bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	3070 Py_VISIT(lz->result);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3071 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3072 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3073
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3074 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3075 izip_next(izipobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3076 {
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3077 Py_ssize_t i;
36518 9f3f87e920a6 Use Py_ssize_t for counts and sizes. Martin v. Löwis <martin@v.loewis.de> parents: 36484 diff changeset	3078 Py_ssize_t tuplesize = lz->tuplesize;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3079 PyObject *result = lz->result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3080 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3081 PyObject *item;
30240 f1ea311127cf For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30238 diff changeset	3082 PyObject *olditem;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3083
30093 0ac4396b1894 Modified itertools.izip() to match the behavior of __builtin__.zip() Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	3084 if (tuplesize == 0)
0ac4396b1894 Modified itertools.izip() to match the behavior of __builtin__.zip() Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	3085 return NULL;
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3086 if (Py_REFCNT(result) == 1) {
30238 a570bb9f133f SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 30093 diff changeset	3087 Py_INCREF(result);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3088 for (i=0 ; i < tuplesize ; i++) {
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3089 it = PyTuple_GET_ITEM(lz->ittuple, i);
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	3090 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3091 item = (*Py_TYPE(it)->tp_iternext)(it);
30238 a570bb9f133f SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 30093 diff changeset	3092 if (item == NULL) {
a570bb9f133f SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 30093 diff changeset	3093 Py_DECREF(result);
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3094 return NULL;
30238 a570bb9f133f SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 30093 diff changeset	3095 }
30240 f1ea311127cf For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30238 diff changeset	3096 olditem = PyTuple_GET_ITEM(result, i);
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3097 PyTuple_SET_ITEM(result, i, item);
30240 f1ea311127cf For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30238 diff changeset	3098 Py_DECREF(olditem);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3099 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3100 } else {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3101 result = PyTuple_New(tuplesize);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3102 if (result == NULL)
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	3103 return NULL;
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3104 for (i=0 ; i < tuplesize ; i++) {
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3105 it = PyTuple_GET_ITEM(lz->ittuple, i);
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	3106 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3107 item = (*Py_TYPE(it)->tp_iternext)(it);
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3108 if (item == NULL) {
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3109 Py_DECREF(result);
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3110 return NULL;
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3111 }
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	3112 PyTuple_SET_ITEM(result, i, item);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3113 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3114 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3115 return result;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3116 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3117
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3118 PyDoc_STRVAR(izip_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3119 "izip(iter1 [,iter2 [...]]) --> izip object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3120 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3121 Return a izip object whose .next() method returns a tuple where\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3122 the i-th element comes from the i-th iterable argument. The .next()\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3123 method continues until the shortest iterable in the argument sequence\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3124 is exhausted and then it raises StopIteration. Works like the zip()\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3125 function but consumes less memory by returning an iterator instead of\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3126 a list.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3127
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	3128 static PyTypeObject izip_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	3129 PyVarObject_HEAD_INIT(NULL, 0)
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3130 "itertools.izip", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3131 sizeof(izipobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3132 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3133 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3134 (destructor)izip_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3135 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3136 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3137 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3138 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3139 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3140 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3141 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3142 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3143 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3144 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3145 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3146 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3147 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3148 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3149 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3150 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3151 izip_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3152 (traverseproc)izip_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3153 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3154 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3155 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	3156 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3157 (iternextfunc)izip_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3158 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3159 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3160 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3161 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3162 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3163 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3164 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3165 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3166 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	3167 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3168 izip_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3169 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3170 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3171
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3172
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3173 /* repeat object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3174
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3175 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3176 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3177 PyObject *element;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3178 Py_ssize_t cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3179 } repeatobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3180
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	3181 static PyTypeObject repeat_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3182
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3183 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3184 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	3185 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3186 repeatobject *ro;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3187 PyObject *element;
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3188 Py_ssize_t cnt = -1;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3189
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 39866 diff changeset	3190 if (type == &repeat_type && !_PyArg_NoKeywords("repeat()", kwds))
35449 1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	3191 return NULL;
1fdfd0130e28 Disallow keyword arguments for type constructors that don't use them. Georg Brandl <georg@python.org> parents: 34108 diff changeset	3192
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3193 if (!PyArg_ParseTuple(args, "O\|n:repeat", &element, &cnt))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3194 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3195
28945 001a90555780 * Added a substantial number of edge case and argument tests for Raymond Hettinger <python@rcn.com> parents: 28930 diff changeset	3196 if (PyTuple_Size(args) == 2 && cnt < 0)
001a90555780 * Added a substantial number of edge case and argument tests for Raymond Hettinger <python@rcn.com> parents: 28930 diff changeset	3197 cnt = 0;
001a90555780 * Added a substantial number of edge case and argument tests for Raymond Hettinger <python@rcn.com> parents: 28930 diff changeset	3198
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3199 ro = (repeatobject *)type->tp_alloc(type, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3200 if (ro == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3201 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3202 Py_INCREF(element);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3203 ro->element = element;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3204 ro->cnt = cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3205 return (PyObject *)ro;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3206 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3207
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3208 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3209 repeat_dealloc(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3210 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3211 PyObject_GC_UnTrack(ro);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3212 Py_XDECREF(ro->element);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3213 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	3214 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3215
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3216 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3217 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	3218 {
32682 bc4dd31c73fd Exercise Jim's VISIT macro. Raymond Hettinger <python@rcn.com> parents: 32010 diff changeset	3219 Py_VISIT(ro->element);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3220 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3221 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3222
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3223 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3224 repeat_next(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3225 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3226 if (ro->cnt == 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3227 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3228 if (ro->cnt > 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3229 ro->cnt--;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3230 Py_INCREF(ro->element);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3231 return ro->element;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3232 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3233
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3234 static PyObject *
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3235 repeat_repr(repeatobject *ro)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3236 {
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3237 PyObject result, objrepr;
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3238
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3239 objrepr = PyObject_Repr(ro->element);
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3240 if (objrepr == NULL)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3241 return NULL;
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3242
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3243 if (ro->cnt == -1)
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3244 result = PyString_FromFormat("repeat(%s)",
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3245 PyString_AS_STRING(objrepr));
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3246 else
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3247 result = PyString_FromFormat("repeat(%s, %zd)",
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3248 PyString_AS_STRING(objrepr), ro->cnt);
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3249 Py_DECREF(objrepr);
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3250 return result;
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3251 }
848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3252
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3253 static PyObject *
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3254 repeat_len(repeatobject *ro)
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3255 {
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3256 if (ro->cnt == -1) {
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3257 PyErr_SetString(PyExc_TypeError, "len() of unsized object");
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3258 return NULL;
e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3259 }
38341 a654d61d36d5 use Py_ssize_t in places that may need it Jack Diederich <jackdied@gmail.com> parents: 37757 diff changeset	3260 return PyInt_FromSize_t(ro->cnt);
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3261 }
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3262
36484 20f75d5e79e9 Renamed _length_cue() to __length_hint__(). See: Armin Rigo <arigo@tunes.org> parents: 36324 diff changeset	3263 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	3264
e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3265 static PyMethodDef repeat_methods[] = {
36484 20f75d5e79e9 Renamed _length_cue() to __length_hint__(). See: Armin Rigo <arigo@tunes.org> parents: 36324 diff changeset	3266 {"__length_hint__", (PyCFunction)repeat_len, METH_NOARGS, length_hint_doc},
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3267 {NULL, NULL} /* sentinel */
31501 14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3268 };
14f860b03be2 Give itertools.repeat() a length method. Raymond Hettinger <python@rcn.com> parents: 31136 diff changeset	3269
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3270 PyDoc_STRVAR(repeat_doc,
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3271 "repeat(element [,times]) -> create an iterator which returns the element\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3272 for the specified number of times. If not specified, returns the element\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3273 endlessly.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3274
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	3275 static PyTypeObject repeat_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	3276 PyVarObject_HEAD_INIT(NULL, 0)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3277 "itertools.repeat", /* tp_name */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3278 sizeof(repeatobject), /* tp_basicsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3279 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3280 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3281 (destructor)repeat_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3282 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3283 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3284 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3285 0, /* tp_compare */
31890 848bce2215d3 Provide more information representations of repeat() and count(). Raymond Hettinger <python@rcn.com> parents: 31675 diff changeset	3286 (reprfunc)repeat_repr, /* tp_repr */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3287 0, /* tp_as_number */
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3288 0, /* tp_as_sequence */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3289 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3290 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3291 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3292 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3293 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3294 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3295 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3296 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3297 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3298 repeat_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3299 (traverseproc)repeat_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3300 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3301 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3302 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	3303 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3304 (iternextfunc)repeat_next, /* tp_iternext */
35636 e22334b1da67 Convert iterator __len__() methods to a private API. Raymond Hettinger <python@rcn.com> parents: 35449 diff changeset	3305 repeat_methods, /* tp_methods */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3306 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3307 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3308 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3309 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3310 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3311 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3312 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3313 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	3314 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3315 repeat_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3316 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3317 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3318
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3319 /* iziplongest object ************************************************************/
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3320
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3321 #include "Python.h"
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3322
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3323 typedef struct {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3324 PyObject_HEAD
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3325 Py_ssize_t tuplesize;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3326 Py_ssize_t numactive;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3327 PyObject ittuple; / tuple of iterators */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3328 PyObject *result;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3329 PyObject *fillvalue;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3330 } iziplongestobject;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3331
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3332 static PyTypeObject iziplongest_type;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3333
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3334 static PyObject *
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3335 izip_longest_new(PyTypeObject type, PyObject args, PyObject *kwds)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3336 {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3337 iziplongestobject *lz;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3338 Py_ssize_t i;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3339 PyObject ittuple; / tuple of iterators */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3340 PyObject *result;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3341 PyObject *fillvalue = Py_None;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3342 Py_ssize_t tuplesize = PySequence_Length(args);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3343
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3344 if (Py_Py3kWarningFlag &&
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3345 PyErr_Warn(PyExc_DeprecationWarning,
45746 d72d527de711 Consistent tense. Raymond Hettinger <python@rcn.com> parents: 45745 diff changeset	3346 "In 3.x, izip_longest() was renamed to zip_longest().") < 0)
45745 d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3347 return NULL;
d5024ee9e88a Add 2-to-3 support for the itertools moved to builtins or renamed. Raymond Hettinger <python@rcn.com> parents: 45699 diff changeset	3348
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3349 if (kwds != NULL && PyDict_CheckExact(kwds) && PyDict_Size(kwds) > 0) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3350 fillvalue = PyDict_GetItemString(kwds, "fillvalue");
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3351 if (fillvalue == NULL \|\| PyDict_Size(kwds) > 1) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3352 PyErr_SetString(PyExc_TypeError,
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3353 "izip_longest() got an unexpected keyword argument");
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3354 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3355 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3356 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3357
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3358 /* args must be a tuple */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3359 assert(PyTuple_Check(args));
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3360
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3361 /* obtain iterators */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3362 ittuple = PyTuple_New(tuplesize);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3363 if (ittuple == NULL)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3364 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3365 for (i=0; i < tuplesize; ++i) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3366 PyObject *item = PyTuple_GET_ITEM(args, i);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3367 PyObject *it = PyObject_GetIter(item);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3368 if (it == NULL) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3369 if (PyErr_ExceptionMatches(PyExc_TypeError))
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3370 PyErr_Format(PyExc_TypeError,
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3371 "izip_longest argument #%zd must support iteration",
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3372 i+1);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3373 Py_DECREF(ittuple);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3374 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3375 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3376 PyTuple_SET_ITEM(ittuple, i, it);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3377 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3378
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3379 /* create a result holder */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3380 result = PyTuple_New(tuplesize);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3381 if (result == NULL) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3382 Py_DECREF(ittuple);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3383 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3384 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3385 for (i=0 ; i < tuplesize ; i++) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3386 Py_INCREF(Py_None);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3387 PyTuple_SET_ITEM(result, i, Py_None);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3388 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3389
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3390 /* create iziplongestobject structure */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3391 lz = (iziplongestobject *)type->tp_alloc(type, 0);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3392 if (lz == NULL) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3393 Py_DECREF(ittuple);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3394 Py_DECREF(result);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3395 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3396 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3397 lz->ittuple = ittuple;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3398 lz->tuplesize = tuplesize;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3399 lz->numactive = tuplesize;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3400 lz->result = result;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3401 Py_INCREF(fillvalue);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3402 lz->fillvalue = fillvalue;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3403 return (PyObject *)lz;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3404 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3405
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3406 static void
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3407 izip_longest_dealloc(iziplongestobject *lz)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3408 {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3409 PyObject_GC_UnTrack(lz);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3410 Py_XDECREF(lz->ittuple);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3411 Py_XDECREF(lz->result);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3412 Py_XDECREF(lz->fillvalue);
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3413 Py_TYPE(lz)->tp_free(lz);
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3414 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3415
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3416 static int
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3417 izip_longest_traverse(iziplongestobject lz, visitproc visit, void arg)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3418 {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3419 Py_VISIT(lz->ittuple);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3420 Py_VISIT(lz->result);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3421 Py_VISIT(lz->fillvalue);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3422 return 0;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3423 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3424
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3425 static PyObject *
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3426 izip_longest_next(iziplongestobject *lz)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3427 {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3428 Py_ssize_t i;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3429 Py_ssize_t tuplesize = lz->tuplesize;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3430 PyObject *result = lz->result;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3431 PyObject *it;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3432 PyObject *item;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3433 PyObject *olditem;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3434
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3435 if (tuplesize == 0)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3436 return NULL;
40878 8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3437 if (lz->numactive == 0)
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3438 return NULL;
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3439 if (Py_REFCNT(result) == 1) {
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3440 Py_INCREF(result);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3441 for (i=0 ; i < tuplesize ; i++) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3442 it = PyTuple_GET_ITEM(lz->ittuple, i);
40878 8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3443 if (it == NULL) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3444 Py_INCREF(lz->fillvalue);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3445 item = lz->fillvalue;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3446 } else {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3447 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3448 item = (*Py_TYPE(it)->tp_iternext)(it);
40878 8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3449 if (item == NULL) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3450 lz->numactive -= 1;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3451 if (lz->numactive == 0) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3452 Py_DECREF(result);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3453 return NULL;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3454 } else {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3455 Py_INCREF(lz->fillvalue);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3456 item = lz->fillvalue;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3457 PyTuple_SET_ITEM(lz->ittuple, i, NULL);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3458 Py_DECREF(it);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3459 }
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3460 }
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3461 }
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3462 olditem = PyTuple_GET_ITEM(result, i);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3463 PyTuple_SET_ITEM(result, i, item);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3464 Py_DECREF(olditem);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3465 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3466 } else {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3467 result = PyTuple_New(tuplesize);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3468 if (result == NULL)
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3469 return NULL;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3470 for (i=0 ; i < tuplesize ; i++) {
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3471 it = PyTuple_GET_ITEM(lz->ittuple, i);
40878 8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3472 if (it == NULL) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3473 Py_INCREF(lz->fillvalue);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3474 item = lz->fillvalue;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3475 } else {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3476 assert(PyIter_Check(it));
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3477 item = (*Py_TYPE(it)->tp_iternext)(it);
40878 8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3478 if (item == NULL) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3479 lz->numactive -= 1;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3480 if (lz->numactive == 0) {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3481 Py_DECREF(result);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3482 return NULL;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3483 } else {
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3484 Py_INCREF(lz->fillvalue);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3485 item = lz->fillvalue;
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3486 PyTuple_SET_ITEM(lz->ittuple, i, NULL);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3487 Py_DECREF(it);
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3488 }
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3489 }
8128619e984d Remove filler struct item and fix leak. Raymond Hettinger <python@rcn.com> parents: 40877 diff changeset	3490 }
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3491 PyTuple_SET_ITEM(result, i, item);
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3492 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3493 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3494 return result;
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3495 }
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3496
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3497 PyDoc_STRVAR(izip_longest_doc,
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3498 "izip_longest(iter1 [,iter2 [...]], [fillvalue=None]) --> izip_longest object\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3499 \n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3500 Return an izip_longest object whose .next() method returns a tuple where\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3501 the i-th element comes from the i-th iterable argument. The .next()\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3502 method continues until the longest iterable in the argument sequence\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3503 is exhausted and then it raises StopIteration. When the shorter iterables\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3504 are exhausted, the fillvalue is substituted in their place. The fillvalue\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3505 defaults to None or can be specified by a keyword argument.\n\
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3506 ");
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3507
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3508 static PyTypeObject iziplongest_type = {
42333 6620a510e9a9 PEP 3123: Provide forward compatibility with Python 3.0, while keeping Martin v. Löwis <martin@v.loewis.de> parents: 41590 diff changeset	3509 PyVarObject_HEAD_INIT(NULL, 0)
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3510 "itertools.izip_longest", /* tp_name */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3511 sizeof(iziplongestobject), /* tp_basicsize */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3512 0, /* tp_itemsize */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3513 /* methods */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3514 (destructor)izip_longest_dealloc, /* tp_dealloc */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3515 0, /* tp_print */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3516 0, /* tp_getattr */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3517 0, /* tp_setattr */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3518 0, /* tp_compare */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3519 0, /* tp_repr */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3520 0, /* tp_as_number */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3521 0, /* tp_as_sequence */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3522 0, /* tp_as_mapping */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3523 0, /* tp_hash */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3524 0, /* tp_call */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3525 0, /* tp_str */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3526 PyObject_GenericGetAttr, /* tp_getattro */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3527 0, /* tp_setattro */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3528 0, /* tp_as_buffer */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3529 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3530 Py_TPFLAGS_BASETYPE, /* tp_flags */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3531 izip_longest_doc, /* tp_doc */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3532 (traverseproc)izip_longest_traverse, /* tp_traverse */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3533 0, /* tp_clear */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3534 0, /* tp_richcompare */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3535 0, /* tp_weaklistoffset */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3536 PyObject_SelfIter, /* tp_iter */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3537 (iternextfunc)izip_longest_next, /* tp_iternext */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3538 0, /* tp_methods */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3539 0, /* tp_members */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3540 0, /* tp_getset */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3541 0, /* tp_base */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3542 0, /* tp_dict */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3543 0, /* tp_descr_get */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3544 0, /* tp_descr_set */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3545 0, /* tp_dictoffset */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3546 0, /* tp_init */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3547 0, /* tp_alloc */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3548 izip_longest_new, /* tp_new */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3549 PyObject_GC_Del, /* tp_free */
6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3550 };
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3551
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3552 /* module level code ********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3553
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3554 PyDoc_STRVAR(module_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3555 "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	3556 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3557 Infinite iterators:\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3558 count([n]) --> n, n+1, n+2, ...\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3559 cycle(p) --> p0, p1, ... plast, p0, p1, ...\n\
28601 58db5fbd9c28 Fix docstring typo Andrew M. Kuchling <amk@amk.ca> parents: 28320 diff changeset	3560 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	3561 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3562 Iterators terminating on the shortest input sequence:\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3563 izip(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ... \n\
40877 6fa0d682e1e5 Add itertools.izip_longest(). Raymond Hettinger <python@rcn.com> parents: 40766 diff changeset	3564 izip_longest(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ... \n\
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3565 ifilter(pred, seq) --> elements of seq where pred(elem) is True\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3566 ifilterfalse(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	3567 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	3568 seq[start:stop:step]\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3569 imap(fun, p, q, ...) --> fun(p0, q0), fun(p1, q1), ...\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3570 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	3571 tee(it, n=2) --> (it1, it2 , ... itn) splits one iterator into n\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3572 chain(p, q, ...) --> p0, p1, ... plast, q0, q1, ... \n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3573 takewhile(pred, seq) --> seq[0], seq[1], until pred fails\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3574 dropwhile(pred, seq) --> seq[n], seq[n+1], starting when pred fails\n\
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	3575 groupby(iterable[, keyfunc]) --> sub-iterators grouped by value of keyfunc(v)\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3576 ");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3577
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3578
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3579 static PyMethodDef module_methods[] = {
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3580 {"tee", (PyCFunction)tee, METH_VARARGS, tee_doc},
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3581 {NULL, NULL} /* sentinel */
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3582 };
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3583
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3584 PyMODINIT_FUNC
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3585 inititertools(void)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3586 {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3587 int i;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3588 PyObject *m;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3589 char *name;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3590 PyTypeObject *typelist[] = {
45539 ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	3591 &combinations_type,
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3592 &cycle_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3593 &dropwhile_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3594 &takewhile_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3595 &islice_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3596 &starmap_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3597 &imap_type,
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3598 &chain_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3599 &ifilter_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3600 &ifilterfalse_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3601 &count_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3602 &izip_type,
45425 3cc57fd87d0e First draft for itertools.product(). Docs and other updates forthcoming. Raymond Hettinger <python@rcn.com> parents: 44717 diff changeset	3603 &iziplongest_type,
45675 c197e18449b4 C implementation of itertools.permutations(). Raymond Hettinger <python@rcn.com> parents: 45658 diff changeset	3604 &permutations_type,
45539 ca45eb71f2d2 Add itertools.combinations(). Raymond Hettinger <python@rcn.com> parents: 45526 diff changeset	3605 &product_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3606 &repeat_type,
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	3607 &groupby_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3608 NULL
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3609 };
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3610
44350 26384b8180c7 #1629: Renamed Py_Size, Py_Type and Py_Refcnt to Py_SIZE, Py_TYPE and Py_REFCNT. Macros for b/w compatibility are available. Christian Heimes <christian@cheimes.de> parents: 43458 diff changeset	3611 Py_TYPE(&teedataobject_type) = &PyType_Type;
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3612 m = Py_InitModule3("itertools", module_methods, module_doc);
36324 1af7f719ed28 Check return result from Py_InitModule(). This API can fail. Neal Norwitz <nnorwitz@gmail.com>* parents: 35999 diff changeset	3613 if (m == NULL)
1af7f719ed28 Check return result from Py_InitModule(). This API can fail. Neal Norwitz <nnorwitz@gmail.com>* parents: 35999 diff changeset	3614 return;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3615
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3616 for (i=0 ; typelist[i] != NULL ; i++) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3617 if (PyType_Ready(typelist[i]) < 0)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3618 return;
29225 f3f306457b43 Fixed dotted name assertion. Raymond Hettinger <python@rcn.com> parents: 28945 diff changeset	3619 name = strchr(typelist[i]->tp_name, '.');
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	3620 assert (name != NULL);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3621 Py_INCREF(typelist[i]);
29225 f3f306457b43 Fixed dotted name assertion. Raymond Hettinger <python@rcn.com> parents: 28945 diff changeset	3622 PyModule_AddObject(m, name+1, (PyObject *)typelist[i]);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	3623 }
30900 3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3624
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3625 if (PyType_Ready(&teedataobject_type) < 0)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3626 return;
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3627 if (PyType_Ready(&tee_type) < 0)
3096fab2a19c Improve the implementation of itertools.tee(). Raymond Hettinger <python@rcn.com> parents: 30800 diff changeset	3628 return;
31136 fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	3629 if (PyType_Ready(&_grouper_type) < 0)
fde74fadf780 Implement itertools.groupby() Raymond Hettinger <python@rcn.com> parents: 30900 diff changeset	3630 return;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3631 }

Mercurial > cpython-lru_cache-weakref

annotate Modules/itertoolsmodule.c @ 45837:c5241069fffd legacy-trunk