File: from_formula_threshold.cpp

package info (click to toggle)
isospec 2.3.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 12,472 kB
  • sloc: cpp: 9,530; python: 2,095; makefile: 180; ansic: 100; sh: 88
file content (77 lines) | stat: -rw-r--r-- 2,875 bytes parent folder | download
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
#include <iostream>
#include "isoSpec++.h"
#include <cassert>

using namespace IsoSpec;

#ifndef ISOSPEC_TESTS_SKIP_MAIN

size_t test_threshold(const char* formula, double threshold, bool print_confs);

int main(int argc, char** argv)
{
	if(argc < 3)
	{
		std::cout << "Proper usage (for example): ./from_formula_threshold C10000H1000O1000N1000 0.01" << std::endl;
		std::cout << "...will the configurations with probability above 0.01 for the above molecule" << std::endl;
                std::cout << "Proper usage (for example): ./from_formula_threshold C10000H1000O1000N1000 0.01 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);

	size_t no_visited = test_threshold(argv[1], atof(argv[2]), print_confs);

	std::cout << "The number of visited configurations is:" << no_visited << std::endl;

}
#endif /* ISOSPEC_TESTS_SKIP_MAIN */


size_t test_threshold(const char* formula, double threshold, bool print_confs)
{

	IsoThresholdGenerator i(formula, threshold, true, 100, 100, true);
        size_t confs_no = i.count_confs();
	//if(print_confs)
        //	std::cout << "No. confs is: " << confs_no << std::endl;
        i.reset();
        IsoThresholdGenerator i2(formula, threshold, true, 100, 100, true);
        IsoThresholdGenerator i3(formula, threshold, true, 100, 100, false);
        int* confspace = new int[i.getAllDim()];
        int* confspace2 = new int[i2.getAllDim()];
        //int* confspace3 = new int[i3.getAllDim()]; // these will be in different order...
        size_t no_visited = 0;
        [[maybe_unused]] double total_prob = 0.0;
	while(i.advanceToNextConfiguration())
	{
                assert(i2.advanceToNextConfiguration());
                assert(i3.advanceToNextConfiguration());
		if(print_confs)
                	std::cout << "lprob: " << i.lprob() << " prob: " << i.prob() << " log(prob): " << log(i.prob()) << " mass: " << i.mass() << " conf: ";
                assert(i.lprob() == i2.lprob());
                assert(i.mass() == i2.mass());
                assert(i.prob() == i2.prob());
                i.get_conf_signature(confspace);
                i2.get_conf_signature(confspace2);
                assert(memcmp(confspace, confspace2, i.getAllDim()*sizeof(int)) == 0);
		if(print_confs)
                	printArray<int>(confspace, i.getAllDim());
		no_visited += 1;
                total_prob += i.prob();
	}
        delete[] confspace;
        delete[] confspace2;
        assert(!i2.advanceToNextConfiguration());
        assert(!i3.advanceToNextConfiguration());
	assert(!i.advanceToNextConfiguration());
	assert(!i2.advanceToNextConfiguration());
	assert(!i3.advanceToNextConfiguration());
        assert(confs_no == no_visited);
	return no_visited;

}