diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
commit | 483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch) | |
tree | e5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/boost/libs/math/tools/hypergeometric_dist_data.cpp | |
parent | Initial commit. (diff) | |
download | ceph-upstream.tar.xz ceph-upstream.zip |
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/boost/libs/math/tools/hypergeometric_dist_data.cpp')
-rw-r--r-- | src/boost/libs/math/tools/hypergeometric_dist_data.cpp | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/src/boost/libs/math/tools/hypergeometric_dist_data.cpp b/src/boost/libs/math/tools/hypergeometric_dist_data.cpp new file mode 100644 index 00000000..55352f03 --- /dev/null +++ b/src/boost/libs/math/tools/hypergeometric_dist_data.cpp @@ -0,0 +1,95 @@ +// (C) Copyright John Maddock 2009. +// Use, modification and distribution are subject to the +// Boost Software License, Version 1.0. (See accompanying file +// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) + +//#define BOOST_MATH_INSTRUMENT + +#include <boost/math/distributions/hypergeometric.hpp> +#include <boost/math/special_functions/trunc.hpp> +#include <boost/math/constants/constants.hpp> +#include <boost/lexical_cast.hpp> +#include <boost/random/uniform_int.hpp> +#include <fstream> +#include <boost/math/tools/test_data.hpp> +#include "mp_t.hpp" + +using namespace boost::math::tools; + +std::mt19937 rnd; + +struct hypergeometric_generator +{ + boost::math::tuple< + mp_t, + mp_t, + mp_t, + mp_t, + mp_t, + mp_t, + mp_t> operator()(mp_t rN, mp_t rr, mp_t rn) + { + using namespace std; + using namespace boost; + using namespace boost::math; + + if((rr > rN) || (rr < rn)) + throw std::domain_error(""); + + try{ + int N = itrunc(rN); + int r = itrunc(rr); + int n = itrunc(rn); + boost::uniform_int<> ui((std::max)(0, n + r - N), (std::min)(n, r)); + int k = ui(rnd); + + hypergeometric_distribution<mp_t> d(r, n, N); + + mp_t p = pdf(d, k); + if((p == 1) || (p == 0)) + { + // trivial case, don't clutter up our table with it: + throw std::domain_error(""); + } + mp_t c = cdf(d, k); + mp_t cc = cdf(complement(d, k)); + + std::cout << "N = " << N << " r = " << r << " n = " << n << " PDF = " << p << " CDF = " << c << " CCDF = " << cc << std::endl; + + return boost::math::make_tuple(r, n, N, k, p, c, cc); + } + catch(const std::exception& e) + { + std::cout << e.what() << std::endl; + throw std::domain_error(""); + } + } +}; + +int main(int argc, char*argv []) +{ + std::string line; + parameter_info<mp_t> arg1, arg2, arg3; + test_data<mp_t> data; + + std::cout << "Welcome.\n" + "This program will generate spot tests hypergeoemtric distribution:\n"; + + arg1 = make_power_param(mp_t(0), 1, 21); + arg2 = make_power_param(mp_t(0), 1, 21); + arg3 = make_power_param(mp_t(0), 1, 21); + + arg1.type |= dummy_param; + arg2.type |= dummy_param; + arg3.type |= dummy_param; + + data.insert(hypergeometric_generator(), arg1, arg2, arg3); + + line = "hypergeometric_dist_data2.ipp"; + std::ofstream ofs(line.c_str()); + ofs << std::scientific << std::setprecision(40); + write_code(ofs, data, "hypergeometric_dist_data2"); + + return 0; +} + |