dc872b6be1
Bug: webrtc:9193,chromium:836790 Change-Id: I589082a18a4a5d1ba5abc170b6cf49d1f545b6cc Reviewed-on: https://webrtc-review.googlesource.com/72480 Reviewed-by: Per Åhgren <peah@webrtc.org> Commit-Queue: Per Åhgren <peah@webrtc.org> Cr-Commit-Position: refs/heads/master@{#23027}
83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) 2018 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 "modules/audio_processing/aec3/echo_audibility.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "modules/audio_processing/aec3/aec3_common.h"
|
|
#include "modules/audio_processing/aec3/stationarity_estimator.h"
|
|
|
|
namespace webrtc {
|
|
|
|
EchoAudibility::EchoAudibility() {
|
|
Reset();
|
|
}
|
|
|
|
void EchoAudibility::Reset() {
|
|
render_stationarity_.Reset();
|
|
}
|
|
|
|
EchoAudibility::~EchoAudibility() = default;
|
|
|
|
void EchoAudibility::UpdateAfterStableDelay(const RenderBuffer& render_buffer,
|
|
size_t delay_blocks,
|
|
size_t capture_block_counter) {
|
|
RTC_DCHECK_GT(capture_block_counter, delay_blocks);
|
|
|
|
size_t num_lookahead = std::min(StationarityEstimator::GetMaxNumLookAhead(),
|
|
render_buffer.Headroom() - delay_blocks + 1);
|
|
int render_block_number = capture_block_counter - delay_blocks;
|
|
|
|
for (size_t k = 0; k < (num_lookahead + 1); ++k) {
|
|
// Delay changes can potentially make that not all the farend blocks
|
|
// are seen. That effect is assumed to have a minimum effect in the
|
|
// estimation.
|
|
render_stationarity_.Update(render_buffer.Spectrum(delay_blocks - k),
|
|
render_block_number + k);
|
|
}
|
|
render_stationarity_.UpdateStationarityFlags(render_block_number,
|
|
num_lookahead);
|
|
}
|
|
|
|
void EchoAudibility::UpdateBeforeStableDelay(
|
|
const RenderBuffer& render_buffer) {
|
|
// If the delay is not set, the read position in the buffer cannot be trust
|
|
// and the write position in the render buffer should be used instead
|
|
|
|
if (first_update_) {
|
|
render_write_prev_ = render_buffer.GetWritePositionSpectrum();
|
|
first_update_ = false;
|
|
return;
|
|
}
|
|
int render_write_current = render_buffer.GetWritePositionSpectrum();
|
|
|
|
for (int idx = render_write_prev_; idx != render_write_current;
|
|
idx = render_buffer.DecIdx(idx)) {
|
|
render_stationarity_.UpdateNoiseEstimator(
|
|
render_buffer.SpectrumFromPosition(idx));
|
|
}
|
|
|
|
render_write_prev_ = render_write_current;
|
|
}
|
|
|
|
void EchoAudibility::Update(const RenderBuffer& render_buffer,
|
|
size_t delay_blocks,
|
|
size_t capture_block_counter,
|
|
bool external_delay_seen) {
|
|
if (external_delay_seen) {
|
|
UpdateAfterStableDelay(render_buffer, delay_blocks, capture_block_counter);
|
|
} else {
|
|
UpdateBeforeStableDelay(render_buffer);
|
|
}
|
|
}
|
|
|
|
} // namespace webrtc
|