summaryrefslogtreecommitdiff
path: root/webrtc/modules/audio_processing/agc2/noise_level_estimator.h
diff options
context:
space:
mode:
Diffstat (limited to 'webrtc/modules/audio_processing/agc2/noise_level_estimator.h')
-rw-r--r--webrtc/modules/audio_processing/agc2/noise_level_estimator.h43
1 files changed, 43 insertions, 0 deletions
diff --git a/webrtc/modules/audio_processing/agc2/noise_level_estimator.h b/webrtc/modules/audio_processing/agc2/noise_level_estimator.h
new file mode 100644
index 0000000..ca2f9f2
--- /dev/null
+++ b/webrtc/modules/audio_processing/agc2/noise_level_estimator.h
@@ -0,0 +1,43 @@
+/*
+ * 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.
+ */
+
+#ifndef MODULES_AUDIO_PROCESSING_AGC2_NOISE_LEVEL_ESTIMATOR_H_
+#define MODULES_AUDIO_PROCESSING_AGC2_NOISE_LEVEL_ESTIMATOR_H_
+
+#include "modules/audio_processing/agc2/signal_classifier.h"
+#include "modules/audio_processing/include/audio_frame_view.h"
+#include "rtc_base/constructor_magic.h"
+
+namespace webrtc {
+class ApmDataDumper;
+
+class NoiseLevelEstimator {
+ public:
+ NoiseLevelEstimator(ApmDataDumper* data_dumper);
+ ~NoiseLevelEstimator();
+ // Returns the estimated noise level in dBFS.
+ float Analyze(const AudioFrameView<const float>& frame);
+
+ private:
+ void Initialize(int sample_rate_hz);
+
+ int sample_rate_hz_;
+ float min_noise_energy_;
+ bool first_update_;
+ float noise_energy_;
+ int noise_energy_hold_counter_;
+ SignalClassifier signal_classifier_;
+
+ RTC_DISALLOW_COPY_AND_ASSIGN(NoiseLevelEstimator);
+};
+
+} // namespace webrtc
+
+#endif // MODULES_AUDIO_PROCESSING_AGC2_NOISE_LEVEL_ESTIMATOR_H_