1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
// (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/bindings/rr.hpp>
#include <boost/test/included/test_exec_monitor.hpp>
#include <boost/math/distributions/hypergeometric.hpp>
#include <boost/math/special_functions/trunc.hpp>
#include <boost/math/constants/constants.hpp>
#include <boost/math/tools/test.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/random/uniform_int.hpp>
#include <fstream>
#include <boost/math/tools/test_data.hpp>
using namespace boost::math::tools;
std::tr1::mt19937 rnd;
struct hypergeometric_generator
{
std::tr1::tuple<
boost::math::ntl::RR,
boost::math::ntl::RR,
boost::math::ntl::RR,
boost::math::ntl::RR,
boost::math::ntl::RR,
boost::math::ntl::RR,
boost::math::ntl::RR> operator()(boost::math::ntl::RR rN, boost::math::ntl::RR rr, boost::math::ntl::RR 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);
uniform_int<> ui((std::max)(0, n + r - N), (std::min)(n, r));
int k = ui(rnd);
hypergeometric_distribution<ntl::RR> d(r, n, N);
ntl::RR p = pdf(d, k);
if((p == 1) || (p == 0))
{
// trivial case, don't clutter up our table with it:
throw std::domain_error("");
}
ntl::RR c = cdf(d, k);
ntl::RR cc = cdf(complement(d, k));
std::cout << "N = " << N << " r = " << r << " n = " << n << " PDF = " << p << " CDF = " << c << " CCDF = " << cc << std::endl;
return tr1::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 test_main(int argc, char*argv [])
{
boost::math::ntl::RR::SetPrecision(1000);
boost::math::ntl::RR::SetOutputPrecision(100);
std::string line;
parameter_info<boost::math::ntl::RR> arg1, arg2, arg3;
test_data<boost::math::ntl::RR> data;
std::cout << "Welcome.\n"
"This program will generate spot tests hypergeoemtric distribution:\n";
arg1 = make_power_param(boost::math::ntl::RR(0), 1, 21);
arg2 = make_power_param(boost::math::ntl::RR(0), 1, 21);
arg3 = make_power_param(boost::math::ntl::RR(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());
write_code(ofs, data, "hypergeometric_dist_data2");
return 0;
}
|
