Index: third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/dynamic_pt_updating_helper.h |
diff --git a/third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/dynamic_pt_updating_helper.h b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/dynamic_pt_updating_helper.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4be58282fd4c4a06b603151672613e1d410666f9 |
--- /dev/null |
+++ b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/dynamic_pt_updating_helper.h |
@@ -0,0 +1,95 @@ |
+/* |
+ * 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_DYNAMIC_PT_UPDATING_HELPER_H |
+#define LATINIME_DYNAMIC_PT_UPDATING_HELPER_H |
+ |
+#include "third_party/android_prediction/defines.h" |
+#include "third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/pt_node_params.h" |
+#include "third_party/android_prediction/utils/int_array_view.h" |
+ |
+namespace latinime { |
+ |
+class BigramProperty; |
+class BufferWithExtendableBuffer; |
+class DynamicPtReadingHelper; |
+class PtNodeReader; |
+class PtNodeWriter; |
+class UnigramProperty; |
+ |
+class DynamicPtUpdatingHelper { |
+ public: |
+ DynamicPtUpdatingHelper(BufferWithExtendableBuffer *const buffer, |
+ const PtNodeReader *const ptNodeReader, PtNodeWriter *const ptNodeWriter) |
+ : mBuffer(buffer), mPtNodeReader(ptNodeReader), mPtNodeWriter(ptNodeWriter) {} |
+ |
+ ~DynamicPtUpdatingHelper() {} |
+ |
+ // Add a word to the dictionary. If the word already exists, update the probability. |
+ bool addUnigramWord(DynamicPtReadingHelper *const readingHelper, |
+ const int *const wordCodePoints, const int codePointCount, |
+ const UnigramProperty *const unigramProperty, bool *const outAddedNewUnigram); |
+ |
+ // Add an n-gram entry. |
+ bool addNgramEntry(const PtNodePosArrayView prevWordsPtNodePos, const int wordPos, |
+ const BigramProperty *const bigramProperty, bool *const outAddedNewEntry); |
+ |
+ // Remove an n-gram entry. |
+ bool removeNgramEntry(const PtNodePosArrayView prevWordsPtNodePos, const int wordPos); |
+ |
+ // Add a shortcut target. |
+ bool addShortcutTarget(const int wordPos, const int *const targetCodePoints, |
+ const int targetCodePointCount, const int shortcutProbability); |
+ |
+ private: |
+ DISALLOW_IMPLICIT_CONSTRUCTORS(DynamicPtUpdatingHelper); |
+ |
+ static const int CHILDREN_POSITION_FIELD_SIZE; |
+ |
+ BufferWithExtendableBuffer *const mBuffer; |
+ const PtNodeReader *const mPtNodeReader; |
+ PtNodeWriter *const mPtNodeWriter; |
+ |
+ bool createAndInsertNodeIntoPtNodeArray(const int parentPos, const int *const nodeCodePoints, |
+ const int nodeCodePointCount, const UnigramProperty *const unigramProperty, |
+ int *const forwardLinkFieldPos); |
+ |
+ bool setPtNodeProbability(const PtNodeParams *const originalPtNodeParams, |
+ const UnigramProperty *const unigramProperty, bool *const outAddedNewUnigram); |
+ |
+ bool createChildrenPtNodeArrayAndAChildPtNode(const PtNodeParams *const parentPtNodeParams, |
+ const UnigramProperty *const unigramProperty, const int *const codePoints, |
+ const int codePointCount); |
+ |
+ bool createNewPtNodeArrayWithAChildPtNode(const int parentPos, const int *const nodeCodePoints, |
+ const int nodeCodePointCount, const UnigramProperty *const unigramProperty); |
+ |
+ bool reallocatePtNodeAndAddNewPtNodes( |
+ const PtNodeParams *const reallocatingPtNodeParams, const int overlappingCodePointCount, |
+ const UnigramProperty *const unigramProperty, const int *const newNodeCodePoints, |
+ const int newNodeCodePointCount); |
+ |
+ const PtNodeParams getUpdatedPtNodeParams(const PtNodeParams *const originalPtNodeParams, |
+ const bool isNotAWord, const bool isBlacklisted, const bool isTerminal, |
+ const int parentPos, const int codePointCount, |
+ const int *const codePoints, const int probability) const; |
+ |
+ const PtNodeParams getPtNodeParamsForNewPtNode(const bool isNotAWord, const bool isBlacklisted, |
+ const bool isTerminal, const int parentPos, |
+ const int codePointCount, const int *const codePoints, const int probability) const; |
+}; |
+} // namespace latinime |
+#endif /* LATINIME_DYNAMIC_PATRICIA_TRIE_UPDATING_HELPER_H */ |