OLD | NEW |
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 #include "webrtc/modules/audio_processing/aec3/render_signal_analyzer.h" | 11 #include "webrtc/modules/audio_processing/aec3/render_signal_analyzer.h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 | 14 |
15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
16 | 16 |
17 namespace webrtc { | 17 namespace webrtc { |
18 | 18 |
19 namespace { | 19 namespace { |
20 constexpr size_t kCounterThreshold = 5; | 20 constexpr size_t kCounterThreshold = 5; |
21 | 21 |
22 } // namespace | 22 } // namespace |
23 | 23 |
24 RenderSignalAnalyzer::RenderSignalAnalyzer() { | 24 RenderSignalAnalyzer::RenderSignalAnalyzer() { |
25 narrow_band_counters_.fill(0); | 25 narrow_band_counters_.fill(0); |
26 } | 26 } |
27 RenderSignalAnalyzer::~RenderSignalAnalyzer() = default; | 27 RenderSignalAnalyzer::~RenderSignalAnalyzer() = default; |
28 | 28 |
29 void RenderSignalAnalyzer::Update( | 29 void RenderSignalAnalyzer::Update( |
30 const FftBuffer& X_buffer, | 30 const RenderBuffer& render_buffer, |
31 const rtc::Optional<size_t>& delay_partitions) { | 31 const rtc::Optional<size_t>& delay_partitions) { |
32 if (!delay_partitions) { | 32 if (!delay_partitions) { |
33 narrow_band_counters_.fill(0); | 33 narrow_band_counters_.fill(0); |
34 return; | 34 return; |
35 } | 35 } |
36 | 36 |
37 const std::array<float, kFftLengthBy2Plus1>& X2 = | 37 const std::array<float, kFftLengthBy2Plus1>& X2 = |
38 X_buffer.Spectrum(*delay_partitions); | 38 render_buffer.Spectrum(*delay_partitions); |
39 | 39 |
40 // Detect narrow band signal regions. | 40 // Detect narrow band signal regions. |
41 for (size_t k = 1; k < (X2.size() - 1); ++k) { | 41 for (size_t k = 1; k < (X2.size() - 1); ++k) { |
42 narrow_band_counters_[k - 1] = X2[k] > 3 * std::max(X2[k - 1], X2[k + 1]) | 42 narrow_band_counters_[k - 1] = X2[k] > 3 * std::max(X2[k - 1], X2[k + 1]) |
43 ? narrow_band_counters_[k - 1] + 1 | 43 ? narrow_band_counters_[k - 1] + 1 |
44 : 0; | 44 : 0; |
45 } | 45 } |
46 } | 46 } |
47 | 47 |
48 void RenderSignalAnalyzer::MaskRegionsAroundNarrowBands( | 48 void RenderSignalAnalyzer::MaskRegionsAroundNarrowBands( |
49 std::array<float, kFftLengthBy2Plus1>* v) const { | 49 std::array<float, kFftLengthBy2Plus1>* v) const { |
50 RTC_DCHECK(v); | 50 RTC_DCHECK(v); |
51 | 51 |
52 // Set v to zero around narrow band signal regions. | 52 // Set v to zero around narrow band signal regions. |
53 if (narrow_band_counters_[0] > kCounterThreshold) { | 53 if (narrow_band_counters_[0] > kCounterThreshold) { |
54 (*v)[1] = (*v)[0] = 0.f; | 54 (*v)[1] = (*v)[0] = 0.f; |
55 } | 55 } |
56 for (size_t k = 2; k < kFftLengthBy2 - 1; ++k) { | 56 for (size_t k = 2; k < kFftLengthBy2 - 1; ++k) { |
57 if (narrow_band_counters_[k - 1] > kCounterThreshold) { | 57 if (narrow_band_counters_[k - 1] > kCounterThreshold) { |
58 (*v)[k - 2] = (*v)[k - 1] = (*v)[k] = (*v)[k + 1] = (*v)[k + 2] = 0.f; | 58 (*v)[k - 2] = (*v)[k - 1] = (*v)[k] = (*v)[k + 1] = (*v)[k + 2] = 0.f; |
59 } | 59 } |
60 } | 60 } |
61 if (narrow_band_counters_[kFftLengthBy2 - 2] > kCounterThreshold) { | 61 if (narrow_band_counters_[kFftLengthBy2 - 2] > kCounterThreshold) { |
62 (*v)[kFftLengthBy2] = (*v)[kFftLengthBy2 - 1] = 0.f; | 62 (*v)[kFftLengthBy2] = (*v)[kFftLengthBy2 - 1] = 0.f; |
63 } | 63 } |
64 } | 64 } |
65 | 65 |
66 } // namespace webrtc | 66 } // namespace webrtc |
OLD | NEW |