1 // (C) Copyright John Maddock 2009.
2 // Use, modification and distribution are subject to the
3 // Boost Software License, Version 1.0. (See accompanying file
4 // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
5
6 //#define BOOST_MATH_INSTRUMENT
7
8 #include <boost/math/distributions/hypergeometric.hpp>
9 #include <boost/math/special_functions/trunc.hpp>
10 #include <boost/math/constants/constants.hpp>
11 #include <boost/lexical_cast.hpp>
12 #include <boost/random/uniform_int.hpp>
13 #include <fstream>
14 #include <boost/math/tools/test_data.hpp>
15 #include "mp_t.hpp"
16
17 using namespace boost::math::tools;
18
19 std::mt19937 rnd;
20
21 struct hypergeometric_generator
22 {
23 boost::math::tuple<
24 mp_t,
25 mp_t,
26 mp_t,
27 mp_t,
28 mp_t,
29 mp_t,
operator ()hypergeometric_generator30 mp_t> operator()(mp_t rN, mp_t rr, mp_t rn)
31 {
32 using namespace std;
33 using namespace boost;
34 using namespace boost::math;
35
36 if((rr > rN) || (rr < rn))
37 throw std::domain_error("");
38
39 try{
40 int N = itrunc(rN);
41 int r = itrunc(rr);
42 int n = itrunc(rn);
43 boost::uniform_int<> ui((std::max)(0, n + r - N), (std::min)(n, r));
44 int k = ui(rnd);
45
46 hypergeometric_distribution<mp_t> d(r, n, N);
47
48 mp_t p = pdf(d, k);
49 if((p == 1) || (p == 0))
50 {
51 // trivial case, don't clutter up our table with it:
52 throw std::domain_error("");
53 }
54 mp_t c = cdf(d, k);
55 mp_t cc = cdf(complement(d, k));
56
57 std::cout << "N = " << N << " r = " << r << " n = " << n << " PDF = " << p << " CDF = " << c << " CCDF = " << cc << std::endl;
58
59 return boost::math::make_tuple(r, n, N, k, p, c, cc);
60 }
61 catch(const std::exception& e)
62 {
63 std::cout << e.what() << std::endl;
64 throw std::domain_error("");
65 }
66 }
67 };
68
main(int argc,char * argv[])69 int main(int argc, char*argv [])
70 {
71 std::string line;
72 parameter_info<mp_t> arg1, arg2, arg3;
73 test_data<mp_t> data;
74
75 std::cout << "Welcome.\n"
76 "This program will generate spot tests hypergeoemtric distribution:\n";
77
78 arg1 = make_power_param(mp_t(0), 1, 21);
79 arg2 = make_power_param(mp_t(0), 1, 21);
80 arg3 = make_power_param(mp_t(0), 1, 21);
81
82 arg1.type |= dummy_param;
83 arg2.type |= dummy_param;
84 arg3.type |= dummy_param;
85
86 data.insert(hypergeometric_generator(), arg1, arg2, arg3);
87
88 line = "hypergeometric_dist_data2.ipp";
89 std::ofstream ofs(line.c_str());
90 ofs << std::scientific << std::setprecision(40);
91 write_code(ofs, data, "hypergeometric_dist_data2");
92
93 return 0;
94 }
95
96