| Index: webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| diff --git a/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc b/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| index b6d6aaeb2e576bb7a9d5fa1c97e047333bbd9383..61238d93d4a6d1f0732009e2c119ebca1f609f79 100644
|
| --- a/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| +++ b/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| @@ -30,7 +30,7 @@ TEST(SuppressionGain, NullOutputGains) {
|
| N2.fill(0.f);
|
| float high_bands_gain;
|
| EXPECT_DEATH(SuppressionGain(DetectOptimization())
|
| - .GetGain(E2, R2, N2, false,
|
| + .GetGain(E2, R2, N2, RenderSignalAnalyzer(), false,
|
| std::vector<std::vector<float>>(
|
| 3, std::vector<float>(kBlockSize, 0.f)),
|
| false, &high_bands_gain, nullptr),
|
| @@ -42,6 +42,7 @@ TEST(SuppressionGain, NullOutputGains) {
|
| // Does a sanity check that the gains are correctly computed.
|
| TEST(SuppressionGain, BasicGainComputation) {
|
| SuppressionGain suppression_gain(DetectOptimization());
|
| + RenderSignalAnalyzer analyzer;
|
| float high_bands_gain;
|
| std::array<float, kFftLengthBy2Plus1> E2;
|
| std::array<float, kFftLengthBy2Plus1> R2;
|
| @@ -54,7 +55,8 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(0.1f);
|
| N2.fill(100.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, false, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, analyzer, false, x, false,
|
| + &high_bands_gain, &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(1.f, a, 0.001); });
|
| @@ -64,7 +66,8 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(0.1f);
|
| N2.fill(0.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, false, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, analyzer, false, x, false,
|
| + &high_bands_gain, &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(1.f, a, 0.001); });
|
| @@ -74,13 +77,15 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(10000000000000.f);
|
| N2.fill(0.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, false, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, analyzer, false, x, false,
|
| + &high_bands_gain, &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(0.f, a, 0.001); });
|
|
|
| // Verify the functionality for forcing a zero gain.
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, true, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, analyzer, false, x, true,
|
| + &high_bands_gain, &g);
|
| std::for_each(g.begin(), g.end(), [](float a) { EXPECT_FLOAT_EQ(0.f, a); });
|
| EXPECT_FLOAT_EQ(0.f, high_bands_gain);
|
| }
|
|
|