mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 14:50:39 +01:00
a837dd790d
Agc2 applies a digital gain to the nearend signal. When the analog level changes, the digital gain calculation is no longer valid. Therefore Agc2 should be notified to analog gain changes. This CL also allow audioproc_f to chain AGC1 and AGC2. In a dependent CL we will allow using AGC1 for analog gain and AGC2 for digital gain. Bug: webrtc:7494 Change-Id: Id75b3728fbf2de1d84b7fba005e4670c7a2985d9 Reviewed-on: https://webrtc-review.googlesource.com/89387 Commit-Queue: Alex Loiko <aleloi@webrtc.org> Reviewed-by: Alessio Bazzica <alessiob@webrtc.org> Cr-Commit-Position: refs/heads/master@{#24231}
57 lines
1.9 KiB
C++
57 lines
1.9 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/agc2/adaptive_agc.h"
|
|
|
|
#include <algorithm>
|
|
#include <numeric>
|
|
|
|
#include "common_audio/include/audio_util.h"
|
|
#include "modules/audio_processing/agc2/vad_with_level.h"
|
|
#include "modules/audio_processing/logging/apm_data_dumper.h"
|
|
|
|
namespace webrtc {
|
|
|
|
AdaptiveAgc::AdaptiveAgc(ApmDataDumper* apm_data_dumper)
|
|
: speech_level_estimator_(apm_data_dumper),
|
|
gain_applier_(apm_data_dumper),
|
|
apm_data_dumper_(apm_data_dumper),
|
|
noise_level_estimator_(apm_data_dumper) {
|
|
RTC_DCHECK(apm_data_dumper);
|
|
}
|
|
|
|
AdaptiveAgc::~AdaptiveAgc() = default;
|
|
|
|
void AdaptiveAgc::Process(AudioFrameView<float> float_frame) {
|
|
const VadWithLevel::LevelAndProbability vad_result =
|
|
vad_.AnalyzeFrame(float_frame);
|
|
apm_data_dumper_->DumpRaw("agc2_vad_probability",
|
|
vad_result.speech_probability);
|
|
apm_data_dumper_->DumpRaw("agc2_vad_rms_dbfs", vad_result.speech_rms_dbfs);
|
|
|
|
apm_data_dumper_->DumpRaw("agc2_vad_peak_dbfs", vad_result.speech_peak_dbfs);
|
|
speech_level_estimator_.UpdateEstimation(vad_result);
|
|
|
|
const float speech_level_dbfs = speech_level_estimator_.LatestLevelEstimate();
|
|
|
|
const float noise_level_dbfs = noise_level_estimator_.Analyze(float_frame);
|
|
|
|
apm_data_dumper_->DumpRaw("agc2_noise_estimate_dbfs", noise_level_dbfs);
|
|
|
|
// The gain applier applies the gain.
|
|
gain_applier_.Process(speech_level_dbfs, noise_level_dbfs, vad_result,
|
|
float_frame);
|
|
}
|
|
|
|
void AdaptiveAgc::Reset() {
|
|
speech_level_estimator_.Reset();
|
|
}
|
|
|
|
} // namespace webrtc
|