855 lines
36 KiB
C++
855 lines
36 KiB
C++
/*
|
|
* Copyright (c) 2016 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_canceller3.h"
|
|
|
|
#include <algorithm>
|
|
#include <utility>
|
|
|
|
#include "modules/audio_processing/aec3/aec3_common.h"
|
|
#include "modules/audio_processing/high_pass_filter.h"
|
|
#include "modules/audio_processing/logging/apm_data_dumper.h"
|
|
#include "rtc_base/atomic_ops.h"
|
|
#include "rtc_base/experiments/field_trial_parser.h"
|
|
#include "rtc_base/logging.h"
|
|
#include "system_wrappers/include/field_trial.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
enum class EchoCanceller3ApiCall { kCapture, kRender };
|
|
|
|
bool DetectSaturation(rtc::ArrayView<const float> y) {
|
|
for (auto y_k : y) {
|
|
if (y_k >= 32700.0f || y_k <= -32700.0f) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// Retrieves a value from a field trial if it is available. If no value is
|
|
// present, the default value is returned. If the retrieved value is beyond the
|
|
// specified limits, the default value is returned instead.
|
|
void RetrieveFieldTrialValue(const char* trial_name,
|
|
float min,
|
|
float max,
|
|
float* value_to_update) {
|
|
const std::string field_trial_str = field_trial::FindFullName(trial_name);
|
|
|
|
FieldTrialParameter<double> field_trial_param(/*key=*/"", *value_to_update);
|
|
|
|
ParseFieldTrial({&field_trial_param}, field_trial_str);
|
|
float field_trial_value = static_cast<float>(field_trial_param.Get());
|
|
|
|
if (field_trial_value >= min && field_trial_value <= max) {
|
|
*value_to_update = field_trial_value;
|
|
}
|
|
}
|
|
|
|
void RetrieveFieldTrialValue(const char* trial_name,
|
|
int min,
|
|
int max,
|
|
int* value_to_update) {
|
|
const std::string field_trial_str = field_trial::FindFullName(trial_name);
|
|
|
|
FieldTrialParameter<int> field_trial_param(/*key=*/"", *value_to_update);
|
|
|
|
ParseFieldTrial({&field_trial_param}, field_trial_str);
|
|
float field_trial_value = field_trial_param.Get();
|
|
|
|
if (field_trial_value >= min && field_trial_value <= max) {
|
|
*value_to_update = field_trial_value;
|
|
}
|
|
}
|
|
|
|
void FillSubFrameView(
|
|
AudioBuffer* frame,
|
|
size_t sub_frame_index,
|
|
std::vector<std::vector<rtc::ArrayView<float>>>* sub_frame_view) {
|
|
RTC_DCHECK_GE(1, sub_frame_index);
|
|
RTC_DCHECK_LE(0, sub_frame_index);
|
|
RTC_DCHECK_EQ(frame->num_bands(), sub_frame_view->size());
|
|
RTC_DCHECK_EQ(frame->num_channels(), (*sub_frame_view)[0].size());
|
|
for (size_t band = 0; band < sub_frame_view->size(); ++band) {
|
|
for (size_t channel = 0; channel < (*sub_frame_view)[0].size(); ++channel) {
|
|
(*sub_frame_view)[band][channel] = rtc::ArrayView<float>(
|
|
&frame->split_bands(channel)[band][sub_frame_index * kSubFrameLength],
|
|
kSubFrameLength);
|
|
}
|
|
}
|
|
}
|
|
|
|
void FillSubFrameView(
|
|
std::vector<std::vector<std::vector<float>>>* frame,
|
|
size_t sub_frame_index,
|
|
std::vector<std::vector<rtc::ArrayView<float>>>* sub_frame_view) {
|
|
RTC_DCHECK_GE(1, sub_frame_index);
|
|
RTC_DCHECK_EQ(frame->size(), sub_frame_view->size());
|
|
RTC_DCHECK_EQ((*frame)[0].size(), (*sub_frame_view)[0].size());
|
|
for (size_t band = 0; band < frame->size(); ++band) {
|
|
for (size_t channel = 0; channel < (*frame)[band].size(); ++channel) {
|
|
(*sub_frame_view)[band][channel] = rtc::ArrayView<float>(
|
|
&(*frame)[band][channel][sub_frame_index * kSubFrameLength],
|
|
kSubFrameLength);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ProcessCaptureFrameContent(
|
|
AudioBuffer* linear_output,
|
|
AudioBuffer* capture,
|
|
bool level_change,
|
|
bool saturated_microphone_signal,
|
|
size_t sub_frame_index,
|
|
FrameBlocker* capture_blocker,
|
|
BlockFramer* linear_output_framer,
|
|
BlockFramer* output_framer,
|
|
BlockProcessor* block_processor,
|
|
std::vector<std::vector<std::vector<float>>>* linear_output_block,
|
|
std::vector<std::vector<rtc::ArrayView<float>>>*
|
|
linear_output_sub_frame_view,
|
|
std::vector<std::vector<std::vector<float>>>* capture_block,
|
|
std::vector<std::vector<rtc::ArrayView<float>>>* capture_sub_frame_view) {
|
|
FillSubFrameView(capture, sub_frame_index, capture_sub_frame_view);
|
|
|
|
if (linear_output) {
|
|
RTC_DCHECK(linear_output_framer);
|
|
RTC_DCHECK(linear_output_block);
|
|
RTC_DCHECK(linear_output_sub_frame_view);
|
|
FillSubFrameView(linear_output, sub_frame_index,
|
|
linear_output_sub_frame_view);
|
|
}
|
|
|
|
capture_blocker->InsertSubFrameAndExtractBlock(*capture_sub_frame_view,
|
|
capture_block);
|
|
block_processor->ProcessCapture(level_change, saturated_microphone_signal,
|
|
linear_output_block, capture_block);
|
|
output_framer->InsertBlockAndExtractSubFrame(*capture_block,
|
|
capture_sub_frame_view);
|
|
|
|
if (linear_output) {
|
|
RTC_DCHECK(linear_output_framer);
|
|
linear_output_framer->InsertBlockAndExtractSubFrame(
|
|
*linear_output_block, linear_output_sub_frame_view);
|
|
}
|
|
}
|
|
|
|
void ProcessRemainingCaptureFrameContent(
|
|
bool level_change,
|
|
bool saturated_microphone_signal,
|
|
FrameBlocker* capture_blocker,
|
|
BlockFramer* linear_output_framer,
|
|
BlockFramer* output_framer,
|
|
BlockProcessor* block_processor,
|
|
std::vector<std::vector<std::vector<float>>>* linear_output_block,
|
|
std::vector<std::vector<std::vector<float>>>* block) {
|
|
if (!capture_blocker->IsBlockAvailable()) {
|
|
return;
|
|
}
|
|
|
|
capture_blocker->ExtractBlock(block);
|
|
block_processor->ProcessCapture(level_change, saturated_microphone_signal,
|
|
linear_output_block, block);
|
|
output_framer->InsertBlock(*block);
|
|
|
|
if (linear_output_framer) {
|
|
RTC_DCHECK(linear_output_block);
|
|
linear_output_framer->InsertBlock(*linear_output_block);
|
|
}
|
|
}
|
|
|
|
void BufferRenderFrameContent(
|
|
std::vector<std::vector<std::vector<float>>>* render_frame,
|
|
size_t sub_frame_index,
|
|
FrameBlocker* render_blocker,
|
|
BlockProcessor* block_processor,
|
|
std::vector<std::vector<std::vector<float>>>* block,
|
|
std::vector<std::vector<rtc::ArrayView<float>>>* sub_frame_view) {
|
|
FillSubFrameView(render_frame, sub_frame_index, sub_frame_view);
|
|
render_blocker->InsertSubFrameAndExtractBlock(*sub_frame_view, block);
|
|
block_processor->BufferRender(*block);
|
|
}
|
|
|
|
void BufferRemainingRenderFrameContent(
|
|
FrameBlocker* render_blocker,
|
|
BlockProcessor* block_processor,
|
|
std::vector<std::vector<std::vector<float>>>* block) {
|
|
if (!render_blocker->IsBlockAvailable()) {
|
|
return;
|
|
}
|
|
render_blocker->ExtractBlock(block);
|
|
block_processor->BufferRender(*block);
|
|
}
|
|
|
|
void CopyBufferIntoFrame(const AudioBuffer& buffer,
|
|
size_t num_bands,
|
|
size_t num_channels,
|
|
std::vector<std::vector<std::vector<float>>>* frame) {
|
|
RTC_DCHECK_EQ(num_bands, frame->size());
|
|
RTC_DCHECK_EQ(num_channels, (*frame)[0].size());
|
|
RTC_DCHECK_EQ(AudioBuffer::kSplitBandSize, (*frame)[0][0].size());
|
|
for (size_t band = 0; band < num_bands; ++band) {
|
|
for (size_t channel = 0; channel < num_channels; ++channel) {
|
|
rtc::ArrayView<const float> buffer_view(
|
|
&buffer.split_bands_const(channel)[band][0],
|
|
AudioBuffer::kSplitBandSize);
|
|
std::copy(buffer_view.begin(), buffer_view.end(),
|
|
(*frame)[band][channel].begin());
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// TODO(webrtc:5298): Move this to a separate file.
|
|
EchoCanceller3Config AdjustConfig(const EchoCanceller3Config& config) {
|
|
EchoCanceller3Config adjusted_cfg = config;
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3UseShortConfigChangeDuration")) {
|
|
adjusted_cfg.filter.config_change_duration_blocks = 10;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3UseZeroInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = 0.f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3UseDot1SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = .1f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3UseDot2SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = .2f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3UseDot3SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = .3f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3UseDot6SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = .6f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3UseDot9SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = .9f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3Use1Dot2SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = 1.2f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3Use1Dot6SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = 1.6f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3Use2Dot0SecondsInitialStateDuration")) {
|
|
adjusted_cfg.filter.initial_state_seconds = 2.0f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3EchoSaturationDetectionKillSwitch")) {
|
|
adjusted_cfg.ep_strength.echo_can_saturate = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3UseDot2ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.2f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot3ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.3f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot4ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.4f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot5ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.5f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot6ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.6f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot7ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.7f;
|
|
} else if (field_trial::IsEnabled("WebRTC-Aec3UseDot8ReverbDefaultLen")) {
|
|
adjusted_cfg.ep_strength.default_len = 0.8f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3ShortHeadroomKillSwitch")) {
|
|
// Two blocks headroom.
|
|
adjusted_cfg.delay.delay_headroom_samples = kBlockSize * 2;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3ClampInstQualityToZeroKillSwitch")) {
|
|
adjusted_cfg.erle.clamp_quality_estimate_to_zero = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3ClampInstQualityToOneKillSwitch")) {
|
|
adjusted_cfg.erle.clamp_quality_estimate_to_one = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3OnsetDetectionKillSwitch")) {
|
|
adjusted_cfg.erle.onset_detection = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceRenderDelayEstimationDownmixing")) {
|
|
adjusted_cfg.delay.render_alignment_mixing.downmix = true;
|
|
adjusted_cfg.delay.render_alignment_mixing.adaptive_selection = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceCaptureDelayEstimationDownmixing")) {
|
|
adjusted_cfg.delay.capture_alignment_mixing.downmix = true;
|
|
adjusted_cfg.delay.capture_alignment_mixing.adaptive_selection = false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceCaptureDelayEstimationLeftRightPrioritization")) {
|
|
adjusted_cfg.delay.capture_alignment_mixing.prefer_first_two_channels =
|
|
true;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-"
|
|
"Aec3RenderDelayEstimationLeftRightPrioritizationKillSwitch")) {
|
|
adjusted_cfg.delay.capture_alignment_mixing.prefer_first_two_channels =
|
|
false;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3SensitiveDominantNearendActivation")) {
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_threshold = 0.5f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3VerySensitiveDominantNearendActivation")) {
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_threshold = 0.75f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3TransparentAntiHowlingGain")) {
|
|
adjusted_cfg.suppressor.high_bands_suppression.anti_howling_gain = 1.f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceMoreTransparentNormalSuppressorTuning")) {
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_transparent = 0.4f;
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_suppress = 0.5f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceMoreTransparentNearendSuppressorTuning")) {
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_transparent = 1.29f;
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_suppress = 1.3f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceMoreTransparentNormalSuppressorHfTuning")) {
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_transparent = 0.3f;
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_suppress = 0.4f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceMoreTransparentNearendSuppressorHfTuning")) {
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_transparent = 1.09f;
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_suppress = 1.1f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceRapidlyAdjustingNormalSuppressorTunings")) {
|
|
adjusted_cfg.suppressor.normal_tuning.max_inc_factor = 2.5f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceRapidlyAdjustingNearendSuppressorTunings")) {
|
|
adjusted_cfg.suppressor.nearend_tuning.max_inc_factor = 2.5f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceSlowlyAdjustingNormalSuppressorTunings")) {
|
|
adjusted_cfg.suppressor.normal_tuning.max_dec_factor_lf = .2f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceSlowlyAdjustingNearendSuppressorTunings")) {
|
|
adjusted_cfg.suppressor.nearend_tuning.max_dec_factor_lf = .2f;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3EnforceStationarityProperties")) {
|
|
adjusted_cfg.echo_audibility.use_stationarity_properties = true;
|
|
}
|
|
|
|
if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceStationarityPropertiesAtInit")) {
|
|
adjusted_cfg.echo_audibility.use_stationarity_properties_at_init = true;
|
|
}
|
|
|
|
if (field_trial::IsEnabled("WebRTC-Aec3EnforceLowActiveRenderLimit")) {
|
|
adjusted_cfg.render_levels.active_render_limit = 50.f;
|
|
} else if (field_trial::IsEnabled(
|
|
"WebRTC-Aec3EnforceVeryLowActiveRenderLimit")) {
|
|
adjusted_cfg.render_levels.active_render_limit = 30.f;
|
|
}
|
|
|
|
// Field-trial based override for the whole suppressor tuning.
|
|
const std::string suppressor_tuning_override_trial_name =
|
|
field_trial::FindFullName("WebRTC-Aec3SuppressorTuningOverride");
|
|
|
|
FieldTrialParameter<double> nearend_tuning_mask_lf_enr_transparent(
|
|
"nearend_tuning_mask_lf_enr_transparent",
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_transparent);
|
|
FieldTrialParameter<double> nearend_tuning_mask_lf_enr_suppress(
|
|
"nearend_tuning_mask_lf_enr_suppress",
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_suppress);
|
|
FieldTrialParameter<double> nearend_tuning_mask_hf_enr_transparent(
|
|
"nearend_tuning_mask_hf_enr_transparent",
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_transparent);
|
|
FieldTrialParameter<double> nearend_tuning_mask_hf_enr_suppress(
|
|
"nearend_tuning_mask_hf_enr_suppress",
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_suppress);
|
|
FieldTrialParameter<double> nearend_tuning_max_inc_factor(
|
|
"nearend_tuning_max_inc_factor",
|
|
adjusted_cfg.suppressor.nearend_tuning.max_inc_factor);
|
|
FieldTrialParameter<double> nearend_tuning_max_dec_factor_lf(
|
|
"nearend_tuning_max_dec_factor_lf",
|
|
adjusted_cfg.suppressor.nearend_tuning.max_dec_factor_lf);
|
|
FieldTrialParameter<double> normal_tuning_mask_lf_enr_transparent(
|
|
"normal_tuning_mask_lf_enr_transparent",
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_transparent);
|
|
FieldTrialParameter<double> normal_tuning_mask_lf_enr_suppress(
|
|
"normal_tuning_mask_lf_enr_suppress",
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_suppress);
|
|
FieldTrialParameter<double> normal_tuning_mask_hf_enr_transparent(
|
|
"normal_tuning_mask_hf_enr_transparent",
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_transparent);
|
|
FieldTrialParameter<double> normal_tuning_mask_hf_enr_suppress(
|
|
"normal_tuning_mask_hf_enr_suppress",
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_suppress);
|
|
FieldTrialParameter<double> normal_tuning_max_inc_factor(
|
|
"normal_tuning_max_inc_factor",
|
|
adjusted_cfg.suppressor.normal_tuning.max_inc_factor);
|
|
FieldTrialParameter<double> normal_tuning_max_dec_factor_lf(
|
|
"normal_tuning_max_dec_factor_lf",
|
|
adjusted_cfg.suppressor.normal_tuning.max_dec_factor_lf);
|
|
FieldTrialParameter<double> dominant_nearend_detection_enr_threshold(
|
|
"dominant_nearend_detection_enr_threshold",
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_threshold);
|
|
FieldTrialParameter<double> dominant_nearend_detection_enr_exit_threshold(
|
|
"dominant_nearend_detection_enr_exit_threshold",
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_exit_threshold);
|
|
FieldTrialParameter<double> dominant_nearend_detection_snr_threshold(
|
|
"dominant_nearend_detection_snr_threshold",
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.snr_threshold);
|
|
FieldTrialParameter<int> dominant_nearend_detection_hold_duration(
|
|
"dominant_nearend_detection_hold_duration",
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.hold_duration);
|
|
FieldTrialParameter<int> dominant_nearend_detection_trigger_threshold(
|
|
"dominant_nearend_detection_trigger_threshold",
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.trigger_threshold);
|
|
FieldTrialParameter<double> ep_strength_default_len(
|
|
"ep_strength_default_len", adjusted_cfg.ep_strength.default_len);
|
|
|
|
ParseFieldTrial(
|
|
{&nearend_tuning_mask_lf_enr_transparent,
|
|
&nearend_tuning_mask_lf_enr_suppress,
|
|
&nearend_tuning_mask_hf_enr_transparent,
|
|
&nearend_tuning_mask_hf_enr_suppress, &nearend_tuning_max_inc_factor,
|
|
&nearend_tuning_max_dec_factor_lf,
|
|
&normal_tuning_mask_lf_enr_transparent,
|
|
&normal_tuning_mask_lf_enr_suppress,
|
|
&normal_tuning_mask_hf_enr_transparent,
|
|
&normal_tuning_mask_hf_enr_suppress, &normal_tuning_max_inc_factor,
|
|
&normal_tuning_max_dec_factor_lf,
|
|
&dominant_nearend_detection_enr_threshold,
|
|
&dominant_nearend_detection_enr_exit_threshold,
|
|
&dominant_nearend_detection_snr_threshold,
|
|
&dominant_nearend_detection_hold_duration,
|
|
&dominant_nearend_detection_trigger_threshold, &ep_strength_default_len},
|
|
suppressor_tuning_override_trial_name);
|
|
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_transparent =
|
|
static_cast<float>(nearend_tuning_mask_lf_enr_transparent.Get());
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_suppress =
|
|
static_cast<float>(nearend_tuning_mask_lf_enr_suppress.Get());
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_transparent =
|
|
static_cast<float>(nearend_tuning_mask_hf_enr_transparent.Get());
|
|
adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_suppress =
|
|
static_cast<float>(nearend_tuning_mask_hf_enr_suppress.Get());
|
|
adjusted_cfg.suppressor.nearend_tuning.max_inc_factor =
|
|
static_cast<float>(nearend_tuning_max_inc_factor.Get());
|
|
adjusted_cfg.suppressor.nearend_tuning.max_dec_factor_lf =
|
|
static_cast<float>(nearend_tuning_max_dec_factor_lf.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_transparent =
|
|
static_cast<float>(normal_tuning_mask_lf_enr_transparent.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_suppress =
|
|
static_cast<float>(normal_tuning_mask_lf_enr_suppress.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_transparent =
|
|
static_cast<float>(normal_tuning_mask_hf_enr_transparent.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_suppress =
|
|
static_cast<float>(normal_tuning_mask_hf_enr_suppress.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.max_inc_factor =
|
|
static_cast<float>(normal_tuning_max_inc_factor.Get());
|
|
adjusted_cfg.suppressor.normal_tuning.max_dec_factor_lf =
|
|
static_cast<float>(normal_tuning_max_dec_factor_lf.Get());
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_threshold =
|
|
static_cast<float>(dominant_nearend_detection_enr_threshold.Get());
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.enr_exit_threshold =
|
|
static_cast<float>(dominant_nearend_detection_enr_exit_threshold.Get());
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.snr_threshold =
|
|
static_cast<float>(dominant_nearend_detection_snr_threshold.Get());
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.hold_duration =
|
|
dominant_nearend_detection_hold_duration.Get();
|
|
adjusted_cfg.suppressor.dominant_nearend_detection.trigger_threshold =
|
|
dominant_nearend_detection_trigger_threshold.Get();
|
|
adjusted_cfg.ep_strength.default_len =
|
|
static_cast<float>(ep_strength_default_len.Get());
|
|
|
|
// Field trial-based overrides of individual suppressor parameters.
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendLfMaskTransparentOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_transparent);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendLfMaskSuppressOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.mask_lf.enr_suppress);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendHfMaskTransparentOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_transparent);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendHfMaskSuppressOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.mask_hf.enr_suppress);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendMaxIncFactorOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.max_inc_factor);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNearendMaxDecFactorLfOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.nearend_tuning.max_dec_factor_lf);
|
|
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalLfMaskTransparentOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_transparent);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalLfMaskSuppressOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.mask_lf.enr_suppress);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalHfMaskTransparentOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_transparent);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalHfMaskSuppressOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.mask_hf.enr_suppress);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalMaxIncFactorOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.max_inc_factor);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorNormalMaxDecFactorLfOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.normal_tuning.max_dec_factor_lf);
|
|
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorDominantNearendEnrThresholdOverride", 0.f, 100.f,
|
|
&adjusted_cfg.suppressor.dominant_nearend_detection.enr_threshold);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorDominantNearendEnrExitThresholdOverride", 0.f,
|
|
100.f,
|
|
&adjusted_cfg.suppressor.dominant_nearend_detection.enr_exit_threshold);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorDominantNearendSnrThresholdOverride", 0.f, 100.f,
|
|
&adjusted_cfg.suppressor.dominant_nearend_detection.snr_threshold);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorDominantNearendHoldDurationOverride", 0, 1000,
|
|
&adjusted_cfg.suppressor.dominant_nearend_detection.hold_duration);
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorDominantNearendTriggerThresholdOverride", 0, 1000,
|
|
&adjusted_cfg.suppressor.dominant_nearend_detection.trigger_threshold);
|
|
|
|
RetrieveFieldTrialValue(
|
|
"WebRTC-Aec3SuppressorAntiHowlingGainOverride", 0.f, 10.f,
|
|
&adjusted_cfg.suppressor.high_bands_suppression.anti_howling_gain);
|
|
|
|
RetrieveFieldTrialValue("WebRTC-Aec3SuppressorEpStrengthDefaultLenOverride",
|
|
-1.f, 1.f, &adjusted_cfg.ep_strength.default_len);
|
|
|
|
return adjusted_cfg;
|
|
}
|
|
|
|
class EchoCanceller3::RenderWriter {
|
|
public:
|
|
RenderWriter(ApmDataDumper* data_dumper,
|
|
SwapQueue<std::vector<std::vector<std::vector<float>>>,
|
|
Aec3RenderQueueItemVerifier>* render_transfer_queue,
|
|
size_t num_bands,
|
|
size_t num_channels);
|
|
~RenderWriter();
|
|
void Insert(const AudioBuffer& input);
|
|
|
|
private:
|
|
ApmDataDumper* data_dumper_;
|
|
const size_t num_bands_;
|
|
const size_t num_channels_;
|
|
HighPassFilter high_pass_filter_;
|
|
std::vector<std::vector<std::vector<float>>> render_queue_input_frame_;
|
|
SwapQueue<std::vector<std::vector<std::vector<float>>>,
|
|
Aec3RenderQueueItemVerifier>* render_transfer_queue_;
|
|
RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(RenderWriter);
|
|
};
|
|
|
|
EchoCanceller3::RenderWriter::RenderWriter(
|
|
ApmDataDumper* data_dumper,
|
|
SwapQueue<std::vector<std::vector<std::vector<float>>>,
|
|
Aec3RenderQueueItemVerifier>* render_transfer_queue,
|
|
size_t num_bands,
|
|
size_t num_channels)
|
|
: data_dumper_(data_dumper),
|
|
num_bands_(num_bands),
|
|
num_channels_(num_channels),
|
|
high_pass_filter_(16000, num_channels),
|
|
render_queue_input_frame_(
|
|
num_bands_,
|
|
std::vector<std::vector<float>>(
|
|
num_channels_,
|
|
std::vector<float>(AudioBuffer::kSplitBandSize, 0.f))),
|
|
render_transfer_queue_(render_transfer_queue) {
|
|
RTC_DCHECK(data_dumper);
|
|
}
|
|
|
|
EchoCanceller3::RenderWriter::~RenderWriter() = default;
|
|
|
|
void EchoCanceller3::RenderWriter::Insert(const AudioBuffer& input) {
|
|
RTC_DCHECK_EQ(AudioBuffer::kSplitBandSize, input.num_frames_per_band());
|
|
RTC_DCHECK_EQ(num_bands_, input.num_bands());
|
|
RTC_DCHECK_EQ(num_channels_, input.num_channels());
|
|
|
|
// TODO(bugs.webrtc.org/8759) Temporary work-around.
|
|
if (num_bands_ != input.num_bands())
|
|
return;
|
|
|
|
data_dumper_->DumpWav("aec3_render_input", AudioBuffer::kSplitBandSize,
|
|
&input.split_bands_const(0)[0][0], 16000, 1);
|
|
|
|
CopyBufferIntoFrame(input, num_bands_, num_channels_,
|
|
&render_queue_input_frame_);
|
|
high_pass_filter_.Process(&render_queue_input_frame_[0]);
|
|
|
|
static_cast<void>(render_transfer_queue_->Insert(&render_queue_input_frame_));
|
|
}
|
|
|
|
int EchoCanceller3::instance_count_ = 0;
|
|
|
|
EchoCanceller3::EchoCanceller3(const EchoCanceller3Config& config,
|
|
int sample_rate_hz,
|
|
size_t num_render_channels,
|
|
size_t num_capture_channels)
|
|
: EchoCanceller3(AdjustConfig(config),
|
|
sample_rate_hz,
|
|
num_render_channels,
|
|
num_capture_channels,
|
|
std::unique_ptr<BlockProcessor>(
|
|
BlockProcessor::Create(AdjustConfig(config),
|
|
sample_rate_hz,
|
|
num_render_channels,
|
|
num_capture_channels))) {}
|
|
EchoCanceller3::EchoCanceller3(const EchoCanceller3Config& config,
|
|
int sample_rate_hz,
|
|
size_t num_render_channels,
|
|
size_t num_capture_channels,
|
|
std::unique_ptr<BlockProcessor> block_processor)
|
|
: data_dumper_(
|
|
new ApmDataDumper(rtc::AtomicOps::Increment(&instance_count_))),
|
|
config_(config),
|
|
sample_rate_hz_(sample_rate_hz),
|
|
num_bands_(NumBandsForRate(sample_rate_hz_)),
|
|
num_render_channels_(num_render_channels),
|
|
num_capture_channels_(num_capture_channels),
|
|
output_framer_(num_bands_, num_capture_channels_),
|
|
capture_blocker_(num_bands_, num_capture_channels_),
|
|
render_blocker_(num_bands_, num_render_channels_),
|
|
render_transfer_queue_(
|
|
kRenderTransferQueueSizeFrames,
|
|
std::vector<std::vector<std::vector<float>>>(
|
|
num_bands_,
|
|
std::vector<std::vector<float>>(
|
|
num_render_channels_,
|
|
std::vector<float>(AudioBuffer::kSplitBandSize, 0.f))),
|
|
Aec3RenderQueueItemVerifier(num_bands_,
|
|
num_render_channels_,
|
|
AudioBuffer::kSplitBandSize)),
|
|
block_processor_(std::move(block_processor)),
|
|
render_queue_output_frame_(
|
|
num_bands_,
|
|
std::vector<std::vector<float>>(
|
|
num_render_channels_,
|
|
std::vector<float>(AudioBuffer::kSplitBandSize, 0.f))),
|
|
render_block_(
|
|
num_bands_,
|
|
std::vector<std::vector<float>>(num_render_channels_,
|
|
std::vector<float>(kBlockSize, 0.f))),
|
|
capture_block_(
|
|
num_bands_,
|
|
std::vector<std::vector<float>>(num_capture_channels_,
|
|
std::vector<float>(kBlockSize, 0.f))),
|
|
render_sub_frame_view_(
|
|
num_bands_,
|
|
std::vector<rtc::ArrayView<float>>(num_render_channels_)),
|
|
capture_sub_frame_view_(
|
|
num_bands_,
|
|
std::vector<rtc::ArrayView<float>>(num_capture_channels_)) {
|
|
RTC_DCHECK(ValidFullBandRate(sample_rate_hz_));
|
|
|
|
if (config_.delay.fixed_capture_delay_samples > 0) {
|
|
block_delay_buffer_.reset(new BlockDelayBuffer(
|
|
num_capture_channels_, num_bands_, AudioBuffer::kSplitBandSize,
|
|
config_.delay.fixed_capture_delay_samples));
|
|
}
|
|
|
|
render_writer_.reset(new RenderWriter(data_dumper_.get(),
|
|
&render_transfer_queue_, num_bands_,
|
|
num_render_channels_));
|
|
|
|
RTC_DCHECK_EQ(num_bands_, std::max(sample_rate_hz_, 16000) / 16000);
|
|
RTC_DCHECK_GE(kMaxNumBands, num_bands_);
|
|
|
|
if (config_.filter.export_linear_aec_output) {
|
|
linear_output_framer_.reset(new BlockFramer(1, num_capture_channels_));
|
|
linear_output_block_ =
|
|
std::make_unique<std::vector<std::vector<std::vector<float>>>>(
|
|
1, std::vector<std::vector<float>>(
|
|
num_capture_channels_, std::vector<float>(kBlockSize, 0.f)));
|
|
linear_output_sub_frame_view_ =
|
|
std::vector<std::vector<rtc::ArrayView<float>>>(
|
|
1, std::vector<rtc::ArrayView<float>>(num_capture_channels_));
|
|
}
|
|
}
|
|
|
|
EchoCanceller3::~EchoCanceller3() = default;
|
|
|
|
void EchoCanceller3::AnalyzeRender(const AudioBuffer& render) {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&render_race_checker_);
|
|
|
|
RTC_DCHECK_EQ(render.num_channels(), num_render_channels_);
|
|
data_dumper_->DumpRaw("aec3_call_order",
|
|
static_cast<int>(EchoCanceller3ApiCall::kRender));
|
|
|
|
return render_writer_->Insert(render);
|
|
}
|
|
|
|
void EchoCanceller3::AnalyzeCapture(const AudioBuffer& capture) {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
|
data_dumper_->DumpWav("aec3_capture_analyze_input", capture.num_frames(),
|
|
capture.channels_const()[0], sample_rate_hz_, 1);
|
|
saturated_microphone_signal_ = false;
|
|
for (size_t channel = 0; channel < capture.num_channels(); ++channel) {
|
|
saturated_microphone_signal_ |=
|
|
DetectSaturation(rtc::ArrayView<const float>(
|
|
capture.channels_const()[channel], capture.num_frames()));
|
|
if (saturated_microphone_signal_) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void EchoCanceller3::ProcessCapture(AudioBuffer* capture, bool level_change) {
|
|
ProcessCapture(capture, nullptr, level_change);
|
|
}
|
|
|
|
void EchoCanceller3::ProcessCapture(AudioBuffer* capture,
|
|
AudioBuffer* linear_output,
|
|
bool level_change) {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
|
RTC_DCHECK(capture);
|
|
RTC_DCHECK_EQ(num_bands_, capture->num_bands());
|
|
RTC_DCHECK_EQ(AudioBuffer::kSplitBandSize, capture->num_frames_per_band());
|
|
RTC_DCHECK_EQ(capture->num_channels(), num_capture_channels_);
|
|
data_dumper_->DumpRaw("aec3_call_order",
|
|
static_cast<int>(EchoCanceller3ApiCall::kCapture));
|
|
|
|
if (linear_output && !linear_output_framer_) {
|
|
RTC_LOG(LS_ERROR) << "Trying to retrieve the linear AEC output without "
|
|
"properly configuring AEC3.";
|
|
RTC_NOTREACHED();
|
|
}
|
|
|
|
// Report capture call in the metrics and periodically update API call
|
|
// metrics.
|
|
api_call_metrics_.ReportCaptureCall();
|
|
|
|
// Optionally delay the capture signal.
|
|
if (config_.delay.fixed_capture_delay_samples > 0) {
|
|
RTC_DCHECK(block_delay_buffer_);
|
|
block_delay_buffer_->DelaySignal(capture);
|
|
}
|
|
|
|
rtc::ArrayView<float> capture_lower_band = rtc::ArrayView<float>(
|
|
&capture->split_bands(0)[0][0], AudioBuffer::kSplitBandSize);
|
|
|
|
data_dumper_->DumpWav("aec3_capture_input", capture_lower_band, 16000, 1);
|
|
|
|
EmptyRenderQueue();
|
|
|
|
ProcessCaptureFrameContent(linear_output, capture, level_change,
|
|
saturated_microphone_signal_, 0, &capture_blocker_,
|
|
linear_output_framer_.get(), &output_framer_,
|
|
block_processor_.get(), linear_output_block_.get(),
|
|
&linear_output_sub_frame_view_, &capture_block_,
|
|
&capture_sub_frame_view_);
|
|
|
|
ProcessCaptureFrameContent(linear_output, capture, level_change,
|
|
saturated_microphone_signal_, 1, &capture_blocker_,
|
|
linear_output_framer_.get(), &output_framer_,
|
|
block_processor_.get(), linear_output_block_.get(),
|
|
&linear_output_sub_frame_view_, &capture_block_,
|
|
&capture_sub_frame_view_);
|
|
|
|
ProcessRemainingCaptureFrameContent(
|
|
level_change, saturated_microphone_signal_, &capture_blocker_,
|
|
linear_output_framer_.get(), &output_framer_, block_processor_.get(),
|
|
linear_output_block_.get(), &capture_block_);
|
|
|
|
data_dumper_->DumpWav("aec3_capture_output", AudioBuffer::kSplitBandSize,
|
|
&capture->split_bands(0)[0][0], 16000, 1);
|
|
}
|
|
|
|
EchoControl::Metrics EchoCanceller3::GetMetrics() const {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
|
Metrics metrics;
|
|
block_processor_->GetMetrics(&metrics);
|
|
return metrics;
|
|
}
|
|
|
|
void EchoCanceller3::SetAudioBufferDelay(int delay_ms) {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
|
block_processor_->SetAudioBufferDelay(delay_ms);
|
|
}
|
|
|
|
bool EchoCanceller3::ActiveProcessing() const {
|
|
return true;
|
|
}
|
|
|
|
EchoCanceller3Config EchoCanceller3::CreateDefaultConfig(
|
|
size_t num_render_channels,
|
|
size_t num_capture_channels) {
|
|
EchoCanceller3Config cfg;
|
|
if (num_render_channels > 1) {
|
|
// Use shorter and more rapidly adapting coarse filter to compensate for
|
|
// thge increased number of total filter parameters to adapt.
|
|
cfg.filter.coarse.length_blocks = 11;
|
|
cfg.filter.coarse.rate = 0.95f;
|
|
cfg.filter.coarse_initial.length_blocks = 11;
|
|
cfg.filter.coarse_initial.rate = 0.95f;
|
|
|
|
// Use more concervative suppressor behavior for non-nearend speech.
|
|
cfg.suppressor.normal_tuning.max_dec_factor_lf = 0.35f;
|
|
cfg.suppressor.normal_tuning.max_inc_factor = 1.5f;
|
|
}
|
|
return cfg;
|
|
}
|
|
|
|
void EchoCanceller3::EmptyRenderQueue() {
|
|
RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
|
bool frame_to_buffer =
|
|
render_transfer_queue_.Remove(&render_queue_output_frame_);
|
|
while (frame_to_buffer) {
|
|
// Report render call in the metrics.
|
|
api_call_metrics_.ReportRenderCall();
|
|
|
|
BufferRenderFrameContent(&render_queue_output_frame_, 0, &render_blocker_,
|
|
block_processor_.get(), &render_block_,
|
|
&render_sub_frame_view_);
|
|
|
|
BufferRenderFrameContent(&render_queue_output_frame_, 1, &render_blocker_,
|
|
block_processor_.get(), &render_block_,
|
|
&render_sub_frame_view_);
|
|
|
|
BufferRemainingRenderFrameContent(&render_blocker_, block_processor_.get(),
|
|
&render_block_);
|
|
|
|
frame_to_buffer =
|
|
render_transfer_queue_.Remove(&render_queue_output_frame_);
|
|
}
|
|
}
|
|
} // namespace webrtc
|