Circulator.hh

#ifndef OPENMESH_PYTHON_CIRCULATOR_HH
#define OPENMESH_PYTHON_CIRCULATOR_HH

#include "Python/Bindings.hh"

namespace OpenMesh {
namespace Python {

template<class Circulator, class CenterEntityHandle>
class CirculatorWrapperT {
        public:

                CirculatorWrapperT(PolyMesh& _mesh, CenterEntityHandle _center) :
                        circulator_(_mesh, _center) {
                }

                CirculatorWrapperT(TriMesh& _mesh, CenterEntityHandle _center) :
                        circulator_(_mesh, _center) {
                }

                CirculatorWrapperT iter() const {
                        return *this;
                }

                typename Circulator::value_type next() {
                        if (circulator_.is_valid()) {
                                typename Circulator::value_type res = *circulator_;
                                ++circulator_;
                                return res;
                        }
                        else {
                                PyErr_SetString(PyExc_StopIteration, "No more data.");
                                boost::python::throw_error_already_set();
                        }
                        return typename Circulator::value_type();
                }

        private:
                Circulator circulator_;
};

template<class Circulator, class CenterEntityHandle>
void expose_circulator(const char *_name) {
        class_<CirculatorWrapperT<Circulator, CenterEntityHandle> >(_name, init<TriMesh&, CenterEntityHandle>())
                .def(init<PolyMesh&, CenterEntityHandle>())
                .def("__iter__", &CirculatorWrapperT<Circulator, CenterEntityHandle>::iter)
                .def("__next__", &CirculatorWrapperT<Circulator, CenterEntityHandle>::next)
                .def("next", &CirculatorWrapperT<Circulator, CenterEntityHandle>::next)
                ;
}

} // namespace OpenMesh
} // namespace Python

#endif