Ticket #3450: rangemodule.cpp

#include <boost/python.hpp>
#include <boost/python/stl_iterator.hpp>
#include <boost/lambda/lambda.hpp>
#include <boost/lambda/casts.hpp>
#include <algorithm>

using namespace std;

namespace bp = boost::python;
namespace bl = boost::lambda;


class C {

        size_t  *v;

        size_t  m_N;

public:

        typedef         const size_t*           const_dim_iter;

        C(size_t N = 3) : v(new size_t[N]),m_N(N) {
                if (N < 1) throw invalid_argument("Wrong input parameter");

                *v = 1;
                transform(v,v+N-1,v+1,bl::_1+static_cast<size_t>(1));
        }

        ~C() { delete []v; }

        const_dim_iter  dbeg() { return v; }

        const_dim_iter  dend() { return v+m_N; }

        template<class InIt>
        void set_dims(InIt b,InIt e) {
                delete []v;
#ifndef WORK_AROUND
                m_N = distance(b,e);

                v = new size_t[m_N];
#else
                v = new size_t[5];
                m_N = 5;
#endif

                transform(b,e,v,bl::ll_static_cast<size_t>(bl::_1));
        }
};

class PyWrap : public C {

public:

        PyWrap(size_t N = 3) : C(N) {}

        void py_set_dims(bp::object o) {

                bp::stl_input_iterator<int> begin(o), end;
                set_dims(begin,end);
        }
};

BOOST_PYTHON_MODULE(test_range) {

        bp::class_<C>("C_base", bp::init< bp::optional<size_t> >("Test wrapper class for a buffer"))
                        .add_property("dims",bp::range(&C::dbeg,&C::dend));

        bp::class_<PyWrap,bp::bases<C> >("C", bp::init< bp::optional<size_t> >("Test wrapper class for a buffer"))
                        .def("set_dims",&PyWrap::py_set_dims);
}