Chromium Code Reviews| 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 |