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
|
#ifndef FREEBAYES_EWENS_H
#define FREEBAYES_EWENS_H
#include <map>
#include <cmath>
#include "Utility.h"
using namespace std;
// genotype priors
long double alleleFrequencyProbability(const map<int, int>& alleleFrequencyCounts, long double theta);
long double alleleFrequencyProbabilityln(const map<int, int>& alleleFrequencyCounts, long double theta);
long double impl_alleleFrequencyProbabilityln(const map<int, int>& alleleFrequencyCounts, long double theta);
class AlleleFrequencyProbabilityCache : public map<map<int, int>, long double> {
public:
long double alleleFrequencyProbabilityln(const map<int, int>& counts, long double theta) {
map<map<int, int>, long double>::iterator p = find(counts);
if (p == end()) {
long double pln = impl_alleleFrequencyProbabilityln(counts, theta);
insert(make_pair(counts, pln));
return pln;
} else {
return p->second;
}
}
};
#endif
|