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

annotate Modules/itertoolsmodule.c @ 33091:48100a01ac7c 2.3

Ignore the locale's encoding if it is an empty string.

author	Martin v. Löwis <martin@v.loewis.de>
date	Thu, 12 Aug 2004 13:14:20 +0000
parents	50708b8f0d28
children

rev	line source
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	2 #include "Python.h"
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	3
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	4 /* Itertools module written and maintained
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	5 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	6 Copyright (c) 2003 Python Software Foundation.
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	7 All rights reserved.
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	8 */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	9
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	10 /* cycle object **********************************************************/
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	11
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	12 typedef struct {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	13 PyObject_HEAD
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	14 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	15 PyObject *saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	16 int firstpass;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	17 } cycleobject;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	18
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	19 static PyTypeObject cycle_type;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	20
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	21 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	22 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	23 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	24 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	25 PyObject *iterable;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	26 PyObject *saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	27 cycleobject *lz;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	28
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	29 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	30 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	31
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	32 /* Get iterator. */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	33 it = PyObject_GetIter(iterable);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	34 if (it == NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	35 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	36
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	37 saved = PyList_New(0);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	38 if (saved == NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	39 Py_DECREF(it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	40 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	41 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	42
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	43 /* create cycleobject structure */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	44 lz = (cycleobject *)type->tp_alloc(type, 0);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	45 if (lz == NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	46 Py_DECREF(it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	47 Py_DECREF(saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	48 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	49 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	50 lz->it = it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	51 lz->saved = saved;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	52 lz->firstpass = 0;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	53
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	54 return (PyObject *)lz;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	55 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	56
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	57 static void
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	58 cycle_dealloc(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	59 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	60 PyObject_GC_UnTrack(lz);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	61 Py_XDECREF(lz->saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	62 Py_XDECREF(lz->it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	63 lz->ob_type->tp_free(lz);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	64 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	65
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	66 static int
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	67 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	68 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	69 int err;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	70
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	71 if (lz->it) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	72 err = visit(lz->it, arg);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	73 if (err)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	74 return err;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	75 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	76 if (lz->saved) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	77 err = visit(lz->saved, arg);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	78 if (err)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	79 return err;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	80 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	81 return 0;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	82 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	83
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	84 static PyObject *
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	85 cycle_next(cycleobject *lz)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	86 {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	87 PyObject *item;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	88 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	89
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	90 while (1) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	91 item = PyIter_Next(lz->it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	92 if (item != NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	93 if (!lz->firstpass)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	94 PyList_Append(lz->saved, item);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	95 return item;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	96 }
32011 50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	97 if (PyErr_Occurred()) {
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	98 if (PyErr_ExceptionMatches(PyExc_StopIteration))
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	99 PyErr_Clear();
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	100 else
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	101 return NULL;
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	102 }
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	103 if (PyList_Size(lz->saved) == 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	104 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	105 it = PyObject_GetIter(lz->saved);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	106 if (it == NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	107 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	108 Py_DECREF(lz->it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	109 lz->it = it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	110 lz->firstpass = 1;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	111 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	112 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	113
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	114 PyDoc_STRVAR(cycle_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	115 "cycle(iterable) --> cycle object\n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	116 \n\
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	117 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	118 Then repeat the sequence indefinitely.");
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	119
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	120 static PyTypeObject cycle_type = {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	121 PyObject_HEAD_INIT(NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	122 0, /* ob_size */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	123 "itertools.cycle", /* tp_name */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	124 sizeof(cycleobject), /* tp_basicsize */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	125 0, /* tp_itemsize */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	126 /* methods */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	127 (destructor)cycle_dealloc, /* tp_dealloc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	128 0, /* tp_print */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	129 0, /* tp_getattr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	130 0, /* tp_setattr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	131 0, /* tp_compare */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	132 0, /* tp_repr */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	133 0, /* tp_as_number */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	134 0, /* tp_as_sequence */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	135 0, /* tp_as_mapping */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	136 0, /* tp_hash */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	137 0, /* tp_call */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	138 0, /* tp_str */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	139 PyObject_GenericGetAttr, /* tp_getattro */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	140 0, /* tp_setattro */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	141 0, /* tp_as_buffer */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	142 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	143 Py_TPFLAGS_BASETYPE, /* tp_flags */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	144 cycle_doc, /* tp_doc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	145 (traverseproc)cycle_traverse, /* tp_traverse */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	146 0, /* tp_clear */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	147 0, /* tp_richcompare */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	148 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	149 PyObject_SelfIter, /* tp_iter */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	150 (iternextfunc)cycle_next, /* tp_iternext */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	151 0, /* tp_methods */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	152 0, /* tp_members */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	153 0, /* tp_getset */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	154 0, /* tp_base */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	155 0, /* tp_dict */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	156 0, /* tp_descr_get */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	157 0, /* tp_descr_set */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	158 0, /* tp_dictoffset */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	159 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	160 0, /* tp_alloc */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	161 cycle_new, /* tp_new */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	162 PyObject_GC_Del, /* tp_free */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	163 };
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	164
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	165
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	166 /* dropwhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	167
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	168 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	169 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	170 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	171 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	172 long start;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	173 } dropwhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	174
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	175 static PyTypeObject dropwhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	176
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	177 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	178 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	179 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	180 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	181 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	182 dropwhileobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	183
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	184 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	185 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	186
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	187 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	188 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	189 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	190 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	191
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	192 /* create dropwhileobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	193 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	194 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	195 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	196 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	197 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	198 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	199 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	200 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	201 lz->start = 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	202
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	203 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	204 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	205
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	206 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	207 dropwhile_dealloc(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	208 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	209 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	210 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	211 Py_XDECREF(lz->it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	212 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	213 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	214
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	215 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	216 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	217 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	218 int err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	219
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	220 if (lz->it) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	221 err = visit(lz->it, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	222 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	223 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	224 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	225 if (lz->func) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	226 err = visit(lz->func, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	227 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	228 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	229 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	230 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	231 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	232
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	233 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	234 dropwhile_next(dropwhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	235 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	236 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	237 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	238 long ok;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	239
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	240 for (;;) {
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	241 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	242 item = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	243 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	244 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	245 if (lz->start == 1)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	246 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	247
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	248 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	249 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	250 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	251 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	252 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	253 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	254 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	255 if (!ok) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	256 lz->start = 1;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	257 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	258 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	259 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	260 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	261 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	262
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	263 PyDoc_STRVAR(dropwhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	264 "dropwhile(predicate, iterable) --> dropwhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	265 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	266 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	267 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	268
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	269 static PyTypeObject dropwhile_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	270 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	271 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	272 "itertools.dropwhile", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	273 sizeof(dropwhileobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	274 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	275 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	276 (destructor)dropwhile_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	277 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	278 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	279 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	280 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	281 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	282 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	283 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	284 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	285 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	286 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	287 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	288 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	289 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	290 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	291 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	292 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	293 dropwhile_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	294 (traverseproc)dropwhile_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	295 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	296 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	297 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	298 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	299 (iternextfunc)dropwhile_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	300 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	301 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	302 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	303 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	304 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	305 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	306 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	307 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	308 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	309 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	310 dropwhile_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	311 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	312 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	313
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	314
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	315 /* takewhile object **********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	316
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	317 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	318 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	319 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	320 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	321 long stop;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	322 } takewhileobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	323
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	324 static PyTypeObject takewhile_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	325
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	326 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	327 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	328 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	329 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	330 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	331 takewhileobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	332
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	333 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	334 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	335
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	336 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	337 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	338 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	339 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	340
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	341 /* create takewhileobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	342 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	343 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	344 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	345 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	346 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	347 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	348 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	349 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	350 lz->stop = 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	351
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	352 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	353 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	354
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	355 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	356 takewhile_dealloc(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	357 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	358 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	359 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	360 Py_XDECREF(lz->it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	361 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	362 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	363
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	364 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	365 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	366 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	367 int err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	368
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	369 if (lz->it) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	370 err = visit(lz->it, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	371 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	372 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	373 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	374 if (lz->func) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	375 err = visit(lz->func, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	376 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	377 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	378 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	379 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	380 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	381
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	382 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	383 takewhile_next(takewhileobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	384 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	385 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	386 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	387 long ok;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	388
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	389 if (lz->stop == 1)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	390 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	391
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	392 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	393 item = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	394 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	395 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	396
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	397 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	398 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	399 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	400 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	401 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	402 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	403 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	404 if (ok)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	405 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	406 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	407 lz->stop = 1;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	408 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	409 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	410
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	411 PyDoc_STRVAR(takewhile_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	412 "takewhile(predicate, iterable) --> takewhile object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	413 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	414 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	415 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	416
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	417 static PyTypeObject takewhile_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	418 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	419 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	420 "itertools.takewhile", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	421 sizeof(takewhileobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	422 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	423 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	424 (destructor)takewhile_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	425 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	426 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	427 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	428 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	429 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	430 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	431 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	432 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	433 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	434 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	435 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	436 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	437 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	438 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	439 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	440 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	441 takewhile_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	442 (traverseproc)takewhile_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	443 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	444 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	445 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	446 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	447 (iternextfunc)takewhile_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	448 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	449 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	450 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	451 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	452 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	453 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	454 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	455 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	456 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	457 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	458 takewhile_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	459 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	460 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	461
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	462
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	463 /* islice object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	464
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	465 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	466 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	467 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	468 long next;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	469 long stop;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	470 long step;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	471 long cnt;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	472 } isliceobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	473
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	474 static PyTypeObject islice_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	475
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	476 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	477 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	478 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	479 PyObject *seq;
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	480 long start=0, stop=-1, step=1;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	481 PyObject it, a1=NULL, *a2=NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	482 int numargs;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	483 isliceobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	484
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	485 numargs = PyTuple_Size(args);
28930 18082b2d2ebd The previous made the stop argument optional. Raymond Hettinger <python@rcn.com> parents: 28929 diff changeset	486 if (!PyArg_ParseTuple(args, "OO\|Ol:islice", &seq, &a1, &a2, &step))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	487 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	488
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	489 if (numargs == 2) {
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	490 if (a1 != Py_None) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	491 stop = PyInt_AsLong(a1);
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	492 if (stop == -1) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	493 if (PyErr_Occurred())
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	494 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	495 PyErr_SetString(PyExc_ValueError,
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	496 "Stop argument must be an integer or None.");
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	497 return NULL;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	498 }
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	499 }
28930 18082b2d2ebd The previous made the stop argument optional. Raymond Hettinger <python@rcn.com> parents: 28929 diff changeset	500 } else {
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	501 start = PyInt_AsLong(a1);
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	502 if (start == -1 && PyErr_Occurred()) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	503 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	504 PyErr_SetString(PyExc_ValueError,
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	505 "Start argument must be an integer.");
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	506 return NULL;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	507 }
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	508 if (a2 != Py_None) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	509 stop = PyInt_AsLong(a2);
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	510 if (stop == -1) {
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	511 if (PyErr_Occurred())
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	512 PyErr_Clear();
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	513 PyErr_SetString(PyExc_ValueError,
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	514 "Stop argument must be an integer or None.");
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	515 return NULL;
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	516 }
92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	517 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	518 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	519
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	520 if (start<0 \|\| stop<-1) {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	521 PyErr_SetString(PyExc_ValueError,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	522 "Indices for islice() must be positive.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	523 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	524 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	525
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	526 if (step<1) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	527 PyErr_SetString(PyExc_ValueError,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	528 "Step must be one or larger for islice().");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	529 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	530 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	531
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	532 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	533 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	534 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	535 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	536
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	537 /* create isliceobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	538 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	539 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	540 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	541 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	542 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	543 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	544 lz->next = start;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	545 lz->stop = stop;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	546 lz->step = step;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	547 lz->cnt = 0L;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	548
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	549 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	550 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	551
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	552 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	553 islice_dealloc(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	554 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	555 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	556 Py_XDECREF(lz->it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	557 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	558 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	559
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	560 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	561 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	562 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	563 if (lz->it)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	564 return visit(lz->it, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	565 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	566 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	567
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	568 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	569 islice_next(isliceobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	570 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	571 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	572 PyObject *it = lz->it;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	573 long oldnext;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	574
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	575 while (lz->cnt < lz->next) {
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	576 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	577 item = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	578 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	579 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	580 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	581 lz->cnt++;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	582 }
28929 92a1e585abf7 SF bug #730685: itertools.islice stop argument is not optional Raymond Hettinger <python@rcn.com> parents: 28601 diff changeset	583 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	584 return NULL;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	585 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	586 item = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	587 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	588 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	589 lz->cnt++;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	590 oldnext = lz->next;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	591 lz->next += lz->step;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	592 if (lz->next < oldnext) /* Check for overflow */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	593 lz->next = lz->stop;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	594 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	595 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	596
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	597 PyDoc_STRVAR(islice_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	598 "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	599 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	600 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	601 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	602 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	603 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	604 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	605 but returns an iterator.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	606
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	607 static PyTypeObject islice_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	608 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	609 0, /* ob_size */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	610 "itertools.islice", /* tp_name */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	611 sizeof(isliceobject), /* tp_basicsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	612 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	613 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	614 (destructor)islice_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	615 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	616 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	617 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	618 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	619 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	620 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	621 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	622 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	623 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	624 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	625 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	626 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	627 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	628 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	629 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	630 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	631 islice_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	632 (traverseproc)islice_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	633 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	634 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	635 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	636 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	637 (iternextfunc)islice_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	638 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	639 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	640 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	641 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	642 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	643 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	644 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	645 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	646 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	647 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	648 islice_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	649 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	650 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	651
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	652
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	653 /* starmap object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	654
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	655 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	656 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	657 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	658 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	659 } starmapobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	660
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	661 static PyTypeObject starmap_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	662
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	663 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	664 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	665 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	666 PyObject func, seq;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	667 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	668 starmapobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	669
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	670 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	671 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	672
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	673 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	674 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	675 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	676 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	677
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	678 /* create starmapobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	679 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	680 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	681 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	682 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	683 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	684 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	685 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	686 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	687
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	688 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	689 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	690
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	691 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	692 starmap_dealloc(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	693 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	694 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	695 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	696 Py_XDECREF(lz->it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	697 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	698 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	699
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	700 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	701 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	702 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	703 int err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	704
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	705 if (lz->it) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	706 err = visit(lz->it, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	707 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	708 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	709 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	710 if (lz->func) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	711 err = visit(lz->func, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	712 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	713 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	714 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	715 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	716 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	717
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	718 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	719 starmap_next(starmapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	720 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	721 PyObject *args;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	722 PyObject *result;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	723 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	724
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	725 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	726 args = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	727 if (args == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	728 return NULL;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	729 if (!PyTuple_CheckExact(args)) {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	730 Py_DECREF(args);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	731 PyErr_SetString(PyExc_TypeError,
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	732 "iterator must return a tuple");
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	733 return NULL;
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	734 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	735 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	736 Py_DECREF(args);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	737 return result;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	738 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	739
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	740 PyDoc_STRVAR(starmap_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	741 "starmap(function, sequence) --> starmap object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	742 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	743 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	744 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	745
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	746 static PyTypeObject starmap_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	747 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	748 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	749 "itertools.starmap", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	750 sizeof(starmapobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	751 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	752 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	753 (destructor)starmap_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	754 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	755 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	756 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	757 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	758 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	759 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	760 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	761 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	762 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	763 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	764 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	765 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	766 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	767 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	768 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	769 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	770 starmap_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	771 (traverseproc)starmap_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	772 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	773 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	774 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	775 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	776 (iternextfunc)starmap_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	777 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	778 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	779 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	780 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	781 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	782 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	783 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	784 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	785 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	786 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	787 starmap_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	788 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	789 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	790
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	791
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	792 /* imap object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	793
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	794 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	795 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	796 PyObject *iters;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	797 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	798 } imapobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	799
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	800 static PyTypeObject imap_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	801
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	802 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	803 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	804 {
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	805 PyObject it, iters, *func;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	806 imapobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	807 int numargs, i;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	808
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	809 numargs = PyTuple_Size(args);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	810 if (numargs < 2) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	811 PyErr_SetString(PyExc_TypeError,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	812 "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	813 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	814 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	815
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	816 iters = PyTuple_New(numargs-1);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	817 if (iters == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	818 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	819
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	820 for (i=1 ; i<numargs ; i++) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	821 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	822 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	823 if (it == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	824 Py_DECREF(iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	825 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	826 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	827 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	828 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	829
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	830 /* create imapobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	831 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	832 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	833 Py_DECREF(iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	834 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	835 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	836 lz->iters = iters;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	837 func = PyTuple_GET_ITEM(args, 0);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	838 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	839 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	840
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	841 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	842 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	843
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	844 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	845 imap_dealloc(imapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	846 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	847 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	848 Py_XDECREF(lz->iters);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	849 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	850 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	851 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	852
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	853 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	854 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	855 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	856 int err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	857
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	858 if (lz->iters) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	859 err = visit(lz->iters, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	860 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	861 return err;
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 if (lz->func) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	864 err = visit(lz->func, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	865 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	866 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	867 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	868 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	869 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	870
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	871 /*
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	872 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	873 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	874 the following reasons:
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	875
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	876 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	877 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	878 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	879 infinite iterators like count() and repeat() (for supplying sequential
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	880 or constant arguments to a function).
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	881
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	882 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	883 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	884 should not pass silently by automatically supplying None.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	885
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	886 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	887 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	888 None.
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	889
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	890 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	891 writing: chain(iterable, repeat(None)).
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	892
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	893 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	894 */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	895
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	896 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	897 imap_next(imapobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	898 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	899 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	900 PyObject *argtuple;
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	901 PyObject *result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	902 int numargs, i;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	903
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	904 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	905 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	906 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	907 return NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	908
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	909 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	910 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	911 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	912 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	913 return NULL;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	914 }
27748 59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	915 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	916 }
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	917 if (lz->func == Py_None)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	918 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	919 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	920 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	921 return result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	922 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	923
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	924 PyDoc_STRVAR(imap_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	925 "imap(func, *iterables) --> imap object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	926 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	927 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	928 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	929 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	930 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	931 iterables.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	932
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	933 static PyTypeObject imap_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	934 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	935 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	936 "itertools.imap", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	937 sizeof(imapobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	938 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	939 /* methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	940 (destructor)imap_dealloc, /* tp_dealloc */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	941 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	942 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	943 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	944 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	945 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	946 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	947 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	948 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	949 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	950 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	951 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	952 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	953 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	954 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	955 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	956 Py_TPFLAGS_BASETYPE, /* tp_flags */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	957 imap_doc, /* tp_doc */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	958 (traverseproc)imap_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	959 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	960 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	961 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	962 PyObject_SelfIter, /* tp_iter */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	963 (iternextfunc)imap_next, /* tp_iternext */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	964 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	965 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	966 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	967 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	968 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	969 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	970 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	971 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	972 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	973 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	974 imap_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	975 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	976 };
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
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	979 /* chain object ************************************************************/
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	980
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	981 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	982 PyObject_HEAD
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	983 long tuplesize;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	984 long iternum; /* which iterator is active */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	985 PyObject ittuple; / tuple of iterators */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	986 } chainobject;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	987
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	988 static PyTypeObject chain_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	989
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	990 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	991 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	992 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	993 chainobject *lz;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	994 int tuplesize = PySequence_Length(args);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	995 int i;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	996 PyObject *ittuple;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	997
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	998 /* obtain iterators */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	999 assert(PyTuple_Check(args));
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1000 ittuple = PyTuple_New(tuplesize);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1001 if(ittuple == NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1002 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1003 for (i=0; i < tuplesize; ++i) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1004 PyObject *item = PyTuple_GET_ITEM(args, i);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1005 PyObject *it = PyObject_GetIter(item);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1006 if (it == NULL) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1007 if (PyErr_ExceptionMatches(PyExc_TypeError))
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1008 PyErr_Format(PyExc_TypeError,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1009 "chain argument #%d must support iteration",
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1010 i+1);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1011 Py_DECREF(ittuple);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1012 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1013 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1014 PyTuple_SET_ITEM(ittuple, i, it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1015 }
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1016
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1017 /* create chainobject structure */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1018 lz = (chainobject *)type->tp_alloc(type, 0);
29522 70e649fa3fcd Add missing DECREF. Raymond Hettinger <python@rcn.com> parents: 29242 diff changeset	1019 if (lz == NULL) {
70e649fa3fcd Add missing DECREF. Raymond Hettinger <python@rcn.com> parents: 29242 diff changeset	1020 Py_DECREF(ittuple);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1021 return NULL;
29522 70e649fa3fcd Add missing DECREF. Raymond Hettinger <python@rcn.com> parents: 29242 diff changeset	1022 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1023
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1024 lz->ittuple = ittuple;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1025 lz->iternum = 0;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1026 lz->tuplesize = tuplesize;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1027
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1028 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1029 }
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 static void
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1032 chain_dealloc(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1033 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1034 PyObject_GC_UnTrack(lz);
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1035 Py_XDECREF(lz->ittuple);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1036 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1037 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1038
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1039 static int
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1040 chain_traverse(chainobject lz, visitproc visit, void arg)
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1041 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1042 if (lz->ittuple)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1043 return visit(lz->ittuple, arg);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1044 return 0;
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1045 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1046
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1047 static PyObject *
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1048 chain_next(chainobject *lz)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1049 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1050 PyObject *it;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1051 PyObject *item;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1052
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1053 while (lz->iternum < lz->tuplesize) {
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1054 it = PyTuple_GET_ITEM(lz->ittuple, lz->iternum);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1055 item = PyIter_Next(it);
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1056 if (item != NULL)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1057 return item;
32011 50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1058 if (PyErr_Occurred()) {
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1059 if (PyErr_ExceptionMatches(PyExc_StopIteration))
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1060 PyErr_Clear();
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1061 else
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1062 return NULL;
50708b8f0d28 SF #950057: itertools.chain doesn't "process" exceptions as they occur Raymond Hettinger <python@rcn.com> parents: 30246 diff changeset	1063 }
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1064 lz->iternum++;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1065 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1066 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1067 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1068
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1069 PyDoc_STRVAR(chain_doc,
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1070 "chain(*iterables) --> chain object\n\
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1071 \n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1072 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	1073 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	1074 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	1075
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1076 static PyTypeObject chain_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1077 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1078 0, /* ob_size */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1079 "itertools.chain", /* tp_name */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1080 sizeof(chainobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1081 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1082 /* methods */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1083 (destructor)chain_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1084 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1085 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1086 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1087 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1088 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1089 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1090 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1091 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1092 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1093 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1094 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1095 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1096 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1097 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1098 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	1099 Py_TPFLAGS_BASETYPE, /* tp_flags */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1100 chain_doc, /* tp_doc */
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1101 (traverseproc)chain_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1102 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1103 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1104 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1105 PyObject_SelfIter, /* tp_iter */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1106 (iternextfunc)chain_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1107 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1108 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1109 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1110 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1111 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1112 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1113 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1114 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1115 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1116 0, /* tp_alloc */
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1117 chain_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1118 PyObject_GC_Del, /* tp_free */
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
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1121
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1122 /* ifilter object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1123
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1124 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1125 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1126 PyObject *func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1127 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1128 } ifilterobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1129
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1130 static PyTypeObject ifilter_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1131
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1132 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1133 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	1134 {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1135 PyObject func, seq;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1136 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1137 ifilterobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1138
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1139 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	1140 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1141
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1142 /* Get iterator. */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1143 it = PyObject_GetIter(seq);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1144 if (it == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1145 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1146
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1147 /* create ifilterobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1148 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	1149 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1150 Py_DECREF(it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1151 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1152 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1153 Py_INCREF(func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1154 lz->func = func;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1155 lz->it = it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1156
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1157 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1158 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1159
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1160 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1161 ifilter_dealloc(ifilterobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1162 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1163 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1164 Py_XDECREF(lz->func);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1165 Py_XDECREF(lz->it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1166 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1167 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1168
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1169 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1170 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	1171 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1172 int err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1173
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1174 if (lz->it) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1175 err = visit(lz->it, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1176 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1177 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1178 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1179 if (lz->func) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1180 err = visit(lz->func, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1181 if (err)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1182 return err;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1183 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1184 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1185 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1186
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1187 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1188 ifilter_next(ifilterobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1189 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1190 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1191 PyObject *it = lz->it;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1192 long ok;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1193
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1194 for (;;) {
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1195 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1196 item = (*it->ob_type->tp_iternext)(it);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1197 if (item == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1198 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1199
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1200 if (lz->func == Py_None) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1201 ok = PyObject_IsTrue(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1202 } else {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1203 PyObject *good;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1204 good = PyObject_CallFunctionObjArgs(lz->func,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1205 item, NULL);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1206 if (good == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1207 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1208 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1209 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1210 ok = PyObject_IsTrue(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1211 Py_DECREF(good);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1212 }
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1213 if (ok)
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1214 return item;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1215 Py_DECREF(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1216 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1217 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1218
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1219 PyDoc_STRVAR(ifilter_doc,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1220 "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	1221 \n\
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1222 Return those items of sequence for which function(item) is true.\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1223 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	1224
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1225 static PyTypeObject ifilter_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1226 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1227 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1228 "itertools.ifilter", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1229 sizeof(ifilterobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1230 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1231 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1232 (destructor)ifilter_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1233 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1234 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1235 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1236 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1237 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1238 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1239 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1240 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1241 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1242 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1243 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1244 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1245 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1246 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1247 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	1248 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1249 ifilter_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1250 (traverseproc)ifilter_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1251 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1252 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1253 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1254 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1255 (iternextfunc)ifilter_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1256 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1257 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1258 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1259 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1260 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1261 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1262 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1263 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1264 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1265 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1266 ifilter_new, /* tp_new */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1267 PyObject_GC_Del, /* tp_free */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1268 };
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1269
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1270
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1271 /* ifilterfalse object ************************************************************/
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1272
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1273 typedef struct {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1274 PyObject_HEAD
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1275 PyObject *func;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1276 PyObject *it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1277 } ifilterfalseobject;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1278
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1279 static PyTypeObject ifilterfalse_type;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1280
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1281 static PyObject *
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1282 ifilterfalse_new(PyTypeObject type, PyObject args, PyObject *kwds)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1283 {
27770 2c389a0448a4 Remove unused variable. Guido van Rossum <guido@python.org> parents: 27769 diff changeset	1284 PyObject func, seq;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1285 PyObject *it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1286 ifilterfalseobject *lz;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1287
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1288 if (!PyArg_UnpackTuple(args, "ifilterfalse", 2, 2, &func, &seq))
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1289 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1290
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1291 /* Get iterator. */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1292 it = PyObject_GetIter(seq);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1293 if (it == NULL)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1294 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1295
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1296 /* create ifilterfalseobject structure */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1297 lz = (ifilterfalseobject *)type->tp_alloc(type, 0);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1298 if (lz == NULL) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1299 Py_DECREF(it);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1300 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1301 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1302 Py_INCREF(func);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1303 lz->func = func;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1304 lz->it = it;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1305
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1306 return (PyObject *)lz;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1307 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1308
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1309 static void
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1310 ifilterfalse_dealloc(ifilterfalseobject *lz)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1311 {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1312 PyObject_GC_UnTrack(lz);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1313 Py_XDECREF(lz->func);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1314 Py_XDECREF(lz->it);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1315 lz->ob_type->tp_free(lz);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1316 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1317
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1318 static int
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1319 ifilterfalse_traverse(ifilterfalseobject lz, visitproc visit, void arg)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1320 {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1321 int err;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1322
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1323 if (lz->it) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1324 err = visit(lz->it, arg);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1325 if (err)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1326 return err;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1327 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1328 if (lz->func) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1329 err = visit(lz->func, arg);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1330 if (err)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1331 return err;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1332 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1333 return 0;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1334 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1335
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1336 static PyObject *
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1337 ifilterfalse_next(ifilterfalseobject *lz)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1338 {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1339 PyObject *item;
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1340 PyObject *it = lz->it;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1341 long ok;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1342
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1343 for (;;) {
28117 9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1344 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1345 item = (*it->ob_type->tp_iternext)(it);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1346 if (item == NULL)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1347 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1348
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1349 if (lz->func == Py_None) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1350 ok = PyObject_IsTrue(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1351 } else {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1352 PyObject *good;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1353 good = PyObject_CallFunctionObjArgs(lz->func,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1354 item, NULL);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1355 if (good == NULL) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1356 Py_DECREF(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1357 return NULL;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1358 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1359 ok = PyObject_IsTrue(good);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1360 Py_DECREF(good);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1361 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1362 if (!ok)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1363 return item;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1364 Py_DECREF(item);
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1365 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1366 }
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1367
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1368 PyDoc_STRVAR(ifilterfalse_doc,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1369 "ifilterfalse(function or None, sequence) --> ifilterfalse object\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1370 \n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1371 Return those items of sequence for which function(item) is false.\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1372 If function is None, return the items that are false.");
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1373
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1374 static PyTypeObject ifilterfalse_type = {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1375 PyObject_HEAD_INIT(NULL)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1376 0, /* ob_size */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1377 "itertools.ifilterfalse", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1378 sizeof(ifilterfalseobject), /* tp_basicsize */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1379 0, /* tp_itemsize */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1380 /* methods */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1381 (destructor)ifilterfalse_dealloc, /* tp_dealloc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1382 0, /* tp_print */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1383 0, /* tp_getattr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1384 0, /* tp_setattr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1385 0, /* tp_compare */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1386 0, /* tp_repr */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1387 0, /* tp_as_number */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1388 0, /* tp_as_sequence */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1389 0, /* tp_as_mapping */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1390 0, /* tp_hash */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1391 0, /* tp_call */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1392 0, /* tp_str */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1393 PyObject_GenericGetAttr, /* tp_getattro */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1394 0, /* tp_setattro */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1395 0, /* tp_as_buffer */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1396 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1397 Py_TPFLAGS_BASETYPE, /* tp_flags */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1398 ifilterfalse_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1399 (traverseproc)ifilterfalse_traverse, /* tp_traverse */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1400 0, /* tp_clear */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1401 0, /* tp_richcompare */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1402 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1403 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1404 (iternextfunc)ifilterfalse_next, /* tp_iternext */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1405 0, /* tp_methods */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1406 0, /* tp_members */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1407 0, /* tp_getset */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1408 0, /* tp_base */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1409 0, /* tp_dict */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1410 0, /* tp_descr_get */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1411 0, /* tp_descr_set */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1412 0, /* tp_dictoffset */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1413 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1414 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1415 ifilterfalse_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1416 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1417 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1418
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1419
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1420 /* count object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1421
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1422 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1423 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1424 long cnt;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1425 } countobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1426
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1427 static PyTypeObject count_type;
27570 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 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1430 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	1431 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1432 countobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1433 long cnt = 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1434
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1435 if (!PyArg_ParseTuple(args, "\|l:count", &cnt))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1436 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1437
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1438 /* create countobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1439 lz = (countobject *)PyObject_New(countobject, &count_type);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1440 if (lz == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1441 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1442 lz->cnt = cnt;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1443
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1444 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1445 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1446
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1447 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1448 count_next(countobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1449 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1450 return PyInt_FromLong(lz->cnt++);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1451 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1452
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1453 PyDoc_STRVAR(count_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1454 "count([firstval]) --> count object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1455 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1456 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	1457 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	1458
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1459 static PyTypeObject count_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1460 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1461 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1462 "itertools.count", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1463 sizeof(countobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1464 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1465 /* methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1466 (destructor)PyObject_Del, /* tp_dealloc */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1467 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1468 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1469 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1470 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1471 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1472 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1473 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1474 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1475 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1476 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1477 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1478 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1479 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1480 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1481 Py_TPFLAGS_DEFAULT, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1482 count_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1483 0, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1484 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1485 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1486 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1487 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1488 (iternextfunc)count_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1489 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1490 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1491 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1492 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1493 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1494 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1495 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1496 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1497 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1498 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1499 count_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1500 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1501
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1502
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1503 /* izip object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1504
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1505 #include "Python.h"
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 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1508 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1509 long tuplesize;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1510 PyObject ittuple; / tuple of iterators */
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1511 PyObject *result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1512 } izipobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1513
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1514 static PyTypeObject izip_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1515
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1516 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1517 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	1518 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1519 izipobject *lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1520 int i;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1521 PyObject ittuple; / tuple of iterators */
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1522 PyObject *result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1523 int tuplesize = PySequence_Length(args);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1524
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1525 /* args must be a tuple */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1526 assert(PyTuple_Check(args));
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1527
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1528 /* obtain iterators */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1529 ittuple = PyTuple_New(tuplesize);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1530 if(ittuple == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1531 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1532 for (i=0; i < tuplesize; ++i) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1533 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	1534 PyObject *it = PyObject_GetIter(item);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1535 if (it == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1536 if (PyErr_ExceptionMatches(PyExc_TypeError))
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1537 PyErr_Format(PyExc_TypeError,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1538 "izip argument #%d must support iteration",
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1539 i+1);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1540 Py_DECREF(ittuple);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1541 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1542 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1543 PyTuple_SET_ITEM(ittuple, i, it);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1544 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1545
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1546 /* create a result holder */
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1547 result = PyTuple_New(tuplesize);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1548 if (result == NULL) {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1549 Py_DECREF(ittuple);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1550 return NULL;
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1551 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1552 for (i=0 ; i < tuplesize ; i++) {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1553 Py_INCREF(Py_None);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1554 PyTuple_SET_ITEM(result, i, Py_None);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1555 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1556
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1557 /* create izipobject structure */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1558 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	1559 if (lz == NULL) {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1560 Py_DECREF(ittuple);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1561 Py_DECREF(result);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1562 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1563 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1564 lz->ittuple = ittuple;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1565 lz->tuplesize = tuplesize;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1566 lz->result = result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1567
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1568 return (PyObject *)lz;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1569 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1570
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1571 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1572 izip_dealloc(izipobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1573 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1574 PyObject_GC_UnTrack(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1575 Py_XDECREF(lz->ittuple);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1576 Py_XDECREF(lz->result);
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1577 lz->ob_type->tp_free(lz);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1578 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1579
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1580 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1581 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	1582 {
29535 c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1583 int err;
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1584
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1585 if (lz->ittuple) {
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1586 err = visit(lz->ittuple, arg);
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1587 if (err)
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1588 return err;
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1589 }
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1590 if (lz->result) {
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1591 err = visit(lz->result, arg);
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1592 if (err)
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1593 return err;
c64687cd5b37 Minor updates: Raymond Hettinger <python@rcn.com> parents: 29522 diff changeset	1594 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1595 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1596 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1597
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1598 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1599 izip_next(izipobject *lz)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1600 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1601 int i;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1602 long tuplesize = lz->tuplesize;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1603 PyObject *result = lz->result;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1604 PyObject *it;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1605 PyObject *item;
30241 3fb3747ce304 For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30239 diff changeset	1606 PyObject *olditem;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1607
30246 97f3d66558fb * Add news item for the previous bugfix Raymond Hettinger <python@rcn.com> parents: 30241 diff changeset	1608 if (tuplesize == 0)
97f3d66558fb * Add news item for the previous bugfix Raymond Hettinger <python@rcn.com> parents: 30241 diff changeset	1609 return NULL;
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1610 if (result->ob_refcnt == 1) {
30239 e83575bb29da SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	1611 Py_INCREF(result);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1612 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	1613 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	1614 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1615 item = (*it->ob_type->tp_iternext)(it);
30239 e83575bb29da SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	1616 if (item == NULL) {
e83575bb29da SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	1617 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	1618 return NULL;
30239 e83575bb29da SF bug #793826: using itertools.izip to mutate tuples Raymond Hettinger <python@rcn.com> parents: 29850 diff changeset	1619 }
30241 3fb3747ce304 For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30239 diff changeset	1620 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	1621 PyTuple_SET_ITEM(result, i, item);
30241 3fb3747ce304 For safety, replace a tuple entry before decreffing it. Raymond Hettinger <python@rcn.com> parents: 30239 diff changeset	1622 Py_DECREF(olditem);
27747 941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1623 }
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1624 } else {
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1625 result = PyTuple_New(tuplesize);
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1626 if (result == NULL)
941de791b17a SF bug #681003: itertools issues Raymond Hettinger <python@rcn.com> parents: 27570 diff changeset	1627 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	1628 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	1629 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	1630 assert(PyIter_Check(it));
9b4eed4bb3e2 Several of the tools can make direct calls the inner iterators. Raymond Hettinger <python@rcn.com> parents: 28024 diff changeset	1631 item = (*it->ob_type->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	1632 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	1633 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	1634 return NULL;
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1635 }
59ae41e04ffb * Eliminated tuple re-use in imap(). Doing it correctly made the code Raymond Hettinger <python@rcn.com> parents: 27747 diff changeset	1636 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	1637 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1638 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1639 return result;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1640 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1641
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1642 PyDoc_STRVAR(izip_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1643 "izip(iter1 [,iter2 [...]]) --> izip object\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1644 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1645 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	1646 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	1647 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	1648 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	1649 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	1650 a list.");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1651
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1652 static PyTypeObject izip_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1653 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1654 0, /* ob_size */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1655 "itertools.izip", /* tp_name */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1656 sizeof(izipobject), /* tp_basicsize */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1657 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1658 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1659 (destructor)izip_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1660 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1661 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1662 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1663 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1664 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1665 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1666 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1667 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1668 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1669 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1670 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1671 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1672 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1673 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1674 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	1675 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1676 izip_doc, /* tp_doc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1677 (traverseproc)izip_traverse, /* tp_traverse */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1678 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1679 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1680 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1681 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1682 (iternextfunc)izip_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1683 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1684 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1685 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1686 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1687 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1688 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1689 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1690 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1691 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1692 0, /* tp_alloc */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1693 izip_new, /* tp_new */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1694 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1695 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1696
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1697
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1698 /* repeat object ************************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1699
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1700 typedef struct {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1701 PyObject_HEAD
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1702 PyObject *element;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1703 long cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1704 } repeatobject;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1705
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1706 static PyTypeObject repeat_type;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1707
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1708 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1709 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	1710 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1711 repeatobject *ro;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1712 PyObject *element;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1713 long cnt = -1;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1714
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1715 if (!PyArg_ParseTuple(args, "O\|l:repeat", &element, &cnt))
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1716 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1717
28945 001a90555780 * Added a substantial number of edge case and argument tests for Raymond Hettinger <python@rcn.com> parents: 28930 diff changeset	1718 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	1719 cnt = 0;
001a90555780 * Added a substantial number of edge case and argument tests for Raymond Hettinger <python@rcn.com> parents: 28930 diff changeset	1720
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1721 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	1722 if (ro == NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1723 return NULL;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1724 Py_INCREF(element);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1725 ro->element = element;
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1726 ro->cnt = cnt;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1727 return (PyObject *)ro;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1728 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1729
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1730 static void
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1731 repeat_dealloc(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1732 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1733 PyObject_GC_UnTrack(ro);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1734 Py_XDECREF(ro->element);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1735 ro->ob_type->tp_free(ro);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1736 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1737
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1738 static int
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1739 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	1740 {
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1741 if (ro->element)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1742 return visit(ro->element, arg);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1743 return 0;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1744 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1745
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1746 static PyObject *
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1747 repeat_next(repeatobject *ro)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1748 {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1749 if (ro->cnt == 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1750 return NULL;
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1751 if (ro->cnt > 0)
61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1752 ro->cnt--;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1753 Py_INCREF(ro->element);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1754 return ro->element;
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1755 }
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1756
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1757 PyDoc_STRVAR(repeat_doc,
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1758 "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	1759 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	1760 endlessly.");
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1761
29850 45dc33433323 SF patch #770521: make itertools type declarations static Raymond Hettinger <python@rcn.com> parents: 29535 diff changeset	1762 static PyTypeObject repeat_type = {
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1763 PyObject_HEAD_INIT(NULL)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1764 0, /* ob_size */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1765 "itertools.repeat", /* tp_name */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1766 sizeof(repeatobject), /* tp_basicsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1767 0, /* tp_itemsize */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1768 /* methods */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1769 (destructor)repeat_dealloc, /* tp_dealloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1770 0, /* tp_print */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1771 0, /* tp_getattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1772 0, /* tp_setattr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1773 0, /* tp_compare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1774 0, /* tp_repr */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1775 0, /* tp_as_number */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1776 0, /* tp_as_sequence */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1777 0, /* tp_as_mapping */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1778 0, /* tp_hash */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1779 0, /* tp_call */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1780 0, /* tp_str */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1781 PyObject_GenericGetAttr, /* tp_getattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1782 0, /* tp_setattro */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1783 0, /* tp_as_buffer */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1784 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	1785 Py_TPFLAGS_BASETYPE, /* tp_flags */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1786 repeat_doc, /* tp_doc */
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1787 (traverseproc)repeat_traverse, /* tp_traverse */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1788 0, /* tp_clear */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1789 0, /* tp_richcompare */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1790 0, /* tp_weaklistoffset */
28320 fd5ef7003469 Renamed PyObject_GenericGetIter to PyObject_SelfIter Raymond Hettinger <python@rcn.com> parents: 28308 diff changeset	1791 PyObject_SelfIter, /* tp_iter */
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1792 (iternextfunc)repeat_next, /* tp_iternext */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1793 0, /* tp_methods */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1794 0, /* tp_members */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1795 0, /* tp_getset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1796 0, /* tp_base */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1797 0, /* tp_dict */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1798 0, /* tp_descr_get */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1799 0, /* tp_descr_set */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1800 0, /* tp_dictoffset */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1801 0, /* tp_init */
29242 d72ad07c739e PyType_GenericAlloc is inherited from object. Raymond Hettinger <python@rcn.com> parents: 29225 diff changeset	1802 0, /* tp_alloc */
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1803 repeat_new, /* tp_new */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1804 PyObject_GC_Del, /* tp_free */
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1805 };
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1806
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1807
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1808 /* module level code ********************************************************/
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1809
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1810 PyDoc_STRVAR(module_doc,
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1811 "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	1812 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1813 Infinite iterators:\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1814 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	1815 cycle(p) --> p0, p1, ... plast, p0, p1, ...\n\
28601 58db5fbd9c28 Fix docstring typo Andrew M. Kuchling <amk@amk.ca> parents: 28320 diff changeset	1816 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	1817 \n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1818 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	1819 izip(p, q, ...) --> (p[0], q[0]), (p[1], q[1]), ... \n\
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1820 ifilter(pred, seq) --> elements of seq where pred(elem) is True\n\
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1821 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	1822 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	1823 seq[start:stop:step]\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1824 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	1825 starmap(fun, seq) --> fun(seq[0]), fun(seq[1]), ...\n\
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1826 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	1827 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	1828 dropwhile(pred, seq) --> seq[n], seq[n+1], starting when pred fails\n\
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1829 ");
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1830
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1831
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1832 PyMODINIT_FUNC
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1833 inititertools(void)
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1834 {
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1835 int i;
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1836 PyObject *m;
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1837 char *name;
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1838 PyTypeObject *typelist[] = {
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1839 &cycle_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1840 &dropwhile_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1841 &takewhile_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1842 &islice_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1843 &starmap_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1844 &imap_type,
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1845 &chain_type,
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1846 &ifilter_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1847 &ifilterfalse_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1848 &count_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1849 &izip_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1850 &repeat_type,
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1851 NULL
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1852 };
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1853
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1854 m = Py_InitModule3("itertools", NULL, module_doc);
ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1855
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1856 for (i=0 ; typelist[i] != NULL ; i++) {
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1857 if (PyType_Ready(typelist[i]) < 0)
dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1858 return;
29225 f3f306457b43 Fixed dotted name assertion. Raymond Hettinger <python@rcn.com> parents: 28945 diff changeset	1859 name = strchr(typelist[i]->tp_name, '.');
28024 61fa6b55c1de User requested changes to the itertools module. Raymond Hettinger <python@rcn.com> parents: 27770 diff changeset	1860 assert (name != NULL);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1861 Py_INCREF(typelist[i]);
29225 f3f306457b43 Fixed dotted name assertion. Raymond Hettinger <python@rcn.com> parents: 28945 diff changeset	1862 PyModule_AddObject(m, name+1, (PyObject *)typelist[i]);
27769 dc5e1d918055 C Code: Raymond Hettinger <python@rcn.com> parents: 27748 diff changeset	1863 }
27570 ba57c3daf6e9 Move itertools module from the sandbox and into production. Raymond Hettinger <python@rcn.com> parents: diff changeset	1864 }

Mercurial > cpython-lru_cache-weakref

annotate Modules/itertoolsmodule.c @ 33091:48100a01ac7c 2.3