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 #include "webrtc/modules/audio_processing/aec3/echo_path_delay_estimator.h" | 10 #include "webrtc/modules/audio_processing/aec3/echo_path_delay_estimator.h" |
11 | 11 |
12 #include <algorithm> | 12 #include <algorithm> |
13 #include <array> | 13 #include <array> |
14 | 14 |
15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
16 #include "webrtc/modules/audio_processing/aec3/aec3_common.h" | 16 #include "webrtc/modules/audio_processing/aec3/aec3_common.h" |
17 #include "webrtc/modules/audio_processing/include/audio_processing.h" | 17 #include "webrtc/modules/audio_processing/include/audio_processing.h" |
18 #include "webrtc/modules/audio_processing/logging/apm_data_dumper.h" | 18 #include "webrtc/modules/audio_processing/logging/apm_data_dumper.h" |
19 | 19 |
20 namespace webrtc { | 20 namespace webrtc { |
21 | 21 |
22 namespace { | 22 namespace { |
23 | 23 |
24 constexpr size_t kNumMatchedFilters = 4; | |
25 constexpr size_t kMatchedFilterWindowSizeSubBlocks = 32; | |
peah-webrtc
2017/03/30 05:36:56
This is now moved to aec3_common.h
| |
26 constexpr size_t kMatchedFilterAlignmentShiftSizeSubBlocks = | |
27 kMatchedFilterWindowSizeSubBlocks * 3 / 4; | |
28 | |
29 constexpr int kDownSamplingFactor = 4; | 24 constexpr int kDownSamplingFactor = 4; |
30 } // namespace | 25 } // namespace |
31 | 26 |
32 EchoPathDelayEstimator::EchoPathDelayEstimator(ApmDataDumper* data_dumper) | 27 EchoPathDelayEstimator::EchoPathDelayEstimator(ApmDataDumper* data_dumper) |
33 : data_dumper_(data_dumper), | 28 : data_dumper_(data_dumper), |
34 matched_filter_(data_dumper_, | 29 matched_filter_(data_dumper_, |
35 DetectOptimization(), | 30 DetectOptimization(), |
36 kMatchedFilterWindowSizeSubBlocks, | 31 kMatchedFilterWindowSizeSubBlocks, |
37 kNumMatchedFilters, | 32 kNumMatchedFilters, |
38 kMatchedFilterAlignmentShiftSizeSubBlocks), | 33 kMatchedFilterAlignmentShiftSizeSubBlocks), |
39 matched_filter_lag_aggregator_(data_dumper_, | 34 matched_filter_lag_aggregator_(data_dumper_, |
40 matched_filter_.NumLagEstimates()) { | 35 matched_filter_.NumLagEstimates()) { |
41 RTC_DCHECK(data_dumper); | 36 RTC_DCHECK(data_dumper); |
42 } | 37 } |
43 | 38 |
44 EchoPathDelayEstimator::~EchoPathDelayEstimator() = default; | 39 EchoPathDelayEstimator::~EchoPathDelayEstimator() = default; |
45 | 40 |
41 void EchoPathDelayEstimator::Reset() { | |
42 matched_filter_lag_aggregator_.Reset(); | |
43 matched_filter_.Reset(); | |
44 } | |
45 | |
46 rtc::Optional<size_t> EchoPathDelayEstimator::EstimateDelay( | 46 rtc::Optional<size_t> EchoPathDelayEstimator::EstimateDelay( |
47 rtc::ArrayView<const float> render, | 47 const DownsampledRenderBuffer& render_buffer, |
48 rtc::ArrayView<const float> capture) { | 48 rtc::ArrayView<const float> capture) { |
49 RTC_DCHECK_EQ(kBlockSize, capture.size()); | 49 RTC_DCHECK_EQ(kBlockSize, capture.size()); |
50 RTC_DCHECK_EQ(render.size(), capture.size()); | |
51 | 50 |
52 std::array<float, kSubBlockSize> downsampled_render; | |
peah-webrtc
2017/03/30 05:36:56
The downsampling of the render is now done in the
| |
53 std::array<float, kSubBlockSize> downsampled_capture; | 51 std::array<float, kSubBlockSize> downsampled_capture; |
54 | 52 capture_decimator_.Decimate(capture, downsampled_capture); |
55 render_decimator_.Decimate(render, &downsampled_render); | 53 matched_filter_.Update(render_buffer, downsampled_capture); |
56 capture_decimator_.Decimate(capture, &downsampled_capture); | |
57 | |
58 matched_filter_.Update(downsampled_render, downsampled_capture); | |
59 | 54 |
60 rtc::Optional<size_t> aggregated_matched_filter_lag = | 55 rtc::Optional<size_t> aggregated_matched_filter_lag = |
61 matched_filter_lag_aggregator_.Aggregate( | 56 matched_filter_lag_aggregator_.Aggregate( |
62 matched_filter_.GetLagEstimates()); | 57 matched_filter_.GetLagEstimates()); |
63 | 58 |
64 // TODO(peah): Move this logging outside of this class once EchoCanceller3 | 59 // TODO(peah): Move this logging outside of this class once EchoCanceller3 |
65 // development is done. | 60 // development is done. |
66 data_dumper_->DumpRaw("aec3_echo_path_delay_estimator_delay", | 61 data_dumper_->DumpRaw("aec3_echo_path_delay_estimator_delay", |
67 aggregated_matched_filter_lag | 62 aggregated_matched_filter_lag |
68 ? static_cast<int>(*aggregated_matched_filter_lag * | 63 ? static_cast<int>(*aggregated_matched_filter_lag * |
69 kDownSamplingFactor) | 64 kDownSamplingFactor) |
70 : -1); | 65 : -1); |
71 | 66 |
72 // Return the detected delay in samples as the aggregated matched filter lag | 67 // Return the detected delay in samples as the aggregated matched filter lag |
73 // compensated by the down sampling factor for the signal being correlated. | 68 // compensated by the down sampling factor for the signal being correlated. |
74 return aggregated_matched_filter_lag | 69 return aggregated_matched_filter_lag |
75 ? rtc::Optional<size_t>(*aggregated_matched_filter_lag * | 70 ? rtc::Optional<size_t>(*aggregated_matched_filter_lag * |
76 kDownSamplingFactor) | 71 kDownSamplingFactor) |
77 : rtc::Optional<size_t>(); | 72 : rtc::Optional<size_t>(); |
78 } | 73 } |
79 | 74 |
80 } // namespace webrtc | 75 } // namespace webrtc |
OLD | NEW |