iframe-proxy | Sunbelt Computer Software

57 lines (51 loc) · 2.72 KB

#include <pybind11/pybind11.h>

#include "common/tick_pybind11_arrays.h"

#include "tick/random/test_rand.h"

namespace py = pybind11;

PYBIND11_MODULE(crandom, m) {

tick::pybind::ensure_numpy_imported();

m.doc() = "tick.random pybind11 bindings";

py::class_<Rand>(m, "Rand")

.def(py::init<>())

.def(py::init<int>(), py::arg("seed"))

.def("uniform_int", &Rand::uniform_int, py::arg("a"), py::arg("b"))

.def("uniform_ulong", &Rand::uniform_ulong, py::arg("a"), py::arg("b"))

.def("uniform", static_cast<double (Rand::*)()>(&Rand::uniform))

.def("uniform", static_cast<double (Rand::*)(double, double)>(&Rand::uniform),

py::arg("a"), py::arg("b"))

.def("gaussian", static_cast<double (Rand::*)()>(&Rand::gaussian))

.def("gaussian",

static_cast<double (Rand::*)(double, double)>(&Rand::gaussian),

py::arg("mean"), py::arg("std"))

.def("exponential", &Rand::exponential, py::arg("intensity"))

.def("poisson", &Rand::poisson, py::arg("rate"))

.def("set_discrete_dist", &Rand::set_discrete_dist, py::arg("probabilities"))

.def("discrete", static_cast<ulong (Rand::*)()>(&Rand::discrete))

.def("discrete", static_cast<ulong (Rand::*)(ArrayDouble)>(&Rand::discrete),

py::arg("probabilities"))

.def("get_seed", &Rand::get_seed)

.def("reseed", &Rand::reseed, py::arg("seed"));

m.def("test_uniform_int", &test_uniform_int, py::arg("a"), py::arg("b"),