blob: 03c4d2d8b704d09391e841ae2cbfa99dcfdf1cb7 [file] [log] [blame]
peah522d71b2017-02-23 05:16:26 -08001/*
2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#ifndef MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_
12#define MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_
peah522d71b2017-02-23 05:16:26 -080013
14#include <array>
peah86afe9d2017-04-06 15:45:32 -070015#include <vector>
peah522d71b2017-02-23 05:16:26 -080016
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020017#include "modules/audio_processing/aec3/aec3_common.h"
18#include "modules/audio_processing/aec3/render_signal_analyzer.h"
19#include "modules/audio_processing/include/audio_processing.h"
20#include "rtc_base/constructormagic.h"
peah522d71b2017-02-23 05:16:26 -080021
22namespace webrtc {
peah522d71b2017-02-23 05:16:26 -080023
24class SuppressionGain {
25 public:
peah8cee56f2017-08-24 22:36:53 -070026 SuppressionGain(const AudioProcessing::Config::EchoCanceller3& config,
27 Aec3Optimization optimization);
peah1d680892017-05-23 04:07:10 -070028 void GetGain(const std::array<float, kFftLengthBy2Plus1>& nearend,
29 const std::array<float, kFftLengthBy2Plus1>& echo,
30 const std::array<float, kFftLengthBy2Plus1>& comfort_noise,
peah14c11a42017-07-11 06:13:43 -070031 const RenderSignalAnalyzer& render_signal_analyzer,
peah86afe9d2017-04-06 15:45:32 -070032 bool saturated_echo,
33 const std::vector<std::vector<float>>& render,
peah6d822ad2017-04-10 13:52:14 -070034 bool force_zero_gain,
Per Åhgrenc65ce782017-10-09 13:01:39 +020035 bool linear_echo_estimate,
peah86afe9d2017-04-06 15:45:32 -070036 float* high_bands_gain,
37 std::array<float, kFftLengthBy2Plus1>* low_band_gain);
peah522d71b2017-02-23 05:16:26 -080038
39 private:
peah1d680892017-05-23 04:07:10 -070040 void LowerBandGain(bool stationary_with_low_power,
peah14c11a42017-07-11 06:13:43 -070041 const rtc::Optional<int>& narrow_peak_band,
peah1d680892017-05-23 04:07:10 -070042 bool saturated_echo,
Per Åhgrenc65ce782017-10-09 13:01:39 +020043 bool linear_echo_estimate,
peah1d680892017-05-23 04:07:10 -070044 const std::array<float, kFftLengthBy2Plus1>& nearend,
45 const std::array<float, kFftLengthBy2Plus1>& echo,
46 const std::array<float, kFftLengthBy2Plus1>& comfort_noise,
47 std::array<float, kFftLengthBy2Plus1>* gain);
48
49 class LowNoiseRenderDetector {
50 public:
51 bool Detect(const std::vector<std::vector<float>>& render);
52
53 private:
54 float average_power_ = 32768.f * 32768.f;
55 };
56
peah522d71b2017-02-23 05:16:26 -080057 const Aec3Optimization optimization_;
peah1d680892017-05-23 04:07:10 -070058 std::array<float, kFftLengthBy2Plus1> last_gain_;
59 std::array<float, kFftLengthBy2Plus1> last_masker_;
60 std::array<float, kFftLengthBy2Plus1> gain_increase_;
61 std::array<float, kFftLengthBy2Plus1> last_echo_;
62
63 LowNoiseRenderDetector low_render_detector_;
64 size_t no_saturation_counter_ = 0;
peah8cee56f2017-08-24 22:36:53 -070065 const AudioProcessing::Config::EchoCanceller3 config_;
66 RTC_DISALLOW_COPY_AND_ASSIGN(SuppressionGain);
peah522d71b2017-02-23 05:16:26 -080067};
68
69} // namespace webrtc
70
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020071#endif // MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_GAIN_H_