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