statistics.cpp

#include <utils/eoStat.h>

#include "PyEO.h"
#include "valueParam.h"

using namespace boost::python;

class StatBaseWrapper : public eoStatBase<PyEO> 
{
    public:
    PyObject* self;
    StatBaseWrapper(PyObject* p) : self(p) {}

    void operator()(const eoPop<PyEO>& pop)
    {
        call_method<void>(self, "__call__", boost::ref(pop));
    }
};

class SortedStatBaseWrapper : public eoSortedStatBase<PyEO> 
{
    public:
    PyObject* self;
    SortedStatBaseWrapper(PyObject* p) : self(p) {}

    void operator()(const std::vector<const PyEO*>& pop)
    {
        call_method<void>(self, "__call__", boost::ref(pop));
    }
};

typedef std::vector<const PyEO*> eoPopView;

const PyEO& popview_getitem(const std::vector<const PyEO*>& pop, int it) 
{ 
    unsigned item = unsigned(it);
    if (item > pop.size())
        throw index_error("too much");
            
    return *pop[item]; 
}

void statistics()
{
    class_<eoStatBase<PyEO>, StatBaseWrapper, boost::noncopyable>
        ("eoStatBase", init<>())
        .def("lastCall", &eoStatBase<PyEO>::lastCall)
        .def("__call__", &StatBaseWrapper::operator())
        ;

    class_< eoPopView >("eoPopView")
        .def("__getitem__", popview_getitem, return_internal_reference<>() )
        .def("__len__", &eoPopView::size)
        ;
    
    class_<eoSortedStatBase<PyEO>, SortedStatBaseWrapper, boost::noncopyable>
        ("eoSortedStatBase", init<>())
        .def("lastCall", &eoSortedStatBase<PyEO>::lastCall)
        .def("__call__", &SortedStatBaseWrapper::operator())
        ;
}

---