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

Side by Side Diff: webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc

Issue 3005483002: Replace remaining gflags usages with rtc_base/flags (Closed)
Patch Set: Rebase Created 3 years, 3 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) 2014 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2014 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 "gflags/gflags.h"
12 #include "webrtc/common_audio/channel_buffer.h" 11 #include "webrtc/common_audio/channel_buffer.h"
13 #include "webrtc/common_audio/include/audio_util.h" 12 #include "webrtc/common_audio/include/audio_util.h"
14 #include "webrtc/common_audio/wav_file.h" 13 #include "webrtc/common_audio/wav_file.h"
15 #include "webrtc/modules/audio_processing/audio_buffer.h" 14 #include "webrtc/modules/audio_processing/audio_buffer.h"
16 #include "webrtc/modules/audio_processing/intelligibility/intelligibility_enhanc er.h" 15 #include "webrtc/modules/audio_processing/intelligibility/intelligibility_enhanc er.h"
17 #include "webrtc/modules/audio_processing/noise_suppression_impl.h" 16 #include "webrtc/modules/audio_processing/noise_suppression_impl.h"
18 #include "webrtc/rtc_base/criticalsection.h" 17 #include "webrtc/rtc_base/criticalsection.h"
19 #include "webrtc/test/gtest.h" 18 #include "webrtc/rtc_base/flags.h"
20 19
21 using std::complex; 20 using std::complex;
22 21
23 namespace webrtc { 22 namespace webrtc {
24 namespace { 23 namespace {
25 24
26 DEFINE_string(clear_file, "speech.wav", "Input file with clear speech."); 25 DEFINE_string(clear_file, "speech.wav", "Input file with clear speech.");
27 DEFINE_string(noise_file, "noise.wav", "Input file with noise data."); 26 DEFINE_string(noise_file, "noise.wav", "Input file with noise data.");
28 DEFINE_string(out_file, "proc_enhanced.wav", "Enhanced output file."); 27 DEFINE_string(out_file, "proc_enhanced.wav", "Enhanced output file.");
28 DEFINE_bool(help, false, "Print this message.");
29 29
30 // void function for gtest 30 int int_main(int argc, char* argv[]) {
31 void void_main(int argc, char* argv[]) { 31 if (rtc::FlagList::SetFlagsFromCommandLine(&argc, argv, true)) {
32 google::SetUsageMessage( 32 return 1;
33 "\n\nInput files must be little-endian 16-bit signed raw PCM.\n"); 33 }
34 google::ParseCommandLineFlags(&argc, &argv, true); 34 if (FLAG_help) {
35 rtc::FlagList::Print(nullptr, false);
36 return 0;
37 }
38 if (argc != 1) {
39 printf("\n\nInput files must be little-endian 16-bit signed raw PCM.\n");
40 return 0;
41 }
35 42
36 WavReader in_file(FLAGS_clear_file); 43 WavReader in_file(FLAG_clear_file);
37 WavReader noise_file(FLAGS_noise_file); 44 WavReader noise_file(FLAG_noise_file);
38 WavWriter out_file(FLAGS_out_file, in_file.sample_rate(), 45 WavWriter out_file(FLAG_out_file, in_file.sample_rate(),
39 in_file.num_channels()); 46 in_file.num_channels());
40 rtc::CriticalSection crit; 47 rtc::CriticalSection crit;
41 NoiseSuppressionImpl ns(&crit); 48 NoiseSuppressionImpl ns(&crit);
42 IntelligibilityEnhancer enh(in_file.sample_rate(), in_file.num_channels(), 1u, 49 IntelligibilityEnhancer enh(in_file.sample_rate(), in_file.num_channels(), 1u,
43 NoiseSuppressionImpl::num_noise_bins()); 50 NoiseSuppressionImpl::num_noise_bins());
44 ns.Initialize(noise_file.num_channels(), noise_file.sample_rate()); 51 ns.Initialize(noise_file.num_channels(), noise_file.sample_rate());
45 ns.Enable(true); 52 ns.Enable(true);
46 const size_t in_samples = noise_file.sample_rate() / 100; 53 const size_t in_samples = noise_file.sample_rate() / 100;
47 const size_t noise_samples = noise_file.sample_rate() / 100; 54 const size_t noise_samples = noise_file.sample_rate() / 100;
48 std::vector<float> in(in_samples * in_file.num_channels()); 55 std::vector<float> in(in_samples * in_file.num_channels());
(...skipping 21 matching lines...) Expand all
70 ns.AnalyzeCaptureAudio(&capture_audio); 77 ns.AnalyzeCaptureAudio(&capture_audio);
71 ns.ProcessCaptureAudio(&capture_audio); 78 ns.ProcessCaptureAudio(&capture_audio);
72 enh.SetCaptureNoiseEstimate(ns.NoiseEstimate(), 1); 79 enh.SetCaptureNoiseEstimate(ns.NoiseEstimate(), 1);
73 enh.ProcessRenderAudio(&render_audio); 80 enh.ProcessRenderAudio(&render_audio);
74 render_audio.CopyTo(in_config, in_buf.channels()); 81 render_audio.CopyTo(in_config, in_buf.channels());
75 Interleave(in_buf.channels(), in_buf.num_frames(), in_buf.num_channels(), 82 Interleave(in_buf.channels(), in_buf.num_frames(), in_buf.num_channels(),
76 in.data()); 83 in.data());
77 FloatToFloatS16(in.data(), in.size(), in.data()); 84 FloatToFloatS16(in.data(), in.size(), in.data());
78 out_file.WriteSamples(in.data(), in.size()); 85 out_file.WriteSamples(in.data(), in.size());
79 } 86 }
87
88 return 0;
80 } 89 }
81 90
82 } // namespace 91 } // namespace
83 } // namespace webrtc 92 } // namespace webrtc
84 93
85 int main(int argc, char* argv[]) { 94 int main(int argc, char* argv[]) {
86 webrtc::void_main(argc, argv); 95 return webrtc::int_main(argc, argv);
87 return 0;
88 } 96 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698