Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(322)

Unified Diff: webrtc/modules/audio_processing/level_controller/gain_selector.cc

Issue 2090583002: New module for the adaptive level controlling functionality in the audio processing module (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Corrected the initial behavior for the peak level estimate, and ensured a nonzero minimum peak leveā€¦ Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: webrtc/modules/audio_processing/level_controller/gain_selector.cc
diff --git a/webrtc/modules/audio_processing/level_controller/gain_selector.cc b/webrtc/modules/audio_processing/level_controller/gain_selector.cc
new file mode 100644
index 0000000000000000000000000000000000000000..de942708c34e89fa510289df4f327259e0587e34
--- /dev/null
+++ b/webrtc/modules/audio_processing/level_controller/gain_selector.cc
@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "webrtc/modules/audio_processing/level_controller/gain_selector.h"
+
+#include <math.h>
+#include <algorithm>
+
+#include "webrtc/base/checks.h"
+#include "webrtc/modules/audio_processing/include/audio_processing.h"
+#include "webrtc/modules/audio_processing/level_controller/lc_constants.h"
+
+namespace webrtc {
+
+GainSelector::GainSelector() {
+ Initialize(AudioProcessing::kSampleRate48kHz);
+}
+
+void GainSelector::Initialize(int sample_rate_hz) {
+ gain_ = 1.f;
+ frame_length_ = sample_rate_hz / 100;
hlundin-webrtc 2016/06/27 11:21:14 rtc::CheckedDivExact
peah-webrtc 2016/06/27 22:51:47 Great! Done.
+}
+
+/* Chooses the gain to apply by the level controller such that
hlundin-webrtc 2016/06/27 11:21:14 Use // style comments.
peah-webrtc 2016/06/27 22:51:48 Done.
+ 1) The level of the stationary noise does not exceed
+ a predefined threshold.
+ 2) The gain does not exceed the that has been deemed
hlundin-webrtc 2016/06/27 11:21:14 deemed to -> found to
hlundin-webrtc 2016/06/27 11:21:14 "the that" ?
peah-webrtc 2016/06/27 22:51:47 Done.
peah-webrtc 2016/06/27 22:51:47 Done.
+ to saturate the signal.
+ 3) Such that the peak level achieves the target peak level.
hlundin-webrtc 2016/06/27 11:21:14 Skip "Such that"; you already have that before the
peah-webrtc 2016/06/27 22:51:47 Done.
+ 4) The gain is not below 1.
+ 5) The gain is not above the maximum gain. */
+float GainSelector::GetNewGain(float peak_level,
+ float noise_energy,
+ float saturating_gain) {
+ float desired_gain = gain_;
+
+ RTC_DCHECK_LT(0.f, peak_level);
hlundin-webrtc 2016/06/27 11:21:14 This will fail if peak_level == 0.f. Is this inten
peah-webrtc 2016/06/27 22:51:47 Yes, the peak level detector should ensure a minim
hlundin-webrtc 2016/06/28 11:29:00 Acknowledged.
peah-webrtc 2016/06/28 22:19:37 Acknowledged.
+ desired_gain = kTargetLcPeakLevel / peak_level;
+
+ float max_noise_energy = kMaxLcNoisePower * frame_length_;
+ if (noise_energy * desired_gain * desired_gain > max_noise_energy) {
+ RTC_DCHECK_LE(0.f, noise_energy);
+ desired_gain = sqrtf(max_noise_energy / noise_energy);
+ }
+
+ gain_ += 0.2f * (desired_gain - gain_);
+ gain_ = std::min(gain_, saturating_gain);
+ gain_ = std::min(gain_, kMaxLcGain);
+ gain_ = std::max(gain_, 1.f);
+
+ return gain_;
+}
+
+} // namespace webrtc

Powered by Google App Engine
This is Rietveld 408576698