Merge "[TestPrep17] Allow fake dictionary for testing" (a04fbce4) · Commits · e / os / android_packages_inputmethods_LatinIME

java/src/com/android/inputmethod/research/MainLogBuffer.java

+13 −0

Original line number	Diff line number	Diff line
		@@ -18,6 +18,7 @@ package com.android.inputmethod.research;

		import android.util.Log;

		import com.android.inputmethod.annotations.UsedForTesting;
		import com.android.inputmethod.latin.Dictionary;
		import com.android.inputmethod.latin.Suggest;
		import com.android.inputmethod.latin.define.ProductionFlag;
		@@ -64,7 +65,11 @@ public abstract class MainLogBuffer extends FixedLogBuffer {
		// The size of the n-grams logged. E.g. N_GRAM_SIZE = 2 means to sample bigrams.
		public static final int N_GRAM_SIZE = 2;

		// TODO: Remove dependence on Suggest, and pass in Dictionary as a parameter to an appropriate
		// method.
		private Suggest mSuggest;
		@UsedForTesting
		private Dictionary mDictionaryForTesting;
		private boolean mIsStopping = false;

		/* package for test */ int mNumWordsBetweenNGrams;
		@@ -83,7 +88,15 @@ public abstract class MainLogBuffer extends FixedLogBuffer {
		mSuggest = suggest;
		}

		@UsedForTesting
		/* package for test */ void setDictionaryForTesting(final Dictionary dictionary) {
		mDictionaryForTesting = dictionary;
		}

		private Dictionary getDictionary() {
		if (mDictionaryForTesting != null) {
		return mDictionaryForTesting;
		}
		if (mSuggest == null \|\| !mSuggest.hasMainDictionary()) return null;
		return mSuggest.getMainDictionary();
		}