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

(limited to 'extractor/suffix_array_sampler.cc')

diff --git a/extractor/suffix_array_sampler.cc b/extractor/suffix_array_sampler.cc
new file mode 100644
index 00000000..4a4ced34
--- /dev/null
+++ b/extractor/suffix_array_sampler.cc
@@ -0,0 +1,40 @@
+#include "suffix_array_sampler.h"
+
+#include "data_array.h"
+#include "phrase_location.h"
+#include "suffix_array.h"
+
+namespace extractor {
+
+SuffixArrayRangeSampler::SuffixArrayRangeSampler(
+    shared_ptr<SuffixArray> source_suffix_array, int max_samples) :
+    BackoffSampler(source_suffix_array->GetData(), max_samples),
+    source_suffix_array(source_suffix_array) {}
+
+SuffixArrayRangeSampler::SuffixArrayRangeSampler() {}
+
+int SuffixArrayRangeSampler::GetNumSubpatterns(const PhraseLocation&) const {
+  return 1;
+}
+
+int SuffixArrayRangeSampler::GetRangeLow(
+    const PhraseLocation& location) const {
+  return location.sa_low;
+}
+
+int SuffixArrayRangeSampler::GetRangeHigh(
+    const PhraseLocation& location) const {
+  return location.sa_high;
+}
+
+int SuffixArrayRangeSampler::GetPosition(
+    const PhraseLocation&, int position) const {
+  return source_suffix_array->GetSuffix(position);
+}
+
+void SuffixArrayRangeSampler::AppendMatching(
+    vector<int>& samples, int index, const PhraseLocation&) const {
+  samples.push_back(source_suffix_array->GetSuffix(index));
+}
+
+} // namespace extractor
-- 
cgit v1.2.3