blob: cc6d39a29f9698b69d749105e96c7201de00b6ad [file] [log] [blame]
satok30088252010-12-01 21:22:15 +09001/*
Ken Wakasa5460ea32012-07-30 16:27:44 +09002 * Copyright (C) 2010, The Android Open Source Project
Ken Wakasa0bbb9172012-07-25 17:51:43 +09003 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
satok30088252010-12-01 21:22:15 +090016
Ken Wakasaf1008c52012-07-31 17:56:40 +090017#include <cassert>
18#include <cstring>
satok30088252010-12-01 21:22:15 +090019
satoke808e432010-12-02 14:53:24 +090020#define LOG_TAG "LatinIME: unigram_dictionary.cpp"
satok30088252010-12-01 21:22:15 +090021
Ken Wakasa77e8e812012-08-02 19:48:08 +090022#include "binary_format.h"
satok30088252010-12-01 21:22:15 +090023#include "char_utils.h"
Ken Wakasa3b088a22012-05-16 23:05:32 +090024#include "defines.h"
satoke808e432010-12-02 14:53:24 +090025#include "dictionary.h"
Ken Wakasa77e8e812012-08-02 19:48:08 +090026#include "proximity_info.h"
Jean Chalardcf9dbbd2011-12-26 15:16:59 +090027#include "terminal_attributes.h"
Ken Wakasa77e8e812012-08-02 19:48:08 +090028#include "unigram_dictionary.h"
29#include "words_priority_queue.h"
30#include "words_priority_queue_pool.h"
Jean Chalard1059f272011-06-28 20:45:05 +090031
satok30088252010-12-01 21:22:15 +090032namespace latinime {
33
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090034const UnigramDictionary::digraph_t UnigramDictionary::GERMAN_UMLAUT_DIGRAPHS[] =
Jean Chalardd3043382012-03-21 18:38:25 +090035 { { 'a', 'e', 0x00E4 }, // U+00E4 : LATIN SMALL LETTER A WITH DIAERESIS
36 { 'o', 'e', 0x00F6 }, // U+00F6 : LATIN SMALL LETTER O WITH DIAERESIS
37 { 'u', 'e', 0x00FC } }; // U+00FC : LATIN SMALL LETTER U WITH DIAERESIS
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090038
Jean Chalardcc78d032012-03-23 16:48:49 +090039const UnigramDictionary::digraph_t UnigramDictionary::FRENCH_LIGATURES_DIGRAPHS[] =
40 { { 'a', 'e', 0x00E6 }, // U+00E6 : LATIN SMALL LETTER AE
41 { 'o', 'e', 0x0153 } }; // U+0153 : LATIN SMALL LIGATURE OE
42
Jean Chalard293ece02011-06-16 20:55:16 +090043// TODO: check the header
Ken Wakasa0bbb9172012-07-25 17:51:43 +090044UnigramDictionary::UnigramDictionary(const uint8_t *const streamStart, int typedLetterMultiplier,
Jean Chalardcd274b12012-04-06 18:26:00 +090045 int fullWordMultiplier, int maxWordLength, int maxWords, const unsigned int flags)
Jean Chalard46a1eec2012-02-27 19:48:47 +090046 : DICT_ROOT(streamStart), MAX_WORD_LENGTH(maxWordLength), MAX_WORDS(maxWords),
satok662fe692010-12-08 17:05:39 +090047 TYPED_LETTER_MULTIPLIER(typedLetterMultiplier), FULL_WORD_MULTIPLIER(fullWordMultiplier),
Jean Chalard1059f272011-06-28 20:45:05 +090048 // TODO : remove this variable.
49 ROOT_POS(0),
satok9df4a452012-03-23 16:05:18 +090050 BYTES_IN_ONE_CHAR(sizeof(int)),
Jean Chalardcd274b12012-04-06 18:26:00 +090051 MAX_DIGRAPH_SEARCH_DEPTH(DEFAULT_MAX_DIGRAPH_SEARCH_DEPTH), FLAGS(flags) {
Ken Wakasade3070a2011-03-19 09:16:42 +090052 if (DEBUG_DICT) {
satok9fb6f472012-01-13 18:01:22 +090053 AKLOGI("UnigramDictionary - constructor");
Ken Wakasade3070a2011-03-19 09:16:42 +090054 }
satok30088252010-12-01 21:22:15 +090055}
56
satok2df30602011-07-15 13:49:00 +090057UnigramDictionary::~UnigramDictionary() {
satok2df30602011-07-15 13:49:00 +090058}
satok30088252010-12-01 21:22:15 +090059
satok9df4a452012-03-23 16:05:18 +090060static inline unsigned int getCodesBufferSize(const int *codes, const int codesSize) {
61 return sizeof(*codes) * codesSize;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090062}
63
Ken Wakasa951ab9d2012-03-09 19:18:59 +090064// TODO: This needs to take a const unsigned short* and not tinker with its contents
satok1147c7b2011-12-14 15:04:58 +090065static inline void addWord(
Jean Chalard926ef062012-08-10 13:14:45 +090066 unsigned short *word, int length, int frequency, WordsPriorityQueue *queue, int type) {
67 queue->push(frequency, word, length, type);
satok1147c7b2011-12-14 15:04:58 +090068}
69
Jean Chalardd3043382012-03-21 18:38:25 +090070// Return the replacement code point for a digraph, or 0 if none.
71int UnigramDictionary::getDigraphReplacement(const int *codes, const int i, const int codesSize,
Ken Wakasa0bbb9172012-07-25 17:51:43 +090072 const digraph_t *const digraphs, const unsigned int digraphsSize) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090073
74 // There can't be a digraph if we don't have at least 2 characters to examine
75 if (i + 2 > codesSize) return false;
76
77 // Search for the first char of some digraph
78 int lastDigraphIndex = -1;
satok9df4a452012-03-23 16:05:18 +090079 const int thisChar = codes[i];
Jean Chalard6c300612012-03-06 19:54:03 +090080 for (lastDigraphIndex = digraphsSize - 1; lastDigraphIndex >= 0; --lastDigraphIndex) {
81 if (thisChar == digraphs[lastDigraphIndex].first) break;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090082 }
83 // No match: return early
Jean Chalardd3043382012-03-21 18:38:25 +090084 if (lastDigraphIndex < 0) return 0;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090085
86 // It's an interesting digraph if the second char matches too.
satok9df4a452012-03-23 16:05:18 +090087 if (digraphs[lastDigraphIndex].second == codes[i + 1]) {
Jean Chalardd3043382012-03-21 18:38:25 +090088 return digraphs[lastDigraphIndex].replacement;
89 } else {
90 return 0;
91 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090092}
93
94// Mostly the same arguments as the non-recursive version, except:
95// codes is the original value. It points to the start of the work buffer, and gets passed as is.
96// codesSize is the size of the user input (thus, it is the size of codesSrc).
97// codesDest is the current point in the work buffer.
98// codesSrc is the current point in the user-input, original, content-unmodified buffer.
99// codesRemain is the remaining size in codesSrc.
satok1d7eaf82011-07-13 10:32:02 +0900100void UnigramDictionary::getWordWithDigraphSuggestionsRec(ProximityInfo *proximityInfo,
satok1147c7b2011-12-14 15:04:58 +0900101 const int *xcoordinates, const int *ycoordinates, const int *codesBuffer,
satok219a5142012-03-08 11:53:18 +0900102 int *xCoordinatesBuffer, int *yCoordinatesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900103 const int codesBufferSize, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900104 const bool useFullEditDistance, const int *codesSrc,
satok1147c7b2011-12-14 15:04:58 +0900105 const int codesRemain, const int currentDepth, int *codesDest, Correction *correction,
Jean Chalard6c300612012-03-06 19:54:03 +0900106 WordsPriorityQueuePool *queuePool,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900107 const digraph_t *const digraphs, const unsigned int digraphsSize) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900108
satok9df4a452012-03-23 16:05:18 +0900109 const int startIndex = codesDest - codesBuffer;
Jean Chalard6c300612012-03-06 19:54:03 +0900110 if (currentDepth < MAX_DIGRAPH_SEARCH_DEPTH) {
Jean Chalarda787dba2011-03-04 12:17:48 +0900111 for (int i = 0; i < codesRemain; ++i) {
satok219a5142012-03-08 11:53:18 +0900112 xCoordinatesBuffer[startIndex + i] = xcoordinates[codesBufferSize - codesRemain + i];
113 yCoordinatesBuffer[startIndex + i] = ycoordinates[codesBufferSize - codesRemain + i];
Jean Chalardd3043382012-03-21 18:38:25 +0900114 const int replacementCodePoint =
115 getDigraphReplacement(codesSrc, i, codesRemain, digraphs, digraphsSize);
116 if (0 != replacementCodePoint) {
Jean Chalarda787dba2011-03-04 12:17:48 +0900117 // Found a digraph. We will try both spellings. eg. the word is "pruefen"
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900118
Jean Chalardd3043382012-03-21 18:38:25 +0900119 // Copy the word up to the first char of the digraph, including proximity chars,
120 // and overwrite the primary code with the replacement code point. Then, continue
121 // processing on the remaining part of the word, skipping the second char of the
122 // digraph.
123 // In our example, copy "pru", replace "u" with the version with the diaeresis and
124 // continue running on "fen".
Jean Chalarda787dba2011-03-04 12:17:48 +0900125 // Make i the index of the second char of the digraph for simplicity. Forgetting
126 // to do that results in an infinite recursion so take care!
127 ++i;
128 memcpy(codesDest, codesSrc, i * BYTES_IN_ONE_CHAR);
Jean Chalardd3043382012-03-21 18:38:25 +0900129 codesDest[(i - 1) * (BYTES_IN_ONE_CHAR / sizeof(codesDest[0]))] =
130 replacementCodePoint;
Jean Chalarda787dba2011-03-04 12:17:48 +0900131 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900132 codesBuffer, xCoordinatesBuffer, yCoordinatesBuffer, codesBufferSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900133 bigramMap, bigramFilter, useFullEditDistance, codesSrc + i + 1,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900134 codesRemain - i - 1, currentDepth + 1, codesDest + i, correction,
Jean Chalard6c300612012-03-06 19:54:03 +0900135 queuePool, digraphs, digraphsSize);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900136
Jean Chalarda787dba2011-03-04 12:17:48 +0900137 // Copy the second char of the digraph in place, then continue processing on
138 // the remaining part of the word.
139 // In our example, after "pru" in the buffer copy the "e", and continue on "fen"
satok9df4a452012-03-23 16:05:18 +0900140 memcpy(codesDest + i, codesSrc + i, BYTES_IN_ONE_CHAR);
Jean Chalarda787dba2011-03-04 12:17:48 +0900141 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900142 codesBuffer, xCoordinatesBuffer, yCoordinatesBuffer, codesBufferSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900143 bigramMap, bigramFilter, useFullEditDistance, codesSrc + i, codesRemain - i,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900144 currentDepth + 1, codesDest + i, correction, queuePool, digraphs,
145 digraphsSize);
Jean Chalarda787dba2011-03-04 12:17:48 +0900146 return;
147 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900148 }
149 }
150
151 // If we come here, we hit the end of the word: let's check it against the dictionary.
152 // In our example, we'll come here once for "prufen" and then once for "pruefen".
153 // If the word contains several digraphs, we'll come it for the product of them.
154 // eg. if the word is "ueberpruefen" we'll test, in order, against
155 // "uberprufen", "uberpruefen", "ueberprufen", "ueberpruefen".
156 const unsigned int remainingBytes = BYTES_IN_ONE_CHAR * codesRemain;
satok219a5142012-03-08 11:53:18 +0900157 if (0 != remainingBytes) {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900158 memcpy(codesDest, codesSrc, remainingBytes);
satok219a5142012-03-08 11:53:18 +0900159 memcpy(&xCoordinatesBuffer[startIndex], &xcoordinates[codesBufferSize - codesRemain],
160 sizeof(int) * codesRemain);
161 memcpy(&yCoordinatesBuffer[startIndex], &ycoordinates[codesBufferSize - codesRemain],
162 sizeof(int) * codesRemain);
163 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900164
satok219a5142012-03-08 11:53:18 +0900165 getWordSuggestions(proximityInfo, xCoordinatesBuffer, yCoordinatesBuffer, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900166 startIndex + codesRemain, bigramMap, bigramFilter, useFullEditDistance, correction,
satoka7e5a5a2011-12-15 16:49:12 +0900167 queuePool);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900168}
169
Jean Chalard8950ce62012-05-07 16:28:30 +0900170// bigramMap contains the association <bigram address> -> <bigram frequency>
171// bigramFilter is a bloom filter for fast rejection: see functions setInFilter and isInFilter
172// in bigram_dictionary.cpp
satoka7e5a5a2011-12-15 16:49:12 +0900173int UnigramDictionary::getSuggestions(ProximityInfo *proximityInfo,
satokb1ed1d42012-06-14 16:35:23 -0700174 const int *xcoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900175 const int *ycoordinates, const int *codes, const int codesSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900176 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard6931df92012-07-12 12:55:48 +0900177 const bool useFullEditDistance, unsigned short *outWords, int *frequencies,
178 int *outputTypes) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900179
satokb1ed1d42012-06-14 16:35:23 -0700180 WordsPriorityQueuePool queuePool(MAX_WORDS, SUB_QUEUE_MAX_WORDS, MAX_WORD_LENGTH);
181 queuePool.clearAll();
satok1bc038c2012-06-14 11:25:50 -0700182 Correction masterCorrection;
183 masterCorrection.resetCorrection();
Jean Chalardaa8df592012-04-06 18:54:20 +0900184 if (BinaryFormat::REQUIRES_GERMAN_UMLAUT_PROCESSING & FLAGS)
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900185 { // Incrementally tune the word and try all possibilities
satok9df4a452012-03-23 16:05:18 +0900186 int codesBuffer[getCodesBufferSize(codes, codesSize)];
satok219a5142012-03-08 11:53:18 +0900187 int xCoordinatesBuffer[codesSize];
188 int yCoordinatesBuffer[codesSize];
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900189 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900190 xCoordinatesBuffer, yCoordinatesBuffer, codesSize, bigramMap, bigramFilter,
satok1bc038c2012-06-14 11:25:50 -0700191 useFullEditDistance, codes, codesSize, 0, codesBuffer, &masterCorrection,
satokb1ed1d42012-06-14 16:35:23 -0700192 &queuePool, GERMAN_UMLAUT_DIGRAPHS,
Jean Chalard6c300612012-03-06 19:54:03 +0900193 sizeof(GERMAN_UMLAUT_DIGRAPHS) / sizeof(GERMAN_UMLAUT_DIGRAPHS[0]));
Jean Chalardaa8df592012-04-06 18:54:20 +0900194 } else if (BinaryFormat::REQUIRES_FRENCH_LIGATURES_PROCESSING & FLAGS) {
satokacb6c542012-03-23 20:58:18 +0900195 int codesBuffer[getCodesBufferSize(codes, codesSize)];
Jean Chalardcc78d032012-03-23 16:48:49 +0900196 int xCoordinatesBuffer[codesSize];
197 int yCoordinatesBuffer[codesSize];
198 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900199 xCoordinatesBuffer, yCoordinatesBuffer, codesSize, bigramMap, bigramFilter,
satok1bc038c2012-06-14 11:25:50 -0700200 useFullEditDistance, codes, codesSize, 0, codesBuffer, &masterCorrection,
satokb1ed1d42012-06-14 16:35:23 -0700201 &queuePool, FRENCH_LIGATURES_DIGRAPHS,
Jean Chalardcc78d032012-03-23 16:48:49 +0900202 sizeof(FRENCH_LIGATURES_DIGRAPHS) / sizeof(FRENCH_LIGATURES_DIGRAPHS[0]));
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900203 } else { // Normal processing
Jean Chalard338d3ec2012-04-06 19:28:34 +0900204 getWordSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, codesSize,
satokb1ed1d42012-06-14 16:35:23 -0700205 bigramMap, bigramFilter, useFullEditDistance, &masterCorrection, &queuePool);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900206 }
207
satok817e5172011-03-04 06:06:45 -0800208 PROF_START(20);
satok8330b482012-01-23 16:52:37 +0900209 if (DEBUG_DICT) {
satokb1ed1d42012-06-14 16:35:23 -0700210 float ns = queuePool.getMasterQueue()->getHighestNormalizedScore(
satok1bc038c2012-06-14 11:25:50 -0700211 masterCorrection.getPrimaryInputWord(), codesSize, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900212 ns += 0;
213 AKLOGI("Max normalized score = %f", ns);
214 }
satoka7e5a5a2011-12-15 16:49:12 +0900215 const int suggestedWordsCount =
Jean Chalard926ef062012-08-10 13:14:45 +0900216 queuePool.getMasterQueue()->outputSuggestions(masterCorrection.getPrimaryInputWord(),
217 codesSize, frequencies, outWords, outputTypes);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900218
219 if (DEBUG_DICT) {
satokb1ed1d42012-06-14 16:35:23 -0700220 float ns = queuePool.getMasterQueue()->getHighestNormalizedScore(
satok1bc038c2012-06-14 11:25:50 -0700221 masterCorrection.getPrimaryInputWord(), codesSize, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900222 ns += 0;
satok9fb6f472012-01-13 18:01:22 +0900223 AKLOGI("Returning %d words", suggestedWordsCount);
Jean Chalard980d6b62011-06-30 17:02:23 +0900224 /// Print the returned words
225 for (int j = 0; j < suggestedWordsCount; ++j) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900226 short unsigned int *w = outWords + j * MAX_WORD_LENGTH;
Jean Chalard980d6b62011-06-30 17:02:23 +0900227 char s[MAX_WORD_LENGTH];
228 for (int i = 0; i <= MAX_WORD_LENGTH; i++) s[i] = w[i];
Jean-Baptiste Querucd7c4132012-05-16 16:56:11 -0700229 (void)s;
satok9fb6f472012-01-13 18:01:22 +0900230 AKLOGI("%s %i", s, frequencies[j]);
Jean Chalard980d6b62011-06-30 17:02:23 +0900231 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900232 }
satok817e5172011-03-04 06:06:45 -0800233 PROF_END(20);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900234 PROF_CLOSE;
235 return suggestedWordsCount;
236}
237
satok1d7eaf82011-07-13 10:32:02 +0900238void UnigramDictionary::getWordSuggestions(ProximityInfo *proximityInfo,
satok1147c7b2011-12-14 15:04:58 +0900239 const int *xcoordinates, const int *ycoordinates, const int *codes,
Jean Chalard8950ce62012-05-07 16:28:30 +0900240 const int inputLength, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900241 const bool useFullEditDistance, Correction *correction,
242 WordsPriorityQueuePool *queuePool) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900243
satok61e2f852011-01-05 14:13:07 +0900244 PROF_OPEN;
245 PROF_START(0);
satok61e2f852011-01-05 14:13:07 +0900246 PROF_END(0);
satok30088252010-12-01 21:22:15 +0900247
satok61e2f852011-01-05 14:13:07 +0900248 PROF_START(1);
Jean Chalard8950ce62012-05-07 16:28:30 +0900249 getOneWordSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, bigramMap, bigramFilter,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900250 useFullEditDistance, inputLength, correction, queuePool);
satok61e2f852011-01-05 14:13:07 +0900251 PROF_END(1);
252
253 PROF_START(2);
satok10266c02011-08-19 22:05:59 +0900254 // Note: This line is intentionally left blank
satok61e2f852011-01-05 14:13:07 +0900255 PROF_END(2);
satokcdbbea72010-12-08 16:04:16 +0900256
satok61e2f852011-01-05 14:13:07 +0900257 PROF_START(3);
satok10266c02011-08-19 22:05:59 +0900258 // Note: This line is intentionally left blank
satok61e2f852011-01-05 14:13:07 +0900259 PROF_END(3);
satok30088252010-12-01 21:22:15 +0900260
satok61e2f852011-01-05 14:13:07 +0900261 PROF_START(4);
satok8330b482012-01-23 16:52:37 +0900262 bool hasAutoCorrectionCandidate = false;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900263 WordsPriorityQueue *masterQueue = queuePool->getMasterQueue();
satok8330b482012-01-23 16:52:37 +0900264 if (masterQueue->size() > 0) {
satok0028ed32012-05-16 20:42:12 +0900265 float nsForMaster = masterQueue->getHighestNormalizedScore(
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900266 correction->getPrimaryInputWord(), inputLength, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900267 hasAutoCorrectionCandidate = (nsForMaster > START_TWO_WORDS_CORRECTION_THRESHOLD);
268 }
satok61e2f852011-01-05 14:13:07 +0900269 PROF_END(4);
satoka3d78f62010-12-09 22:08:33 +0900270
satok61e2f852011-01-05 14:13:07 +0900271 PROF_START(5);
satok99557162012-01-26 22:49:13 +0900272 // Multiple word suggestions
273 if (SUGGEST_MULTIPLE_WORDS
274 && inputLength >= MIN_USER_TYPED_LENGTH_FOR_MULTIPLE_WORD_SUGGESTION) {
satoka85f4922012-01-30 18:18:30 +0900275 getSplitMultipleWordsSuggestions(proximityInfo, xcoordinates, ycoordinates, codes,
satok1f6b52e2012-01-30 13:53:58 +0900276 useFullEditDistance, inputLength, correction, queuePool,
277 hasAutoCorrectionCandidate);
satok662fe692010-12-08 17:05:39 +0900278 }
satok61e2f852011-01-05 14:13:07 +0900279 PROF_END(5);
satok817e5172011-03-04 06:06:45 -0800280
281 PROF_START(6);
satok99557162012-01-26 22:49:13 +0900282 // Note: This line is intentionally left blank
satok817e5172011-03-04 06:06:45 -0800283 PROF_END(6);
satok99557162012-01-26 22:49:13 +0900284
satok29dc8062012-01-17 15:59:15 +0900285 if (DEBUG_DICT) {
satok6ad15fc2012-01-16 16:21:21 +0900286 queuePool->dumpSubQueue1TopSuggestions();
satok29dc8062012-01-17 15:59:15 +0900287 for (int i = 0; i < SUB_QUEUE_MAX_COUNT; ++i) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900288 WordsPriorityQueue *queue = queuePool->getSubQueue(FIRST_WORD_INDEX, i);
satok29dc8062012-01-17 15:59:15 +0900289 if (queue->size() > 0) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900290 WordsPriorityQueue::SuggestedWord *sw = queue->top();
satok29dc8062012-01-17 15:59:15 +0900291 const int score = sw->mScore;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900292 const unsigned short *word = sw->mWord;
satok29dc8062012-01-17 15:59:15 +0900293 const int wordLength = sw->mWordLength;
satok0028ed32012-05-16 20:42:12 +0900294 float ns = Correction::RankingAlgorithm::calcNormalizedScore(
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900295 correction->getPrimaryInputWord(), i, word, wordLength, score);
satok29dc8062012-01-17 15:59:15 +0900296 ns += 0;
297 AKLOGI("--- TOP SUB WORDS for %d --- %d %f [%d]", i, score, ns,
satok54af64a2012-01-17 15:58:23 +0900298 (ns > TWO_WORDS_CORRECTION_WITH_OTHER_ERROR_THRESHOLD));
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900299 DUMP_WORD(correction->getPrimaryInputWord(), i);
satok29dc8062012-01-17 15:59:15 +0900300 DUMP_WORD(word, wordLength);
301 }
302 }
satok6ad15fc2012-01-16 16:21:21 +0900303 }
satok30088252010-12-01 21:22:15 +0900304}
305
Yusuke Nojima258bfe62011-09-28 12:59:43 +0900306void UnigramDictionary::initSuggestions(ProximityInfo *proximityInfo, const int *xCoordinates,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900307 const int *yCoordinates, const int *codes, const int inputLength,
308 Correction *correction) const {
Ken Wakasade3070a2011-03-19 09:16:42 +0900309 if (DEBUG_DICT) {
satok9fb6f472012-01-13 18:01:22 +0900310 AKLOGI("initSuggest");
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900311 DUMP_WORD_INT(codes, inputLength);
Ken Wakasade3070a2011-03-19 09:16:42 +0900312 }
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900313 correction->initInputParams(proximityInfo, codes, inputLength, xCoordinates, yCoordinates);
satok1a6da632011-12-16 23:15:06 +0900314 const int maxDepth = min(inputLength * MAX_DEPTH_MULTIPLIER, MAX_WORD_LENGTH);
315 correction->initCorrection(proximityInfo, inputLength, maxDepth);
satok30088252010-12-01 21:22:15 +0900316}
317
satok715514d2010-12-02 20:19:59 +0900318static const char QUOTE = '\'';
satok662fe692010-12-08 17:05:39 +0900319static const char SPACE = ' ';
satok30088252010-12-01 21:22:15 +0900320
satok744dab62011-12-15 22:29:05 +0900321void UnigramDictionary::getOneWordSuggestions(ProximityInfo *proximityInfo,
322 const int *xcoordinates, const int *ycoordinates, const int *codes,
Jean Chalard8950ce62012-05-07 16:28:30 +0900323 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
324 const bool useFullEditDistance, const int inputLength,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900325 Correction *correction, WordsPriorityQueuePool *queuePool) const {
satok6ad15fc2012-01-16 16:21:21 +0900326 initSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, inputLength, correction);
Jean Chalard8950ce62012-05-07 16:28:30 +0900327 getSuggestionCandidates(useFullEditDistance, inputLength, bigramMap, bigramFilter, correction,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900328 queuePool, true /* doAutoCompletion */, DEFAULT_MAX_ERRORS, FIRST_WORD_INDEX);
satok744dab62011-12-15 22:29:05 +0900329}
330
satok1147c7b2011-12-14 15:04:58 +0900331void UnigramDictionary::getSuggestionCandidates(const bool useFullEditDistance,
Jean Chalard8950ce62012-05-07 16:28:30 +0900332 const int inputLength, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900333 Correction *correction, WordsPriorityQueuePool *queuePool,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900334 const bool doAutoCompletion, const int maxErrors, const int currentWordIndex) const {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900335 uint8_t totalTraverseCount = correction->pushAndGetTotalTraverseCount();
336 if (DEBUG_DICT) {
337 AKLOGI("Traverse count %d", totalTraverseCount);
338 }
339 if (totalTraverseCount > MULTIPLE_WORDS_SUGGESTION_MAX_TOTAL_TRAVERSE_COUNT) {
340 if (DEBUG_DICT) {
341 AKLOGI("Abort traversing %d", totalTraverseCount);
342 }
343 return;
344 }
satok10266c02011-08-19 22:05:59 +0900345 // TODO: Remove setCorrectionParams
satok1147c7b2011-12-14 15:04:58 +0900346 correction->setCorrectionParams(0, 0, 0,
satokd03317c2011-12-14 21:38:11 +0900347 -1 /* spaceProximityPos */, -1 /* missingSpacePos */, useFullEditDistance,
satok1a6da632011-12-16 23:15:06 +0900348 doAutoCompletion, maxErrors);
satok662fe692010-12-08 17:05:39 +0900349 int rootPosition = ROOT_POS;
Jean Chalard980d6b62011-06-30 17:02:23 +0900350 // Get the number of children of root, then increment the position
Jean Chalard6d419812012-01-16 15:19:47 +0900351 int childCount = BinaryFormat::getGroupCountAndForwardPointer(DICT_ROOT, &rootPosition);
satok208268d2011-08-10 15:44:08 +0900352 int outputIndex = 0;
satokd2997922010-12-07 13:08:39 +0900353
satok1147c7b2011-12-14 15:04:58 +0900354 correction->initCorrectionState(rootPosition, childCount, (inputLength <= 0));
satokd2997922010-12-07 13:08:39 +0900355
satok662fe692010-12-08 17:05:39 +0900356 // Depth first search
satok208268d2011-08-10 15:44:08 +0900357 while (outputIndex >= 0) {
satok1147c7b2011-12-14 15:04:58 +0900358 if (correction->initProcessState(outputIndex)) {
359 int siblingPos = correction->getTreeSiblingPos(outputIndex);
satokd2997922010-12-07 13:08:39 +0900360 int firstChildPos;
satok0f6c8e82011-08-03 02:19:44 +0900361
satok4e4e74e2011-08-03 23:27:32 +0900362 const bool needsToTraverseChildrenNodes = processCurrentNode(siblingPos,
Jean Chalard8950ce62012-05-07 16:28:30 +0900363 bigramMap, bigramFilter, correction, &childCount, &firstChildPos, &siblingPos,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900364 queuePool, currentWordIndex);
satok662fe692010-12-08 17:05:39 +0900365 // Update next sibling pos
satok1147c7b2011-12-14 15:04:58 +0900366 correction->setTreeSiblingPos(outputIndex, siblingPos);
satok208268d2011-08-10 15:44:08 +0900367
satokd2997922010-12-07 13:08:39 +0900368 if (needsToTraverseChildrenNodes) {
369 // Goes to child node
satok1147c7b2011-12-14 15:04:58 +0900370 outputIndex = correction->goDownTree(outputIndex, childCount, firstChildPos);
satokd2997922010-12-07 13:08:39 +0900371 }
372 } else {
satokcdbbea72010-12-08 16:04:16 +0900373 // Goes to parent sibling node
satok1147c7b2011-12-14 15:04:58 +0900374 outputIndex = correction->getTreeParentIndex(outputIndex);
satokd2997922010-12-07 13:08:39 +0900375 }
376 }
377}
378
Jean Chalard171d1802012-04-23 18:51:00 +0900379inline void UnigramDictionary::onTerminal(const int probability,
Jean Chalardcf9dbbd2011-12-26 15:16:59 +0900380 const TerminalAttributes& terminalAttributes, Correction *correction,
satok8330b482012-01-23 16:52:37 +0900381 WordsPriorityQueuePool *queuePool, const bool addToMasterQueue,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900382 const int currentWordIndex) const {
satok6ad15fc2012-01-16 16:21:21 +0900383 const int inputIndex = correction->getInputIndex();
384 const bool addToSubQueue = inputIndex < SUB_QUEUE_MAX_COUNT;
satok54af64a2012-01-17 15:58:23 +0900385
satok8876b752011-08-04 18:31:57 +0900386 int wordLength;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900387 unsigned short *wordPointer;
satok54af64a2012-01-17 15:58:23 +0900388
satok1f6b52e2012-01-30 13:53:58 +0900389 if ((currentWordIndex == FIRST_WORD_INDEX) && addToMasterQueue) {
satok54af64a2012-01-17 15:58:23 +0900390 WordsPriorityQueue *masterQueue = queuePool->getMasterQueue();
Jean Chalard171d1802012-04-23 18:51:00 +0900391 const int finalProbability =
392 correction->getFinalProbability(probability, &wordPointer, &wordLength);
393 if (finalProbability != NOT_A_PROBABILITY) {
Jean Chalardb14fc88e2012-08-10 16:52:27 +0900394 if (0 != finalProbability) {
395 // If the probability is 0, we don't want to add this word. However we still
396 // want to add its shortcuts (including a possible whitelist entry) if any.
397 addWord(wordPointer, wordLength, finalProbability, masterQueue,
398 Dictionary::KIND_CORRECTION);
399 }
satok54af64a2012-01-17 15:58:23 +0900400
Jean Chalard171d1802012-04-23 18:51:00 +0900401 const int shortcutProbability = finalProbability > 0 ? finalProbability - 1 : 0;
satok54af64a2012-01-17 15:58:23 +0900402 // Please note that the shortcut candidates will be added to the master queue only.
403 TerminalAttributes::ShortcutIterator iterator =
404 terminalAttributes.getShortcutIterator();
405 while (iterator.hasNextShortcutTarget()) {
406 // TODO: addWord only supports weak ordering, meaning we have no means
407 // to control the order of the shortcuts relative to one another or to the word.
Jean Chalard171d1802012-04-23 18:51:00 +0900408 // We need to either modulate the probability of each shortcut according
409 // to its own shortcut probability or to make the queue
satok54af64a2012-01-17 15:58:23 +0900410 // so that the insert order is protected inside the queue for words
Jean Chalard9a933a72012-03-27 19:56:23 +0900411 // with the same score. For the moment we use -1 to make sure the shortcut will
412 // never be in front of the word.
satok54af64a2012-01-17 15:58:23 +0900413 uint16_t shortcutTarget[MAX_WORD_LENGTH_INTERNAL];
Jean Chalardb14fc88e2012-08-10 16:52:27 +0900414 int shortcutFrequency;
satok54af64a2012-01-17 15:58:23 +0900415 const int shortcutTargetStringLength = iterator.getNextShortcutTarget(
Jean Chalardb14fc88e2012-08-10 16:52:27 +0900416 MAX_WORD_LENGTH_INTERNAL, shortcutTarget, &shortcutFrequency);
417 int shortcutScore;
418 int kind;
419 if (shortcutFrequency == BinaryFormat::WHITELIST_SHORTCUT_FREQUENCY
420 && correction->sameAsTyped()) {
421 shortcutScore = S_INT_MAX;
422 kind = Dictionary::KIND_WHITELIST;
423 } else {
424 shortcutScore = shortcutProbability;
425 kind = Dictionary::KIND_CORRECTION;
426 }
427 addWord(shortcutTarget, shortcutTargetStringLength, shortcutScore,
428 masterQueue, kind);
satok6ad15fc2012-01-16 16:21:21 +0900429 }
Jean Chalardcf9dbbd2011-12-26 15:16:59 +0900430 }
satok54af64a2012-01-17 15:58:23 +0900431 }
satok6ad15fc2012-01-16 16:21:21 +0900432
satok54af64a2012-01-17 15:58:23 +0900433 // We only allow two words + other error correction for words with SUB_QUEUE_MIN_WORD_LENGTH
434 // or more length.
435 if (inputIndex >= SUB_QUEUE_MIN_WORD_LENGTH && addToSubQueue) {
satok8330b482012-01-23 16:52:37 +0900436 WordsPriorityQueue *subQueue;
satok7409d152012-01-26 16:13:25 +0900437 subQueue = queuePool->getSubQueue(currentWordIndex, inputIndex);
438 if (!subQueue) {
satok8330b482012-01-23 16:52:37 +0900439 return;
440 }
Jean Chalard171d1802012-04-23 18:51:00 +0900441 const int finalProbability = correction->getFinalProbabilityForSubQueue(
442 probability, &wordPointer, &wordLength, inputIndex);
Jean Chalard926ef062012-08-10 13:14:45 +0900443 addWord(wordPointer, wordLength, finalProbability, subQueue, Dictionary::KIND_CORRECTION);
Jean Chalardca5ef282011-06-17 15:36:26 +0900444 }
445}
446
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900447int UnigramDictionary::getSubStringSuggestion(
satok7409d152012-01-26 16:13:25 +0900448 ProximityInfo *proximityInfo, const int *xcoordinates, const int *ycoordinates,
satok3c09bb12012-01-26 18:36:19 +0900449 const int *codes, const bool useFullEditDistance, Correction *correction,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900450 WordsPriorityQueuePool *queuePool, const int inputLength,
satok3c09bb12012-01-26 18:36:19 +0900451 const bool hasAutoCorrectionCandidate, const int currentWordIndex,
452 const int inputWordStartPos, const int inputWordLength,
satok99557162012-01-26 22:49:13 +0900453 const int outputWordStartPos, const bool isSpaceProximity, int *freqArray,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900454 int*wordLengthArray, unsigned short *outputWord, int *outputWordLength) const {
Satoshi Kataoka67e3cc82012-05-31 15:04:58 +0900455 if (inputWordLength > MULTIPLE_WORDS_SUGGESTION_MAX_WORD_LENGTH) {
456 return FLAG_MULTIPLE_SUGGEST_ABORT;
457 }
458
459 /////////////////////////////////////////////
460 // safety net for multiple word suggestion //
461 // TODO: Remove this safety net //
462 /////////////////////////////////////////////
463 int smallWordCount = 0;
464 int singleLetterWordCount = 0;
465 if (inputWordLength == 1) {
466 ++singleLetterWordCount;
467 }
468 if (inputWordLength <= 2) {
469 // small word == single letter or 2-letter word
470 ++smallWordCount;
471 }
472 for (int i = 0; i < currentWordIndex; ++i) {
473 const int length = wordLengthArray[i];
474 if (length == 1) {
475 ++singleLetterWordCount;
476 // Safety net to avoid suggesting sequential single letter words
477 if (i < (currentWordIndex - 1)) {
478 if (wordLengthArray[i + 1] == 1) {
479 return FLAG_MULTIPLE_SUGGEST_ABORT;
480 }
481 } else if (inputWordLength == 1) {
482 return FLAG_MULTIPLE_SUGGEST_ABORT;
483 }
484 }
485 if (length <= 2) {
486 ++smallWordCount;
487 }
488 // Safety net to avoid suggesting multiple words with many (4 or more, for now) small words
489 if (singleLetterWordCount >= 3 || smallWordCount >= 4) {
490 return FLAG_MULTIPLE_SUGGEST_ABORT;
491 }
492 }
493 //////////////////////////////////////////////
494 // TODO: Remove the safety net above //
495 //////////////////////////////////////////////
496
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900497 unsigned short *tempOutputWord = 0;
satok1f6b52e2012-01-30 13:53:58 +0900498 int nextWordLength = 0;
satok99557162012-01-26 22:49:13 +0900499 // TODO: Optimize init suggestion
500 initSuggestions(proximityInfo, xcoordinates, ycoordinates, codes,
501 inputLength, correction);
502
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900503 unsigned short word[MAX_WORD_LENGTH_INTERNAL];
satok3c09bb12012-01-26 18:36:19 +0900504 int freq = getMostFrequentWordLike(
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900505 inputWordStartPos, inputWordLength, correction, word);
satok3c09bb12012-01-26 18:36:19 +0900506 if (freq > 0) {
satok1f6b52e2012-01-30 13:53:58 +0900507 nextWordLength = inputWordLength;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900508 tempOutputWord = word;
satok3c09bb12012-01-26 18:36:19 +0900509 } else if (!hasAutoCorrectionCandidate) {
510 if (inputWordStartPos > 0) {
511 const int offset = inputWordStartPos;
512 initSuggestions(proximityInfo, &xcoordinates[offset], &ycoordinates[offset],
satok9df4a452012-03-23 16:05:18 +0900513 codes + offset, inputWordLength, correction);
satok3c09bb12012-01-26 18:36:19 +0900514 queuePool->clearSubQueue(currentWordIndex);
Jean Chalard4d9b2022012-04-23 19:25:28 +0900515 // TODO: pass the bigram list for substring suggestion
516 getSuggestionCandidates(useFullEditDistance, inputWordLength,
Jean Chalard8950ce62012-05-07 16:28:30 +0900517 0 /* bigramMap */, 0 /* bigramFilter */, correction, queuePool,
518 false /* doAutoCompletion */, MAX_ERRORS_FOR_TWO_WORDS, currentWordIndex);
satok3c09bb12012-01-26 18:36:19 +0900519 if (DEBUG_DICT) {
satok1f6b52e2012-01-30 13:53:58 +0900520 if (currentWordIndex < MULTIPLE_WORDS_SUGGESTION_MAX_WORDS) {
satok3c09bb12012-01-26 18:36:19 +0900521 AKLOGI("Dump word candidates(%d) %d", currentWordIndex, inputWordLength);
522 for (int i = 0; i < SUB_QUEUE_MAX_COUNT; ++i) {
523 queuePool->getSubQueue(currentWordIndex, i)->dumpTopWord();
524 }
525 }
526 }
527 }
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900528 WordsPriorityQueue *queue = queuePool->getSubQueue(currentWordIndex, inputWordLength);
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900529 // TODO: Return the correct value depending on doAutoCompletion
530 if (!queue || queue->size() <= 0) {
531 return FLAG_MULTIPLE_SUGGEST_ABORT;
satok3c09bb12012-01-26 18:36:19 +0900532 }
533 int score = 0;
satok0028ed32012-05-16 20:42:12 +0900534 const float ns = queue->getHighestNormalizedScore(
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900535 correction->getPrimaryInputWord(), inputWordLength,
satok1f6b52e2012-01-30 13:53:58 +0900536 &tempOutputWord, &score, &nextWordLength);
satok3c09bb12012-01-26 18:36:19 +0900537 if (DEBUG_DICT) {
538 AKLOGI("NS(%d) = %f, Score = %d", currentWordIndex, ns, score);
539 }
540 // Two words correction won't be done if the score of the first word doesn't exceed the
541 // threshold.
542 if (ns < TWO_WORDS_CORRECTION_WITH_OTHER_ERROR_THRESHOLD
satok1f6b52e2012-01-30 13:53:58 +0900543 || nextWordLength < SUB_QUEUE_MIN_WORD_LENGTH) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900544 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900545 }
satok1f6b52e2012-01-30 13:53:58 +0900546 freq = score >> (nextWordLength + TWO_WORDS_PLUS_OTHER_ERROR_CORRECTION_DEMOTION_DIVIDER);
satok3c09bb12012-01-26 18:36:19 +0900547 }
548 if (DEBUG_DICT) {
satok1f6b52e2012-01-30 13:53:58 +0900549 AKLOGI("Freq(%d): %d, length: %d, input length: %d, input start: %d (%d)"
550 , currentWordIndex, freq, nextWordLength, inputWordLength, inputWordStartPos,
551 wordLengthArray[0]);
satok3c09bb12012-01-26 18:36:19 +0900552 }
satok1f6b52e2012-01-30 13:53:58 +0900553 if (freq <= 0 || nextWordLength <= 0
554 || MAX_WORD_LENGTH <= (outputWordStartPos + nextWordLength)) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900555 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900556 }
satok1f6b52e2012-01-30 13:53:58 +0900557 for (int i = 0; i < nextWordLength; ++i) {
satok3c09bb12012-01-26 18:36:19 +0900558 outputWord[outputWordStartPos + i] = tempOutputWord[i];
559 }
satok99557162012-01-26 22:49:13 +0900560
561 // Put output values
satok1f6b52e2012-01-30 13:53:58 +0900562 freqArray[currentWordIndex] = freq;
satok99557162012-01-26 22:49:13 +0900563 // TODO: put output length instead of input length
satok1f6b52e2012-01-30 13:53:58 +0900564 wordLengthArray[currentWordIndex] = inputWordLength;
565 const int tempOutputWordLength = outputWordStartPos + nextWordLength;
566 if (outputWordLength) {
567 *outputWordLength = tempOutputWordLength;
568 }
satok99557162012-01-26 22:49:13 +0900569
satok3c09bb12012-01-26 18:36:19 +0900570 if ((inputWordStartPos + inputWordLength) < inputLength) {
satok1f6b52e2012-01-30 13:53:58 +0900571 if (outputWordStartPos + nextWordLength >= MAX_WORD_LENGTH) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900572 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900573 }
satoka85f4922012-01-30 18:18:30 +0900574 outputWord[tempOutputWordLength] = SPACE;
satok1f6b52e2012-01-30 13:53:58 +0900575 if (outputWordLength) {
576 ++*outputWordLength;
577 }
578 } else if (currentWordIndex >= 1) {
satok99557162012-01-26 22:49:13 +0900579 // TODO: Handle 3 or more words
satoka85f4922012-01-30 18:18:30 +0900580 const int pairFreq = correction->getFreqForSplitMultipleWords(
581 freqArray, wordLengthArray, currentWordIndex + 1, isSpaceProximity, outputWord);
satok99557162012-01-26 22:49:13 +0900582 if (DEBUG_DICT) {
satoka85f4922012-01-30 18:18:30 +0900583 DUMP_WORD(outputWord, tempOutputWordLength);
satoka0ac31f2012-05-23 19:55:27 +0900584 for (int i = 0; i < currentWordIndex + 1; ++i) {
585 AKLOGI("Split %d,%d words: freq = %d, length = %d", i, currentWordIndex + 1,
586 freqArray[i], wordLengthArray[i]);
587 }
588 AKLOGI("Split two words: freq = %d, length = %d, %d, isSpace ? %d", pairFreq,
589 inputLength, tempOutputWordLength, isSpaceProximity);
satok99557162012-01-26 22:49:13 +0900590 }
Jean Chalard926ef062012-08-10 13:14:45 +0900591 addWord(outputWord, tempOutputWordLength, pairFreq, queuePool->getMasterQueue(),
592 Dictionary::KIND_CORRECTION);
satok3c09bb12012-01-26 18:36:19 +0900593 }
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900594 return FLAG_MULTIPLE_SUGGEST_CONTINUE;
satok7409d152012-01-26 16:13:25 +0900595}
596
satok1f6b52e2012-01-30 13:53:58 +0900597void UnigramDictionary::getMultiWordsSuggestionRec(ProximityInfo *proximityInfo,
598 const int *xcoordinates, const int *ycoordinates, const int *codes,
599 const bool useFullEditDistance, const int inputLength,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900600 Correction *correction, WordsPriorityQueuePool *queuePool,
satok1f6b52e2012-01-30 13:53:58 +0900601 const bool hasAutoCorrectionCandidate, const int startInputPos, const int startWordIndex,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900602 const int outputWordLength, int *freqArray, int *wordLengthArray,
603 unsigned short *outputWord) const {
satok1f6b52e2012-01-30 13:53:58 +0900604 if (startWordIndex >= (MULTIPLE_WORDS_SUGGESTION_MAX_WORDS - 1)) {
605 // Return if the last word index
606 return;
607 }
satoka85f4922012-01-30 18:18:30 +0900608 if (startWordIndex >= 1
609 && (hasAutoCorrectionCandidate
610 || inputLength < MIN_INPUT_LENGTH_FOR_THREE_OR_MORE_WORDS_CORRECTION)) {
611 // Do not suggest 3+ words if already has auto correction candidate
612 return;
613 }
614 for (int i = startInputPos + 1; i < inputLength; ++i) {
satok1f6b52e2012-01-30 13:53:58 +0900615 if (DEBUG_CORRECTION_FREQ) {
satoka85f4922012-01-30 18:18:30 +0900616 AKLOGI("Multi words(%d), start in %d sep %d start out %d",
617 startWordIndex, startInputPos, i, outputWordLength);
618 DUMP_WORD(outputWord, outputWordLength);
satok1f6b52e2012-01-30 13:53:58 +0900619 }
satoka85f4922012-01-30 18:18:30 +0900620 int tempOutputWordLength = 0;
621 // Current word
622 int inputWordStartPos = startInputPos;
623 int inputWordLength = i - startInputPos;
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900624 const int suggestionFlag = getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates,
625 codes, useFullEditDistance, correction, queuePool, inputLength,
626 hasAutoCorrectionCandidate, startWordIndex, inputWordStartPos, inputWordLength,
627 outputWordLength, true /* not used */, freqArray, wordLengthArray, outputWord,
628 &tempOutputWordLength);
629 if (suggestionFlag == FLAG_MULTIPLE_SUGGEST_ABORT) {
630 // TODO: break here
631 continue;
632 } else if (suggestionFlag == FLAG_MULTIPLE_SUGGEST_SKIP) {
satok1f6b52e2012-01-30 13:53:58 +0900633 continue;
634 }
635
satoka85f4922012-01-30 18:18:30 +0900636 if (DEBUG_CORRECTION_FREQ) {
637 AKLOGI("Do missing space correction");
638 }
639 // Next word
satok1f6b52e2012-01-30 13:53:58 +0900640 // Missing space
641 inputWordStartPos = i;
642 inputWordLength = inputLength - i;
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900643 if(getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates, codes,
satok1f6b52e2012-01-30 13:53:58 +0900644 useFullEditDistance, correction, queuePool, inputLength, hasAutoCorrectionCandidate,
satoka85f4922012-01-30 18:18:30 +0900645 startWordIndex + 1, inputWordStartPos, inputWordLength, tempOutputWordLength,
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900646 false /* missing space */, freqArray, wordLengthArray, outputWord, 0)
647 != FLAG_MULTIPLE_SUGGEST_CONTINUE) {
satoka85f4922012-01-30 18:18:30 +0900648 getMultiWordsSuggestionRec(proximityInfo, xcoordinates, ycoordinates, codes,
649 useFullEditDistance, inputLength, correction, queuePool,
650 hasAutoCorrectionCandidate, inputWordStartPos, startWordIndex + 1,
651 tempOutputWordLength, freqArray, wordLengthArray, outputWord);
652 }
satok1f6b52e2012-01-30 13:53:58 +0900653
654 // Mistyped space
655 ++inputWordStartPos;
656 --inputWordLength;
657
658 if (inputWordLength <= 0) {
659 continue;
660 }
661
662 const int x = xcoordinates[inputWordStartPos - 1];
663 const int y = ycoordinates[inputWordStartPos - 1];
664 if (!proximityInfo->hasSpaceProximity(x, y)) {
665 continue;
666 }
667
satoka85f4922012-01-30 18:18:30 +0900668 if (DEBUG_CORRECTION_FREQ) {
669 AKLOGI("Do mistyped space correction");
670 }
satok1f6b52e2012-01-30 13:53:58 +0900671 getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates, codes,
672 useFullEditDistance, correction, queuePool, inputLength, hasAutoCorrectionCandidate,
satoka85f4922012-01-30 18:18:30 +0900673 startWordIndex + 1, inputWordStartPos, inputWordLength, tempOutputWordLength,
674 true /* mistyped space */, freqArray, wordLengthArray, outputWord, 0);
satok1f6b52e2012-01-30 13:53:58 +0900675 }
676}
677
satoka85f4922012-01-30 18:18:30 +0900678void UnigramDictionary::getSplitMultipleWordsSuggestions(ProximityInfo *proximityInfo,
satok744dab62011-12-15 22:29:05 +0900679 const int *xcoordinates, const int *ycoordinates, const int *codes,
satok1f6b52e2012-01-30 13:53:58 +0900680 const bool useFullEditDistance, const int inputLength,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900681 Correction *correction, WordsPriorityQueuePool *queuePool,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900682 const bool hasAutoCorrectionCandidate) const {
satokbd6ccdd2012-01-23 12:30:20 +0900683 if (inputLength >= MAX_WORD_LENGTH) return;
satok612c6e42011-08-01 19:35:27 +0900684 if (DEBUG_DICT) {
satok1f6b52e2012-01-30 13:53:58 +0900685 AKLOGI("--- Suggest multiple words");
686 }
satok54af64a2012-01-17 15:58:23 +0900687
satokbd6ccdd2012-01-23 12:30:20 +0900688 // Allocating fixed length array on stack
689 unsigned short outputWord[MAX_WORD_LENGTH];
satok1f6b52e2012-01-30 13:53:58 +0900690 int freqArray[MULTIPLE_WORDS_SUGGESTION_MAX_WORDS];
691 int wordLengthArray[MULTIPLE_WORDS_SUGGESTION_MAX_WORDS];
692 const int outputWordLength = 0;
693 const int startInputPos = 0;
694 const int startWordIndex = 0;
695 getMultiWordsSuggestionRec(proximityInfo, xcoordinates, ycoordinates, codes,
696 useFullEditDistance, inputLength, correction, queuePool, hasAutoCorrectionCandidate,
697 startInputPos, startWordIndex, outputWordLength, freqArray, wordLengthArray,
698 outputWord);
Jean Chalarde6715e32011-06-30 19:47:25 +0900699}
700
Jean Chalard1059f272011-06-28 20:45:05 +0900701// Wrapper for getMostFrequentWordLikeInner, which matches it to the previous
702// interface.
703inline int UnigramDictionary::getMostFrequentWordLike(const int startInputIndex,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900704 const int inputLength, Correction *correction, unsigned short *word) const {
Jean Chalard1059f272011-06-28 20:45:05 +0900705 uint16_t inWord[inputLength];
706
707 for (int i = 0; i < inputLength; ++i) {
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900708 inWord[i] = (uint16_t)correction->getPrimaryCharAt(startInputIndex + i);
Jean Chalard1059f272011-06-28 20:45:05 +0900709 }
710 return getMostFrequentWordLikeInner(inWord, inputLength, word);
711}
712
713// This function will take the position of a character array within a CharGroup,
714// and check it actually like-matches the word in inWord starting at startInputIndex,
715// that is, it matches it with case and accents squashed.
716// The function returns true if there was a full match, false otherwise.
717// The function will copy on-the-fly the characters in the CharGroup to outNewWord.
718// It will also place the end position of the array in outPos; in outInputIndex,
719// it will place the index of the first char AFTER the match if there was a match,
720// and the initial position if there was not. It makes sense because if there was
721// a match we want to continue searching, but if there was not, we want to go to
722// the next CharGroup.
723// In and out parameters may point to the same location. This function takes care
724// not to use any input parameters after it wrote into its outputs.
725static inline bool testCharGroupForContinuedLikeness(const uint8_t flags,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900726 const uint8_t *const root, const int startPos,
727 const uint16_t *const inWord, const int startInputIndex,
728 int32_t *outNewWord, int *outInputIndex, int *outPos) {
Jean Chalard19560502012-07-31 23:45:32 +0900729 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
Jean Chalard1059f272011-06-28 20:45:05 +0900730 int pos = startPos;
731 int32_t character = BinaryFormat::getCharCodeAndForwardPointer(root, &pos);
Tadashi G. Takaoka6e3cb272011-11-11 14:26:13 +0900732 int32_t baseChar = toBaseLowerCase(character);
733 const uint16_t wChar = toBaseLowerCase(inWord[startInputIndex]);
Jean Chalard1059f272011-06-28 20:45:05 +0900734
735 if (baseChar != wChar) {
736 *outPos = hasMultipleChars ? BinaryFormat::skipOtherCharacters(root, pos) : pos;
737 *outInputIndex = startInputIndex;
738 return false;
739 }
740 int inputIndex = startInputIndex;
741 outNewWord[inputIndex] = character;
742 if (hasMultipleChars) {
743 character = BinaryFormat::getCharCodeAndForwardPointer(root, &pos);
744 while (NOT_A_CHARACTER != character) {
Tadashi G. Takaoka6e3cb272011-11-11 14:26:13 +0900745 baseChar = toBaseLowerCase(character);
746 if (toBaseLowerCase(inWord[++inputIndex]) != baseChar) {
Jean Chalard1059f272011-06-28 20:45:05 +0900747 *outPos = BinaryFormat::skipOtherCharacters(root, pos);
748 *outInputIndex = startInputIndex;
749 return false;
750 }
751 outNewWord[inputIndex] = character;
752 character = BinaryFormat::getCharCodeAndForwardPointer(root, &pos);
753 }
754 }
755 *outInputIndex = inputIndex + 1;
756 *outPos = pos;
757 return true;
758}
759
760// This function is invoked when a word like the word searched for is found.
761// It will compare the frequency to the max frequency, and if greater, will
762// copy the word into the output buffer. In output value maxFreq, it will
763// write the new maximum frequency if it changed.
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900764static inline void onTerminalWordLike(const int freq, int32_t *newWord, const int length,
765 short unsigned int *outWord, int *maxFreq) {
Jean Chalard1059f272011-06-28 20:45:05 +0900766 if (freq > *maxFreq) {
767 for (int q = 0; q < length; ++q)
768 outWord[q] = newWord[q];
769 outWord[length] = 0;
770 *maxFreq = freq;
771 }
772}
773
774// Will find the highest frequency of the words like the one passed as an argument,
775// that is, everything that only differs by case/accents.
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900776int UnigramDictionary::getMostFrequentWordLikeInner(const uint16_t *const inWord,
777 const int length, short unsigned int *outWord) const {
Jean Chalard1059f272011-06-28 20:45:05 +0900778 int32_t newWord[MAX_WORD_LENGTH_INTERNAL];
779 int depth = 0;
780 int maxFreq = -1;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900781 const uint8_t *const root = DICT_ROOT;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900782 int stackChildCount[MAX_WORD_LENGTH_INTERNAL];
783 int stackInputIndex[MAX_WORD_LENGTH_INTERNAL];
784 int stackSiblingPos[MAX_WORD_LENGTH_INTERNAL];
Jean Chalard1059f272011-06-28 20:45:05 +0900785
Jean Chalard4c0eca62012-01-16 15:15:53 +0900786 int startPos = 0;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900787 stackChildCount[0] = BinaryFormat::getGroupCountAndForwardPointer(root, &startPos);
788 stackInputIndex[0] = 0;
789 stackSiblingPos[0] = startPos;
Jean Chalard1059f272011-06-28 20:45:05 +0900790 while (depth >= 0) {
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900791 const int charGroupCount = stackChildCount[depth];
792 int pos = stackSiblingPos[depth];
Jean Chalard1059f272011-06-28 20:45:05 +0900793 for (int charGroupIndex = charGroupCount - 1; charGroupIndex >= 0; --charGroupIndex) {
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900794 int inputIndex = stackInputIndex[depth];
Jean Chalard1059f272011-06-28 20:45:05 +0900795 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(root, &pos);
796 // Test whether all chars in this group match with the word we are searching for. If so,
797 // we want to traverse its children (or if the length match, evaluate its frequency).
798 // Note that this function will output the position regardless, but will only write
799 // into inputIndex if there is a match.
800 const bool isAlike = testCharGroupForContinuedLikeness(flags, root, pos, inWord,
801 inputIndex, newWord, &inputIndex, &pos);
Jean Chalard19560502012-07-31 23:45:32 +0900802 if (isAlike && (BinaryFormat::FLAG_IS_TERMINAL & flags) && (inputIndex == length)) {
Jean Chalard1059f272011-06-28 20:45:05 +0900803 const int frequency = BinaryFormat::readFrequencyWithoutMovingPointer(root, pos);
804 onTerminalWordLike(frequency, newWord, inputIndex, outWord, &maxFreq);
805 }
806 pos = BinaryFormat::skipFrequency(flags, pos);
807 const int siblingPos = BinaryFormat::skipChildrenPosAndAttributes(root, flags, pos);
808 const int childrenNodePos = BinaryFormat::readChildrenPosition(root, flags, pos);
809 // If we had a match and the word has children, we want to traverse them. We don't have
810 // to traverse words longer than the one we are searching for, since they will not match
811 // anyway, so don't traverse unless inputIndex < length.
812 if (isAlike && (-1 != childrenNodePos) && (inputIndex < length)) {
813 // Save position for this depth, to get back to this once children are done
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900814 stackChildCount[depth] = charGroupIndex;
815 stackSiblingPos[depth] = siblingPos;
Jean Chalard1059f272011-06-28 20:45:05 +0900816 // Prepare stack values for next depth
817 ++depth;
818 int childrenPos = childrenNodePos;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900819 stackChildCount[depth] =
Jean Chalard1059f272011-06-28 20:45:05 +0900820 BinaryFormat::getGroupCountAndForwardPointer(root, &childrenPos);
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900821 stackSiblingPos[depth] = childrenPos;
822 stackInputIndex[depth] = inputIndex;
Jean Chalard1059f272011-06-28 20:45:05 +0900823 pos = childrenPos;
824 // Go to the next depth level.
825 ++depth;
826 break;
827 } else {
828 // No match, or no children, or word too long to ever match: go the next sibling.
829 pos = siblingPos;
830 }
831 }
832 --depth;
833 }
834 return maxFreq;
835}
836
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900837int UnigramDictionary::getFrequency(const int32_t *const inWord, const int length) const {
838 const uint8_t *const root = DICT_ROOT;
Jean Chalarde9a86e22012-06-28 21:01:29 +0900839 int pos = BinaryFormat::getTerminalPosition(root, inWord, length,
840 false /* forceLowerCaseSearch */);
Satoshi Kataoka2f854e12012-05-29 15:58:13 +0900841 if (NOT_VALID_WORD == pos) {
842 return NOT_A_PROBABILITY;
843 }
844 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(root, &pos);
Jean Chalard19560502012-07-31 23:45:32 +0900845 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
Satoshi Kataoka2f854e12012-05-29 15:58:13 +0900846 if (hasMultipleChars) {
847 pos = BinaryFormat::skipOtherCharacters(root, pos);
848 } else {
849 BinaryFormat::getCharCodeAndForwardPointer(DICT_ROOT, &pos);
850 }
851 const int unigramFreq = BinaryFormat::readFrequencyWithoutMovingPointer(root, pos);
852 return unigramFreq;
Jean Chalard1059f272011-06-28 20:45:05 +0900853}
854
855// TODO: remove this function.
856int UnigramDictionary::getBigramPosition(int pos, unsigned short *word, int offset,
857 int length) const {
858 return -1;
859}
860
861// ProcessCurrentNode returns a boolean telling whether to traverse children nodes or not.
862// If the return value is false, then the caller should read in the output "nextSiblingPosition"
863// to find out the address of the next sibling node and pass it to a new call of processCurrentNode.
864// It is worthy to note that when false is returned, the output values other than
865// nextSiblingPosition are undefined.
866// If the return value is true, then the caller must proceed to traverse the children of this
867// node. processCurrentNode will output the information about the children: their count in
868// newCount, their position in newChildrenPosition, the traverseAllNodes flag in
869// newTraverseAllNodes, the match weight into newMatchRate, the input index into newInputIndex, the
870// diffs into newDiffs, the sibling position in nextSiblingPosition, and the output index into
871// newOutputIndex. Please also note the following caveat: processCurrentNode does not know when
872// there aren't any more nodes at this level, it merely returns the address of the first byte after
873// the current node in nextSiblingPosition. Thus, the caller must keep count of the nodes at any
874// given level, as output into newCount when traversing this level's parent.
satok8876b752011-08-04 18:31:57 +0900875inline bool UnigramDictionary::processCurrentNode(const int initialPos,
Jean Chalard8950ce62012-05-07 16:28:30 +0900876 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter, Correction *correction,
877 int *newCount, int *newChildrenPosition, int *nextSiblingPosition,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900878 WordsPriorityQueuePool *queuePool, const int currentWordIndex) const {
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900879 if (DEBUG_DICT) {
satokcfca3c62011-08-10 14:30:10 +0900880 correction->checkState();
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900881 }
Jean Chalard0584f022011-06-30 19:23:16 +0900882 int pos = initialPos;
Jean Chalard0584f022011-06-30 19:23:16 +0900883
Jean Chalard1059f272011-06-28 20:45:05 +0900884 // Flags contain the following information:
885 // - Address type (MASK_GROUP_ADDRESS_TYPE) on two bits:
886 // - FLAG_GROUP_ADDRESS_TYPE_{ONE,TWO,THREE}_BYTES means there are children and their address
887 // is on the specified number of bytes.
888 // - FLAG_GROUP_ADDRESS_TYPE_NOADDRESS means there are no children, and therefore no address.
889 // - FLAG_HAS_MULTIPLE_CHARS: whether this node has multiple char or not.
890 // - FLAG_IS_TERMINAL: whether this node is a terminal or not (it may still have children)
891 // - FLAG_HAS_BIGRAMS: whether this node has bigrams or not
892 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(DICT_ROOT, &pos);
Jean Chalard19560502012-07-31 23:45:32 +0900893 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
894 const bool isTerminalNode = (0 != (BinaryFormat::FLAG_IS_TERMINAL & flags));
satok8876b752011-08-04 18:31:57 +0900895
896 bool needsToInvokeOnTerminal = false;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900897
Jean Chalard1059f272011-06-28 20:45:05 +0900898 // This gets only ONE character from the stream. Next there will be:
899 // if FLAG_HAS_MULTIPLE CHARS: the other characters of the same node
900 // else if FLAG_IS_TERMINAL: the frequency
901 // else if MASK_GROUP_ADDRESS_TYPE is not NONE: the children address
902 // Note that you can't have a node that both is not a terminal and has no children.
903 int32_t c = BinaryFormat::getCharCodeAndForwardPointer(DICT_ROOT, &pos);
904 assert(NOT_A_CHARACTER != c);
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900905
Jean Chalard1059f272011-06-28 20:45:05 +0900906 // We are going to loop through each character and make it look like it's a different
907 // node each time. To do that, we will process characters in this node in order until
908 // we find the character terminator. This is signalled by getCharCode* returning
909 // NOT_A_CHARACTER.
910 // As a special case, if there is only one character in this node, we must not read the
911 // next bytes so we will simulate the NOT_A_CHARACTER return by testing the flags.
912 // This way, each loop run will look like a "virtual node".
913 do {
914 // We prefetch the next char. If 'c' is the last char of this node, we will have
915 // NOT_A_CHARACTER in the next char. From this we can decide whether this virtual node
916 // should behave as a terminal or not and whether we have children.
917 const int32_t nextc = hasMultipleChars
918 ? BinaryFormat::getCharCodeAndForwardPointer(DICT_ROOT, &pos) : NOT_A_CHARACTER;
919 const bool isLastChar = (NOT_A_CHARACTER == nextc);
920 // If there are more chars in this nodes, then this virtual node is not a terminal.
921 // If we are on the last char, this virtual node is a terminal if this node is.
satok8876b752011-08-04 18:31:57 +0900922 const bool isTerminal = isLastChar && isTerminalNode;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900923
satokcfca3c62011-08-10 14:30:10 +0900924 Correction::CorrectionType stateType = correction->processCharAndCalcState(
satok8876b752011-08-04 18:31:57 +0900925 c, isTerminal);
satokcfca3c62011-08-10 14:30:10 +0900926 if (stateType == Correction::TRAVERSE_ALL_ON_TERMINAL
927 || stateType == Correction::ON_TERMINAL) {
satok8876b752011-08-04 18:31:57 +0900928 needsToInvokeOnTerminal = true;
satokd03317c2011-12-14 21:38:11 +0900929 } else if (stateType == Correction::UNRELATED || correction->needsToPrune()) {
satok8876b752011-08-04 18:31:57 +0900930 // We found that this is an unrelated character, so we should give up traversing
931 // this node and its children entirely.
932 // However we may not be on the last virtual node yet so we skip the remaining
933 // characters in this node, the frequency if it's there, read the next sibling
934 // position to output it, then return false.
935 // We don't have to output other values because we return false, as in
936 // "don't traverse children".
Jean Chalard1059f272011-06-28 20:45:05 +0900937 if (!isLastChar) {
938 pos = BinaryFormat::skipOtherCharacters(DICT_ROOT, pos);
939 }
940 pos = BinaryFormat::skipFrequency(flags, pos);
941 *nextSiblingPosition =
942 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
943 return false;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900944 }
945
Jean Chalard1059f272011-06-28 20:45:05 +0900946 // Prepare for the next character. Promote the prefetched char to current char - the loop
947 // will take care of prefetching the next. If we finally found our last char, nextc will
948 // contain NOT_A_CHARACTER.
949 c = nextc;
Jean Chalard1059f272011-06-28 20:45:05 +0900950 } while (NOT_A_CHARACTER != c);
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900951
satok8876b752011-08-04 18:31:57 +0900952 if (isTerminalNode) {
satok6ad15fc2012-01-16 16:21:21 +0900953 // The frequency should be here, because we come here only if this is actually
954 // a terminal node, and we are on its last char.
Jean Chalard171d1802012-04-23 18:51:00 +0900955 const int unigramFreq = BinaryFormat::readFrequencyWithoutMovingPointer(DICT_ROOT, pos);
satok6ad15fc2012-01-16 16:21:21 +0900956 const int childrenAddressPos = BinaryFormat::skipFrequency(flags, pos);
957 const int attributesPos = BinaryFormat::skipChildrenPosition(flags, childrenAddressPos);
958 TerminalAttributes terminalAttributes(DICT_ROOT, flags, attributesPos);
Jean Chalard8950ce62012-05-07 16:28:30 +0900959 // bigramMap contains the bigram frequencies indexed by addresses for fast lookup.
960 // bigramFilter is a bloom filter of said frequencies for even faster rejection.
Jean Chalard49ba1352012-05-07 20:14:00 +0900961 const int probability = BinaryFormat::getProbability(initialPos, bigramMap, bigramFilter,
962 unigramFreq);
Jean Chalard171d1802012-04-23 18:51:00 +0900963 onTerminal(probability, terminalAttributes, correction, queuePool, needsToInvokeOnTerminal,
satok8330b482012-01-23 16:52:37 +0900964 currentWordIndex);
Jean Chalard1059f272011-06-28 20:45:05 +0900965
satok8876b752011-08-04 18:31:57 +0900966 // If there are more chars in this node, then this virtual node has children.
967 // If we are on the last char, this virtual node has children if this node has.
968 const bool hasChildren = BinaryFormat::hasChildrenInFlags(flags);
969
970 // This character matched the typed character (enough to traverse the node at least)
971 // so we just evaluated it. Now we should evaluate this virtual node's children - that
972 // is, if it has any. If it has no children, we're done here - so we skip the end of
973 // the node, output the siblings position, and return false "don't traverse children".
974 // Note that !hasChildren implies isLastChar, so we know we don't have to skip any
975 // remaining char in this group for there can't be any.
976 if (!hasChildren) {
977 pos = BinaryFormat::skipFrequency(flags, pos);
978 *nextSiblingPosition =
979 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
980 return false;
981 }
982
983 // Optimization: Prune out words that are too long compared to how much was typed.
satokcfca3c62011-08-10 14:30:10 +0900984 if (correction->needsToPrune()) {
satok8876b752011-08-04 18:31:57 +0900985 pos = BinaryFormat::skipFrequency(flags, pos);
986 *nextSiblingPosition =
987 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
satok10266c02011-08-19 22:05:59 +0900988 if (DEBUG_DICT_FULL) {
satok9fb6f472012-01-13 18:01:22 +0900989 AKLOGI("Traversing was pruned.");
satok10266c02011-08-19 22:05:59 +0900990 }
satok8876b752011-08-04 18:31:57 +0900991 return false;
992 }
993 }
Jean Chalard1059f272011-06-28 20:45:05 +0900994
995 // Now we finished processing this node, and we want to traverse children. If there are no
996 // children, we can't come here.
997 assert(BinaryFormat::hasChildrenInFlags(flags));
998
999 // If this node was a terminal it still has the frequency under the pointer (it may have been
1000 // read, but not skipped - see readFrequencyWithoutMovingPointer).
1001 // Next come the children position, then possibly attributes (attributes are bigrams only for
1002 // now, maybe something related to shortcuts in the future).
1003 // Once this is read, we still need to output the number of nodes in the immediate children of
1004 // this node, so we read and output it before returning true, as in "please traverse children".
1005 pos = BinaryFormat::skipFrequency(flags, pos);
1006 int childrenPos = BinaryFormat::readChildrenPosition(DICT_ROOT, flags, pos);
1007 *nextSiblingPosition = BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
1008 *newCount = BinaryFormat::getGroupCountAndForwardPointer(DICT_ROOT, &childrenPos);
1009 *newChildrenPosition = childrenPos;
1010 return true;
Jean Chalard85a1d1e2011-06-21 22:23:21 +09001011}
satok30088252010-12-01 21:22:15 +09001012} // namespace latinime