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/level_controller/peak_level_estimator. h" | 11 #include "webrtc/modules/audio_processing/level_controller/peak_level_estimator. h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 | 14 |
15 #include "webrtc/modules/audio_processing/audio_buffer.h" | 15 #include "webrtc/modules/audio_processing/audio_buffer.h" |
16 #include "webrtc/modules/audio_processing/level_controller/lc_constants.h" | |
16 #include "webrtc/modules/audio_processing/logging/apm_data_dumper.h" | 17 #include "webrtc/modules/audio_processing/logging/apm_data_dumper.h" |
17 | 18 |
18 namespace webrtc { | 19 namespace webrtc { |
19 | 20 |
20 PeakLevelEstimator::PeakLevelEstimator() { | 21 PeakLevelEstimator::PeakLevelEstimator() { |
21 Initialize(); | 22 Initialize(); |
22 } | 23 } |
23 | 24 |
24 PeakLevelEstimator::~PeakLevelEstimator() {} | 25 PeakLevelEstimator::~PeakLevelEstimator() {} |
25 | 26 |
26 void PeakLevelEstimator::Initialize() { | 27 void PeakLevelEstimator::Initialize() { |
27 peak_level_ = 1000.f; | 28 peak_level_ = kTargetLcPeakLevel; |
28 hold_counter_ = 0; | 29 hold_counter_ = 0; |
30 initialization_phase_ = true; | |
29 } | 31 } |
30 | 32 |
31 float PeakLevelEstimator::Analyze(SignalClassifier::SignalType signal_type, | 33 float PeakLevelEstimator::Analyze(SignalClassifier::SignalType signal_type, |
32 float frame_peak_level) { | 34 float frame_peak_level) { |
33 if (frame_peak_level > 0) { | 35 if (frame_peak_level == 0) { |
34 if (peak_level_ < frame_peak_level) { | 36 return peak_level_; |
hlundin-webrtc
2016/06/30 07:36:23
I'm all in favor of early returns, but you are no
peah-webrtc
2016/06/30 14:38:18
Good point! Added a DCHECK.
Done.
| |
35 // Smoothly update the estimate upwards when the frame peak level is | 37 } |
36 // higher than the estimate. | |
37 peak_level_ += 0.1f * (frame_peak_level - peak_level_); | |
38 hold_counter_ = 100; | |
39 } else { | |
40 hold_counter_ = std::max(0, hold_counter_ - 1); | |
41 | 38 |
42 // When the signal is highly non-stationary, update the estimate slowly | 39 if (peak_level_ < frame_peak_level) { |
43 // downwards if the estimate is lower than the frame peak level. | 40 // Smoothly update the estimate upwards when the frame peak level is |
44 if (signal_type == SignalClassifier::SignalType::kHighlyNonStationary && | 41 // higher than the estimate. |
45 hold_counter_ == 0) { | 42 peak_level_ += 0.1f * (frame_peak_level - peak_level_); |
46 peak_level_ = | 43 hold_counter_ = 100; |
47 std::max(peak_level_ + 0.01f * (frame_peak_level - peak_level_), | 44 initialization_phase_ = false; |
48 peak_level_ * 0.995f); | 45 } else { |
49 } | 46 hold_counter_ = std::max(0, hold_counter_ - 1); |
47 | |
48 // When the signal is highly non-stationary, update the estimate slowly | |
49 // downwards if the estimate is lower than the frame peak level. | |
50 if ((signal_type == SignalClassifier::SignalType::kHighlyNonStationary && | |
51 hold_counter_ == 0) || | |
52 initialization_phase_) { | |
53 peak_level_ = | |
54 std::max(peak_level_ + 0.01f * (frame_peak_level - peak_level_), | |
55 peak_level_ * 0.995f); | |
50 } | 56 } |
51 } | 57 } |
52 | 58 |
53 peak_level_ = std::max(peak_level_, 30.f); | 59 peak_level_ = std::max(peak_level_, 30.f); |
54 | 60 |
55 return peak_level_; | 61 return peak_level_; |
56 } | 62 } |
57 | 63 |
58 } // namespace webrtc | 64 } // namespace webrtc |
OLD | NEW |