Merge "Add prev words context to the SuggestedWordInfo." (7df95c54) · Commits · e / os / android_packages_inputmethods_LatinIME

java-overridable/src/com/android/inputmethod/latin/utils/StatsUtils.java

+2 −1

Original line number	Diff line number	Diff line
		@@ -72,7 +72,8 @@ public final class StatsUtils {
		}

		public static void onAutoCorrection(final String typedWord, final String autoCorrectionWord,
		final boolean isBatchInput, final DictionaryFacilitator dictionaryType) {
		final boolean isBatchInput, final DictionaryFacilitator dictionaryFacilitator,
		final String prevWordsContext) {
		}

		public static void onWordCommitUserTyped(final String commitWord, final boolean isBatchMode) {

java/src/com/android/inputmethod/latin/BinaryDictionary.java

+3 −2

Original line number	Diff line number	Diff line
		@@ -27,7 +27,6 @@ import com.android.inputmethod.latin.common.Constants;
		import com.android.inputmethod.latin.common.FileUtils;
		import com.android.inputmethod.latin.common.InputPointers;
		import com.android.inputmethod.latin.common.StringUtils;
		import com.android.inputmethod.latin.define.DecoderSpecificConstants;
		import com.android.inputmethod.latin.makedict.DictionaryHeader;
		import com.android.inputmethod.latin.makedict.FormatSpec;
		import com.android.inputmethod.latin.makedict.FormatSpec.DictionaryOptions;
		@@ -330,7 +329,9 @@ public final class BinaryDictionary extends Dictionary {
		if (len > 0) {
		suggestions.add(new SuggestedWordInfo(
		new String(session.mOutputCodePoints, start, len),
		(int)(session.mOutputScores[j] * weightForLocale), session.mOutputTypes[j],
		"" /* prevWordsContext */,
		(int)(session.mOutputScores[j] * weightForLocale),
		session.mOutputTypes[j],
		this /* sourceDict */,
		session.mSpaceIndices[j] /* indexOfTouchPointOfSecondWord */,
		session.mOutputAutoCommitFirstWordConfidence[0]));

java/src/com/android/inputmethod/latin/PunctuationSuggestions.java

+2 −1

Original line number	Diff line number	Diff line
		@@ -114,7 +114,8 @@ public final class PunctuationSuggestions extends SuggestedWords {
		}

		private static SuggestedWordInfo newHardCodedWordInfo(final String keySpec) {
		return new SuggestedWordInfo(keySpec, SuggestedWordInfo.MAX_SCORE,
		return new SuggestedWordInfo(keySpec, "" /* prevWordsContext */,
		SuggestedWordInfo.MAX_SCORE,
		SuggestedWordInfo.KIND_HARDCODED,
		Dictionary.DICTIONARY_HARDCODED,
		SuggestedWordInfo.NOT_AN_INDEX /* indexOfTouchPointOfSecondWord */,

java/src/com/android/inputmethod/latin/Suggest.java

+4 −3

Original line number	Diff line number	Diff line
		@@ -36,7 +36,6 @@ import java.util.HashMap;
		import java.util.Locale;

		import javax.annotation.Nonnull;
		import javax.annotation.Nullable;

		/**
		* This class loads a dictionary and provides a list of suggestions for a given sequence of
		@@ -249,7 +248,8 @@ public final class Suggest {
		}

		final SuggestedWordInfo typedWordInfo = new SuggestedWordInfo(typedWordString,
		SuggestedWordInfo.MAX_SCORE, SuggestedWordInfo.KIND_TYPED,
		"" /* prevWordsContext */, SuggestedWordInfo.MAX_SCORE,
		SuggestedWordInfo.KIND_TYPED,
		null == sourceDictionaryOfRemovedWord ? Dictionary.DICTIONARY_USER_TYPED
		: sourceDictionaryOfRemovedWord,
		SuggestedWordInfo.NOT_AN_INDEX /* indexOfTouchPointOfSecondWord */,
		@@ -426,7 +426,8 @@ public final class Suggest {
		for (int i = quotesToAppend - 1; i >= 0; --i) {
		sb.appendCodePoint(Constants.CODE_SINGLE_QUOTE);
		}
		return new SuggestedWordInfo(sb.toString(), wordInfo.mScore, wordInfo.mKindAndFlags,
		return new SuggestedWordInfo(sb.toString(), wordInfo.mPrevWordsContext,
		wordInfo.mScore, wordInfo.mKindAndFlags,
		wordInfo.mSourceDict, wordInfo.mIndexOfTouchPointOfSecondWord,
		wordInfo.mAutoCommitFirstWordConfidence);
		}

java/src/com/android/inputmethod/latin/SuggestedWords.java

+6 −24

Original line number	Diff line number	Diff line
		@@ -17,11 +17,9 @@
		package com.android.inputmethod.latin;

		import android.text.TextUtils;
		import android.util.Log;
		import android.view.inputmethod.CompletionInfo;

		import com.android.inputmethod.annotations.UsedForTesting;
		import com.android.inputmethod.latin.common.Constants;
		import com.android.inputmethod.latin.common.StringUtils;
		import com.android.inputmethod.latin.define.DebugFlags;

		@@ -264,6 +262,7 @@ public class SuggestedWords {
		public static final int KIND_FLAG_APPROPRIATE_FOR_AUTO_CORRECTION = 0x10000000;

		public final String mWord;
		public final String mPrevWordsContext;
		// The completion info from the application. Null for suggestions that don't come from
		// the application (including keyboard-computed ones, so this is almost always null)
		public final CompletionInfo mApplicationSpecifiedCompletionInfo;
		@@ -284,6 +283,7 @@ public class SuggestedWords {
		/**
		* Create a new suggested word info.
		* @param word The string to suggest.
		* @param prevWordsContext previous words context.
		* @param score A measure of how likely this suggestion is.
		* @param kindAndFlags The kind of suggestion, as one of the above KIND_* constants with
		* flags.
		@@ -291,10 +291,12 @@ public class SuggestedWords {
		* @param indexOfTouchPointOfSecondWord See mIndexOfTouchPointOfSecondWord.
		* @param autoCommitFirstWordConfidence See mAutoCommitFirstWordConfidence.
		*/
		public SuggestedWordInfo(final String word, final int score, final int kindAndFlags,
		public SuggestedWordInfo(final String word, final String prevWordsContext,
		final int score, final int kindAndFlags,
		final Dictionary sourceDict, final int indexOfTouchPointOfSecondWord,
		final int autoCommitFirstWordConfidence) {
		mWord = word;
		mPrevWordsContext = prevWordsContext;
		mApplicationSpecifiedCompletionInfo = null;
		mScore = score;
		mKindAndFlags = kindAndFlags;
		@@ -311,6 +313,7 @@ public class SuggestedWords {
		*/
		public SuggestedWordInfo(final CompletionInfo applicationSpecifiedCompletion) {
		mWord = applicationSpecifiedCompletion.getText().toString();
		mPrevWordsContext = "";
		mApplicationSpecifiedCompletionInfo = applicationSpecifiedCompletion;
		mScore = SuggestedWordInfo.MAX_SCORE;
		mKindAndFlags = SuggestedWordInfo.KIND_APP_DEFINED;
		@@ -429,27 +432,6 @@ public class SuggestedWords {
		return isPrediction(mInputStyle);
		}

		// Creates a new SuggestedWordInfo from the currently suggested words that removes all but the
		// last word of all suggestions, separated by a space. This is necessary because when we commit
		// a multiple-word suggestion, the IME only retains the last word as the composing word, and
		// we should only suggest replacements for this last word.
		// TODO: make this work with languages without spaces.
		public SuggestedWords getSuggestedWordsForLastWordOfPhraseGesture() {
		final ArrayList<SuggestedWordInfo> newSuggestions = new ArrayList<>();
		for (int i = 0; i < mSuggestedWordInfoList.size(); ++i) {
		final SuggestedWordInfo info = mSuggestedWordInfoList.get(i);
		final int indexOfLastSpace = info.mWord.lastIndexOf(Constants.CODE_SPACE) + 1;
		final String lastWord = info.mWord.substring(indexOfLastSpace);
		newSuggestions.add(new SuggestedWordInfo(lastWord, info.mScore, info.mKindAndFlags,
		info.mSourceDict, SuggestedWordInfo.NOT_AN_INDEX,
		SuggestedWordInfo.NOT_A_CONFIDENCE));
		}
		return new SuggestedWords(newSuggestions, null /* rawSuggestions */,
		newSuggestions.isEmpty() ? null : newSuggestions.get(0) /* typedWordInfo */,
		mTypedWordValid, mWillAutoCorrect, mIsObsoleteSuggestions, INPUT_STYLE_TAIL_BATCH,
		NOT_A_SEQUENCE_NUMBER);
		}

		/**
		* @return the {@link SuggestedWordInfo} which corresponds to the word that is originally
		* typed by the user. Otherwise returns {@code null}. Note that gesture input is not