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