blob: 3b485a055a665aed938a5fda91f6b1f0546d842c [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 Wakasab02ee3d2012-10-08 11:46:14 +090044UnigramDictionary::UnigramDictionary(const uint8_t *const streamStart, int fullWordMultiplier,
45 int maxWordLength, int maxWords, const unsigned int flags)
46 : DICT_ROOT(streamStart), MAX_WORD_LENGTH(maxWordLength), MAX_WORDS(maxWords),
47 FULL_WORD_MULTIPLIER(fullWordMultiplier), // TODO : remove this variable.
48 ROOT_POS(0), BYTES_IN_ONE_CHAR(sizeof(int)),
49 MAX_DIGRAPH_SEARCH_DEPTH(DEFAULT_MAX_DIGRAPH_SEARCH_DEPTH), FLAGS(flags) {
Ken Wakasade3070a2011-03-19 09:16:42 +090050 if (DEBUG_DICT) {
satok9fb6f472012-01-13 18:01:22 +090051 AKLOGI("UnigramDictionary - constructor");
Ken Wakasade3070a2011-03-19 09:16:42 +090052 }
satok30088252010-12-01 21:22:15 +090053}
54
satok2df30602011-07-15 13:49:00 +090055UnigramDictionary::~UnigramDictionary() {
satok2df30602011-07-15 13:49:00 +090056}
satok30088252010-12-01 21:22:15 +090057
satok9df4a452012-03-23 16:05:18 +090058static inline unsigned int getCodesBufferSize(const int *codes, const int codesSize) {
Ken Wakasaf2789812012-09-04 12:49:46 +090059 return static_cast<unsigned int>(sizeof(*codes)) * codesSize;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090060}
61
Ken Wakasa951ab9d2012-03-09 19:18:59 +090062// TODO: This needs to take a const unsigned short* and not tinker with its contents
Ken Wakasaf2789812012-09-04 12:49:46 +090063static inline void addWord(unsigned short *word, int length, int frequency,
64 WordsPriorityQueue *queue, int type) {
Jean Chalard926ef062012-08-10 13:14:45 +090065 queue->push(frequency, word, length, type);
satok1147c7b2011-12-14 15:04:58 +090066}
67
Jean Chalardd3043382012-03-21 18:38:25 +090068// Return the replacement code point for a digraph, or 0 if none.
69int UnigramDictionary::getDigraphReplacement(const int *codes, const int i, const int codesSize,
Ken Wakasa0bbb9172012-07-25 17:51:43 +090070 const digraph_t *const digraphs, const unsigned int digraphsSize) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090071
72 // There can't be a digraph if we don't have at least 2 characters to examine
73 if (i + 2 > codesSize) return false;
74
75 // Search for the first char of some digraph
76 int lastDigraphIndex = -1;
satok9df4a452012-03-23 16:05:18 +090077 const int thisChar = codes[i];
Jean Chalard6c300612012-03-06 19:54:03 +090078 for (lastDigraphIndex = digraphsSize - 1; lastDigraphIndex >= 0; --lastDigraphIndex) {
79 if (thisChar == digraphs[lastDigraphIndex].first) break;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090080 }
81 // No match: return early
Jean Chalardd3043382012-03-21 18:38:25 +090082 if (lastDigraphIndex < 0) return 0;
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090083
84 // It's an interesting digraph if the second char matches too.
satok9df4a452012-03-23 16:05:18 +090085 if (digraphs[lastDigraphIndex].second == codes[i + 1]) {
Jean Chalardd3043382012-03-21 18:38:25 +090086 return digraphs[lastDigraphIndex].replacement;
87 } else {
88 return 0;
89 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +090090}
91
92// Mostly the same arguments as the non-recursive version, except:
93// codes is the original value. It points to the start of the work buffer, and gets passed as is.
94// codesSize is the size of the user input (thus, it is the size of codesSrc).
95// codesDest is the current point in the work buffer.
96// codesSrc is the current point in the user-input, original, content-unmodified buffer.
97// codesRemain is the remaining size in codesSrc.
satok1d7eaf82011-07-13 10:32:02 +090098void UnigramDictionary::getWordWithDigraphSuggestionsRec(ProximityInfo *proximityInfo,
satok1147c7b2011-12-14 15:04:58 +090099 const int *xcoordinates, const int *ycoordinates, const int *codesBuffer,
satok219a5142012-03-08 11:53:18 +0900100 int *xCoordinatesBuffer, int *yCoordinatesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900101 const int codesBufferSize, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900102 const bool useFullEditDistance, const int *codesSrc,
satok1147c7b2011-12-14 15:04:58 +0900103 const int codesRemain, const int currentDepth, int *codesDest, Correction *correction,
Jean Chalard6c300612012-03-06 19:54:03 +0900104 WordsPriorityQueuePool *queuePool,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900105 const digraph_t *const digraphs, const unsigned int digraphsSize) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900106
Ken Wakasaf2789812012-09-04 12:49:46 +0900107 const int startIndex = static_cast<int>(codesDest - codesBuffer);
Jean Chalard6c300612012-03-06 19:54:03 +0900108 if (currentDepth < MAX_DIGRAPH_SEARCH_DEPTH) {
Jean Chalarda787dba2011-03-04 12:17:48 +0900109 for (int i = 0; i < codesRemain; ++i) {
satok219a5142012-03-08 11:53:18 +0900110 xCoordinatesBuffer[startIndex + i] = xcoordinates[codesBufferSize - codesRemain + i];
111 yCoordinatesBuffer[startIndex + i] = ycoordinates[codesBufferSize - codesRemain + i];
Jean Chalardd3043382012-03-21 18:38:25 +0900112 const int replacementCodePoint =
113 getDigraphReplacement(codesSrc, i, codesRemain, digraphs, digraphsSize);
114 if (0 != replacementCodePoint) {
Jean Chalarda787dba2011-03-04 12:17:48 +0900115 // Found a digraph. We will try both spellings. eg. the word is "pruefen"
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900116
Jean Chalardd3043382012-03-21 18:38:25 +0900117 // Copy the word up to the first char of the digraph, including proximity chars,
118 // and overwrite the primary code with the replacement code point. Then, continue
119 // processing on the remaining part of the word, skipping the second char of the
120 // digraph.
121 // In our example, copy "pru", replace "u" with the version with the diaeresis and
122 // continue running on "fen".
Jean Chalarda787dba2011-03-04 12:17:48 +0900123 // Make i the index of the second char of the digraph for simplicity. Forgetting
124 // to do that results in an infinite recursion so take care!
125 ++i;
126 memcpy(codesDest, codesSrc, i * BYTES_IN_ONE_CHAR);
Jean Chalardd3043382012-03-21 18:38:25 +0900127 codesDest[(i - 1) * (BYTES_IN_ONE_CHAR / sizeof(codesDest[0]))] =
128 replacementCodePoint;
Jean Chalarda787dba2011-03-04 12:17:48 +0900129 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900130 codesBuffer, xCoordinatesBuffer, yCoordinatesBuffer, codesBufferSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900131 bigramMap, bigramFilter, useFullEditDistance, codesSrc + i + 1,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900132 codesRemain - i - 1, currentDepth + 1, codesDest + i, correction,
Jean Chalard6c300612012-03-06 19:54:03 +0900133 queuePool, digraphs, digraphsSize);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900134
Jean Chalarda787dba2011-03-04 12:17:48 +0900135 // Copy the second char of the digraph in place, then continue processing on
136 // the remaining part of the word.
137 // In our example, after "pru" in the buffer copy the "e", and continue on "fen"
satok9df4a452012-03-23 16:05:18 +0900138 memcpy(codesDest + i, codesSrc + i, BYTES_IN_ONE_CHAR);
Jean Chalarda787dba2011-03-04 12:17:48 +0900139 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900140 codesBuffer, xCoordinatesBuffer, yCoordinatesBuffer, codesBufferSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900141 bigramMap, bigramFilter, useFullEditDistance, codesSrc + i, codesRemain - i,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900142 currentDepth + 1, codesDest + i, correction, queuePool, digraphs,
143 digraphsSize);
Jean Chalarda787dba2011-03-04 12:17:48 +0900144 return;
145 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900146 }
147 }
148
149 // If we come here, we hit the end of the word: let's check it against the dictionary.
150 // In our example, we'll come here once for "prufen" and then once for "pruefen".
151 // If the word contains several digraphs, we'll come it for the product of them.
152 // eg. if the word is "ueberpruefen" we'll test, in order, against
153 // "uberprufen", "uberpruefen", "ueberprufen", "ueberpruefen".
154 const unsigned int remainingBytes = BYTES_IN_ONE_CHAR * codesRemain;
satok219a5142012-03-08 11:53:18 +0900155 if (0 != remainingBytes) {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900156 memcpy(codesDest, codesSrc, remainingBytes);
satok219a5142012-03-08 11:53:18 +0900157 memcpy(&xCoordinatesBuffer[startIndex], &xcoordinates[codesBufferSize - codesRemain],
158 sizeof(int) * codesRemain);
159 memcpy(&yCoordinatesBuffer[startIndex], &ycoordinates[codesBufferSize - codesRemain],
160 sizeof(int) * codesRemain);
161 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900162
satok219a5142012-03-08 11:53:18 +0900163 getWordSuggestions(proximityInfo, xCoordinatesBuffer, yCoordinatesBuffer, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900164 startIndex + codesRemain, bigramMap, bigramFilter, useFullEditDistance, correction,
satoka7e5a5a2011-12-15 16:49:12 +0900165 queuePool);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900166}
167
Jean Chalard8950ce62012-05-07 16:28:30 +0900168// bigramMap contains the association <bigram address> -> <bigram frequency>
169// bigramFilter is a bloom filter for fast rejection: see functions setInFilter and isInFilter
170// in bigram_dictionary.cpp
Ken Wakasaf2789812012-09-04 12:49:46 +0900171int UnigramDictionary::getSuggestions(ProximityInfo *proximityInfo, const int *xcoordinates,
Jean Chalard338d3ec2012-04-06 19:28:34 +0900172 const int *ycoordinates, const int *codes, const int codesSize,
Jean Chalard8950ce62012-05-07 16:28:30 +0900173 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard6931df92012-07-12 12:55:48 +0900174 const bool useFullEditDistance, unsigned short *outWords, int *frequencies,
175 int *outputTypes) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900176
satokb1ed1d42012-06-14 16:35:23 -0700177 WordsPriorityQueuePool queuePool(MAX_WORDS, SUB_QUEUE_MAX_WORDS, MAX_WORD_LENGTH);
178 queuePool.clearAll();
satok1bc038c2012-06-14 11:25:50 -0700179 Correction masterCorrection;
180 masterCorrection.resetCorrection();
Jean Chalardaa8df592012-04-06 18:54:20 +0900181 if (BinaryFormat::REQUIRES_GERMAN_UMLAUT_PROCESSING & FLAGS)
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900182 { // Incrementally tune the word and try all possibilities
satok9df4a452012-03-23 16:05:18 +0900183 int codesBuffer[getCodesBufferSize(codes, codesSize)];
satok219a5142012-03-08 11:53:18 +0900184 int xCoordinatesBuffer[codesSize];
185 int yCoordinatesBuffer[codesSize];
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900186 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900187 xCoordinatesBuffer, yCoordinatesBuffer, codesSize, bigramMap, bigramFilter,
satok1bc038c2012-06-14 11:25:50 -0700188 useFullEditDistance, codes, codesSize, 0, codesBuffer, &masterCorrection,
Ken Wakasab02ee3d2012-10-08 11:46:14 +0900189 &queuePool, GERMAN_UMLAUT_DIGRAPHS, NELEMS(GERMAN_UMLAUT_DIGRAPHS));
Jean Chalardaa8df592012-04-06 18:54:20 +0900190 } else if (BinaryFormat::REQUIRES_FRENCH_LIGATURES_PROCESSING & FLAGS) {
satokacb6c542012-03-23 20:58:18 +0900191 int codesBuffer[getCodesBufferSize(codes, codesSize)];
Jean Chalardcc78d032012-03-23 16:48:49 +0900192 int xCoordinatesBuffer[codesSize];
193 int yCoordinatesBuffer[codesSize];
194 getWordWithDigraphSuggestionsRec(proximityInfo, xcoordinates, ycoordinates, codesBuffer,
Jean Chalard8950ce62012-05-07 16:28:30 +0900195 xCoordinatesBuffer, yCoordinatesBuffer, codesSize, bigramMap, bigramFilter,
satok1bc038c2012-06-14 11:25:50 -0700196 useFullEditDistance, codes, codesSize, 0, codesBuffer, &masterCorrection,
Ken Wakasab02ee3d2012-10-08 11:46:14 +0900197 &queuePool, FRENCH_LIGATURES_DIGRAPHS, NELEMS(FRENCH_LIGATURES_DIGRAPHS));
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900198 } else { // Normal processing
Jean Chalard338d3ec2012-04-06 19:28:34 +0900199 getWordSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, codesSize,
satokb1ed1d42012-06-14 16:35:23 -0700200 bigramMap, bigramFilter, useFullEditDistance, &masterCorrection, &queuePool);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900201 }
202
satok817e5172011-03-04 06:06:45 -0800203 PROF_START(20);
satok8330b482012-01-23 16:52:37 +0900204 if (DEBUG_DICT) {
satokb1ed1d42012-06-14 16:35:23 -0700205 float ns = queuePool.getMasterQueue()->getHighestNormalizedScore(
satok1bc038c2012-06-14 11:25:50 -0700206 masterCorrection.getPrimaryInputWord(), codesSize, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900207 ns += 0;
208 AKLOGI("Max normalized score = %f", ns);
209 }
satoka7e5a5a2011-12-15 16:49:12 +0900210 const int suggestedWordsCount =
Jean Chalard926ef062012-08-10 13:14:45 +0900211 queuePool.getMasterQueue()->outputSuggestions(masterCorrection.getPrimaryInputWord(),
212 codesSize, frequencies, outWords, outputTypes);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900213
214 if (DEBUG_DICT) {
satokb1ed1d42012-06-14 16:35:23 -0700215 float ns = queuePool.getMasterQueue()->getHighestNormalizedScore(
satok1bc038c2012-06-14 11:25:50 -0700216 masterCorrection.getPrimaryInputWord(), codesSize, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900217 ns += 0;
satok9fb6f472012-01-13 18:01:22 +0900218 AKLOGI("Returning %d words", suggestedWordsCount);
Jean Chalard980d6b62011-06-30 17:02:23 +0900219 /// Print the returned words
220 for (int j = 0; j < suggestedWordsCount; ++j) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900221 short unsigned int *w = outWords + j * MAX_WORD_LENGTH;
Jean Chalard980d6b62011-06-30 17:02:23 +0900222 char s[MAX_WORD_LENGTH];
223 for (int i = 0; i <= MAX_WORD_LENGTH; i++) s[i] = w[i];
Ken Wakasa7914e902012-09-07 08:55:16 +0900224 (void)s; // To suppress compiler warning
satok9fb6f472012-01-13 18:01:22 +0900225 AKLOGI("%s %i", s, frequencies[j]);
Jean Chalard980d6b62011-06-30 17:02:23 +0900226 }
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900227 }
satok817e5172011-03-04 06:06:45 -0800228 PROF_END(20);
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900229 PROF_CLOSE;
230 return suggestedWordsCount;
231}
232
satok1d7eaf82011-07-13 10:32:02 +0900233void UnigramDictionary::getWordSuggestions(ProximityInfo *proximityInfo,
satok1147c7b2011-12-14 15:04:58 +0900234 const int *xcoordinates, const int *ycoordinates, const int *codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900235 const int inputSize, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900236 const bool useFullEditDistance, Correction *correction,
237 WordsPriorityQueuePool *queuePool) const {
Jean Chalardc2bbc6a2011-02-25 17:56:53 +0900238
satok61e2f852011-01-05 14:13:07 +0900239 PROF_OPEN;
240 PROF_START(0);
satok61e2f852011-01-05 14:13:07 +0900241 PROF_END(0);
satok30088252010-12-01 21:22:15 +0900242
satok61e2f852011-01-05 14:13:07 +0900243 PROF_START(1);
Jean Chalard8950ce62012-05-07 16:28:30 +0900244 getOneWordSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, bigramMap, bigramFilter,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900245 useFullEditDistance, inputSize, correction, queuePool);
satok61e2f852011-01-05 14:13:07 +0900246 PROF_END(1);
247
248 PROF_START(2);
satok10266c02011-08-19 22:05:59 +0900249 // Note: This line is intentionally left blank
satok61e2f852011-01-05 14:13:07 +0900250 PROF_END(2);
satokcdbbea72010-12-08 16:04:16 +0900251
satok61e2f852011-01-05 14:13:07 +0900252 PROF_START(3);
satok10266c02011-08-19 22:05:59 +0900253 // Note: This line is intentionally left blank
satok61e2f852011-01-05 14:13:07 +0900254 PROF_END(3);
satok30088252010-12-01 21:22:15 +0900255
satok61e2f852011-01-05 14:13:07 +0900256 PROF_START(4);
satok8330b482012-01-23 16:52:37 +0900257 bool hasAutoCorrectionCandidate = false;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900258 WordsPriorityQueue *masterQueue = queuePool->getMasterQueue();
satok8330b482012-01-23 16:52:37 +0900259 if (masterQueue->size() > 0) {
satok0028ed32012-05-16 20:42:12 +0900260 float nsForMaster = masterQueue->getHighestNormalizedScore(
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900261 correction->getPrimaryInputWord(), inputSize, 0, 0, 0);
satok8330b482012-01-23 16:52:37 +0900262 hasAutoCorrectionCandidate = (nsForMaster > START_TWO_WORDS_CORRECTION_THRESHOLD);
263 }
satok61e2f852011-01-05 14:13:07 +0900264 PROF_END(4);
satoka3d78f62010-12-09 22:08:33 +0900265
satok61e2f852011-01-05 14:13:07 +0900266 PROF_START(5);
satok99557162012-01-26 22:49:13 +0900267 // Multiple word suggestions
268 if (SUGGEST_MULTIPLE_WORDS
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900269 && inputSize >= MIN_USER_TYPED_LENGTH_FOR_MULTIPLE_WORD_SUGGESTION) {
satoka85f4922012-01-30 18:18:30 +0900270 getSplitMultipleWordsSuggestions(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900271 useFullEditDistance, inputSize, correction, queuePool,
satok1f6b52e2012-01-30 13:53:58 +0900272 hasAutoCorrectionCandidate);
satok662fe692010-12-08 17:05:39 +0900273 }
satok61e2f852011-01-05 14:13:07 +0900274 PROF_END(5);
satok817e5172011-03-04 06:06:45 -0800275
276 PROF_START(6);
satok99557162012-01-26 22:49:13 +0900277 // Note: This line is intentionally left blank
satok817e5172011-03-04 06:06:45 -0800278 PROF_END(6);
satok99557162012-01-26 22:49:13 +0900279
satok29dc8062012-01-17 15:59:15 +0900280 if (DEBUG_DICT) {
satok6ad15fc2012-01-16 16:21:21 +0900281 queuePool->dumpSubQueue1TopSuggestions();
satok29dc8062012-01-17 15:59:15 +0900282 for (int i = 0; i < SUB_QUEUE_MAX_COUNT; ++i) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900283 WordsPriorityQueue *queue = queuePool->getSubQueue(FIRST_WORD_INDEX, i);
satok29dc8062012-01-17 15:59:15 +0900284 if (queue->size() > 0) {
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900285 WordsPriorityQueue::SuggestedWord *sw = queue->top();
satok29dc8062012-01-17 15:59:15 +0900286 const int score = sw->mScore;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900287 const unsigned short *word = sw->mWord;
satok29dc8062012-01-17 15:59:15 +0900288 const int wordLength = sw->mWordLength;
satok0028ed32012-05-16 20:42:12 +0900289 float ns = Correction::RankingAlgorithm::calcNormalizedScore(
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900290 correction->getPrimaryInputWord(), i, word, wordLength, score);
satok29dc8062012-01-17 15:59:15 +0900291 ns += 0;
292 AKLOGI("--- TOP SUB WORDS for %d --- %d %f [%d]", i, score, ns,
satok54af64a2012-01-17 15:58:23 +0900293 (ns > TWO_WORDS_CORRECTION_WITH_OTHER_ERROR_THRESHOLD));
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900294 DUMP_WORD(correction->getPrimaryInputWord(), i);
satok29dc8062012-01-17 15:59:15 +0900295 DUMP_WORD(word, wordLength);
296 }
297 }
satok6ad15fc2012-01-16 16:21:21 +0900298 }
satok30088252010-12-01 21:22:15 +0900299}
300
Yusuke Nojima258bfe62011-09-28 12:59:43 +0900301void UnigramDictionary::initSuggestions(ProximityInfo *proximityInfo, const int *xCoordinates,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900302 const int *yCoordinates, const int *codes, const int inputSize,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900303 Correction *correction) const {
Ken Wakasade3070a2011-03-19 09:16:42 +0900304 if (DEBUG_DICT) {
satok9fb6f472012-01-13 18:01:22 +0900305 AKLOGI("initSuggest");
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900306 DUMP_WORD_INT(codes, inputSize);
Ken Wakasade3070a2011-03-19 09:16:42 +0900307 }
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900308 correction->initInputParams(proximityInfo, codes, inputSize, xCoordinates, yCoordinates);
309 const int maxDepth = min(inputSize * MAX_DEPTH_MULTIPLIER, MAX_WORD_LENGTH);
310 correction->initCorrection(proximityInfo, inputSize, maxDepth);
satok30088252010-12-01 21:22:15 +0900311}
312
satok744dab62011-12-15 22:29:05 +0900313void UnigramDictionary::getOneWordSuggestions(ProximityInfo *proximityInfo,
314 const int *xcoordinates, const int *ycoordinates, const int *codes,
Jean Chalard8950ce62012-05-07 16:28:30 +0900315 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900316 const bool useFullEditDistance, const int inputSize,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900317 Correction *correction, WordsPriorityQueuePool *queuePool) const {
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900318 initSuggestions(proximityInfo, xcoordinates, ycoordinates, codes, inputSize, correction);
319 getSuggestionCandidates(useFullEditDistance, inputSize, bigramMap, bigramFilter, correction,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900320 queuePool, true /* doAutoCompletion */, DEFAULT_MAX_ERRORS, FIRST_WORD_INDEX);
satok744dab62011-12-15 22:29:05 +0900321}
322
satok1147c7b2011-12-14 15:04:58 +0900323void UnigramDictionary::getSuggestionCandidates(const bool useFullEditDistance,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900324 const int inputSize, const std::map<int, int> *bigramMap, const uint8_t *bigramFilter,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900325 Correction *correction, WordsPriorityQueuePool *queuePool,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900326 const bool doAutoCompletion, const int maxErrors, const int currentWordIndex) const {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900327 uint8_t totalTraverseCount = correction->pushAndGetTotalTraverseCount();
328 if (DEBUG_DICT) {
329 AKLOGI("Traverse count %d", totalTraverseCount);
330 }
331 if (totalTraverseCount > MULTIPLE_WORDS_SUGGESTION_MAX_TOTAL_TRAVERSE_COUNT) {
332 if (DEBUG_DICT) {
333 AKLOGI("Abort traversing %d", totalTraverseCount);
334 }
335 return;
336 }
satok10266c02011-08-19 22:05:59 +0900337 // TODO: Remove setCorrectionParams
satok1147c7b2011-12-14 15:04:58 +0900338 correction->setCorrectionParams(0, 0, 0,
satokd03317c2011-12-14 21:38:11 +0900339 -1 /* spaceProximityPos */, -1 /* missingSpacePos */, useFullEditDistance,
satok1a6da632011-12-16 23:15:06 +0900340 doAutoCompletion, maxErrors);
satok662fe692010-12-08 17:05:39 +0900341 int rootPosition = ROOT_POS;
Jean Chalard980d6b62011-06-30 17:02:23 +0900342 // Get the number of children of root, then increment the position
Jean Chalard6d419812012-01-16 15:19:47 +0900343 int childCount = BinaryFormat::getGroupCountAndForwardPointer(DICT_ROOT, &rootPosition);
satok208268d2011-08-10 15:44:08 +0900344 int outputIndex = 0;
satokd2997922010-12-07 13:08:39 +0900345
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900346 correction->initCorrectionState(rootPosition, childCount, (inputSize <= 0));
satokd2997922010-12-07 13:08:39 +0900347
satok662fe692010-12-08 17:05:39 +0900348 // Depth first search
satok208268d2011-08-10 15:44:08 +0900349 while (outputIndex >= 0) {
satok1147c7b2011-12-14 15:04:58 +0900350 if (correction->initProcessState(outputIndex)) {
351 int siblingPos = correction->getTreeSiblingPos(outputIndex);
satokd2997922010-12-07 13:08:39 +0900352 int firstChildPos;
satok0f6c8e82011-08-03 02:19:44 +0900353
satok4e4e74e2011-08-03 23:27:32 +0900354 const bool needsToTraverseChildrenNodes = processCurrentNode(siblingPos,
Jean Chalard8950ce62012-05-07 16:28:30 +0900355 bigramMap, bigramFilter, correction, &childCount, &firstChildPos, &siblingPos,
Jean Chalard4d9b2022012-04-23 19:25:28 +0900356 queuePool, currentWordIndex);
satok662fe692010-12-08 17:05:39 +0900357 // Update next sibling pos
satok1147c7b2011-12-14 15:04:58 +0900358 correction->setTreeSiblingPos(outputIndex, siblingPos);
satok208268d2011-08-10 15:44:08 +0900359
satokd2997922010-12-07 13:08:39 +0900360 if (needsToTraverseChildrenNodes) {
361 // Goes to child node
satok1147c7b2011-12-14 15:04:58 +0900362 outputIndex = correction->goDownTree(outputIndex, childCount, firstChildPos);
satokd2997922010-12-07 13:08:39 +0900363 }
364 } else {
satokcdbbea72010-12-08 16:04:16 +0900365 // Goes to parent sibling node
satok1147c7b2011-12-14 15:04:58 +0900366 outputIndex = correction->getTreeParentIndex(outputIndex);
satokd2997922010-12-07 13:08:39 +0900367 }
368 }
369}
370
Jean Chalard171d1802012-04-23 18:51:00 +0900371inline void UnigramDictionary::onTerminal(const int probability,
Jean Chalardcf9dbbd2011-12-26 15:16:59 +0900372 const TerminalAttributes& terminalAttributes, Correction *correction,
satok8330b482012-01-23 16:52:37 +0900373 WordsPriorityQueuePool *queuePool, const bool addToMasterQueue,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900374 const int currentWordIndex) const {
satok6ad15fc2012-01-16 16:21:21 +0900375 const int inputIndex = correction->getInputIndex();
376 const bool addToSubQueue = inputIndex < SUB_QUEUE_MAX_COUNT;
satok54af64a2012-01-17 15:58:23 +0900377
satok8876b752011-08-04 18:31:57 +0900378 int wordLength;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900379 unsigned short *wordPointer;
satok54af64a2012-01-17 15:58:23 +0900380
satok1f6b52e2012-01-30 13:53:58 +0900381 if ((currentWordIndex == FIRST_WORD_INDEX) && addToMasterQueue) {
satok54af64a2012-01-17 15:58:23 +0900382 WordsPriorityQueue *masterQueue = queuePool->getMasterQueue();
Jean Chalard171d1802012-04-23 18:51:00 +0900383 const int finalProbability =
384 correction->getFinalProbability(probability, &wordPointer, &wordLength);
satok54af64a2012-01-17 15:58:23 +0900385
Jean Chalard72b1c932012-08-31 15:24:39 +0900386 if (0 != finalProbability && !terminalAttributes.isBlacklistedOrNotAWord()) {
Jean Chalard8af8c152012-08-17 09:37:39 +0900387 // If the probability is 0, we don't want to add this word. However we still
388 // want to add its shortcuts (including a possible whitelist entry) if any.
Jean Chalard72b1c932012-08-31 15:24:39 +0900389 // Furthermore, if this is not a word (shortcut only for example) or a blacklisted
390 // entry then we never want to suggest this.
Jean Chalard8af8c152012-08-17 09:37:39 +0900391 addWord(wordPointer, wordLength, finalProbability, masterQueue,
392 Dictionary::KIND_CORRECTION);
393 }
394
395 const int shortcutProbability = finalProbability > 0 ? finalProbability - 1 : 0;
396 // Please note that the shortcut candidates will be added to the master queue only.
397 TerminalAttributes::ShortcutIterator iterator =
398 terminalAttributes.getShortcutIterator();
399 while (iterator.hasNextShortcutTarget()) {
400 // TODO: addWord only supports weak ordering, meaning we have no means
401 // to control the order of the shortcuts relative to one another or to the word.
402 // We need to either modulate the probability of each shortcut according
403 // to its own shortcut probability or to make the queue
404 // so that the insert order is protected inside the queue for words
405 // with the same score. For the moment we use -1 to make sure the shortcut will
406 // never be in front of the word.
407 uint16_t shortcutTarget[MAX_WORD_LENGTH_INTERNAL];
408 int shortcutFrequency;
409 const int shortcutTargetStringLength = iterator.getNextShortcutTarget(
410 MAX_WORD_LENGTH_INTERNAL, shortcutTarget, &shortcutFrequency);
411 int shortcutScore;
412 int kind;
413 if (shortcutFrequency == BinaryFormat::WHITELIST_SHORTCUT_FREQUENCY
414 && correction->sameAsTyped()) {
415 shortcutScore = S_INT_MAX;
416 kind = Dictionary::KIND_WHITELIST;
417 } else {
418 shortcutScore = shortcutProbability;
419 kind = Dictionary::KIND_CORRECTION;
satok6ad15fc2012-01-16 16:21:21 +0900420 }
Jean Chalard8af8c152012-08-17 09:37:39 +0900421 addWord(shortcutTarget, shortcutTargetStringLength, shortcutScore,
422 masterQueue, kind);
Jean Chalardcf9dbbd2011-12-26 15:16:59 +0900423 }
satok54af64a2012-01-17 15:58:23 +0900424 }
satok6ad15fc2012-01-16 16:21:21 +0900425
satok54af64a2012-01-17 15:58:23 +0900426 // We only allow two words + other error correction for words with SUB_QUEUE_MIN_WORD_LENGTH
427 // or more length.
428 if (inputIndex >= SUB_QUEUE_MIN_WORD_LENGTH && addToSubQueue) {
satok8330b482012-01-23 16:52:37 +0900429 WordsPriorityQueue *subQueue;
satok7409d152012-01-26 16:13:25 +0900430 subQueue = queuePool->getSubQueue(currentWordIndex, inputIndex);
431 if (!subQueue) {
satok8330b482012-01-23 16:52:37 +0900432 return;
433 }
Jean Chalard171d1802012-04-23 18:51:00 +0900434 const int finalProbability = correction->getFinalProbabilityForSubQueue(
435 probability, &wordPointer, &wordLength, inputIndex);
Jean Chalard926ef062012-08-10 13:14:45 +0900436 addWord(wordPointer, wordLength, finalProbability, subQueue, Dictionary::KIND_CORRECTION);
Jean Chalardca5ef282011-06-17 15:36:26 +0900437 }
438}
439
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900440int UnigramDictionary::getSubStringSuggestion(
satok7409d152012-01-26 16:13:25 +0900441 ProximityInfo *proximityInfo, const int *xcoordinates, const int *ycoordinates,
satok3c09bb12012-01-26 18:36:19 +0900442 const int *codes, const bool useFullEditDistance, Correction *correction,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900443 WordsPriorityQueuePool *queuePool, const int inputSize,
satok3c09bb12012-01-26 18:36:19 +0900444 const bool hasAutoCorrectionCandidate, const int currentWordIndex,
445 const int inputWordStartPos, const int inputWordLength,
satok99557162012-01-26 22:49:13 +0900446 const int outputWordStartPos, const bool isSpaceProximity, int *freqArray,
Jean Chalard18ebba32012-09-06 20:37:55 +0900447 int *wordLengthArray, unsigned short *outputWord, int *outputWordLength) const {
Satoshi Kataoka67e3cc82012-05-31 15:04:58 +0900448 if (inputWordLength > MULTIPLE_WORDS_SUGGESTION_MAX_WORD_LENGTH) {
449 return FLAG_MULTIPLE_SUGGEST_ABORT;
450 }
451
452 /////////////////////////////////////////////
453 // safety net for multiple word suggestion //
454 // TODO: Remove this safety net //
455 /////////////////////////////////////////////
456 int smallWordCount = 0;
457 int singleLetterWordCount = 0;
458 if (inputWordLength == 1) {
459 ++singleLetterWordCount;
460 }
461 if (inputWordLength <= 2) {
462 // small word == single letter or 2-letter word
463 ++smallWordCount;
464 }
465 for (int i = 0; i < currentWordIndex; ++i) {
466 const int length = wordLengthArray[i];
467 if (length == 1) {
468 ++singleLetterWordCount;
469 // Safety net to avoid suggesting sequential single letter words
470 if (i < (currentWordIndex - 1)) {
471 if (wordLengthArray[i + 1] == 1) {
472 return FLAG_MULTIPLE_SUGGEST_ABORT;
473 }
474 } else if (inputWordLength == 1) {
475 return FLAG_MULTIPLE_SUGGEST_ABORT;
476 }
477 }
478 if (length <= 2) {
479 ++smallWordCount;
480 }
481 // Safety net to avoid suggesting multiple words with many (4 or more, for now) small words
482 if (singleLetterWordCount >= 3 || smallWordCount >= 4) {
483 return FLAG_MULTIPLE_SUGGEST_ABORT;
484 }
485 }
486 //////////////////////////////////////////////
487 // TODO: Remove the safety net above //
488 //////////////////////////////////////////////
489
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900490 unsigned short *tempOutputWord = 0;
satok1f6b52e2012-01-30 13:53:58 +0900491 int nextWordLength = 0;
satok99557162012-01-26 22:49:13 +0900492 // TODO: Optimize init suggestion
493 initSuggestions(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900494 inputSize, correction);
satok99557162012-01-26 22:49:13 +0900495
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900496 unsigned short word[MAX_WORD_LENGTH_INTERNAL];
satok3c09bb12012-01-26 18:36:19 +0900497 int freq = getMostFrequentWordLike(
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900498 inputWordStartPos, inputWordLength, correction, word);
satok3c09bb12012-01-26 18:36:19 +0900499 if (freq > 0) {
satok1f6b52e2012-01-30 13:53:58 +0900500 nextWordLength = inputWordLength;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900501 tempOutputWord = word;
satok3c09bb12012-01-26 18:36:19 +0900502 } else if (!hasAutoCorrectionCandidate) {
503 if (inputWordStartPos > 0) {
504 const int offset = inputWordStartPos;
505 initSuggestions(proximityInfo, &xcoordinates[offset], &ycoordinates[offset],
satok9df4a452012-03-23 16:05:18 +0900506 codes + offset, inputWordLength, correction);
satok3c09bb12012-01-26 18:36:19 +0900507 queuePool->clearSubQueue(currentWordIndex);
Jean Chalard4d9b2022012-04-23 19:25:28 +0900508 // TODO: pass the bigram list for substring suggestion
509 getSuggestionCandidates(useFullEditDistance, inputWordLength,
Jean Chalard8950ce62012-05-07 16:28:30 +0900510 0 /* bigramMap */, 0 /* bigramFilter */, correction, queuePool,
511 false /* doAutoCompletion */, MAX_ERRORS_FOR_TWO_WORDS, currentWordIndex);
satok3c09bb12012-01-26 18:36:19 +0900512 if (DEBUG_DICT) {
satok1f6b52e2012-01-30 13:53:58 +0900513 if (currentWordIndex < MULTIPLE_WORDS_SUGGESTION_MAX_WORDS) {
satok3c09bb12012-01-26 18:36:19 +0900514 AKLOGI("Dump word candidates(%d) %d", currentWordIndex, inputWordLength);
515 for (int i = 0; i < SUB_QUEUE_MAX_COUNT; ++i) {
516 queuePool->getSubQueue(currentWordIndex, i)->dumpTopWord();
517 }
518 }
519 }
520 }
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900521 WordsPriorityQueue *queue = queuePool->getSubQueue(currentWordIndex, inputWordLength);
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900522 // TODO: Return the correct value depending on doAutoCompletion
523 if (!queue || queue->size() <= 0) {
524 return FLAG_MULTIPLE_SUGGEST_ABORT;
satok3c09bb12012-01-26 18:36:19 +0900525 }
526 int score = 0;
satok0028ed32012-05-16 20:42:12 +0900527 const float ns = queue->getHighestNormalizedScore(
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900528 correction->getPrimaryInputWord(), inputWordLength,
satok1f6b52e2012-01-30 13:53:58 +0900529 &tempOutputWord, &score, &nextWordLength);
satok3c09bb12012-01-26 18:36:19 +0900530 if (DEBUG_DICT) {
531 AKLOGI("NS(%d) = %f, Score = %d", currentWordIndex, ns, score);
532 }
533 // Two words correction won't be done if the score of the first word doesn't exceed the
534 // threshold.
535 if (ns < TWO_WORDS_CORRECTION_WITH_OTHER_ERROR_THRESHOLD
satok1f6b52e2012-01-30 13:53:58 +0900536 || nextWordLength < SUB_QUEUE_MIN_WORD_LENGTH) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900537 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900538 }
satok1f6b52e2012-01-30 13:53:58 +0900539 freq = score >> (nextWordLength + TWO_WORDS_PLUS_OTHER_ERROR_CORRECTION_DEMOTION_DIVIDER);
satok3c09bb12012-01-26 18:36:19 +0900540 }
541 if (DEBUG_DICT) {
Jean Chalard18ebba32012-09-06 20:37:55 +0900542 AKLOGI("Freq(%d): %d, length: %d, input length: %d, input start: %d (%d)",
543 currentWordIndex, freq, nextWordLength, inputWordLength, inputWordStartPos,
544 (currentWordIndex > 0) ? wordLengthArray[0] : 0);
satok3c09bb12012-01-26 18:36:19 +0900545 }
satok1f6b52e2012-01-30 13:53:58 +0900546 if (freq <= 0 || nextWordLength <= 0
547 || MAX_WORD_LENGTH <= (outputWordStartPos + nextWordLength)) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900548 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900549 }
satok1f6b52e2012-01-30 13:53:58 +0900550 for (int i = 0; i < nextWordLength; ++i) {
satok3c09bb12012-01-26 18:36:19 +0900551 outputWord[outputWordStartPos + i] = tempOutputWord[i];
552 }
satok99557162012-01-26 22:49:13 +0900553
554 // Put output values
satok1f6b52e2012-01-30 13:53:58 +0900555 freqArray[currentWordIndex] = freq;
satok99557162012-01-26 22:49:13 +0900556 // TODO: put output length instead of input length
satok1f6b52e2012-01-30 13:53:58 +0900557 wordLengthArray[currentWordIndex] = inputWordLength;
558 const int tempOutputWordLength = outputWordStartPos + nextWordLength;
559 if (outputWordLength) {
560 *outputWordLength = tempOutputWordLength;
561 }
satok99557162012-01-26 22:49:13 +0900562
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900563 if ((inputWordStartPos + inputWordLength) < inputSize) {
satok1f6b52e2012-01-30 13:53:58 +0900564 if (outputWordStartPos + nextWordLength >= MAX_WORD_LENGTH) {
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900565 return FLAG_MULTIPLE_SUGGEST_SKIP;
satok3c09bb12012-01-26 18:36:19 +0900566 }
Ken Wakasab02ee3d2012-10-08 11:46:14 +0900567 outputWord[tempOutputWordLength] = KEYCODE_SPACE;
satok1f6b52e2012-01-30 13:53:58 +0900568 if (outputWordLength) {
569 ++*outputWordLength;
570 }
571 } else if (currentWordIndex >= 1) {
satok99557162012-01-26 22:49:13 +0900572 // TODO: Handle 3 or more words
satoka85f4922012-01-30 18:18:30 +0900573 const int pairFreq = correction->getFreqForSplitMultipleWords(
574 freqArray, wordLengthArray, currentWordIndex + 1, isSpaceProximity, outputWord);
satok99557162012-01-26 22:49:13 +0900575 if (DEBUG_DICT) {
satoka85f4922012-01-30 18:18:30 +0900576 DUMP_WORD(outputWord, tempOutputWordLength);
satoka0ac31f2012-05-23 19:55:27 +0900577 for (int i = 0; i < currentWordIndex + 1; ++i) {
578 AKLOGI("Split %d,%d words: freq = %d, length = %d", i, currentWordIndex + 1,
579 freqArray[i], wordLengthArray[i]);
580 }
581 AKLOGI("Split two words: freq = %d, length = %d, %d, isSpace ? %d", pairFreq,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900582 inputSize, tempOutputWordLength, isSpaceProximity);
satok99557162012-01-26 22:49:13 +0900583 }
Jean Chalard926ef062012-08-10 13:14:45 +0900584 addWord(outputWord, tempOutputWordLength, pairFreq, queuePool->getMasterQueue(),
585 Dictionary::KIND_CORRECTION);
satok3c09bb12012-01-26 18:36:19 +0900586 }
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900587 return FLAG_MULTIPLE_SUGGEST_CONTINUE;
satok7409d152012-01-26 16:13:25 +0900588}
589
satok1f6b52e2012-01-30 13:53:58 +0900590void UnigramDictionary::getMultiWordsSuggestionRec(ProximityInfo *proximityInfo,
591 const int *xcoordinates, const int *ycoordinates, const int *codes,
Ken Wakasaf2789812012-09-04 12:49:46 +0900592 const bool useFullEditDistance, const int inputSize, Correction *correction,
593 WordsPriorityQueuePool *queuePool, const bool hasAutoCorrectionCandidate,
594 const int startInputPos, const int startWordIndex, const int outputWordLength,
595 int *freqArray, int *wordLengthArray, unsigned short *outputWord) const {
satok1f6b52e2012-01-30 13:53:58 +0900596 if (startWordIndex >= (MULTIPLE_WORDS_SUGGESTION_MAX_WORDS - 1)) {
597 // Return if the last word index
598 return;
599 }
satoka85f4922012-01-30 18:18:30 +0900600 if (startWordIndex >= 1
601 && (hasAutoCorrectionCandidate
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900602 || inputSize < MIN_INPUT_LENGTH_FOR_THREE_OR_MORE_WORDS_CORRECTION)) {
satoka85f4922012-01-30 18:18:30 +0900603 // Do not suggest 3+ words if already has auto correction candidate
604 return;
605 }
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900606 for (int i = startInputPos + 1; i < inputSize; ++i) {
satok1f6b52e2012-01-30 13:53:58 +0900607 if (DEBUG_CORRECTION_FREQ) {
satoka85f4922012-01-30 18:18:30 +0900608 AKLOGI("Multi words(%d), start in %d sep %d start out %d",
609 startWordIndex, startInputPos, i, outputWordLength);
610 DUMP_WORD(outputWord, outputWordLength);
satok1f6b52e2012-01-30 13:53:58 +0900611 }
satoka85f4922012-01-30 18:18:30 +0900612 int tempOutputWordLength = 0;
613 // Current word
614 int inputWordStartPos = startInputPos;
615 int inputWordLength = i - startInputPos;
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900616 const int suggestionFlag = getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900617 codes, useFullEditDistance, correction, queuePool, inputSize,
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900618 hasAutoCorrectionCandidate, startWordIndex, inputWordStartPos, inputWordLength,
619 outputWordLength, true /* not used */, freqArray, wordLengthArray, outputWord,
620 &tempOutputWordLength);
621 if (suggestionFlag == FLAG_MULTIPLE_SUGGEST_ABORT) {
622 // TODO: break here
623 continue;
624 } else if (suggestionFlag == FLAG_MULTIPLE_SUGGEST_SKIP) {
satok1f6b52e2012-01-30 13:53:58 +0900625 continue;
626 }
627
satoka85f4922012-01-30 18:18:30 +0900628 if (DEBUG_CORRECTION_FREQ) {
629 AKLOGI("Do missing space correction");
630 }
631 // Next word
satok1f6b52e2012-01-30 13:53:58 +0900632 // Missing space
633 inputWordStartPos = i;
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900634 inputWordLength = inputSize - i;
Ken Wakasaf2789812012-09-04 12:49:46 +0900635 if (getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900636 useFullEditDistance, correction, queuePool, inputSize, hasAutoCorrectionCandidate,
satoka85f4922012-01-30 18:18:30 +0900637 startWordIndex + 1, inputWordStartPos, inputWordLength, tempOutputWordLength,
Satoshi Kataoka6cbe2042012-05-30 17:28:34 +0900638 false /* missing space */, freqArray, wordLengthArray, outputWord, 0)
639 != FLAG_MULTIPLE_SUGGEST_CONTINUE) {
satoka85f4922012-01-30 18:18:30 +0900640 getMultiWordsSuggestionRec(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900641 useFullEditDistance, inputSize, correction, queuePool,
satoka85f4922012-01-30 18:18:30 +0900642 hasAutoCorrectionCandidate, inputWordStartPos, startWordIndex + 1,
643 tempOutputWordLength, freqArray, wordLengthArray, outputWord);
644 }
satok1f6b52e2012-01-30 13:53:58 +0900645
646 // Mistyped space
647 ++inputWordStartPos;
648 --inputWordLength;
649
650 if (inputWordLength <= 0) {
651 continue;
652 }
653
654 const int x = xcoordinates[inputWordStartPos - 1];
655 const int y = ycoordinates[inputWordStartPos - 1];
656 if (!proximityInfo->hasSpaceProximity(x, y)) {
657 continue;
658 }
659
satoka85f4922012-01-30 18:18:30 +0900660 if (DEBUG_CORRECTION_FREQ) {
661 AKLOGI("Do mistyped space correction");
662 }
satok1f6b52e2012-01-30 13:53:58 +0900663 getSubStringSuggestion(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900664 useFullEditDistance, correction, queuePool, inputSize, hasAutoCorrectionCandidate,
satoka85f4922012-01-30 18:18:30 +0900665 startWordIndex + 1, inputWordStartPos, inputWordLength, tempOutputWordLength,
666 true /* mistyped space */, freqArray, wordLengthArray, outputWord, 0);
satok1f6b52e2012-01-30 13:53:58 +0900667 }
668}
669
satoka85f4922012-01-30 18:18:30 +0900670void UnigramDictionary::getSplitMultipleWordsSuggestions(ProximityInfo *proximityInfo,
satok744dab62011-12-15 22:29:05 +0900671 const int *xcoordinates, const int *ycoordinates, const int *codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900672 const bool useFullEditDistance, const int inputSize,
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900673 Correction *correction, WordsPriorityQueuePool *queuePool,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900674 const bool hasAutoCorrectionCandidate) const {
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900675 if (inputSize >= MAX_WORD_LENGTH) return;
satok612c6e42011-08-01 19:35:27 +0900676 if (DEBUG_DICT) {
satok1f6b52e2012-01-30 13:53:58 +0900677 AKLOGI("--- Suggest multiple words");
678 }
satok54af64a2012-01-17 15:58:23 +0900679
satokbd6ccdd2012-01-23 12:30:20 +0900680 // Allocating fixed length array on stack
681 unsigned short outputWord[MAX_WORD_LENGTH];
satok1f6b52e2012-01-30 13:53:58 +0900682 int freqArray[MULTIPLE_WORDS_SUGGESTION_MAX_WORDS];
683 int wordLengthArray[MULTIPLE_WORDS_SUGGESTION_MAX_WORDS];
684 const int outputWordLength = 0;
685 const int startInputPos = 0;
686 const int startWordIndex = 0;
687 getMultiWordsSuggestionRec(proximityInfo, xcoordinates, ycoordinates, codes,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900688 useFullEditDistance, inputSize, correction, queuePool, hasAutoCorrectionCandidate,
satok1f6b52e2012-01-30 13:53:58 +0900689 startInputPos, startWordIndex, outputWordLength, freqArray, wordLengthArray,
690 outputWord);
Jean Chalarde6715e32011-06-30 19:47:25 +0900691}
692
Jean Chalard1059f272011-06-28 20:45:05 +0900693// Wrapper for getMostFrequentWordLikeInner, which matches it to the previous
694// interface.
695inline int UnigramDictionary::getMostFrequentWordLike(const int startInputIndex,
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900696 const int inputSize, Correction *correction, unsigned short *word) const {
697 uint16_t inWord[inputSize];
Jean Chalard1059f272011-06-28 20:45:05 +0900698
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900699 for (int i = 0; i < inputSize; ++i) {
Satoshi Kataoka4a3db702012-06-08 15:29:44 +0900700 inWord[i] = (uint16_t)correction->getPrimaryCharAt(startInputIndex + i);
Jean Chalard1059f272011-06-28 20:45:05 +0900701 }
Satoshi Kataoka687a2442012-08-23 15:46:43 +0900702 return getMostFrequentWordLikeInner(inWord, inputSize, word);
Jean Chalard1059f272011-06-28 20:45:05 +0900703}
704
705// This function will take the position of a character array within a CharGroup,
706// and check it actually like-matches the word in inWord starting at startInputIndex,
707// that is, it matches it with case and accents squashed.
708// The function returns true if there was a full match, false otherwise.
709// The function will copy on-the-fly the characters in the CharGroup to outNewWord.
710// It will also place the end position of the array in outPos; in outInputIndex,
711// it will place the index of the first char AFTER the match if there was a match,
712// and the initial position if there was not. It makes sense because if there was
713// a match we want to continue searching, but if there was not, we want to go to
714// the next CharGroup.
715// In and out parameters may point to the same location. This function takes care
716// not to use any input parameters after it wrote into its outputs.
717static inline bool testCharGroupForContinuedLikeness(const uint8_t flags,
Ken Wakasaf2789812012-09-04 12:49:46 +0900718 const uint8_t *const root, const int startPos, const uint16_t *const inWord,
719 const int startInputIndex, const int inputSize, int32_t *outNewWord, int *outInputIndex,
720 int *outPos) {
Jean Chalard19560502012-07-31 23:45:32 +0900721 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
Jean Chalard1059f272011-06-28 20:45:05 +0900722 int pos = startPos;
Ken Wakasaf2789812012-09-04 12:49:46 +0900723 int32_t codePoint = BinaryFormat::getCodePointAndForwardPointer(root, &pos);
724 int32_t baseChar = toBaseLowerCase(codePoint);
Tadashi G. Takaoka6e3cb272011-11-11 14:26:13 +0900725 const uint16_t wChar = toBaseLowerCase(inWord[startInputIndex]);
Jean Chalard1059f272011-06-28 20:45:05 +0900726
727 if (baseChar != wChar) {
728 *outPos = hasMultipleChars ? BinaryFormat::skipOtherCharacters(root, pos) : pos;
729 *outInputIndex = startInputIndex;
730 return false;
731 }
732 int inputIndex = startInputIndex;
Ken Wakasaf2789812012-09-04 12:49:46 +0900733 outNewWord[inputIndex] = codePoint;
Jean Chalard1059f272011-06-28 20:45:05 +0900734 if (hasMultipleChars) {
Ken Wakasaf2789812012-09-04 12:49:46 +0900735 codePoint = BinaryFormat::getCodePointAndForwardPointer(root, &pos);
736 while (NOT_A_CODE_POINT != codePoint) {
737 baseChar = toBaseLowerCase(codePoint);
738 if (inputIndex + 1 >= inputSize || toBaseLowerCase(inWord[++inputIndex]) != baseChar) {
Jean Chalard1059f272011-06-28 20:45:05 +0900739 *outPos = BinaryFormat::skipOtherCharacters(root, pos);
740 *outInputIndex = startInputIndex;
741 return false;
742 }
Ken Wakasaf2789812012-09-04 12:49:46 +0900743 outNewWord[inputIndex] = codePoint;
744 codePoint = BinaryFormat::getCodePointAndForwardPointer(root, &pos);
Jean Chalard1059f272011-06-28 20:45:05 +0900745 }
746 }
747 *outInputIndex = inputIndex + 1;
748 *outPos = pos;
749 return true;
750}
751
752// This function is invoked when a word like the word searched for is found.
753// It will compare the frequency to the max frequency, and if greater, will
754// copy the word into the output buffer. In output value maxFreq, it will
755// write the new maximum frequency if it changed.
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900756static inline void onTerminalWordLike(const int freq, int32_t *newWord, const int length,
757 short unsigned int *outWord, int *maxFreq) {
Jean Chalard1059f272011-06-28 20:45:05 +0900758 if (freq > *maxFreq) {
Ken Wakasaf2789812012-09-04 12:49:46 +0900759 for (int q = 0; q < length; ++q) {
Jean Chalard1059f272011-06-28 20:45:05 +0900760 outWord[q] = newWord[q];
Ken Wakasaf2789812012-09-04 12:49:46 +0900761 }
Jean Chalard1059f272011-06-28 20:45:05 +0900762 outWord[length] = 0;
763 *maxFreq = freq;
764 }
765}
766
767// Will find the highest frequency of the words like the one passed as an argument,
768// that is, everything that only differs by case/accents.
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900769int UnigramDictionary::getMostFrequentWordLikeInner(const uint16_t *const inWord,
Ken Wakasaf2789812012-09-04 12:49:46 +0900770 const int inputSize, short unsigned int *outWord) const {
Jean Chalard1059f272011-06-28 20:45:05 +0900771 int32_t newWord[MAX_WORD_LENGTH_INTERNAL];
772 int depth = 0;
773 int maxFreq = -1;
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900774 const uint8_t *const root = DICT_ROOT;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900775 int stackChildCount[MAX_WORD_LENGTH_INTERNAL];
776 int stackInputIndex[MAX_WORD_LENGTH_INTERNAL];
777 int stackSiblingPos[MAX_WORD_LENGTH_INTERNAL];
Jean Chalard1059f272011-06-28 20:45:05 +0900778
Jean Chalard4c0eca62012-01-16 15:15:53 +0900779 int startPos = 0;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900780 stackChildCount[0] = BinaryFormat::getGroupCountAndForwardPointer(root, &startPos);
781 stackInputIndex[0] = 0;
782 stackSiblingPos[0] = startPos;
Jean Chalard1059f272011-06-28 20:45:05 +0900783 while (depth >= 0) {
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900784 const int charGroupCount = stackChildCount[depth];
785 int pos = stackSiblingPos[depth];
Jean Chalard1059f272011-06-28 20:45:05 +0900786 for (int charGroupIndex = charGroupCount - 1; charGroupIndex >= 0; --charGroupIndex) {
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900787 int inputIndex = stackInputIndex[depth];
Jean Chalard1059f272011-06-28 20:45:05 +0900788 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(root, &pos);
789 // Test whether all chars in this group match with the word we are searching for. If so,
Ken Wakasaf2789812012-09-04 12:49:46 +0900790 // we want to traverse its children (or if the inputSize match, evaluate its frequency).
Jean Chalard1059f272011-06-28 20:45:05 +0900791 // Note that this function will output the position regardless, but will only write
792 // into inputIndex if there is a match.
793 const bool isAlike = testCharGroupForContinuedLikeness(flags, root, pos, inWord,
Ken Wakasaf2789812012-09-04 12:49:46 +0900794 inputIndex, inputSize, newWord, &inputIndex, &pos);
795 if (isAlike && (BinaryFormat::FLAG_IS_TERMINAL & flags) && (inputIndex == inputSize)) {
Jean Chalard1059f272011-06-28 20:45:05 +0900796 const int frequency = BinaryFormat::readFrequencyWithoutMovingPointer(root, pos);
797 onTerminalWordLike(frequency, newWord, inputIndex, outWord, &maxFreq);
798 }
799 pos = BinaryFormat::skipFrequency(flags, pos);
800 const int siblingPos = BinaryFormat::skipChildrenPosAndAttributes(root, flags, pos);
801 const int childrenNodePos = BinaryFormat::readChildrenPosition(root, flags, pos);
802 // If we had a match and the word has children, we want to traverse them. We don't have
803 // to traverse words longer than the one we are searching for, since they will not match
Ken Wakasaf2789812012-09-04 12:49:46 +0900804 // anyway, so don't traverse unless inputIndex < inputSize.
805 if (isAlike && (-1 != childrenNodePos) && (inputIndex < inputSize)) {
Jean Chalard1059f272011-06-28 20:45:05 +0900806 // Save position for this depth, to get back to this once children are done
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900807 stackChildCount[depth] = charGroupIndex;
808 stackSiblingPos[depth] = siblingPos;
Jean Chalard1059f272011-06-28 20:45:05 +0900809 // Prepare stack values for next depth
810 ++depth;
811 int childrenPos = childrenNodePos;
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900812 stackChildCount[depth] =
Jean Chalard1059f272011-06-28 20:45:05 +0900813 BinaryFormat::getGroupCountAndForwardPointer(root, &childrenPos);
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900814 stackSiblingPos[depth] = childrenPos;
815 stackInputIndex[depth] = inputIndex;
Jean Chalard1059f272011-06-28 20:45:05 +0900816 pos = childrenPos;
817 // Go to the next depth level.
818 ++depth;
819 break;
820 } else {
821 // No match, or no children, or word too long to ever match: go the next sibling.
822 pos = siblingPos;
823 }
824 }
825 --depth;
826 }
827 return maxFreq;
828}
829
Ken Wakasa0bbb9172012-07-25 17:51:43 +0900830int UnigramDictionary::getFrequency(const int32_t *const inWord, const int length) const {
831 const uint8_t *const root = DICT_ROOT;
Jean Chalarde9a86e22012-06-28 21:01:29 +0900832 int pos = BinaryFormat::getTerminalPosition(root, inWord, length,
833 false /* forceLowerCaseSearch */);
Satoshi Kataoka2f854e12012-05-29 15:58:13 +0900834 if (NOT_VALID_WORD == pos) {
835 return NOT_A_PROBABILITY;
836 }
837 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(root, &pos);
Jean Chalard72b1c932012-08-31 15:24:39 +0900838 if (flags & (BinaryFormat::FLAG_IS_BLACKLISTED | BinaryFormat::FLAG_IS_NOT_A_WORD)) {
839 // If this is not a word, or if it's a blacklisted entry, it should behave as
840 // having no frequency outside of the suggestion process (where it should be used
841 // for shortcuts).
842 return NOT_A_PROBABILITY;
843 }
Jean Chalard19560502012-07-31 23:45:32 +0900844 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
Satoshi Kataoka2f854e12012-05-29 15:58:13 +0900845 if (hasMultipleChars) {
846 pos = BinaryFormat::skipOtherCharacters(root, pos);
847 } else {
Ken Wakasaf2789812012-09-04 12:49:46 +0900848 BinaryFormat::getCodePointAndForwardPointer(DICT_ROOT, &pos);
Satoshi Kataoka2f854e12012-05-29 15:58:13 +0900849 }
850 const int unigramFreq = BinaryFormat::readFrequencyWithoutMovingPointer(root, pos);
851 return unigramFreq;
Jean Chalard1059f272011-06-28 20:45:05 +0900852}
853
854// TODO: remove this function.
855int UnigramDictionary::getBigramPosition(int pos, unsigned short *word, int offset,
856 int length) const {
857 return -1;
858}
859
860// ProcessCurrentNode returns a boolean telling whether to traverse children nodes or not.
861// If the return value is false, then the caller should read in the output "nextSiblingPosition"
862// to find out the address of the next sibling node and pass it to a new call of processCurrentNode.
863// It is worthy to note that when false is returned, the output values other than
864// nextSiblingPosition are undefined.
865// If the return value is true, then the caller must proceed to traverse the children of this
866// node. processCurrentNode will output the information about the children: their count in
867// newCount, their position in newChildrenPosition, the traverseAllNodes flag in
868// newTraverseAllNodes, the match weight into newMatchRate, the input index into newInputIndex, the
869// diffs into newDiffs, the sibling position in nextSiblingPosition, and the output index into
870// newOutputIndex. Please also note the following caveat: processCurrentNode does not know when
871// there aren't any more nodes at this level, it merely returns the address of the first byte after
872// the current node in nextSiblingPosition. Thus, the caller must keep count of the nodes at any
873// given level, as output into newCount when traversing this level's parent.
satok8876b752011-08-04 18:31:57 +0900874inline bool UnigramDictionary::processCurrentNode(const int initialPos,
Jean Chalard8950ce62012-05-07 16:28:30 +0900875 const std::map<int, int> *bigramMap, const uint8_t *bigramFilter, Correction *correction,
876 int *newCount, int *newChildrenPosition, int *nextSiblingPosition,
Satoshi Kataoka6bc051d2012-06-08 19:52:19 +0900877 WordsPriorityQueuePool *queuePool, const int currentWordIndex) const {
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900878 if (DEBUG_DICT) {
satokcfca3c62011-08-10 14:30:10 +0900879 correction->checkState();
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900880 }
Jean Chalard0584f022011-06-30 19:23:16 +0900881 int pos = initialPos;
Jean Chalard0584f022011-06-30 19:23:16 +0900882
Jean Chalard1059f272011-06-28 20:45:05 +0900883 // Flags contain the following information:
884 // - Address type (MASK_GROUP_ADDRESS_TYPE) on two bits:
885 // - FLAG_GROUP_ADDRESS_TYPE_{ONE,TWO,THREE}_BYTES means there are children and their address
886 // is on the specified number of bytes.
887 // - FLAG_GROUP_ADDRESS_TYPE_NOADDRESS means there are no children, and therefore no address.
888 // - FLAG_HAS_MULTIPLE_CHARS: whether this node has multiple char or not.
889 // - FLAG_IS_TERMINAL: whether this node is a terminal or not (it may still have children)
890 // - FLAG_HAS_BIGRAMS: whether this node has bigrams or not
891 const uint8_t flags = BinaryFormat::getFlagsAndForwardPointer(DICT_ROOT, &pos);
Jean Chalard19560502012-07-31 23:45:32 +0900892 const bool hasMultipleChars = (0 != (BinaryFormat::FLAG_HAS_MULTIPLE_CHARS & flags));
893 const bool isTerminalNode = (0 != (BinaryFormat::FLAG_IS_TERMINAL & flags));
satok8876b752011-08-04 18:31:57 +0900894
895 bool needsToInvokeOnTerminal = false;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900896
Jean Chalard1059f272011-06-28 20:45:05 +0900897 // This gets only ONE character from the stream. Next there will be:
898 // if FLAG_HAS_MULTIPLE CHARS: the other characters of the same node
899 // else if FLAG_IS_TERMINAL: the frequency
900 // else if MASK_GROUP_ADDRESS_TYPE is not NONE: the children address
901 // Note that you can't have a node that both is not a terminal and has no children.
Ken Wakasaf2789812012-09-04 12:49:46 +0900902 int32_t c = BinaryFormat::getCodePointAndForwardPointer(DICT_ROOT, &pos);
903 assert(NOT_A_CODE_POINT != c);
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900904
Jean Chalard1059f272011-06-28 20:45:05 +0900905 // We are going to loop through each character and make it look like it's a different
906 // node each time. To do that, we will process characters in this node in order until
Ken Wakasaf2789812012-09-04 12:49:46 +0900907 // we find the character terminator. This is signalled by getCodePoint* returning
908 // NOT_A_CODE_POINT.
Jean Chalard1059f272011-06-28 20:45:05 +0900909 // As a special case, if there is only one character in this node, we must not read the
Ken Wakasaf2789812012-09-04 12:49:46 +0900910 // next bytes so we will simulate the NOT_A_CODE_POINT return by testing the flags.
Jean Chalard1059f272011-06-28 20:45:05 +0900911 // This way, each loop run will look like a "virtual node".
912 do {
913 // We prefetch the next char. If 'c' is the last char of this node, we will have
Ken Wakasaf2789812012-09-04 12:49:46 +0900914 // NOT_A_CODE_POINT in the next char. From this we can decide whether this virtual node
Jean Chalard1059f272011-06-28 20:45:05 +0900915 // should behave as a terminal or not and whether we have children.
916 const int32_t nextc = hasMultipleChars
Ken Wakasaf2789812012-09-04 12:49:46 +0900917 ? BinaryFormat::getCodePointAndForwardPointer(DICT_ROOT, &pos) : NOT_A_CODE_POINT;
918 const bool isLastChar = (NOT_A_CODE_POINT == nextc);
Jean Chalard1059f272011-06-28 20:45:05 +0900919 // If there are more chars in this nodes, then this virtual node is not a terminal.
920 // If we are on the last char, this virtual node is a terminal if this node is.
satok8876b752011-08-04 18:31:57 +0900921 const bool isTerminal = isLastChar && isTerminalNode;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900922
satokcfca3c62011-08-10 14:30:10 +0900923 Correction::CorrectionType stateType = correction->processCharAndCalcState(
satok8876b752011-08-04 18:31:57 +0900924 c, isTerminal);
satokcfca3c62011-08-10 14:30:10 +0900925 if (stateType == Correction::TRAVERSE_ALL_ON_TERMINAL
926 || stateType == Correction::ON_TERMINAL) {
satok8876b752011-08-04 18:31:57 +0900927 needsToInvokeOnTerminal = true;
satokd03317c2011-12-14 21:38:11 +0900928 } else if (stateType == Correction::UNRELATED || correction->needsToPrune()) {
satok8876b752011-08-04 18:31:57 +0900929 // We found that this is an unrelated character, so we should give up traversing
930 // this node and its children entirely.
931 // However we may not be on the last virtual node yet so we skip the remaining
932 // characters in this node, the frequency if it's there, read the next sibling
933 // position to output it, then return false.
934 // We don't have to output other values because we return false, as in
935 // "don't traverse children".
Jean Chalard1059f272011-06-28 20:45:05 +0900936 if (!isLastChar) {
937 pos = BinaryFormat::skipOtherCharacters(DICT_ROOT, pos);
938 }
939 pos = BinaryFormat::skipFrequency(flags, pos);
940 *nextSiblingPosition =
941 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
942 return false;
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900943 }
944
Jean Chalard1059f272011-06-28 20:45:05 +0900945 // Prepare for the next character. Promote the prefetched char to current char - the loop
946 // will take care of prefetching the next. If we finally found our last char, nextc will
Ken Wakasaf2789812012-09-04 12:49:46 +0900947 // contain NOT_A_CODE_POINT.
Jean Chalard1059f272011-06-28 20:45:05 +0900948 c = nextc;
Ken Wakasaf2789812012-09-04 12:49:46 +0900949 } while (NOT_A_CODE_POINT != c);
Jean Chalard85a1d1e2011-06-21 22:23:21 +0900950
satok8876b752011-08-04 18:31:57 +0900951 if (isTerminalNode) {
satok6ad15fc2012-01-16 16:21:21 +0900952 // The frequency should be here, because we come here only if this is actually
953 // a terminal node, and we are on its last char.
Jean Chalard171d1802012-04-23 18:51:00 +0900954 const int unigramFreq = BinaryFormat::readFrequencyWithoutMovingPointer(DICT_ROOT, pos);
satok6ad15fc2012-01-16 16:21:21 +0900955 const int childrenAddressPos = BinaryFormat::skipFrequency(flags, pos);
956 const int attributesPos = BinaryFormat::skipChildrenPosition(flags, childrenAddressPos);
957 TerminalAttributes terminalAttributes(DICT_ROOT, flags, attributesPos);
Jean Chalard8950ce62012-05-07 16:28:30 +0900958 // bigramMap contains the bigram frequencies indexed by addresses for fast lookup.
959 // bigramFilter is a bloom filter of said frequencies for even faster rejection.
Jean Chalard49ba1352012-05-07 20:14:00 +0900960 const int probability = BinaryFormat::getProbability(initialPos, bigramMap, bigramFilter,
961 unigramFreq);
Jean Chalard171d1802012-04-23 18:51:00 +0900962 onTerminal(probability, terminalAttributes, correction, queuePool, needsToInvokeOnTerminal,
satok8330b482012-01-23 16:52:37 +0900963 currentWordIndex);
Jean Chalard1059f272011-06-28 20:45:05 +0900964
satok8876b752011-08-04 18:31:57 +0900965 // If there are more chars in this node, then this virtual node has children.
966 // If we are on the last char, this virtual node has children if this node has.
967 const bool hasChildren = BinaryFormat::hasChildrenInFlags(flags);
968
969 // This character matched the typed character (enough to traverse the node at least)
970 // so we just evaluated it. Now we should evaluate this virtual node's children - that
971 // is, if it has any. If it has no children, we're done here - so we skip the end of
972 // the node, output the siblings position, and return false "don't traverse children".
973 // Note that !hasChildren implies isLastChar, so we know we don't have to skip any
974 // remaining char in this group for there can't be any.
975 if (!hasChildren) {
976 pos = BinaryFormat::skipFrequency(flags, pos);
977 *nextSiblingPosition =
978 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
979 return false;
980 }
981
982 // Optimization: Prune out words that are too long compared to how much was typed.
satokcfca3c62011-08-10 14:30:10 +0900983 if (correction->needsToPrune()) {
satok8876b752011-08-04 18:31:57 +0900984 pos = BinaryFormat::skipFrequency(flags, pos);
985 *nextSiblingPosition =
986 BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
satok10266c02011-08-19 22:05:59 +0900987 if (DEBUG_DICT_FULL) {
satok9fb6f472012-01-13 18:01:22 +0900988 AKLOGI("Traversing was pruned.");
satok10266c02011-08-19 22:05:59 +0900989 }
satok8876b752011-08-04 18:31:57 +0900990 return false;
991 }
992 }
Jean Chalard1059f272011-06-28 20:45:05 +0900993
994 // Now we finished processing this node, and we want to traverse children. If there are no
995 // children, we can't come here.
996 assert(BinaryFormat::hasChildrenInFlags(flags));
997
998 // If this node was a terminal it still has the frequency under the pointer (it may have been
999 // read, but not skipped - see readFrequencyWithoutMovingPointer).
1000 // Next come the children position, then possibly attributes (attributes are bigrams only for
1001 // now, maybe something related to shortcuts in the future).
1002 // Once this is read, we still need to output the number of nodes in the immediate children of
1003 // this node, so we read and output it before returning true, as in "please traverse children".
1004 pos = BinaryFormat::skipFrequency(flags, pos);
1005 int childrenPos = BinaryFormat::readChildrenPosition(DICT_ROOT, flags, pos);
1006 *nextSiblingPosition = BinaryFormat::skipChildrenPosAndAttributes(DICT_ROOT, flags, pos);
1007 *newCount = BinaryFormat::getGroupCountAndForwardPointer(DICT_ROOT, &childrenPos);
1008 *newChildrenPosition = childrenPos;
1009 return true;
Jean Chalard85a1d1e2011-06-21 22:23:21 +09001010}
satok30088252010-12-01 21:22:15 +09001011} // namespace latinime