| Index: webrtc/modules/audio_processing/aec3/matched_filter_lag_aggregator.cc
|
| diff --git a/webrtc/modules/audio_processing/aec3/matched_filter_lag_aggregator.cc b/webrtc/modules/audio_processing/aec3/matched_filter_lag_aggregator.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..d9176efa94fbd76f6d2fbbffa121e7ba2f68e54d
|
| --- /dev/null
|
| +++ b/webrtc/modules/audio_processing/aec3/matched_filter_lag_aggregator.cc
|
| @@ -0,0 +1,66 @@
|
| +/*
|
| + * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +#include "webrtc/modules/audio_processing/aec3/matched_filter_lag_aggregator.h"
|
| +
|
| +#include "webrtc/modules/audio_processing/logging/apm_data_dumper.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +MatchedFilterLagAggregator::MatchedFilterLagAggregator(
|
| + ApmDataDumper* data_dumper,
|
| + size_t num_lag_estimates)
|
| + : data_dumper_(data_dumper), lag_updates_in_a_row_(num_lag_estimates, 0) {
|
| + RTC_DCHECK(data_dumper);
|
| + RTC_DCHECK_LT(0, num_lag_estimates);
|
| +}
|
| +
|
| +MatchedFilterLagAggregator::~MatchedFilterLagAggregator() = default;
|
| +
|
| +rtc::Optional<size_t> MatchedFilterLagAggregator::Aggregate(
|
| + rtc::ArrayView<const MatchedFilter::LagEstimate> lag_estimates) {
|
| + RTC_DCHECK_EQ(lag_updates_in_a_row_.size(), lag_estimates.size());
|
| +
|
| + // Count the number of lag updates in a row to ensure that only stable lags
|
| + // are taken into account.
|
| + for (size_t k = 0; k < lag_estimates.size(); ++k) {
|
| + lag_updates_in_a_row_[k] =
|
| + lag_estimates[k].updated ? lag_updates_in_a_row_[k] + 1 : 0;
|
| + }
|
| +
|
| + // If available, choose the strongest lag estimate as the best one.
|
| + int best_lag_estimate_index = -1;
|
| + for (size_t k = 0; k < lag_estimates.size(); ++k) {
|
| + if (lag_updates_in_a_row_[k] > 10 && lag_estimates[k].reliable &&
|
| + (best_lag_estimate_index == -1 ||
|
| + lag_estimates[k].accuracy >
|
| + lag_estimates[best_lag_estimate_index].accuracy)) {
|
| + best_lag_estimate_index = k;
|
| + }
|
| + }
|
| +
|
| + // TODO(peah): Remove this logging once all development is done.
|
| + data_dumper_->DumpRaw("aec3_echo_path_delay_estimator_best_index",
|
| + best_lag_estimate_index);
|
| +
|
| + // Require the same lag to be detected 10 times in a row before considering
|
| + // it reliable.
|
| + if (best_lag_estimate_index >= 0) {
|
| + candidate_counter_ =
|
| + (candidate_ == lag_estimates[best_lag_estimate_index].lag)
|
| + ? candidate_counter_ + 1
|
| + : 0;
|
| + candidate_ = lag_estimates[best_lag_estimate_index].lag;
|
| + }
|
| +
|
| + return candidate_counter_ >= 10 ? rtc::Optional<size_t>(candidate_)
|
| + : rtc::Optional<size_t>();
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|