blob: be4aa1bb21bb8533eaca9f9c3f28657ea3abd7ba (
plain)
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
|
#ifndef _SAMPLER_H_
#define _SAMPLER_H_
#include <memory>
using namespace std;
namespace extractor {
class PhraseLocation;
class SuffixArray;
/**
* Provides uniform sampling for a PhraseLocation.
*/
class Sampler {
public:
Sampler(shared_ptr<SuffixArray> suffix_array, int max_samples);
virtual ~Sampler();
// Samples uniformly at most max_samples phrase occurrences.
virtual PhraseLocation Sample(const PhraseLocation& location) const;
protected:
Sampler();
private:
// Round floating point number to the nearest integer.
int Round(double x) const;
shared_ptr<SuffixArray> suffix_array;
int max_samples;
};
} // namespace extractor
#endif
|