Make iterators length transparent where possible.

This commit is contained in:
Raymond Hettinger 2004-03-18 22:43:10 +00:00
parent 1e5809ff02
commit 435bf58b7b
3 changed files with 42 additions and 3 deletions

View File

@ -71,6 +71,19 @@ iter_iternext(PyObject *iterator)
return NULL;
}
static int
iter_len(seqiterobject *it)
{
if (it->it_seq)
return PyObject_Size(it->it_seq) - it->it_index;
return 0;
}
static PySequenceMethods iter_as_sequence = {
(inquiry)iter_len, /* sq_length */
0, /* sq_concat */
};
PyTypeObject PySeqIter_Type = {
PyObject_HEAD_INIT(&PyType_Type)
0, /* ob_size */
@ -85,7 +98,7 @@ PyTypeObject PySeqIter_Type = {
0, /* tp_compare */
0, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
&iter_as_sequence, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */

View File

@ -2707,6 +2707,19 @@ listiter_next(listiterobject *it)
return NULL;
}
static int
listiter_len(listiterobject *it)
{
if (it->it_seq)
return PyList_GET_SIZE(it->it_seq) - it->it_index;
return 0;
}
static PySequenceMethods listiter_as_sequence = {
(inquiry)listiter_len, /* sq_length */
0, /* sq_concat */
};
PyTypeObject PyListIter_Type = {
PyObject_HEAD_INIT(&PyType_Type)
0, /* ob_size */
@ -2721,7 +2734,7 @@ PyTypeObject PyListIter_Type = {
0, /* tp_compare */
0, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
&listiter_as_sequence, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */

View File

@ -839,6 +839,19 @@ tupleiter_next(tupleiterobject *it)
return NULL;
}
static int
tupleiter_len(tupleiterobject *it)
{
if (it->it_seq)
return PyTuple_GET_SIZE(it->it_seq) - it->it_index;
return 0;
}
static PySequenceMethods tupleiter_as_sequence = {
(inquiry)tupleiter_len, /* sq_length */
0, /* sq_concat */
};
PyTypeObject PyTupleIter_Type = {
PyObject_HEAD_INIT(&PyType_Type)
0, /* ob_size */
@ -853,7 +866,7 @@ PyTypeObject PyTupleIter_Type = {
0, /* tp_compare */
0, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
&tupleiter_as_sequence, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */