summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/math/tools/hypergeometric_dist_data.cpp
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
commit483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch)
treee5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/boost/libs/math/tools/hypergeometric_dist_data.cpp
parentInitial commit. (diff)
downloadceph-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.cpp95
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;
+}
+