Index: third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h |
diff --git a/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7e03322651c0e0a51b30b1d4c15095b94e214bbb |
--- /dev/null |
+++ b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h |
@@ -0,0 +1,60 @@ |
+/* |
+ * Copyright (C) 2013 The Android Open Source Project |
+ * |
+ * Licensed under the Apache License, Version 2.0 (the "License"); |
+ * you may not use this file except in compliance with the License. |
+ * You may obtain a copy of the License at |
+ * |
+ * http://www.apache.org/licenses/LICENSE-2.0 |
+ * |
+ * Unless required by applicable law or agreed to in writing, software |
+ * distributed under the License is distributed on an "AS IS" BASIS, |
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
+ * See the License for the specific language governing permissions and |
+ * limitations under the License. |
+ */ |
+ |
+#ifndef LATINIME_BIGRAM_LIST_POLICY_H |
+#define LATINIME_BIGRAM_LIST_POLICY_H |
+ |
+#include <cstdint> |
+ |
+#include "third_party/android_prediction/defines.h" |
+#include "third_party/android_prediction/suggest/core/policy/dictionary_bigrams_structure_policy.h" |
+#include "third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/bigram/bigram_list_read_write_utils.h" |
+ |
+namespace latinime { |
+ |
+class BigramListPolicy : public DictionaryBigramsStructurePolicy { |
+ public: |
+ BigramListPolicy(const uint8_t *const bigramsBuf, const int bufSize) |
+ : mBigramsBuf(bigramsBuf), mBufSize(bufSize) {} |
+ |
+ ~BigramListPolicy() {} |
+ |
+ void getNextBigram(int *const outBigramPos, int *const outProbability, bool *const outHasNext, |
+ int *const pos) const { |
+ BigramListReadWriteUtils::BigramFlags flags; |
+ if (!BigramListReadWriteUtils::getBigramEntryPropertiesAndAdvancePosition(mBigramsBuf, |
+ mBufSize, &flags, outBigramPos, pos)) { |
+ AKLOGE("Cannot read bigram entry. mBufSize: %d, pos: %d. ", mBufSize, *pos); |
+ *outProbability = NOT_A_PROBABILITY; |
+ *outHasNext = false; |
+ return; |
+ } |
+ *outProbability = BigramListReadWriteUtils::getProbabilityFromFlags(flags); |
+ *outHasNext = BigramListReadWriteUtils::hasNext(flags); |
+ } |
+ |
+ bool skipAllBigrams(int *const pos) const { |
+ return BigramListReadWriteUtils::skipExistingBigrams(mBigramsBuf, mBufSize, pos); |
+ } |
+ |
+ private: |
+ DISALLOW_IMPLICIT_CONSTRUCTORS(BigramListPolicy); |
+ |
+ const uint8_t *const mBigramsBuf; |
+ const int mBufSize; |
+}; |
+} // namespace latinime |
+#endif // LATINIME_BIGRAM_LIST_POLICY_H |