| OLD | NEW |
| (Empty) |
| 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 "webrtc/modules/audio_processing/agc/standalone_vad.h" | |
| 12 | |
| 13 #include <string.h> | |
| 14 | |
| 15 #include "gtest/gtest.h" | |
| 16 #include "webrtc/base/scoped_ptr.h" | |
| 17 #include "webrtc/modules/interface/module_common_types.h" | |
| 18 #include "webrtc/test/testsupport/fileutils.h" | |
| 19 | |
| 20 namespace webrtc { | |
| 21 | |
| 22 TEST(StandaloneVadTest, Api) { | |
| 23 rtc::scoped_ptr<StandaloneVad> vad(StandaloneVad::Create()); | |
| 24 int16_t data[kLength10Ms] = { 0 }; | |
| 25 | |
| 26 // Valid frame length (for 32 kHz rate), but not what the VAD is expecting. | |
| 27 EXPECT_EQ(-1, vad->AddAudio(data, 320)); | |
| 28 | |
| 29 const int kMaxNumFrames = 3; | |
| 30 double p[kMaxNumFrames]; | |
| 31 for (int n = 0; n < kMaxNumFrames; n++) | |
| 32 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); | |
| 33 | |
| 34 // Pretend |p| is shorter that it should be. | |
| 35 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames - 1)); | |
| 36 | |
| 37 EXPECT_EQ(0, vad->GetActivity(p, kMaxNumFrames)); | |
| 38 | |
| 39 // Ask for activity when buffer is empty. | |
| 40 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames)); | |
| 41 | |
| 42 // Should reset and result in one buffer. | |
| 43 for (int n = 0; n < kMaxNumFrames + 1; n++) | |
| 44 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); | |
| 45 EXPECT_EQ(0, vad->GetActivity(p, 1)); | |
| 46 | |
| 47 // Wrong modes | |
| 48 EXPECT_EQ(-1, vad->set_mode(-1)); | |
| 49 EXPECT_EQ(-1, vad->set_mode(4)); | |
| 50 | |
| 51 // Valid mode. | |
| 52 const int kMode = 2; | |
| 53 EXPECT_EQ(0, vad->set_mode(kMode)); | |
| 54 EXPECT_EQ(kMode, vad->mode()); | |
| 55 } | |
| 56 | |
| 57 TEST(StandaloneVadTest, ActivityDetection) { | |
| 58 rtc::scoped_ptr<StandaloneVad> vad(StandaloneVad::Create()); | |
| 59 const size_t kDataLength = kLength10Ms; | |
| 60 int16_t data[kDataLength] = { 0 }; | |
| 61 | |
| 62 FILE* pcm_file = | |
| 63 fopen(test::ResourcePath("audio_processing/agc/agc_audio", "pcm").c_str(), | |
| 64 "rb"); | |
| 65 ASSERT_TRUE(pcm_file != NULL); | |
| 66 | |
| 67 FILE* reference_file = fopen( | |
| 68 test::ResourcePath("audio_processing/agc/agc_vad", "dat").c_str(), "rb"); | |
| 69 ASSERT_TRUE(reference_file != NULL); | |
| 70 | |
| 71 // Reference activities are prepared with 0 aggressiveness. | |
| 72 ASSERT_EQ(0, vad->set_mode(0)); | |
| 73 | |
| 74 // Stand-alone VAD can operate on 1, 2 or 3 frames of length 10 ms. The | |
| 75 // reference file is created for 30 ms frame. | |
| 76 const int kNumVadFramesToProcess = 3; | |
| 77 int num_frames = 0; | |
| 78 while (fread(data, sizeof(int16_t), kDataLength, pcm_file) == kDataLength) { | |
| 79 vad->AddAudio(data, kDataLength); | |
| 80 num_frames++; | |
| 81 if (num_frames == kNumVadFramesToProcess) { | |
| 82 num_frames = 0; | |
| 83 int referece_activity; | |
| 84 double p[kNumVadFramesToProcess]; | |
| 85 EXPECT_EQ(1u, fread(&referece_activity, sizeof(referece_activity), 1, | |
| 86 reference_file)); | |
| 87 int activity = vad->GetActivity(p, kNumVadFramesToProcess); | |
| 88 EXPECT_EQ(referece_activity, activity); | |
| 89 if (activity != 0) { | |
| 90 // When active, probabilities are set to 0.5. | |
| 91 for (int n = 0; n < kNumVadFramesToProcess; n++) | |
| 92 EXPECT_EQ(0.5, p[n]); | |
| 93 } else { | |
| 94 // When inactive, probabilities are set to 0.01. | |
| 95 for (int n = 0; n < kNumVadFramesToProcess; n++) | |
| 96 EXPECT_EQ(0.01, p[n]); | |
| 97 } | |
| 98 } | |
| 99 } | |
| 100 fclose(reference_file); | |
| 101 fclose(pcm_file); | |
| 102 } | |
| 103 } | |
| OLD | NEW |