2009-03-13 22:11:42 +00:00
|
|
|
/*
|
2010-03-26 22:07:10 +00:00
|
|
|
* Copyright (C) 2008 The Android Open Source Project
|
2011-01-07 06:01:51 +00:00
|
|
|
*
|
2013-01-21 12:52:57 +00:00
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
2011-01-07 06:01:51 +00:00
|
|
|
*
|
2013-01-21 12:52:57 +00:00
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
2011-01-07 06:01:51 +00:00
|
|
|
*
|
2009-03-13 22:11:42 +00:00
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
2013-01-21 12:52:57 +00:00
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
2009-03-13 22:11:42 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
package com.android.inputmethod.latin;
|
|
|
|
|
2012-04-27 06:50:21 +00:00
|
|
|
import android.text.TextUtils;
|
2014-02-21 06:09:37 +00:00
|
|
|
import android.util.Log;
|
2012-08-14 13:49:59 +00:00
|
|
|
import android.util.SparseArray;
|
2010-11-29 08:57:48 +00:00
|
|
|
|
2013-09-17 09:07:16 +00:00
|
|
|
import com.android.inputmethod.annotations.UsedForTesting;
|
2011-10-04 01:54:23 +00:00
|
|
|
import com.android.inputmethod.keyboard.ProximityInfo;
|
2012-06-12 18:31:46 +00:00
|
|
|
import com.android.inputmethod.latin.SuggestedWords.SuggestedWordInfo;
|
2014-02-04 12:39:44 +00:00
|
|
|
import com.android.inputmethod.latin.makedict.DictionaryHeader;
|
|
|
|
import com.android.inputmethod.latin.makedict.FormatSpec;
|
2014-03-05 08:42:36 +00:00
|
|
|
import com.android.inputmethod.latin.makedict.FormatSpec.DictionaryOptions;
|
2014-02-04 12:39:44 +00:00
|
|
|
import com.android.inputmethod.latin.makedict.UnsupportedFormatException;
|
2014-02-06 06:13:33 +00:00
|
|
|
import com.android.inputmethod.latin.makedict.WordProperty;
|
2013-07-22 03:43:37 +00:00
|
|
|
import com.android.inputmethod.latin.settings.NativeSuggestOptions;
|
2014-03-27 08:46:35 +00:00
|
|
|
import com.android.inputmethod.latin.utils.BinaryDictionaryUtils;
|
2013-06-23 16:11:32 +00:00
|
|
|
import com.android.inputmethod.latin.utils.CollectionUtils;
|
2014-03-27 08:46:35 +00:00
|
|
|
import com.android.inputmethod.latin.utils.FileUtils;
|
2013-06-23 16:11:32 +00:00
|
|
|
import com.android.inputmethod.latin.utils.JniUtils;
|
2014-01-15 07:04:05 +00:00
|
|
|
import com.android.inputmethod.latin.utils.LanguageModelParam;
|
2013-06-23 16:11:32 +00:00
|
|
|
import com.android.inputmethod.latin.utils.StringUtils;
|
2011-10-04 01:54:23 +00:00
|
|
|
|
2013-09-24 13:57:15 +00:00
|
|
|
import java.io.File;
|
2012-06-12 18:31:46 +00:00
|
|
|
import java.util.ArrayList;
|
2009-03-13 22:11:42 +00:00
|
|
|
import java.util.Arrays;
|
2014-02-04 12:39:44 +00:00
|
|
|
import java.util.HashMap;
|
2012-04-04 12:39:23 +00:00
|
|
|
import java.util.Locale;
|
2009-03-13 22:11:42 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Implements a static, compacted, binary dictionary of standard words.
|
|
|
|
*/
|
2013-09-25 09:32:40 +00:00
|
|
|
// TODO: All methods which should be locked need to have a suffix "Locked".
|
2012-09-27 09:16:16 +00:00
|
|
|
public final class BinaryDictionary extends Dictionary {
|
2012-10-08 02:46:14 +00:00
|
|
|
private static final String TAG = BinaryDictionary.class.getSimpleName();
|
2011-03-14 18:46:15 +00:00
|
|
|
|
2013-01-22 04:14:53 +00:00
|
|
|
// Must be equal to MAX_WORD_LENGTH in native/jni/src/defines.h
|
2013-07-26 03:35:11 +00:00
|
|
|
private static final int MAX_WORD_LENGTH = Constants.DICTIONARY_MAX_WORD_LENGTH;
|
2013-01-22 04:14:53 +00:00
|
|
|
// Must be equal to MAX_RESULTS in native/jni/src/defines.h
|
2013-01-11 16:18:00 +00:00
|
|
|
private static final int MAX_RESULTS = 18;
|
2013-10-01 08:30:40 +00:00
|
|
|
// The cutoff returned by native for auto-commit confidence.
|
|
|
|
// Must be equal to CONFIDENCE_TO_AUTO_COMMIT in native/jni/src/defines.h
|
|
|
|
private static final int CONFIDENCE_TO_AUTO_COMMIT = 1000000;
|
2009-03-13 22:11:42 +00:00
|
|
|
|
2013-09-27 14:12:12 +00:00
|
|
|
@UsedForTesting
|
|
|
|
public static final String UNIGRAM_COUNT_QUERY = "UNIGRAM_COUNT";
|
|
|
|
@UsedForTesting
|
|
|
|
public static final String BIGRAM_COUNT_QUERY = "BIGRAM_COUNT";
|
2013-10-07 08:05:24 +00:00
|
|
|
@UsedForTesting
|
|
|
|
public static final String MAX_UNIGRAM_COUNT_QUERY = "MAX_UNIGRAM_COUNT";
|
|
|
|
@UsedForTesting
|
|
|
|
public static final String MAX_BIGRAM_COUNT_QUERY = "MAX_BIGRAM_COUNT";
|
2013-09-27 14:12:12 +00:00
|
|
|
|
2013-12-13 08:09:16 +00:00
|
|
|
public static final int NOT_A_VALID_TIMESTAMP = -1;
|
|
|
|
|
2014-01-31 02:06:42 +00:00
|
|
|
// Format to get unigram flags from native side via getWordPropertyNative().
|
|
|
|
private static final int FORMAT_WORD_PROPERTY_OUTPUT_FLAG_COUNT = 4;
|
|
|
|
private static final int FORMAT_WORD_PROPERTY_IS_NOT_A_WORD_INDEX = 0;
|
|
|
|
private static final int FORMAT_WORD_PROPERTY_IS_BLACKLISTED_INDEX = 1;
|
|
|
|
private static final int FORMAT_WORD_PROPERTY_HAS_BIGRAMS_INDEX = 2;
|
|
|
|
private static final int FORMAT_WORD_PROPERTY_HAS_SHORTCUTS_INDEX = 3;
|
2013-12-13 08:09:16 +00:00
|
|
|
|
2014-01-31 11:32:44 +00:00
|
|
|
// Format to get probability and historical info from native side via getWordPropertyNative().
|
|
|
|
public static final int FORMAT_WORD_PROPERTY_OUTPUT_PROBABILITY_INFO_COUNT = 4;
|
|
|
|
public static final int FORMAT_WORD_PROPERTY_PROBABILITY_INDEX = 0;
|
|
|
|
public static final int FORMAT_WORD_PROPERTY_TIMESTAMP_INDEX = 1;
|
|
|
|
public static final int FORMAT_WORD_PROPERTY_LEVEL_INDEX = 2;
|
|
|
|
public static final int FORMAT_WORD_PROPERTY_COUNT_INDEX = 3;
|
2013-12-13 08:09:16 +00:00
|
|
|
|
2014-03-27 08:46:35 +00:00
|
|
|
public static final String DICT_FILE_NAME_SUFFIX_FOR_MIGRATION = ".migrate";
|
|
|
|
|
2011-10-31 11:44:01 +00:00
|
|
|
private long mNativeDict;
|
2012-08-14 13:49:59 +00:00
|
|
|
private final Locale mLocale;
|
2013-09-12 09:47:56 +00:00
|
|
|
private final long mDictSize;
|
2013-09-17 09:07:16 +00:00
|
|
|
private final String mDictFilePath;
|
2014-02-21 06:09:37 +00:00
|
|
|
private final boolean mIsUpdatable;
|
2014-04-03 09:08:23 +00:00
|
|
|
private boolean mHasUpdated;
|
|
|
|
|
2012-08-10 08:16:39 +00:00
|
|
|
private final int[] mInputCodePoints = new int[MAX_WORD_LENGTH];
|
2014-03-07 10:36:19 +00:00
|
|
|
private final int[] mOutputSuggestionCount = new int[1];
|
2012-10-29 09:06:22 +00:00
|
|
|
private final int[] mOutputCodePoints = new int[MAX_WORD_LENGTH * MAX_RESULTS];
|
2013-01-11 09:59:01 +00:00
|
|
|
private final int[] mSpaceIndices = new int[MAX_RESULTS];
|
2012-07-10 11:18:10 +00:00
|
|
|
private final int[] mOutputScores = new int[MAX_RESULTS];
|
2012-07-12 03:55:48 +00:00
|
|
|
private final int[] mOutputTypes = new int[MAX_RESULTS];
|
2013-09-30 12:39:43 +00:00
|
|
|
// Only one result is ever used
|
|
|
|
private final int[] mOutputAutoCommitFirstWordConfidence = new int[1];
|
2014-04-02 16:01:51 +00:00
|
|
|
private final float[] mInputOutputLanguageWeight = new float[1];
|
2009-03-13 22:11:42 +00:00
|
|
|
|
2013-05-24 17:21:04 +00:00
|
|
|
private final NativeSuggestOptions mNativeSuggestOptions = new NativeSuggestOptions();
|
2012-08-14 13:49:59 +00:00
|
|
|
|
|
|
|
private final SparseArray<DicTraverseSession> mDicTraverseSessions =
|
2012-08-22 05:22:20 +00:00
|
|
|
CollectionUtils.newSparseArray();
|
2012-08-20 05:37:16 +00:00
|
|
|
|
|
|
|
// TODO: There should be a way to remove used DicTraverseSession objects from
|
|
|
|
// {@code mDicTraverseSessions}.
|
2012-10-03 06:19:43 +00:00
|
|
|
private DicTraverseSession getTraverseSession(final int traverseSessionId) {
|
2012-08-20 05:37:16 +00:00
|
|
|
synchronized(mDicTraverseSessions) {
|
|
|
|
DicTraverseSession traverseSession = mDicTraverseSessions.get(traverseSessionId);
|
|
|
|
if (traverseSession == null) {
|
2012-08-14 13:49:59 +00:00
|
|
|
traverseSession = mDicTraverseSessions.get(traverseSessionId);
|
|
|
|
if (traverseSession == null) {
|
2013-09-12 09:47:56 +00:00
|
|
|
traverseSession = new DicTraverseSession(mLocale, mNativeDict, mDictSize);
|
2012-08-14 13:49:59 +00:00
|
|
|
mDicTraverseSessions.put(traverseSessionId, traverseSession);
|
|
|
|
}
|
|
|
|
}
|
2012-08-20 05:37:16 +00:00
|
|
|
return traverseSession;
|
2012-08-14 13:49:59 +00:00
|
|
|
}
|
|
|
|
}
|
2009-03-13 22:11:42 +00:00
|
|
|
|
|
|
|
/**
|
2011-04-26 12:49:09 +00:00
|
|
|
* Constructor for the binary dictionary. This is supposed to be called from the
|
|
|
|
* dictionary factory.
|
|
|
|
* @param filename the name of the file to read through native code.
|
|
|
|
* @param offset the offset of the dictionary data within the file.
|
|
|
|
* @param length the length of the binary data.
|
2012-04-06 10:12:05 +00:00
|
|
|
* @param useFullEditDistance whether to use the full edit distance in suggestions
|
2012-06-27 08:31:09 +00:00
|
|
|
* @param dictType the dictionary type, as a human-readable string
|
2013-06-25 08:39:06 +00:00
|
|
|
* @param isUpdatable whether to open the dictionary file in writable mode.
|
2009-03-13 22:11:42 +00:00
|
|
|
*/
|
2013-01-11 09:59:01 +00:00
|
|
|
public BinaryDictionary(final String filename, final long offset, final long length,
|
2013-06-25 08:39:06 +00:00
|
|
|
final boolean useFullEditDistance, final Locale locale, final String dictType,
|
|
|
|
final boolean isUpdatable) {
|
2012-06-27 08:31:09 +00:00
|
|
|
super(dictType);
|
2012-08-14 13:49:59 +00:00
|
|
|
mLocale = locale;
|
2013-09-12 09:47:56 +00:00
|
|
|
mDictSize = length;
|
2013-09-17 09:07:16 +00:00
|
|
|
mDictFilePath = filename;
|
2014-02-21 06:09:37 +00:00
|
|
|
mIsUpdatable = isUpdatable;
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = false;
|
2013-05-24 17:21:04 +00:00
|
|
|
mNativeSuggestOptions.setUseFullEditDistance(useFullEditDistance);
|
2013-06-25 08:39:06 +00:00
|
|
|
loadDictionary(filename, offset, length, isUpdatable);
|
2010-08-20 05:35:02 +00:00
|
|
|
}
|
|
|
|
|
2011-02-25 09:21:02 +00:00
|
|
|
static {
|
2012-03-08 08:07:02 +00:00
|
|
|
JniUtils.loadNativeLibrary();
|
2011-02-25 09:21:02 +00:00
|
|
|
}
|
2011-02-25 08:56:53 +00:00
|
|
|
|
2013-06-25 05:25:01 +00:00
|
|
|
private static native long openNative(String sourceDir, long dictOffset, long dictSize,
|
|
|
|
boolean isUpdatable);
|
2014-02-04 12:39:44 +00:00
|
|
|
private static native void getHeaderInfoNative(long dict, int[] outHeaderSize,
|
|
|
|
int[] outFormatVersion, ArrayList<int[]> outAttributeKeys,
|
|
|
|
ArrayList<int[]> outAttributeValues);
|
2013-09-17 09:07:16 +00:00
|
|
|
private static native void flushNative(long dict, String filePath);
|
2013-09-30 04:57:54 +00:00
|
|
|
private static native boolean needsToRunGCNative(long dict, boolean mindsBlockByGC);
|
2013-09-17 09:07:16 +00:00
|
|
|
private static native void flushWithGCNative(long dict, String filePath);
|
2013-01-11 16:18:00 +00:00
|
|
|
private static native void closeNative(long dict);
|
2013-12-13 08:09:16 +00:00
|
|
|
private static native int getFormatVersionNative(long dict);
|
2013-03-18 04:08:31 +00:00
|
|
|
private static native int getProbabilityNative(long dict, int[] word);
|
2013-09-18 02:18:28 +00:00
|
|
|
private static native int getBigramProbabilityNative(long dict, int[] word0, int[] word1);
|
2014-01-31 02:06:42 +00:00
|
|
|
private static native void getWordPropertyNative(long dict, int[] word,
|
2014-01-31 11:32:44 +00:00
|
|
|
int[] outCodePoints, boolean[] outFlags, int[] outProbabilityInfo,
|
|
|
|
ArrayList<int[]> outBigramTargets, ArrayList<int[]> outBigramProbabilityInfo,
|
|
|
|
ArrayList<int[]> outShortcutTargets, ArrayList<Integer> outShortcutProbabilities);
|
2014-02-03 05:51:58 +00:00
|
|
|
private static native int getNextWordNative(long dict, int token, int[] outCodePoints);
|
2014-03-07 10:36:19 +00:00
|
|
|
private static native void getSuggestionsNative(long dict, long proximityInfo,
|
2013-01-11 16:18:00 +00:00
|
|
|
long traverseSession, int[] xCoordinates, int[] yCoordinates, int[] times,
|
2014-03-20 09:47:45 +00:00
|
|
|
int[] pointerIds, int[] inputCodePoints, int inputSize, int[] suggestOptions,
|
|
|
|
int[] prevWordCodePointArray, int[] outputSuggestionCount, int[] outputCodePoints,
|
|
|
|
int[] outputScores, int[] outputIndices, int[] outputTypes,
|
2014-04-02 16:01:51 +00:00
|
|
|
int[] outputAutoCommitFirstWordConfidence, float[] inOutLanguageWeight);
|
2013-12-13 08:09:16 +00:00
|
|
|
private static native void addUnigramWordNative(long dict, int[] word, int probability,
|
|
|
|
int[] shortcutTarget, int shortcutProbability, boolean isNotAWord,
|
|
|
|
boolean isBlacklisted, int timestamp);
|
2013-06-26 03:51:33 +00:00
|
|
|
private static native void addBigramWordsNative(long dict, int[] word0, int[] word1,
|
2013-12-13 08:09:16 +00:00
|
|
|
int probability, int timestamp);
|
2013-06-26 03:51:33 +00:00
|
|
|
private static native void removeBigramWordsNative(long dict, int[] word0, int[] word1);
|
2013-12-13 08:09:16 +00:00
|
|
|
private static native int addMultipleDictionaryEntriesNative(long dict,
|
|
|
|
LanguageModelParam[] languageModelParams, int startIndex);
|
2013-09-18 02:18:28 +00:00
|
|
|
private static native int calculateProbabilityNative(long dict, int unigramProbability,
|
|
|
|
int bigramProbability);
|
2013-09-27 14:12:12 +00:00
|
|
|
private static native String getPropertyNative(long dict, String query);
|
2014-02-21 06:09:37 +00:00
|
|
|
private static native boolean isCorruptedNative(long dict);
|
2010-08-20 05:35:02 +00:00
|
|
|
|
2012-08-09 14:23:08 +00:00
|
|
|
// TODO: Move native dict into session
|
2012-10-03 06:19:43 +00:00
|
|
|
private final void loadDictionary(final String path, final long startOffset,
|
2013-06-25 08:39:06 +00:00
|
|
|
final long length, final boolean isUpdatable) {
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = false;
|
2013-06-25 08:39:06 +00:00
|
|
|
mNativeDict = openNative(path, startOffset, length, isUpdatable);
|
2010-08-20 05:35:02 +00:00
|
|
|
}
|
|
|
|
|
2014-02-21 06:09:37 +00:00
|
|
|
// TODO: Check isCorrupted() for main dictionaries.
|
|
|
|
public boolean isCorrupted() {
|
|
|
|
if (!isValidDictionary()) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
if (!isCorruptedNative(mNativeDict)) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
// TODO: Record the corruption.
|
|
|
|
Log.e(TAG, "BinaryDictionary (" + mDictFilePath + ") is corrupted.");
|
|
|
|
Log.e(TAG, "locale: " + mLocale);
|
|
|
|
Log.e(TAG, "dict size: " + mDictSize);
|
|
|
|
Log.e(TAG, "updatable: " + mIsUpdatable);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-02-04 12:39:44 +00:00
|
|
|
public DictionaryHeader getHeader() throws UnsupportedFormatException {
|
|
|
|
if (mNativeDict == 0) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
final int[] outHeaderSize = new int[1];
|
|
|
|
final int[] outFormatVersion = new int[1];
|
|
|
|
final ArrayList<int[]> outAttributeKeys = CollectionUtils.newArrayList();
|
|
|
|
final ArrayList<int[]> outAttributeValues = CollectionUtils.newArrayList();
|
|
|
|
getHeaderInfoNative(mNativeDict, outHeaderSize, outFormatVersion, outAttributeKeys,
|
|
|
|
outAttributeValues);
|
|
|
|
final HashMap<String, String> attributes = new HashMap<String, String>();
|
|
|
|
for (int i = 0; i < outAttributeKeys.size(); i++) {
|
|
|
|
final String attributeKey = StringUtils.getStringFromNullTerminatedCodePointArray(
|
|
|
|
outAttributeKeys.get(i));
|
|
|
|
final String attributeValue = StringUtils.getStringFromNullTerminatedCodePointArray(
|
|
|
|
outAttributeValues.get(i));
|
|
|
|
attributes.put(attributeKey, attributeValue);
|
|
|
|
}
|
2014-02-06 08:55:45 +00:00
|
|
|
final boolean hasHistoricalInfo = DictionaryHeader.ATTRIBUTE_VALUE_TRUE.equals(
|
|
|
|
attributes.get(DictionaryHeader.HAS_HISTORICAL_INFO_KEY));
|
2014-02-04 12:39:44 +00:00
|
|
|
return new DictionaryHeader(outHeaderSize[0], new DictionaryOptions(attributes),
|
|
|
|
new FormatSpec.FormatOptions(outFormatVersion[0], hasHistoricalInfo));
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-07-09 09:25:27 +00:00
|
|
|
@Override
|
|
|
|
public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
|
2013-05-01 11:36:36 +00:00
|
|
|
final String prevWord, final ProximityInfo proximityInfo,
|
2014-04-02 15:47:28 +00:00
|
|
|
final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
|
|
|
|
final float[] inOutLanguageWeight) {
|
2013-05-01 11:36:36 +00:00
|
|
|
return getSuggestionsWithSessionId(composer, prevWord, proximityInfo, blockOffensiveWords,
|
2014-04-02 15:47:28 +00:00
|
|
|
additionalFeaturesOptions, 0 /* sessionId */, inOutLanguageWeight);
|
2012-08-14 13:49:59 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ArrayList<SuggestedWordInfo> getSuggestionsWithSessionId(final WordComposer composer,
|
2013-05-01 11:36:36 +00:00
|
|
|
final String prevWord, final ProximityInfo proximityInfo,
|
2013-08-28 10:24:28 +00:00
|
|
|
final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
|
2014-04-02 15:47:28 +00:00
|
|
|
final int sessionId, final float[] inOutLanguageWeight) {
|
2012-07-10 11:21:13 +00:00
|
|
|
if (!isValidDictionary()) return null;
|
2012-08-14 13:49:59 +00:00
|
|
|
|
2012-08-21 05:05:57 +00:00
|
|
|
Arrays.fill(mInputCodePoints, Constants.NOT_A_CODE);
|
2012-07-10 11:26:34 +00:00
|
|
|
// TODO: toLowerCase in the native code
|
2012-07-11 07:07:08 +00:00
|
|
|
final int[] prevWordCodePointArray = (null == prevWord)
|
2012-10-03 06:19:43 +00:00
|
|
|
? null : StringUtils.toCodePointArray(prevWord);
|
2014-03-28 04:25:04 +00:00
|
|
|
final int composerSize = composer.sizeWithoutTrailingSingleQuotes();
|
2012-07-10 11:43:19 +00:00
|
|
|
|
2012-07-10 11:50:52 +00:00
|
|
|
final boolean isGesture = composer.isBatchMode();
|
|
|
|
if (composerSize <= 1 || !isGesture) {
|
2012-07-10 11:43:19 +00:00
|
|
|
if (composerSize > MAX_WORD_LENGTH - 1) return null;
|
2014-03-27 09:43:45 +00:00
|
|
|
for (int i = 0; i < composerSize; i++) {
|
|
|
|
mInputCodePoints[i] = composer.getCodeAt(i);
|
2012-07-10 11:43:19 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-11 03:56:57 +00:00
|
|
|
final InputPointers ips = composer.getInputPointers();
|
2013-01-11 16:18:00 +00:00
|
|
|
final int inputSize = isGesture ? ips.getPointerSize() : composerSize;
|
2013-05-24 17:21:04 +00:00
|
|
|
mNativeSuggestOptions.setIsGesture(isGesture);
|
2013-08-28 10:24:28 +00:00
|
|
|
mNativeSuggestOptions.setAdditionalFeaturesOptions(additionalFeaturesOptions);
|
2014-04-02 16:01:51 +00:00
|
|
|
if (inOutLanguageWeight != null) {
|
|
|
|
mInputOutputLanguageWeight[0] = inOutLanguageWeight[0];
|
|
|
|
} else {
|
|
|
|
mInputOutputLanguageWeight[0] = Dictionary.NOT_A_LANGUAGE_WEIGHT;
|
|
|
|
}
|
2012-07-11 03:56:57 +00:00
|
|
|
// proximityInfo and/or prevWordForBigrams may not be null.
|
2014-03-07 10:36:19 +00:00
|
|
|
getSuggestionsNative(mNativeDict, proximityInfo.getNativeProximityInfo(),
|
2013-01-11 09:59:01 +00:00
|
|
|
getTraverseSession(sessionId).getSession(), ips.getXCoordinates(),
|
|
|
|
ips.getYCoordinates(), ips.getTimes(), ips.getPointerIds(), mInputCodePoints,
|
2014-03-20 09:47:45 +00:00
|
|
|
inputSize, mNativeSuggestOptions.getOptions(),
|
2014-03-07 10:36:19 +00:00
|
|
|
prevWordCodePointArray, mOutputSuggestionCount, mOutputCodePoints, mOutputScores,
|
2014-04-02 16:01:51 +00:00
|
|
|
mSpaceIndices, mOutputTypes, mOutputAutoCommitFirstWordConfidence,
|
|
|
|
mInputOutputLanguageWeight);
|
|
|
|
if (inOutLanguageWeight != null) {
|
|
|
|
inOutLanguageWeight[0] = mInputOutputLanguageWeight[0];
|
|
|
|
}
|
2014-03-07 10:36:19 +00:00
|
|
|
final int count = mOutputSuggestionCount[0];
|
2012-08-21 07:34:55 +00:00
|
|
|
final ArrayList<SuggestedWordInfo> suggestions = CollectionUtils.newArrayList();
|
2010-12-06 12:28:24 +00:00
|
|
|
for (int j = 0; j < count; ++j) {
|
|
|
|
final int start = j * MAX_WORD_LENGTH;
|
2010-08-20 05:35:02 +00:00
|
|
|
int len = 0;
|
2012-10-29 09:06:22 +00:00
|
|
|
while (len < MAX_WORD_LENGTH && mOutputCodePoints[start + len] != 0) {
|
2010-12-06 12:28:24 +00:00
|
|
|
++len;
|
2010-08-20 05:35:02 +00:00
|
|
|
}
|
|
|
|
if (len > 0) {
|
2013-04-22 10:48:23 +00:00
|
|
|
final int flags = mOutputTypes[j] & SuggestedWordInfo.KIND_MASK_FLAGS;
|
2013-05-01 11:36:36 +00:00
|
|
|
if (blockOffensiveWords
|
2013-04-23 03:19:31 +00:00
|
|
|
&& 0 != (flags & SuggestedWordInfo.KIND_FLAG_POSSIBLY_OFFENSIVE)
|
2013-04-22 10:48:23 +00:00
|
|
|
&& 0 == (flags & SuggestedWordInfo.KIND_FLAG_EXACT_MATCH)) {
|
2013-04-23 03:19:31 +00:00
|
|
|
// If we block potentially offensive words, and if the word is possibly
|
|
|
|
// offensive, then we don't output it unless it's also an exact match.
|
2013-04-22 10:48:23 +00:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
final int kind = mOutputTypes[j] & SuggestedWordInfo.KIND_MASK_KIND;
|
|
|
|
final int score = SuggestedWordInfo.KIND_WHITELIST == kind
|
2012-08-10 04:14:45 +00:00
|
|
|
? SuggestedWordInfo.MAX_SCORE : mOutputScores[j];
|
2013-04-22 10:48:23 +00:00
|
|
|
// TODO: check that all users of the `kind' parameter are ready to accept
|
|
|
|
// flags too and pass mOutputTypes[j] instead of kind
|
2012-10-29 09:06:22 +00:00
|
|
|
suggestions.add(new SuggestedWordInfo(new String(mOutputCodePoints, start, len),
|
2013-08-20 07:11:03 +00:00
|
|
|
score, kind, this /* sourceDict */,
|
2013-09-10 08:48:00 +00:00
|
|
|
mSpaceIndices[j] /* indexOfTouchPointOfSecondWord */,
|
2013-08-20 09:00:21 +00:00
|
|
|
mOutputAutoCommitFirstWordConfidence[0]));
|
2010-08-20 05:35:02 +00:00
|
|
|
}
|
|
|
|
}
|
2012-06-12 21:57:40 +00:00
|
|
|
return suggestions;
|
2009-03-13 22:11:42 +00:00
|
|
|
}
|
|
|
|
|
2012-10-03 08:36:45 +00:00
|
|
|
public boolean isValidDictionary() {
|
2011-02-28 22:54:04 +00:00
|
|
|
return mNativeDict != 0;
|
|
|
|
}
|
|
|
|
|
2013-12-13 08:09:16 +00:00
|
|
|
public int getFormatVersion() {
|
|
|
|
return getFormatVersionNative(mNativeDict);
|
|
|
|
}
|
|
|
|
|
2009-03-13 22:11:42 +00:00
|
|
|
@Override
|
2012-10-03 06:19:43 +00:00
|
|
|
public boolean isValidWord(final String word) {
|
2013-09-18 02:18:28 +00:00
|
|
|
return getFrequency(word) != NOT_A_PROBABILITY;
|
2012-05-29 10:07:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
2012-10-03 06:19:43 +00:00
|
|
|
public int getFrequency(final String word) {
|
2013-09-18 02:18:28 +00:00
|
|
|
if (word == null) return NOT_A_PROBABILITY;
|
2012-10-03 06:19:43 +00:00
|
|
|
int[] codePoints = StringUtils.toCodePointArray(word);
|
2013-03-18 04:08:31 +00:00
|
|
|
return getProbabilityNative(mNativeDict, codePoints);
|
2009-03-13 22:11:42 +00:00
|
|
|
}
|
2009-06-04 19:20:45 +00:00
|
|
|
|
2012-04-27 06:50:21 +00:00
|
|
|
// TODO: Add a batch process version (isValidBigramMultiple?) to avoid excessive numbers of jni
|
|
|
|
// calls when checking for changes in an entire dictionary.
|
2013-07-04 12:17:49 +00:00
|
|
|
public boolean isValidBigram(final String word0, final String word1) {
|
2013-09-18 02:18:28 +00:00
|
|
|
return getBigramProbability(word0, word1) != NOT_A_PROBABILITY;
|
|
|
|
}
|
|
|
|
|
|
|
|
public int getBigramProbability(final String word0, final String word1) {
|
|
|
|
if (TextUtils.isEmpty(word0) || TextUtils.isEmpty(word1)) return NOT_A_PROBABILITY;
|
2013-07-04 12:17:49 +00:00
|
|
|
final int[] codePoints0 = StringUtils.toCodePointArray(word0);
|
2012-10-29 09:06:22 +00:00
|
|
|
final int[] codePoints1 = StringUtils.toCodePointArray(word1);
|
2013-09-18 02:18:28 +00:00
|
|
|
return getBigramProbabilityNative(mNativeDict, codePoints0, codePoints1);
|
2013-07-04 12:17:49 +00:00
|
|
|
}
|
|
|
|
|
2014-01-31 02:06:42 +00:00
|
|
|
public WordProperty getWordProperty(final String word) {
|
2013-12-13 08:09:16 +00:00
|
|
|
if (TextUtils.isEmpty(word)) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
final int[] codePoints = StringUtils.toCodePointArray(word);
|
|
|
|
final int[] outCodePoints = new int[MAX_WORD_LENGTH];
|
2014-01-31 02:06:42 +00:00
|
|
|
final boolean[] outFlags = new boolean[FORMAT_WORD_PROPERTY_OUTPUT_FLAG_COUNT];
|
2014-01-31 11:32:44 +00:00
|
|
|
final int[] outProbabilityInfo =
|
|
|
|
new int[FORMAT_WORD_PROPERTY_OUTPUT_PROBABILITY_INFO_COUNT];
|
|
|
|
final ArrayList<int[]> outBigramTargets = CollectionUtils.newArrayList();
|
|
|
|
final ArrayList<int[]> outBigramProbabilityInfo = CollectionUtils.newArrayList();
|
2013-12-13 08:09:16 +00:00
|
|
|
final ArrayList<int[]> outShortcutTargets = CollectionUtils.newArrayList();
|
|
|
|
final ArrayList<Integer> outShortcutProbabilities = CollectionUtils.newArrayList();
|
2014-01-31 11:32:44 +00:00
|
|
|
getWordPropertyNative(mNativeDict, codePoints, outCodePoints, outFlags, outProbabilityInfo,
|
|
|
|
outBigramTargets, outBigramProbabilityInfo, outShortcutTargets,
|
|
|
|
outShortcutProbabilities);
|
2014-01-31 02:06:42 +00:00
|
|
|
return new WordProperty(codePoints,
|
|
|
|
outFlags[FORMAT_WORD_PROPERTY_IS_NOT_A_WORD_INDEX],
|
|
|
|
outFlags[FORMAT_WORD_PROPERTY_IS_BLACKLISTED_INDEX],
|
|
|
|
outFlags[FORMAT_WORD_PROPERTY_HAS_BIGRAMS_INDEX],
|
2014-01-31 11:32:44 +00:00
|
|
|
outFlags[FORMAT_WORD_PROPERTY_HAS_SHORTCUTS_INDEX], outProbabilityInfo,
|
|
|
|
outBigramTargets, outBigramProbabilityInfo, outShortcutTargets,
|
|
|
|
outShortcutProbabilities);
|
2013-12-13 08:09:16 +00:00
|
|
|
}
|
|
|
|
|
2014-02-03 05:51:58 +00:00
|
|
|
public static class GetNextWordPropertyResult {
|
|
|
|
public WordProperty mWordProperty;
|
|
|
|
public int mNextToken;
|
|
|
|
|
|
|
|
public GetNextWordPropertyResult(final WordProperty wordPreperty, final int nextToken) {
|
|
|
|
mWordProperty = wordPreperty;
|
|
|
|
mNextToken = nextToken;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Method to iterate all words in the dictionary for makedict.
|
|
|
|
* If token is 0, this method newly starts iterating the dictionary.
|
|
|
|
*/
|
|
|
|
public GetNextWordPropertyResult getNextWordProperty(final int token) {
|
|
|
|
final int[] codePoints = new int[MAX_WORD_LENGTH];
|
|
|
|
final int nextToken = getNextWordNative(mNativeDict, token, codePoints);
|
2014-02-04 12:42:06 +00:00
|
|
|
final String word = StringUtils.getStringFromNullTerminatedCodePointArray(codePoints);
|
2014-02-03 05:51:58 +00:00
|
|
|
return new GetNextWordPropertyResult(getWordProperty(word), nextToken);
|
|
|
|
}
|
|
|
|
|
2013-12-13 08:09:16 +00:00
|
|
|
// Add a unigram entry to binary dictionary with unigram attributes in native code.
|
|
|
|
public void addUnigramWord(final String word, final int probability,
|
|
|
|
final String shortcutTarget, final int shortcutProbability, final boolean isNotAWord,
|
|
|
|
final boolean isBlacklisted, final int timestamp) {
|
2013-07-04 12:17:49 +00:00
|
|
|
if (TextUtils.isEmpty(word)) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
final int[] codePoints = StringUtils.toCodePointArray(word);
|
2013-12-13 08:09:16 +00:00
|
|
|
final int[] shortcutTargetCodePoints = (shortcutTarget != null) ?
|
|
|
|
StringUtils.toCodePointArray(shortcutTarget) : null;
|
|
|
|
addUnigramWordNative(mNativeDict, codePoints, probability, shortcutTargetCodePoints,
|
|
|
|
shortcutProbability, isNotAWord, isBlacklisted, timestamp);
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = true;
|
2013-07-04 12:17:49 +00:00
|
|
|
}
|
|
|
|
|
2013-12-13 08:09:16 +00:00
|
|
|
// Add a bigram entry to binary dictionary with timestamp in native code.
|
|
|
|
public void addBigramWords(final String word0, final String word1, final int probability,
|
|
|
|
final int timestamp) {
|
2013-07-04 12:17:49 +00:00
|
|
|
if (TextUtils.isEmpty(word0) || TextUtils.isEmpty(word1)) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
final int[] codePoints0 = StringUtils.toCodePointArray(word0);
|
|
|
|
final int[] codePoints1 = StringUtils.toCodePointArray(word1);
|
2013-12-13 08:09:16 +00:00
|
|
|
addBigramWordsNative(mNativeDict, codePoints0, codePoints1, probability, timestamp);
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = true;
|
2013-07-04 12:17:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Remove a bigram entry form binary dictionary in native code.
|
|
|
|
public void removeBigramWords(final String word0, final String word1) {
|
|
|
|
if (TextUtils.isEmpty(word0) || TextUtils.isEmpty(word1)) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
final int[] codePoints0 = StringUtils.toCodePointArray(word0);
|
|
|
|
final int[] codePoints1 = StringUtils.toCodePointArray(word1);
|
|
|
|
removeBigramWordsNative(mNativeDict, codePoints0, codePoints1);
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = true;
|
2012-04-27 06:50:21 +00:00
|
|
|
}
|
|
|
|
|
2013-12-13 08:09:16 +00:00
|
|
|
public void addMultipleDictionaryEntries(final LanguageModelParam[] languageModelParams) {
|
|
|
|
if (!isValidDictionary()) return;
|
|
|
|
int processedParamCount = 0;
|
|
|
|
while (processedParamCount < languageModelParams.length) {
|
|
|
|
if (needsToRunGC(true /* mindsBlockByGC */)) {
|
|
|
|
flushWithGC();
|
|
|
|
}
|
|
|
|
processedParamCount = addMultipleDictionaryEntriesNative(mNativeDict,
|
|
|
|
languageModelParams, processedParamCount);
|
2014-04-03 09:08:23 +00:00
|
|
|
mHasUpdated = true;
|
2013-12-13 08:09:16 +00:00
|
|
|
if (processedParamCount <= 0) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-25 09:32:40 +00:00
|
|
|
private void reopen() {
|
|
|
|
close();
|
2013-09-24 13:57:15 +00:00
|
|
|
final File dictFile = new File(mDictFilePath);
|
2013-12-13 08:09:16 +00:00
|
|
|
// WARNING: Because we pass 0 as the offset and file.length() as the length, this can
|
|
|
|
// only be called for actual files. Right now it's only called by the flush() family of
|
|
|
|
// functions, which require an updatable dictionary, so it's okay. But beware.
|
|
|
|
loadDictionary(dictFile.getAbsolutePath(), 0 /* startOffset */,
|
2014-02-21 06:09:37 +00:00
|
|
|
dictFile.length(), mIsUpdatable);
|
2013-09-17 09:07:16 +00:00
|
|
|
}
|
|
|
|
|
2013-09-25 09:32:40 +00:00
|
|
|
public void flush() {
|
|
|
|
if (!isValidDictionary()) return;
|
2014-04-03 09:08:23 +00:00
|
|
|
if (mHasUpdated) {
|
|
|
|
flushNative(mNativeDict, mDictFilePath);
|
|
|
|
reopen();
|
|
|
|
}
|
2013-09-25 09:32:40 +00:00
|
|
|
}
|
|
|
|
|
2013-09-17 09:07:16 +00:00
|
|
|
public void flushWithGC() {
|
|
|
|
if (!isValidDictionary()) return;
|
|
|
|
flushWithGCNative(mNativeDict, mDictFilePath);
|
2013-09-25 09:32:40 +00:00
|
|
|
reopen();
|
2013-09-17 09:07:16 +00:00
|
|
|
}
|
|
|
|
|
2013-09-30 04:57:54 +00:00
|
|
|
/**
|
|
|
|
* Checks whether GC is needed to run or not.
|
|
|
|
* @param mindsBlockByGC Whether to mind operations blocked by GC. We don't need to care about
|
|
|
|
* the blocking in some situations such as in idle time or just before closing.
|
|
|
|
* @return whether GC is needed to run or not.
|
|
|
|
*/
|
|
|
|
public boolean needsToRunGC(final boolean mindsBlockByGC) {
|
2013-09-17 09:07:16 +00:00
|
|
|
if (!isValidDictionary()) return false;
|
2013-09-30 04:57:54 +00:00
|
|
|
return needsToRunGCNative(mNativeDict, mindsBlockByGC);
|
2013-09-17 09:07:16 +00:00
|
|
|
}
|
|
|
|
|
2014-03-27 08:46:35 +00:00
|
|
|
public boolean migrateTo(final int newFormatVersion) {
|
|
|
|
if (!isValidDictionary()) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
final String tmpDictFilePath = mDictFilePath + DICT_FILE_NAME_SUFFIX_FOR_MIGRATION;
|
|
|
|
// TODO: Implement migrateNative(tmpDictFilePath, newFormatVersion).
|
|
|
|
close();
|
|
|
|
final File dictFile = new File(mDictFilePath);
|
|
|
|
final File tmpDictFile = new File(tmpDictFilePath);
|
|
|
|
FileUtils.deleteRecursively(dictFile);
|
|
|
|
if (!BinaryDictionaryUtils.renameDict(tmpDictFile, dictFile)) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
loadDictionary(dictFile.getAbsolutePath(), 0 /* startOffset */,
|
|
|
|
dictFile.length(), mIsUpdatable);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2013-09-18 02:18:28 +00:00
|
|
|
@UsedForTesting
|
|
|
|
public int calculateProbability(final int unigramProbability, final int bigramProbability) {
|
|
|
|
if (!isValidDictionary()) return NOT_A_PROBABILITY;
|
|
|
|
return calculateProbabilityNative(mNativeDict, unigramProbability, bigramProbability);
|
|
|
|
}
|
|
|
|
|
2013-09-27 14:12:12 +00:00
|
|
|
@UsedForTesting
|
2014-01-15 22:55:57 +00:00
|
|
|
public String getPropertyForTest(final String query) {
|
2013-09-27 14:12:12 +00:00
|
|
|
if (!isValidDictionary()) return "";
|
|
|
|
return getPropertyNative(mNativeDict, query);
|
|
|
|
}
|
|
|
|
|
2013-08-20 09:00:21 +00:00
|
|
|
@Override
|
|
|
|
public boolean shouldAutoCommit(final SuggestedWordInfo candidate) {
|
2013-10-01 08:30:40 +00:00
|
|
|
return candidate.mAutoCommitFirstWordConfidence > CONFIDENCE_TO_AUTO_COMMIT;
|
2013-08-20 09:00:21 +00:00
|
|
|
}
|
|
|
|
|
2009-10-12 20:48:35 +00:00
|
|
|
@Override
|
2012-08-20 05:37:16 +00:00
|
|
|
public void close() {
|
|
|
|
synchronized (mDicTraverseSessions) {
|
|
|
|
final int sessionsSize = mDicTraverseSessions.size();
|
|
|
|
for (int index = 0; index < sessionsSize; ++index) {
|
|
|
|
final DicTraverseSession traverseSession = mDicTraverseSessions.valueAt(index);
|
|
|
|
if (traverseSession != null) {
|
|
|
|
traverseSession.close();
|
|
|
|
}
|
2012-08-14 13:49:59 +00:00
|
|
|
}
|
2013-09-25 09:32:40 +00:00
|
|
|
mDicTraverseSessions.clear();
|
2012-08-14 13:49:59 +00:00
|
|
|
}
|
2013-09-25 09:32:40 +00:00
|
|
|
closeInternalLocked();
|
2011-01-17 06:13:45 +00:00
|
|
|
}
|
|
|
|
|
2013-09-25 09:32:40 +00:00
|
|
|
private synchronized void closeInternalLocked() {
|
2009-03-13 22:11:42 +00:00
|
|
|
if (mNativeDict != 0) {
|
|
|
|
closeNative(mNativeDict);
|
|
|
|
mNativeDict = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-25 09:32:40 +00:00
|
|
|
// TODO: Manage BinaryDictionary instances without using WeakReference or something.
|
2009-03-13 22:11:42 +00:00
|
|
|
@Override
|
|
|
|
protected void finalize() throws Throwable {
|
2011-01-17 06:13:45 +00:00
|
|
|
try {
|
2013-09-25 09:32:40 +00:00
|
|
|
closeInternalLocked();
|
2011-01-17 06:13:45 +00:00
|
|
|
} finally {
|
|
|
|
super.finalize();
|
|
|
|
}
|
2009-03-13 22:11:42 +00:00
|
|
|
}
|
|
|
|
}
|