OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 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/common_audio/blocker.h" | 11 #include "webrtc/common_audio/blocker.h" |
12 | 12 |
13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 #include "webrtc/base/arraysize.h" |
14 | 15 |
15 namespace { | 16 namespace { |
16 | 17 |
17 // Callback Function to add 3 to every sample in the signal. | 18 // Callback Function to add 3 to every sample in the signal. |
18 class PlusThreeBlockerCallback : public webrtc::BlockerCallback { | 19 class PlusThreeBlockerCallback : public webrtc::BlockerCallback { |
19 public: | 20 public: |
20 void ProcessBlock(const float* const* input, | 21 void ProcessBlock(const float* const* input, |
21 size_t num_frames, | 22 size_t num_frames, |
22 int num_input_channels, | 23 int num_input_channels, |
23 int num_output_channels, | 24 int num_output_channels, |
(...skipping 25 matching lines...) Expand all Loading... |
49 } // namespace | 50 } // namespace |
50 | 51 |
51 namespace webrtc { | 52 namespace webrtc { |
52 | 53 |
53 // Tests blocking with a window that multiplies the signal by 2, a callback | 54 // Tests blocking with a window that multiplies the signal by 2, a callback |
54 // that adds 3 to each sample in the signal, and different combinations of chunk | 55 // that adds 3 to each sample in the signal, and different combinations of chunk |
55 // size, block size, and shift amount. | 56 // size, block size, and shift amount. |
56 class BlockerTest : public ::testing::Test { | 57 class BlockerTest : public ::testing::Test { |
57 protected: | 58 protected: |
58 void RunTest(Blocker* blocker, | 59 void RunTest(Blocker* blocker, |
59 int chunk_size, | 60 size_t chunk_size, |
60 int num_frames, | 61 size_t num_frames, |
61 const float* const* input, | 62 const float* const* input, |
62 float* const* input_chunk, | 63 float* const* input_chunk, |
63 float* const* output, | 64 float* const* output, |
64 float* const* output_chunk, | 65 float* const* output_chunk, |
65 int num_input_channels, | 66 int num_input_channels, |
66 int num_output_channels) { | 67 int num_output_channels) { |
67 int start = 0; | 68 size_t start = 0; |
68 int end = chunk_size - 1; | 69 size_t end = chunk_size - 1; |
69 while (end < num_frames) { | 70 while (end < num_frames) { |
70 CopyTo(input_chunk, 0, start, num_input_channels, chunk_size, input); | 71 CopyTo(input_chunk, 0, start, num_input_channels, chunk_size, input); |
71 blocker->ProcessChunk(input_chunk, | 72 blocker->ProcessChunk(input_chunk, |
72 chunk_size, | 73 chunk_size, |
73 num_input_channels, | 74 num_input_channels, |
74 num_output_channels, | 75 num_output_channels, |
75 output_chunk); | 76 output_chunk); |
76 CopyTo(output, start, 0, num_output_channels, chunk_size, output_chunk); | 77 CopyTo(output, start, 0, num_output_channels, chunk_size, output_chunk); |
77 | 78 |
78 start = start + chunk_size; | 79 start += chunk_size; |
79 end = end + chunk_size; | 80 end += chunk_size; |
80 } | 81 } |
81 } | 82 } |
82 | 83 |
83 void ValidateSignalEquality(const float* const* expected, | 84 void ValidateSignalEquality(const float* const* expected, |
84 const float* const* actual, | 85 const float* const* actual, |
85 int num_channels, | 86 int num_channels, |
86 int num_frames) { | 87 size_t num_frames) { |
87 for (int i = 0; i < num_channels; ++i) { | 88 for (int i = 0; i < num_channels; ++i) { |
88 for (int j = 0; j < num_frames; ++j) { | 89 for (size_t j = 0; j < num_frames; ++j) { |
89 EXPECT_FLOAT_EQ(expected[i][j], actual[i][j]); | 90 EXPECT_FLOAT_EQ(expected[i][j], actual[i][j]); |
90 } | 91 } |
91 } | 92 } |
92 } | 93 } |
93 | 94 |
94 void ValidateInitialDelay(const float* const* output, | 95 void ValidateInitialDelay(const float* const* output, |
95 int num_channels, | 96 int num_channels, |
96 int num_frames, | 97 size_t num_frames, |
97 int initial_delay) { | 98 size_t initial_delay) { |
98 for (int i = 0; i < num_channels; ++i) { | 99 for (int i = 0; i < num_channels; ++i) { |
99 for (int j = 0; j < num_frames; ++j) { | 100 for (size_t j = 0; j < num_frames; ++j) { |
100 if (j < initial_delay) { | 101 if (j < initial_delay) { |
101 EXPECT_FLOAT_EQ(output[i][j], 0.f); | 102 EXPECT_FLOAT_EQ(output[i][j], 0.f); |
102 } else { | 103 } else { |
103 EXPECT_GT(output[i][j], 0.f); | 104 EXPECT_GT(output[i][j], 0.f); |
104 } | 105 } |
105 } | 106 } |
106 } | 107 } |
107 } | 108 } |
108 | 109 |
109 static void CopyTo(float* const* dst, | 110 static void CopyTo(float* const* dst, |
110 int start_index_dst, | 111 size_t start_index_dst, |
111 int start_index_src, | 112 size_t start_index_src, |
112 int num_channels, | 113 int num_channels, |
113 int num_frames, | 114 size_t num_frames, |
114 const float* const* src) { | 115 const float* const* src) { |
115 for (int i = 0; i < num_channels; ++i) { | 116 for (int i = 0; i < num_channels; ++i) { |
116 memcpy(&dst[i][start_index_dst], | 117 memcpy(&dst[i][start_index_dst], |
117 &src[i][start_index_src], | 118 &src[i][start_index_src], |
118 num_frames * sizeof(float)); | 119 num_frames * sizeof(float)); |
119 } | 120 } |
120 } | 121 } |
121 }; | 122 }; |
122 | 123 |
123 TEST_F(BlockerTest, TestBlockerMutuallyPrimeChunkandBlockSize) { | 124 TEST_F(BlockerTest, TestBlockerMutuallyPrimeChunkandBlockSize) { |
124 const int kNumInputChannels = 3; | 125 const int kNumInputChannels = 3; |
125 const int kNumOutputChannels = 2; | 126 const int kNumOutputChannels = 2; |
126 const int kNumFrames = 10; | 127 const size_t kNumFrames = 10; |
127 const int kBlockSize = 4; | 128 const size_t kBlockSize = 4; |
128 const int kChunkSize = 5; | 129 const size_t kChunkSize = 5; |
129 const int kShiftAmount = 2; | 130 const size_t kShiftAmount = 2; |
130 | 131 |
131 const float kInput[kNumInputChannels][kNumFrames] = { | 132 const float kInput[kNumInputChannels][kNumFrames] = { |
132 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, | 133 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, |
133 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, | 134 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, |
134 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; | 135 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; |
135 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); | 136 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); |
136 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); | 137 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); |
137 | 138 |
138 const float kExpectedOutput[kNumInputChannels][kNumFrames] = { | 139 const float kExpectedOutput[kNumInputChannels][kNumFrames] = { |
139 {6, 6, 12, 20, 20, 20, 20, 20, 20, 20}, | 140 {6, 6, 12, 20, 20, 20, 20, 20, 20, 20}, |
(...skipping 29 matching lines...) Expand all Loading... |
169 | 170 |
170 ValidateSignalEquality(expected_output_cb.channels(), | 171 ValidateSignalEquality(expected_output_cb.channels(), |
171 actual_output_cb.channels(), | 172 actual_output_cb.channels(), |
172 kNumOutputChannels, | 173 kNumOutputChannels, |
173 kNumFrames); | 174 kNumFrames); |
174 } | 175 } |
175 | 176 |
176 TEST_F(BlockerTest, TestBlockerMutuallyPrimeShiftAndBlockSize) { | 177 TEST_F(BlockerTest, TestBlockerMutuallyPrimeShiftAndBlockSize) { |
177 const int kNumInputChannels = 3; | 178 const int kNumInputChannels = 3; |
178 const int kNumOutputChannels = 2; | 179 const int kNumOutputChannels = 2; |
179 const int kNumFrames = 12; | 180 const size_t kNumFrames = 12; |
180 const int kBlockSize = 4; | 181 const size_t kBlockSize = 4; |
181 const int kChunkSize = 6; | 182 const size_t kChunkSize = 6; |
182 const int kShiftAmount = 3; | 183 const size_t kShiftAmount = 3; |
183 | 184 |
184 const float kInput[kNumInputChannels][kNumFrames] = { | 185 const float kInput[kNumInputChannels][kNumFrames] = { |
185 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, | 186 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, |
186 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, | 187 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, |
187 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; | 188 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; |
188 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); | 189 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); |
189 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); | 190 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); |
190 | 191 |
191 const float kExpectedOutput[kNumOutputChannels][kNumFrames] = { | 192 const float kExpectedOutput[kNumOutputChannels][kNumFrames] = { |
192 {6, 10, 10, 20, 10, 10, 20, 10, 10, 20, 10, 10}, | 193 {6, 10, 10, 20, 10, 10, 20, 10, 10, 20, 10, 10}, |
(...skipping 29 matching lines...) Expand all Loading... |
222 | 223 |
223 ValidateSignalEquality(expected_output_cb.channels(), | 224 ValidateSignalEquality(expected_output_cb.channels(), |
224 actual_output_cb.channels(), | 225 actual_output_cb.channels(), |
225 kNumOutputChannels, | 226 kNumOutputChannels, |
226 kNumFrames); | 227 kNumFrames); |
227 } | 228 } |
228 | 229 |
229 TEST_F(BlockerTest, TestBlockerNoOverlap) { | 230 TEST_F(BlockerTest, TestBlockerNoOverlap) { |
230 const int kNumInputChannels = 3; | 231 const int kNumInputChannels = 3; |
231 const int kNumOutputChannels = 2; | 232 const int kNumOutputChannels = 2; |
232 const int kNumFrames = 12; | 233 const size_t kNumFrames = 12; |
233 const int kBlockSize = 4; | 234 const size_t kBlockSize = 4; |
234 const int kChunkSize = 4; | 235 const size_t kChunkSize = 4; |
235 const int kShiftAmount = 4; | 236 const size_t kShiftAmount = 4; |
236 | 237 |
237 const float kInput[kNumInputChannels][kNumFrames] = { | 238 const float kInput[kNumInputChannels][kNumFrames] = { |
238 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, | 239 {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, |
239 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, | 240 {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2}, |
240 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; | 241 {3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3}}; |
241 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); | 242 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); |
242 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); | 243 input_cb.SetDataForTesting(kInput[0], sizeof(kInput) / sizeof(**kInput)); |
243 | 244 |
244 const float kExpectedOutput[kNumOutputChannels][kNumFrames] = { | 245 const float kExpectedOutput[kNumOutputChannels][kNumFrames] = { |
245 {10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10}, | 246 {10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10}, |
(...skipping 29 matching lines...) Expand all Loading... |
275 | 276 |
276 ValidateSignalEquality(expected_output_cb.channels(), | 277 ValidateSignalEquality(expected_output_cb.channels(), |
277 actual_output_cb.channels(), | 278 actual_output_cb.channels(), |
278 kNumOutputChannels, | 279 kNumOutputChannels, |
279 kNumFrames); | 280 kNumFrames); |
280 } | 281 } |
281 | 282 |
282 TEST_F(BlockerTest, InitialDelaysAreMinimum) { | 283 TEST_F(BlockerTest, InitialDelaysAreMinimum) { |
283 const int kNumInputChannels = 3; | 284 const int kNumInputChannels = 3; |
284 const int kNumOutputChannels = 2; | 285 const int kNumOutputChannels = 2; |
285 const int kNumFrames = 1280; | 286 const size_t kNumFrames = 1280; |
286 const int kChunkSize[] = | 287 const size_t kChunkSize[] = |
287 {80, 80, 80, 80, 80, 80, 160, 160, 160, 160, 160, 160}; | 288 {80, 80, 80, 80, 80, 80, 160, 160, 160, 160, 160, 160}; |
288 const int kBlockSize[] = | 289 const size_t kBlockSize[] = |
289 {64, 64, 64, 128, 128, 128, 128, 128, 128, 256, 256, 256}; | 290 {64, 64, 64, 128, 128, 128, 128, 128, 128, 256, 256, 256}; |
290 const int kShiftAmount[] = | 291 const size_t kShiftAmount[] = |
291 {16, 32, 64, 32, 64, 128, 32, 64, 128, 64, 128, 256}; | 292 {16, 32, 64, 32, 64, 128, 32, 64, 128, 64, 128, 256}; |
292 const int kInitialDelay[] = | 293 const size_t kInitialDelay[] = |
293 {48, 48, 48, 112, 112, 112, 96, 96, 96, 224, 224, 224}; | 294 {48, 48, 48, 112, 112, 112, 96, 96, 96, 224, 224, 224}; |
294 | 295 |
295 float input[kNumInputChannels][kNumFrames]; | 296 float input[kNumInputChannels][kNumFrames]; |
296 for (int i = 0; i < kNumInputChannels; ++i) { | 297 for (int i = 0; i < kNumInputChannels; ++i) { |
297 for (int j = 0; j < kNumFrames; ++j) { | 298 for (size_t j = 0; j < kNumFrames; ++j) { |
298 input[i][j] = i + 1; | 299 input[i][j] = i + 1; |
299 } | 300 } |
300 } | 301 } |
301 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); | 302 ChannelBuffer<float> input_cb(kNumFrames, kNumInputChannels); |
302 input_cb.SetDataForTesting(input[0], sizeof(input) / sizeof(**input)); | 303 input_cb.SetDataForTesting(input[0], sizeof(input) / sizeof(**input)); |
303 | 304 |
304 ChannelBuffer<float> output_cb(kNumFrames, kNumOutputChannels); | 305 ChannelBuffer<float> output_cb(kNumFrames, kNumOutputChannels); |
305 | 306 |
306 CopyBlockerCallback callback; | 307 CopyBlockerCallback callback; |
307 | 308 |
308 for (size_t i = 0; i < (sizeof(kChunkSize) / sizeof(*kChunkSize)); ++i) { | 309 for (size_t i = 0; i < arraysize(kChunkSize); ++i) { |
309 rtc::scoped_ptr<float[]> window(new float[kBlockSize[i]]); | 310 rtc::scoped_ptr<float[]> window(new float[kBlockSize[i]]); |
310 for (int j = 0; j < kBlockSize[i]; ++j) { | 311 for (size_t j = 0; j < kBlockSize[i]; ++j) { |
311 window[j] = 1.f; | 312 window[j] = 1.f; |
312 } | 313 } |
313 | 314 |
314 ChannelBuffer<float> input_chunk_cb(kChunkSize[i], kNumInputChannels); | 315 ChannelBuffer<float> input_chunk_cb(kChunkSize[i], kNumInputChannels); |
315 ChannelBuffer<float> output_chunk_cb(kChunkSize[i], kNumOutputChannels); | 316 ChannelBuffer<float> output_chunk_cb(kChunkSize[i], kNumOutputChannels); |
316 | 317 |
317 Blocker blocker(kChunkSize[i], | 318 Blocker blocker(kChunkSize[i], |
318 kBlockSize[i], | 319 kBlockSize[i], |
319 kNumInputChannels, | 320 kNumInputChannels, |
320 kNumOutputChannels, | 321 kNumOutputChannels, |
(...skipping 12 matching lines...) Expand all Loading... |
333 kNumOutputChannels); | 334 kNumOutputChannels); |
334 | 335 |
335 ValidateInitialDelay(output_cb.channels(), | 336 ValidateInitialDelay(output_cb.channels(), |
336 kNumOutputChannels, | 337 kNumOutputChannels, |
337 kNumFrames, | 338 kNumFrames, |
338 kInitialDelay[i]); | 339 kInitialDelay[i]); |
339 } | 340 } |
340 } | 341 } |
341 | 342 |
342 } // namespace webrtc | 343 } // namespace webrtc |
OLD | NEW |