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 |