APM Quality Assessment tool, audioproc_f Python wrapper

webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/audioproc_wrapper.py

Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/audioproc_wrapper.py
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/audioproc_wrapper.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/audioproc_wrapper.py
new file mode 100644
index 0000000000000000000000000000000000000000..ba46c579d01f8f9254e4699fc04899ff735315b9
--- /dev/null
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/audioproc_wrapper.py
@@ -0,0 +1,66 @@
+# Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
+#
+# Use of this source code is governed by a BSD-style license
+# that can be found in the LICENSE file in the root of the source
+# tree. An additional intellectual property rights grant can be found
+# in the file PATENTS.  All contributing project authors may
+# be found in the AUTHORS file in the root of the source tree.
+
+import cProfile
+import logging
+import os
+import subprocess
+
+from .data_access import AudioProcConfigFile
+
+class AudioProcWrapper(object):
+
+  OUTPUT_FILENAME = 'output.wav'
+  _AUDIOPROC_F_BIN_PATH = os.path.abspath('audioproc_f')
+
+  def __init__(self):
+    self._config = None
+    self._input_signal_filepath = None
+    self._output_signal_filepath = None
+
+    # Profiler instance to measure audioproc_f running time.
+    self._profiler = cProfile.Profile()
+
+  @property
+  def output_filepath(self):
+    return self._output_signal_filepath
+
+  def run(self, config_filepath, input_filepath, output_path):
+    # Init.
+    self._input_signal_filepath = input_filepath
+    self._output_signal_filepath = os.path.join(
+        output_path, self.OUTPUT_FILENAME)
+    profiling_stats_filepath = os.path.join(output_path, 'profiling.stats')
+
+    # Skip if the output has already been generated.
+    if os.path.exists(self._output_signal_filepath) and os.path.exists(
+        profiling_stats_filepath):
+      return
+
+    # Load configuration.
+    self._config = AudioProcConfigFile.load(config_filepath)
+
+    # Set remaining parametrs.
+    self._config['-i'] = self._input_signal_filepath
+    self._config['-o'] = self._output_signal_filepath
+
+    # Build arguments list.
+    args = [self._AUDIOPROC_F_BIN_PATH]
+    for param_name in self._config:
+      args.append(param_name)
+      if self._config[param_name] is not None:
+        args.append(str(self._config[param_name]))
+    logging.debug(' '.join(args))
+
+    # Run.
+    self._profiler.enable()
+    subprocess.call(args)
+    self._profiler.disable()
+
+    # Save profiling stats.
+    self._profiler.dump_stats(profiling_stats_filepath)