]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // (C) Copyright John Maddock 2006. |
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 | #include <boost/math/special_functions/gamma.hpp> | |
7 | #include <boost/math/constants/constants.hpp> | |
8 | #include <boost/math/tools/test_data.hpp> | |
9 | #include <fstream> | |
10 | #include "mp_t.hpp" | |
11 | ||
12 | using namespace boost::math::tools; | |
13 | ||
14 | struct beta_data_generator | |
15 | { | |
16 | mp_t operator()(mp_t a, mp_t b) | |
17 | { | |
18 | if(a < b) | |
19 | throw std::domain_error(""); | |
20 | // very naively calculate spots: | |
21 | mp_t g1, g2, g3; | |
22 | int s1, s2, s3; | |
23 | g1 = boost::math::lgamma(a, &s1); | |
24 | g2 = boost::math::lgamma(b, &s2); | |
25 | g3 = boost::math::lgamma(a+b, &s3); | |
26 | g1 += g2 - g3; | |
27 | g1 = exp(g1); | |
28 | g1 *= s1 * s2 * s3; | |
29 | return g1; | |
30 | } | |
31 | }; | |
32 | ||
33 | ||
34 | int main() | |
35 | { | |
36 | parameter_info<mp_t> arg1, arg2; | |
37 | test_data<mp_t> data; | |
38 | ||
39 | std::cout << "Welcome.\n" | |
40 | "This program will generate spot tests for the beta function:\n" | |
41 | " beta(a, b)\n\n"; | |
42 | ||
43 | bool cont; | |
44 | std::string line; | |
45 | ||
46 | do{ | |
47 | get_user_parameter_info(arg1, "a"); | |
48 | get_user_parameter_info(arg2, "b"); | |
49 | data.insert(beta_data_generator(), arg1, arg2); | |
50 | ||
51 | std::cout << "Any more data [y/n]?"; | |
52 | std::getline(std::cin, line); | |
53 | boost::algorithm::trim(line); | |
54 | cont = (line == "y"); | |
55 | }while(cont); | |
56 | ||
57 | std::cout << "Enter name of test data file [default=beta_data.ipp]"; | |
58 | std::getline(std::cin, line); | |
59 | boost::algorithm::trim(line); | |
60 | if(line == "") | |
61 | line = "beta_data.ipp"; | |
62 | std::ofstream ofs(line.c_str()); | |
63 | ofs << std::scientific << std::setprecision(40); | |
64 | write_code(ofs, data, "beta_data"); | |
65 | ||
66 | return 0; | |
67 | } | |
68 |