1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
/*
* Copyright (C) 2001 Ichiro Fujinaga, Michael Droettboom, and Karl MacMillan
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
//////////////////////////////////////////////////////////////////////////////
// Iterators
/* These objects make it (somewhat) easier to create Python iterator objects.
To create a Python iterator, inherit from IteratorObject and define:
An initializer (constructor):
void init([any signature]);
An iteration function:
PyObject* next(IteratorObject* self);
An optional destructor method:
static void dealloc(IteratorObject* self);
Plus any data members needed to keep the iteration going.
Within your 'next' function, simply return 0 to stop iteration. All the
other details of handling Python iteration are taken care of by this base
class.
To create an iterator and return it from a function:
MyIterator* iterator = iterator_new_simple<MyIterator>();
iterator->init([initialization parameters]);
return (PyObject*)iterator;
See also below the BasicIterator template which makes it easy to iterate
through C++ STL sequences.
*/
#ifndef mgd010105_iterator_hpp
#define mgd010105_iterator_hpp
#include <Python.h>
#include "gameramodule.hpp"
struct IteratorObject {
PyObject_HEAD
PyObject*(*m_fp_next)(IteratorObject*);
void(*m_fp_dealloc)(IteratorObject*);
static void dealloc(IteratorObject* self) { };
};
void init_IteratorType();
// PyTypeObject* get_IteratorType();
template<class T>
T* iterator_new();
template<class T>
T* iterator_new() {
IteratorObject* so;
PyTypeObject* type = get_IteratorType();
type->tp_basicsize = sizeof(T);
so = (IteratorObject*)(type->tp_alloc(type, 0));
so->m_fp_next = T::next;
so->m_fp_dealloc = T::dealloc;
return (T*)so;
}
// CONCRETE ITERATORS
template<class T>
struct BasicIterator : IteratorObject {
int init(typename T::iterator begin, typename T::iterator end) {
m_it = begin;
m_end = end;
return 1;
}
static PyObject* next(IteratorObject* self) {
BasicIterator<T>* so = (BasicIterator<T>*)self;
if (so->m_it == so->m_end)
return 0;
return (PyObject*)*((so->m_it)++);
}
typename T::iterator m_it, m_end;
};
#endif
|
