mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
0ae390fd17
Adding a delay unit to be used in the APM Transient Suppressor (TS) sub-module through which the observerd voice probabilities are temporally aligned to the audio processed by TS, which introduces algorithmic delay. Bug: webrtc:13663 Change-Id: I2136c303914580851c742d8db89478a13b06dacb Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/255680 Reviewed-by: Hanna Silen <silen@webrtc.org> Commit-Queue: Alessio Bazzica <alessiob@webrtc.org> Cr-Commit-Position: refs/heads/main@{#36487}
108 lines
3.9 KiB
C++
108 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2022 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/transient/voice_probability_delay_unit.h"
|
|
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
|
|
// Checks that with zero delay, the observed value is immediately returned as
|
|
// delayed value.
|
|
TEST(VoiceProbabilityDelayUnit, NoDelay) {
|
|
VoiceProbabilityDelayUnit delay_unit(/*delay_num_samples=*/0,
|
|
/*sample_rate_hz=*/48000);
|
|
constexpr int kMax = 5;
|
|
for (int i = 0; i <= kMax; ++i) {
|
|
SCOPED_TRACE(i);
|
|
float voice_probability = static_cast<float>(i) / kMax;
|
|
EXPECT_EQ(voice_probability, delay_unit.Delay(voice_probability));
|
|
}
|
|
}
|
|
|
|
// Checks that with integer delays, an exact copy of a previously observed value
|
|
// is returned.
|
|
TEST(VoiceProbabilityDelayUnit, IntegerDelay) {
|
|
VoiceProbabilityDelayUnit delay_unit_10ms(/*delay_num_samples=*/480,
|
|
/*sample_rate_hz=*/48000);
|
|
delay_unit_10ms.Delay(0.125f);
|
|
EXPECT_EQ(0.125f, delay_unit_10ms.Delay(0.9f));
|
|
|
|
VoiceProbabilityDelayUnit delay_unit_20ms(/*delay_num_samples=*/960,
|
|
/*sample_rate_hz=*/48000);
|
|
delay_unit_20ms.Delay(0.125f);
|
|
delay_unit_20ms.Delay(0.8f);
|
|
EXPECT_EQ(0.125f, delay_unit_20ms.Delay(0.9f));
|
|
}
|
|
|
|
// Checks that with a fractional delay < 10 ms, interpolation is applied.
|
|
TEST(VoiceProbabilityDelayUnit, FractionalDelayLessThan10ms) {
|
|
// Create delay unit with fractional delay of 6 ms.
|
|
VoiceProbabilityDelayUnit delay_unit(/*delay_num_samples=*/288,
|
|
/*sample_rate_hz=*/48000);
|
|
// frame 0
|
|
// --------- frame 1
|
|
// ---------
|
|
// 0000001111
|
|
delay_unit.Delay(1.0f);
|
|
EXPECT_FLOAT_EQ(0.68f, delay_unit.Delay(0.2f));
|
|
}
|
|
|
|
// Checks that with a fractional delay > 10 ms, interpolation is applied.
|
|
TEST(VoiceProbabilityDelayUnit, FractionalDelayGreaterThan10ms) {
|
|
// Create delay unit with fractional delay of 14 ms.
|
|
VoiceProbabilityDelayUnit delay_unit(/*delay_num_samples=*/672,
|
|
/*sample_rate_hz=*/48000);
|
|
// frame 0
|
|
// --------- frame 1
|
|
// --------- frame 2
|
|
// ---------
|
|
// 0000111111
|
|
delay_unit.Delay(1.0f);
|
|
delay_unit.Delay(0.2f);
|
|
EXPECT_FLOAT_EQ(0.52f, delay_unit.Delay(1.0f));
|
|
}
|
|
|
|
// Checks that `Initialize()` resets the delay unit.
|
|
TEST(VoiceProbabilityDelayUnit, InitializeResetsDelayUnit) {
|
|
VoiceProbabilityDelayUnit delay_unit(/*delay_num_samples=*/960,
|
|
/*sample_rate_hz=*/48000);
|
|
delay_unit.Delay(1.0f);
|
|
delay_unit.Delay(0.9f);
|
|
|
|
delay_unit.Initialize(/*delay_num_samples=*/160, /*sample_rate_hz=*/8000);
|
|
EXPECT_EQ(0.0f, delay_unit.Delay(0.1f));
|
|
EXPECT_EQ(0.0f, delay_unit.Delay(0.2f));
|
|
EXPECT_EQ(0.1f, delay_unit.Delay(0.3f));
|
|
}
|
|
|
|
// Checks that `Initialize()` handles delay changes.
|
|
TEST(VoiceProbabilityDelayUnit, InitializeHandlesDelayChanges) {
|
|
// Start with a 20 ms delay.
|
|
VoiceProbabilityDelayUnit delay_unit(/*delay_num_samples=*/960,
|
|
/*sample_rate_hz=*/48000);
|
|
delay_unit.Delay(1.0f);
|
|
delay_unit.Delay(0.9f);
|
|
|
|
// Lower the delay to 10 ms.
|
|
delay_unit.Initialize(/*delay_num_samples=*/80, /*sample_rate_hz=*/8000);
|
|
EXPECT_EQ(0.0f, delay_unit.Delay(0.1f));
|
|
EXPECT_EQ(0.1f, delay_unit.Delay(0.2f));
|
|
|
|
// Increase the delay to 15 ms.
|
|
delay_unit.Initialize(/*delay_num_samples=*/120, /*sample_rate_hz=*/8000);
|
|
EXPECT_EQ(0.0f, delay_unit.Delay(0.1f));
|
|
EXPECT_EQ(0.05f, delay_unit.Delay(0.2f));
|
|
EXPECT_EQ(0.15f, delay_unit.Delay(0.3f));
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace webrtc
|