Add spellcheck and word suggestion to the prediction service

third_party/prediction/suggest/core/session/dic_traverse_session.cpp

Index: third_party/prediction/suggest/core/session/dic_traverse_session.cpp
diff --git a/third_party/prediction/suggest/core/session/dic_traverse_session.cpp b/third_party/prediction/suggest/core/session/dic_traverse_session.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..3d236f875ffcf10b06fccea932d1bd7e3000bb76
--- /dev/null
+++ b/third_party/prediction/suggest/core/session/dic_traverse_session.cpp
@@ -0,0 +1,93 @@
+/*
+ * Copyright (C) 2012 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.
+ */
+
+#include "third_party/prediction/suggest/core/session/dic_traverse_session.h"
+#include "third_party/prediction/defines.h"
+#include "third_party/prediction/suggest/core/dictionary/dictionary.h"
+#include "third_party/prediction/suggest/core/policy/dictionary_header_structure_policy.h"
+#include "third_party/prediction/suggest/core/policy/dictionary_structure_with_buffer_policy.h"
+#include "third_party/prediction/suggest/core/session/prev_words_info.h"
+
+namespace latinime {
+
+// 256K bytes threshold is heuristically used to distinguish dictionaries
+// containing many unigrams
+// (e.g. main dictionary) from small dictionaries (e.g. contacts...)
+const int DicTraverseSession::
+    DICTIONARY_SIZE_THRESHOLD_TO_USE_LARGE_CACHE_FOR_SUGGESTION = 256 * 1024;
+
+void DicTraverseSession::init(const Dictionary* const dictionary,
+                              const PrevWordsInfo* const prevWordsInfo,
+                              const SuggestOptions* const suggestOptions) {
+  mDictionary = dictionary;
+  mMultiWordCostMultiplier = getDictionaryStructurePolicy()
+                                 ->getHeaderStructurePolicy()
+                                 ->getMultiWordCostMultiplier();
+  mSuggestOptions = suggestOptions;
+  prevWordsInfo->getPrevWordsTerminalPtNodePos(getDictionaryStructurePolicy(),
+                                               mPrevWordsPtNodePos,
+                                               true /* tryLowerCaseSearch */);
+}
+
+void DicTraverseSession::setupForGetSuggestions(const ProximityInfo* pInfo,
+                                                const int* inputCodePoints,
+                                                const int inputSize,
+                                                const int* const inputXs,
+                                                const int* const inputYs,
+                                                const int* const times,
+                                                const int* const pointerIds,
+                                                const float maxSpatialDistance,
+                                                const int maxPointerCount) {
+  mProximityInfo = pInfo;
+  mMaxPointerCount = maxPointerCount;
+  initializeProximityInfoStates(inputCodePoints, inputXs, inputYs, times,
+                                pointerIds, inputSize, maxSpatialDistance,
+                                maxPointerCount);
+}
+
+const DictionaryStructureWithBufferPolicy*
+DicTraverseSession::getDictionaryStructurePolicy() const {
+  return mDictionary->getDictionaryStructurePolicy();
+}
+
+void DicTraverseSession::resetCache(const int thresholdForNextActiveDicNodes,
+                                    const int maxWords) {
+  mDicNodesCache.reset(thresholdForNextActiveDicNodes /* nextActiveSize */,
+                       maxWords /* terminalSize */);
+  mMultiBigramMap.clear();
+}
+
+void DicTraverseSession::initializeProximityInfoStates(
+    const int* const inputCodePoints,
+    const int* const inputXs,
+    const int* const inputYs,
+    const int* const times,
+    const int* const pointerIds,
+    const int inputSize,
+    const float maxSpatialDistance,
+    const int maxPointerCount) {
+  ASSERT(1 <= maxPointerCount && maxPointerCount <= MAX_POINTER_COUNT_G);
+  mInputSize = 0;
+  for (int i = 0; i < maxPointerCount; ++i) {
+    mProximityInfoStates[i].initInputParams(
+        i, maxSpatialDistance, getProximityInfo(), inputCodePoints, inputSize,
+        inputXs, inputYs, times, pointerIds,
+        maxPointerCount == MAX_POINTER_COUNT_G
+        /* TODO: this is a hack. fix proximity info state */);
+    mInputSize += mProximityInfoStates[i].size();
+  }
+}
+}  // namespace latinime