Lines Matching refs:lz
324 groupby_reduce(groupbyobject *lz, PyObject *Py_UNUSED(ignored)) in groupby_reduce() argument
330 if (lz->tgtkey && lz->currkey && lz->currvalue) in groupby_reduce()
331 value = Py_BuildValue("O(OO)(OOO)", Py_TYPE(lz), in groupby_reduce()
332 lz->it, lz->keyfunc, lz->currkey, lz->currvalue, lz->tgtkey); in groupby_reduce()
334 value = Py_BuildValue("O(OO)", Py_TYPE(lz), in groupby_reduce()
335 lz->it, lz->keyfunc); in groupby_reduce()
343 groupby_setstate(groupbyobject *lz, PyObject *state) in groupby_setstate() argument
354 Py_XSETREF(lz->currkey, currkey); in groupby_setstate()
356 Py_XSETREF(lz->currvalue, currvalue); in groupby_setstate()
358 Py_XSETREF(lz->tgtkey, tgtkey); in groupby_setstate()
505 _grouper_reduce(_grouperobject *lz, PyObject *Py_UNUSED(ignored)) in _grouper_reduce() argument
508 if (((groupbyobject *)lz->parent)->currgrouper != lz) { in _grouper_reduce()
511 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->parent, lz->tgtkey); in _grouper_reduce()
1098 cycleobject *lz; in itertools_cycle_impl() local
1112 lz = (cycleobject *)type->tp_alloc(type, 0); in itertools_cycle_impl()
1113 if (lz == NULL) { in itertools_cycle_impl()
1118 lz->it = it; in itertools_cycle_impl()
1119 lz->saved = saved; in itertools_cycle_impl()
1120 lz->index = 0; in itertools_cycle_impl()
1121 lz->firstpass = 0; in itertools_cycle_impl()
1123 return (PyObject *)lz; in itertools_cycle_impl()
1127 cycle_dealloc(cycleobject *lz) in cycle_dealloc() argument
1129 PyObject_GC_UnTrack(lz); in cycle_dealloc()
1130 Py_XDECREF(lz->it); in cycle_dealloc()
1131 Py_XDECREF(lz->saved); in cycle_dealloc()
1132 Py_TYPE(lz)->tp_free(lz); in cycle_dealloc()
1136 cycle_traverse(cycleobject *lz, visitproc visit, void *arg) in cycle_traverse() argument
1138 Py_VISIT(lz->it); in cycle_traverse()
1139 Py_VISIT(lz->saved); in cycle_traverse()
1144 cycle_next(cycleobject *lz) in cycle_next() argument
1148 if (lz->it != NULL) { in cycle_next()
1149 item = PyIter_Next(lz->it); in cycle_next()
1151 if (lz->firstpass) in cycle_next()
1153 if (PyList_Append(lz->saved, item)) { in cycle_next()
1162 Py_CLEAR(lz->it); in cycle_next()
1164 if (PyList_GET_SIZE(lz->saved) == 0) in cycle_next()
1166 item = PyList_GET_ITEM(lz->saved, lz->index); in cycle_next()
1167 lz->index++; in cycle_next()
1168 if (lz->index >= PyList_GET_SIZE(lz->saved)) in cycle_next()
1169 lz->index = 0; in cycle_next()
1175 cycle_reduce(cycleobject *lz, PyObject *Py_UNUSED(ignored)) in cycle_reduce() argument
1178 if (lz->it == NULL) { in cycle_reduce()
1179 PyObject *it = PyObject_GetIter(lz->saved); in cycle_reduce()
1182 if (lz->index != 0) { in cycle_reduce()
1185 "n", lz->index); in cycle_reduce()
1192 return Py_BuildValue("O(N)(OO)", Py_TYPE(lz), it, lz->saved, Py_True); in cycle_reduce()
1194 return Py_BuildValue("O(O)(OO)", Py_TYPE(lz), lz->it, lz->saved, in cycle_reduce()
1195 lz->firstpass ? Py_True : Py_False); in cycle_reduce()
1199 cycle_setstate(cycleobject *lz, PyObject *state) in cycle_setstate() argument
1211 Py_XSETREF(lz->saved, saved); in cycle_setstate()
1212 lz->firstpass = firstpass != 0; in cycle_setstate()
1213 lz->index = 0; in cycle_setstate()
1295 dropwhileobject *lz; in itertools_dropwhile_impl() local
1303 lz = (dropwhileobject *)type->tp_alloc(type, 0); in itertools_dropwhile_impl()
1304 if (lz == NULL) { in itertools_dropwhile_impl()
1309 lz->func = func; in itertools_dropwhile_impl()
1310 lz->it = it; in itertools_dropwhile_impl()
1311 lz->start = 0; in itertools_dropwhile_impl()
1313 return (PyObject *)lz; in itertools_dropwhile_impl()
1317 dropwhile_dealloc(dropwhileobject *lz) in dropwhile_dealloc() argument
1319 PyObject_GC_UnTrack(lz); in dropwhile_dealloc()
1320 Py_XDECREF(lz->func); in dropwhile_dealloc()
1321 Py_XDECREF(lz->it); in dropwhile_dealloc()
1322 Py_TYPE(lz)->tp_free(lz); in dropwhile_dealloc()
1326 dropwhile_traverse(dropwhileobject *lz, visitproc visit, void *arg) in dropwhile_traverse() argument
1328 Py_VISIT(lz->it); in dropwhile_traverse()
1329 Py_VISIT(lz->func); in dropwhile_traverse()
1334 dropwhile_next(dropwhileobject *lz) in dropwhile_next() argument
1337 PyObject *it = lz->it; in dropwhile_next()
1346 if (lz->start == 1) in dropwhile_next()
1349 good = PyObject_CallOneArg(lz->func, item); in dropwhile_next()
1357 lz->start = 1; in dropwhile_next()
1367 dropwhile_reduce(dropwhileobject *lz, PyObject *Py_UNUSED(ignored)) in dropwhile_reduce() argument
1369 return Py_BuildValue("O(OO)l", Py_TYPE(lz), lz->func, lz->it, lz->start); in dropwhile_reduce()
1373 dropwhile_setstate(dropwhileobject *lz, PyObject *state) in dropwhile_setstate() argument
1378 lz->start = start; in dropwhile_setstate()
1458 takewhileobject *lz; in itertools_takewhile_impl() local
1466 lz = (takewhileobject *)type->tp_alloc(type, 0); in itertools_takewhile_impl()
1467 if (lz == NULL) { in itertools_takewhile_impl()
1472 lz->func = func; in itertools_takewhile_impl()
1473 lz->it = it; in itertools_takewhile_impl()
1474 lz->stop = 0; in itertools_takewhile_impl()
1476 return (PyObject *)lz; in itertools_takewhile_impl()
1480 takewhile_dealloc(takewhileobject *lz) in takewhile_dealloc() argument
1482 PyObject_GC_UnTrack(lz); in takewhile_dealloc()
1483 Py_XDECREF(lz->func); in takewhile_dealloc()
1484 Py_XDECREF(lz->it); in takewhile_dealloc()
1485 Py_TYPE(lz)->tp_free(lz); in takewhile_dealloc()
1489 takewhile_traverse(takewhileobject *lz, visitproc visit, void *arg) in takewhile_traverse() argument
1491 Py_VISIT(lz->it); in takewhile_traverse()
1492 Py_VISIT(lz->func); in takewhile_traverse()
1497 takewhile_next(takewhileobject *lz) in takewhile_next() argument
1500 PyObject *it = lz->it; in takewhile_next()
1503 if (lz->stop == 1) in takewhile_next()
1510 good = PyObject_CallOneArg(lz->func, item); in takewhile_next()
1521 lz->stop = 1; in takewhile_next()
1526 takewhile_reduce(takewhileobject *lz, PyObject *Py_UNUSED(ignored)) in takewhile_reduce() argument
1528 return Py_BuildValue("O(OO)l", Py_TYPE(lz), lz->func, lz->it, lz->stop); in takewhile_reduce()
1532 takewhile_reduce_setstate(takewhileobject *lz, PyObject *state) in takewhile_reduce_setstate() argument
1538 lz->stop = stop; in takewhile_reduce_setstate()
1615 isliceobject *lz; in islice_new() local
1678 lz = (isliceobject *)type->tp_alloc(type, 0); in islice_new()
1679 if (lz == NULL) { in islice_new()
1683 lz->it = it; in islice_new()
1684 lz->next = start; in islice_new()
1685 lz->stop = stop; in islice_new()
1686 lz->step = step; in islice_new()
1687 lz->cnt = 0L; in islice_new()
1689 return (PyObject *)lz; in islice_new()
1693 islice_dealloc(isliceobject *lz) in islice_dealloc() argument
1695 PyObject_GC_UnTrack(lz); in islice_dealloc()
1696 Py_XDECREF(lz->it); in islice_dealloc()
1697 Py_TYPE(lz)->tp_free(lz); in islice_dealloc()
1701 islice_traverse(isliceobject *lz, visitproc visit, void *arg) in islice_traverse() argument
1703 Py_VISIT(lz->it); in islice_traverse()
1708 islice_next(isliceobject *lz) in islice_next() argument
1711 PyObject *it = lz->it; in islice_next()
1712 Py_ssize_t stop = lz->stop; in islice_next()
1720 while (lz->cnt < lz->next) { in islice_next()
1725 lz->cnt++; in islice_next()
1727 if (stop != -1 && lz->cnt >= stop) in islice_next()
1732 lz->cnt++; in islice_next()
1733 oldnext = lz->next; in islice_next()
1736 lz->next += (size_t)lz->step; in islice_next()
1737 if (lz->next < oldnext || (stop != -1 && lz->next > stop)) in islice_next()
1738 lz->next = stop; in islice_next()
1742 Py_CLEAR(lz->it); in islice_next()
1747 islice_reduce(isliceobject *lz, PyObject *Py_UNUSED(ignored)) in islice_reduce() argument
1754 if (lz->it == NULL) { in islice_reduce()
1764 return Py_BuildValue("O(Nn)n", Py_TYPE(lz), empty_it, 0, 0); in islice_reduce()
1766 if (lz->stop == -1) { in islice_reduce()
1770 stop = PyLong_FromSsize_t(lz->stop); in islice_reduce()
1774 return Py_BuildValue("O(OnNn)n", Py_TYPE(lz), in islice_reduce()
1775 lz->it, lz->next, stop, lz->step, in islice_reduce()
1776 lz->cnt); in islice_reduce()
1780 islice_setstate(isliceobject *lz, PyObject *state) in islice_setstate() argument
1786 lz->cnt = cnt; in islice_setstate()
1876 starmapobject *lz; in itertools_starmap_impl() local
1884 lz = (starmapobject *)type->tp_alloc(type, 0); in itertools_starmap_impl()
1885 if (lz == NULL) { in itertools_starmap_impl()
1890 lz->func = func; in itertools_starmap_impl()
1891 lz->it = it; in itertools_starmap_impl()
1893 return (PyObject *)lz; in itertools_starmap_impl()
1897 starmap_dealloc(starmapobject *lz) in starmap_dealloc() argument
1899 PyObject_GC_UnTrack(lz); in starmap_dealloc()
1900 Py_XDECREF(lz->func); in starmap_dealloc()
1901 Py_XDECREF(lz->it); in starmap_dealloc()
1902 Py_TYPE(lz)->tp_free(lz); in starmap_dealloc()
1906 starmap_traverse(starmapobject *lz, visitproc visit, void *arg) in starmap_traverse() argument
1908 Py_VISIT(lz->it); in starmap_traverse()
1909 Py_VISIT(lz->func); in starmap_traverse()
1914 starmap_next(starmapobject *lz) in starmap_next() argument
1918 PyObject *it = lz->it; in starmap_next()
1930 result = PyObject_Call(lz->func, args, NULL); in starmap_next()
1936 starmap_reduce(starmapobject *lz, PyObject *Py_UNUSED(ignored)) in starmap_reduce() argument
1939 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->func, lz->it); in starmap_reduce()
2006 chainobject *lz; in chain_new_internal() local
2008 lz = (chainobject *)type->tp_alloc(type, 0); in chain_new_internal()
2009 if (lz == NULL) { in chain_new_internal()
2014 lz->source = source; in chain_new_internal()
2015 lz->active = NULL; in chain_new_internal()
2016 return (PyObject *)lz; in chain_new_internal()
2056 chain_dealloc(chainobject *lz) in chain_dealloc() argument
2058 PyObject_GC_UnTrack(lz); in chain_dealloc()
2059 Py_XDECREF(lz->active); in chain_dealloc()
2060 Py_XDECREF(lz->source); in chain_dealloc()
2061 Py_TYPE(lz)->tp_free(lz); in chain_dealloc()
2065 chain_traverse(chainobject *lz, visitproc visit, void *arg) in chain_traverse() argument
2067 Py_VISIT(lz->source); in chain_traverse()
2068 Py_VISIT(lz->active); in chain_traverse()
2073 chain_next(chainobject *lz) in chain_next() argument
2080 while (lz->source != NULL) { in chain_next()
2081 if (lz->active == NULL) { in chain_next()
2082 PyObject *iterable = PyIter_Next(lz->source); in chain_next()
2084 Py_CLEAR(lz->source); in chain_next()
2087 lz->active = PyObject_GetIter(iterable); in chain_next()
2089 if (lz->active == NULL) { in chain_next()
2090 Py_CLEAR(lz->source); in chain_next()
2094 item = (*Py_TYPE(lz->active)->tp_iternext)(lz->active); in chain_next()
2104 Py_CLEAR(lz->active); in chain_next()
2111 chain_reduce(chainobject *lz, PyObject *Py_UNUSED(ignored)) in chain_reduce() argument
2113 if (lz->source) { in chain_reduce()
2118 if (lz->active) { in chain_reduce()
2119 return Py_BuildValue("O()(OO)", Py_TYPE(lz), lz->source, lz->active); in chain_reduce()
2121 return Py_BuildValue("O()(O)", Py_TYPE(lz), lz->source); in chain_reduce()
2124 return Py_BuildValue("O()", Py_TYPE(lz)); /* exhausted */ in chain_reduce()
2130 chain_setstate(chainobject *lz, PyObject *state) in chain_setstate() argument
2147 Py_XSETREF(lz->source, source); in chain_setstate()
2149 Py_XSETREF(lz->active, active); in chain_setstate()
2231 productobject *lz; in product_new() local
2293 lz = (productobject *)type->tp_alloc(type, 0); in product_new()
2294 if (lz == NULL) in product_new()
2297 lz->pools = pools; in product_new()
2298 lz->indices = indices; in product_new()
2299 lz->result = NULL; in product_new()
2300 lz->stopped = 0; in product_new()
2302 return (PyObject *)lz; in product_new()
2312 product_dealloc(productobject *lz) in product_dealloc() argument
2314 PyObject_GC_UnTrack(lz); in product_dealloc()
2315 Py_XDECREF(lz->pools); in product_dealloc()
2316 Py_XDECREF(lz->result); in product_dealloc()
2317 if (lz->indices != NULL) in product_dealloc()
2318 PyMem_Free(lz->indices); in product_dealloc()
2319 Py_TYPE(lz)->tp_free(lz); in product_dealloc()
2323 product_sizeof(productobject *lz, void *unused) in product_sizeof() argument
2327 res = _PyObject_SIZE(Py_TYPE(lz)); in product_sizeof()
2328 res += PyTuple_GET_SIZE(lz->pools) * sizeof(Py_ssize_t); in product_sizeof()
2335 product_traverse(productobject *lz, visitproc visit, void *arg) in product_traverse() argument
2337 Py_VISIT(lz->pools); in product_traverse()
2338 Py_VISIT(lz->result); in product_traverse()
2343 product_next(productobject *lz) in product_next() argument
2348 PyObject *pools = lz->pools; in product_next()
2349 PyObject *result = lz->result; in product_next()
2353 if (lz->stopped) in product_next()
2362 lz->result = result; in product_next()
2372 Py_ssize_t *indices = lz->indices; in product_next()
2380 lz->result = result; in product_next()
2425 lz->stopped = 1; in product_next()
2430 product_reduce(productobject *lz, PyObject *Py_UNUSED(ignored)) in product_reduce() argument
2432 if (lz->stopped) { in product_reduce()
2433 return Py_BuildValue("O(())", Py_TYPE(lz)); in product_reduce()
2434 } else if (lz->result == NULL) { in product_reduce()
2435 return Py_BuildValue("OO", Py_TYPE(lz), lz->pools); in product_reduce()
2443 n = PyTuple_GET_SIZE(lz->pools); in product_reduce()
2448 PyObject* index = PyLong_FromSsize_t(lz->indices[i]); in product_reduce()
2455 return Py_BuildValue("OON", Py_TYPE(lz), lz->pools, indices); in product_reduce()
2460 product_setstate(productobject *lz, PyObject *state) in product_setstate() argument
2465 n = PyTuple_GET_SIZE(lz->pools); in product_setstate()
2478 pool = PyTuple_GET_ITEM(lz->pools, i); in product_setstate()
2481 lz->stopped = 1; in product_setstate()
2489 lz->indices[i] = index; in product_setstate()
2496 PyObject *pool = PyTuple_GET_ITEM(lz->pools, i); in product_setstate()
2497 PyObject *element = PyTuple_GET_ITEM(pool, lz->indices[i]); in product_setstate()
2501 Py_XSETREF(lz->result, result); in product_setstate()
2761 combinations_reduce(combinationsobject *lz, PyObject *Py_UNUSED(ignored)) in combinations_reduce() argument
2763 if (lz->result == NULL) { in combinations_reduce()
2764 return Py_BuildValue("O(On)", Py_TYPE(lz), lz->pool, lz->r); in combinations_reduce()
2765 } else if (lz->stopped) { in combinations_reduce()
2766 return Py_BuildValue("O(()n)", Py_TYPE(lz), lz->r); in combinations_reduce()
2772 indices = PyTuple_New(lz->r); in combinations_reduce()
2775 for (i=0; i<lz->r; i++) in combinations_reduce()
2777 PyObject* index = PyLong_FromSsize_t(lz->indices[i]); in combinations_reduce()
2785 return Py_BuildValue("O(On)N", Py_TYPE(lz), lz->pool, lz->r, indices); in combinations_reduce()
2790 combinations_setstate(combinationsobject *lz, PyObject *state) in combinations_setstate() argument
2794 Py_ssize_t n = PyTuple_GET_SIZE(lz->pool); in combinations_setstate()
2796 if (!PyTuple_Check(state) || PyTuple_GET_SIZE(state) != lz->r) { in combinations_setstate()
2801 for (i=0; i<lz->r; i++) { in combinations_setstate()
2808 max = i + n - lz->r; in combinations_setstate()
2814 lz->indices[i] = index; in combinations_setstate()
2817 result = PyTuple_New(lz->r); in combinations_setstate()
2820 for (i=0; i<lz->r; i++) { in combinations_setstate()
2821 PyObject *element = PyTuple_GET_ITEM(lz->pool, lz->indices[i]); in combinations_setstate()
2826 Py_XSETREF(lz->result, result); in combinations_setstate()
3092 cwr_reduce(cwrobject *lz, PyObject *Py_UNUSED(ignored)) in cwr_reduce() argument
3094 if (lz->result == NULL) { in cwr_reduce()
3095 return Py_BuildValue("O(On)", Py_TYPE(lz), lz->pool, lz->r); in cwr_reduce()
3096 } else if (lz->stopped) { in cwr_reduce()
3097 return Py_BuildValue("O(()n)", Py_TYPE(lz), lz->r); in cwr_reduce()
3103 indices = PyTuple_New(lz->r); in cwr_reduce()
3106 for (i=0; i<lz->r; i++) { in cwr_reduce()
3107 PyObject* index = PyLong_FromSsize_t(lz->indices[i]); in cwr_reduce()
3115 return Py_BuildValue("O(On)N", Py_TYPE(lz), lz->pool, lz->r, indices); in cwr_reduce()
3120 cwr_setstate(cwrobject *lz, PyObject *state) in cwr_setstate() argument
3125 if (!PyTuple_Check(state) || PyTuple_GET_SIZE(state) != lz->r) in cwr_setstate()
3131 n = PyTuple_GET_SIZE(lz->pool); in cwr_setstate()
3132 for (i=0; i<lz->r; i++) { in cwr_setstate()
3143 lz->indices[i] = index; in cwr_setstate()
3145 result = PyTuple_New(lz->r); in cwr_setstate()
3148 for (i=0; i<lz->r; i++) { in cwr_setstate()
3149 PyObject *element = PyTuple_GET_ITEM(lz->pool, lz->indices[i]); in cwr_setstate()
3153 Py_XSETREF(lz->result, result); in cwr_setstate()
3630 accumulateobject *lz; in itertools_accumulate_impl() local
3638 lz = (accumulateobject *)type->tp_alloc(type, 0); in itertools_accumulate_impl()
3639 if (lz == NULL) { in itertools_accumulate_impl()
3646 lz->binop = binop; in itertools_accumulate_impl()
3648 lz->total = NULL; in itertools_accumulate_impl()
3649 lz->it = it; in itertools_accumulate_impl()
3651 lz->initial = initial; in itertools_accumulate_impl()
3652 return (PyObject *)lz; in itertools_accumulate_impl()
3656 accumulate_dealloc(accumulateobject *lz) in accumulate_dealloc() argument
3658 PyObject_GC_UnTrack(lz); in accumulate_dealloc()
3659 Py_XDECREF(lz->binop); in accumulate_dealloc()
3660 Py_XDECREF(lz->total); in accumulate_dealloc()
3661 Py_XDECREF(lz->it); in accumulate_dealloc()
3662 Py_XDECREF(lz->initial); in accumulate_dealloc()
3663 Py_TYPE(lz)->tp_free(lz); in accumulate_dealloc()
3667 accumulate_traverse(accumulateobject *lz, visitproc visit, void *arg) in accumulate_traverse() argument
3669 Py_VISIT(lz->binop); in accumulate_traverse()
3670 Py_VISIT(lz->it); in accumulate_traverse()
3671 Py_VISIT(lz->total); in accumulate_traverse()
3672 Py_VISIT(lz->initial); in accumulate_traverse()
3677 accumulate_next(accumulateobject *lz) in accumulate_next() argument
3681 if (lz->initial != Py_None) { in accumulate_next()
3682 lz->total = lz->initial; in accumulate_next()
3684 lz->initial = Py_None; in accumulate_next()
3685 Py_INCREF(lz->total); in accumulate_next()
3686 return lz->total; in accumulate_next()
3688 val = (*Py_TYPE(lz->it)->tp_iternext)(lz->it); in accumulate_next()
3692 if (lz->total == NULL) { in accumulate_next()
3694 lz->total = val; in accumulate_next()
3695 return lz->total; in accumulate_next()
3698 if (lz->binop == NULL) in accumulate_next()
3699 newtotal = PyNumber_Add(lz->total, val); in accumulate_next()
3701 newtotal = PyObject_CallFunctionObjArgs(lz->binop, lz->total, val, NULL); in accumulate_next()
3707 Py_SETREF(lz->total, newtotal); in accumulate_next()
3712 accumulate_reduce(accumulateobject *lz, PyObject *Py_UNUSED(ignored)) in accumulate_reduce() argument
3714 if (lz->initial != Py_None) { in accumulate_reduce()
3717 assert(lz->total == NULL); in accumulate_reduce()
3721 lz->initial, lz->it); in accumulate_reduce()
3724 return Py_BuildValue("O(NO)O", Py_TYPE(lz), in accumulate_reduce()
3725 it, lz->binop?lz->binop:Py_None, Py_None); in accumulate_reduce()
3727 if (lz->total == Py_None) { in accumulate_reduce()
3735 lz->total, lz->it); in accumulate_reduce()
3738 it = PyObject_CallFunction((PyObject *)Py_TYPE(lz), "NO", in accumulate_reduce()
3739 it, lz->binop ? lz->binop : Py_None); in accumulate_reduce()
3744 return Py_BuildValue("O(OO)O", Py_TYPE(lz), in accumulate_reduce()
3745 lz->it, lz->binop?lz->binop:Py_None, in accumulate_reduce()
3746 lz->total?lz->total:Py_None); in accumulate_reduce()
3750 accumulate_setstate(accumulateobject *lz, PyObject *state) in accumulate_setstate() argument
3753 Py_XSETREF(lz->total, state); in accumulate_setstate()
3841 compressobject *lz; in itertools_compress_impl() local
3851 lz = (compressobject *)type->tp_alloc(type, 0); in itertools_compress_impl()
3852 if (lz == NULL) in itertools_compress_impl()
3854 lz->data = data; in itertools_compress_impl()
3855 lz->selectors = selectors; in itertools_compress_impl()
3856 return (PyObject *)lz; in itertools_compress_impl()
3865 compress_dealloc(compressobject *lz) in compress_dealloc() argument
3867 PyObject_GC_UnTrack(lz); in compress_dealloc()
3868 Py_XDECREF(lz->data); in compress_dealloc()
3869 Py_XDECREF(lz->selectors); in compress_dealloc()
3870 Py_TYPE(lz)->tp_free(lz); in compress_dealloc()
3874 compress_traverse(compressobject *lz, visitproc visit, void *arg) in compress_traverse() argument
3876 Py_VISIT(lz->data); in compress_traverse()
3877 Py_VISIT(lz->selectors); in compress_traverse()
3882 compress_next(compressobject *lz) in compress_next() argument
3884 PyObject *data = lz->data, *selectors = lz->selectors; in compress_next()
3918 compress_reduce(compressobject *lz, PyObject *Py_UNUSED(ignored)) in compress_reduce() argument
3920 return Py_BuildValue("O(OO)", Py_TYPE(lz), in compress_reduce()
3921 lz->data, lz->selectors); in compress_reduce()
3999 filterfalseobject *lz; in itertools_filterfalse_impl() local
4007 lz = (filterfalseobject *)type->tp_alloc(type, 0); in itertools_filterfalse_impl()
4008 if (lz == NULL) { in itertools_filterfalse_impl()
4013 lz->func = func; in itertools_filterfalse_impl()
4014 lz->it = it; in itertools_filterfalse_impl()
4016 return (PyObject *)lz; in itertools_filterfalse_impl()
4020 filterfalse_dealloc(filterfalseobject *lz) in filterfalse_dealloc() argument
4022 PyObject_GC_UnTrack(lz); in filterfalse_dealloc()
4023 Py_XDECREF(lz->func); in filterfalse_dealloc()
4024 Py_XDECREF(lz->it); in filterfalse_dealloc()
4025 Py_TYPE(lz)->tp_free(lz); in filterfalse_dealloc()
4029 filterfalse_traverse(filterfalseobject *lz, visitproc visit, void *arg) in filterfalse_traverse() argument
4031 Py_VISIT(lz->it); in filterfalse_traverse()
4032 Py_VISIT(lz->func); in filterfalse_traverse()
4037 filterfalse_next(filterfalseobject *lz) in filterfalse_next() argument
4040 PyObject *it = lz->it; in filterfalse_next()
4050 if (lz->func == Py_None || lz->func == (PyObject *)&PyBool_Type) { in filterfalse_next()
4054 good = PyObject_CallOneArg(lz->func, item); in filterfalse_next()
4071 filterfalse_reduce(filterfalseobject *lz, PyObject *Py_UNUSED(ignored)) in filterfalse_reduce() argument
4073 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->func, lz->it); in filterfalse_reduce()
4173 countobject *lz; in itertools_count_impl() local
4233 lz = (countobject *)type->tp_alloc(type, 0); in itertools_count_impl()
4234 if (lz == NULL) { in itertools_count_impl()
4239 lz->cnt = cnt; in itertools_count_impl()
4240 lz->long_cnt = long_cnt; in itertools_count_impl()
4241 lz->long_step = long_step; in itertools_count_impl()
4243 return (PyObject *)lz; in itertools_count_impl()
4247 count_dealloc(countobject *lz) in count_dealloc() argument
4249 PyObject_GC_UnTrack(lz); in count_dealloc()
4250 Py_XDECREF(lz->long_cnt); in count_dealloc()
4251 Py_XDECREF(lz->long_step); in count_dealloc()
4252 Py_TYPE(lz)->tp_free(lz); in count_dealloc()
4256 count_traverse(countobject *lz, visitproc visit, void *arg) in count_traverse() argument
4258 Py_VISIT(lz->long_cnt); in count_traverse()
4259 Py_VISIT(lz->long_step); in count_traverse()
4264 count_nextlong(countobject *lz) in count_nextlong() argument
4269 long_cnt = lz->long_cnt; in count_nextlong()
4276 assert(lz->cnt == PY_SSIZE_T_MAX && long_cnt != NULL); in count_nextlong()
4278 stepped_up = PyNumber_Add(long_cnt, lz->long_step); in count_nextlong()
4281 lz->long_cnt = stepped_up; in count_nextlong()
4286 count_next(countobject *lz) in count_next() argument
4288 if (lz->cnt == PY_SSIZE_T_MAX) in count_next()
4289 return count_nextlong(lz); in count_next()
4290 return PyLong_FromSsize_t(lz->cnt++); in count_next()
4294 count_repr(countobject *lz) in count_repr() argument
4296 if (lz->cnt != PY_SSIZE_T_MAX) in count_repr()
4298 _PyType_Name(Py_TYPE(lz)), lz->cnt); in count_repr()
4300 if (PyLong_Check(lz->long_step)) { in count_repr()
4301 long step = PyLong_AsLong(lz->long_step); in count_repr()
4308 _PyType_Name(Py_TYPE(lz)), in count_repr()
4309 lz->long_cnt); in count_repr()
4313 _PyType_Name(Py_TYPE(lz)), in count_repr()
4314 lz->long_cnt, lz->long_step); in count_repr()
4318 count_reduce(countobject *lz, PyObject *Py_UNUSED(ignored)) in count_reduce() argument
4320 if (lz->cnt == PY_SSIZE_T_MAX) in count_reduce()
4321 return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->long_cnt, lz->long_step); in count_reduce()
4322 return Py_BuildValue("O(n)", Py_TYPE(lz), lz->cnt); in count_reduce()
4548 ziplongestobject *lz; in zip_longest_new() local
4599 lz = (ziplongestobject *)type->tp_alloc(type, 0); in zip_longest_new()
4600 if (lz == NULL) { in zip_longest_new()
4605 lz->ittuple = ittuple; in zip_longest_new()
4606 lz->tuplesize = tuplesize; in zip_longest_new()
4607 lz->numactive = tuplesize; in zip_longest_new()
4608 lz->result = result; in zip_longest_new()
4610 lz->fillvalue = fillvalue; in zip_longest_new()
4611 return (PyObject *)lz; in zip_longest_new()
4615 zip_longest_dealloc(ziplongestobject *lz) in zip_longest_dealloc() argument
4617 PyObject_GC_UnTrack(lz); in zip_longest_dealloc()
4618 Py_XDECREF(lz->ittuple); in zip_longest_dealloc()
4619 Py_XDECREF(lz->result); in zip_longest_dealloc()
4620 Py_XDECREF(lz->fillvalue); in zip_longest_dealloc()
4621 Py_TYPE(lz)->tp_free(lz); in zip_longest_dealloc()
4625 zip_longest_traverse(ziplongestobject *lz, visitproc visit, void *arg) in zip_longest_traverse() argument
4627 Py_VISIT(lz->ittuple); in zip_longest_traverse()
4628 Py_VISIT(lz->result); in zip_longest_traverse()
4629 Py_VISIT(lz->fillvalue); in zip_longest_traverse()
4634 zip_longest_next(ziplongestobject *lz) in zip_longest_next() argument
4637 Py_ssize_t tuplesize = lz->tuplesize; in zip_longest_next()
4638 PyObject *result = lz->result; in zip_longest_next()
4645 if (lz->numactive == 0) in zip_longest_next()
4650 it = PyTuple_GET_ITEM(lz->ittuple, i); in zip_longest_next()
4652 Py_INCREF(lz->fillvalue); in zip_longest_next()
4653 item = lz->fillvalue; in zip_longest_next()
4657 lz->numactive -= 1; in zip_longest_next()
4658 if (lz->numactive == 0 || PyErr_Occurred()) { in zip_longest_next()
4659 lz->numactive = 0; in zip_longest_next()
4663 Py_INCREF(lz->fillvalue); in zip_longest_next()
4664 item = lz->fillvalue; in zip_longest_next()
4665 PyTuple_SET_ITEM(lz->ittuple, i, NULL); in zip_longest_next()
4684 it = PyTuple_GET_ITEM(lz->ittuple, i); in zip_longest_next()
4686 Py_INCREF(lz->fillvalue); in zip_longest_next()
4687 item = lz->fillvalue; in zip_longest_next()
4691 lz->numactive -= 1; in zip_longest_next()
4692 if (lz->numactive == 0 || PyErr_Occurred()) { in zip_longest_next()
4693 lz->numactive = 0; in zip_longest_next()
4697 Py_INCREF(lz->fillvalue); in zip_longest_next()
4698 item = lz->fillvalue; in zip_longest_next()
4699 PyTuple_SET_ITEM(lz->ittuple, i, NULL); in zip_longest_next()
4711 zip_longest_reduce(ziplongestobject *lz, PyObject *Py_UNUSED(ignored)) in zip_longest_reduce() argument
4718 PyObject *args = PyTuple_New(PyTuple_GET_SIZE(lz->ittuple)); in zip_longest_reduce()
4722 for (i=0; i<PyTuple_GET_SIZE(lz->ittuple); i++) { in zip_longest_reduce()
4723 PyObject *elem = PyTuple_GET_ITEM(lz->ittuple, i); in zip_longest_reduce()
4734 return Py_BuildValue("ONO", Py_TYPE(lz), args, lz->fillvalue); in zip_longest_reduce()
4738 zip_longest_setstate(ziplongestobject *lz, PyObject *state) in zip_longest_setstate() argument
4741 Py_XSETREF(lz->fillvalue, state); in zip_longest_setstate()