| Index: third_party/prediction/suggest/core/layout/proximity_info_state_utils.cpp
|
| diff --git a/third_party/prediction/suggest/core/layout/proximity_info_state_utils.cpp b/third_party/prediction/suggest/core/layout/proximity_info_state_utils.cpp
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..d968bb6b2768daf3d724f394e6da12f1bfcca459
|
| --- /dev/null
|
| +++ b/third_party/prediction/suggest/core/layout/proximity_info_state_utils.cpp
|
| @@ -0,0 +1,1215 @@
|
| +/*
|
| + * Copyright (C) 2013 The Android Open Source Project
|
| + *
|
| + * Licensed under the Apache License, Version 2.0 (the "License");
|
| + * you may not use this file except in compliance with the License.
|
| + * You may obtain a copy of the License at
|
| + *
|
| + * http://www.apache.org/licenses/LICENSE-2.0
|
| + *
|
| + * Unless required by applicable law or agreed to in writing, software
|
| + * distributed under the License is distributed on an "AS IS" BASIS,
|
| + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| + * See the License for the specific language governing permissions and
|
| + * limitations under the License.
|
| + */
|
| +
|
| +#include "third_party/prediction/suggest/core/layout/proximity_info_state_utils.h"
|
| +
|
| +#include <algorithm>
|
| +#include <cmath>
|
| +#include <cstring> // for memset()
|
| +#include <sstream> // for debug prints
|
| +#include <unordered_map>
|
| +#include <vector>
|
| +
|
| +#include "third_party/prediction/defines.h"
|
| +#include "third_party/prediction/suggest/core/layout/geometry_utils.h"
|
| +#include "third_party/prediction/suggest/core/layout/normal_distribution_2d.h"
|
| +#include "third_party/prediction/suggest/core/layout/proximity_info.h"
|
| +#include "third_party/prediction/suggest/core/layout/proximity_info_params.h"
|
| +
|
| +namespace latinime {
|
| +
|
| +/* static */ int ProximityInfoStateUtils::trimLastTwoTouchPoints(
|
| + std::vector<int>* sampledInputXs,
|
| + std::vector<int>* sampledInputYs,
|
| + std::vector<int>* sampledInputTimes,
|
| + std::vector<int>* sampledLengthCache,
|
| + std::vector<int>* sampledInputIndice) {
|
| + const int nextStartIndex =
|
| + (*sampledInputIndice)[sampledInputIndice->size() - 2];
|
| + popInputData(sampledInputXs, sampledInputYs, sampledInputTimes,
|
| + sampledLengthCache, sampledInputIndice);
|
| + popInputData(sampledInputXs, sampledInputYs, sampledInputTimes,
|
| + sampledLengthCache, sampledInputIndice);
|
| + return nextStartIndex;
|
| +}
|
| +
|
| +/* static */ int ProximityInfoStateUtils::updateTouchPoints(
|
| + const ProximityInfo* const proximityInfo,
|
| + const int maxPointToKeyLength,
|
| + const int* const inputProximities,
|
| + const int* const inputXCoordinates,
|
| + const int* const inputYCoordinates,
|
| + const int* const times,
|
| + const int* const pointerIds,
|
| + const int inputSize,
|
| + const bool isGeometric,
|
| + const int pointerId,
|
| + const int pushTouchPointStartIndex,
|
| + std::vector<int>* sampledInputXs,
|
| + std::vector<int>* sampledInputYs,
|
| + std::vector<int>* sampledInputTimes,
|
| + std::vector<int>* sampledLengthCache,
|
| + std::vector<int>* sampledInputIndice) {
|
| + if (DEBUG_SAMPLING_POINTS) {
|
| + if (times) {
|
| + for (int i = 0; i < inputSize; ++i) {
|
| + AKLOGI("(%d) x %d, y %d, time %d", i, inputXCoordinates[i],
|
| + inputYCoordinates[i], times[i]);
|
| + }
|
| + }
|
| + }
|
| +#ifdef DO_ASSERT_TEST
|
| + if (times) {
|
| + for (int i = 0; i < inputSize; ++i) {
|
| + if (i > 0) {
|
| + if (times[i] < times[i - 1]) {
|
| + AKLOGI("Invalid time sequence. %d, %d", times[i - 1], times[i]);
|
| + ASSERT(false);
|
| + }
|
| + }
|
| + }
|
| + }
|
| +#endif
|
| + const bool proximityOnly =
|
| + !isGeometric && (inputXCoordinates[0] < 0 || inputYCoordinates[0] < 0);
|
| + int lastInputIndex = pushTouchPointStartIndex;
|
| + for (int i = lastInputIndex; i < inputSize; ++i) {
|
| + const int pid = pointerIds ? pointerIds[i] : 0;
|
| + if (pointerId == pid) {
|
| + lastInputIndex = i;
|
| + }
|
| + }
|
| + if (DEBUG_GEO_FULL) {
|
| + AKLOGI("Init ProximityInfoState: last input index = %d", lastInputIndex);
|
| + }
|
| + // Working space to save near keys distances for current, prev and prevprev
|
| + // input point.
|
| + NearKeysDistanceMap nearKeysDistances[3];
|
| + // These pointers are swapped for each inputs points.
|
| + NearKeysDistanceMap* currentNearKeysDistances = &nearKeysDistances[0];
|
| + NearKeysDistanceMap* prevNearKeysDistances = &nearKeysDistances[1];
|
| + NearKeysDistanceMap* prevPrevNearKeysDistances = &nearKeysDistances[2];
|
| + // "sumAngle" is accumulated by each angle of input points. And when
|
| + // "sumAngle" exceeds
|
| + // the threshold we save that point, reset sumAngle. This aims to keep the
|
| + // figure of
|
| + // the curve.
|
| + float sumAngle = 0.0f;
|
| +
|
| + for (int i = pushTouchPointStartIndex; i <= lastInputIndex; ++i) {
|
| + // Assuming pointerId == 0 if pointerIds is null.
|
| + const int pid = pointerIds ? pointerIds[i] : 0;
|
| + if (DEBUG_GEO_FULL) {
|
| + AKLOGI("Init ProximityInfoState: (%d)PID = %d", i, pid);
|
| + }
|
| + if (pointerId == pid) {
|
| + const int c = isGeometric ? NOT_A_COORDINATE
|
| + : getPrimaryCodePointAt(inputProximities, i);
|
| + const int x = proximityOnly ? NOT_A_COORDINATE : inputXCoordinates[i];
|
| + const int y = proximityOnly ? NOT_A_COORDINATE : inputYCoordinates[i];
|
| + const int time = times ? times[i] : -1;
|
| +
|
| + if (i > 1) {
|
| + const float prevAngle = GeometryUtils::getAngle(
|
| + inputXCoordinates[i - 2], inputYCoordinates[i - 2],
|
| + inputXCoordinates[i - 1], inputYCoordinates[i - 1]);
|
| + const float currentAngle = GeometryUtils::getAngle(
|
| + inputXCoordinates[i - 1], inputYCoordinates[i - 1], x, y);
|
| + sumAngle += GeometryUtils::getAngleDiff(prevAngle, currentAngle);
|
| + }
|
| +
|
| + if (pushTouchPoint(
|
| + proximityInfo, maxPointToKeyLength, i, c, x, y, time, isGeometric,
|
| + isGeometric /* doSampling */, i == lastInputIndex, sumAngle,
|
| + currentNearKeysDistances, prevNearKeysDistances,
|
| + prevPrevNearKeysDistances, sampledInputXs, sampledInputYs,
|
| + sampledInputTimes, sampledLengthCache, sampledInputIndice)) {
|
| + // Previous point information was popped.
|
| + NearKeysDistanceMap* tmp = prevNearKeysDistances;
|
| + prevNearKeysDistances = currentNearKeysDistances;
|
| + currentNearKeysDistances = tmp;
|
| + } else {
|
| + NearKeysDistanceMap* tmp = prevPrevNearKeysDistances;
|
| + prevPrevNearKeysDistances = prevNearKeysDistances;
|
| + prevNearKeysDistances = currentNearKeysDistances;
|
| + currentNearKeysDistances = tmp;
|
| + sumAngle = 0.0f;
|
| + }
|
| + }
|
| + }
|
| + return sampledInputXs->size();
|
| +}
|
| +
|
| +/* static */ const int* ProximityInfoStateUtils::getProximityCodePointsAt(
|
| + const int* const inputProximities,
|
| + const int index) {
|
| + return inputProximities + (index * MAX_PROXIMITY_CHARS_SIZE);
|
| +}
|
| +
|
| +/* static */ int ProximityInfoStateUtils::getPrimaryCodePointAt(
|
| + const int* const inputProximities,
|
| + const int index) {
|
| + return getProximityCodePointsAt(inputProximities, index)[0];
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::initPrimaryInputWord(
|
| + const int inputSize,
|
| + const int* const inputProximities,
|
| + int* primaryInputWord) {
|
| + memset(primaryInputWord, 0, sizeof(primaryInputWord[0]) * MAX_WORD_LENGTH);
|
| + for (int i = 0; i < inputSize; ++i) {
|
| + primaryInputWord[i] = getPrimaryCodePointAt(inputProximities, i);
|
| + }
|
| +}
|
| +
|
| +/* static */ float
|
| +ProximityInfoStateUtils::calculateSquaredDistanceFromSweetSpotCenter(
|
| + const ProximityInfo* const proximityInfo,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const int keyIndex,
|
| + const int inputIndex) {
|
| + const float sweetSpotCenterX = proximityInfo->getSweetSpotCenterXAt(keyIndex);
|
| + const float sweetSpotCenterY = proximityInfo->getSweetSpotCenterYAt(keyIndex);
|
| + const float inputX = static_cast<float>((*sampledInputXs)[inputIndex]);
|
| + const float inputY = static_cast<float>((*sampledInputYs)[inputIndex]);
|
| + return GeometryUtils::SQUARE_FLOAT(inputX - sweetSpotCenterX) +
|
| + GeometryUtils::SQUARE_FLOAT(inputY - sweetSpotCenterY);
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::calculateNormalizedSquaredDistance(
|
| + const ProximityInfo* const proximityInfo,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const int keyIndex,
|
| + const int inputIndex) {
|
| + if (keyIndex == NOT_AN_INDEX) {
|
| + return ProximityInfoParams::NOT_A_DISTANCE_FLOAT;
|
| + }
|
| + if (!proximityInfo->hasSweetSpotData(keyIndex)) {
|
| + return ProximityInfoParams::NOT_A_DISTANCE_FLOAT;
|
| + }
|
| + if (NOT_A_COORDINATE == (*sampledInputXs)[inputIndex]) {
|
| + return ProximityInfoParams::NOT_A_DISTANCE_FLOAT;
|
| + }
|
| + const float squaredDistance = calculateSquaredDistanceFromSweetSpotCenter(
|
| + proximityInfo, sampledInputXs, sampledInputYs, keyIndex, inputIndex);
|
| + const float squaredRadius =
|
| + GeometryUtils::SQUARE_FLOAT(proximityInfo->getSweetSpotRadiiAt(keyIndex));
|
| + return squaredDistance / squaredRadius;
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::initGeometricDistanceInfos(
|
| + const ProximityInfo* const proximityInfo,
|
| + const int sampledInputSize,
|
| + const int lastSavedInputSize,
|
| + const bool isGeometric,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + std::vector<float>* sampledNormalizedSquaredLengthCache) {
|
| + const int keyCount = proximityInfo->getKeyCount();
|
| + sampledNormalizedSquaredLengthCache->resize(sampledInputSize * keyCount);
|
| + for (int i = lastSavedInputSize; i < sampledInputSize; ++i) {
|
| + for (int k = 0; k < keyCount; ++k) {
|
| + const int index = i * keyCount + k;
|
| + const int x = (*sampledInputXs)[i];
|
| + const int y = (*sampledInputYs)[i];
|
| + const float normalizedSquaredDistance =
|
| + proximityInfo->getNormalizedSquaredDistanceFromCenterFloatG(
|
| + k, x, y, isGeometric);
|
| + (*sampledNormalizedSquaredLengthCache)[index] = normalizedSquaredDistance;
|
| + }
|
| + }
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::popInputData(
|
| + std::vector<int>* sampledInputXs,
|
| + std::vector<int>* sampledInputYs,
|
| + std::vector<int>* sampledInputTimes,
|
| + std::vector<int>* sampledLengthCache,
|
| + std::vector<int>* sampledInputIndice) {
|
| + sampledInputXs->pop_back();
|
| + sampledInputYs->pop_back();
|
| + sampledInputTimes->pop_back();
|
| + sampledLengthCache->pop_back();
|
| + sampledInputIndice->pop_back();
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::refreshSpeedRates(
|
| + const int inputSize,
|
| + const int* const xCoordinates,
|
| + const int* const yCoordinates,
|
| + const int* const times,
|
| + const int lastSavedInputSize,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<int>* const sampledInputTimes,
|
| + const std::vector<int>* const sampledLengthCache,
|
| + const std::vector<int>* const sampledInputIndice,
|
| + std::vector<float>* sampledSpeedRates,
|
| + std::vector<float>* sampledDirections) {
|
| + // Relative speed calculation.
|
| + const int sumDuration =
|
| + sampledInputTimes->back() - sampledInputTimes->front();
|
| + const int sumLength =
|
| + sampledLengthCache->back() - sampledLengthCache->front();
|
| + const float averageSpeed =
|
| + static_cast<float>(sumLength) / static_cast<float>(sumDuration);
|
| + sampledSpeedRates->resize(sampledInputSize);
|
| + for (int i = lastSavedInputSize; i < sampledInputSize; ++i) {
|
| + const int index = (*sampledInputIndice)[i];
|
| + int length = 0;
|
| + int duration = 0;
|
| +
|
| + // Calculate velocity by using distances and durations of
|
| + // ProximityInfoParams::NUM_POINTS_FOR_SPEED_CALCULATION points for both
|
| + // forward and
|
| + // backward.
|
| + const int forwardNumPoints =
|
| + std::min(inputSize - 1,
|
| + index + ProximityInfoParams::NUM_POINTS_FOR_SPEED_CALCULATION);
|
| + for (int j = index; j < forwardNumPoints; ++j) {
|
| + if (i < sampledInputSize - 1 && j >= (*sampledInputIndice)[i + 1]) {
|
| + break;
|
| + }
|
| + length += GeometryUtils::getDistanceInt(xCoordinates[j], yCoordinates[j],
|
| + xCoordinates[j + 1],
|
| + yCoordinates[j + 1]);
|
| + duration += times[j + 1] - times[j];
|
| + }
|
| + const int backwardNumPoints = std::max(
|
| + 0, index - ProximityInfoParams::NUM_POINTS_FOR_SPEED_CALCULATION);
|
| + for (int j = index - 1; j >= backwardNumPoints; --j) {
|
| + if (i > 0 && j < (*sampledInputIndice)[i - 1]) {
|
| + break;
|
| + }
|
| + // TODO: use mSampledLengthCache instead?
|
| + length += GeometryUtils::getDistanceInt(xCoordinates[j], yCoordinates[j],
|
| + xCoordinates[j + 1],
|
| + yCoordinates[j + 1]);
|
| + duration += times[j + 1] - times[j];
|
| + }
|
| + if (duration == 0 || sumDuration == 0) {
|
| + // Cannot calculate speed; thus, it gives an average value (1.0);
|
| + (*sampledSpeedRates)[i] = 1.0f;
|
| + } else {
|
| + const float speed =
|
| + static_cast<float>(length) / static_cast<float>(duration);
|
| + (*sampledSpeedRates)[i] = speed / averageSpeed;
|
| + }
|
| + }
|
| +
|
| + // Direction calculation.
|
| + sampledDirections->resize(sampledInputSize - 1);
|
| + for (int i = std::max(0, lastSavedInputSize - 1); i < sampledInputSize - 1;
|
| + ++i) {
|
| + (*sampledDirections)[i] =
|
| + getDirection(sampledInputXs, sampledInputYs, i, i + 1);
|
| + }
|
| + return averageSpeed;
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::refreshBeelineSpeedRates(
|
| + const int mostCommonKeyWidth,
|
| + const float averageSpeed,
|
| + const int inputSize,
|
| + const int* const xCoordinates,
|
| + const int* const yCoordinates,
|
| + const int* times,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<int>* const inputIndice,
|
| + std::vector<int>* beelineSpeedPercentiles) {
|
| + if (DEBUG_SAMPLING_POINTS) {
|
| + AKLOGI("--- refresh beeline speed rates");
|
| + }
|
| + beelineSpeedPercentiles->resize(sampledInputSize);
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + (*beelineSpeedPercentiles)[i] = static_cast<int>(
|
| + calculateBeelineSpeedRate(mostCommonKeyWidth, averageSpeed, i,
|
| + inputSize, xCoordinates, yCoordinates, times,
|
| + sampledInputSize, sampledInputXs,
|
| + sampledInputYs, inputIndice) *
|
| + MAX_PERCENTILE);
|
| + }
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::getDirection(
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const int index0,
|
| + const int index1) {
|
| + ASSERT(sampledInputXs && sampledInputYs);
|
| + const int sampledInputSize = sampledInputXs->size();
|
| + if (index0 < 0 || index0 > sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + if (index1 < 0 || index1 > sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + const int x1 = (*sampledInputXs)[index0];
|
| + const int y1 = (*sampledInputYs)[index0];
|
| + const int x2 = (*sampledInputXs)[index1];
|
| + const int y2 = (*sampledInputYs)[index1];
|
| + return GeometryUtils::getAngle(x1, y1, x2, y2);
|
| +}
|
| +
|
| +// Calculating point to key distance for all near keys and returning the
|
| +// distance between
|
| +// the given point and the nearest key position.
|
| +/* static */ float ProximityInfoStateUtils::updateNearKeysDistances(
|
| + const ProximityInfo* const proximityInfo,
|
| + const float maxPointToKeyLength,
|
| + const int x,
|
| + const int y,
|
| + const bool isGeometric,
|
| + NearKeysDistanceMap* const currentNearKeysDistances) {
|
| + currentNearKeysDistances->clear();
|
| + const int keyCount = proximityInfo->getKeyCount();
|
| + float nearestKeyDistance = maxPointToKeyLength;
|
| + for (int k = 0; k < keyCount; ++k) {
|
| + const float dist =
|
| + proximityInfo->getNormalizedSquaredDistanceFromCenterFloatG(
|
| + k, x, y, isGeometric);
|
| + if (dist < ProximityInfoParams::NEAR_KEY_THRESHOLD_FOR_DISTANCE) {
|
| + currentNearKeysDistances->insert(std::pair<int, float>(k, dist));
|
| + }
|
| + if (nearestKeyDistance > dist) {
|
| + nearestKeyDistance = dist;
|
| + }
|
| + }
|
| + return nearestKeyDistance;
|
| +}
|
| +
|
| +// Check if previous point is at local minimum position to near keys.
|
| +/* static */ bool ProximityInfoStateUtils::isPrevLocalMin(
|
| + const NearKeysDistanceMap* const currentNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevPrevNearKeysDistances) {
|
| + for (NearKeysDistanceMap::const_iterator it = prevNearKeysDistances->begin();
|
| + it != prevNearKeysDistances->end(); ++it) {
|
| + NearKeysDistanceMap::const_iterator itPP =
|
| + prevPrevNearKeysDistances->find(it->first);
|
| + NearKeysDistanceMap::const_iterator itC =
|
| + currentNearKeysDistances->find(it->first);
|
| + const bool isPrevPrevNear =
|
| + (itPP == prevPrevNearKeysDistances->end() ||
|
| + itPP->second >
|
| + it->second + ProximityInfoParams::MARGIN_FOR_PREV_LOCAL_MIN);
|
| + const bool isCurrentNear =
|
| + (itC == currentNearKeysDistances->end() ||
|
| + itC->second >
|
| + it->second + ProximityInfoParams::MARGIN_FOR_PREV_LOCAL_MIN);
|
| + if (isPrevPrevNear && isCurrentNear) {
|
| + return true;
|
| + }
|
| + }
|
| + return false;
|
| +}
|
| +
|
| +// Calculating a point score that indicates usefulness of the point.
|
| +/* static */ float ProximityInfoStateUtils::getPointScore(
|
| + const int mostCommonKeyWidth,
|
| + const int x,
|
| + const int y,
|
| + const int time,
|
| + const bool lastPoint,
|
| + const float nearest,
|
| + const float sumAngle,
|
| + const NearKeysDistanceMap* const currentNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevPrevNearKeysDistances,
|
| + std::vector<int>* sampledInputXs,
|
| + std::vector<int>* sampledInputYs) {
|
| + const size_t size = sampledInputXs->size();
|
| + // If there is only one point, add this point. Besides, if the previous
|
| + // point's distance map
|
| + // is empty, we re-compute nearby keys distances from the current point.
|
| + // Note that the current point is the first point in the incremental input
|
| + // that needs to
|
| + // be re-computed.
|
| + if (size <= 1 || prevNearKeysDistances->empty()) {
|
| + return 0.0f;
|
| + }
|
| +
|
| + const int baseSampleRate = mostCommonKeyWidth;
|
| + const int distPrev =
|
| + GeometryUtils::getDistanceInt(
|
| + sampledInputXs->back(), sampledInputYs->back(),
|
| + (*sampledInputXs)[size - 2], (*sampledInputYs)[size - 2]) *
|
| + ProximityInfoParams::DISTANCE_BASE_SCALE;
|
| + float score = 0.0f;
|
| +
|
| + // Location
|
| + if (!isPrevLocalMin(currentNearKeysDistances, prevNearKeysDistances,
|
| + prevPrevNearKeysDistances)) {
|
| + score += ProximityInfoParams::NOT_LOCALMIN_DISTANCE_SCORE;
|
| + } else if (nearest <
|
| + ProximityInfoParams::NEAR_KEY_THRESHOLD_FOR_POINT_SCORE) {
|
| + // Promote points nearby keys
|
| + score += ProximityInfoParams::LOCALMIN_DISTANCE_AND_NEAR_TO_KEY_SCORE;
|
| + }
|
| + // Angle
|
| + const float angle1 = GeometryUtils::getAngle(x, y, sampledInputXs->back(),
|
| + sampledInputYs->back());
|
| + const float angle2 = GeometryUtils::getAngle(
|
| + sampledInputXs->back(), sampledInputYs->back(),
|
| + (*sampledInputXs)[size - 2], (*sampledInputYs)[size - 2]);
|
| + const float angleDiff = GeometryUtils::getAngleDiff(angle1, angle2);
|
| +
|
| + // Save corner
|
| + if (distPrev >
|
| + baseSampleRate *
|
| + ProximityInfoParams::CORNER_CHECK_DISTANCE_THRESHOLD_SCALE &&
|
| + (sumAngle > ProximityInfoParams::CORNER_SUM_ANGLE_THRESHOLD ||
|
| + angleDiff >
|
| + ProximityInfoParams::CORNER_ANGLE_THRESHOLD_FOR_POINT_SCORE)) {
|
| + score += ProximityInfoParams::CORNER_SCORE;
|
| + }
|
| + return score;
|
| +}
|
| +
|
| +// Sampling touch point and pushing information to vectors.
|
| +// Returning if previous point is popped or not.
|
| +/* static */ bool ProximityInfoStateUtils::pushTouchPoint(
|
| + const ProximityInfo* const proximityInfo,
|
| + const int maxPointToKeyLength,
|
| + const int inputIndex,
|
| + const int nodeCodePoint,
|
| + int x,
|
| + int y,
|
| + const int time,
|
| + const bool isGeometric,
|
| + const bool doSampling,
|
| + const bool isLastPoint,
|
| + const float sumAngle,
|
| + NearKeysDistanceMap* const currentNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevNearKeysDistances,
|
| + const NearKeysDistanceMap* const prevPrevNearKeysDistances,
|
| + std::vector<int>* sampledInputXs,
|
| + std::vector<int>* sampledInputYs,
|
| + std::vector<int>* sampledInputTimes,
|
| + std::vector<int>* sampledLengthCache,
|
| + std::vector<int>* sampledInputIndice) {
|
| + const int mostCommonKeyWidth = proximityInfo->getMostCommonKeyWidth();
|
| +
|
| + size_t size = sampledInputXs->size();
|
| + bool popped = false;
|
| + if (nodeCodePoint < 0 && doSampling) {
|
| + const float nearest =
|
| + updateNearKeysDistances(proximityInfo, maxPointToKeyLength, x, y,
|
| + isGeometric, currentNearKeysDistances);
|
| + const float score = getPointScore(
|
| + mostCommonKeyWidth, x, y, time, isLastPoint, nearest, sumAngle,
|
| + currentNearKeysDistances, prevNearKeysDistances,
|
| + prevPrevNearKeysDistances, sampledInputXs, sampledInputYs);
|
| + if (score < 0) {
|
| + // Pop previous point because it would be useless.
|
| + popInputData(sampledInputXs, sampledInputYs, sampledInputTimes,
|
| + sampledLengthCache, sampledInputIndice);
|
| + size = sampledInputXs->size();
|
| + popped = true;
|
| + } else {
|
| + popped = false;
|
| + }
|
| + // Check if the last point should be skipped.
|
| + if (isLastPoint && size > 0) {
|
| + if (GeometryUtils::getDistanceInt(x, y, sampledInputXs->back(),
|
| + sampledInputYs->back()) *
|
| + ProximityInfoParams::LAST_POINT_SKIP_DISTANCE_SCALE <
|
| + mostCommonKeyWidth) {
|
| + // This point is not used because it's too close to the previous point.
|
| + if (DEBUG_GEO_FULL) {
|
| + AKLOGI(
|
| + "p0: size = %zd, x = %d, y = %d, lx = %d, ly = %d, dist = %d, "
|
| + "width = %d",
|
| + size, x, y, sampledInputXs->back(), sampledInputYs->back(),
|
| + GeometryUtils::getDistanceInt(x, y, sampledInputXs->back(),
|
| + sampledInputYs->back()),
|
| + mostCommonKeyWidth /
|
| + ProximityInfoParams::LAST_POINT_SKIP_DISTANCE_SCALE);
|
| + }
|
| + return popped;
|
| + }
|
| + }
|
| + }
|
| +
|
| + if (nodeCodePoint >= 0 && (x < 0 || y < 0)) {
|
| + const int keyId = proximityInfo->getKeyIndexOf(nodeCodePoint);
|
| + if (keyId >= 0) {
|
| + x = proximityInfo->getKeyCenterXOfKeyIdG(keyId, NOT_AN_INDEX,
|
| + isGeometric);
|
| + y = proximityInfo->getKeyCenterYOfKeyIdG(keyId, NOT_AN_INDEX,
|
| + isGeometric);
|
| + }
|
| + }
|
| +
|
| + // Pushing point information.
|
| + if (size > 0) {
|
| + sampledLengthCache->push_back(
|
| + sampledLengthCache->back() +
|
| + GeometryUtils::getDistanceInt(x, y, sampledInputXs->back(),
|
| + sampledInputYs->back()));
|
| + } else {
|
| + sampledLengthCache->push_back(0);
|
| + }
|
| + sampledInputXs->push_back(x);
|
| + sampledInputYs->push_back(y);
|
| + sampledInputTimes->push_back(time);
|
| + sampledInputIndice->push_back(inputIndex);
|
| + if (DEBUG_GEO_FULL) {
|
| + AKLOGI(
|
| + "pushTouchPoint: x = %03d, y = %03d, time = %d, index = %d, popped ? "
|
| + "%01d",
|
| + x, y, time, inputIndex, popped);
|
| + }
|
| + return popped;
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::calculateBeelineSpeedRate(
|
| + const int mostCommonKeyWidth,
|
| + const float averageSpeed,
|
| + const int id,
|
| + const int inputSize,
|
| + const int* const xCoordinates,
|
| + const int* const yCoordinates,
|
| + const int* times,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<int>* const sampledInputIndices) {
|
| + if (sampledInputSize <= 0 || averageSpeed < 0.001f) {
|
| + if (DEBUG_SAMPLING_POINTS) {
|
| + AKLOGI("--- invalid state: cancel. size = %d, ave = %f", sampledInputSize,
|
| + averageSpeed);
|
| + }
|
| + return 1.0f;
|
| + }
|
| + const int lookupRadius = mostCommonKeyWidth *
|
| + ProximityInfoParams::LOOKUP_RADIUS_PERCENTILE /
|
| + MAX_PERCENTILE;
|
| + const int x0 = (*sampledInputXs)[id];
|
| + const int y0 = (*sampledInputYs)[id];
|
| + const int actualInputIndex = (*sampledInputIndices)[id];
|
| + int tempTime = 0;
|
| + int tempBeelineDistance = 0;
|
| + int start = actualInputIndex;
|
| + // lookup forward
|
| + while (start > 0 && tempBeelineDistance < lookupRadius) {
|
| + tempTime += times[start] - times[start - 1];
|
| + --start;
|
| + tempBeelineDistance = GeometryUtils::getDistanceInt(
|
| + x0, y0, xCoordinates[start], yCoordinates[start]);
|
| + }
|
| + // Exclusive unless this is an edge point
|
| + if (start > 0 && start < actualInputIndex) {
|
| + ++start;
|
| + }
|
| + tempTime = 0;
|
| + tempBeelineDistance = 0;
|
| + int end = actualInputIndex;
|
| + // lookup backward
|
| + while (end < (inputSize - 1) && tempBeelineDistance < lookupRadius) {
|
| + tempTime += times[end + 1] - times[end];
|
| + ++end;
|
| + tempBeelineDistance = GeometryUtils::getDistanceInt(
|
| + x0, y0, xCoordinates[end], yCoordinates[end]);
|
| + }
|
| + // Exclusive unless this is an edge point
|
| + if (end > actualInputIndex && end < (inputSize - 1)) {
|
| + --end;
|
| + }
|
| +
|
| + if (start >= end) {
|
| + if (DEBUG_DOUBLE_LETTER) {
|
| + AKLOGI("--- double letter: start == end %d", start);
|
| + }
|
| + return 1.0f;
|
| + }
|
| +
|
| + const int x2 = xCoordinates[start];
|
| + const int y2 = yCoordinates[start];
|
| + const int x3 = xCoordinates[end];
|
| + const int y3 = yCoordinates[end];
|
| + const int beelineDistance = GeometryUtils::getDistanceInt(x2, y2, x3, y3);
|
| + int adjustedStartTime = times[start];
|
| + if (start == 0 && actualInputIndex == 0 && inputSize > 1) {
|
| + adjustedStartTime += ProximityInfoParams::FIRST_POINT_TIME_OFFSET_MILLIS;
|
| + }
|
| + int adjustedEndTime = times[end];
|
| + if (end == (inputSize - 1) && inputSize > 1) {
|
| + adjustedEndTime -= ProximityInfoParams::FIRST_POINT_TIME_OFFSET_MILLIS;
|
| + }
|
| + const int time = adjustedEndTime - adjustedStartTime;
|
| + if (time <= 0) {
|
| + return 1.0f;
|
| + }
|
| +
|
| + if (time >= ProximityInfoParams::STRONG_DOUBLE_LETTER_TIME_MILLIS) {
|
| + return 0.0f;
|
| + }
|
| + if (DEBUG_DOUBLE_LETTER) {
|
| + AKLOGI(
|
| + "--- (%d, %d) double letter: start = %d, end = %d, dist = %d, time = "
|
| + "%d,"
|
| + " speed = %f, ave = %f, val = %f, start time = %d, end time = %d",
|
| + id, (*sampledInputIndices)[id], start, end, beelineDistance, time,
|
| + (static_cast<float>(beelineDistance) / static_cast<float>(time)),
|
| + averageSpeed,
|
| + ((static_cast<float>(beelineDistance) / static_cast<float>(time)) /
|
| + averageSpeed),
|
| + adjustedStartTime, adjustedEndTime);
|
| + }
|
| + // Offset 1%
|
| + // TODO: Detect double letter more smartly
|
| + return 0.01f +
|
| + static_cast<float>(beelineDistance) / static_cast<float>(time) /
|
| + averageSpeed;
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::getPointAngle(
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const int index) {
|
| + if (!sampledInputXs || !sampledInputYs) {
|
| + return 0.0f;
|
| + }
|
| + const int sampledInputSize = sampledInputXs->size();
|
| + if (index <= 0 || index >= sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + const float previousDirection =
|
| + getDirection(sampledInputXs, sampledInputYs, index - 1, index);
|
| + const float nextDirection =
|
| + getDirection(sampledInputXs, sampledInputYs, index, index + 1);
|
| + const float directionDiff =
|
| + GeometryUtils::getAngleDiff(previousDirection, nextDirection);
|
| + return directionDiff;
|
| +}
|
| +
|
| +/* static */ float ProximityInfoStateUtils::getPointsAngle(
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const int index0,
|
| + const int index1,
|
| + const int index2) {
|
| + if (!sampledInputXs || !sampledInputYs) {
|
| + return 0.0f;
|
| + }
|
| + const int sampledInputSize = sampledInputXs->size();
|
| + if (index0 < 0 || index0 > sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + if (index1 < 0 || index1 > sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + if (index2 < 0 || index2 > sampledInputSize - 1) {
|
| + return 0.0f;
|
| + }
|
| + const float previousDirection =
|
| + getDirection(sampledInputXs, sampledInputYs, index0, index1);
|
| + const float nextDirection =
|
| + getDirection(sampledInputXs, sampledInputYs, index1, index2);
|
| + return GeometryUtils::getAngleDiff(previousDirection, nextDirection);
|
| +}
|
| +
|
| +// This function basically converts from a length to an edit distance.
|
| +// Accordingly, it's obviously
|
| +// wrong to compare with mMaxPointToKeyLength.
|
| +/* static */ float ProximityInfoStateUtils::getPointToKeyByIdLength(
|
| + const float maxPointToKeyLength,
|
| + const std::vector<float>* const sampledNormalizedSquaredLengthCache,
|
| + const int keyCount,
|
| + const int inputIndex,
|
| + const int keyId) {
|
| + if (keyId != NOT_AN_INDEX) {
|
| + const int index = inputIndex * keyCount + keyId;
|
| + return std::min((*sampledNormalizedSquaredLengthCache)[index],
|
| + maxPointToKeyLength);
|
| + }
|
| + // If the char is not a key on the keyboard then return the max length.
|
| + return static_cast<float>(MAX_VALUE_FOR_WEIGHTING);
|
| +}
|
| +
|
| +// Updates probabilities of aligning to some keys and skipping.
|
| +// Word suggestion should be based on this probabilities.
|
| +/* static */ void ProximityInfoStateUtils::updateAlignPointProbabilities(
|
| + const float maxPointToKeyLength,
|
| + const int mostCommonKeyWidth,
|
| + const int keyCount,
|
| + const int start,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<float>* const sampledSpeedRates,
|
| + const std::vector<int>* const sampledLengthCache,
|
| + const std::vector<float>* const sampledNormalizedSquaredLengthCache,
|
| + const ProximityInfo* const proximityInfo,
|
| + std::vector<std::unordered_map<int, float>>* charProbabilities) {
|
| + charProbabilities->resize(sampledInputSize);
|
| + // Calculates probabilities of using a point as a correlated point with the
|
| + // character
|
| + // for each point.
|
| + for (int i = start; i < sampledInputSize; ++i) {
|
| + (*charProbabilities)[i].clear();
|
| + // First, calculates skip probability. Starts from MAX_SKIP_PROBABILITY.
|
| + // Note that all values that are multiplied to this probability should be in
|
| + // [0.0, 1.0];
|
| + float skipProbability = ProximityInfoParams::MAX_SKIP_PROBABILITY;
|
| +
|
| + const float currentAngle = getPointAngle(sampledInputXs, sampledInputYs, i);
|
| + const float speedRate = (*sampledSpeedRates)[i];
|
| +
|
| + float nearestKeyDistance = static_cast<float>(MAX_VALUE_FOR_WEIGHTING);
|
| + for (int j = 0; j < keyCount; ++j) {
|
| + const float distance = getPointToKeyByIdLength(
|
| + maxPointToKeyLength, sampledNormalizedSquaredLengthCache, keyCount, i,
|
| + j);
|
| + if (distance < nearestKeyDistance) {
|
| + nearestKeyDistance = distance;
|
| + }
|
| + }
|
| +
|
| + if (i == 0) {
|
| + skipProbability *=
|
| + std::min(1.0f, nearestKeyDistance *
|
| + ProximityInfoParams::NEAREST_DISTANCE_WEIGHT +
|
| + ProximityInfoParams::NEAREST_DISTANCE_BIAS);
|
| + // Promote the first point
|
| + skipProbability *= ProximityInfoParams::SKIP_FIRST_POINT_PROBABILITY;
|
| + } else if (i == sampledInputSize - 1) {
|
| + skipProbability *= std::min(
|
| + 1.0f, nearestKeyDistance *
|
| + ProximityInfoParams::NEAREST_DISTANCE_WEIGHT_FOR_LAST +
|
| + ProximityInfoParams::NEAREST_DISTANCE_BIAS_FOR_LAST);
|
| + // Promote the last point
|
| + skipProbability *= ProximityInfoParams::SKIP_LAST_POINT_PROBABILITY;
|
| + } else {
|
| + // If the current speed is relatively slower than adjacent keys, we
|
| + // promote this point.
|
| + if ((*sampledSpeedRates)[i - 1] - ProximityInfoParams::SPEED_MARGIN >
|
| + speedRate &&
|
| + speedRate <
|
| + (*sampledSpeedRates)[i + 1] - ProximityInfoParams::SPEED_MARGIN) {
|
| + if (currentAngle < ProximityInfoParams::CORNER_ANGLE_THRESHOLD) {
|
| + skipProbability *= std::min(
|
| + 1.0f, speedRate * ProximityInfoParams::
|
| + SLOW_STRAIGHT_WEIGHT_FOR_SKIP_PROBABILITY);
|
| + } else {
|
| + // If the angle is small enough, we promote this point more. (e.g. pit
|
| + // vs put)
|
| + skipProbability *= std::min(
|
| + 1.0f,
|
| + speedRate *
|
| + ProximityInfoParams::SPEED_WEIGHT_FOR_SKIP_PROBABILITY +
|
| + ProximityInfoParams::MIN_SPEED_RATE_FOR_SKIP_PROBABILITY);
|
| + }
|
| + }
|
| +
|
| + skipProbability *=
|
| + std::min(1.0f, speedRate * nearestKeyDistance *
|
| + ProximityInfoParams::NEAREST_DISTANCE_WEIGHT +
|
| + ProximityInfoParams::NEAREST_DISTANCE_BIAS);
|
| +
|
| + // Adjusts skip probability by a rate depending on angle.
|
| + // ANGLE_RATE of skipProbability is adjusted by current angle.
|
| + skipProbability *=
|
| + (M_PI_F - currentAngle) / M_PI_F * ProximityInfoParams::ANGLE_WEIGHT +
|
| + (1.0f - ProximityInfoParams::ANGLE_WEIGHT);
|
| + if (currentAngle > ProximityInfoParams::DEEP_CORNER_ANGLE_THRESHOLD) {
|
| + skipProbability *= ProximityInfoParams::SKIP_DEEP_CORNER_PROBABILITY;
|
| + }
|
| + // We assume the angle of this point is the angle for point[i], point[i -
|
| + // 2]
|
| + // and point[i - 3]. The reason why we don't use the angle for point[i],
|
| + // point[i - 1]
|
| + // and point[i - 2] is this angle can be more affected by the noise.
|
| + const float prevAngle =
|
| + getPointsAngle(sampledInputXs, sampledInputYs, i, i - 2, i - 3);
|
| + if (i >= 3 && prevAngle < ProximityInfoParams::STRAIGHT_ANGLE_THRESHOLD &&
|
| + currentAngle > ProximityInfoParams::CORNER_ANGLE_THRESHOLD) {
|
| + skipProbability *= ProximityInfoParams::SKIP_CORNER_PROBABILITY;
|
| + }
|
| + }
|
| +
|
| + // probabilities must be in [0.0,
|
| + // ProximityInfoParams::MAX_SKIP_PROBABILITY];
|
| + ASSERT(skipProbability >= 0.0f);
|
| + ASSERT(skipProbability <= ProximityInfoParams::MAX_SKIP_PROBABILITY);
|
| + (*charProbabilities)[i][NOT_AN_INDEX] = skipProbability;
|
| +
|
| + // Second, calculates key probabilities by dividing the rest probability
|
| + // (1.0f - skipProbability).
|
| + const float inputCharProbability = 1.0f - skipProbability;
|
| +
|
| + const float speedMultipliedByAngleRate = std::min(
|
| + speedRate * currentAngle / M_PI_F *
|
| + ProximityInfoParams::SPEEDxANGLE_WEIGHT_FOR_STANDARD_DEVIATION,
|
| + ProximityInfoParams::MAX_SPEEDxANGLE_RATE_FOR_STANDARD_DEVIATION);
|
| + const float speedMultipliedByNearestKeyDistanceRate = std::min(
|
| + speedRate * nearestKeyDistance *
|
| + ProximityInfoParams::SPEEDxNEAREST_WEIGHT_FOR_STANDARD_DEVIATION,
|
| + ProximityInfoParams::MAX_SPEEDxNEAREST_RATE_FOR_STANDARD_DEVIATION);
|
| + const float sigma =
|
| + (speedMultipliedByAngleRate + speedMultipliedByNearestKeyDistanceRate +
|
| + ProximityInfoParams::MIN_STANDARD_DEVIATION) *
|
| + mostCommonKeyWidth;
|
| + float theta = 0.0f;
|
| + // TODO: Use different metrics to compute sigmas.
|
| + float sigmaX = sigma;
|
| + float sigmaY = sigma;
|
| + if (i == 0 && i != sampledInputSize - 1) {
|
| + // First point
|
| + theta = getDirection(sampledInputXs, sampledInputYs, i + 1, i);
|
| + sigmaX *= ProximityInfoParams::STANDARD_DEVIATION_X_WEIGHT_FOR_FIRST;
|
| + sigmaY *= ProximityInfoParams::STANDARD_DEVIATION_Y_WEIGHT_FOR_FIRST;
|
| + } else {
|
| + if (i == sampledInputSize - 1) {
|
| + // Last point
|
| + sigmaX *= ProximityInfoParams::STANDARD_DEVIATION_X_WEIGHT_FOR_LAST;
|
| + sigmaY *= ProximityInfoParams::STANDARD_DEVIATION_Y_WEIGHT_FOR_LAST;
|
| + } else {
|
| + sigmaX *= ProximityInfoParams::STANDARD_DEVIATION_X_WEIGHT;
|
| + sigmaY *= ProximityInfoParams::STANDARD_DEVIATION_Y_WEIGHT;
|
| + }
|
| + theta = getDirection(sampledInputXs, sampledInputYs, i, i - 1);
|
| + }
|
| + NormalDistribution2D distribution((*sampledInputXs)[i], sigmaX,
|
| + (*sampledInputYs)[i], sigmaY, theta);
|
| + // Summing up probability densities of all near keys.
|
| + float sumOfProbabilityDensities = 0.0f;
|
| + for (int j = 0; j < keyCount; ++j) {
|
| + sumOfProbabilityDensities += distribution.getProbabilityDensity(
|
| + proximityInfo->getKeyCenterXOfKeyIdG(
|
| + j, NOT_A_COORDINATE /* referencePointX */,
|
| + true /* isGeometric */),
|
| + proximityInfo->getKeyCenterYOfKeyIdG(
|
| + j, NOT_A_COORDINATE /* referencePointY */,
|
| + true /* isGeometric */));
|
| + }
|
| +
|
| + // Split the probability of an input point to keys that are close to the
|
| + // input point.
|
| + for (int j = 0; j < keyCount; ++j) {
|
| + const float probabilityDensity = distribution.getProbabilityDensity(
|
| + proximityInfo->getKeyCenterXOfKeyIdG(
|
| + j, NOT_A_COORDINATE /* referencePointX */,
|
| + true /* isGeometric */),
|
| + proximityInfo->getKeyCenterYOfKeyIdG(
|
| + j, NOT_A_COORDINATE /* referencePointY */,
|
| + true /* isGeometric */));
|
| + const float probability =
|
| + inputCharProbability * probabilityDensity / sumOfProbabilityDensities;
|
| + (*charProbabilities)[i][j] = probability;
|
| + }
|
| + }
|
| +
|
| + if (DEBUG_POINTS_PROBABILITY) {
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + std::stringstream sstream;
|
| + sstream << i << ", ";
|
| + sstream << "(" << (*sampledInputXs)[i] << ", " << (*sampledInputYs)[i]
|
| + << "), ";
|
| + sstream << "Speed: " << (*sampledSpeedRates)[i] << ", ";
|
| + sstream << "Angle: " << getPointAngle(sampledInputXs, sampledInputYs, i)
|
| + << ", \n";
|
| +
|
| + for (std::unordered_map<int, float>::iterator it =
|
| + (*charProbabilities)[i].begin();
|
| + it != (*charProbabilities)[i].end(); ++it) {
|
| + if (it->first == NOT_AN_INDEX) {
|
| + sstream << it->first << "(skip):" << it->second << "\n";
|
| + } else {
|
| + sstream
|
| + << it->first << "("
|
| + //<< static_cast<char>(mProximityInfo->getCodePointOf(it->first))
|
| + << "):" << it->second << "\n";
|
| + }
|
| + }
|
| + AKLOGI("%s", sstream.str().c_str());
|
| + }
|
| + }
|
| +
|
| + // Decrease key probabilities of points which don't have the highest
|
| + // probability of that key
|
| + // among nearby points. Probabilities of the first point and the last point
|
| + // are not suppressed.
|
| + for (int i = std::max(start, 1); i < sampledInputSize; ++i) {
|
| + for (int j = i + 1; j < sampledInputSize; ++j) {
|
| + if (!suppressCharProbabilities(mostCommonKeyWidth, sampledInputSize,
|
| + sampledLengthCache, i, j,
|
| + charProbabilities)) {
|
| + break;
|
| + }
|
| + }
|
| + for (int j = i - 1; j >= std::max(start, 0); --j) {
|
| + if (!suppressCharProbabilities(mostCommonKeyWidth, sampledInputSize,
|
| + sampledLengthCache, i, j,
|
| + charProbabilities)) {
|
| + break;
|
| + }
|
| + }
|
| + }
|
| +
|
| + // Converting from raw probabilities to log probabilities to calculate spatial
|
| + // distance.
|
| + for (int i = start; i < sampledInputSize; ++i) {
|
| + for (int j = 0; j < keyCount; ++j) {
|
| + std::unordered_map<int, float>::iterator it =
|
| + (*charProbabilities)[i].find(j);
|
| + if (it == (*charProbabilities)[i].end()) {
|
| + continue;
|
| + } else if (it->second < ProximityInfoParams::MIN_PROBABILITY) {
|
| + // Erases from near keys vector because it has very low probability.
|
| + (*charProbabilities)[i].erase(j);
|
| + } else {
|
| + it->second = -logf(it->second);
|
| + }
|
| + }
|
| + (*charProbabilities)[i][NOT_AN_INDEX] =
|
| + -logf((*charProbabilities)[i][NOT_AN_INDEX]);
|
| + }
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::updateSampledSearchKeySets(
|
| + const ProximityInfo* const proximityInfo,
|
| + const int sampledInputSize,
|
| + const int lastSavedInputSize,
|
| + const std::vector<int>* const sampledLengthCache,
|
| + const std::vector<std::unordered_map<int, float>>* const charProbabilities,
|
| + std::vector<NearKeycodesSet>* sampledSearchKeySets,
|
| + std::vector<std::vector<int>>* sampledSearchKeyVectors) {
|
| + sampledSearchKeySets->resize(sampledInputSize);
|
| + sampledSearchKeyVectors->resize(sampledInputSize);
|
| + const int readForwordLength =
|
| + static_cast<int>(hypotf(proximityInfo->getKeyboardWidth(),
|
| + proximityInfo->getKeyboardHeight()) *
|
| + ProximityInfoParams::SEARCH_KEY_RADIUS_RATIO);
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + if (i >= lastSavedInputSize) {
|
| + (*sampledSearchKeySets)[i].reset();
|
| + }
|
| + for (int j = std::max(i, lastSavedInputSize); j < sampledInputSize; ++j) {
|
| + // TODO: Investigate if this is required. This may not fail.
|
| + if ((*sampledLengthCache)[j] - (*sampledLengthCache)[i] >=
|
| + readForwordLength) {
|
| + break;
|
| + }
|
| + for (const auto& charProbability : charProbabilities->at(j)) {
|
| + if (charProbability.first == NOT_AN_INDEX) {
|
| + continue;
|
| + }
|
| + (*sampledSearchKeySets)[i].set(charProbability.first);
|
| + }
|
| + }
|
| + }
|
| + const int keyCount = proximityInfo->getKeyCount();
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + std::vector<int>* searchKeyVector = &(*sampledSearchKeyVectors)[i];
|
| + searchKeyVector->clear();
|
| + for (int j = 0; j < keyCount; ++j) {
|
| + if ((*sampledSearchKeySets)[i].test(j)) {
|
| + const int keyCodePoint = proximityInfo->getCodePointOf(j);
|
| + if (std::find(searchKeyVector->begin(), searchKeyVector->end(),
|
| + keyCodePoint) == searchKeyVector->end()) {
|
| + searchKeyVector->push_back(keyCodePoint);
|
| + }
|
| + }
|
| + }
|
| + }
|
| +}
|
| +
|
| +// Decreases char probabilities of index0 by checking probabilities of a near
|
| +// point (index1) and
|
| +// increases char probabilities of index1 by checking probabilities of index0.
|
| +/* static */ bool ProximityInfoStateUtils::suppressCharProbabilities(
|
| + const int mostCommonKeyWidth,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const lengthCache,
|
| + const int index0,
|
| + const int index1,
|
| + std::vector<std::unordered_map<int, float>>* charProbabilities) {
|
| + ASSERT(0 <= index0 && index0 < sampledInputSize);
|
| + ASSERT(0 <= index1 && index1 < sampledInputSize);
|
| + const float keyWidthFloat = static_cast<float>(mostCommonKeyWidth);
|
| + const float diff = fabsf(
|
| + static_cast<float>((*lengthCache)[index0] - (*lengthCache)[index1]));
|
| + if (diff > keyWidthFloat * ProximityInfoParams::SUPPRESSION_LENGTH_WEIGHT) {
|
| + return false;
|
| + }
|
| + const float suppressionRate =
|
| + ProximityInfoParams::MIN_SUPPRESSION_RATE +
|
| + diff / keyWidthFloat / ProximityInfoParams::SUPPRESSION_LENGTH_WEIGHT *
|
| + ProximityInfoParams::SUPPRESSION_WEIGHT;
|
| + for (std::unordered_map<int, float>::iterator it =
|
| + (*charProbabilities)[index0].begin();
|
| + it != (*charProbabilities)[index0].end(); ++it) {
|
| + std::unordered_map<int, float>::iterator it2 =
|
| + (*charProbabilities)[index1].find(it->first);
|
| + if (it2 != (*charProbabilities)[index1].end() && it->second < it2->second) {
|
| + const float newProbability = it->second * suppressionRate;
|
| + const float suppression = it->second - newProbability;
|
| + it->second = newProbability;
|
| + // mCharProbabilities[index0][NOT_AN_INDEX] is the probability of skipping
|
| + // this point.
|
| + (*charProbabilities)[index0][NOT_AN_INDEX] += suppression;
|
| +
|
| + // Add the probability of the same key nearby index1
|
| + const float probabilityGain = std::min(
|
| + suppression *
|
| + ProximityInfoParams::SUPPRESSION_WEIGHT_FOR_PROBABILITY_GAIN,
|
| + (*charProbabilities)[index1][NOT_AN_INDEX] *
|
| + ProximityInfoParams::
|
| + SKIP_PROBABALITY_WEIGHT_FOR_PROBABILITY_GAIN);
|
| + it2->second += probabilityGain;
|
| + (*charProbabilities)[index1][NOT_AN_INDEX] -= probabilityGain;
|
| + }
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +/* static */ bool
|
| +ProximityInfoStateUtils::checkAndReturnIsContinuousSuggestionPossible(
|
| + const int inputSize,
|
| + const int* const xCoordinates,
|
| + const int* const yCoordinates,
|
| + const int* const times,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<int>* const sampledTimes,
|
| + const std::vector<int>* const sampledInputIndices) {
|
| + if (inputSize < sampledInputSize) {
|
| + return false;
|
| + }
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + const int index = (*sampledInputIndices)[i];
|
| + if (index >= inputSize) {
|
| + return false;
|
| + }
|
| + if (xCoordinates[index] != (*sampledInputXs)[i] ||
|
| + yCoordinates[index] != (*sampledInputYs)[i]) {
|
| + return false;
|
| + }
|
| + if (!times) {
|
| + continue;
|
| + }
|
| + if (times[index] != (*sampledTimes)[i]) {
|
| + return false;
|
| + }
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +// Get a word that is detected by tracing the most probable string into
|
| +// codePointBuf and
|
| +// returns probability of generating the word.
|
| +/* static */ float ProximityInfoStateUtils::getMostProbableString(
|
| + const ProximityInfo* const proximityInfo,
|
| + const int sampledInputSize,
|
| + const std::vector<std::unordered_map<int, float>>* const charProbabilities,
|
| + int* const codePointBuf) {
|
| + ASSERT(sampledInputSize >= 0);
|
| + memset(codePointBuf, 0, sizeof(codePointBuf[0]) * MAX_WORD_LENGTH);
|
| + int index = 0;
|
| + float sumLogProbability = 0.0f;
|
| + // TODO: Current implementation is greedy algorithm. DP would be efficient for
|
| + // many cases.
|
| + for (int i = 0; i < sampledInputSize && index < MAX_WORD_LENGTH - 1; ++i) {
|
| + float minLogProbability = static_cast<float>(MAX_VALUE_FOR_WEIGHTING);
|
| + int character = NOT_AN_INDEX;
|
| + for (std::unordered_map<int, float>::const_iterator it =
|
| + (*charProbabilities)[i].begin();
|
| + it != (*charProbabilities)[i].end(); ++it) {
|
| + const float logProbability =
|
| + (it->first != NOT_AN_INDEX)
|
| + ? it->second + ProximityInfoParams::DEMOTION_LOG_PROBABILITY
|
| + : it->second;
|
| + if (logProbability < minLogProbability) {
|
| + minLogProbability = logProbability;
|
| + character = it->first;
|
| + }
|
| + }
|
| + if (character != NOT_AN_INDEX) {
|
| + const int codePoint = proximityInfo->getCodePointOf(character);
|
| + if (codePoint == NOT_A_CODE_POINT) {
|
| + AKLOGE(
|
| + "Key index(%d) is not found. Cannot construct most probable string",
|
| + character);
|
| + ASSERT(false);
|
| + // Make the length zero, which means most probable string won't be used.
|
| + index = 0;
|
| + break;
|
| + }
|
| + codePointBuf[index] = codePoint;
|
| + index++;
|
| + }
|
| + sumLogProbability += minLogProbability;
|
| + }
|
| + codePointBuf[index] = '\0';
|
| + return sumLogProbability;
|
| +}
|
| +
|
| +/* static */ void ProximityInfoStateUtils::dump(
|
| + const bool isGeometric,
|
| + const int inputSize,
|
| + const int* const inputXCoordinates,
|
| + const int* const inputYCoordinates,
|
| + const int sampledInputSize,
|
| + const std::vector<int>* const sampledInputXs,
|
| + const std::vector<int>* const sampledInputYs,
|
| + const std::vector<int>* const sampledTimes,
|
| + const std::vector<float>* const sampledSpeedRates,
|
| + const std::vector<int>* const sampledBeelineSpeedPercentiles) {
|
| + if (DEBUG_GEO_FULL) {
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + AKLOGI("Sampled(%d): x = %d, y = %d, time = %d", i, (*sampledInputXs)[i],
|
| + (*sampledInputYs)[i], sampledTimes ? (*sampledTimes)[i] : -1);
|
| + }
|
| + }
|
| +
|
| + std::stringstream originalX, originalY, sampledX, sampledY;
|
| + for (int i = 0; i < inputSize; ++i) {
|
| + originalX << inputXCoordinates[i];
|
| + originalY << inputYCoordinates[i];
|
| + if (i != inputSize - 1) {
|
| + originalX << ";";
|
| + originalY << ";";
|
| + }
|
| + }
|
| + AKLOGI("===== sampled points =====");
|
| + for (int i = 0; i < sampledInputSize; ++i) {
|
| + if (isGeometric) {
|
| + AKLOGI(
|
| + "%d: x = %d, y = %d, time = %d, relative speed = %.4f, beeline speed "
|
| + "= %d",
|
| + i, (*sampledInputXs)[i], (*sampledInputYs)[i], (*sampledTimes)[i],
|
| + (*sampledSpeedRates)[i], (*sampledBeelineSpeedPercentiles)[i]);
|
| + }
|
| + sampledX << (*sampledInputXs)[i];
|
| + sampledY << (*sampledInputYs)[i];
|
| + if (i != sampledInputSize - 1) {
|
| + sampledX << ";";
|
| + sampledY << ";";
|
| + }
|
| + }
|
| + AKLOGI("original points:\n%s, %s,\nsampled points:\n%s, %s,\n",
|
| + originalX.str().c_str(), originalY.str().c_str(),
|
| + sampledX.str().c_str(), sampledY.str().c_str());
|
| +}
|
| +} // namespace latinime
|
|
|