speech_level_estimator_unittest.cc | Explore in Territory

/*
 *  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/speech_level_estimator.h"

#include <memory>

#include "api/audio/audio_processing.h"
#include "modules/audio_processing/agc2/agc2_common.h"
#include "modules/audio_processing/logging/apm_data_dumper.h"
#include "rtc_base/gunit.h"

namespace webrtc {
namespace {

AdaptiveDigitalConfig;

// Number of speech frames that the level estimator must observe in order to
// become confident about the estimated level.
constexpr int kNumFramesToConfidence = …;
static_assert …;

constexpr float kConvergenceSpeedTestsLevelTolerance = …;

// Provides the `vad_level` value `num_iterations` times to `level_estimator`.
void RunOnConstantLevel(int num_iterations,
                        float rms_dbfs,
                        float peak_dbfs,
                        float speech_probability,
                        SpeechLevelEstimator& level_estimator) { … }

constexpr float kNoSpeechProbability = …;
constexpr float kLowSpeechProbability = …;
constexpr float kMaxSpeechProbability = …;

// Level estimator with data dumper.
struct TestLevelEstimator { … };

// Checks that the level estimator converges to a constant input speech level.
TEST(GainController2SpeechLevelEstimator, LevelStabilizes) { … }

// Checks that the level controller does not become confident when too few
// speech frames are observed.
TEST(GainController2SpeechLevelEstimator, IsNotConfident) { … }

// Checks that the level controller becomes confident when enough speech frames
// are observed.
TEST(GainController2SpeechLevelEstimator, IsConfident) { … }

// Checks that the estimated level is not affected by the level of non-speech
// frames.
TEST(GainController2SpeechLevelEstimator, EstimatorIgnoresNonSpeechFrames) { … }

// Checks the convergence speed of the estimator before it becomes confident.
TEST(GainController2SpeechLevelEstimator, ConvergenceSpeedBeforeConfidence) { … }

// Checks the convergence speed of the estimator after it becomes confident.
TEST(GainController2SpeechLevelEstimator, ConvergenceSpeedAfterConfidence) { … }

class SpeechLevelEstimatorParametrization
    : public ::testing::TestWithParam<int> { … };

TEST_P(SpeechLevelEstimatorParametrization, DoNotAdaptToShortSpeechSegments) { … }

TEST_P(SpeechLevelEstimatorParametrization, AdaptToEnoughSpeechSegments) { … }

INSTANTIATE_TEST_SUITE_P(…);

}  // namespace
}  // namespace webrtc
chromium/third_party/webrtc/modules/audio_processing/agc2/speech_level_estimator_unittest.cc
