2001-04-20 23:06:46 +02:00
|
|
|
/* Iterator objects */
|
|
|
|
|
|
|
|
#include "Python.h"
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
PyObject_HEAD
|
|
|
|
long it_index;
|
2002-07-16 22:24:46 +02:00
|
|
|
PyObject *it_seq; /* Set to NULL when iterator is exhausted */
|
2001-04-23 16:08:49 +02:00
|
|
|
} seqiterobject;
|
2001-04-20 23:06:46 +02:00
|
|
|
|
|
|
|
PyObject *
|
2001-04-23 16:08:49 +02:00
|
|
|
PySeqIter_New(PyObject *seq)
|
2001-04-20 23:06:46 +02:00
|
|
|
{
|
2001-04-23 16:08:49 +02:00
|
|
|
seqiterobject *it;
|
2002-05-08 10:44:21 +02:00
|
|
|
|
|
|
|
if (!PySequence_Check(seq)) {
|
|
|
|
PyErr_BadInternalCall();
|
|
|
|
return NULL;
|
|
|
|
}
|
2002-03-18 21:43:51 +01:00
|
|
|
it = PyObject_GC_New(seqiterobject, &PySeqIter_Type);
|
2001-04-20 23:06:46 +02:00
|
|
|
if (it == NULL)
|
|
|
|
return NULL;
|
|
|
|
it->it_index = 0;
|
|
|
|
Py_INCREF(seq);
|
|
|
|
it->it_seq = seq;
|
2002-03-18 21:43:51 +01:00
|
|
|
_PyObject_GC_TRACK(it);
|
2001-04-20 23:06:46 +02:00
|
|
|
return (PyObject *)it;
|
|
|
|
}
|
2002-07-16 22:24:46 +02:00
|
|
|
|
2001-04-20 23:06:46 +02:00
|
|
|
static void
|
2001-04-23 16:08:49 +02:00
|
|
|
iter_dealloc(seqiterobject *it)
|
2001-04-20 23:06:46 +02:00
|
|
|
{
|
2002-03-18 21:43:51 +01:00
|
|
|
_PyObject_GC_UNTRACK(it);
|
2002-07-16 22:24:46 +02:00
|
|
|
Py_XDECREF(it->it_seq);
|
2002-03-18 21:43:51 +01:00
|
|
|
PyObject_GC_Del(it);
|
2001-04-20 23:06:46 +02:00
|
|
|
}
|
|
|
|
|
2001-07-12 15:27:25 +02:00
|
|
|
static int
|
|
|
|
iter_traverse(seqiterobject *it, visitproc visit, void *arg)
|
|
|
|
{
|
2002-07-16 22:24:46 +02:00
|
|
|
if (it->it_seq == NULL)
|
|
|
|
return 0;
|
2001-07-12 15:27:25 +02:00
|
|
|
return visit(it->it_seq, arg);
|
|
|
|
}
|
|
|
|
|
2001-04-23 16:08:49 +02:00
|
|
|
static PyObject *
|
|
|
|
iter_getiter(PyObject *it)
|
|
|
|
{
|
|
|
|
Py_INCREF(it);
|
|
|
|
return it;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
iter_iternext(PyObject *iterator)
|
|
|
|
{
|
|
|
|
seqiterobject *it;
|
|
|
|
PyObject *seq;
|
2002-08-09 03:30:17 +02:00
|
|
|
PyObject *result;
|
2001-04-23 16:08:49 +02:00
|
|
|
|
|
|
|
assert(PySeqIter_Check(iterator));
|
|
|
|
it = (seqiterobject *)iterator;
|
|
|
|
seq = it->it_seq;
|
2002-07-16 22:24:46 +02:00
|
|
|
if (seq == NULL)
|
|
|
|
return NULL;
|
2001-04-20 23:06:46 +02:00
|
|
|
|
2002-08-09 03:30:17 +02:00
|
|
|
result = PySequence_GetItem(seq, it->it_index);
|
|
|
|
if (result != NULL) {
|
|
|
|
it->it_index++;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
if (PyErr_ExceptionMatches(PyExc_IndexError) ||
|
|
|
|
PyErr_ExceptionMatches(PyExc_StopIteration))
|
|
|
|
{
|
|
|
|
PyErr_Clear();
|
2002-07-16 22:24:46 +02:00
|
|
|
Py_DECREF(seq);
|
|
|
|
it->it_seq = NULL;
|
2001-04-20 23:06:46 +02:00
|
|
|
}
|
2002-08-09 03:30:17 +02:00
|
|
|
return NULL;
|
2001-04-20 23:06:46 +02:00
|
|
|
}
|
|
|
|
|
2001-04-23 16:08:49 +02:00
|
|
|
PyTypeObject PySeqIter_Type = {
|
2001-04-20 23:06:46 +02:00
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
|
|
0, /* ob_size */
|
|
|
|
"iterator", /* tp_name */
|
2002-03-18 21:43:51 +01:00
|
|
|
sizeof(seqiterobject), /* tp_basicsize */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_itemsize */
|
|
|
|
/* methods */
|
|
|
|
(destructor)iter_dealloc, /* tp_dealloc */
|
|
|
|
0, /* tp_print */
|
2001-08-02 06:15:00 +02:00
|
|
|
0, /* tp_getattr */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_setattr */
|
|
|
|
0, /* tp_compare */
|
|
|
|
0, /* tp_repr */
|
|
|
|
0, /* tp_as_number */
|
|
|
|
0, /* tp_as_sequence */
|
|
|
|
0, /* tp_as_mapping */
|
|
|
|
0, /* tp_hash */
|
|
|
|
0, /* tp_call */
|
|
|
|
0, /* tp_str */
|
2001-08-02 06:15:00 +02:00
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_setattro */
|
|
|
|
0, /* tp_as_buffer */
|
2002-03-18 21:43:51 +01:00
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_doc */
|
2001-07-12 15:27:25 +02:00
|
|
|
(traverseproc)iter_traverse, /* tp_traverse */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_clear */
|
|
|
|
0, /* tp_richcompare */
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
(getiterfunc)iter_getiter, /* tp_iter */
|
2001-04-23 16:08:49 +02:00
|
|
|
(iternextfunc)iter_iternext, /* tp_iternext */
|
2002-07-16 22:24:46 +02:00
|
|
|
0, /* tp_methods */
|
2001-08-02 06:15:00 +02:00
|
|
|
0, /* tp_members */
|
|
|
|
0, /* tp_getset */
|
|
|
|
0, /* tp_base */
|
|
|
|
0, /* tp_dict */
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
0, /* tp_descr_set */
|
2001-04-20 23:06:46 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
/* -------------------------------------- */
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
PyObject_HEAD
|
2002-07-16 22:24:46 +02:00
|
|
|
PyObject *it_callable; /* Set to NULL when iterator is exhausted */
|
|
|
|
PyObject *it_sentinel; /* Set to NULL when iterator is exhausted */
|
2001-04-20 23:06:46 +02:00
|
|
|
} calliterobject;
|
|
|
|
|
|
|
|
PyObject *
|
|
|
|
PyCallIter_New(PyObject *callable, PyObject *sentinel)
|
|
|
|
{
|
|
|
|
calliterobject *it;
|
2002-03-18 21:43:51 +01:00
|
|
|
it = PyObject_GC_New(calliterobject, &PyCallIter_Type);
|
2001-04-20 23:06:46 +02:00
|
|
|
if (it == NULL)
|
|
|
|
return NULL;
|
|
|
|
Py_INCREF(callable);
|
|
|
|
it->it_callable = callable;
|
|
|
|
Py_INCREF(sentinel);
|
|
|
|
it->it_sentinel = sentinel;
|
2002-03-18 21:43:51 +01:00
|
|
|
_PyObject_GC_TRACK(it);
|
2001-04-20 23:06:46 +02:00
|
|
|
return (PyObject *)it;
|
|
|
|
}
|
|
|
|
static void
|
|
|
|
calliter_dealloc(calliterobject *it)
|
|
|
|
{
|
2002-03-18 21:43:51 +01:00
|
|
|
_PyObject_GC_UNTRACK(it);
|
2002-07-16 22:24:46 +02:00
|
|
|
Py_XDECREF(it->it_callable);
|
|
|
|
Py_XDECREF(it->it_sentinel);
|
2002-03-18 21:43:51 +01:00
|
|
|
PyObject_GC_Del(it);
|
2001-04-20 23:06:46 +02:00
|
|
|
}
|
2001-04-23 16:08:49 +02:00
|
|
|
|
2001-07-12 15:27:25 +02:00
|
|
|
static int
|
|
|
|
calliter_traverse(calliterobject *it, visitproc visit, void *arg)
|
|
|
|
{
|
|
|
|
int err;
|
2002-07-16 22:24:46 +02:00
|
|
|
if (it->it_callable != NULL && (err = visit(it->it_callable, arg)))
|
2001-07-12 15:27:25 +02:00
|
|
|
return err;
|
2002-07-16 22:24:46 +02:00
|
|
|
if (it->it_sentinel != NULL && (err = visit(it->it_sentinel, arg)))
|
2001-07-12 15:27:25 +02:00
|
|
|
return err;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2001-04-23 16:08:49 +02:00
|
|
|
static PyObject *
|
|
|
|
calliter_iternext(calliterobject *it)
|
|
|
|
{
|
2002-07-16 22:24:46 +02:00
|
|
|
if (it->it_callable != NULL) {
|
|
|
|
PyObject *result = PyObject_CallObject(it->it_callable, NULL);
|
|
|
|
if (result != NULL) {
|
|
|
|
int ok;
|
|
|
|
ok = PyObject_RichCompareBool(result,
|
|
|
|
it->it_sentinel,
|
|
|
|
Py_EQ);
|
|
|
|
if (ok == 0)
|
|
|
|
return result; /* Common case, fast path */
|
2001-04-23 16:08:49 +02:00
|
|
|
Py_DECREF(result);
|
2002-07-16 22:24:46 +02:00
|
|
|
if (ok > 0) {
|
|
|
|
Py_DECREF(it->it_callable);
|
|
|
|
it->it_callable = NULL;
|
|
|
|
Py_DECREF(it->it_sentinel);
|
|
|
|
it->it_sentinel = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (PyErr_ExceptionMatches(PyExc_StopIteration)) {
|
|
|
|
PyErr_Clear();
|
|
|
|
Py_DECREF(it->it_callable);
|
|
|
|
it->it_callable = NULL;
|
|
|
|
Py_DECREF(it->it_sentinel);
|
|
|
|
it->it_sentinel = NULL;
|
2001-04-23 16:08:49 +02:00
|
|
|
}
|
|
|
|
}
|
2002-07-16 22:24:46 +02:00
|
|
|
return NULL;
|
2001-04-23 16:08:49 +02:00
|
|
|
}
|
|
|
|
|
2001-04-20 23:06:46 +02:00
|
|
|
PyTypeObject PyCallIter_Type = {
|
|
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
|
|
0, /* ob_size */
|
|
|
|
"callable-iterator", /* tp_name */
|
2002-03-18 21:43:51 +01:00
|
|
|
sizeof(calliterobject), /* tp_basicsize */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_itemsize */
|
|
|
|
/* methods */
|
|
|
|
(destructor)calliter_dealloc, /* tp_dealloc */
|
|
|
|
0, /* tp_print */
|
2001-08-02 06:15:00 +02:00
|
|
|
0, /* tp_getattr */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_setattr */
|
|
|
|
0, /* tp_compare */
|
|
|
|
0, /* tp_repr */
|
|
|
|
0, /* tp_as_number */
|
|
|
|
0, /* tp_as_sequence */
|
|
|
|
0, /* tp_as_mapping */
|
|
|
|
0, /* tp_hash */
|
|
|
|
0, /* tp_call */
|
|
|
|
0, /* tp_str */
|
2001-08-02 06:15:00 +02:00
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_setattro */
|
|
|
|
0, /* tp_as_buffer */
|
2002-03-18 21:43:51 +01:00
|
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_doc */
|
2001-07-12 15:27:25 +02:00
|
|
|
(traverseproc)calliter_traverse, /* tp_traverse */
|
2001-04-20 23:06:46 +02:00
|
|
|
0, /* tp_clear */
|
|
|
|
0, /* tp_richcompare */
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
(getiterfunc)iter_getiter, /* tp_iter */
|
2001-04-23 16:08:49 +02:00
|
|
|
(iternextfunc)calliter_iternext, /* tp_iternext */
|
2002-07-16 22:24:46 +02:00
|
|
|
0, /* tp_methods */
|
2001-08-02 06:15:00 +02:00
|
|
|
0, /* tp_members */
|
|
|
|
0, /* tp_getset */
|
|
|
|
0, /* tp_base */
|
|
|
|
0, /* tp_dict */
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
0, /* tp_descr_set */
|
2001-04-20 23:06:46 +02:00
|
|
|
};
|