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
|
#include <iostream>
#include <cassert>
#include <sstream>
#define private public // MWAHAHAHAHA!!!
#include "../../IsoSpec++/unity-build.cpp"
using namespace IsoSpec;
#ifndef ISOSPEC_TESTS_SKIP_MAIN
size_t test_ordered(const char* formula, double total_prob, bool print_confs = false);
int main(int argc, char** argv)
{
if(argc < 3)
{
std::cout << "Proper usage (for example): " << argv[0] << " C10000H1000O1000N1000 0.9999" << std::endl;
std::cout << "...will print the minimal number of configurations necessary to cover 0.9999 probability of the above molecule" << std::endl;
std::cout << argv[0] << " C10000H1000O1000N1000 0.9999 false" << std::endl;
std::cout << "will just count them" << std::endl;
return -1;
}
bool print_confs = true;
if(argc > 3)
print_confs = (strcmp(argv[3], "true") == 0);
#ifndef ISOSPEC_TESTS_MEMSAN
size_t no_confs =
#endif
test_ordered(argv[1], atof(argv[2]), print_confs);
#ifndef ISOSPEC_TESTS_MEMSAN
std::cout << "The number of visited configurations is: " << no_confs << std::endl;
#endif
}
#endif /* ISOSPEC_TESTS_SKIP_MAIN */
size_t test_ordered(const char* formula, double total_prob, bool print_confs)
{
IsoOrderedGenerator i(formula);
MarginalTrek& MT = *(i.marginalResults[0]);
double target_prob = total_prob;
size_t no_visited = 0;
while(target_prob > 0.0 && MT.add_next_conf())
{
double curr_p = exp(MT._conf_lprobs[no_visited]);
target_prob -= curr_p;
no_visited += 1;
}
return no_visited;
}
|
