| Index: third_party/android_prediction/suggest/core/layout/normal_distribution_2d.h
|
| diff --git a/third_party/android_prediction/suggest/core/layout/normal_distribution_2d.h b/third_party/android_prediction/suggest/core/layout/normal_distribution_2d.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..2f8c4da6c1cb0330355aa0ed014041b43ea543fb
|
| --- /dev/null
|
| +++ b/third_party/android_prediction/suggest/core/layout/normal_distribution_2d.h
|
| @@ -0,0 +1,59 @@
|
| +/*
|
| + * Copyright (C) 2014 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.
|
| + */
|
| +
|
| +#ifndef LATINIME_NORMAL_DISTRIBUTION_2D_H
|
| +#define LATINIME_NORMAL_DISTRIBUTION_2D_H
|
| +
|
| +#include <cmath>
|
| +
|
| +#include "third_party/android_prediction/defines.h"
|
| +#include "third_party/android_prediction/suggest/core/layout/geometry_utils.h"
|
| +#include "third_party/android_prediction/suggest/core/layout/normal_distribution.h"
|
| +
|
| +namespace latinime {
|
| +
|
| +// Normal distribution on a 2D plane. The covariance is always zero, but the distribution can be
|
| +// rotated.
|
| +class NormalDistribution2D {
|
| + public:
|
| + NormalDistribution2D(const float uX, const float sigmaX, const float uY, const float sigmaY,
|
| + const float theta)
|
| + : mXDistribution(0.0f, sigmaX), mYDistribution(0.0f, sigmaY), mUX(uX), mUY(uY),
|
| + mSinTheta(sinf(theta)), mCosTheta(cosf(theta)) {}
|
| +
|
| + float getProbabilityDensity(const float x, const float y) const {
|
| + // Shift
|
| + const float shiftedX = x - mUX;
|
| + const float shiftedY = y - mUY;
|
| + // Rotate
|
| + const float rotatedShiftedX = mCosTheta * shiftedX + mSinTheta * shiftedY;
|
| + const float rotatedShiftedY = -mSinTheta * shiftedX + mCosTheta * shiftedY;
|
| + return mXDistribution.getProbabilityDensity(rotatedShiftedX)
|
| + * mYDistribution.getProbabilityDensity(rotatedShiftedY);
|
| + }
|
| +
|
| + private:
|
| + DISALLOW_IMPLICIT_CONSTRUCTORS(NormalDistribution2D);
|
| +
|
| + const NormalDistribution mXDistribution;
|
| + const NormalDistribution mYDistribution;
|
| + const float mUX;
|
| + const float mUY;
|
| + const float mSinTheta;
|
| + const float mCosTheta;
|
| +};
|
| +} // namespace latinime
|
| +#endif // LATINIME_NORMAL_DISTRIBUTION_2D_H
|
|
|