Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(484)

Side by Side Diff: webrtc/modules/audio_processing/aec3/render_signal_analyzer.cc

Issue 2784023002: Major AEC3 render pipeline changes (Closed)
Patch Set: Disabled one more DEATH test that caused issues due to bug on bots Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698