1 /*
2 * Copyright (c) 2012 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
11 // Unit tests for Accelerate and PreemptiveExpand classes.
12
13 #include "modules/audio_coding/neteq/accelerate.h"
14 #include "modules/audio_coding/neteq/preemptive_expand.h"
15
16 #include <map>
17 #include <memory>
18
19 #include "common_audio/signal_processing/include/signal_processing_library.h"
20 #include "modules/audio_coding/neteq/background_noise.h"
21 #include "modules/audio_coding/neteq/tools/input_audio_file.h"
22 #include "rtc_base/checks.h"
23 #include "test/gtest.h"
24 #include "test/testsupport/fileutils.h"
25
26 namespace webrtc {
27
28 namespace {
29 const size_t kNumChannels = 1;
30 }
31
TEST(TimeStretch,CreateAndDestroy)32 TEST(TimeStretch, CreateAndDestroy) {
33 const int kSampleRate = 8000;
34 const int kOverlapSamples = 5 * kSampleRate / 8000;
35 BackgroundNoise bgn(kNumChannels);
36 Accelerate accelerate(kSampleRate, kNumChannels, bgn);
37 PreemptiveExpand preemptive_expand(
38 kSampleRate, kNumChannels, bgn, kOverlapSamples);
39 }
40
TEST(TimeStretch,CreateUsingFactory)41 TEST(TimeStretch, CreateUsingFactory) {
42 const int kSampleRate = 8000;
43 const int kOverlapSamples = 5 * kSampleRate / 8000;
44 BackgroundNoise bgn(kNumChannels);
45
46 AccelerateFactory accelerate_factory;
47 Accelerate* accelerate =
48 accelerate_factory.Create(kSampleRate, kNumChannels, bgn);
49 EXPECT_TRUE(accelerate != NULL);
50 delete accelerate;
51
52 PreemptiveExpandFactory preemptive_expand_factory;
53 PreemptiveExpand* preemptive_expand = preemptive_expand_factory.Create(
54 kSampleRate, kNumChannels, bgn, kOverlapSamples);
55 EXPECT_TRUE(preemptive_expand != NULL);
56 delete preemptive_expand;
57 }
58
59 class TimeStretchTest : public ::testing::Test {
60 protected:
TimeStretchTest()61 TimeStretchTest()
62 : input_file_(new test::InputAudioFile(
63 test::ResourcePath("audio_coding/testfile32kHz", "pcm"))),
64 sample_rate_hz_(32000),
65 block_size_(30 * sample_rate_hz_ / 1000), // 30 ms
66 audio_(new int16_t[block_size_]),
67 background_noise_(kNumChannels) {
68 WebRtcSpl_Init();
69 }
70
Next30Ms()71 const int16_t* Next30Ms() {
72 RTC_CHECK(input_file_->Read(block_size_, audio_.get()));
73 return audio_.get();
74 }
75
76 // Returns the total length change (in samples) that the accelerate operation
77 // resulted in during the run.
TestAccelerate(size_t loops,bool fast_mode)78 size_t TestAccelerate(size_t loops, bool fast_mode) {
79 Accelerate accelerate(sample_rate_hz_, kNumChannels, background_noise_);
80 size_t total_length_change = 0;
81 for (size_t i = 0; i < loops; ++i) {
82 AudioMultiVector output(kNumChannels);
83 size_t length_change;
84 UpdateReturnStats(accelerate.Process(Next30Ms(), block_size_, fast_mode,
85 &output, &length_change));
86 total_length_change += length_change;
87 }
88 return total_length_change;
89 }
90
UpdateReturnStats(TimeStretch::ReturnCodes ret)91 void UpdateReturnStats(TimeStretch::ReturnCodes ret) {
92 switch (ret) {
93 case TimeStretch::kSuccess:
94 case TimeStretch::kSuccessLowEnergy:
95 case TimeStretch::kNoStretch:
96 ++return_stats_[ret];
97 break;
98 case TimeStretch::kError:
99 FAIL() << "Process returned an error";
100 }
101 }
102
103 std::unique_ptr<test::InputAudioFile> input_file_;
104 const int sample_rate_hz_;
105 const size_t block_size_;
106 std::unique_ptr<int16_t[]> audio_;
107 std::map<TimeStretch::ReturnCodes, int> return_stats_;
108 BackgroundNoise background_noise_;
109 };
110
TEST_F(TimeStretchTest,Accelerate)111 TEST_F(TimeStretchTest, Accelerate) {
112 // TestAccelerate returns the total length change in samples.
113 EXPECT_EQ(15268U, TestAccelerate(100, false));
114 EXPECT_EQ(9, return_stats_[TimeStretch::kSuccess]);
115 EXPECT_EQ(58, return_stats_[TimeStretch::kSuccessLowEnergy]);
116 EXPECT_EQ(33, return_stats_[TimeStretch::kNoStretch]);
117 }
118
TEST_F(TimeStretchTest,AccelerateFastMode)119 TEST_F(TimeStretchTest, AccelerateFastMode) {
120 // TestAccelerate returns the total length change in samples.
121 EXPECT_EQ(21400U, TestAccelerate(100, true));
122 EXPECT_EQ(31, return_stats_[TimeStretch::kSuccess]);
123 EXPECT_EQ(58, return_stats_[TimeStretch::kSuccessLowEnergy]);
124 EXPECT_EQ(11, return_stats_[TimeStretch::kNoStretch]);
125 }
126
127 } // namespace webrtc
128