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

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

Issue 1192863006: Revert "Pull the Voice Activity Detector out from the AGC" (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: 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/vad/pitch_based_vad.h" 11 #include "webrtc/modules/audio_processing/agc/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>
15 16
16 #include <string> 17 #include "gtest/gtest.h"
17
18 #include "testing/gtest/include/gtest/gtest.h"
19 #include "webrtc/test/testsupport/fileutils.h" 18 #include "webrtc/test/testsupport/fileutils.h"
20 19
21 namespace webrtc { 20 namespace webrtc {
22 21
23 TEST(PitchBasedVadTest, VoicingProbabilityTest) { 22 TEST(PitchBasedVadTest, VoicingProbabilityTest) {
24 std::string spectral_peak_file_name = 23 std::string spectral_peak_file_name = test::ResourcePath(
25 test::ResourcePath("audio_processing/agc/agc_spectral_peak", "dat"); 24 "audio_processing/agc/agc_spectral_peak", "dat");
26 FILE* spectral_peak_file = fopen(spectral_peak_file_name.c_str(), "rb"); 25 FILE* spectral_peak_file = fopen(spectral_peak_file_name.c_str(), "rb");
27 ASSERT_TRUE(spectral_peak_file != NULL); 26 ASSERT_TRUE(spectral_peak_file != NULL);
28 27
29 std::string pitch_gain_file_name = 28 std::string pitch_gain_file_name =
30 test::ResourcePath("audio_processing/agc/agc_pitch_gain", "dat"); 29 test::ResourcePath("audio_processing/agc/agc_pitch_gain", "dat");
31 FILE* pitch_gain_file = fopen(pitch_gain_file_name.c_str(), "rb"); 30 FILE* pitch_gain_file = fopen(pitch_gain_file_name.c_str(), "rb");
32 ASSERT_TRUE(pitch_gain_file != NULL); 31 ASSERT_TRUE(pitch_gain_file != NULL);
33 32
34 std::string pitch_lag_file_name = 33 std::string pitch_lag_file_name =
35 test::ResourcePath("audio_processing/agc/agc_pitch_lag", "dat"); 34 test::ResourcePath("audio_processing/agc/agc_pitch_lag", "dat");
36 FILE* pitch_lag_file = fopen(pitch_lag_file_name.c_str(), "rb"); 35 FILE* pitch_lag_file = fopen(pitch_lag_file_name.c_str(), "rb");
37 ASSERT_TRUE(pitch_lag_file != NULL); 36 ASSERT_TRUE(pitch_lag_file != NULL);
38 37
39 std::string voicing_prob_file_name = 38 std::string voicing_prob_file_name =
40 test::ResourcePath("audio_processing/agc/agc_voicing_prob", "dat"); 39 test::ResourcePath("audio_processing/agc/agc_voicing_prob", "dat");
41 FILE* voicing_prob_file = fopen(voicing_prob_file_name.c_str(), "rb"); 40 FILE* voicing_prob_file = fopen(voicing_prob_file_name.c_str(), "rb");
42 ASSERT_TRUE(voicing_prob_file != NULL); 41 ASSERT_TRUE(voicing_prob_file != NULL);
43 42
44 PitchBasedVad vad_; 43 PitchBasedVad vad_;
45 44
46 double reference_activity_probability; 45 double reference_activity_probability;
47 46
48 AudioFeatures audio_features; 47 AudioFeatures audio_features;
49 memset(&audio_features, 0, sizeof(audio_features)); 48 memset(&audio_features, 0, sizeof(audio_features));
50 audio_features.num_frames = 1; 49 audio_features.num_frames = 1;
51 while (fread(audio_features.spectral_peak, 50 while (fread(audio_features.spectral_peak,
52 sizeof(audio_features.spectral_peak[0]), 1, 51 sizeof(audio_features.spectral_peak[0]), 1,
53 spectral_peak_file) == 1u) { 52 spectral_peak_file) == 1u) {
54 double p; 53 double p;
55 ASSERT_EQ(1u, fread(audio_features.log_pitch_gain, 54 ASSERT_EQ(1u, fread(audio_features.log_pitch_gain, sizeof(
56 sizeof(audio_features.log_pitch_gain[0]), 1, 55 audio_features.log_pitch_gain[0]), 1, pitch_gain_file));
57 pitch_gain_file)); 56 ASSERT_EQ(1u, fread(audio_features.pitch_lag_hz, sizeof(
58 ASSERT_EQ(1u, 57 audio_features.pitch_lag_hz[0]), 1, pitch_lag_file));
59 fread(audio_features.pitch_lag_hz, 58 ASSERT_EQ(1u, fread(&reference_activity_probability, sizeof(
60 sizeof(audio_features.pitch_lag_hz[0]), 1, pitch_lag_file)); 59 reference_activity_probability), 1, voicing_prob_file));
61 ASSERT_EQ(1u, fread(&reference_activity_probability,
62 sizeof(reference_activity_probability), 1,
63 voicing_prob_file));
64 60
65 p = 0.5; // Initialize to the neutral value for combining probabilities. 61 p = 0.5; // Initialize to the neutral value for combining probabilities.
66 EXPECT_EQ(0, vad_.VoicingProbability(audio_features, &p)); 62 EXPECT_EQ(0, vad_.VoicingProbability(audio_features, &p));
67 EXPECT_NEAR(p, reference_activity_probability, 0.01); 63 EXPECT_NEAR(p, reference_activity_probability, 0.01);
68 } 64 }
69 65
70 fclose(spectral_peak_file); 66 fclose(spectral_peak_file);
71 fclose(pitch_gain_file); 67 fclose(pitch_gain_file);
72 fclose(pitch_lag_file); 68 fclose(pitch_lag_file);
73 } 69 }
74 70
75 } // namespace webrtc 71 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/modules/audio_processing/agc/pitch_based_vad.cc ('k') | webrtc/modules/audio_processing/agc/pitch_internal.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698