| 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 | 10 |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 68 return rtc::Optional<size_t>(); | 68 return rtc::Optional<size_t>(); |
| 69 } | 69 } |
| 70 | 70 |
| 71 constexpr int kEchoPathChangeCounterInitial = kNumBlocksPerSecond / 5; | 71 constexpr int kEchoPathChangeCounterInitial = kNumBlocksPerSecond / 5; |
| 72 constexpr int kEchoPathChangeCounterMax = 2 * kNumBlocksPerSecond; | 72 constexpr int kEchoPathChangeCounterMax = 2 * kNumBlocksPerSecond; |
| 73 | 73 |
| 74 } // namespace | 74 } // namespace |
| 75 | 75 |
| 76 int AecState::instance_count_ = 0; | 76 int AecState::instance_count_ = 0; |
| 77 | 77 |
| 78 AecState::AecState(float reverb_decay) | 78 AecState::AecState(const AudioProcessing::Config::EchoCanceller3& config) |
| 79 : data_dumper_( | 79 : data_dumper_( |
| 80 new ApmDataDumper(rtc::AtomicOps::Increment(&instance_count_))), | 80 new ApmDataDumper(rtc::AtomicOps::Increment(&instance_count_))), |
| 81 erle_estimator_(config.param.erle.min, |
| 82 config.param.erle.max_l, |
| 83 config.param.erle.max_h), |
| 81 echo_path_change_counter_(kEchoPathChangeCounterInitial), | 84 echo_path_change_counter_(kEchoPathChangeCounterInitial), |
| 82 reverb_decay_(reverb_decay) {} | 85 config_(config), |
| 86 reverb_decay_(config_.param.ep_strength.default_len) {} |
| 83 | 87 |
| 84 AecState::~AecState() = default; | 88 AecState::~AecState() = default; |
| 85 | 89 |
| 86 void AecState::HandleEchoPathChange( | 90 void AecState::HandleEchoPathChange( |
| 87 const EchoPathVariability& echo_path_variability) { | 91 const EchoPathVariability& echo_path_variability) { |
| 88 if (echo_path_variability.AudioPathChanged()) { | 92 if (echo_path_variability.AudioPathChanged()) { |
| 89 blocks_since_last_saturation_ = 0; | 93 blocks_since_last_saturation_ = 0; |
| 90 usable_linear_estimate_ = false; | 94 usable_linear_estimate_ = false; |
| 91 echo_leakage_detected_ = false; | 95 echo_leakage_detected_ = false; |
| 92 capture_signal_saturation_ = false; | 96 capture_signal_saturation_ = false; |
| (...skipping 152 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 245 | 249 |
| 246 // If all reverb candidates have been evaluated, choose the best one as the | 250 // If all reverb candidates have been evaluated, choose the best one as the |
| 247 // reverb decay. | 251 // reverb decay. |
| 248 if (reverb_decay_to_test_ >= 0.9965f) { | 252 if (reverb_decay_to_test_ >= 0.9965f) { |
| 249 if (reverb_decay_candidate_residual_ < 0.f) { | 253 if (reverb_decay_candidate_residual_ < 0.f) { |
| 250 // Transform the decay to be in the unit of blocks. | 254 // Transform the decay to be in the unit of blocks. |
| 251 reverb_decay_ = powf(reverb_decay_candidate_, kFftLengthBy2); | 255 reverb_decay_ = powf(reverb_decay_candidate_, kFftLengthBy2); |
| 252 | 256 |
| 253 // Limit the estimated reverb_decay_ to the maximum one needed in practice | 257 // Limit the estimated reverb_decay_ to the maximum one needed in practice |
| 254 // to minimize the impact of incorrect estimates. | 258 // to minimize the impact of incorrect estimates. |
| 255 reverb_decay_ = std::min(0.8f, reverb_decay_); | 259 reverb_decay_ = |
| 260 std::min(config_.param.ep_strength.default_len, reverb_decay_); |
| 256 } | 261 } |
| 257 reverb_decay_to_test_ = 0.9f; | 262 reverb_decay_to_test_ = 0.9f; |
| 258 reverb_decay_candidate_residual_ = -1.f; | 263 reverb_decay_candidate_residual_ = -1.f; |
| 259 } | 264 } |
| 260 | 265 |
| 261 // For noisy impulse responses, assume a fixed tail length. | 266 // For noisy impulse responses, assume a fixed tail length. |
| 262 if (tail_power > 0.0005f) { | 267 if (tail_power > 0.0005f) { |
| 263 reverb_decay_ = 0.7f; | 268 reverb_decay_ = config_.param.ep_strength.default_len; |
| 264 } | 269 } |
| 265 data_dumper_->DumpRaw("aec3_reverb_decay", reverb_decay_); | 270 data_dumper_->DumpRaw("aec3_reverb_decay", reverb_decay_); |
| 266 data_dumper_->DumpRaw("aec3_tail_power", tail_power); | 271 data_dumper_->DumpRaw("aec3_tail_power", tail_power); |
| 267 } | 272 } |
| 268 | 273 |
| 269 void AecState::EchoAudibility::Update(rtc::ArrayView<const float> x, | 274 void AecState::EchoAudibility::Update(rtc::ArrayView<const float> x, |
| 270 const std::array<float, kBlockSize>& s) { | 275 const std::array<float, kBlockSize>& s) { |
| 271 auto result_x = std::minmax_element(x.begin(), x.end()); | 276 auto result_x = std::minmax_element(x.begin(), x.end()); |
| 272 auto result_s = std::minmax_element(s.begin(), s.end()); | 277 auto result_s = std::minmax_element(s.begin(), s.end()); |
| 273 const float x_abs = | 278 const float x_abs = |
| (...skipping 26 matching lines...) Expand all Loading... |
| 300 max_nearend_ = e_abs; | 305 max_nearend_ = e_abs; |
| 301 max_nearend_counter_ = 0; | 306 max_nearend_counter_ = 0; |
| 302 } else { | 307 } else { |
| 303 if (++max_nearend_counter_ > 5 * kNumBlocksPerSecond) { | 308 if (++max_nearend_counter_ > 5 * kNumBlocksPerSecond) { |
| 304 max_nearend_ *= 0.995f; | 309 max_nearend_ *= 0.995f; |
| 305 } | 310 } |
| 306 } | 311 } |
| 307 } | 312 } |
| 308 | 313 |
| 309 } // namespace webrtc | 314 } // namespace webrtc |
| OLD | NEW |