Add languageWeight as an argument of getSuggestions. (18f21f96) · Commits · e / os / android_packages_inputmethods_LatinIME

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

+4 −3

Original line number	Diff line number	Diff line
		@@ -231,16 +231,17 @@ public final class BinaryDictionary extends Dictionary {
		@Override
		public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final float[] inOutLanguageWeight) {
		return getSuggestionsWithSessionId(composer, prevWord, proximityInfo, blockOffensiveWords,
		additionalFeaturesOptions, 0 /* sessionId */);
		additionalFeaturesOptions, 0 /* sessionId */, inOutLanguageWeight);
		}

		@Override
		public ArrayList<SuggestedWordInfo> getSuggestionsWithSessionId(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final int sessionId) {
		final int sessionId, final float[] inOutLanguageWeight) {
		if (!isValidDictionary()) return null;

		Arrays.fill(mInputCodePoints, Constants.NOT_A_CODE);

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

+10 −4

Original line number	Diff line number	Diff line
		@@ -27,6 +27,7 @@ import java.util.ArrayList;
		*/
		public abstract class Dictionary {
		public static final int NOT_A_PROBABILITY = -1;
		public static final float NOT_A_LANGUAGE_WEIGHT = -1.0f;

		// The following types do not actually come from real dictionary instances, so we create
		// corresponding instances.
		@@ -70,22 +71,26 @@ public abstract class Dictionary {
		* @param proximityInfo the object for key proximity. May be ignored by some implementations.
		* @param blockOffensiveWords whether to block potentially offensive words
		* @param additionalFeaturesOptions options about additional features used for the suggestion.
		* @param inOutLanguageWeight the language weight used for generating suggestions.
		* inOutLanguageWeight is a float array that has only one element. This can be updated when the
		* different language weight is used.
		* @return the list of suggestions (possibly null if none)
		*/
		// TODO: pass more context than just the previous word, to enable better suggestions (n-gram
		// and more)
		abstract public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions);
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final float[] inOutLanguageWeight);

		// The default implementation of this method ignores sessionId.
		// Subclasses that want to use sessionId need to override this method.
		public ArrayList<SuggestedWordInfo> getSuggestionsWithSessionId(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final int sessionId) {
		final int sessionId, final float[] inOutLanguageWeight) {
		return getSuggestions(composer, prevWord, proximityInfo, blockOffensiveWords,
		additionalFeaturesOptions);
		additionalFeaturesOptions, inOutLanguageWeight);
		}

		/**
		@@ -159,7 +164,8 @@ public abstract class Dictionary {
		@Override
		public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final float[] inOutLanguageWeight) {
		return null;
		}

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

+6 −3

Original line number	Diff line number	Diff line
		@@ -58,18 +58,21 @@ public final class DictionaryCollection extends Dictionary {
		@Override
		public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final float[] inOutLanguageWeight) {
		final CopyOnWriteArrayList<Dictionary> dictionaries = mDictionaries;
		if (dictionaries.isEmpty()) return null;
		// To avoid creating unnecessary objects, we get the list out of the first
		// dictionary and add the rest to it if not null, hence the get(0)
		ArrayList<SuggestedWordInfo> suggestions = dictionaries.get(0).getSuggestions(composer,
		prevWord, proximityInfo, blockOffensiveWords, additionalFeaturesOptions);
		prevWord, proximityInfo, blockOffensiveWords, additionalFeaturesOptions,
		inOutLanguageWeight);
		if (null == suggestions) suggestions = CollectionUtils.newArrayList();
		final int length = dictionaries.size();
		for (int i = 1; i < length; ++ i) {
		final ArrayList<SuggestedWordInfo> sugg = dictionaries.get(i).getSuggestions(composer,
		prevWord, proximityInfo, blockOffensiveWords, additionalFeaturesOptions);
		prevWord, proximityInfo, blockOffensiveWords, additionalFeaturesOptions,
		inOutLanguageWeight);
		if (null != sugg) suggestions.addAll(sugg);
		}
		return suggestions;

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

+3 −1

Original line number	Diff line number	Diff line
		@@ -461,12 +461,14 @@ public class DictionaryFacilitatorForSuggest {
		final Map<String, Dictionary> dictMap = dictionaries.mDictMap;
		final SuggestionResults suggestionResults =
		new SuggestionResults(dictionaries.mLocale, SuggestedWords.MAX_SUGGESTIONS);
		final float[] languageWeight = new float[] { Dictionary.NOT_A_LANGUAGE_WEIGHT };
		for (final String dictType : dictTypesOrderedToGetSuggestion) {
		final Dictionary dictionary = dictMap.get(dictType);
		if (null == dictionary) continue;
		final ArrayList<SuggestedWordInfo> dictionarySuggestions =
		dictionary.getSuggestionsWithSessionId(composer, prevWord, proximityInfo,
		blockOffensiveWords, additionalFeaturesOptions, sessionId);
		blockOffensiveWords, additionalFeaturesOptions, sessionId,
		languageWeight);
		if (null == dictionarySuggestions) continue;
		suggestionResults.addAll(dictionarySuggestions);
		if (null != rawSuggestions) {

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

+5 −4

Original line number	Diff line number	Diff line
		@@ -387,7 +387,7 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
		public ArrayList<SuggestedWordInfo> getSuggestionsWithSessionId(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final int sessionId) {
		final int sessionId, final float[] inOutLanguageWeight) {
		reloadDictionaryIfRequired();
		if (processingLargeTask()) {
		return null;
		@@ -404,7 +404,7 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
		final ArrayList<SuggestedWordInfo> binarySuggestion =
		mBinaryDictionary.getSuggestionsWithSessionId(composer, prevWord,
		proximityInfo, blockOffensiveWords, additionalFeaturesOptions,
		sessionId);
		sessionId, inOutLanguageWeight);
		holder.set(binarySuggestion);
		if (mBinaryDictionary.isCorrupted()) {
		removeBinaryDictionaryLocked();
		@@ -417,9 +417,10 @@ abstract public class ExpandableBinaryDictionary extends Dictionary {
		@Override
		public ArrayList<SuggestedWordInfo> getSuggestions(final WordComposer composer,
		final String prevWord, final ProximityInfo proximityInfo,
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions) {
		final boolean blockOffensiveWords, final int[] additionalFeaturesOptions,
		final float[] inOutLanguageWeight) {
		return getSuggestionsWithSessionId(composer, prevWord, proximityInfo, blockOffensiveWords,
		additionalFeaturesOptions, 0 /* sessionId */);
		additionalFeaturesOptions, 0 /* sessionId */, inOutLanguageWeight);
		}

		@Override