LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
_random.cc
Go to the documentation of this file.
1/*
2 * LSST Data Management System
3 * Copyright 2008-2016 AURA/LSST.
4 *
5 * This product includes software developed by the
6 * LSST Project (http://www.lsst.org/).
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the LSST License Statement and
19 * the GNU General Public License along with this program. If not,
20 * see <https://www.lsstcorp.org/LegalNotices/>.
21 */
22
23#include <pybind11/pybind11.h>
24#include <lsst/utils/python.h>
25#include <pybind11/stl.h>
26
29
30namespace py = pybind11;
31
32using namespace lsst::afw::math;
33using namespace pybind11::literals;
34namespace lsst {
35namespace afw {
36namespace math {
37template <typename ImageT>
38void declareRandomImage(lsst::utils::python::WrapperCollection &wrappers) {
39 wrappers.wrap([](auto &mod) {
40 mod.def("randomUniformImage", (void (*)(ImageT *, Random &))randomUniformImage<ImageT>);
41 mod.def("randomUniformPosImage", (void (*)(ImageT *, Random &))randomUniformPosImage<ImageT>);
42 mod.def("randomUniformIntImage",
43 (void (*)(ImageT *, Random &, unsigned long))randomUniformIntImage<ImageT>);
44 mod.def("randomFlatImage",
45 (void (*)(ImageT *, Random &, double const, double const))randomFlatImage<ImageT>);
46 mod.def("randomGaussianImage", (void (*)(ImageT *, Random &))randomGaussianImage<ImageT>);
47 mod.def("randomChisqImage", (void (*)(ImageT *, Random &, double const))randomChisqImage<ImageT>);
48 mod.def("randomPoissonImage", (void (*)(ImageT *, Random &, double const))randomPoissonImage<ImageT>);
49 });
50}
51
52void declareRandom(lsst::utils::python::WrapperCollection &wrappers) {
53 auto clsRandom = wrappers.wrapType(py::class_<Random>(wrappers.module, "Random"), [](auto &mod,
54 auto &cls) {
55 /* Constructors */
56 cls.def(py::init<Random::Algorithm, unsigned long>(), "algorithm"_a = Random::Algorithm::MT19937,
57 "seed"_a = 1);
58 cls.def(py::init<std::string const &, unsigned long>(), "algorithm"_a, "seed"_a = 1);
59
60 /* Members */
61 cls.def("deepCopy", &Random::deepCopy);
62 cls.def("getAlgorithm", &Random::getAlgorithm);
63 cls.def("getAlgorithmName", &Random::getAlgorithmName);
64 cls.def_static("getAlgorithmNames", &Random::getAlgorithmNames);
65 cls.def("getSeed", &Random::getSeed);
66 cls.def("uniform", &Random::uniform);
67 cls.def("uniformPos", &Random::uniformPos);
68 cls.def("uniformInt", &Random::uniformInt);
69 cls.def("flat", &Random::flat);
70 cls.def("gaussian", &Random::gaussian);
71 cls.def("chisq", &Random::chisq);
72 cls.def("poisson", &Random::poisson);
73
74 // getState and setState are special, their std::string cannot
75 // be converted to a Python string (needs to go to bytes instead)
76 // thus use the same solution as employed with Swig
77 cls.def("getState", [](Random &self) -> py::object {
78 std::string state = self.getState();
79 return py::reinterpret_steal<py::object>(PyBytes_FromStringAndSize(state.data(), state.size()));
80 });
81 cls.def("setState", [](Random &self, py::bytes const &state) { self.setState(state); });
82 });
83 /* Member types and enums */
84 wrappers.wrapType(py::enum_<Random::Algorithm>(clsRandom, "Algorithm"), [](auto &mod, auto &enm) {
85 enm.value("MT19937", Random::Algorithm::MT19937);
86 enm.value("RANLXS0", Random::Algorithm::RANLXS0);
87 enm.value("RANLXS1", Random::Algorithm::RANLXS1);
88 enm.value("RANLXS2", Random::Algorithm::RANLXS2);
89 enm.value("RANLXD1", Random::Algorithm::RANLXD1);
90 enm.value("RANLXD2", Random::Algorithm::RANLXD2);
91 enm.value("RANLUX", Random::Algorithm::RANLUX);
92 enm.value("RANLUX389", Random::Algorithm::RANLUX389);
93 enm.value("CMRG", Random::Algorithm::CMRG);
94 enm.value("MRG", Random::Algorithm::MRG);
95 enm.value("TAUS", Random::Algorithm::TAUS);
96 enm.value("TAUS2", Random::Algorithm::TAUS2);
97 enm.value("GFSR4", Random::Algorithm::GFSR4);
98 enm.value("NUM_ALGORITHMS", Random::Algorithm::NUM_ALGORITHMS);
99 enm.export_values();
100 });
101}
102
103void wrapRandom(lsst::utils::python::WrapperCollection &wrappers) {
104 wrappers.addSignatureDependency("lsst.afw.image");
105 declareRandom(wrappers);
106 declareRandomImage<lsst::afw::image::Image<double>>(wrappers);
107 declareRandomImage<lsst::afw::image::Image<float>>(wrappers);
108}
109} // namespace math
110} // namespace afw
111} // namespace lsst
A class that can be used to generate sequences of random numbers according to a number of different a...
Definition Random.h:57
@ MT19937
The MT19937 "Mersenne Twister" generator of Makoto Matsumoto and Takuji Nishimura.
Definition Random.h:62
@ TAUS
A maximally equidistributed combined Tausworthe generator by L'Ecuyer.
Definition Random.h:85
@ RANLUX
Original version of the RANLUX algorithm, 24-bit output.
Definition Random.h:77
@ RANLXS2
Second-generation version of the RANLUX algorithm of Lüscher, 24-bit output, luxury level 2 (stronges...
Definition Random.h:71
@ RANLXD1
Double precision (48-bit) output using the RANLXS algorithm, luxury level 1 (weakest).
Definition Random.h:73
@ NUM_ALGORITHMS
Number of supported algorithms.
Definition Random.h:92
@ MRG
Fifth-order multiple recursive generator by L'Ecuyer, Blouin, and Coutre.
Definition Random.h:83
@ GFSR4
A fifth-order multiple recursive generator by L'Ecuyer, Blouin, and Coutre.
Definition Random.h:90
@ RANLXS1
Second-generation version of the RANLUX algorithm of Lüscher, 24-bit output, luxury level 1 (stronger...
Definition Random.h:68
@ RANLXD2
Double precision (48-bit) output using the RANLXS algorithm, luxury level 2 (strongest).
Definition Random.h:75
@ RANLXS0
Second-generation version of the RANLUX algorithm of Lüscher, 24-bit output, luxury level 0 (weakest)
Definition Random.h:65
@ CMRG
Combined multiple recursive generator by L'Ecuyer.
Definition Random.h:81
@ TAUS2
A maximally equidistributed combined Tausworthe generator by L'Ecuyer with improved seeding relative ...
Definition Random.h:88
@ RANLUX389
Original version of the RANLUX algorithm, 24-bit output (all bits are decorrelated).
Definition Random.h:79
void wrapRandom(lsst::utils::python::WrapperCollection &)
Definition _random.cc:103
void declareRandomImage(lsst::utils::python::WrapperCollection &wrappers)
Definition _random.cc:38
void declareRandom(lsst::utils::python::WrapperCollection &wrappers)
Definition _random.cc:52