| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include "webrtc/modules/audio_processing/echo_detector/mean_variance_estimator.
h" | 11 #include "webrtc/modules/audio_processing/echo_detector/mean_variance_estimator.
h" |
| 12 | 12 |
| 13 #include <math.h> | 13 #include <math.h> |
| 14 | 14 |
| 15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
| 16 | 16 |
| 17 namespace webrtc { | 17 namespace webrtc { |
| 18 namespace { | 18 namespace { |
| 19 | 19 |
| 20 // Parameter controlling the adaptation speed. | 20 // Parameter controlling the adaptation speed. |
| 21 constexpr float kAlpha = 0.001f; | 21 constexpr float kAlpha = 0.001f; |
| 22 | 22 |
| 23 } // namespace | 23 } // namespace |
| 24 | 24 |
| 25 void MeanVarianceEstimator::Update(float value) { | 25 void MeanVarianceEstimator::Update(float value) { |
| 26 mean_ = (1.f - kAlpha) * mean_ + kAlpha * value; | 26 mean_ = (1.f - kAlpha) * mean_ + kAlpha * value; |
| 27 variance_ = | 27 variance_ = |
| 28 (1.f - kAlpha) * variance_ + kAlpha * (value - mean_) * (value - mean_); | 28 (1.f - kAlpha) * variance_ + kAlpha * (value - mean_) * (value - mean_); |
| 29 RTC_DCHECK(isfinite(mean_)); |
| 30 RTC_DCHECK(isfinite(variance_)); |
| 29 } | 31 } |
| 30 | 32 |
| 31 float MeanVarianceEstimator::std_deviation() const { | 33 float MeanVarianceEstimator::std_deviation() const { |
| 32 RTC_DCHECK_GE(variance_, 0.f); | 34 RTC_DCHECK_GE(variance_, 0.f); |
| 33 return sqrtf(variance_); | 35 return sqrtf(variance_); |
| 34 } | 36 } |
| 35 | 37 |
| 36 float MeanVarianceEstimator::mean() const { | 38 float MeanVarianceEstimator::mean() const { |
| 37 return mean_; | 39 return mean_; |
| 38 } | 40 } |
| 39 | 41 |
| 40 void MeanVarianceEstimator::Clear() { | 42 void MeanVarianceEstimator::Clear() { |
| 41 mean_ = 0.f; | 43 mean_ = 0.f; |
| 42 variance_ = 0.f; | 44 variance_ = 0.f; |
| 43 } | 45 } |
| 44 | 46 |
| 45 } // namespace webrtc | 47 } // namespace webrtc |
| OLD | NEW |