Gilles Debunne | 0eea668 | 2011-08-29 13:30:31 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 The Android Open Source Project |
| 3 | * |
| 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 | */ |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 16 | |
| 17 | package android.widget; |
| 18 | |
| 19 | import android.content.Context; |
| 20 | import android.text.Editable; |
| 21 | import android.text.Selection; |
Gilles Debunne | 653d3a2 | 2011-12-07 10:35:59 -0800 | [diff] [blame] | 22 | import android.text.SpannableStringBuilder; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 23 | import android.text.Spanned; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 24 | import android.text.method.WordIterator; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 25 | import android.text.style.SpellCheckSpan; |
| 26 | import android.text.style.SuggestionSpan; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 27 | import android.view.textservice.SpellCheckerSession; |
| 28 | import android.view.textservice.SpellCheckerSession.SpellCheckerSessionListener; |
| 29 | import android.view.textservice.SuggestionsInfo; |
| 30 | import android.view.textservice.TextInfo; |
| 31 | import android.view.textservice.TextServicesManager; |
| 32 | |
| 33 | import com.android.internal.util.ArrayUtils; |
| 34 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 35 | import java.text.BreakIterator; |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 36 | import java.util.Locale; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 37 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 38 | |
| 39 | /** |
| 40 | * Helper class for TextView. Bridge between the TextView and the Dictionnary service. |
| 41 | * |
| 42 | * @hide |
| 43 | */ |
| 44 | public class SpellChecker implements SpellCheckerSessionListener { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 45 | |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 46 | // No more than this number of words will be parsed on each iteration to ensure a minimum |
| 47 | // lock of the UI thread |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 48 | public static final int MAX_NUMBER_OF_WORDS = 50; |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 49 | |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 50 | // Rough estimate, such that the word iterator interval usually does not need to be shifted |
| 51 | public static final int AVERAGE_WORD_LENGTH = 7; |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 52 | |
| 53 | // When parsing, use a character window of that size. Will be shifted if needed |
| 54 | public static final int WORD_ITERATOR_INTERVAL = AVERAGE_WORD_LENGTH * MAX_NUMBER_OF_WORDS; |
| 55 | |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 56 | // Pause between each spell check to keep the UI smooth |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 57 | private final static int SPELL_PAUSE_DURATION = 400; // milliseconds |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 58 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 59 | private final TextView mTextView; |
| 60 | |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 61 | SpellCheckerSession mSpellCheckerSession; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 62 | final int mCookie; |
| 63 | |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 64 | // Paired arrays for the (id, spellCheckSpan) pair. A negative id means the associated |
| 65 | // SpellCheckSpan has been recycled and can be-reused. |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 66 | // Contains null SpellCheckSpans after index mLength. |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 67 | private int[] mIds; |
| 68 | private SpellCheckSpan[] mSpellCheckSpans; |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 69 | // The mLength first elements of the above arrays have been initialized |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 70 | private int mLength; |
| 71 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 72 | // Parsers on chunck of text, cutting text into words that will be checked |
| 73 | private SpellParser[] mSpellParsers = new SpellParser[0]; |
| 74 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 75 | private int mSpanSequenceCounter = 0; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 76 | |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 77 | private Locale mCurrentLocale; |
| 78 | |
| 79 | // Shared by all SpellParsers. Cannot be shared with TextView since it may be used |
| 80 | // concurrently due to the asynchronous nature of onGetSuggestions. |
| 81 | private WordIterator mWordIterator; |
| 82 | |
Gilles Debunne | a49ba2f | 2011-12-01 17:41:15 -0800 | [diff] [blame] | 83 | private TextServicesManager mTextServicesManager; |
| 84 | |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 85 | private Runnable mSpellRunnable; |
| 86 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 87 | public SpellChecker(TextView textView) { |
| 88 | mTextView = textView; |
| 89 | |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 90 | // Arbitrary: these arrays will automatically double their sizes on demand |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 91 | final int size = ArrayUtils.idealObjectArraySize(1); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 92 | mIds = new int[size]; |
| 93 | mSpellCheckSpans = new SpellCheckSpan[size]; |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 94 | |
satok | 05f2470 | 2011-11-02 19:29:35 +0900 | [diff] [blame] | 95 | setLocale(mTextView.getTextServicesLocale()); |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 96 | |
| 97 | mCookie = hashCode(); |
| 98 | } |
| 99 | |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 100 | private void resetSession() { |
Marco Nelissen | 5673544 | 2011-11-09 16:07:33 -0800 | [diff] [blame] | 101 | closeSession(); |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 102 | |
| 103 | mTextServicesManager = (TextServicesManager) mTextView.getContext(). |
| 104 | getSystemService(Context.TEXT_SERVICES_MANAGER_SERVICE); |
| 105 | if (!mTextServicesManager.isSpellCheckerEnabled()) { |
satok | 9b3855b | 2011-11-02 17:01:28 +0900 | [diff] [blame] | 106 | mSpellCheckerSession = null; |
| 107 | } else { |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 108 | mSpellCheckerSession = mTextServicesManager.newSpellCheckerSession( |
satok | 9b3855b | 2011-11-02 17:01:28 +0900 | [diff] [blame] | 109 | null /* Bundle not currently used by the textServicesManager */, |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 110 | mCurrentLocale, this, |
satok | 9b3855b | 2011-11-02 17:01:28 +0900 | [diff] [blame] | 111 | false /* means any available languages from current spell checker */); |
| 112 | } |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 113 | |
| 114 | // Restore SpellCheckSpans in pool |
| 115 | for (int i = 0; i < mLength; i++) { |
| 116 | mSpellCheckSpans[i].setSpellCheckInProgress(false); |
| 117 | mIds[i] = -1; |
| 118 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 119 | mLength = 0; |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 120 | |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 121 | // Remove existing misspelled SuggestionSpans |
| 122 | mTextView.removeMisspelledSpans((Editable) mTextView.getText()); |
| 123 | } |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 124 | |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 125 | private void setLocale(Locale locale) { |
| 126 | mCurrentLocale = locale; |
| 127 | |
| 128 | resetSession(); |
| 129 | |
| 130 | // Change SpellParsers' wordIterator locale |
| 131 | mWordIterator = new WordIterator(locale); |
| 132 | |
| 133 | // This class is the listener for locale change: warn other locale-aware objects |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 134 | mTextView.onLocaleChanged(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 135 | } |
| 136 | |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 137 | /** |
| 138 | * @return true if a spell checker session has successfully been created. Returns false if not, |
| 139 | * for instance when spell checking has been disabled in settings. |
| 140 | */ |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 141 | private boolean isSessionActive() { |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 142 | return mSpellCheckerSession != null; |
| 143 | } |
| 144 | |
| 145 | public void closeSession() { |
| 146 | if (mSpellCheckerSession != null) { |
| 147 | mSpellCheckerSession.close(); |
| 148 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 149 | |
| 150 | final int length = mSpellParsers.length; |
| 151 | for (int i = 0; i < length; i++) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 152 | mSpellParsers[i].stop(); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 153 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 154 | |
| 155 | if (mSpellRunnable != null) { |
| 156 | mTextView.removeCallbacks(mSpellRunnable); |
| 157 | } |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 158 | } |
| 159 | |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 160 | private int nextSpellCheckSpanIndex() { |
| 161 | for (int i = 0; i < mLength; i++) { |
| 162 | if (mIds[i] < 0) return i; |
| 163 | } |
| 164 | |
| 165 | if (mLength == mSpellCheckSpans.length) { |
| 166 | final int newSize = mLength * 2; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 167 | int[] newIds = new int[newSize]; |
| 168 | SpellCheckSpan[] newSpellCheckSpans = new SpellCheckSpan[newSize]; |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 169 | System.arraycopy(mIds, 0, newIds, 0, mLength); |
| 170 | System.arraycopy(mSpellCheckSpans, 0, newSpellCheckSpans, 0, mLength); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 171 | mIds = newIds; |
| 172 | mSpellCheckSpans = newSpellCheckSpans; |
| 173 | } |
| 174 | |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 175 | mSpellCheckSpans[mLength] = new SpellCheckSpan(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 176 | mLength++; |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 177 | return mLength - 1; |
| 178 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 179 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 180 | private void addSpellCheckSpan(Editable editable, int start, int end) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 181 | final int index = nextSpellCheckSpanIndex(); |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 182 | editable.setSpan(mSpellCheckSpans[index], start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 183 | mIds[index] = mSpanSequenceCounter++; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 184 | } |
| 185 | |
| 186 | public void removeSpellCheckSpan(SpellCheckSpan spellCheckSpan) { |
| 187 | for (int i = 0; i < mLength; i++) { |
| 188 | if (mSpellCheckSpans[i] == spellCheckSpan) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 189 | mSpellCheckSpans[i].setSpellCheckInProgress(false); |
| 190 | mIds[i] = -1; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 191 | return; |
| 192 | } |
| 193 | } |
| 194 | } |
| 195 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 196 | public void onSelectionChanged() { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 197 | spellCheck(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 198 | } |
| 199 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 200 | public void spellCheck(int start, int end) { |
satok | 05f2470 | 2011-11-02 19:29:35 +0900 | [diff] [blame] | 201 | final Locale locale = mTextView.getTextServicesLocale(); |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 202 | if (mCurrentLocale == null || (!(mCurrentLocale.equals(locale)))) { |
| 203 | setLocale(locale); |
| 204 | // Re-check the entire text |
| 205 | start = 0; |
| 206 | end = mTextView.getText().length(); |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 207 | } else { |
| 208 | final boolean spellCheckerActivated = mTextServicesManager.isSpellCheckerEnabled(); |
| 209 | if (isSessionActive() != spellCheckerActivated) { |
| 210 | // Spell checker has been turned of or off since last spellCheck |
| 211 | resetSession(); |
| 212 | } |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 213 | } |
| 214 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 215 | if (!isSessionActive()) return; |
| 216 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 217 | // Find first available SpellParser from pool |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 218 | final int length = mSpellParsers.length; |
| 219 | for (int i = 0; i < length; i++) { |
| 220 | final SpellParser spellParser = mSpellParsers[i]; |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 221 | if (spellParser.isFinished()) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 222 | spellParser.init(start, end); |
| 223 | spellParser.parse(); |
| 224 | return; |
| 225 | } |
| 226 | } |
| 227 | |
| 228 | // No available parser found in pool, create a new one |
| 229 | SpellParser[] newSpellParsers = new SpellParser[length + 1]; |
| 230 | System.arraycopy(mSpellParsers, 0, newSpellParsers, 0, length); |
| 231 | mSpellParsers = newSpellParsers; |
| 232 | |
| 233 | SpellParser spellParser = new SpellParser(); |
| 234 | mSpellParsers[length] = spellParser; |
| 235 | spellParser.init(start, end); |
| 236 | spellParser.parse(); |
| 237 | } |
| 238 | |
| 239 | private void spellCheck() { |
Gilles Debunne | 0eea668 | 2011-08-29 13:30:31 -0700 | [diff] [blame] | 240 | if (mSpellCheckerSession == null) return; |
Gilles Debunne | 9906847 | 2011-08-29 12:05:11 -0700 | [diff] [blame] | 241 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 242 | Editable editable = (Editable) mTextView.getText(); |
| 243 | final int selectionStart = Selection.getSelectionStart(editable); |
| 244 | final int selectionEnd = Selection.getSelectionEnd(editable); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 245 | |
| 246 | TextInfo[] textInfos = new TextInfo[mLength]; |
| 247 | int textInfosCount = 0; |
| 248 | |
| 249 | for (int i = 0; i < mLength; i++) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 250 | final SpellCheckSpan spellCheckSpan = mSpellCheckSpans[i]; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 251 | if (spellCheckSpan.isSpellCheckInProgress()) continue; |
| 252 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 253 | final int start = editable.getSpanStart(spellCheckSpan); |
| 254 | final int end = editable.getSpanEnd(spellCheckSpan); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 255 | |
| 256 | // Do not check this word if the user is currently editing it |
Gilles Debunne | d6e3494 | 2011-08-26 10:13:15 -0700 | [diff] [blame] | 257 | if (start >= 0 && end > start && (selectionEnd < start || selectionStart > end)) { |
Gilles Debunne | 653d3a2 | 2011-12-07 10:35:59 -0800 | [diff] [blame] | 258 | final String word = (editable instanceof SpannableStringBuilder) ? |
| 259 | ((SpannableStringBuilder) editable).substring(start, end) : |
| 260 | editable.subSequence(start, end).toString(); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 261 | spellCheckSpan.setSpellCheckInProgress(true); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 262 | textInfos[textInfosCount++] = new TextInfo(word, mCookie, mIds[i]); |
| 263 | } |
| 264 | } |
| 265 | |
| 266 | if (textInfosCount > 0) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 267 | if (textInfosCount < textInfos.length) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 268 | TextInfo[] textInfosCopy = new TextInfo[textInfosCount]; |
| 269 | System.arraycopy(textInfos, 0, textInfosCopy, 0, textInfosCount); |
| 270 | textInfos = textInfosCopy; |
| 271 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 272 | |
Gilles Debunne | 0eea668 | 2011-08-29 13:30:31 -0700 | [diff] [blame] | 273 | mSpellCheckerSession.getSuggestions(textInfos, SuggestionSpan.SUGGESTIONS_MAX_SIZE, |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 274 | false /* TODO Set sequentialWords to true for initial spell check */); |
| 275 | } |
| 276 | } |
| 277 | |
| 278 | @Override |
satok | 0dc1f64 | 2011-11-18 11:27:10 +0900 | [diff] [blame] | 279 | public void onGetSuggestionsForSentence(SuggestionsInfo[] results) { |
| 280 | // TODO: Handle the position and length for each suggestion |
| 281 | onGetSuggestions(results); |
| 282 | } |
| 283 | |
| 284 | @Override |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 285 | public void onGetSuggestions(SuggestionsInfo[] results) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 286 | Editable editable = (Editable) mTextView.getText(); |
| 287 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 288 | for (int i = 0; i < results.length; i++) { |
| 289 | SuggestionsInfo suggestionsInfo = results[i]; |
| 290 | if (suggestionsInfo.getCookie() != mCookie) continue; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 291 | final int sequenceNumber = suggestionsInfo.getSequence(); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 292 | |
| 293 | for (int j = 0; j < mLength; j++) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 294 | if (sequenceNumber == mIds[j]) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 295 | final int attributes = suggestionsInfo.getSuggestionsAttributes(); |
| 296 | boolean isInDictionary = |
| 297 | ((attributes & SuggestionsInfo.RESULT_ATTR_IN_THE_DICTIONARY) > 0); |
| 298 | boolean looksLikeTypo = |
| 299 | ((attributes & SuggestionsInfo.RESULT_ATTR_LOOKS_LIKE_TYPO) > 0); |
| 300 | |
Gilles Debunne | e1fc4f6 | 2011-10-03 17:01:19 -0700 | [diff] [blame] | 301 | SpellCheckSpan spellCheckSpan = mSpellCheckSpans[j]; |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 302 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 303 | if (!isInDictionary && looksLikeTypo) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 304 | createMisspelledSuggestionSpan(editable, suggestionsInfo, spellCheckSpan); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 305 | } |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 306 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 307 | editable.removeSpan(spellCheckSpan); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 308 | break; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 309 | } |
| 310 | } |
| 311 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 312 | |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 313 | scheduleNewSpellCheck(); |
| 314 | } |
| 315 | |
| 316 | private void scheduleNewSpellCheck() { |
| 317 | if (mSpellRunnable == null) { |
| 318 | mSpellRunnable = new Runnable() { |
| 319 | @Override |
| 320 | public void run() { |
| 321 | final int length = mSpellParsers.length; |
| 322 | for (int i = 0; i < length; i++) { |
| 323 | final SpellParser spellParser = mSpellParsers[i]; |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 324 | if (!spellParser.isFinished()) { |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 325 | spellParser.parse(); |
| 326 | break; // run one spell parser at a time to bound running time |
| 327 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 328 | } |
| 329 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 330 | }; |
| 331 | } else { |
| 332 | mTextView.removeCallbacks(mSpellRunnable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 333 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 334 | |
| 335 | mTextView.postDelayed(mSpellRunnable, SPELL_PAUSE_DURATION); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 336 | } |
| 337 | |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 338 | private void createMisspelledSuggestionSpan(Editable editable, SuggestionsInfo suggestionsInfo, |
| 339 | SpellCheckSpan spellCheckSpan) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 340 | final int start = editable.getSpanStart(spellCheckSpan); |
| 341 | final int end = editable.getSpanEnd(spellCheckSpan); |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 342 | if (start < 0 || end <= start) return; // span was removed in the meantime |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 343 | |
| 344 | // Other suggestion spans may exist on that region, with identical suggestions, filter |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 345 | // them out to avoid duplicates. |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 346 | SuggestionSpan[] suggestionSpans = editable.getSpans(start, end, SuggestionSpan.class); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 347 | final int length = suggestionSpans.length; |
| 348 | for (int i = 0; i < length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 349 | final int spanStart = editable.getSpanStart(suggestionSpans[i]); |
| 350 | final int spanEnd = editable.getSpanEnd(suggestionSpans[i]); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 351 | if (spanStart != start || spanEnd != end) { |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 352 | // Nulled (to avoid new array allocation) if not on that exact same region |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 353 | suggestionSpans[i] = null; |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 354 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 355 | } |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 356 | |
| 357 | final int suggestionsCount = suggestionsInfo.getSuggestionsCount(); |
| 358 | String[] suggestions; |
| 359 | if (suggestionsCount <= 0) { |
| 360 | // A negative suggestion count is possible |
| 361 | suggestions = ArrayUtils.emptyArray(String.class); |
| 362 | } else { |
| 363 | int numberOfSuggestions = 0; |
| 364 | suggestions = new String[suggestionsCount]; |
| 365 | |
| 366 | for (int i = 0; i < suggestionsCount; i++) { |
| 367 | final String spellSuggestion = suggestionsInfo.getSuggestionAt(i); |
| 368 | if (spellSuggestion == null) break; |
| 369 | boolean suggestionFound = false; |
| 370 | |
| 371 | for (int j = 0; j < length && !suggestionFound; j++) { |
| 372 | if (suggestionSpans[j] == null) break; |
| 373 | |
| 374 | String[] suggests = suggestionSpans[j].getSuggestions(); |
| 375 | for (int k = 0; k < suggests.length; k++) { |
| 376 | if (spellSuggestion.equals(suggests[k])) { |
| 377 | // The suggestion is already provided by an other SuggestionSpan |
| 378 | suggestionFound = true; |
| 379 | break; |
| 380 | } |
| 381 | } |
| 382 | } |
| 383 | |
| 384 | if (!suggestionFound) { |
| 385 | suggestions[numberOfSuggestions++] = spellSuggestion; |
| 386 | } |
| 387 | } |
| 388 | |
| 389 | if (numberOfSuggestions != suggestionsCount) { |
| 390 | String[] newSuggestions = new String[numberOfSuggestions]; |
| 391 | System.arraycopy(suggestions, 0, newSuggestions, 0, numberOfSuggestions); |
| 392 | suggestions = newSuggestions; |
| 393 | } |
| 394 | } |
| 395 | |
| 396 | SuggestionSpan suggestionSpan = new SuggestionSpan(mTextView.getContext(), suggestions, |
| 397 | SuggestionSpan.FLAG_EASY_CORRECT | SuggestionSpan.FLAG_MISSPELLED); |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 398 | editable.setSpan(suggestionSpan, start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 399 | |
Gilles Debunne | 961ebb9 | 2011-12-12 10:16:04 -0800 | [diff] [blame] | 400 | mTextView.invalidateRegion(start, end, false /* No cursor involved */); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 401 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 402 | |
| 403 | private class SpellParser { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 404 | private Object mRange = new Object(); |
| 405 | |
| 406 | public void init(int start, int end) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 407 | setRangeSpan((Editable) mTextView.getText(), start, end); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 408 | } |
| 409 | |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 410 | public boolean isFinished() { |
| 411 | return ((Editable) mTextView.getText()).getSpanStart(mRange) < 0; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 412 | } |
| 413 | |
Gilles Debunne | 249d1e8 | 2011-12-12 20:06:29 -0800 | [diff] [blame] | 414 | public void stop() { |
| 415 | removeRangeSpan((Editable) mTextView.getText()); |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 416 | } |
| 417 | |
| 418 | private void setRangeSpan(Editable editable, int start, int end) { |
| 419 | editable.setSpan(mRange, start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
| 420 | } |
| 421 | |
| 422 | private void removeRangeSpan(Editable editable) { |
| 423 | editable.removeSpan(mRange); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 424 | } |
| 425 | |
| 426 | public void parse() { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 427 | Editable editable = (Editable) mTextView.getText(); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 428 | // Iterate over the newly added text and schedule new SpellCheckSpans |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 429 | final int start = editable.getSpanStart(mRange); |
| 430 | final int end = editable.getSpanEnd(mRange); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 431 | |
| 432 | int wordIteratorWindowEnd = Math.min(end, start + WORD_ITERATOR_INTERVAL); |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 433 | mWordIterator.setCharSequence(editable, start, wordIteratorWindowEnd); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 434 | |
| 435 | // Move back to the beginning of the current word, if any |
| 436 | int wordStart = mWordIterator.preceding(start); |
| 437 | int wordEnd; |
| 438 | if (wordStart == BreakIterator.DONE) { |
| 439 | wordEnd = mWordIterator.following(start); |
| 440 | if (wordEnd != BreakIterator.DONE) { |
| 441 | wordStart = mWordIterator.getBeginning(wordEnd); |
| 442 | } |
| 443 | } else { |
| 444 | wordEnd = mWordIterator.getEnd(wordStart); |
| 445 | } |
| 446 | if (wordEnd == BreakIterator.DONE) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 447 | removeRangeSpan(editable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 448 | return; |
| 449 | } |
| 450 | |
| 451 | // We need to expand by one character because we want to include the spans that |
| 452 | // end/start at position start/end respectively. |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 453 | SpellCheckSpan[] spellCheckSpans = editable.getSpans(start - 1, end + 1, |
| 454 | SpellCheckSpan.class); |
| 455 | SuggestionSpan[] suggestionSpans = editable.getSpans(start - 1, end + 1, |
| 456 | SuggestionSpan.class); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 457 | |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 458 | int wordCount = 0; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 459 | boolean scheduleOtherSpellCheck = false; |
| 460 | |
| 461 | while (wordStart <= end) { |
| 462 | if (wordEnd >= start && wordEnd > wordStart) { |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 463 | if (wordCount >= MAX_NUMBER_OF_WORDS) { |
| 464 | scheduleOtherSpellCheck = true; |
| 465 | break; |
| 466 | } |
| 467 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 468 | // A new word has been created across the interval boundaries with this edit. |
| 469 | // Previous spans (ended on start / started on end) removed, not valid anymore |
| 470 | if (wordStart < start && wordEnd > start) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 471 | removeSpansAt(editable, start, spellCheckSpans); |
| 472 | removeSpansAt(editable, start, suggestionSpans); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 473 | } |
| 474 | |
| 475 | if (wordStart < end && wordEnd > end) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 476 | removeSpansAt(editable, end, spellCheckSpans); |
| 477 | removeSpansAt(editable, end, suggestionSpans); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 478 | } |
| 479 | |
| 480 | // Do not create new boundary spans if they already exist |
| 481 | boolean createSpellCheckSpan = true; |
| 482 | if (wordEnd == start) { |
| 483 | for (int i = 0; i < spellCheckSpans.length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 484 | final int spanEnd = editable.getSpanEnd(spellCheckSpans[i]); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 485 | if (spanEnd == start) { |
| 486 | createSpellCheckSpan = false; |
| 487 | break; |
| 488 | } |
| 489 | } |
| 490 | } |
| 491 | |
| 492 | if (wordStart == end) { |
| 493 | for (int i = 0; i < spellCheckSpans.length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 494 | final int spanStart = editable.getSpanStart(spellCheckSpans[i]); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 495 | if (spanStart == end) { |
| 496 | createSpellCheckSpan = false; |
| 497 | break; |
| 498 | } |
| 499 | } |
| 500 | } |
| 501 | |
| 502 | if (createSpellCheckSpan) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 503 | addSpellCheckSpan(editable, wordStart, wordEnd); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 504 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 505 | wordCount++; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 506 | } |
| 507 | |
| 508 | // iterate word by word |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 509 | int originalWordEnd = wordEnd; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 510 | wordEnd = mWordIterator.following(wordEnd); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 511 | if ((wordIteratorWindowEnd < end) && |
| 512 | (wordEnd == BreakIterator.DONE || wordEnd >= wordIteratorWindowEnd)) { |
| 513 | wordIteratorWindowEnd = Math.min(end, originalWordEnd + WORD_ITERATOR_INTERVAL); |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 514 | mWordIterator.setCharSequence(editable, originalWordEnd, wordIteratorWindowEnd); |
| 515 | wordEnd = mWordIterator.following(originalWordEnd); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 516 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 517 | if (wordEnd == BreakIterator.DONE) break; |
| 518 | wordStart = mWordIterator.getBeginning(wordEnd); |
| 519 | if (wordStart == BreakIterator.DONE) { |
| 520 | break; |
| 521 | } |
| 522 | } |
| 523 | |
| 524 | if (scheduleOtherSpellCheck) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 525 | // Update range span: start new spell check from last wordStart |
| 526 | setRangeSpan(editable, wordStart, end); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 527 | } else { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 528 | removeRangeSpan(editable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 529 | } |
| 530 | |
| 531 | spellCheck(); |
| 532 | } |
| 533 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 534 | private <T> void removeSpansAt(Editable editable, int offset, T[] spans) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 535 | final int length = spans.length; |
| 536 | for (int i = 0; i < length; i++) { |
| 537 | final T span = spans[i]; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 538 | final int start = editable.getSpanStart(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 539 | if (start > offset) continue; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 540 | final int end = editable.getSpanEnd(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 541 | if (end < offset) continue; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 542 | editable.removeSpan(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 543 | } |
| 544 | } |
| 545 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 546 | } |