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

Side by Side Diff: webrtc/modules/audio_processing/vad/pitch_based_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
1 /* 1 /*
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 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 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 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include "webrtc/modules/audio_processing/agc/pitch_based_vad.h" 11 #include "webrtc/modules/audio_processing/vad/pitch_based_vad.h"
12 12
13 #include <math.h> 13 #include <math.h>
14 #include <stdio.h> 14 #include <stdio.h>
15 #include <string.h>
16 15
17 #include "gtest/gtest.h" 16 #include <string>
17
18 #include "testing/gtest/include/gtest/gtest.h"
18 #include "webrtc/test/testsupport/fileutils.h" 19 #include "webrtc/test/testsupport/fileutils.h"
19 20
20 namespace webrtc { 21 namespace webrtc {
21 22
22 TEST(PitchBasedVadTest, VoicingProbabilityTest) { 23 TEST(PitchBasedVadTest, VoicingProbabilityTest) {
23 std::string spectral_peak_file_name = test::ResourcePath( 24 std::string spectral_peak_file_name =
24 "audio_processing/agc/agc_spectral_peak", "dat"); 25 test::ResourcePath("audio_processing/agc/agc_spectral_peak", "dat");
25 FILE* spectral_peak_file = fopen(spectral_peak_file_name.c_str(), "rb"); 26 FILE* spectral_peak_file = fopen(spectral_peak_file_name.c_str(), "rb");
26 ASSERT_TRUE(spectral_peak_file != NULL); 27 ASSERT_TRUE(spectral_peak_file != NULL);
27 28
28 std::string pitch_gain_file_name = 29 std::string pitch_gain_file_name =
29 test::ResourcePath("audio_processing/agc/agc_pitch_gain", "dat"); 30 test::ResourcePath("audio_processing/agc/agc_pitch_gain", "dat");
30 FILE* pitch_gain_file = fopen(pitch_gain_file_name.c_str(), "rb"); 31 FILE* pitch_gain_file = fopen(pitch_gain_file_name.c_str(), "rb");
31 ASSERT_TRUE(pitch_gain_file != NULL); 32 ASSERT_TRUE(pitch_gain_file != NULL);
32 33
33 std::string pitch_lag_file_name = 34 std::string pitch_lag_file_name =
34 test::ResourcePath("audio_processing/agc/agc_pitch_lag", "dat"); 35 test::ResourcePath("audio_processing/agc/agc_pitch_lag", "dat");
35 FILE* pitch_lag_file = fopen(pitch_lag_file_name.c_str(), "rb"); 36 FILE* pitch_lag_file = fopen(pitch_lag_file_name.c_str(), "rb");
36 ASSERT_TRUE(pitch_lag_file != NULL); 37 ASSERT_TRUE(pitch_lag_file != NULL);
37 38
38 std::string voicing_prob_file_name = 39 std::string voicing_prob_file_name =
39 test::ResourcePath("audio_processing/agc/agc_voicing_prob", "dat"); 40 test::ResourcePath("audio_processing/agc/agc_voicing_prob", "dat");
40 FILE* voicing_prob_file = fopen(voicing_prob_file_name.c_str(), "rb"); 41 FILE* voicing_prob_file = fopen(voicing_prob_file_name.c_str(), "rb");
41 ASSERT_TRUE(voicing_prob_file != NULL); 42 ASSERT_TRUE(voicing_prob_file != NULL);
42 43
43 PitchBasedVad vad_; 44 PitchBasedVad vad_;
44 45
45 double reference_activity_probability; 46 double reference_activity_probability;
46 47
47 AudioFeatures audio_features; 48 AudioFeatures audio_features;
48 memset(&audio_features, 0, sizeof(audio_features)); 49 memset(&audio_features, 0, sizeof(audio_features));
49 audio_features.num_frames = 1; 50 audio_features.num_frames = 1;
50 while (fread(audio_features.spectral_peak, 51 while (fread(audio_features.spectral_peak,
51 sizeof(audio_features.spectral_peak[0]), 1, 52 sizeof(audio_features.spectral_peak[0]), 1,
52 spectral_peak_file) == 1u) { 53 spectral_peak_file) == 1u) {
53 double p; 54 double p;
54 ASSERT_EQ(1u, fread(audio_features.log_pitch_gain, sizeof( 55 ASSERT_EQ(1u, fread(audio_features.log_pitch_gain,
55 audio_features.log_pitch_gain[0]), 1, pitch_gain_file)); 56 sizeof(audio_features.log_pitch_gain[0]), 1,
56 ASSERT_EQ(1u, fread(audio_features.pitch_lag_hz, sizeof( 57 pitch_gain_file));
57 audio_features.pitch_lag_hz[0]), 1, pitch_lag_file)); 58 ASSERT_EQ(1u,
58 ASSERT_EQ(1u, fread(&reference_activity_probability, sizeof( 59 fread(audio_features.pitch_lag_hz,
59 reference_activity_probability), 1, voicing_prob_file)); 60 sizeof(audio_features.pitch_lag_hz[0]), 1, pitch_lag_file));
61 ASSERT_EQ(1u, fread(&reference_activity_probability,
62 sizeof(reference_activity_probability), 1,
63 voicing_prob_file));
60 64
61 p = 0.5; // Initialize to the neutral value for combining probabilities. 65 p = 0.5; // Initialize to the neutral value for combining probabilities.
62 EXPECT_EQ(0, vad_.VoicingProbability(audio_features, &p)); 66 EXPECT_EQ(0, vad_.VoicingProbability(audio_features, &p));
63 EXPECT_NEAR(p, reference_activity_probability, 0.01); 67 EXPECT_NEAR(p, reference_activity_probability, 0.01);
64 } 68 }
65 69
66 fclose(spectral_peak_file); 70 fclose(spectral_peak_file);
67 fclose(pitch_gain_file); 71 fclose(pitch_gain_file);
68 fclose(pitch_lag_file); 72 fclose(pitch_lag_file);
69 } 73 }
70 74
71 } // namespace webrtc 75 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/modules/audio_processing/vad/pitch_based_vad.cc ('k') | webrtc/modules/audio_processing/vad/pitch_internal.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698