luckyimnotanigger/HeliBoard
1
0
Fork 0
mirror of https://github.com/Helium314/HeliBoard.git synced 2025-04-25 17:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make history learn next word after auto-capitalized words, improves #135

Browse source
This commit is contained in:
Helium314 2023-09-11 11:39:02 +02:00
parent 8840411039
commit 12cbd43bda
2 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
app/src/main/java/org/dslul/openboard/inputmethod/latin/DictionaryFacilitatorImpl.java
View file

@ -79,6 +79,11 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
private volatile CountDownLatch mLatchForWaitingLoadingMainDictionaries = new CountDownLatch(0);
// To synchronize assigning mDictionaryGroup to ensure closing dictionaries.
private final Object mLock = new Object();
// library does not deal well with ngram history for auto-capitalized words, so we adjust the ngram
// context to store next word suggestions for such cases
private boolean mTryChangingWords = false;
private String mChangeFrom = "";
private String mChangeTo = "";
public static final Map<String, Class<? extends ExpandableBinaryDictionary>>
DICT_TYPE_TO_CLASS = new HashMap<>();
@ -731,6 +736,8 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
if (maxFreq == 0 && blockPotentiallyOffensive) {
return;
}
if (mTryChangingWords)
mTryChangingWords = ngramContext.changeWordIfAfterBeginningOfSentence(mChangeFrom, mChangeTo);
final String secondWord;
if (wasAutoCapitalized) {
// used word with lower-case first letter instead of all lower-case, as auto-capitalize
@ -747,6 +754,9 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
// If however the word is not in the dictionary, or exists as a de-capitalized word
// only, then we consider that was a lower-case word that had been auto-capitalized.
secondWord = decapitalizedWord;
mTryChangingWords = true;
mChangeFrom = word;
mChangeTo = secondWord;
}
} else {
// HACK: We'd like to avoid adding the capitalized form of common words to the User

13
app/src/main/java/org/dslul/openboard/inputmethod/latin/NgramContext.java
View file

@ -119,6 +119,19 @@ public class NgramContext {
mMaxPrevWordCount = maxPrevWordCount;
}
public boolean changeWordIfAfterBeginningOfSentence(final String from, final String to) {
boolean beginning = false;
for (int i = mPrevWordsCount - 1; i >= 0; i--) {
WordInfo info = mPrevWordsInfo[i];
if (beginning && TextUtils.equals(info.mWord, from)) {
mPrevWordsInfo[i] = new WordInfo(to);
return true;
}
beginning = info.mIsBeginningOfSentence;
}
return false;
}
/**
* Create next prevWordsInfo using current prevWordsInfo.
*/