satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 1 | /* |
| 2 | ** |
| 3 | ** Copyright 2010, The Android Open Source Project |
| 4 | ** |
| 5 | ** Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | ** you may not use this file except in compliance with the License. |
| 7 | ** You may obtain a copy of the License at |
| 8 | ** |
| 9 | ** http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | ** |
| 11 | ** Unless required by applicable law or agreed to in writing, software |
| 12 | ** distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | ** See the License for the specific language governing permissions and |
| 15 | ** limitations under the License. |
| 16 | */ |
| 17 | |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 18 | #include <assert.h> |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 19 | #include <fcntl.h> |
satok | f5cded1 | 2010-12-06 21:28:24 +0900 | [diff] [blame] | 20 | #include <stdio.h> |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 21 | #include <string.h> |
| 22 | |
satok | e808e43 | 2010-12-02 14:53:24 +0900 | [diff] [blame] | 23 | #define LOG_TAG "LatinIME: unigram_dictionary.cpp" |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 24 | |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 25 | #include "basechars.h" |
| 26 | #include "char_utils.h" |
satok | e808e43 | 2010-12-02 14:53:24 +0900 | [diff] [blame] | 27 | #include "dictionary.h" |
| 28 | #include "unigram_dictionary.h" |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 29 | |
| 30 | namespace latinime { |
| 31 | |
satok | e808e43 | 2010-12-02 14:53:24 +0900 | [diff] [blame] | 32 | UnigramDictionary::UnigramDictionary(const unsigned char *dict, int typedLetterMultiplier, |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 33 | int fullWordMultiplier, int maxWordLength, int maxWords, int maxProximityChars, |
satok | 18c28f4 | 2010-12-02 18:11:54 +0900 | [diff] [blame] | 34 | const bool isLatestDictVersion) |
Tadashi G. Takaoka | 887f11e | 2011-02-10 20:53:58 +0900 | [diff] [blame^] | 35 | : DICT(dict), MAX_WORD_LENGTH(maxWordLength), MAX_WORDS(maxWords), |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 36 | MAX_PROXIMITY_CHARS(maxProximityChars), IS_LATEST_DICT_VERSION(isLatestDictVersion), |
| 37 | TYPED_LETTER_MULTIPLIER(typedLetterMultiplier), FULL_WORD_MULTIPLIER(fullWordMultiplier), |
| 38 | ROOT_POS(isLatestDictVersion ? DICTIONARY_HEADER_SIZE : 0) { |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 39 | if (DEBUG_DICT) LOGI("UnigramDictionary - constructor"); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 40 | } |
| 41 | |
satok | 18c28f4 | 2010-12-02 18:11:54 +0900 | [diff] [blame] | 42 | UnigramDictionary::~UnigramDictionary() {} |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 43 | |
satok | 18c28f4 | 2010-12-02 18:11:54 +0900 | [diff] [blame] | 44 | int UnigramDictionary::getSuggestions(int *codes, int codesSize, unsigned short *outWords, |
Tadashi G. Takaoka | 887f11e | 2011-02-10 20:53:58 +0900 | [diff] [blame^] | 45 | int *frequencies) { |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 46 | PROF_OPEN; |
| 47 | PROF_START(0); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 48 | initSuggestions(codes, codesSize, outWords, frequencies); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 49 | if (DEBUG_DICT) assert(codesSize == mInputLength); |
| 50 | |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 51 | const int MAX_DEPTH = min(mInputLength * MAX_DEPTH_MULTIPLIER, MAX_WORD_LENGTH); |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 52 | PROF_END(0); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 53 | |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 54 | PROF_START(1); |
Tadashi G. Takaoka | 887f11e | 2011-02-10 20:53:58 +0900 | [diff] [blame^] | 55 | getSuggestionCandidates(-1, -1, -1, mNextLettersFrequency, NEXT_LETTERS_SIZE, MAX_DEPTH); |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 56 | PROF_END(1); |
| 57 | |
| 58 | PROF_START(2); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 59 | // Suggestion with missing character |
| 60 | if (SUGGEST_WORDS_WITH_MISSING_CHARACTER) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 61 | for (int i = 0; i < codesSize; ++i) { |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 62 | if (DEBUG_DICT) LOGI("--- Suggest missing characters %d", i); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 63 | getSuggestionCandidates(i, -1, -1, NULL, 0, MAX_DEPTH); |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 64 | } |
| 65 | } |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 66 | PROF_END(2); |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 67 | |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 68 | PROF_START(3); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 69 | // Suggestion with excessive character |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 70 | if (SUGGEST_WORDS_WITH_EXCESSIVE_CHARACTER |
| 71 | && mInputLength >= MIN_USER_TYPED_LENGTH_FOR_EXCESSIVE_CHARACTER_SUGGESTION) { |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 72 | for (int i = 0; i < codesSize; ++i) { |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 73 | if (DEBUG_DICT) LOGI("--- Suggest excessive characters %d", i); |
| 74 | getSuggestionCandidates(-1, i, -1, NULL, 0, MAX_DEPTH); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 75 | } |
| 76 | } |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 77 | PROF_END(3); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 78 | |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 79 | PROF_START(4); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 80 | // Suggestion with transposed characters |
| 81 | // Only suggest words that length is mInputLength |
| 82 | if (SUGGEST_WORDS_WITH_TRANSPOSED_CHARACTERS) { |
| 83 | for (int i = 0; i < codesSize; ++i) { |
| 84 | if (DEBUG_DICT) LOGI("--- Suggest transposed characters %d", i); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 85 | getSuggestionCandidates(-1, -1, i, NULL, 0, mInputLength - 1); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 86 | } |
| 87 | } |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 88 | PROF_END(4); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 89 | |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 90 | PROF_START(5); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 91 | // Suggestions with missing space |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 92 | if (SUGGEST_WORDS_WITH_MISSING_SPACE_CHARACTER |
| 93 | && mInputLength >= MIN_USER_TYPED_LENGTH_FOR_MISSING_SPACE_SUGGESTION) { |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 94 | for (int i = 1; i < codesSize; ++i) { |
| 95 | if (DEBUG_DICT) LOGI("--- Suggest missing space characters %d", i); |
| 96 | getMissingSpaceWords(mInputLength, i); |
| 97 | } |
| 98 | } |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 99 | PROF_END(5); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 100 | |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 101 | PROF_START(6); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 102 | // Get the word count |
| 103 | int suggestedWordsCount = 0; |
| 104 | while (suggestedWordsCount < MAX_WORDS && mFrequencies[suggestedWordsCount] > 0) { |
| 105 | suggestedWordsCount++; |
| 106 | } |
| 107 | |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 108 | if (DEBUG_DICT) { |
| 109 | LOGI("Returning %d words", suggestedWordsCount); |
| 110 | LOGI("Next letters: "); |
Tadashi G. Takaoka | 887f11e | 2011-02-10 20:53:58 +0900 | [diff] [blame^] | 111 | for (int k = 0; k < NEXT_LETTERS_SIZE; k++) { |
| 112 | if (mNextLettersFrequency[k] > 0) { |
| 113 | LOGI("%c = %d,", k, mNextLettersFrequency[k]); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 114 | } |
| 115 | } |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 116 | } |
satok | 61e2f85 | 2011-01-05 14:13:07 +0900 | [diff] [blame] | 117 | PROF_END(6); |
| 118 | PROF_CLOSE; |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 119 | return suggestedWordsCount; |
| 120 | } |
| 121 | |
| 122 | void UnigramDictionary::initSuggestions(int *codes, int codesSize, unsigned short *outWords, |
| 123 | int *frequencies) { |
satok | f5cded1 | 2010-12-06 21:28:24 +0900 | [diff] [blame] | 124 | if (DEBUG_DICT) LOGI("initSuggest"); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 125 | mFrequencies = frequencies; |
| 126 | mOutputChars = outWords; |
| 127 | mInputCodes = codes; |
| 128 | mInputLength = codesSize; |
| 129 | mMaxEditDistance = mInputLength < 5 ? 2 : mInputLength / 2; |
| 130 | } |
| 131 | |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 132 | void UnigramDictionary::registerNextLetter( |
| 133 | unsigned short c, int *nextLetters, int nextLettersSize) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 134 | if (c < nextLettersSize) { |
| 135 | nextLetters[c]++; |
| 136 | } |
| 137 | } |
| 138 | |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 139 | // TODO: We need to optimize addWord by using STL or something |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 140 | bool UnigramDictionary::addWord(unsigned short *word, int length, int frequency) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 141 | word[length] = 0; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 142 | if (DEBUG_DICT && DEBUG_SHOW_FOUND_WORD) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 143 | char s[length + 1]; |
| 144 | for (int i = 0; i <= length; i++) s[i] = word[i]; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 145 | LOGI("Found word = %s, freq = %d", s, frequency); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 146 | } |
satok | f5cded1 | 2010-12-06 21:28:24 +0900 | [diff] [blame] | 147 | if (length > MAX_WORD_LENGTH) { |
| 148 | if (DEBUG_DICT) LOGI("Exceeded max word length."); |
| 149 | return false; |
| 150 | } |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 151 | |
| 152 | // Find the right insertion point |
| 153 | int insertAt = 0; |
| 154 | while (insertAt < MAX_WORDS) { |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 155 | if (frequency > mFrequencies[insertAt] || (mFrequencies[insertAt] == frequency |
| 156 | && length < Dictionary::wideStrLen(mOutputChars + insertAt * MAX_WORD_LENGTH))) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 157 | break; |
| 158 | } |
| 159 | insertAt++; |
| 160 | } |
| 161 | if (insertAt < MAX_WORDS) { |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 162 | if (DEBUG_DICT) { |
| 163 | char s[length + 1]; |
| 164 | for (int i = 0; i <= length; i++) s[i] = word[i]; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 165 | LOGI("Added word = %s, freq = %d", s, frequency); |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 166 | } |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 167 | memmove((char*) mFrequencies + (insertAt + 1) * sizeof(mFrequencies[0]), |
| 168 | (char*) mFrequencies + insertAt * sizeof(mFrequencies[0]), |
| 169 | (MAX_WORDS - insertAt - 1) * sizeof(mFrequencies[0])); |
| 170 | mFrequencies[insertAt] = frequency; |
| 171 | memmove((char*) mOutputChars + (insertAt + 1) * MAX_WORD_LENGTH * sizeof(short), |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 172 | (char*) mOutputChars + insertAt * MAX_WORD_LENGTH * sizeof(short), |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 173 | (MAX_WORDS - insertAt - 1) * sizeof(short) * MAX_WORD_LENGTH); |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 174 | unsigned short *dest = mOutputChars + insertAt * MAX_WORD_LENGTH; |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 175 | while (length--) { |
| 176 | *dest++ = *word++; |
| 177 | } |
| 178 | *dest = 0; // NULL terminate |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 179 | if (DEBUG_DICT) LOGI("Added word at %d", insertAt); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 180 | return true; |
| 181 | } |
| 182 | return false; |
| 183 | } |
| 184 | |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 185 | unsigned short UnigramDictionary::toLowerCase(unsigned short c) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 186 | if (c < sizeof(BASE_CHARS) / sizeof(BASE_CHARS[0])) { |
| 187 | c = BASE_CHARS[c]; |
| 188 | } |
| 189 | if (c >='A' && c <= 'Z') { |
| 190 | c |= 32; |
| 191 | } else if (c > 127) { |
| 192 | c = latin_tolower(c); |
| 193 | } |
| 194 | return c; |
| 195 | } |
| 196 | |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 197 | bool UnigramDictionary::sameAsTyped(unsigned short *word, int length) { |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 198 | if (length != mInputLength) { |
| 199 | return false; |
| 200 | } |
| 201 | int *inputCodes = mInputCodes; |
| 202 | while (length--) { |
| 203 | if ((unsigned int) *inputCodes != (unsigned int) *word) { |
| 204 | return false; |
| 205 | } |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 206 | inputCodes += MAX_PROXIMITY_CHARS; |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 207 | word++; |
| 208 | } |
| 209 | return true; |
| 210 | } |
| 211 | |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 212 | static const char QUOTE = '\''; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 213 | static const char SPACE = ' '; |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 214 | |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 215 | void UnigramDictionary::getSuggestionCandidates(const int skipPos, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 216 | const int excessivePos, const int transposedPos, int *nextLetters, |
| 217 | const int nextLettersSize, const int maxDepth) { |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 218 | if (DEBUG_DICT) { |
| 219 | LOGI("getSuggestionCandidates %d", maxDepth); |
| 220 | assert(transposedPos + 1 < mInputLength); |
| 221 | assert(excessivePos < mInputLength); |
| 222 | assert(missingPos < mInputLength); |
| 223 | } |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 224 | int rootPosition = ROOT_POS; |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 225 | // Get the number of child of root, then increment the position |
| 226 | int childCount = Dictionary::getCount(DICT, &rootPosition); |
| 227 | int depth = 0; |
| 228 | |
| 229 | mStackChildCount[0] = childCount; |
| 230 | mStackTraverseAll[0] = (mInputLength <= 0); |
| 231 | mStackNodeFreq[0] = 1; |
| 232 | mStackInputIndex[0] = 0; |
| 233 | mStackDiffs[0] = 0; |
| 234 | mStackSiblingPos[0] = rootPosition; |
| 235 | |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 236 | // Depth first search |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 237 | while (depth >= 0) { |
| 238 | if (mStackChildCount[depth] > 0) { |
| 239 | --mStackChildCount[depth]; |
| 240 | bool traverseAllNodes = mStackTraverseAll[depth]; |
| 241 | int snr = mStackNodeFreq[depth]; |
| 242 | int inputIndex = mStackInputIndex[depth]; |
| 243 | int diffs = mStackDiffs[depth]; |
| 244 | int siblingPos = mStackSiblingPos[depth]; |
| 245 | int firstChildPos; |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 246 | // depth will never be greater than maxDepth because in that case, |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 247 | // needsToTraverseChildrenNodes should be false |
| 248 | const bool needsToTraverseChildrenNodes = processCurrentNode(siblingPos, depth, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 249 | maxDepth, traverseAllNodes, snr, inputIndex, diffs, skipPos, excessivePos, |
| 250 | transposedPos, nextLetters, nextLettersSize, &childCount, &firstChildPos, |
| 251 | &traverseAllNodes, &snr, &inputIndex, &diffs, &siblingPos); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 252 | // Update next sibling pos |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 253 | mStackSiblingPos[depth] = siblingPos; |
| 254 | if (needsToTraverseChildrenNodes) { |
| 255 | // Goes to child node |
| 256 | ++depth; |
| 257 | mStackChildCount[depth] = childCount; |
| 258 | mStackTraverseAll[depth] = traverseAllNodes; |
| 259 | mStackNodeFreq[depth] = snr; |
| 260 | mStackInputIndex[depth] = inputIndex; |
| 261 | mStackDiffs[depth] = diffs; |
| 262 | mStackSiblingPos[depth] = firstChildPos; |
| 263 | } |
| 264 | } else { |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 265 | // Goes to parent sibling node |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 266 | --depth; |
| 267 | } |
| 268 | } |
| 269 | } |
| 270 | |
satok | f7425bb | 2011-01-05 16:37:53 +0900 | [diff] [blame] | 271 | inline static void multiplyRate(const int rate, int *freq) { |
| 272 | if (rate > 1000000) { |
| 273 | *freq = (*freq / 100) * rate; |
| 274 | } else { |
| 275 | *freq = *freq * rate / 100; |
| 276 | } |
| 277 | } |
| 278 | |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 279 | bool UnigramDictionary::getMissingSpaceWords(const int inputLength, const int missingSpacePos) { |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 280 | if (missingSpacePos <= 0 || missingSpacePos >= inputLength |
| 281 | || inputLength >= MAX_WORD_LENGTH) return false; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 282 | const int newWordLength = inputLength + 1; |
| 283 | // Allocating variable length array on stack |
| 284 | unsigned short word[newWordLength]; |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 285 | const int firstFreq = getBestWordFreq(0, missingSpacePos, mWord); |
| 286 | if (DEBUG_DICT) LOGI("First freq: %d", firstFreq); |
| 287 | if (firstFreq <= 0) return false; |
| 288 | |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 289 | for (int i = 0; i < missingSpacePos; ++i) { |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 290 | word[i] = mWord[i]; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 291 | } |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 292 | |
| 293 | const int secondFreq = getBestWordFreq(missingSpacePos, inputLength - missingSpacePos, mWord); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 294 | if (DEBUG_DICT) LOGI("Second freq: %d", secondFreq); |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 295 | if (secondFreq <= 0) return false; |
| 296 | |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 297 | word[missingSpacePos] = SPACE; |
| 298 | for (int i = (missingSpacePos + 1); i < newWordLength; ++i) { |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 299 | word[i] = mWord[i - missingSpacePos - 1]; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 300 | } |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 301 | |
| 302 | int pairFreq = ((firstFreq + secondFreq) / 2); |
| 303 | for (int i = 0; i < inputLength; ++i) pairFreq *= TYPED_LETTER_MULTIPLIER; |
satok | f7425bb | 2011-01-05 16:37:53 +0900 | [diff] [blame] | 304 | multiplyRate(WORDS_WITH_MISSING_SPACE_CHARACTER_DEMOTION_RATE, &pairFreq); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 305 | addWord(word, newWordLength, pairFreq); |
| 306 | return true; |
| 307 | } |
| 308 | |
| 309 | // Keep this for comparing spec to new getWords |
| 310 | void UnigramDictionary::getWordsOld(const int initialPos, const int inputLength, const int skipPos, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 311 | const int excessivePos, const int transposedPos,int *nextLetters, |
| 312 | const int nextLettersSize) { |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 313 | int initialPosition = initialPos; |
| 314 | const int count = Dictionary::getCount(DICT, &initialPosition); |
| 315 | getWordsRec(count, initialPosition, 0, |
| 316 | min(inputLength * MAX_DEPTH_MULTIPLIER, MAX_WORD_LENGTH), |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 317 | mInputLength <= 0, 1, 0, 0, skipPos, excessivePos, transposedPos, nextLetters, |
| 318 | nextLettersSize); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 319 | } |
| 320 | |
satok | 6831926 | 2010-12-03 19:38:08 +0900 | [diff] [blame] | 321 | void UnigramDictionary::getWordsRec(const int childrenCount, const int pos, const int depth, |
| 322 | const int maxDepth, const bool traverseAllNodes, const int snr, const int inputIndex, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 323 | const int diffs, const int skipPos, const int excessivePos, const int transposedPos, |
| 324 | int *nextLetters, const int nextLettersSize) { |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 325 | int siblingPos = pos; |
satok | 6831926 | 2010-12-03 19:38:08 +0900 | [diff] [blame] | 326 | for (int i = 0; i < childrenCount; ++i) { |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 327 | int newCount; |
| 328 | int newChildPosition; |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 329 | const int newDepth = depth + 1; |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 330 | bool newTraverseAllNodes; |
| 331 | int newSnr; |
| 332 | int newInputIndex; |
| 333 | int newDiffs; |
| 334 | int newSiblingPos; |
| 335 | const bool needsToTraverseChildrenNodes = processCurrentNode(siblingPos, depth, maxDepth, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 336 | traverseAllNodes, snr, inputIndex, diffs, skipPos, excessivePos, transposedPos, |
| 337 | nextLetters, nextLettersSize, |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 338 | &newCount, &newChildPosition, &newTraverseAllNodes, &newSnr, |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 339 | &newInputIndex, &newDiffs, &newSiblingPos); |
| 340 | siblingPos = newSiblingPos; |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 341 | |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 342 | if (needsToTraverseChildrenNodes) { |
| 343 | getWordsRec(newCount, newChildPosition, newDepth, maxDepth, newTraverseAllNodes, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 344 | newSnr, newInputIndex, newDiffs, skipPos, excessivePos, transposedPos, |
| 345 | nextLetters, nextLettersSize); |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 346 | } |
| 347 | } |
| 348 | } |
| 349 | |
satok | 58c49b9 | 2011-01-27 03:23:39 +0900 | [diff] [blame] | 350 | inline int UnigramDictionary::calculateFinalFreq(const int inputIndex, const int depth, |
| 351 | const int snr, const int skipPos, const int excessivePos, const int transposedPos, |
| 352 | const int freq, const bool sameLength) { |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 353 | // TODO: Demote by edit distance |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 354 | int finalFreq = freq * snr; |
satok | f7425bb | 2011-01-05 16:37:53 +0900 | [diff] [blame] | 355 | if (skipPos >= 0) multiplyRate(WORDS_WITH_MISSING_CHARACTER_DEMOTION_RATE, &finalFreq); |
| 356 | if (transposedPos >= 0) multiplyRate( |
| 357 | WORDS_WITH_TRANSPOSED_CHARACTERS_DEMOTION_RATE, &finalFreq); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 358 | if (excessivePos >= 0) { |
satok | f7425bb | 2011-01-05 16:37:53 +0900 | [diff] [blame] | 359 | multiplyRate(WORDS_WITH_EXCESSIVE_CHARACTER_DEMOTION_RATE, &finalFreq); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 360 | if (!existsAdjacentProximityChars(inputIndex, mInputLength)) { |
satok | f7425bb | 2011-01-05 16:37:53 +0900 | [diff] [blame] | 361 | multiplyRate(WORDS_WITH_EXCESSIVE_CHARACTER_OUT_OF_PROXIMITY_DEMOTION_RATE, &finalFreq); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 362 | } |
| 363 | } |
satok | 58c49b9 | 2011-01-27 03:23:39 +0900 | [diff] [blame] | 364 | int lengthFreq = TYPED_LETTER_MULTIPLIER; |
| 365 | for (int i = 0; i < depth; ++i) lengthFreq *= TYPED_LETTER_MULTIPLIER; |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 366 | if (lengthFreq == snr) { |
| 367 | if (depth > 1) { |
| 368 | if (DEBUG_DICT) LOGI("Found full matched word."); |
| 369 | multiplyRate(FULL_MATCHED_WORDS_PROMOTION_RATE, &finalFreq); |
| 370 | } |
| 371 | if (sameLength && transposedPos < 0 && skipPos < 0 && excessivePos < 0) { |
| 372 | finalFreq *= FULL_MATCH_ACCENTS_OR_CAPITALIZATION_DIFFER_MULTIPLIER; |
| 373 | } |
satok | 58c49b9 | 2011-01-27 03:23:39 +0900 | [diff] [blame] | 374 | } |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 375 | if (sameLength && skipPos < 0) finalFreq *= FULL_WORD_MULTIPLIER; |
| 376 | return finalFreq; |
| 377 | } |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 378 | |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 379 | inline void UnigramDictionary::onTerminalWhenUserTypedLengthIsGreaterThanInputLength( |
| 380 | unsigned short *word, const int inputIndex, const int depth, const int snr, |
| 381 | int *nextLetters, const int nextLettersSize, const int skipPos, const int excessivePos, |
| 382 | const int transposedPos, const int freq) { |
satok | 58c49b9 | 2011-01-27 03:23:39 +0900 | [diff] [blame] | 383 | const int finalFreq = calculateFinalFreq(inputIndex, depth, snr, skipPos, excessivePos, |
| 384 | transposedPos, freq, false); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 385 | if (depth >= MIN_SUGGEST_DEPTH) addWord(word, depth + 1, finalFreq); |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 386 | if (depth >= mInputLength && skipPos < 0) { |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 387 | registerNextLetter(mWord[mInputLength], nextLetters, nextLettersSize); |
| 388 | } |
| 389 | } |
| 390 | |
| 391 | inline void UnigramDictionary::onTerminalWhenUserTypedLengthIsSameAsInputLength( |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 392 | unsigned short *word, const int inputIndex, const int depth, const int snr, |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 393 | const int skipPos, const int excessivePos, const int transposedPos, const int freq) { |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 394 | if (sameAsTyped(word, depth + 1)) return; |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 395 | const int finalFreq = calculateFinalFreq(inputIndex, depth, snr, skipPos, |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 396 | excessivePos, transposedPos, freq, true); |
| 397 | // Proximity collection will promote a word of the same length as what user typed. |
| 398 | if (depth >= MIN_SUGGEST_DEPTH) addWord(word, depth + 1, finalFreq); |
satok | 715514d | 2010-12-02 20:19:59 +0900 | [diff] [blame] | 399 | } |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 400 | |
| 401 | inline bool UnigramDictionary::needsToSkipCurrentNode(const unsigned short c, |
satok | 6831926 | 2010-12-03 19:38:08 +0900 | [diff] [blame] | 402 | const int inputIndex, const int skipPos, const int depth) { |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 403 | const unsigned short userTypedChar = (mInputCodes + (inputIndex * MAX_PROXIMITY_CHARS))[0]; |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 404 | // Skip the ' or other letter and continue deeper |
| 405 | return (c == QUOTE && userTypedChar != QUOTE) || skipPos == depth; |
| 406 | } |
| 407 | |
satok | e07baa6 | 2010-12-09 21:55:40 +0900 | [diff] [blame] | 408 | inline bool UnigramDictionary::existsAdjacentProximityChars(const int inputIndex, |
| 409 | const int inputLength) { |
| 410 | if (inputIndex < 0 || inputIndex >= inputLength) return false; |
| 411 | const int currentChar = *getInputCharsAt(inputIndex); |
| 412 | const int leftIndex = inputIndex - 1; |
| 413 | if (leftIndex >= 0) { |
| 414 | int *leftChars = getInputCharsAt(leftIndex); |
| 415 | int i = 0; |
| 416 | while (leftChars[i] > 0 && i < MAX_PROXIMITY_CHARS) { |
| 417 | if (leftChars[i++] == currentChar) return true; |
| 418 | } |
| 419 | } |
| 420 | const int rightIndex = inputIndex + 1; |
| 421 | if (rightIndex < inputLength) { |
| 422 | int *rightChars = getInputCharsAt(rightIndex); |
| 423 | int i = 0; |
| 424 | while (rightChars[i] > 0 && i < MAX_PROXIMITY_CHARS) { |
| 425 | if (rightChars[i++] == currentChar) return true; |
| 426 | } |
| 427 | } |
| 428 | return false; |
| 429 | } |
| 430 | |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 431 | inline UnigramDictionary::ProximityType UnigramDictionary::getMatchedProximityId( |
| 432 | const int *currentChars, const unsigned short c, const int skipPos, |
| 433 | const int excessivePos, const int transposedPos) { |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 434 | const unsigned short lowerC = toLowerCase(c); |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 435 | int j = 0; |
satok | e07baa6 | 2010-12-09 21:55:40 +0900 | [diff] [blame] | 436 | while (currentChars[j] > 0 && j < MAX_PROXIMITY_CHARS) { |
satok | 6831926 | 2010-12-03 19:38:08 +0900 | [diff] [blame] | 437 | const bool matched = (currentChars[j] == lowerC || currentChars[j] == c); |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 438 | // If skipPos is defined, not to search proximity collections. |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 439 | // First char is what user typed. |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 440 | if (matched) { |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 441 | if (j > 0) return NEAR_PROXIMITY_CHAR; |
| 442 | return SAME_OR_ACCENTED_OR_CAPITALIZED_CHAR; |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 443 | } else if (skipPos >= 0 || excessivePos >= 0 || transposedPos >= 0) { |
| 444 | // Not to check proximity characters |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 445 | return UNRELATED_CHAR; |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 446 | } |
| 447 | ++j; |
| 448 | } |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 449 | return UNRELATED_CHAR; |
satok | 28bd03b | 2010-12-03 16:39:16 +0900 | [diff] [blame] | 450 | } |
| 451 | |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 452 | inline bool UnigramDictionary::processCurrentNode(const int pos, const int depth, |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 453 | const int maxDepth, const bool traverseAllNodes, int snr, int inputIndex, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 454 | const int diffs, const int skipPos, const int excessivePos, const int transposedPos, |
| 455 | int *nextLetters, const int nextLettersSize, int *newCount, int *newChildPosition, |
| 456 | bool *newTraverseAllNodes, int *newSnr, int*newInputIndex, int *newDiffs, |
| 457 | int *nextSiblingPosition) { |
| 458 | if (DEBUG_DICT) { |
| 459 | int inputCount = 0; |
| 460 | if (skipPos >= 0) ++inputCount; |
| 461 | if (excessivePos >= 0) ++inputCount; |
| 462 | if (transposedPos >= 0) ++inputCount; |
| 463 | assert(inputCount <= 1); |
| 464 | } |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 465 | unsigned short c; |
| 466 | int childPosition; |
| 467 | bool terminal; |
| 468 | int freq; |
satok | fd16f1d | 2011-01-27 16:25:16 +0900 | [diff] [blame] | 469 | bool isSameAsUserTypedLength = false; |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 470 | |
satok | fd16f1d | 2011-01-27 16:25:16 +0900 | [diff] [blame] | 471 | if (excessivePos == depth && inputIndex < mInputLength - 1) ++inputIndex; |
satok | cdbbea7 | 2010-12-08 16:04:16 +0900 | [diff] [blame] | 472 | |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 473 | *nextSiblingPosition = Dictionary::setDictionaryValues(DICT, IS_LATEST_DICT_VERSION, pos, &c, |
| 474 | &childPosition, &terminal, &freq); |
| 475 | |
| 476 | const bool needsToTraverseChildrenNodes = childPosition != 0; |
| 477 | |
| 478 | // If we are only doing traverseAllNodes, no need to look at the typed characters. |
| 479 | if (traverseAllNodes || needsToSkipCurrentNode(c, inputIndex, skipPos, depth)) { |
| 480 | mWord[depth] = c; |
| 481 | if (traverseAllNodes && terminal) { |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 482 | onTerminalWhenUserTypedLengthIsGreaterThanInputLength(mWord, inputIndex, depth, |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 483 | snr, nextLetters, nextLettersSize, skipPos, excessivePos, transposedPos, freq); |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 484 | } |
| 485 | if (!needsToTraverseChildrenNodes) return false; |
| 486 | *newTraverseAllNodes = traverseAllNodes; |
| 487 | *newSnr = snr; |
| 488 | *newDiffs = diffs; |
| 489 | *newInputIndex = inputIndex; |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 490 | } else { |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 491 | int *currentChars = mInputCodes + (inputIndex * MAX_PROXIMITY_CHARS); |
satok | a3d78f6 | 2010-12-09 22:08:33 +0900 | [diff] [blame] | 492 | |
| 493 | if (transposedPos >= 0) { |
| 494 | if (inputIndex == transposedPos) currentChars += MAX_PROXIMITY_CHARS; |
| 495 | if (inputIndex == (transposedPos + 1)) currentChars -= MAX_PROXIMITY_CHARS; |
| 496 | } |
| 497 | |
| 498 | int matchedProximityCharId = getMatchedProximityId(currentChars, c, skipPos, excessivePos, |
| 499 | transposedPos); |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 500 | if (UNRELATED_CHAR == matchedProximityCharId) return false; |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 501 | mWord[depth] = c; |
| 502 | // If inputIndex is greater than mInputLength, that means there is no |
| 503 | // proximity chars. So, we don't need to check proximity. |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 504 | if (SAME_OR_ACCENTED_OR_CAPITALIZED_CHAR == matchedProximityCharId) { |
| 505 | snr = snr * TYPED_LETTER_MULTIPLIER; |
| 506 | } |
satok | fd16f1d | 2011-01-27 16:25:16 +0900 | [diff] [blame] | 507 | bool isSameAsUserTypedLength = mInputLength == inputIndex + 1 |
| 508 | || (excessivePos == mInputLength - 1 && inputIndex == mInputLength - 2); |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 509 | if (isSameAsUserTypedLength && terminal) { |
satok | 54fe9e0 | 2010-12-13 14:42:35 +0900 | [diff] [blame] | 510 | onTerminalWhenUserTypedLengthIsSameAsInputLength(mWord, inputIndex, depth, snr, |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 511 | skipPos, excessivePos, transposedPos, freq); |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 512 | } |
| 513 | if (!needsToTraverseChildrenNodes) return false; |
| 514 | // Start traversing all nodes after the index exceeds the user typed length |
| 515 | *newTraverseAllNodes = isSameAsUserTypedLength; |
Jean Chalard | 8dc754a | 2011-01-27 14:20:22 +0900 | [diff] [blame] | 516 | *newSnr = snr; |
| 517 | *newDiffs = diffs + ((NEAR_PROXIMITY_CHAR == matchedProximityCharId) ? 1 : 0); |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 518 | *newInputIndex = inputIndex + 1; |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 519 | } |
| 520 | // Optimization: Prune out words that are too long compared to how much was typed. |
satok | d299792 | 2010-12-07 13:08:39 +0900 | [diff] [blame] | 521 | if (depth >= maxDepth || *newDiffs > mMaxEditDistance) { |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 522 | return false; |
| 523 | } |
| 524 | |
| 525 | // If inputIndex is greater than mInputLength, that means there are no proximity chars. |
satok | fd16f1d | 2011-01-27 16:25:16 +0900 | [diff] [blame] | 526 | // TODO: Check if this can be isSameAsUserTypedLength only. |
| 527 | if (isSameAsUserTypedLength || mInputLength <= *newInputIndex) { |
satok | 48e432c | 2010-12-06 17:38:58 +0900 | [diff] [blame] | 528 | *newTraverseAllNodes = true; |
| 529 | } |
| 530 | // get the count of nodes and increment childAddress. |
| 531 | *newCount = Dictionary::getCount(DICT, &childPosition); |
| 532 | *newChildPosition = childPosition; |
| 533 | if (DEBUG_DICT) assert(needsToTraverseChildrenNodes); |
| 534 | return needsToTraverseChildrenNodes; |
| 535 | } |
| 536 | |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 537 | inline int UnigramDictionary::getBestWordFreq(const int startInputIndex, const int inputLength, |
| 538 | unsigned short *word) { |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 539 | int pos = ROOT_POS; |
| 540 | int count = Dictionary::getCount(DICT, &pos); |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 541 | int maxFreq = 0; |
| 542 | int depth = 0; |
| 543 | unsigned short newWord[MAX_WORD_LENGTH_INTERNAL]; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 544 | bool terminal = false; |
| 545 | |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 546 | mStackChildCount[0] = count; |
| 547 | mStackSiblingPos[0] = pos; |
| 548 | |
| 549 | while (depth >= 0) { |
| 550 | if (mStackChildCount[depth] > 0) { |
| 551 | --mStackChildCount[depth]; |
| 552 | int firstChildPos; |
| 553 | int newFreq; |
| 554 | int siblingPos = mStackSiblingPos[depth]; |
| 555 | const bool needsToTraverseChildrenNodes = processCurrentNodeForExactMatch(siblingPos, |
| 556 | startInputIndex, depth, newWord, &firstChildPos, &count, &terminal, &newFreq, |
| 557 | &siblingPos); |
| 558 | mStackSiblingPos[depth] = siblingPos; |
| 559 | if (depth == (inputLength - 1)) { |
| 560 | // Traverse sibling node |
| 561 | if (terminal) { |
| 562 | if (newFreq > maxFreq) { |
| 563 | for (int i = 0; i < inputLength; ++i) word[i] = newWord[i]; |
| 564 | if (DEBUG_DICT && DEBUG_NODE) { |
| 565 | char s[inputLength + 1]; |
| 566 | for (int i = 0; i < inputLength; ++i) s[i] = word[i]; |
| 567 | s[inputLength] = 0; |
| 568 | LOGI("New missing space word found: %d > %d (%s), %d, %d", |
| 569 | newFreq, maxFreq, s, inputLength, depth); |
| 570 | } |
| 571 | maxFreq = newFreq; |
| 572 | } |
| 573 | } |
| 574 | } else if (needsToTraverseChildrenNodes) { |
| 575 | // Traverse children nodes |
| 576 | ++depth; |
| 577 | mStackChildCount[depth] = count; |
| 578 | mStackSiblingPos[depth] = firstChildPos; |
| 579 | } |
| 580 | } else { |
| 581 | // Traverse parent node |
| 582 | --depth; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 583 | } |
| 584 | } |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 585 | |
| 586 | word[inputLength] = 0; |
| 587 | return maxFreq; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 588 | } |
| 589 | |
| 590 | inline bool UnigramDictionary::processCurrentNodeForExactMatch(const int firstChildPos, |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 591 | const int startInputIndex, const int depth, unsigned short *word, int *newChildPosition, |
| 592 | int *newCount, bool *newTerminal, int *newFreq, int *siblingPos) { |
| 593 | const int inputIndex = startInputIndex + depth; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 594 | const int *currentChars = mInputCodes + (inputIndex * MAX_PROXIMITY_CHARS); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 595 | unsigned short c; |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 596 | *siblingPos = Dictionary::setDictionaryValues(DICT, IS_LATEST_DICT_VERSION, firstChildPos, &c, |
| 597 | newChildPosition, newTerminal, newFreq); |
| 598 | const unsigned int inputC = currentChars[0]; |
| 599 | if (DEBUG_DICT) assert(inputC <= U_SHORT_MAX); |
| 600 | const unsigned short lowerC = toLowerCase(c); |
| 601 | const bool matched = (inputC == lowerC || inputC == c); |
| 602 | const bool hasChild = *newChildPosition != 0; |
| 603 | if (matched) { |
| 604 | word[depth] = c; |
| 605 | if (DEBUG_DICT && DEBUG_NODE) { |
| 606 | LOGI("Node(%c, %c)<%d>, %d, %d", inputC, c, matched, hasChild, *newFreq); |
| 607 | if (*newTerminal) LOGI("Terminal %d", *newFreq); |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 608 | } |
satok | aee09dc | 2010-12-09 19:21:51 +0900 | [diff] [blame] | 609 | if (hasChild) { |
| 610 | *newCount = Dictionary::getCount(DICT, newChildPosition); |
| 611 | return true; |
| 612 | } else { |
| 613 | return false; |
| 614 | } |
| 615 | } else { |
| 616 | // If this node is not user typed character, this method treats this word as unmatched. |
| 617 | // Thus newTerminal shouldn't be true. |
| 618 | *newTerminal = false; |
| 619 | return false; |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 620 | } |
satok | 662fe69 | 2010-12-08 17:05:39 +0900 | [diff] [blame] | 621 | } |
satok | 3008825 | 2010-12-01 21:22:15 +0900 | [diff] [blame] | 622 | } // namespace latinime |