156 lines
4.6 KiB
C++
156 lines
4.6 KiB
C++
/*
|
|
PyEO
|
|
|
|
Copyright (C) 2003 Maarten Keijzer
|
|
|
|
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#ifndef PYEO_H
|
|
#define PYEO_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <exception>
|
|
#include <boost/python.hpp>
|
|
|
|
#include <EO.h>
|
|
|
|
struct index_error : public std::exception
|
|
{
|
|
index_error(std::string w) : what(w) {};
|
|
virtual ~index_error() throw() {}
|
|
std::string what;
|
|
};
|
|
|
|
class PyFitness : public boost::python::object
|
|
{
|
|
public :
|
|
|
|
typedef PyFitness fitness_traits; // it's its own traits class :-)
|
|
|
|
PyFitness() : boost::python::object() {}
|
|
|
|
template <class T>
|
|
PyFitness(const T& o) : boost::python::object(o) {}
|
|
|
|
static unsigned nObjectives() { return objective_info.size(); }
|
|
static double tol() { return 1e-6; }
|
|
static bool maximizing(int which) { return objective_info[which] > 0; }
|
|
|
|
static void setObjectivesSize(int sz) { objective_info.resize(sz, 0); }
|
|
static void setObjectivesValue(unsigned which, int value)
|
|
{
|
|
if (which >= objective_info.size())
|
|
{
|
|
throw index_error("Too few elements allocated, resize objectives first");
|
|
}
|
|
|
|
objective_info[which] = value;
|
|
}
|
|
|
|
static std::vector<int> objective_info;
|
|
|
|
bool dominates(const PyFitness& oth) const;
|
|
|
|
double operator[](int i) const
|
|
{
|
|
boost::python::extract<double> x(object::operator[](i));
|
|
|
|
if (!x.check())
|
|
throw std::runtime_error("PyFitness: does not contain doubles");
|
|
return x();
|
|
}
|
|
|
|
bool operator<(const PyFitness& other) const
|
|
{
|
|
if (objective_info.size() == 0)
|
|
{
|
|
const object& self = *this;
|
|
const object& oth = other;
|
|
return self < oth;
|
|
}
|
|
// otherwise use objective_info
|
|
|
|
for (unsigned i = 0; i < objective_info.size(); ++i)
|
|
{
|
|
double a = objective_info[i] * operator[](i);
|
|
double b = objective_info[i] * other[i];
|
|
|
|
if ( fabs(a - b) > tol())
|
|
{
|
|
if (a < b)
|
|
return true;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool operator>(const PyFitness& other) const
|
|
{
|
|
return other.operator<(*this);
|
|
}
|
|
|
|
void printOn(std::ostream& os) const { const boost::python::object& o = *this; boost::python::api::operator<<(os,o); }
|
|
friend std::ostream& operator<<(std::ostream& os, const PyFitness& p) { p.printOn(os); return os; }
|
|
friend std::istream& operator>>(std::istream& is, PyFitness& p) { boost::python::object o; is >> o; p = o; return is; }
|
|
};
|
|
|
|
struct PyEO : public EO< PyFitness >
|
|
{
|
|
typedef PyFitness Fitness;
|
|
|
|
boost::python::object getFitness() const { return invalid()? Fitness(): fitness(); }
|
|
void setFitness(boost::python::object f) { if (f == Fitness()) invalidate(); else fitness(f); }
|
|
|
|
boost::python::object getGenome() const { return genome; }
|
|
void setGenome(boost::python::object g) { genome = g; }
|
|
boost::python::object genome;
|
|
|
|
std::string to_string() const
|
|
{
|
|
std::string result;
|
|
result += boost::python::extract<const char*>(boost::python::str(getFitness()));
|
|
result += ' ';
|
|
result += boost::python::extract<const char*>(boost::python::str(genome));
|
|
return result;
|
|
}
|
|
|
|
bool operator<(const PyEO& other) const { return EO<Fitness>::operator<(other); }
|
|
bool operator>(const PyEO& other) const { return EO<Fitness>::operator>(other); }
|
|
|
|
};
|
|
|
|
std::ostream& operator<<(std::ostream& os, const PyEO& _eo);
|
|
|
|
struct PyEO_pickle_suite : boost::python::pickle_suite
|
|
{
|
|
typedef PyEO::Fitness Fitness;
|
|
|
|
static boost::python::tuple getstate(const PyEO& _eo)
|
|
{
|
|
return boost::python::make_tuple(_eo.getFitness(), _eo.genome);
|
|
}
|
|
|
|
static void setstate(PyEO& _eo, boost::python::tuple pickled)
|
|
{
|
|
_eo.setFitness( Fitness(pickled[0]) );
|
|
_eo.genome = pickled[1];
|
|
}
|
|
};
|
|
|
|
#endif
|