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
17extern "C" {
18#include "webrtc/common_audio/vad/vad_core.h"
19#include "webrtc/common_audio/vad/vad_sp.h"
20}
21
22namespace {
23
24TEST_F(VadTest, vad_sp) {
25  VadInstT* self = reinterpret_cast<VadInstT*>(malloc(sizeof(VadInstT)));
26  const int 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 (int16_t i = 0; i < kMaxFrameLenSp; ++i) {
44    data_in[i] = (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 (int16_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