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

Side by Side Diff: webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment.py

Issue 2793903006: Style fix and docstring for the APM QA Python tool (Closed)
Patch Set: _ prefix for internal use functions Created 3 years, 8 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 #!/usr/bin/env python 1 #!/usr/bin/env python
2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. 2 # Copyright (c) 2017 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 """Perform APM module quality assessment on one or more input files using one or 10 """Perform APM module quality assessment on one or more input files using one or
(...skipping 14 matching lines...) Expand all
25 import quality_assessment.noise_generation as noise_generation 25 import quality_assessment.noise_generation as noise_generation
26 import quality_assessment.simulation as simulation 26 import quality_assessment.simulation as simulation
27 27
28 _NOISE_GENERATOR_CLASSES = noise_generation.NoiseGenerator.REGISTERED_CLASSES 28 _NOISE_GENERATOR_CLASSES = noise_generation.NoiseGenerator.REGISTERED_CLASSES
29 _NOISE_GENERATORS_NAMES = _NOISE_GENERATOR_CLASSES.keys() 29 _NOISE_GENERATORS_NAMES = _NOISE_GENERATOR_CLASSES.keys()
30 _EVAL_SCORE_WORKER_CLASSES = eval_scores.EvaluationScore.REGISTERED_CLASSES 30 _EVAL_SCORE_WORKER_CLASSES = eval_scores.EvaluationScore.REGISTERED_CLASSES
31 _EVAL_SCORE_WORKER_NAMES = _EVAL_SCORE_WORKER_CLASSES.keys() 31 _EVAL_SCORE_WORKER_NAMES = _EVAL_SCORE_WORKER_CLASSES.keys()
32 32
33 _DEFAULT_CONFIG_FILE = 'apm_configs/default.json' 33 _DEFAULT_CONFIG_FILE = 'apm_configs/default.json'
34 34
35 def _instance_arguments_parser(): 35
36 def _InstanceArgumentsParser():
37 """Arguments parser factory.
38 """
36 parser = argparse.ArgumentParser(description=( 39 parser = argparse.ArgumentParser(description=(
37 'Perform APM module quality assessment on one or more input files using ' 40 'Perform APM module quality assessment on one or more input files using '
38 'one or more audioproc_f configuration files and one or more noise ' 41 'one or more audioproc_f configuration files and one or more noise '
39 'generators.')) 42 'generators.'))
40 43
41 parser.add_argument('-c', '--config_files', nargs='+', required=False, 44 parser.add_argument('-c', '--config_files', nargs='+', required=False,
42 help=('path to the configuration files defining the ' 45 help=('path to the configuration files defining the '
43 'arguments with which the audioproc_f tool is ' 46 'arguments with which the audioproc_f tool is '
44 'called'), 47 'called'),
45 default=[_DEFAULT_CONFIG_FILE]) 48 default=[_DEFAULT_CONFIG_FILE])
(...skipping 23 matching lines...) Expand all
69 parser.add_argument('--air_db_path', required=True, 72 parser.add_argument('--air_db_path', required=True,
70 help='path to the Aechen IR database') 73 help='path to the Aechen IR database')
71 74
72 return parser 75 return parser
73 76
74 77
75 def main(): 78 def main():
76 # TODO(alessiob): level = logging.INFO once debugged. 79 # TODO(alessiob): level = logging.INFO once debugged.
77 logging.basicConfig(level=logging.DEBUG) 80 logging.basicConfig(level=logging.DEBUG)
78 81
79 parser = _instance_arguments_parser() 82 parser = _InstanceArgumentsParser()
80 args = parser.parse_args() 83 args = parser.parse_args()
81 84
82 simulator = simulation.ApmModuleSimulator( 85 simulator = simulation.ApmModuleSimulator(
83 aechen_ir_database_path=args.air_db_path, 86 aechen_ir_database_path=args.air_db_path,
84 polqa_tool_path=args.polqa_path) 87 polqa_tool_path=args.polqa_path)
85 simulator.run( 88 simulator.Run(
86 config_filepaths=args.config_files, 89 config_filepaths=args.config_files,
87 input_filepaths=args.input_files, 90 input_filepaths=args.input_files,
88 noise_generator_names=args.noise_generators, 91 noise_generator_names=args.noise_generators,
89 eval_score_names=args.eval_scores, 92 eval_score_names=args.eval_scores,
90 output_dir=args.output_dir) 93 output_dir=args.output_dir)
91 94
92 sys.exit(0) 95 sys.exit(0)
93 96
94 97
95 if __name__ == '__main__': 98 if __name__ == '__main__':
96 main() 99 main()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698