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 | |
| 100 | private void setLocale(Locale locale) { |
Marco Nelissen | 5673544 | 2011-11-09 16:07:33 -0800 | [diff] [blame] | 101 | closeSession(); |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 102 | final TextServicesManager textServicesManager = (TextServicesManager) |
| 103 | mTextView.getContext().getSystemService(Context.TEXT_SERVICES_MANAGER_SERVICE); |
satok | 9b3855b | 2011-11-02 17:01:28 +0900 | [diff] [blame] | 104 | if (!textServicesManager.isSpellCheckerEnabled()) { |
| 105 | mSpellCheckerSession = null; |
| 106 | } else { |
| 107 | mSpellCheckerSession = textServicesManager.newSpellCheckerSession( |
| 108 | null /* Bundle not currently used by the textServicesManager */, |
| 109 | locale, this, |
| 110 | false /* means any available languages from current spell checker */); |
| 111 | } |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 112 | mCurrentLocale = locale; |
| 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 | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 121 | mSpellParsers = new SpellParser[0]; |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 122 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 123 | // This class is the global listener for locale change: warn other locale-aware objects |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 124 | mTextView.onLocaleChanged(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 125 | } |
| 126 | |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 127 | /** |
| 128 | * @return true if a spell checker session has successfully been created. Returns false if not, |
| 129 | * for instance when spell checking has been disabled in settings. |
| 130 | */ |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 131 | private boolean isSessionActive() { |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 132 | return mSpellCheckerSession != null; |
| 133 | } |
| 134 | |
| 135 | public void closeSession() { |
| 136 | if (mSpellCheckerSession != null) { |
| 137 | mSpellCheckerSession.close(); |
| 138 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 139 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 140 | stopAllSpellParsers(); |
| 141 | } |
| 142 | |
| 143 | private void stopAllSpellParsers() { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 144 | final int length = mSpellParsers.length; |
| 145 | for (int i = 0; i < length; i++) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 146 | mSpellParsers[i].stop(); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 147 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 148 | |
| 149 | if (mSpellRunnable != null) { |
| 150 | mTextView.removeCallbacks(mSpellRunnable); |
| 151 | } |
Gilles Debunne | 186aaf9 | 2011-09-16 14:26:12 -0700 | [diff] [blame] | 152 | } |
| 153 | |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 154 | private int nextSpellCheckSpanIndex() { |
| 155 | for (int i = 0; i < mLength; i++) { |
| 156 | if (mIds[i] < 0) return i; |
| 157 | } |
| 158 | |
| 159 | if (mLength == mSpellCheckSpans.length) { |
| 160 | final int newSize = mLength * 2; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 161 | int[] newIds = new int[newSize]; |
| 162 | SpellCheckSpan[] newSpellCheckSpans = new SpellCheckSpan[newSize]; |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 163 | System.arraycopy(mIds, 0, newIds, 0, mLength); |
| 164 | System.arraycopy(mSpellCheckSpans, 0, newSpellCheckSpans, 0, mLength); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 165 | mIds = newIds; |
| 166 | mSpellCheckSpans = newSpellCheckSpans; |
| 167 | } |
| 168 | |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 169 | mSpellCheckSpans[mLength] = new SpellCheckSpan(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 170 | mLength++; |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 171 | return mLength - 1; |
| 172 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 173 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 174 | private void addSpellCheckSpan(Editable editable, int start, int end) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 175 | final int index = nextSpellCheckSpanIndex(); |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 176 | editable.setSpan(mSpellCheckSpans[index], start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 177 | mIds[index] = mSpanSequenceCounter++; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 178 | } |
| 179 | |
| 180 | public void removeSpellCheckSpan(SpellCheckSpan spellCheckSpan) { |
| 181 | for (int i = 0; i < mLength; i++) { |
| 182 | if (mSpellCheckSpans[i] == spellCheckSpan) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 183 | mSpellCheckSpans[i].setSpellCheckInProgress(false); |
| 184 | mIds[i] = -1; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 185 | return; |
| 186 | } |
| 187 | } |
| 188 | } |
| 189 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 190 | public void onSelectionChanged() { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 191 | spellCheck(); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 192 | } |
| 193 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 194 | public void spellCheck(int start, int end) { |
satok | 05f2470 | 2011-11-02 19:29:35 +0900 | [diff] [blame] | 195 | final Locale locale = mTextView.getTextServicesLocale(); |
Gilles Debunne | 9d8d3f1 | 2011-10-13 12:15:10 -0700 | [diff] [blame] | 196 | if (mCurrentLocale == null || (!(mCurrentLocale.equals(locale)))) { |
| 197 | setLocale(locale); |
| 198 | // Re-check the entire text |
| 199 | start = 0; |
| 200 | end = mTextView.getText().length(); |
| 201 | } |
| 202 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 203 | if (!isSessionActive()) return; |
| 204 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 205 | // Find first available SpellParser from pool |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 206 | final int length = mSpellParsers.length; |
| 207 | for (int i = 0; i < length; i++) { |
| 208 | final SpellParser spellParser = mSpellParsers[i]; |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 209 | if (!spellParser.isParsing()) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 210 | spellParser.init(start, end); |
| 211 | spellParser.parse(); |
| 212 | return; |
| 213 | } |
| 214 | } |
| 215 | |
| 216 | // No available parser found in pool, create a new one |
| 217 | SpellParser[] newSpellParsers = new SpellParser[length + 1]; |
| 218 | System.arraycopy(mSpellParsers, 0, newSpellParsers, 0, length); |
| 219 | mSpellParsers = newSpellParsers; |
| 220 | |
| 221 | SpellParser spellParser = new SpellParser(); |
| 222 | mSpellParsers[length] = spellParser; |
| 223 | spellParser.init(start, end); |
| 224 | spellParser.parse(); |
| 225 | } |
| 226 | |
| 227 | private void spellCheck() { |
Gilles Debunne | 0eea668 | 2011-08-29 13:30:31 -0700 | [diff] [blame] | 228 | if (mSpellCheckerSession == null) return; |
Gilles Debunne | 9906847 | 2011-08-29 12:05:11 -0700 | [diff] [blame] | 229 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 230 | Editable editable = (Editable) mTextView.getText(); |
| 231 | final int selectionStart = Selection.getSelectionStart(editable); |
| 232 | final int selectionEnd = Selection.getSelectionEnd(editable); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 233 | |
| 234 | TextInfo[] textInfos = new TextInfo[mLength]; |
| 235 | int textInfosCount = 0; |
| 236 | |
| 237 | for (int i = 0; i < mLength; i++) { |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 238 | final SpellCheckSpan spellCheckSpan = mSpellCheckSpans[i]; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 239 | if (spellCheckSpan.isSpellCheckInProgress()) continue; |
| 240 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 241 | final int start = editable.getSpanStart(spellCheckSpan); |
| 242 | final int end = editable.getSpanEnd(spellCheckSpan); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 243 | |
| 244 | // Do not check this word if the user is currently editing it |
Gilles Debunne | d6e3494 | 2011-08-26 10:13:15 -0700 | [diff] [blame] | 245 | if (start >= 0 && end > start && (selectionEnd < start || selectionStart > end)) { |
Gilles Debunne | 653d3a2 | 2011-12-07 10:35:59 -0800 | [diff] [blame] | 246 | final String word = (editable instanceof SpannableStringBuilder) ? |
| 247 | ((SpannableStringBuilder) editable).substring(start, end) : |
| 248 | editable.subSequence(start, end).toString(); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 249 | spellCheckSpan.setSpellCheckInProgress(true); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 250 | textInfos[textInfosCount++] = new TextInfo(word, mCookie, mIds[i]); |
| 251 | } |
| 252 | } |
| 253 | |
| 254 | if (textInfosCount > 0) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 255 | if (textInfosCount < textInfos.length) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 256 | TextInfo[] textInfosCopy = new TextInfo[textInfosCount]; |
| 257 | System.arraycopy(textInfos, 0, textInfosCopy, 0, textInfosCount); |
| 258 | textInfos = textInfosCopy; |
| 259 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 260 | |
Gilles Debunne | 0eea668 | 2011-08-29 13:30:31 -0700 | [diff] [blame] | 261 | mSpellCheckerSession.getSuggestions(textInfos, SuggestionSpan.SUGGESTIONS_MAX_SIZE, |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 262 | false /* TODO Set sequentialWords to true for initial spell check */); |
| 263 | } |
| 264 | } |
| 265 | |
| 266 | @Override |
satok | 0dc1f64 | 2011-11-18 11:27:10 +0900 | [diff] [blame] | 267 | public void onGetSuggestionsForSentence(SuggestionsInfo[] results) { |
| 268 | // TODO: Handle the position and length for each suggestion |
| 269 | onGetSuggestions(results); |
| 270 | } |
| 271 | |
| 272 | @Override |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 273 | public void onGetSuggestions(SuggestionsInfo[] results) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 274 | Editable editable = (Editable) mTextView.getText(); |
| 275 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 276 | for (int i = 0; i < results.length; i++) { |
| 277 | SuggestionsInfo suggestionsInfo = results[i]; |
| 278 | if (suggestionsInfo.getCookie() != mCookie) continue; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 279 | final int sequenceNumber = suggestionsInfo.getSequence(); |
Gilles Debunne | b062e81 | 2011-09-27 14:58:37 -0700 | [diff] [blame] | 280 | |
| 281 | for (int j = 0; j < mLength; j++) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 282 | if (sequenceNumber == mIds[j]) { |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 283 | final int attributes = suggestionsInfo.getSuggestionsAttributes(); |
| 284 | boolean isInDictionary = |
| 285 | ((attributes & SuggestionsInfo.RESULT_ATTR_IN_THE_DICTIONARY) > 0); |
| 286 | boolean looksLikeTypo = |
| 287 | ((attributes & SuggestionsInfo.RESULT_ATTR_LOOKS_LIKE_TYPO) > 0); |
| 288 | |
Gilles Debunne | e1fc4f6 | 2011-10-03 17:01:19 -0700 | [diff] [blame] | 289 | SpellCheckSpan spellCheckSpan = mSpellCheckSpans[j]; |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 290 | |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 291 | if (!isInDictionary && looksLikeTypo) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 292 | createMisspelledSuggestionSpan(editable, suggestionsInfo, spellCheckSpan); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 293 | } |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 294 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 295 | editable.removeSpan(spellCheckSpan); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 296 | break; |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 297 | } |
| 298 | } |
| 299 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 300 | |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 301 | scheduleNewSpellCheck(); |
| 302 | } |
| 303 | |
| 304 | private void scheduleNewSpellCheck() { |
| 305 | if (mSpellRunnable == null) { |
| 306 | mSpellRunnable = new Runnable() { |
| 307 | @Override |
| 308 | public void run() { |
| 309 | final int length = mSpellParsers.length; |
| 310 | for (int i = 0; i < length; i++) { |
| 311 | final SpellParser spellParser = mSpellParsers[i]; |
| 312 | if (!spellParser.isFinished()) { |
| 313 | spellParser.parse(); |
| 314 | break; // run one spell parser at a time to bound running time |
| 315 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 316 | } |
| 317 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 318 | }; |
| 319 | } else { |
| 320 | mTextView.removeCallbacks(mSpellRunnable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 321 | } |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 322 | |
| 323 | mTextView.postDelayed(mSpellRunnable, SPELL_PAUSE_DURATION); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 324 | } |
| 325 | |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 326 | private void createMisspelledSuggestionSpan(Editable editable, SuggestionsInfo suggestionsInfo, |
| 327 | SpellCheckSpan spellCheckSpan) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 328 | final int start = editable.getSpanStart(spellCheckSpan); |
| 329 | final int end = editable.getSpanEnd(spellCheckSpan); |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 330 | if (start < 0 || end <= start) return; // span was removed in the meantime |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 331 | |
| 332 | // Other suggestion spans may exist on that region, with identical suggestions, filter |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 333 | // them out to avoid duplicates. |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 334 | SuggestionSpan[] suggestionSpans = editable.getSpans(start, end, SuggestionSpan.class); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 335 | final int length = suggestionSpans.length; |
| 336 | for (int i = 0; i < length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 337 | final int spanStart = editable.getSpanStart(suggestionSpans[i]); |
| 338 | final int spanEnd = editable.getSpanEnd(suggestionSpans[i]); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 339 | if (spanStart != start || spanEnd != end) { |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 340 | // 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] | 341 | suggestionSpans[i] = null; |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 342 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 343 | } |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 344 | |
| 345 | final int suggestionsCount = suggestionsInfo.getSuggestionsCount(); |
| 346 | String[] suggestions; |
| 347 | if (suggestionsCount <= 0) { |
| 348 | // A negative suggestion count is possible |
| 349 | suggestions = ArrayUtils.emptyArray(String.class); |
| 350 | } else { |
| 351 | int numberOfSuggestions = 0; |
| 352 | suggestions = new String[suggestionsCount]; |
| 353 | |
| 354 | for (int i = 0; i < suggestionsCount; i++) { |
| 355 | final String spellSuggestion = suggestionsInfo.getSuggestionAt(i); |
| 356 | if (spellSuggestion == null) break; |
| 357 | boolean suggestionFound = false; |
| 358 | |
| 359 | for (int j = 0; j < length && !suggestionFound; j++) { |
| 360 | if (suggestionSpans[j] == null) break; |
| 361 | |
| 362 | String[] suggests = suggestionSpans[j].getSuggestions(); |
| 363 | for (int k = 0; k < suggests.length; k++) { |
| 364 | if (spellSuggestion.equals(suggests[k])) { |
| 365 | // The suggestion is already provided by an other SuggestionSpan |
| 366 | suggestionFound = true; |
| 367 | break; |
| 368 | } |
| 369 | } |
| 370 | } |
| 371 | |
| 372 | if (!suggestionFound) { |
| 373 | suggestions[numberOfSuggestions++] = spellSuggestion; |
| 374 | } |
| 375 | } |
| 376 | |
| 377 | if (numberOfSuggestions != suggestionsCount) { |
| 378 | String[] newSuggestions = new String[numberOfSuggestions]; |
| 379 | System.arraycopy(suggestions, 0, newSuggestions, 0, numberOfSuggestions); |
| 380 | suggestions = newSuggestions; |
| 381 | } |
| 382 | } |
| 383 | |
| 384 | SuggestionSpan suggestionSpan = new SuggestionSpan(mTextView.getContext(), suggestions, |
| 385 | SuggestionSpan.FLAG_EASY_CORRECT | SuggestionSpan.FLAG_MISSPELLED); |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 386 | editable.setSpan(suggestionSpan, start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
Gilles Debunne | 176cd0d | 2011-09-29 16:37:27 -0700 | [diff] [blame] | 387 | |
Gilles Debunne | 8615ac9 | 2011-11-29 15:25:03 -0800 | [diff] [blame] | 388 | mTextView.invalidateRegion(start, end); |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 389 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 390 | |
| 391 | private class SpellParser { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 392 | private Object mRange = new Object(); |
| 393 | |
| 394 | public void init(int start, int end) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 395 | setRangeSpan((Editable) mTextView.getText(), start, end); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 396 | } |
| 397 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 398 | public void stop() { |
| 399 | removeRangeSpan((Editable) mTextView.getText()); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 400 | } |
| 401 | |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 402 | public boolean isParsing() { |
| 403 | return ((Editable) mTextView.getText()).getSpanStart(mRange) >= 0; |
| 404 | } |
| 405 | |
| 406 | private void setRangeSpan(Editable editable, int start, int end) { |
| 407 | editable.setSpan(mRange, start, end, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE); |
| 408 | } |
| 409 | |
| 410 | private void removeRangeSpan(Editable editable) { |
| 411 | editable.removeSpan(mRange); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 412 | } |
| 413 | |
| 414 | public void parse() { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 415 | Editable editable = (Editable) mTextView.getText(); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 416 | // Iterate over the newly added text and schedule new SpellCheckSpans |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 417 | final int start = editable.getSpanStart(mRange); |
| 418 | final int end = editable.getSpanEnd(mRange); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 419 | |
| 420 | int wordIteratorWindowEnd = Math.min(end, start + WORD_ITERATOR_INTERVAL); |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 421 | mWordIterator.setCharSequence(editable, start, wordIteratorWindowEnd); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 422 | |
| 423 | // Move back to the beginning of the current word, if any |
| 424 | int wordStart = mWordIterator.preceding(start); |
| 425 | int wordEnd; |
| 426 | if (wordStart == BreakIterator.DONE) { |
| 427 | wordEnd = mWordIterator.following(start); |
| 428 | if (wordEnd != BreakIterator.DONE) { |
| 429 | wordStart = mWordIterator.getBeginning(wordEnd); |
| 430 | } |
| 431 | } else { |
| 432 | wordEnd = mWordIterator.getEnd(wordStart); |
| 433 | } |
| 434 | if (wordEnd == BreakIterator.DONE) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 435 | removeRangeSpan(editable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 436 | return; |
| 437 | } |
| 438 | |
| 439 | // We need to expand by one character because we want to include the spans that |
| 440 | // end/start at position start/end respectively. |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 441 | SpellCheckSpan[] spellCheckSpans = editable.getSpans(start - 1, end + 1, |
| 442 | SpellCheckSpan.class); |
| 443 | SuggestionSpan[] suggestionSpans = editable.getSpans(start - 1, end + 1, |
| 444 | SuggestionSpan.class); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 445 | |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 446 | int wordCount = 0; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 447 | boolean scheduleOtherSpellCheck = false; |
| 448 | |
| 449 | while (wordStart <= end) { |
| 450 | if (wordEnd >= start && wordEnd > wordStart) { |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 451 | if (wordCount >= MAX_NUMBER_OF_WORDS) { |
| 452 | scheduleOtherSpellCheck = true; |
| 453 | break; |
| 454 | } |
| 455 | |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 456 | // A new word has been created across the interval boundaries with this edit. |
| 457 | // Previous spans (ended on start / started on end) removed, not valid anymore |
| 458 | if (wordStart < start && wordEnd > start) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 459 | removeSpansAt(editable, start, spellCheckSpans); |
| 460 | removeSpansAt(editable, start, suggestionSpans); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 461 | } |
| 462 | |
| 463 | if (wordStart < end && wordEnd > end) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 464 | removeSpansAt(editable, end, spellCheckSpans); |
| 465 | removeSpansAt(editable, end, suggestionSpans); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 466 | } |
| 467 | |
| 468 | // Do not create new boundary spans if they already exist |
| 469 | boolean createSpellCheckSpan = true; |
| 470 | if (wordEnd == start) { |
| 471 | for (int i = 0; i < spellCheckSpans.length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 472 | final int spanEnd = editable.getSpanEnd(spellCheckSpans[i]); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 473 | if (spanEnd == start) { |
| 474 | createSpellCheckSpan = false; |
| 475 | break; |
| 476 | } |
| 477 | } |
| 478 | } |
| 479 | |
| 480 | if (wordStart == end) { |
| 481 | for (int i = 0; i < spellCheckSpans.length; i++) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 482 | final int spanStart = editable.getSpanStart(spellCheckSpans[i]); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 483 | if (spanStart == end) { |
| 484 | createSpellCheckSpan = false; |
| 485 | break; |
| 486 | } |
| 487 | } |
| 488 | } |
| 489 | |
| 490 | if (createSpellCheckSpan) { |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 491 | addSpellCheckSpan(editable, wordStart, wordEnd); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 492 | } |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 493 | wordCount++; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 494 | } |
| 495 | |
| 496 | // iterate word by word |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 497 | int originalWordEnd = wordEnd; |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 498 | wordEnd = mWordIterator.following(wordEnd); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 499 | if ((wordIteratorWindowEnd < end) && |
| 500 | (wordEnd == BreakIterator.DONE || wordEnd >= wordIteratorWindowEnd)) { |
| 501 | wordIteratorWindowEnd = Math.min(end, originalWordEnd + WORD_ITERATOR_INTERVAL); |
Gilles Debunne | be5f49f | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 502 | mWordIterator.setCharSequence(editable, originalWordEnd, wordIteratorWindowEnd); |
| 503 | wordEnd = mWordIterator.following(originalWordEnd); |
Gilles Debunne | 35199f5 | 2011-10-25 15:05:16 -0700 | [diff] [blame] | 504 | } |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 505 | if (wordEnd == BreakIterator.DONE) break; |
| 506 | wordStart = mWordIterator.getBeginning(wordEnd); |
| 507 | if (wordStart == BreakIterator.DONE) { |
| 508 | break; |
| 509 | } |
| 510 | } |
| 511 | |
| 512 | if (scheduleOtherSpellCheck) { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 513 | // Update range span: start new spell check from last wordStart |
| 514 | setRangeSpan(editable, wordStart, end); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 515 | } else { |
Gilles Debunne | e9b8280 | 2011-10-27 14:38:27 -0700 | [diff] [blame] | 516 | removeRangeSpan(editable); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 517 | } |
| 518 | |
| 519 | spellCheck(); |
| 520 | } |
| 521 | |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 522 | private <T> void removeSpansAt(Editable editable, int offset, T[] spans) { |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 523 | final int length = spans.length; |
| 524 | for (int i = 0; i < length; i++) { |
| 525 | final T span = spans[i]; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 526 | final int start = editable.getSpanStart(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 527 | if (start > offset) continue; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 528 | final int end = editable.getSpanEnd(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 529 | if (end < offset) continue; |
Gilles Debunne | f656030 | 2011-10-10 15:03:55 -0700 | [diff] [blame] | 530 | editable.removeSpan(span); |
Gilles Debunne | 287d6c6 | 2011-10-05 18:22:11 -0700 | [diff] [blame] | 531 | } |
| 532 | } |
| 533 | } |
Gilles Debunne | 6435a56 | 2011-08-04 21:22:30 -0700 | [diff] [blame] | 534 | } |