74 lines
3.4 KiB
C++
74 lines
3.4 KiB
C++
/* @file pydistribution.cpp */
|
|
|
|
#include "pydistribution.hpp"
|
|
#include "xo/distribution/Normal.hpp"
|
|
#include "xo/distribution/ExplicitDist.hpp"
|
|
//#include "xo/reflect/SelfTagging.hpp"
|
|
#include "xo/pyutil/pyutil.hpp"
|
|
#include <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
|
|
namespace xo {
|
|
using xo::distribution::Normal;
|
|
using xo::distribution::Distribution;
|
|
using xo::distribution::ExplicitDist;
|
|
using xo::ref::rp;
|
|
|
|
namespace sim {
|
|
namespace py = pybind11;
|
|
|
|
PYBIND11_MODULE(PYDISTRIBUTION_MODULE_NAME(), m) {
|
|
m.doc() = "pybind11 distribution plugin"; // optional module docstring
|
|
|
|
m.def("normalcdf",
|
|
&Normal::cdf_impl,
|
|
"cumulative normal distribution",
|
|
py::arg("x"));
|
|
|
|
py::class_<Distribution<double>,
|
|
rp<Distribution<double>>>(m, "Distribution")
|
|
.def("cdf", &Distribution<double>::cdf,
|
|
"return cumulative distribution function at x",
|
|
py::arg("x"));
|
|
|
|
py::class_<ExplicitDist<double>,
|
|
Distribution<double>,
|
|
rp<ExplicitDist<double>>>(m, "ExplicitDist")
|
|
.def_static("make", &ExplicitDist<double>::make,
|
|
"create instance",
|
|
py::arg("bucket_dx"), py::arg("ref_value"))
|
|
.def_static("make_n", &ExplicitDist<double>::make_n,
|
|
"create instance with n buckets",
|
|
py::arg("n"), py::arg("bucket_dx"), py::arg("ref_value"))
|
|
.def("n_bucket", &ExplicitDist<double>::n_bucket,
|
|
"return number of explicitly-represented buckets in distribution")
|
|
.def("lo", &ExplicitDist<double>::lo,
|
|
"return least upper bound x: cdf(x)=0")
|
|
.def("hi", &ExplicitDist<double>::hi,
|
|
"return greatest lower bound x: cdf(x)=1")
|
|
.def("density", &ExplicitDist<double>::density,
|
|
"return probability density at x",
|
|
py::arg("x"))
|
|
.def("density_v", &ExplicitDist<double>::density_v,
|
|
"return probability density vector for all explicit buckets."
|
|
" each member is pair {lh bucket edge, density}")
|
|
.def("signed_bucket_index", &ExplicitDist<double>::signed_bucket_index,
|
|
"signed index to probability bucket. ref_value -> 0",
|
|
py::arg("x"))
|
|
.def("scale_bucket", &ExplicitDist<double>::scale_bucket,
|
|
"scale probability weight in bucket containing x by k",
|
|
py::arg("x"), py::arg("k"))
|
|
.def("scale_by_normal_cdf", &ExplicitDist<double>::scale_by_normal_cdf,
|
|
"scale by normal cumulative distribution N(sign.(x-mean)/sigma)."
|
|
" expect sign in {+1, -1}",
|
|
py::arg("sign"), py::arg("mean"), py::arg("sigma"))
|
|
.def("renormalize", &ExplicitDist<double>::renormalize,
|
|
"renormalize to ensure sum of weights=1")
|
|
.def("check_renormalize", &ExplicitDist<double>::check_renormalize,
|
|
"renormalize if needed, otherwise do nothing")
|
|
.def("__repr__", &ExplicitDist<double>::display_string);
|
|
}
|
|
} /*namespace sim*/
|
|
} /*namespace xo*/
|
|
|
|
/* end pydistribution.cpp */
|