peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 1 | /* |
| 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 Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 11 | #ifndef MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_ |
| 12 | #define MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_ |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 13 | |
Yves Gerey | 988cc08 | 2018-10-23 12:03:01 +0200 | [diff] [blame] | 14 | #include <algorithm> |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 15 | #include <array> |
Yves Gerey | 988cc08 | 2018-10-23 12:03:01 +0200 | [diff] [blame] | 16 | #include <cstddef> |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 17 | |
Danil Chapovalov | db9f7ab | 2018-06-19 10:50:11 +0200 | [diff] [blame] | 18 | #include "absl/types/optional.h" |
Per Åhgren | 971de07 | 2018-03-14 23:23:47 +0100 | [diff] [blame] | 19 | #include "api/audio/echo_canceller3_config.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 20 | #include "modules/audio_processing/aec3/aec3_common.h" |
| 21 | #include "modules/audio_processing/aec3/render_buffer.h" |
Yves Gerey | 988cc08 | 2018-10-23 12:03:01 +0200 | [diff] [blame] | 22 | #include "rtc_base/checks.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame^] | 23 | #include "rtc_base/constructor_magic.h" |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 24 | |
| 25 | namespace webrtc { |
| 26 | |
| 27 | // Provides functionality for analyzing the properties of the render signal. |
| 28 | class RenderSignalAnalyzer { |
| 29 | public: |
Per Åhgren | 971de07 | 2018-03-14 23:23:47 +0100 | [diff] [blame] | 30 | explicit RenderSignalAnalyzer(const EchoCanceller3Config& config); |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 31 | ~RenderSignalAnalyzer(); |
| 32 | |
| 33 | // Updates the render signal analysis with the most recent render signal. |
peah | 86afe9d | 2017-04-06 15:45:32 -0700 | [diff] [blame] | 34 | void Update(const RenderBuffer& render_buffer, |
Danil Chapovalov | db9f7ab | 2018-06-19 10:50:11 +0200 | [diff] [blame] | 35 | const absl::optional<size_t>& delay_partitions); |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 36 | |
| 37 | // Returns true if the render signal is poorly exciting. |
| 38 | bool PoorSignalExcitation() const { |
| 39 | RTC_DCHECK_LT(2, narrow_band_counters_.size()); |
| 40 | return std::any_of(narrow_band_counters_.begin(), |
| 41 | narrow_band_counters_.end(), |
| 42 | [](size_t a) { return a > 10; }); |
| 43 | } |
| 44 | |
| 45 | // Zeros the array around regions with narrow bands signal characteristics. |
| 46 | void MaskRegionsAroundNarrowBands( |
| 47 | std::array<float, kFftLengthBy2Plus1>* v) const; |
| 48 | |
Danil Chapovalov | db9f7ab | 2018-06-19 10:50:11 +0200 | [diff] [blame] | 49 | absl::optional<int> NarrowPeakBand() const { return narrow_peak_band_; } |
peah | 14c11a4 | 2017-07-11 06:13:43 -0700 | [diff] [blame] | 50 | |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 51 | private: |
Per Åhgren | 971de07 | 2018-03-14 23:23:47 +0100 | [diff] [blame] | 52 | const int strong_peak_freeze_duration_; |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 53 | std::array<size_t, kFftLengthBy2 - 1> narrow_band_counters_; |
Danil Chapovalov | db9f7ab | 2018-06-19 10:50:11 +0200 | [diff] [blame] | 54 | absl::optional<int> narrow_peak_band_; |
peah | 14c11a4 | 2017-07-11 06:13:43 -0700 | [diff] [blame] | 55 | size_t narrow_peak_counter_; |
peah | 522d71b | 2017-02-23 05:16:26 -0800 | [diff] [blame] | 56 | |
| 57 | RTC_DISALLOW_COPY_AND_ASSIGN(RenderSignalAnalyzer); |
| 58 | }; |
| 59 | |
| 60 | } // namespace webrtc |
| 61 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 62 | #endif // MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_ |