vad_sp_unittest.cc (2636B)
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 #include <cstdint> 12 #include <cstdlib> 13 14 #include "common_audio/vad/vad_unittest.h" 15 #include "test/gtest.h" 16 17 extern "C" { 18 #include "common_audio/vad/vad_core.h" 19 #include "common_audio/vad/vad_sp.h" 20 } 21 22 namespace webrtc { 23 namespace test { 24 25 TEST_F(VadTest, vad_sp) { 26 VadInstT* self = reinterpret_cast<VadInstT*>(malloc(sizeof(VadInstT))); 27 const size_t kMaxFrameLenSp = 960; // Maximum frame length in this unittest. 28 int16_t zeros[kMaxFrameLenSp] = {0}; 29 int32_t state[2] = {0}; 30 int16_t data_in[kMaxFrameLenSp]; 31 int16_t data_out[kMaxFrameLenSp]; 32 33 // We expect the first value to be 1600 as long as `frame_counter` is zero, 34 // which is true for the first iteration. 35 static const int16_t kReferenceMin[32] = { 36 1600, 720, 509, 512, 532, 552, 570, 588, 606, 624, 642, 37 659, 675, 691, 707, 723, 1600, 544, 502, 522, 542, 561, 38 579, 597, 615, 633, 651, 667, 683, 699, 715, 731}; 39 40 // Construct a speech signal that will trigger the VAD in all modes. It is 41 // known that (i * i) will wrap around, but that doesn't matter in this case. 42 for (size_t i = 0; i < kMaxFrameLenSp; ++i) { 43 data_in[i] = static_cast<int16_t>(i * i); 44 } 45 // Input values all zeros, expect all zeros out. 46 WebRtcVad_Downsampling(zeros, data_out, state, kMaxFrameLenSp); 47 EXPECT_EQ(0, state[0]); 48 EXPECT_EQ(0, state[1]); 49 for (size_t i = 0; i < kMaxFrameLenSp / 2; ++i) { 50 EXPECT_EQ(0, data_out[i]); 51 } 52 // Make a simple non-zero data test. 53 WebRtcVad_Downsampling(data_in, data_out, state, kMaxFrameLenSp); 54 EXPECT_EQ(207, state[0]); 55 EXPECT_EQ(2270, state[1]); 56 57 ASSERT_EQ(0, WebRtcVad_InitCore(self)); 58 // TODO(bjornv): Replace this part of the test with taking values from an 59 // array and calculate the reference value here. Make sure the values are not 60 // ordered. 61 for (int16_t i = 0; i < 16; ++i) { 62 int16_t value = 500 * (i + 1); 63 for (int j = 0; j < kNumChannels; ++j) { 64 // Use values both above and below initialized value. 65 EXPECT_EQ(kReferenceMin[i], WebRtcVad_FindMinimum(self, value, j)); 66 EXPECT_EQ(kReferenceMin[i + 16], WebRtcVad_FindMinimum(self, 12000, j)); 67 } 68 self->frame_counter++; 69 } 70 71 free(self); 72 } 73 } // namespace test 74 } // namespace webrtc