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.cc | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 extractor/phrase_location_sampler.cc

(limited to 'extractor/phrase_location_sampler.cc')

diff --git a/extractor/phrase_location_sampler.cc b/extractor/phrase_location_sampler.cc
new file mode 100644
index 00000000..a2eec105
--- /dev/null
+++ b/extractor/phrase_location_sampler.cc
@@ -0,0 +1,34 @@
+#include "phrase_location_sampler.h"
+
+#include "matchings_sampler.h"
+#include "phrase_location.h"
+#include "suffix_array.h"
+#include "suffix_array_sampler.h"
+
+namespace extractor {
+
+PhraseLocationSampler::PhraseLocationSampler(
+    shared_ptr<SuffixArray> suffix_array, int max_samples) {
+  matchings_sampler = make_shared<MatchingsSampler>(
+      suffix_array->GetData(), max_samples);
+  suffix_array_sampler = make_shared<SuffixArrayRangeSampler>(
+      suffix_array, max_samples);
+}
+
+PhraseLocationSampler::PhraseLocationSampler(
+    shared_ptr<MatchingsSampler> matchings_sampler,
+    shared_ptr<SuffixArrayRangeSampler> suffix_array_sampler) :
+    matchings_sampler(matchings_sampler),
+    suffix_array_sampler(suffix_array_sampler) {}
+
+PhraseLocation PhraseLocationSampler::Sample(
+    const PhraseLocation& location,
+    const unordered_set<int>& blacklisted_sentence_ids) const {
+  if (location.matchings == NULL) {
+    return suffix_array_sampler->Sample(location, blacklisted_sentence_ids);
+  } else {
+    return matchings_sampler->Sample(location, blacklisted_sentence_ids);
+  }
+}
+
+} // namespace extractor
-- 
cgit v1.2.3