diff options
Diffstat (limited to 'webrtc/modules/audio_processing/aec3/suppression_filter.h')
-rw-r--r-- | webrtc/modules/audio_processing/aec3/suppression_filter.h | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/webrtc/modules/audio_processing/aec3/suppression_filter.h b/webrtc/modules/audio_processing/aec3/suppression_filter.h new file mode 100644 index 0000000..dcf2292 --- /dev/null +++ b/webrtc/modules/audio_processing/aec3/suppression_filter.h @@ -0,0 +1,48 @@ +/* + * Copyright (c) 2017 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_AEC3_SUPPRESSION_FILTER_H_ +#define MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_FILTER_H_ + +#include <array> +#include <vector> + +#include "modules/audio_processing/aec3/aec3_common.h" +#include "modules/audio_processing/aec3/aec3_fft.h" +#include "modules/audio_processing/aec3/fft_data.h" +#include "rtc_base/constructor_magic.h" + +namespace webrtc { + +class SuppressionFilter { + public: + SuppressionFilter(Aec3Optimization optimization, + int sample_rate_hz, + size_t num_capture_channels_); + ~SuppressionFilter(); + void ApplyGain(rtc::ArrayView<const FftData> comfort_noise, + rtc::ArrayView<const FftData> comfort_noise_high_bands, + const std::array<float, kFftLengthBy2Plus1>& suppression_gain, + float high_bands_gain, + rtc::ArrayView<const FftData> E_lowest_band, + std::vector<std::vector<std::vector<float>>>* e); + + private: + const Aec3Optimization optimization_; + const int sample_rate_hz_; + const size_t num_capture_channels_; + const Aec3Fft fft_; + std::vector<std::vector<std::array<float, kFftLengthBy2>>> e_output_old_; + RTC_DISALLOW_COPY_AND_ASSIGN(SuppressionFilter); +}; + +} // namespace webrtc + +#endif // MODULES_AUDIO_PROCESSING_AEC3_SUPPRESSION_FILTER_H_ |