Iterator.hh

#ifndef OPENMESH_PYTHON_ITERATOR_HH
#define OPENMESH_PYTHON_ITERATOR_HH

#include "Python/Bindings.hh"

namespace OpenMesh {
namespace Python {

template<class Iterator, size_t (OpenMesh::ArrayKernel::*n_items)() const>
class IteratorWrapperT {
        public:

                IteratorWrapperT(const PolyMesh& _mesh, typename Iterator::value_type _hnd, bool _skip = false) :
                        mesh_(_mesh), n_items_(n_items),
                        iterator_(_mesh, _hnd, _skip),
                        iterator_end_(_mesh, typename Iterator::value_type(int((_mesh.*n_items)()))) {
                }

                IteratorWrapperT(const TriMesh& _mesh, typename Iterator::value_type _hnd, bool _skip = false) :
                        mesh_(_mesh), n_items_(n_items),
                        iterator_(_mesh, _hnd, _skip),
                        iterator_end_(_mesh, typename Iterator::value_type(int((_mesh.*n_items)()))) {
                }

                IteratorWrapperT iter() const {
                        return *this;
                }

                typename Iterator::value_type next() {
                        if (iterator_ != iterator_end_) {
                                typename Iterator::value_type res = *iterator_;
                                ++iterator_;
                                return res;
                        }
                        else {
                                PyErr_SetString(PyExc_StopIteration, "No more data.");
                                boost::python::throw_error_already_set();
                        }
                        return typename Iterator::value_type();
                }

                unsigned int len() const {
                        return (mesh_.*n_items_)();
                }

        private:
                const OpenMesh::PolyConnectivity& mesh_;
                size_t (OpenMesh::ArrayKernel::*n_items_)() const;
                Iterator iterator_;
                Iterator iterator_end_;
};

template<class Iterator, size_t (OpenMesh::ArrayKernel::*n_items)() const>
void expose_iterator(const char *_name) {
        class_<IteratorWrapperT<Iterator, n_items> >(_name, init<PolyMesh&, typename Iterator::value_type, optional<bool> >())
                .def(init<TriMesh&, typename Iterator::value_type, optional<bool> >())
                .def("__iter__", &IteratorWrapperT<Iterator, n_items>::iter)
                .def("__next__", &IteratorWrapperT<Iterator, n_items>::next)
                .def("__len__", &IteratorWrapperT<Iterator, n_items>::len)
                .def("next", &IteratorWrapperT<Iterator, n_items>::next)
                ;
}

} // namespace OpenMesh
} // namespace Python

#endif