New module for the adaptive level controlling functionality in the audio processing module

webrtc/modules/audio_processing/level_controller/noise_level_estimator.cc

Index: webrtc/modules/audio_processing/level_controller/noise_level_estimator.cc
diff --git a/webrtc/modules/audio_processing/level_controller/noise_level_estimator.cc b/webrtc/modules/audio_processing/level_controller/noise_level_estimator.cc
new file mode 100644
index 0000000000000000000000000000000000000000..ced192a9534c3f484f1ddc9550dab09cfaa082ff
--- /dev/null
+++ b/webrtc/modules/audio_processing/level_controller/noise_level_estimator.cc
@@ -0,0 +1,74 @@
+/*
+ *  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/noise_level_estimator.h"
+
+#include <algorithm>
+
+#include "webrtc/modules/audio_processing/audio_buffer.h"
+#include "webrtc/modules/audio_processing/logging/apm_data_dumper.h"
+
+namespace webrtc {
+
+NoiseLevelEstimator::NoiseLevelEstimator() {
+  Initialize(AudioProcessing::kSampleRate48kHz);
+}
+
+NoiseLevelEstimator::~NoiseLevelEstimator() {}
+
+void NoiseLevelEstimator::Initialize(int sample_rate_hz) {
+  noise_energy_ = 1.f;
+  first_update_ = true;
+  min_noise_energy_ = sample_rate_hz * 2.f * 2.f / 100.f;
+  noise_energy_hold_counter_ = 0;
+}
+
+float NoiseLevelEstimator::Analyze(SignalClassifier::SignalType signal_type,
+                                   float frame_energy) {
+  if (frame_energy > 0) {

[hlundin-webrtc, 2016/06/27 11:21:16]

You can save yourself one level of indentation on the rest of the function by
simply returning early when the energy is <= 0
if (frame_energy <= 0) {
  return noise_energy_;
}

[peah-webrtc, 2016/06/27 22:51:49]

Good point!

Done.

+    if (first_update_) {
+      // Initialize the noise energy to the frame energy.
+      noise_energy_ = frame_energy;

[hlundin-webrtc, 2016/06/27 11:21:16]

first_update_ = false;
return noise_energy_ = std::max(frame_energy, min_noise_energy_);

[peah-webrtc, 2016/06/27 22:51:49]

Nice!

Done.

+    } else {
+      // Update the noise estimate in a minimum statistics-type manner.
+      if (signal_type == SignalClassifier::SignalType::kStationary) {
+        if (frame_energy > noise_energy_) {
+          // Leak the estimate upwards towards the frame energy if no recent
+          // downward update.
+          (noise_energy_hold_counter_) =

[hlundin-webrtc, 2016/06/27 11:21:16]

Why the parentheses around the lhs?

[peah-webrtc, 2016/06/27 22:51:49]

That is a leftover from a previous code construct. Removed it.

Done.

+              std::max(noise_energy_hold_counter_ - 1, 0);
+
+          if (noise_energy_hold_counter_ == 0) {
+            noise_energy_ = std::min(noise_energy_ * 1.01f, frame_energy);
+          }
+        } else {
+          // Update smothly downwards with a limited maximum update magnitude.

[hlundin-webrtc, 2016/06/27 11:21:16]

smothly -> smoothly

[peah-webrtc, 2016/06/27 22:51:49]

Done.

+          noise_energy_ =
+              std::max(noise_energy_ * 0.9f,
+                       noise_energy_ + 0.05f * (frame_energy - noise_energy_));
+          noise_energy_hold_counter_ = 1000;
+        }
+      } else {
+        // For a non-stationary signal, leak the estimate downwards in order to
+        // avoid estimate locking due to incorrect signal classification.
+        noise_energy_ = noise_energy_ * 0.99f;
+      }
+    }
+
+    // Ensure a minimum of the estimate.
+    noise_energy_ = std::max(noise_energy_, min_noise_energy_);
+
+    first_update_ = false;

[hlundin-webrtc, 2016/06/27 11:21:16]

Delete.

[peah-webrtc, 2016/06/27 22:51:49]

Done.

+  }
+
+  return noise_energy_;
+}
+
+}  // namespace webrtc