mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-17 07:37:51 +01:00
ef5d5af3a0
This CL introduces an adaptive estimation of the early reverb in the estimation for the room reverberation. The benefits of this is that for room with long early reflections there is a lower risk of underestimating the reverberation. This CL is for a landing the code in https://webrtc-review.googlesource.com/c/src/+/87420, and the review of the code was done in that CL. The author of code is devicentepena@webrtc.org Bug: webrtc:9479, chromium:865397 Change-Id: Id6f57e2a684664aef96e8c502e66775f37da59da Reviewed-on: https://webrtc-review.googlesource.com/91162 Commit-Queue: Per Åhgren <peah@webrtc.org> Reviewed-by: Sam Zackrisson <saza@webrtc.org> Cr-Commit-Position: refs/heads/master@{#24146}
61 lines
2.1 KiB
C++
61 lines
2.1 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.
|
|
*/
|
|
|
|
#ifndef MODULES_AUDIO_PROCESSING_AEC3_REVERB_MODEL_ESTIMATOR_H_
|
|
#define MODULES_AUDIO_PROCESSING_AEC3_REVERB_MODEL_ESTIMATOR_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "absl/types/optional.h"
|
|
#include "api/array_view.h"
|
|
#include "api/audio/echo_canceller3_config.h"
|
|
#include "modules/audio_processing/aec3/reverb_decay_estimator.h"
|
|
#include "modules/audio_processing/aec3/reverb_frequency_response.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class ApmDataDumper;
|
|
|
|
// Class for estimating the model parameters for the reverberant echo.
|
|
class ReverbModelEstimator {
|
|
public:
|
|
explicit ReverbModelEstimator(const EchoCanceller3Config& config);
|
|
~ReverbModelEstimator();
|
|
|
|
// Updates the estimates based on new data.
|
|
void Update(rtc::ArrayView<const float> impulse_response,
|
|
const std::vector<std::array<float, kFftLengthBy2Plus1>>&
|
|
frequency_response,
|
|
const absl::optional<float>& linear_filter_quality,
|
|
int filter_delay_blocks,
|
|
bool usable_linear_estimate,
|
|
bool stationary_block);
|
|
|
|
// Returns the exponential decay of the reverberant echo.
|
|
float ReverbDecay() const { return reverb_decay_estimator_.Decay(); }
|
|
|
|
// Return the frequency response of the reverberant echo.
|
|
rtc::ArrayView<const float> GetReverbFrequencyResponse() const {
|
|
return reverb_frequency_response_.FrequencyResponse();
|
|
}
|
|
|
|
// Dumps debug data.
|
|
void Dump(ApmDataDumper* data_dumper) const {
|
|
reverb_decay_estimator_.Dump(data_dumper);
|
|
}
|
|
|
|
private:
|
|
ReverbDecayEstimator reverb_decay_estimator_;
|
|
ReverbFrequencyResponse reverb_frequency_response_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_AUDIO_PROCESSING_AEC3_REVERB_MODEL_ESTIMATOR_H_
|