OLD | NEW |
---|---|
(Empty) | |
1 /* | |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | |
3 * | |
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 | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 #include <vector> | |
11 | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 #include "webrtc/base/array_view.h" | |
14 #include "webrtc/modules/audio_processing/audio_buffer.h" | |
15 #include "webrtc/modules/audio_processing/level_estimator_impl.h" | |
16 #include "webrtc/modules/audio_processing/test/audio_buffer_tools.h" | |
17 #include "webrtc/modules/audio_processing/test/bitexactness_tools.h" | |
18 | |
19 namespace webrtc { | |
20 namespace { | |
21 | |
22 const int kNumFramesToProcess = 1000; | |
23 | |
24 // Processes a specified amount of frames, verifies the results and reports | |
25 // any errors. | |
26 void RunBitexactnessTest(int sample_rate_hz, | |
27 size_t num_channels, | |
28 int rms_reference) { | |
29 rtc::CriticalSection crit_capture; | |
30 LevelEstimatorImpl level_estimator(&crit_capture); | |
31 level_estimator.Initialize(); | |
32 level_estimator.Enable(true); | |
33 | |
34 int samples_per_channel = rtc::CheckedDivExact(sample_rate_hz, 100); | |
35 StreamConfig capture_config(sample_rate_hz, num_channels, false); | |
36 AudioBuffer capture_buffer( | |
37 capture_config.num_frames(), capture_config.num_channels(), | |
38 capture_config.num_frames(), capture_config.num_channels(), | |
39 capture_config.num_frames()); | |
40 | |
41 test::InputAudioFile capture_file( | |
42 test::GetApmCaptureTestVectorFileName(sample_rate_hz)); | |
43 std::vector<float> capture_input(samples_per_channel * num_channels); | |
44 for (size_t frame_no = 0; frame_no < kNumFramesToProcess; ++frame_no) { | |
45 ReadFloatSamplesFromStereoFile(samples_per_channel, num_channels, | |
46 &capture_file, capture_input); | |
47 | |
48 test::CopyVectorToAudioBuffer(capture_config, capture_input, | |
49 &capture_buffer); | |
50 | |
51 level_estimator.ProcessStream(&capture_buffer); | |
52 } | |
53 | |
54 // Extract test results. | |
55 int rms = level_estimator.RMS(); | |
hlundin-webrtc
2016/03/17 15:20:57
Skip this line and simply do a one-liner:
EXPECT_E
| |
56 | |
57 // Compare the output to the reference. | |
58 EXPECT_EQ(rms_reference, rms); | |
59 } | |
60 | |
61 } // namespace | |
62 | |
63 TEST(LevelEstimatorBitExactnessTest, Mono8kHz) { | |
64 const int kRmsReference = 31; | |
65 | |
66 RunBitexactnessTest(8000, 1, kRmsReference); | |
67 } | |
68 | |
69 TEST(LevelEstimatorBitExactnessTest, Mono16kHz) { | |
70 const int kRmsReference = 31; | |
71 | |
72 RunBitexactnessTest(16000, 1, kRmsReference); | |
73 } | |
74 | |
75 TEST(LevelEstimatorBitExactnessTest, Mono32kHz) { | |
76 const int kRmsReference = 31; | |
77 | |
78 RunBitexactnessTest(32000, 1, kRmsReference); | |
79 } | |
80 | |
81 TEST(LevelEstimatorBitExactnessTest, Mono48kHz) { | |
82 const int kRmsReference = 31; | |
83 | |
84 RunBitexactnessTest(48000, 1, kRmsReference); | |
85 } | |
86 | |
87 TEST(LevelEstimatorBitExactnessTest, Stereo16kHz) { | |
88 const int kRmsReference = 30; | |
89 | |
90 RunBitexactnessTest(16000, 2, kRmsReference); | |
91 } | |
92 | |
93 } // namespace webrtc | |
OLD | NEW |