Merge "refactor space proximity"

2 files changed, 7 insertions, 0 deletions

diff --git a/native/jni/src/proximity_info_state.cpp b/native/jni/src/proximity_info_state.cpp
index 1e1413a5d..757487004 100644
--- a/native/jni/src/proximity_info_state.cpp
+++ b/native/jni/src/proximity_info_state.cpp
@@ -1185,6 +1185,11 @@ float ProximityInfoState::getMostProbableString(int *const codePointBuf) const {
     return sumLogProbability;
 }
 
+bool ProximityInfoState::hasSpaceProximity(const int index) const {
+    ASSERT(0 <= index && index < mSampledInputSize);
+    return mProximityInfo->hasSpaceProximity(getInputX(index), getInputY(index));
+}
+
 // Returns a probability of mapping index to keyIndex.
 float ProximityInfoState::getProbability(const int index, const int keyIndex) const {
     ASSERT(0 <= index && index < mSampledInputSize);
diff --git a/native/jni/src/proximity_info_state.h b/native/jni/src/proximity_info_state.h
index bc2cf505c..9258744f9 100644
--- a/native/jni/src/proximity_info_state.h
+++ b/native/jni/src/proximity_info_state.h
@@ -146,6 +146,8 @@ class ProximityInfoState {
         return mSampledInputYs[index];
     }
 
+    bool hasSpaceProximity(const int index) const;
+
     int getLengthCache(const int index) const {
         return mLengthCache[index];
     }