Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(67)

Side by Side Diff: webrtc/modules/audio_processing/agc/standalone_vad_unittest.cc

Issue 1181933002: Pull the Voice Activity Detector out from the AGC (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Remove unused files from isolate Created 5 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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 }
OLDNEW
« no previous file with comments | « webrtc/modules/audio_processing/agc/standalone_vad.cc ('k') | webrtc/modules/audio_processing/agc/voice_gmm_tables.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698