Index: webrtc/modules/audio_processing/level_controller/level_controller_unittest.cc |
diff --git a/webrtc/modules/audio_processing/level_controller/level_controller_unittest.cc b/webrtc/modules/audio_processing/level_controller/level_controller_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4058db94b177bbff5e5f7071cd7a9914137e4990 |
--- /dev/null |
+++ b/webrtc/modules/audio_processing/level_controller/level_controller_unittest.cc |
@@ -0,0 +1,122 @@ |
+/* |
+ * Copyright (c) 2016 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 <vector> |
+ |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "webrtc/base/array_view.h" |
+#include "webrtc/modules/audio_processing/audio_buffer.h" |
+#include "webrtc/modules/audio_processing/include/audio_processing.h" |
+#include "webrtc/modules/audio_processing/level_controller/level_controller.h" |
+#include "webrtc/modules/audio_processing/test/audio_buffer_tools.h" |
+#include "webrtc/modules/audio_processing/test/bitexactness_tools.h" |
+ |
+namespace webrtc { |
+namespace { |
+ |
+const int kNumFramesToProcess = 1000; |
+ |
+// Processes a specified amount of frames, verifies the results and reports |
+// any errors. |
+void RunBitexactnessTest(int sample_rate_hz, |
+ size_t num_channels, |
+ rtc::ArrayView<const float> output_reference) { |
+ LevelController level_controller; |
+ level_controller.Initialize(sample_rate_hz); |
+ |
+ int samples_per_channel = rtc::CheckedDivExact(sample_rate_hz, 100); |
+ const StreamConfig capture_config(sample_rate_hz, num_channels, false); |
+ AudioBuffer capture_buffer( |
+ capture_config.num_frames(), capture_config.num_channels(), |
+ capture_config.num_frames(), capture_config.num_channels(), |
+ capture_config.num_frames()); |
+ test::InputAudioFile capture_file( |
+ test::GetApmCaptureTestVectorFileName(sample_rate_hz)); |
+ std::vector<float> capture_input(samples_per_channel * num_channels); |
+ for (size_t frame_no = 0; frame_no < kNumFramesToProcess; ++frame_no) { |
+ ReadFloatSamplesFromStereoFile(samples_per_channel, num_channels, |
+ &capture_file, capture_input); |
+ |
+ test::CopyVectorToAudioBuffer(capture_config, capture_input, |
+ &capture_buffer); |
+ |
+ level_controller.Process(&capture_buffer); |
+ } |
+ |
+ // Extract test results. |
+ std::vector<float> capture_output; |
+ test::ExtractVectorFromAudioBuffer(capture_config, &capture_buffer, |
+ &capture_output); |
+ |
+ // Compare the output with the reference. Only the first values of the output |
+ // from last frame processed are compared in order not having to specify all |
+ // preceding frames as testvectors. As the algorithm being tested has a |
+ // memory, testing only the last frame implicitly also tests the preceeding |
+ // frames. |
+ const float kVectorElementErrorBound = 1.0f / 32768.0f; |
+ EXPECT_TRUE(test::VerifyDeinterleavedArray( |
+ capture_config.num_frames(), capture_config.num_channels(), |
+ output_reference, capture_output, kVectorElementErrorBound)); |
+} |
+ |
+} // namespace |
+ |
+TEST(LevelControlBitExactnessTest, Mono8kHz) { |
+ const float kOutputReference[] = {-0.023242f, -0.020266f, -0.015097f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate8kHz, 1, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Mono16kHz) { |
+ const float kOutputReference[] = {-0.019461f, -0.018761f, -0.018481f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate16kHz, 1, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Mono32kHz) { |
+ const float kOutputReference[] = {-0.016872f, -0.019118f, -0.018722f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate32kHz, 1, kOutputReference); |
+} |
+ |
+// TODO(peah): Investigate why this particular testcase differ between Android |
+// and the rest of the platforms. |
+TEST(LevelControlBitExactnessTest, Mono48kHz) { |
+#if !(defined(WEBRTC_ARCH_ARM64) || defined(WEBRTC_ARCH_ARM) || \ |
+ defined(WEBRTC_ANDROID)) |
+ const float kOutputReference[] = {-0.016771f, -0.017831f, -0.020482f}; |
+#else |
+ const float kOutputReference[] = {-0.015949f, -0.016957f, -0.019478f}; |
+#endif |
+ RunBitexactnessTest(AudioProcessing::kSampleRate48kHz, 1, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Stereo8kHz) { |
+ const float kOutputReference[] = {-0.019304f, -0.011600f, -0.016690f, |
+ -0.071335f, -0.031849f, -0.065694f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate8kHz, 2, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Stereo16kHz) { |
+ const float kOutputReference[] = {-0.016302f, -0.007559f, -0.015668f, |
+ -0.068346f, -0.031476f, -0.066065f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate16kHz, 2, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Stereo32kHz) { |
+ const float kOutputReference[] = {-0.013944f, -0.008337f, -0.015972f, |
+ -0.063563f, -0.031233f, -0.066784f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate32kHz, 2, kOutputReference); |
+} |
+ |
+TEST(LevelControlBitExactnessTest, Stereo48kHz) { |
+ const float kOutputReference[] = {-0.013652f, -0.008125f, -0.014593f, |
+ -0.062963f, -0.030270f, -0.064727f}; |
+ RunBitexactnessTest(AudioProcessing::kSampleRate48kHz, 2, kOutputReference); |
+} |
+ |
+} // namespace webrtc |