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

Side by Side Diff: webrtc/modules/audio_processing/aec3/suppression_gain.h

Issue 2980493002: Add adaptive notch filter to remove narrowband echo components in AEC3 (Closed)
Patch Set: Created 3 years, 5 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 unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2017 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 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_ 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_
12 #define WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_ 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_
13 13
14 #include <array> 14 #include <array>
15 #include <vector> 15 #include <vector>
16 16
17 #include "webrtc/modules/audio_processing/aec3/aec3_common.h" 17 #include "webrtc/modules/audio_processing/aec3/aec3_common.h"
18 #include "webrtc/modules/audio_processing/aec3/render_signal_analyzer.h"
18 #include "webrtc/rtc_base/constructormagic.h" 19 #include "webrtc/rtc_base/constructormagic.h"
19 20
20 namespace webrtc { 21 namespace webrtc {
21 22
22 class SuppressionGain { 23 class SuppressionGain {
23 public: 24 public:
24 explicit SuppressionGain(Aec3Optimization optimization); 25 explicit SuppressionGain(Aec3Optimization optimization);
25 void GetGain(const std::array<float, kFftLengthBy2Plus1>& nearend, 26 void GetGain(const std::array<float, kFftLengthBy2Plus1>& nearend,
26 const std::array<float, kFftLengthBy2Plus1>& echo, 27 const std::array<float, kFftLengthBy2Plus1>& echo,
27 const std::array<float, kFftLengthBy2Plus1>& comfort_noise, 28 const std::array<float, kFftLengthBy2Plus1>& comfort_noise,
29 const RenderSignalAnalyzer& render_signal_analyzer,
28 bool saturated_echo, 30 bool saturated_echo,
29 const std::vector<std::vector<float>>& render, 31 const std::vector<std::vector<float>>& render,
30 bool force_zero_gain, 32 bool force_zero_gain,
31 float* high_bands_gain, 33 float* high_bands_gain,
32 std::array<float, kFftLengthBy2Plus1>* low_band_gain); 34 std::array<float, kFftLengthBy2Plus1>* low_band_gain);
33 35
34 private: 36 private:
35 void LowerBandGain(bool stationary_with_low_power, 37 void LowerBandGain(bool stationary_with_low_power,
38 const RenderSignalAnalyzer& render_signal_analyzer,
36 bool saturated_echo, 39 bool saturated_echo,
37 const std::array<float, kFftLengthBy2Plus1>& nearend, 40 const std::array<float, kFftLengthBy2Plus1>& nearend,
38 const std::array<float, kFftLengthBy2Plus1>& echo, 41 const std::array<float, kFftLengthBy2Plus1>& echo,
39 const std::array<float, kFftLengthBy2Plus1>& comfort_noise, 42 const std::array<float, kFftLengthBy2Plus1>& comfort_noise,
40 std::array<float, kFftLengthBy2Plus1>* gain); 43 std::array<float, kFftLengthBy2Plus1>* gain);
41 44
42 class LowNoiseRenderDetector { 45 class LowNoiseRenderDetector {
43 public: 46 public:
44 bool Detect(const std::vector<std::vector<float>>& render); 47 bool Detect(const std::vector<std::vector<float>>& render);
45 48
46 private: 49 private:
47 float average_power_ = 32768.f * 32768.f; 50 float average_power_ = 32768.f * 32768.f;
48 }; 51 };
49 52
50 const Aec3Optimization optimization_; 53 const Aec3Optimization optimization_;
51 std::array<float, kFftLengthBy2Plus1> last_gain_; 54 std::array<float, kFftLengthBy2Plus1> last_gain_;
52 std::array<float, kFftLengthBy2Plus1> last_masker_; 55 std::array<float, kFftLengthBy2Plus1> last_masker_;
53 std::array<float, kFftLengthBy2Plus1> gain_increase_; 56 std::array<float, kFftLengthBy2Plus1> gain_increase_;
54 std::array<float, kFftLengthBy2Plus1> last_echo_; 57 std::array<float, kFftLengthBy2Plus1> last_echo_;
55 58
56 LowNoiseRenderDetector low_render_detector_; 59 LowNoiseRenderDetector low_render_detector_;
57 size_t no_saturation_counter_ = 0; 60 size_t no_saturation_counter_ = 0;
58 RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(SuppressionGain); 61 RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(SuppressionGain);
59 }; 62 };
60 63
61 } // namespace webrtc 64 } // namespace webrtc
62 65
63 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_ 66 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698