OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/modules/audio_processing/aec3/residual_echo_estimator.h" | 11 #include "webrtc/modules/audio_processing/aec3/residual_echo_estimator.h" |
12 | 12 |
13 #include "webrtc/base/random.h" | 13 #include "webrtc/base/random.h" |
14 #include "webrtc/modules/audio_processing/aec3/aec_state.h" | 14 #include "webrtc/modules/audio_processing/aec3/aec_state.h" |
15 #include "webrtc/modules/audio_processing/aec3/aec3_fft.h" | 15 #include "webrtc/modules/audio_processing/aec3/aec3_fft.h" |
16 #include "webrtc/modules/audio_processing/test/echo_canceller_test_tools.h" | 16 #include "webrtc/modules/audio_processing/test/echo_canceller_test_tools.h" |
17 #include "webrtc/test/gtest.h" | 17 #include "webrtc/test/gtest.h" |
18 | 18 |
19 namespace webrtc { | 19 namespace webrtc { |
20 | 20 |
21 #if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) | 21 #if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
22 | 22 |
23 // Verifies that the check for non-null output residual echo power works. | 23 // Verifies that the check for non-null output residual echo power works. |
24 TEST(ResidualEchoEstimator, NullResidualEchoPowerOutput) { | 24 TEST(ResidualEchoEstimator, NullResidualEchoPowerOutput) { |
25 AecState aec_state; | 25 AecState aec_state(0.f); |
26 RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 10, | 26 RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 10, |
27 std::vector<size_t>(1, 10)); | 27 std::vector<size_t>(1, 10)); |
28 std::vector<std::array<float, kFftLengthBy2Plus1>> H2; | 28 std::vector<std::array<float, kFftLengthBy2Plus1>> H2; |
29 std::array<float, kFftLengthBy2Plus1> S2_linear; | 29 std::array<float, kFftLengthBy2Plus1> S2_linear; |
30 std::array<float, kFftLengthBy2Plus1> Y2; | 30 std::array<float, kFftLengthBy2Plus1> Y2; |
31 EXPECT_DEATH(ResidualEchoEstimator().Estimate(true, aec_state, render_buffer, | 31 EXPECT_DEATH(ResidualEchoEstimator().Estimate(true, aec_state, render_buffer, |
32 S2_linear, Y2, nullptr), | 32 S2_linear, Y2, nullptr), |
33 ""); | 33 ""); |
34 } | 34 } |
35 | 35 |
36 #endif | 36 #endif |
37 | 37 |
38 TEST(ResidualEchoEstimator, BasicTest) { | 38 TEST(ResidualEchoEstimator, BasicTest) { |
39 ResidualEchoEstimator estimator; | 39 ResidualEchoEstimator estimator; |
40 AecState aec_state; | 40 AecState aec_state(0.f); |
41 RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 10, | 41 RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 10, |
42 std::vector<size_t>(1, 10)); | 42 std::vector<size_t>(1, 10)); |
43 std::array<float, kFftLengthBy2Plus1> E2_main; | 43 std::array<float, kFftLengthBy2Plus1> E2_main; |
44 std::array<float, kFftLengthBy2Plus1> E2_shadow; | 44 std::array<float, kFftLengthBy2Plus1> E2_shadow; |
45 std::array<float, kFftLengthBy2Plus1> S2_linear; | 45 std::array<float, kFftLengthBy2Plus1> S2_linear; |
46 std::array<float, kFftLengthBy2Plus1> S2_fallback; | 46 std::array<float, kFftLengthBy2Plus1> S2_fallback; |
47 std::array<float, kFftLengthBy2Plus1> Y2; | 47 std::array<float, kFftLengthBy2Plus1> Y2; |
48 std::array<float, kFftLengthBy2Plus1> R2; | 48 std::array<float, kFftLengthBy2Plus1> R2; |
49 EchoPathVariability echo_path_variability(false, false); | 49 EchoPathVariability echo_path_variability(false, false); |
50 std::vector<std::vector<float>> x(3, std::vector<float>(kBlockSize, 0.f)); | 50 std::vector<std::vector<float>> x(3, std::vector<float>(kBlockSize, 0.f)); |
(...skipping 26 matching lines...) Expand all Loading... |
77 aec_state.Update(H2, rtc::Optional<size_t>(2), render_buffer, E2_main, Y2, | 77 aec_state.Update(H2, rtc::Optional<size_t>(2), render_buffer, E2_main, Y2, |
78 x[0], false); | 78 x[0], false); |
79 | 79 |
80 estimator.Estimate(true, aec_state, render_buffer, S2_linear, Y2, &R2); | 80 estimator.Estimate(true, aec_state, render_buffer, S2_linear, Y2, &R2); |
81 } | 81 } |
82 std::for_each(R2.begin(), R2.end(), | 82 std::for_each(R2.begin(), R2.end(), |
83 [&](float a) { EXPECT_NEAR(kLevel, a, 0.1f); }); | 83 [&](float a) { EXPECT_NEAR(kLevel, a, 0.1f); }); |
84 } | 84 } |
85 | 85 |
86 } // namespace webrtc | 86 } // namespace webrtc |
OLD | NEW |