mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-19 08:37:54 +01:00
87a7b82520
This CL adds proper multichannel support to the noise suppressor. To accomplish that in a safe way, a full refactoring of the noise suppressor code has been done. Due to floating point precision, the changes made are not entirely bitexact. They are, however, very close to being bitexact. As a safety measure, the former noise suppressor code is preserved and a kill-switch is added to allow revering to the legacy noise suppressor in case issues arise. Bug: webrtc:10895, b/143344262 Change-Id: I0b071011b23265ac12e6d4b3956499d122286657 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/158407 Commit-Queue: Per Åhgren <peah@webrtc.org> Reviewed-by: Gustaf Ullberg <gustaf@webrtc.org> Cr-Commit-Position: refs/heads/master@{#29646}
51 lines
1.8 KiB
C++
51 lines
1.8 KiB
C++
/*
|
|
* Copyright (c) 2019 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_NS_SPEECH_PROBABILITY_ESTIMATOR_H_
|
|
#define MODULES_AUDIO_PROCESSING_NS_SPEECH_PROBABILITY_ESTIMATOR_H_
|
|
|
|
#include <array>
|
|
|
|
#include "api/array_view.h"
|
|
#include "modules/audio_processing/ns/ns_common.h"
|
|
#include "modules/audio_processing/ns/signal_model_estimator.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// Class for estimating the probability of speech.
|
|
class SpeechProbabilityEstimator {
|
|
public:
|
|
SpeechProbabilityEstimator();
|
|
SpeechProbabilityEstimator(const SpeechProbabilityEstimator&) = delete;
|
|
SpeechProbabilityEstimator& operator=(const SpeechProbabilityEstimator&) =
|
|
delete;
|
|
|
|
// Compute speech probability.
|
|
void Update(
|
|
int32_t num_analyzed_frames,
|
|
rtc::ArrayView<const float, kFftSizeBy2Plus1> prior_snr,
|
|
rtc::ArrayView<const float, kFftSizeBy2Plus1> post_snr,
|
|
rtc::ArrayView<const float, kFftSizeBy2Plus1> conservative_noise_spectrum,
|
|
rtc::ArrayView<const float, kFftSizeBy2Plus1> signal_spectrum,
|
|
float signal_spectral_sum,
|
|
float signal_energy);
|
|
|
|
float get_prior_probability() const { return prior_speech_prob_; }
|
|
rtc::ArrayView<const float> get_probability() { return speech_probability_; }
|
|
|
|
private:
|
|
SignalModelEstimator signal_model_estimator_;
|
|
float prior_speech_prob_ = .5f;
|
|
std::array<float, kFftSizeBy2Plus1> speech_probability_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_AUDIO_PROCESSING_NS_SPEECH_PROBABILITY_ESTIMATOR_H_
|