1 /*
2 * Copyright (c) 2013 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 #include "testing/gmock/include/gmock/gmock.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13 #include "webrtc/common_audio/include/audio_util.h"
14 #include "webrtc/typedefs.h"
15
16 namespace webrtc {
17 namespace {
18
19 using ::testing::ElementsAreArray;
20
ExpectArraysEq(const int16_t * ref,const int16_t * test,size_t length)21 void ExpectArraysEq(const int16_t* ref, const int16_t* test, size_t length) {
22 for (size_t i = 0; i < length; ++i) {
23 EXPECT_EQ(ref[i], test[i]);
24 }
25 }
26
ExpectArraysEq(const float * ref,const float * test,size_t length)27 void ExpectArraysEq(const float* ref, const float* test, size_t length) {
28 for (size_t i = 0; i < length; ++i) {
29 EXPECT_FLOAT_EQ(ref[i], test[i]);
30 }
31 }
32
TEST(AudioUtilTest,FloatToS16)33 TEST(AudioUtilTest, FloatToS16) {
34 const size_t kSize = 9;
35 const float kInput[kSize] = {0.f,
36 0.4f / 32767.f,
37 0.6f / 32767.f,
38 -0.4f / 32768.f,
39 -0.6f / 32768.f,
40 1.f,
41 -1.f,
42 1.1f,
43 -1.1f};
44 const int16_t kReference[kSize] = {0, 0, 1, 0, -1,
45 32767, -32768, 32767, -32768};
46 int16_t output[kSize];
47 FloatToS16(kInput, kSize, output);
48 ExpectArraysEq(kReference, output, kSize);
49 }
50
TEST(AudioUtilTest,S16ToFloat)51 TEST(AudioUtilTest, S16ToFloat) {
52 const size_t kSize = 7;
53 const int16_t kInput[kSize] = {0, 1, -1, 16384, -16384, 32767, -32768};
54 const float kReference[kSize] = {
55 0.f, 1.f / 32767.f, -1.f / 32768.f, 16384.f / 32767.f, -0.5f, 1.f, -1.f};
56 float output[kSize];
57 S16ToFloat(kInput, kSize, output);
58 ExpectArraysEq(kReference, output, kSize);
59 }
60
TEST(AudioUtilTest,FloatS16ToS16)61 TEST(AudioUtilTest, FloatS16ToS16) {
62 const size_t kSize = 7;
63 const float kInput[kSize] = {0.f, 0.4f, 0.5f, -0.4f,
64 -0.5f, 32768.f, -32769.f};
65 const int16_t kReference[kSize] = {0, 0, 1, 0, -1, 32767, -32768};
66 int16_t output[kSize];
67 FloatS16ToS16(kInput, kSize, output);
68 ExpectArraysEq(kReference, output, kSize);
69 }
70
TEST(AudioUtilTest,FloatToFloatS16)71 TEST(AudioUtilTest, FloatToFloatS16) {
72 const size_t kSize = 9;
73 const float kInput[kSize] = {0.f,
74 0.4f / 32767.f,
75 0.6f / 32767.f,
76 -0.4f / 32768.f,
77 -0.6f / 32768.f,
78 1.f,
79 -1.f,
80 1.1f,
81 -1.1f};
82 const float kReference[kSize] = {0.f, 0.4f, 0.6f, -0.4f, -0.6f,
83 32767.f, -32768.f, 36043.7f, -36044.8f};
84 float output[kSize];
85 FloatToFloatS16(kInput, kSize, output);
86 ExpectArraysEq(kReference, output, kSize);
87 }
88
TEST(AudioUtilTest,FloatS16ToFloat)89 TEST(AudioUtilTest, FloatS16ToFloat) {
90 const size_t kSize = 9;
91 const float kInput[kSize] = {0.f, 0.4f, 0.6f, -0.4f, -0.6f,
92 32767.f, -32768.f, 36043.7f, -36044.8f};
93 const float kReference[kSize] = {0.f,
94 0.4f / 32767.f,
95 0.6f / 32767.f,
96 -0.4f / 32768.f,
97 -0.6f / 32768.f,
98 1.f,
99 -1.f,
100 1.1f,
101 -1.1f};
102 float output[kSize];
103 FloatS16ToFloat(kInput, kSize, output);
104 ExpectArraysEq(kReference, output, kSize);
105 }
106
TEST(AudioUtilTest,InterleavingStereo)107 TEST(AudioUtilTest, InterleavingStereo) {
108 const int16_t kInterleaved[] = {2, 3, 4, 9, 8, 27, 16, 81};
109 const size_t kSamplesPerChannel = 4;
110 const int kNumChannels = 2;
111 const size_t kLength = kSamplesPerChannel * kNumChannels;
112 int16_t left[kSamplesPerChannel], right[kSamplesPerChannel];
113 int16_t* deinterleaved[] = {left, right};
114 Deinterleave(kInterleaved, kSamplesPerChannel, kNumChannels, deinterleaved);
115 const int16_t kRefLeft[] = {2, 4, 8, 16};
116 const int16_t kRefRight[] = {3, 9, 27, 81};
117 ExpectArraysEq(kRefLeft, left, kSamplesPerChannel);
118 ExpectArraysEq(kRefRight, right, kSamplesPerChannel);
119
120 int16_t interleaved[kLength];
121 Interleave(deinterleaved, kSamplesPerChannel, kNumChannels, interleaved);
122 ExpectArraysEq(kInterleaved, interleaved, kLength);
123 }
124
TEST(AudioUtilTest,InterleavingMonoIsIdentical)125 TEST(AudioUtilTest, InterleavingMonoIsIdentical) {
126 const int16_t kInterleaved[] = {1, 2, 3, 4, 5};
127 const size_t kSamplesPerChannel = 5;
128 const int kNumChannels = 1;
129 int16_t mono[kSamplesPerChannel];
130 int16_t* deinterleaved[] = {mono};
131 Deinterleave(kInterleaved, kSamplesPerChannel, kNumChannels, deinterleaved);
132 ExpectArraysEq(kInterleaved, mono, kSamplesPerChannel);
133
134 int16_t interleaved[kSamplesPerChannel];
135 Interleave(deinterleaved, kSamplesPerChannel, kNumChannels, interleaved);
136 ExpectArraysEq(mono, interleaved, kSamplesPerChannel);
137 }
138
TEST(AudioUtilTest,DownmixInterleavedToMono)139 TEST(AudioUtilTest, DownmixInterleavedToMono) {
140 {
141 const size_t kNumFrames = 4;
142 const int kNumChannels = 1;
143 const int16_t interleaved[kNumChannels * kNumFrames] = {1, 2, -1, -3};
144 int16_t deinterleaved[kNumFrames];
145
146 DownmixInterleavedToMono(interleaved, kNumFrames, kNumChannels,
147 deinterleaved);
148
149 EXPECT_THAT(deinterleaved, ElementsAreArray(interleaved));
150 }
151 {
152 const size_t kNumFrames = 2;
153 const int kNumChannels = 2;
154 const int16_t interleaved[kNumChannels * kNumFrames] = {10, 20, -10, -30};
155 int16_t deinterleaved[kNumFrames];
156
157 DownmixInterleavedToMono(interleaved, kNumFrames, kNumChannels,
158 deinterleaved);
159 const int16_t expected[kNumFrames] = {15, -20};
160
161 EXPECT_THAT(deinterleaved, ElementsAreArray(expected));
162 }
163 {
164 const size_t kNumFrames = 3;
165 const int kNumChannels = 3;
166 const int16_t interleaved[kNumChannels * kNumFrames] = {
167 30000, 30000, 24001, -5, -10, -20, -30000, -30999, -30000};
168 int16_t deinterleaved[kNumFrames];
169
170 DownmixInterleavedToMono(interleaved, kNumFrames, kNumChannels,
171 deinterleaved);
172 const int16_t expected[kNumFrames] = {28000, -11, -30333};
173
174 EXPECT_THAT(deinterleaved, ElementsAreArray(expected));
175 }
176 }
177
TEST(AudioUtilTest,DownmixToMonoTest)178 TEST(AudioUtilTest, DownmixToMonoTest) {
179 {
180 const size_t kNumFrames = 4;
181 const int kNumChannels = 1;
182 const float input_data[kNumChannels][kNumFrames] = {{1.f, 2.f, -1.f, -3.f}};
183 const float* input[kNumChannels];
184 for (int i = 0; i < kNumChannels; ++i) {
185 input[i] = input_data[i];
186 }
187
188 float downmixed[kNumFrames];
189
190 DownmixToMono<float, float>(input, kNumFrames, kNumChannels, downmixed);
191
192 EXPECT_THAT(downmixed, ElementsAreArray(input_data[0]));
193 }
194 {
195 const size_t kNumFrames = 3;
196 const int kNumChannels = 2;
197 const float input_data[kNumChannels][kNumFrames] = {{1.f, 2.f, -1.f},
198 {3.f, 0.f, 1.f}};
199 const float* input[kNumChannels];
200 for (int i = 0; i < kNumChannels; ++i) {
201 input[i] = input_data[i];
202 }
203
204 float downmixed[kNumFrames];
205 const float expected[kNumFrames] = {2.f, 1.f, 0.f};
206
207 DownmixToMono<float, float>(input, kNumFrames, kNumChannels, downmixed);
208
209 EXPECT_THAT(downmixed, ElementsAreArray(expected));
210 }
211 {
212 const size_t kNumFrames = 3;
213 const int kNumChannels = 3;
214 const int16_t input_data[kNumChannels][kNumFrames] = {
215 {30000, -5, -30000}, {30000, -10, -30999}, {24001, -20, -30000}};
216 const int16_t* input[kNumChannels];
217 for (int i = 0; i < kNumChannels; ++i) {
218 input[i] = input_data[i];
219 }
220
221 int16_t downmixed[kNumFrames];
222 const int16_t expected[kNumFrames] = {28000, -11, -30333};
223
224 DownmixToMono<int16_t, int32_t>(input, kNumFrames, kNumChannels, downmixed);
225
226 EXPECT_THAT(downmixed, ElementsAreArray(expected));
227 }
228 }
229
230 } // namespace
231 } // namespace webrtc
232