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 
11 #ifndef MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_
12 #define MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_
13 
14 #include <array>
15 #include <memory>
16 
17 #include "api/optional.h"
18 #include "modules/audio_processing/aec3/aec3_common.h"
19 #include "modules/audio_processing/aec3/render_buffer.h"
20 #include "rtc_base/constructormagic.h"
21 
22 namespace webrtc {
23 
24 // Provides functionality for analyzing the properties of the render signal.
25 class RenderSignalAnalyzer {
26  public:
27   RenderSignalAnalyzer();
28   ~RenderSignalAnalyzer();
29 
30   // Updates the render signal analysis with the most recent render signal.
31   void Update(const RenderBuffer& render_buffer,
32               const rtc::Optional<size_t>& delay_partitions);
33 
34   // Returns true if the render signal is poorly exciting.
PoorSignalExcitation()35   bool PoorSignalExcitation() const {
36     RTC_DCHECK_LT(2, narrow_band_counters_.size());
37     return std::any_of(narrow_band_counters_.begin(),
38                        narrow_band_counters_.end(),
39                        [](size_t a) { return a > 10; });
40   }
41 
42   // Zeros the array around regions with narrow bands signal characteristics.
43   void MaskRegionsAroundNarrowBands(
44       std::array<float, kFftLengthBy2Plus1>* v) const;
45 
NarrowPeakBand()46   rtc::Optional<int> NarrowPeakBand() const { return narrow_peak_band_; }
47 
48  private:
49   std::array<size_t, kFftLengthBy2 - 1> narrow_band_counters_;
50   rtc::Optional<int> narrow_peak_band_;
51   size_t narrow_peak_counter_;
52 
53   RTC_DISALLOW_COPY_AND_ASSIGN(RenderSignalAnalyzer);
54 };
55 
56 }  // namespace webrtc
57 
58 #endif  // MODULES_AUDIO_PROCESSING_AEC3_RENDER_SIGNAL_ANALYZER_H_
59