blob: f0e57a90b9cdba9304eab4706b8b2feebf53604e [file] [log] [blame]
andrew@webrtc.org08df9b22014-12-16 20:57:15 +00001/*
2 * Copyright (c) 2014 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
Ivo Creusen2cb41052018-03-15 12:22:52 +010011#include <string.h>
12
aluebsb0ad43b2015-11-20 00:11:53 -080013#include <iostream>
kwiberg62eaacf2016-02-17 06:39:05 -080014#include <memory>
Ivo Creusen2cb41052018-03-15 12:22:52 +010015#include <string>
16#include <utility>
Alessio Bazzica68170382018-11-20 12:27:20 +010017#include <vector>
andrew@webrtc.org08df9b22014-12-16 20:57:15 +000018
Alessio Bazzica68170382018-11-20 12:27:20 +010019#include "absl/strings/string_view.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020020#include "modules/audio_processing/include/audio_processing.h"
21#include "modules/audio_processing/test/aec_dump_based_simulator.h"
22#include "modules/audio_processing/test/audio_processing_simulator.h"
Ivo Creusen2cb41052018-03-15 12:22:52 +010023#include "modules/audio_processing/test/audioproc_float_impl.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020024#include "modules/audio_processing/test/wav_based_simulator.h"
Alessio Bazzica68170382018-11-20 12:27:20 +010025#include "rtc_base/checks.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020026#include "rtc_base/flags.h"
Alessio Bazzica68170382018-11-20 12:27:20 +010027#include "rtc_base/strings/string_builder.h"
andrew@webrtc.org08df9b22014-12-16 20:57:15 +000028
peah60a189f2016-05-24 20:54:40 -070029namespace webrtc {
30namespace test {
Peter Kasting69558702016-01-12 16:26:35 -080031namespace {
32
peah60a189f2016-05-24 20:54:40 -070033const int kParameterNotSpecifiedValue = -10000;
34
35const char kUsageDescription[] =
36 "Usage: audioproc_f [options] -i <input.wav>\n"
37 " or\n"
38 " audioproc_f [options] -dump_input <aec_dump>\n"
39 "\n\n"
40 "Command-line tool to simulate a call using the audio "
41 "processing module, either based on wav files or "
oprypin6e09d872017-08-31 03:21:39 -070042 "protobuf debug dump recordings.\n";
peah60a189f2016-05-24 20:54:40 -070043
Mirko Bonadei2dfa9982018-10-18 11:35:32 +020044WEBRTC_DEFINE_string(dump_input, "", "Aec dump input filename");
45WEBRTC_DEFINE_string(dump_output, "", "Aec dump output filename");
46WEBRTC_DEFINE_string(i, "", "Forward stream input wav filename");
47WEBRTC_DEFINE_string(o, "", "Forward stream output wav filename");
48WEBRTC_DEFINE_string(ri, "", "Reverse stream input wav filename");
49WEBRTC_DEFINE_string(ro, "", "Reverse stream output wav filename");
50WEBRTC_DEFINE_string(artificial_nearend, "", "Artificial nearend wav filename");
51WEBRTC_DEFINE_int(output_num_channels,
52 kParameterNotSpecifiedValue,
53 "Number of forward stream output channels");
54WEBRTC_DEFINE_int(reverse_output_num_channels,
55 kParameterNotSpecifiedValue,
56 "Number of Reverse stream output channels");
57WEBRTC_DEFINE_int(output_sample_rate_hz,
58 kParameterNotSpecifiedValue,
59 "Forward stream output sample rate in Hz");
60WEBRTC_DEFINE_int(reverse_output_sample_rate_hz,
61 kParameterNotSpecifiedValue,
62 "Reverse stream output sample rate in Hz");
63WEBRTC_DEFINE_bool(fixed_interface,
64 false,
65 "Use the fixed interface when operating on wav files");
66WEBRTC_DEFINE_int(aec,
67 kParameterNotSpecifiedValue,
68 "Activate (1) or deactivate(0) the echo canceller");
69WEBRTC_DEFINE_int(aecm,
70 kParameterNotSpecifiedValue,
71 "Activate (1) or deactivate(0) the mobile echo controller");
72WEBRTC_DEFINE_int(ed,
73 kParameterNotSpecifiedValue,
74 "Activate (1) or deactivate (0) the residual echo detector");
75WEBRTC_DEFINE_string(ed_graph,
76 "",
77 "Output filename for graph of echo likelihood");
78WEBRTC_DEFINE_int(agc,
79 kParameterNotSpecifiedValue,
80 "Activate (1) or deactivate(0) the AGC");
81WEBRTC_DEFINE_int(agc2,
82 kParameterNotSpecifiedValue,
83 "Activate (1) or deactivate(0) the AGC2");
84WEBRTC_DEFINE_int(pre_amplifier,
85 kParameterNotSpecifiedValue,
86 "Activate (1) or deactivate(0) the pre amplifier");
87WEBRTC_DEFINE_int(hpf,
88 kParameterNotSpecifiedValue,
89 "Activate (1) or deactivate(0) the high-pass filter");
90WEBRTC_DEFINE_int(ns,
91 kParameterNotSpecifiedValue,
92 "Activate (1) or deactivate(0) the noise suppressor");
93WEBRTC_DEFINE_int(ts,
94 kParameterNotSpecifiedValue,
95 "Activate (1) or deactivate(0) the transient suppressor");
96WEBRTC_DEFINE_int(vad,
97 kParameterNotSpecifiedValue,
98 "Activate (1) or deactivate(0) the voice activity detector");
99WEBRTC_DEFINE_int(le,
100 kParameterNotSpecifiedValue,
101 "Activate (1) or deactivate(0) the level estimator");
102WEBRTC_DEFINE_bool(
103 all_default,
104 false,
105 "Activate all of the default components (will be overridden by any "
106 "other settings)");
107WEBRTC_DEFINE_int(aec_suppression_level,
108 kParameterNotSpecifiedValue,
109 "Set the aec suppression level (0-2)");
110WEBRTC_DEFINE_int(delay_agnostic,
111 kParameterNotSpecifiedValue,
112 "Activate (1) or deactivate(0) the AEC delay agnostic mode");
113WEBRTC_DEFINE_int(extended_filter,
114 kParameterNotSpecifiedValue,
115 "Activate (1) or deactivate(0) the AEC extended filter mode");
Per Åhgren200feba2019-03-06 04:16:46 +0100116WEBRTC_DEFINE_int(use_legacy_aec,
117 kParameterNotSpecifiedValue,
118 "Activate (1) or deactivate(0) the legacy AEC");
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200119WEBRTC_DEFINE_int(experimental_agc,
120 kParameterNotSpecifiedValue,
121 "Activate (1) or deactivate(0) the experimental AGC");
122WEBRTC_DEFINE_int(
123 experimental_agc_disable_digital_adaptive,
124 kParameterNotSpecifiedValue,
125 "Force-deactivate (1) digital adaptation in "
126 "experimental AGC. Digital adaptation is active by default (0).");
127WEBRTC_DEFINE_int(experimental_agc_analyze_before_aec,
128 kParameterNotSpecifiedValue,
129 "Make level estimation happen before AEC"
130 " in the experimental AGC. After AEC is the default (0)");
131WEBRTC_DEFINE_int(
Alex Loikod9342442018-09-10 13:59:41 +0200132 experimental_agc_agc2_level_estimator,
133 kParameterNotSpecifiedValue,
134 "AGC2 level estimation"
135 " in the experimental AGC. AGC1 level estimation is the default (0)");
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200136WEBRTC_DEFINE_int(
peah60a189f2016-05-24 20:54:40 -0700137 refined_adaptive_filter,
138 kParameterNotSpecifiedValue,
139 "Activate (1) or deactivate(0) the refined adaptive filter functionality");
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200140WEBRTC_DEFINE_int(agc_mode,
141 kParameterNotSpecifiedValue,
142 "Specify the AGC mode (0-2)");
143WEBRTC_DEFINE_int(agc_target_level,
144 kParameterNotSpecifiedValue,
145 "Specify the AGC target level (0-31)");
146WEBRTC_DEFINE_int(agc_limiter,
147 kParameterNotSpecifiedValue,
148 "Activate (1) or deactivate(0) the level estimator");
149WEBRTC_DEFINE_int(agc_compression_gain,
150 kParameterNotSpecifiedValue,
151 "Specify the AGC compression gain (0-90)");
152WEBRTC_DEFINE_float(agc2_enable_adaptive_gain,
153 kParameterNotSpecifiedValue,
154 "Activate (1) or deactivate(0) the AGC2 adaptive gain");
155WEBRTC_DEFINE_float(agc2_fixed_gain_db, 0.f, "AGC2 fixed gain (dB) to apply");
Alessio Bazzica68170382018-11-20 12:27:20 +0100156
157std::vector<std::string> GetAgc2AdaptiveLevelEstimatorNames() {
158 return {"RMS", "peak"};
159}
160WEBRTC_DEFINE_string(
161 agc2_adaptive_level_estimator,
162 "RMS",
163 "AGC2 adaptive digital level estimator to use [RMS, peak]");
164
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200165WEBRTC_DEFINE_float(pre_amplifier_gain_factor,
166 1.f,
167 "Pre-amplifier gain factor (linear) to apply");
168WEBRTC_DEFINE_int(vad_likelihood,
169 kParameterNotSpecifiedValue,
170 "Specify the VAD likelihood (0-3)");
171WEBRTC_DEFINE_int(ns_level,
172 kParameterNotSpecifiedValue,
173 "Specify the NS level (0-3)");
174WEBRTC_DEFINE_int(stream_delay,
175 kParameterNotSpecifiedValue,
176 "Specify the stream delay in ms to use");
177WEBRTC_DEFINE_int(use_stream_delay,
178 kParameterNotSpecifiedValue,
179 "Activate (1) or deactivate(0) reporting the stream delay");
180WEBRTC_DEFINE_int(stream_drift_samples,
181 kParameterNotSpecifiedValue,
182 "Specify the number of stream drift samples to use");
183WEBRTC_DEFINE_int(initial_mic_level, 100, "Initial mic level (0-255)");
184WEBRTC_DEFINE_int(
185 simulate_mic_gain,
186 0,
187 "Activate (1) or deactivate(0) the analog mic gain simulation");
188WEBRTC_DEFINE_int(
189 simulated_mic_kind,
190 kParameterNotSpecifiedValue,
191 "Specify which microphone kind to use for microphone simulation");
192WEBRTC_DEFINE_bool(performance_report, false, "Report the APM performance ");
193WEBRTC_DEFINE_bool(verbose, false, "Produce verbose output");
194WEBRTC_DEFINE_bool(quiet,
195 false,
196 "Avoid producing information about the progress.");
197WEBRTC_DEFINE_bool(bitexactness_report,
198 false,
199 "Report bitexactness for aec dump result reproduction");
200WEBRTC_DEFINE_bool(discard_settings_in_aecdump,
201 false,
202 "Discard any config settings specified in the aec dump");
203WEBRTC_DEFINE_bool(store_intermediate_output,
204 false,
205 "Creates new output files after each init");
206WEBRTC_DEFINE_string(custom_call_order_file,
207 "",
208 "Custom process API call order file");
Ivo Creusen9a66d5e2019-03-13 13:55:12 +0100209WEBRTC_DEFINE_string(
210 output_custom_call_order_file,
211 "",
212 "Generate custom process API call order file from AEC dump");
Per Åhgren200feba2019-03-06 04:16:46 +0100213WEBRTC_DEFINE_bool(print_aec_parameter_values,
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200214 false,
Per Åhgren200feba2019-03-06 04:16:46 +0100215 "Print parameter values used in AEC in JSON-format");
216WEBRTC_DEFINE_string(aec_settings,
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200217 "",
Per Åhgren200feba2019-03-06 04:16:46 +0100218 "File in JSON-format with custom AEC settings");
Per Åhgren7a95e0f2018-10-25 09:56:49 +0200219WEBRTC_DEFINE_bool(dump_data,
220 false,
221 "Dump internal data during the call (requires build flag)");
Alessio Bazzica4bc60452018-11-20 12:44:15 +0100222WEBRTC_DEFINE_string(dump_data_output_dir,
223 "",
224 "Internal data dump output directory");
Mirko Bonadei2dfa9982018-10-18 11:35:32 +0200225WEBRTC_DEFINE_bool(help, false, "Print this message");
peah60a189f2016-05-24 20:54:40 -0700226
Alex Loiko890988c2017-08-31 10:25:48 +0200227void SetSettingIfSpecified(const std::string& value,
Danil Chapovalovdb9f7ab2018-06-19 10:50:11 +0200228 absl::optional<std::string>* parameter) {
peah60a189f2016-05-24 20:54:40 -0700229 if (value.compare("") != 0) {
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100230 *parameter = value;
peah60a189f2016-05-24 20:54:40 -0700231 }
232}
233
Danil Chapovalovdb9f7ab2018-06-19 10:50:11 +0200234void SetSettingIfSpecified(int value, absl::optional<int>* parameter) {
peah60a189f2016-05-24 20:54:40 -0700235 if (value != kParameterNotSpecifiedValue) {
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100236 *parameter = value;
peah60a189f2016-05-24 20:54:40 -0700237 }
238}
239
Danil Chapovalovdb9f7ab2018-06-19 10:50:11 +0200240void SetSettingIfFlagSet(int32_t flag, absl::optional<bool>* parameter) {
peah60a189f2016-05-24 20:54:40 -0700241 if (flag == 0) {
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100242 *parameter = false;
peah60a189f2016-05-24 20:54:40 -0700243 } else if (flag == 1) {
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100244 *parameter = true;
peah60a189f2016-05-24 20:54:40 -0700245 }
246}
247
Alessio Bazzica68170382018-11-20 12:27:20 +0100248AudioProcessing::Config::GainController2::LevelEstimator
249MapAgc2AdaptiveLevelEstimator(absl::string_view name) {
250 if (name.compare("RMS") == 0) {
251 return AudioProcessing::Config::GainController2::LevelEstimator::kRms;
252 }
253 if (name.compare("peak") == 0) {
254 return AudioProcessing::Config::GainController2::LevelEstimator::kPeak;
255 }
256 auto concat_strings =
257 [](const std::vector<std::string>& strings) -> std::string {
258 rtc::StringBuilder ss;
259 for (const auto& s : strings) {
260 ss << " " << s;
261 }
262 return ss.Release();
263 };
264 RTC_CHECK(false)
265 << "Invalid value for agc2_adaptive_level_estimator, valid options:"
266 << concat_strings(GetAgc2AdaptiveLevelEstimatorNames()) << ".";
267}
268
peah60a189f2016-05-24 20:54:40 -0700269SimulationSettings CreateSettings() {
270 SimulationSettings settings;
oprypin6e09d872017-08-31 03:21:39 -0700271 if (FLAG_all_default) {
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100272 settings.use_le = true;
273 settings.use_vad = true;
274 settings.use_ie = false;
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100275 settings.use_ts = true;
276 settings.use_ns = true;
277 settings.use_hpf = true;
278 settings.use_agc = true;
279 settings.use_agc2 = false;
Alex Loiko5feb30e2018-04-16 13:52:32 +0200280 settings.use_pre_amplifier = false;
Oskar Sundbomaa8b67d2017-11-17 14:34:48 +0100281 settings.use_aec = true;
282 settings.use_aecm = false;
283 settings.use_ed = false;
peah60a189f2016-05-24 20:54:40 -0700284 }
oprypin6e09d872017-08-31 03:21:39 -0700285 SetSettingIfSpecified(FLAG_dump_input, &settings.aec_dump_input_filename);
286 SetSettingIfSpecified(FLAG_dump_output, &settings.aec_dump_output_filename);
287 SetSettingIfSpecified(FLAG_i, &settings.input_filename);
288 SetSettingIfSpecified(FLAG_o, &settings.output_filename);
289 SetSettingIfSpecified(FLAG_ri, &settings.reverse_input_filename);
290 SetSettingIfSpecified(FLAG_ro, &settings.reverse_output_filename);
291 SetSettingIfSpecified(FLAG_artificial_nearend,
peahdf80fd12016-12-09 02:43:40 -0800292 &settings.artificial_nearend_filename);
oprypin6e09d872017-08-31 03:21:39 -0700293 SetSettingIfSpecified(FLAG_output_num_channels,
peah60a189f2016-05-24 20:54:40 -0700294 &settings.output_num_channels);
oprypin6e09d872017-08-31 03:21:39 -0700295 SetSettingIfSpecified(FLAG_reverse_output_num_channels,
peah60a189f2016-05-24 20:54:40 -0700296 &settings.reverse_output_num_channels);
oprypin6e09d872017-08-31 03:21:39 -0700297 SetSettingIfSpecified(FLAG_output_sample_rate_hz,
peah60a189f2016-05-24 20:54:40 -0700298 &settings.output_sample_rate_hz);
oprypin6e09d872017-08-31 03:21:39 -0700299 SetSettingIfSpecified(FLAG_reverse_output_sample_rate_hz,
peah60a189f2016-05-24 20:54:40 -0700300 &settings.reverse_output_sample_rate_hz);
oprypin6e09d872017-08-31 03:21:39 -0700301 SetSettingIfFlagSet(FLAG_aec, &settings.use_aec);
302 SetSettingIfFlagSet(FLAG_aecm, &settings.use_aecm);
303 SetSettingIfFlagSet(FLAG_ed, &settings.use_ed);
304 SetSettingIfSpecified(FLAG_ed_graph, &settings.ed_graph_output_filename);
305 SetSettingIfFlagSet(FLAG_agc, &settings.use_agc);
306 SetSettingIfFlagSet(FLAG_agc2, &settings.use_agc2);
Alex Loiko5feb30e2018-04-16 13:52:32 +0200307 SetSettingIfFlagSet(FLAG_pre_amplifier, &settings.use_pre_amplifier);
oprypin6e09d872017-08-31 03:21:39 -0700308 SetSettingIfFlagSet(FLAG_hpf, &settings.use_hpf);
309 SetSettingIfFlagSet(FLAG_ns, &settings.use_ns);
310 SetSettingIfFlagSet(FLAG_ts, &settings.use_ts);
oprypin6e09d872017-08-31 03:21:39 -0700311 SetSettingIfFlagSet(FLAG_vad, &settings.use_vad);
312 SetSettingIfFlagSet(FLAG_le, &settings.use_le);
313 SetSettingIfSpecified(FLAG_aec_suppression_level,
peah60a189f2016-05-24 20:54:40 -0700314 &settings.aec_suppression_level);
oprypin6e09d872017-08-31 03:21:39 -0700315 SetSettingIfFlagSet(FLAG_delay_agnostic, &settings.use_delay_agnostic);
316 SetSettingIfFlagSet(FLAG_extended_filter, &settings.use_extended_filter);
oprypin6e09d872017-08-31 03:21:39 -0700317 SetSettingIfFlagSet(FLAG_refined_adaptive_filter,
peah60a189f2016-05-24 20:54:40 -0700318 &settings.use_refined_adaptive_filter);
319
Per Åhgren200feba2019-03-06 04:16:46 +0100320 SetSettingIfFlagSet(FLAG_use_legacy_aec, &settings.use_legacy_aec);
oprypin6e09d872017-08-31 03:21:39 -0700321 SetSettingIfFlagSet(FLAG_experimental_agc, &settings.use_experimental_agc);
Alex Loiko9489c3a2018-08-09 15:04:24 +0200322 SetSettingIfFlagSet(FLAG_experimental_agc_disable_digital_adaptive,
323 &settings.experimental_agc_disable_digital_adaptive);
Alex Loikod9342442018-09-10 13:59:41 +0200324 SetSettingIfFlagSet(FLAG_experimental_agc_analyze_before_aec,
325 &settings.experimental_agc_analyze_before_aec);
326 SetSettingIfFlagSet(FLAG_experimental_agc_agc2_level_estimator,
327 &settings.use_experimental_agc_agc2_level_estimator);
oprypin6e09d872017-08-31 03:21:39 -0700328 SetSettingIfSpecified(FLAG_agc_mode, &settings.agc_mode);
329 SetSettingIfSpecified(FLAG_agc_target_level, &settings.agc_target_level);
330 SetSettingIfFlagSet(FLAG_agc_limiter, &settings.use_agc_limiter);
331 SetSettingIfSpecified(FLAG_agc_compression_gain,
peah60a189f2016-05-24 20:54:40 -0700332 &settings.agc_compression_gain);
Per Åhgrene4d23b12018-10-02 23:40:07 +0200333 SetSettingIfFlagSet(FLAG_agc2_enable_adaptive_gain,
334 &settings.agc2_use_adaptive_gain);
Alessio Bazzica270f7b52017-10-13 11:05:17 +0200335 settings.agc2_fixed_gain_db = FLAG_agc2_fixed_gain_db;
Alessio Bazzica68170382018-11-20 12:27:20 +0100336 settings.agc2_adaptive_level_estimator =
337 MapAgc2AdaptiveLevelEstimator(FLAG_agc2_adaptive_level_estimator);
Alex Loiko5feb30e2018-04-16 13:52:32 +0200338 settings.pre_amplifier_gain_factor = FLAG_pre_amplifier_gain_factor;
oprypin6e09d872017-08-31 03:21:39 -0700339 SetSettingIfSpecified(FLAG_vad_likelihood, &settings.vad_likelihood);
340 SetSettingIfSpecified(FLAG_ns_level, &settings.ns_level);
341 SetSettingIfSpecified(FLAG_stream_delay, &settings.stream_delay);
Per Åhgrenc5efb0c2018-05-28 09:10:59 +0200342 SetSettingIfFlagSet(FLAG_use_stream_delay, &settings.use_stream_delay);
oprypin6e09d872017-08-31 03:21:39 -0700343 SetSettingIfSpecified(FLAG_stream_drift_samples,
peah60a189f2016-05-24 20:54:40 -0700344 &settings.stream_drift_samples);
oprypin6e09d872017-08-31 03:21:39 -0700345 SetSettingIfSpecified(FLAG_custom_call_order_file,
Ivo Creusen9a66d5e2019-03-13 13:55:12 +0100346 &settings.call_order_input_filename);
347 SetSettingIfSpecified(FLAG_output_custom_call_order_file,
348 &settings.call_order_output_filename);
Per Åhgren200feba2019-03-06 04:16:46 +0100349 SetSettingIfSpecified(FLAG_aec_settings, &settings.aec_settings_filename);
Alessio Bazzicaca90a552017-09-27 11:51:10 +0200350 settings.initial_mic_level = FLAG_initial_mic_level;
351 settings.simulate_mic_gain = FLAG_simulate_mic_gain;
352 SetSettingIfSpecified(FLAG_simulated_mic_kind, &settings.simulated_mic_kind);
oprypin6e09d872017-08-31 03:21:39 -0700353 settings.report_performance = FLAG_performance_report;
354 settings.use_verbose_logging = FLAG_verbose;
Per Åhgren5a72a5e2018-08-24 07:18:24 +0200355 settings.use_quiet_output = FLAG_quiet;
oprypin6e09d872017-08-31 03:21:39 -0700356 settings.report_bitexactness = FLAG_bitexactness_report;
357 settings.discard_all_settings_in_aecdump = FLAG_discard_settings_in_aecdump;
358 settings.fixed_interface = FLAG_fixed_interface;
359 settings.store_intermediate_output = FLAG_store_intermediate_output;
Per Åhgren200feba2019-03-06 04:16:46 +0100360 settings.print_aec_parameter_values = FLAG_print_aec_parameter_values;
Per Åhgren7a95e0f2018-10-25 09:56:49 +0200361 settings.dump_internal_data = FLAG_dump_data;
Alessio Bazzica4bc60452018-11-20 12:44:15 +0100362 SetSettingIfSpecified(FLAG_dump_data_output_dir,
363 &settings.dump_internal_data_output_dir);
peah60a189f2016-05-24 20:54:40 -0700364
365 return settings;
366}
367
Alex Loiko890988c2017-08-31 10:25:48 +0200368void ReportConditionalErrorAndExit(bool condition, const std::string& message) {
peah60a189f2016-05-24 20:54:40 -0700369 if (condition) {
370 std::cerr << message << std::endl;
371 exit(1);
372 }
373}
374
375void PerformBasicParameterSanityChecks(const SimulationSettings& settings) {
376 if (settings.input_filename || settings.reverse_input_filename) {
377 ReportConditionalErrorAndExit(!!settings.aec_dump_input_filename,
378 "Error: The aec dump cannot be specified "
379 "together with input wav files!\n");
380
peahdf80fd12016-12-09 02:43:40 -0800381 ReportConditionalErrorAndExit(!!settings.artificial_nearend_filename,
382 "Error: The artificial nearend cannot be "
383 "specified together with input wav files!\n");
384
peah60a189f2016-05-24 20:54:40 -0700385 ReportConditionalErrorAndExit(!settings.input_filename,
386 "Error: When operating at wav files, the "
387 "input wav filename must be "
388 "specified!\n");
389
390 ReportConditionalErrorAndExit(
391 settings.reverse_output_filename && !settings.reverse_input_filename,
392 "Error: When operating at wav files, the reverse input wav filename "
393 "must be specified if the reverse output wav filename is specified!\n");
394 } else {
395 ReportConditionalErrorAndExit(!settings.aec_dump_input_filename,
396 "Error: Either the aec dump or the wav "
397 "input files must be specified!\n");
398 }
399
400 ReportConditionalErrorAndExit(
401 settings.use_aec && *settings.use_aec && settings.use_aecm &&
402 *settings.use_aecm,
403 "Error: The AEC and the AECM cannot be activated at the same time!\n");
404
405 ReportConditionalErrorAndExit(
406 settings.output_sample_rate_hz && *settings.output_sample_rate_hz <= 0,
407 "Error: --output_sample_rate_hz must be positive!\n");
408
409 ReportConditionalErrorAndExit(
410 settings.reverse_output_sample_rate_hz &&
411 settings.output_sample_rate_hz &&
412 *settings.output_sample_rate_hz <= 0,
413 "Error: --reverse_output_sample_rate_hz must be positive!\n");
414
415 ReportConditionalErrorAndExit(
416 settings.output_num_channels && *settings.output_num_channels <= 0,
417 "Error: --output_num_channels must be positive!\n");
418
419 ReportConditionalErrorAndExit(
420 settings.reverse_output_num_channels &&
421 *settings.reverse_output_num_channels <= 0,
422 "Error: --reverse_output_num_channels must be positive!\n");
423
Sam Zackrissoncdf0e6d2018-09-17 11:05:17 +0200424 ReportConditionalErrorAndExit(settings.aec_suppression_level &&
425 ((*settings.aec_suppression_level) < 1 ||
426 (*settings.aec_suppression_level) > 2),
427 "Error: --aec_suppression_level must be "
428 "specified between 1 and 2. 0 is "
429 "deprecated.\n");
peah60a189f2016-05-24 20:54:40 -0700430
431 ReportConditionalErrorAndExit(
peah60a189f2016-05-24 20:54:40 -0700432 settings.agc_target_level && ((*settings.agc_target_level) < 0 ||
433 (*settings.agc_target_level) > 31),
434 "Error: --agc_target_level must be specified between 0 and 31.\n");
435
436 ReportConditionalErrorAndExit(
437 settings.agc_compression_gain && ((*settings.agc_compression_gain) < 0 ||
438 (*settings.agc_compression_gain) > 90),
439 "Error: --agc_compression_gain must be specified between 0 and 90.\n");
440
441 ReportConditionalErrorAndExit(
Alessio Bazzica270f7b52017-10-13 11:05:17 +0200442 settings.use_agc2 && *settings.use_agc2 &&
443 ((settings.agc2_fixed_gain_db) < 0 ||
444 (settings.agc2_fixed_gain_db) > 90),
445 "Error: --agc2_fixed_gain_db must be specified between 0 and 90.\n");
446
447 ReportConditionalErrorAndExit(
peah60a189f2016-05-24 20:54:40 -0700448 settings.vad_likelihood &&
449 ((*settings.vad_likelihood) < 0 || (*settings.vad_likelihood) > 3),
450 "Error: --vad_likelihood must be specified between 0 and 3.\n");
451
452 ReportConditionalErrorAndExit(
453 settings.ns_level &&
454 ((*settings.ns_level) < 0 || (*settings.ns_level) > 3),
455 "Error: --ns_level must be specified between 0 and 3.\n");
456
457 ReportConditionalErrorAndExit(
458 settings.report_bitexactness && !settings.aec_dump_input_filename,
459 "Error: --bitexactness_report can only be used when operating on an "
460 "aecdump\n");
461
peah5ad5de32016-12-09 03:18:22 -0800462 ReportConditionalErrorAndExit(
Ivo Creusen9a66d5e2019-03-13 13:55:12 +0100463 settings.call_order_input_filename && settings.aec_dump_input_filename,
peah5ad5de32016-12-09 03:18:22 -0800464 "Error: --custom_call_order_file cannot be used when operating on an "
465 "aecdump\n");
466
Alessio Bazzicaca90a552017-09-27 11:51:10 +0200467 ReportConditionalErrorAndExit(
468 (settings.initial_mic_level < 0 || settings.initial_mic_level > 255),
469 "Error: --initial_mic_level must be specified between 0 and 255.\n");
470
471 ReportConditionalErrorAndExit(
472 settings.simulated_mic_kind && !settings.simulate_mic_gain,
473 "Error: --simulated_mic_kind cannot be specified when mic simulation is "
474 "disabled\n");
475
476 ReportConditionalErrorAndExit(
477 !settings.simulated_mic_kind && settings.simulate_mic_gain,
478 "Error: --simulated_mic_kind must be specified when mic simulation is "
479 "enabled\n");
480
peah60a189f2016-05-24 20:54:40 -0700481 auto valid_wav_name = [](const std::string& wav_file_name) {
482 if (wav_file_name.size() < 5) {
483 return false;
484 }
485 if ((wav_file_name.compare(wav_file_name.size() - 4, 4, ".wav") == 0) ||
486 (wav_file_name.compare(wav_file_name.size() - 4, 4, ".WAV") == 0)) {
487 return true;
488 }
489 return false;
490 };
491
492 ReportConditionalErrorAndExit(
493 settings.input_filename && (!valid_wav_name(*settings.input_filename)),
494 "Error: --i must be a valid .wav file name.\n");
495
496 ReportConditionalErrorAndExit(
497 settings.output_filename && (!valid_wav_name(*settings.output_filename)),
498 "Error: --o must be a valid .wav file name.\n");
499
500 ReportConditionalErrorAndExit(
501 settings.reverse_input_filename &&
502 (!valid_wav_name(*settings.reverse_input_filename)),
503 "Error: --ri must be a valid .wav file name.\n");
504
505 ReportConditionalErrorAndExit(
506 settings.reverse_output_filename &&
507 (!valid_wav_name(*settings.reverse_output_filename)),
508 "Error: --ro must be a valid .wav file name.\n");
peahdf80fd12016-12-09 02:43:40 -0800509
510 ReportConditionalErrorAndExit(
511 settings.artificial_nearend_filename &&
512 !valid_wav_name(*settings.artificial_nearend_filename),
513 "Error: --artifical_nearend must be a valid .wav file name.\n");
Per Åhgren7a95e0f2018-10-25 09:56:49 +0200514
515 ReportConditionalErrorAndExit(
516 WEBRTC_APM_DEBUG_DUMP == 0 && settings.dump_internal_data,
517 "Error: --dump_data cannot be set without proper build support.\n");
Alessio Bazzica4bc60452018-11-20 12:44:15 +0100518
519 ReportConditionalErrorAndExit(
520 !settings.dump_internal_data &&
521 settings.dump_internal_data_output_dir.has_value(),
522 "Error: --dump_data_output_dir cannot be set without --dump_data.\n");
Ivo Creusen9a66d5e2019-03-13 13:55:12 +0100523
524 ReportConditionalErrorAndExit(
525 !settings.aec_dump_input_filename &&
526 settings.call_order_output_filename.has_value(),
527 "Error: --output_custom_call_order_file needs an AEC dump input file.\n");
Peter Kasting69558702016-01-12 16:26:35 -0800528}
529
530} // namespace
531
Ivo Creusen2cb41052018-03-15 12:22:52 +0100532int AudioprocFloatImpl(std::unique_ptr<AudioProcessingBuilder> ap_builder,
533 int argc,
534 char* argv[]) {
Yves Gerey665174f2018-06-19 15:03:05 +0200535 if (rtc::FlagList::SetFlagsFromCommandLine(&argc, argv, true) || FLAG_help ||
536 argc != 1) {
oprypin6e09d872017-08-31 03:21:39 -0700537 printf("%s", kUsageDescription);
538 if (FLAG_help) {
539 rtc::FlagList::Print(nullptr, false);
540 return 0;
541 }
542 return 1;
543 }
andrew@webrtc.org08df9b22014-12-16 20:57:15 +0000544
peah60a189f2016-05-24 20:54:40 -0700545 SimulationSettings settings = CreateSettings();
546 PerformBasicParameterSanityChecks(settings);
547 std::unique_ptr<AudioProcessingSimulator> processor;
andrew@webrtc.org08df9b22014-12-16 20:57:15 +0000548
peah60a189f2016-05-24 20:54:40 -0700549 if (settings.aec_dump_input_filename) {
Ivo Creusen8c812f32018-03-09 17:33:04 +0100550 processor.reset(new AecDumpBasedSimulator(settings, std::move(ap_builder)));
aluebsb0ad43b2015-11-20 00:11:53 -0800551 } else {
Ivo Creusen8c812f32018-03-09 17:33:04 +0100552 processor.reset(new WavBasedSimulator(settings, std::move(ap_builder)));
andrewbdafe312015-10-29 23:42:54 -0700553 }
554
peah60a189f2016-05-24 20:54:40 -0700555 processor->Process();
aluebsb0ad43b2015-11-20 00:11:53 -0800556
peah60a189f2016-05-24 20:54:40 -0700557 if (settings.report_performance) {
aluebsb0ad43b2015-11-20 00:11:53 -0800558 const auto& proc_time = processor->proc_time();
Niels Möllerd28db7f2016-05-10 16:31:47 +0200559 int64_t exec_time_us = proc_time.sum / rtc::kNumNanosecsPerMicrosec;
peah60a189f2016-05-24 20:54:40 -0700560 std::cout << std::endl
561 << "Execution time: " << exec_time_us * 1e-6 << " s, File time: "
562 << processor->get_num_process_stream_calls() * 1.f /
563 AudioProcessingSimulator::kChunksPerSecond
564 << std::endl
565 << "Time per fwd stream chunk (mean, max, min): " << std::endl
566 << exec_time_us * 1.f / processor->get_num_process_stream_calls()
567 << " us, " << 1.f * proc_time.max / rtc::kNumNanosecsPerMicrosec
568 << " us, " << 1.f * proc_time.min / rtc::kNumNanosecsPerMicrosec
569 << " us" << std::endl;
570 }
571
572 if (settings.report_bitexactness && settings.aec_dump_input_filename) {
573 if (processor->OutputWasBitexact()) {
574 std::cout << "The processing was bitexact.";
575 } else {
576 std::cout << "The processing was not bitexact.";
577 }
Alejandro Luebs5d22c002015-04-15 11:26:40 -0700578 }
aluebsb0ad43b2015-11-20 00:11:53 -0800579
andrew@webrtc.org08df9b22014-12-16 20:57:15 +0000580 return 0;
581}
582
peah60a189f2016-05-24 20:54:40 -0700583} // namespace test
andrew@webrtc.org08df9b22014-12-16 20:57:15 +0000584} // namespace webrtc