cf2a57dd88
Remove support for (outdated) <strstream>, require <sstream>. Require uint32_t for now, defined in stdint.h according to C99. Some general cleanup and more documentation.
107 lines
2.6 KiB
C++
107 lines
2.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
|
|
*/
|
|
|
|
#include <utils/eoRNG.h>
|
|
#include <boost/python.hpp>
|
|
|
|
using namespace boost::python;
|
|
|
|
#include <sstream>
|
|
#include <boost/python/detail/api_placeholder.hpp>
|
|
|
|
using namespace boost::python;
|
|
|
|
namespace eo
|
|
{
|
|
eoRng rng(time(0));
|
|
}
|
|
|
|
eoRng& get_rng() { return rng; }
|
|
double normal(eoRng& rng) { return rng.normal(); }
|
|
|
|
std::string rng_to_string(const eoRng& _rng)
|
|
{
|
|
std::ostringstream os;
|
|
_rng.printOn(os);
|
|
os << std::ends;
|
|
return os.str();
|
|
}
|
|
|
|
|
|
void rng_from_string(eoRng& _rng, std::string s)
|
|
{
|
|
std::istringstream is(s);
|
|
_rng.readFrom(is);
|
|
}
|
|
|
|
struct RNG_pickle_suite : boost::python::pickle_suite
|
|
{
|
|
static
|
|
boost::python::tuple getstate(const eoRng& _rng)
|
|
{
|
|
return boost::python::make_tuple(str(rng_to_string(_rng)));
|
|
}
|
|
static
|
|
void setstate(eoRng& _rng, boost::python::tuple pickled)
|
|
{
|
|
std::string state = extract<std::string>(pickled[0]);
|
|
rng_from_string(_rng, state);
|
|
}
|
|
};
|
|
|
|
int spin(eoRng& _rng, numeric::array values, double total)
|
|
{
|
|
if (total == 0.0)
|
|
{
|
|
unsigned sz = len(values);
|
|
for (unsigned i = 0; i < sz; ++i)
|
|
{
|
|
total += extract<double>(values[i]); //extract?
|
|
}
|
|
}
|
|
|
|
double chance = _rng.uniform() * total;
|
|
|
|
int i = 0;
|
|
while (chance >= 0.0)
|
|
chance -= extract<double>(values[i++]);
|
|
|
|
return --i;
|
|
}
|
|
|
|
void random_numbers()
|
|
{
|
|
class_<eoRng, boost::noncopyable>("eoRng", init<uint32>())
|
|
.def("flip", &eoRng::flip)
|
|
.def("random", &eoRng::random)
|
|
.def("rand", &eoRng::rand)
|
|
.def("rand_max", &eoRng::rand_max)
|
|
.def("reseed", &eoRng::reseed)
|
|
.def("uniform", &eoRng::uniform)
|
|
.def("normal", normal)
|
|
.def("negexp", &eoRng::negexp)
|
|
.def("to_string", rng_to_string)
|
|
.def("from_string", rng_from_string)
|
|
.def("roulette_wheel", spin)
|
|
.def_pickle(RNG_pickle_suite())
|
|
;
|
|
|
|
def("rng", get_rng, return_value_policy<reference_existing_object>());
|
|
}
|