From a6e6a369f40d8fb6a191fd7f74fc5efa8bfae2a0 Mon Sep 17 00:00:00 2001
From: Paul Baltescu <pauldb89@gmail.com>
Date: Wed, 27 Nov 2013 14:33:36 +0000
Subject: Unify sampling backoff strategy.

---
 extractor/phrase_location_sampler.h | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 extractor/phrase_location_sampler.h

(limited to 'extractor/phrase_location_sampler.h')

diff --git a/extractor/phrase_location_sampler.h b/extractor/phrase_location_sampler.h
new file mode 100644
index 00000000..0e88335e
--- /dev/null
+++ b/extractor/phrase_location_sampler.h
@@ -0,0 +1,35 @@
+#ifndef _PHRASE_LOCATION_SAMPLER_H_
+#define _PHRASE_LOCATION_SAMPLER_H_
+
+#include <memory>
+
+#include "sampler.h"
+
+namespace extractor {
+
+class MatchingsSampler;
+class PhraseLocation;
+class SuffixArray;
+class SuffixArrayRangeSampler;
+
+class PhraseLocationSampler : public Sampler {
+ public:
+  PhraseLocationSampler(shared_ptr<SuffixArray> suffix_array, int max_samples);
+
+  // For testing only.
+  PhraseLocationSampler(
+      shared_ptr<MatchingsSampler> matchings_sampler,
+      shared_ptr<SuffixArrayRangeSampler> suffix_array_sampler);
+
+  PhraseLocation Sample(
+      const PhraseLocation& location,
+      const unordered_set<int>& blacklisted_sentence_ids) const;
+
+ private:
+  shared_ptr<MatchingsSampler> matchings_sampler;
+  shared_ptr<SuffixArrayRangeSampler> suffix_array_sampler;
+};
+
+} // namespace extractor
+
+#endif
-- 
cgit v1.2.3