Refactor some utility code to audio_helper.
BUG=None
TEST=Run audiovideo_LineOutToMicInLoopback, desktopui_AudioFeedback,
desktopui_MediaAudioFeedback test work good.
Change-Id: I65b9d52f43869333f1041a91c5200ff2d5fa225d
Reviewed-on: https://gerrit.chromium.org/gerrit/19809
Commit-Ready: Hsinyu Chao <hychao@chromium.org>
Reviewed-by: Hsinyu Chao <hychao@chromium.org>
Tested-by: Hsinyu Chao <hychao@chromium.org>
diff --git a/client/cros/audio/audio_helper.py b/client/cros/audio/audio_helper.py
index 79a8a02..b5b5f54 100644
--- a/client/cros/audio/audio_helper.py
+++ b/client/cros/audio/audio_helper.py
@@ -5,18 +5,29 @@
import logging
import os
+import re
from autotest_lib.client.bin import utils
from autotest_lib.client.common_lib import error
LD_LIBRARY_PATH = 'LD_LIBRARY_PATH'
+_DEFAULT_NUM_CHANNELS = 2
+_DEFAULT_INPUT_DEVICE = 'default'
+_DEFAULT_SOX_FORMAT = '-t raw -b 16 -e signed -r 48000 -L'
+
+_SOX_RMS_AMPLITUDE_RE = re.compile('RMS\s+amplitude:\s+(.+)')
+_SOX_FORMAT = '-t raw -b 16 -e signed -r 48000 -L'
+
class AudioHelper(object):
'''
A helper class contains audio related utility functions.
'''
- def __init__(self, test):
+ def __init__(self, test, sox_format=_DEFAULT_SOX_FORMAT,
+ input_device=_DEFAULT_INPUT_DEVICE):
self._test = test
+ self._sox_format = sox_format
+ self._input_device = input_device
def setup_deps(self, deps):
'''
@@ -73,3 +84,60 @@
# A card is allowed not to support all the controls, so don't
# fail the test here if we get an error.
logging.info('amixer command failed: %s' % cmd)
+
+ def get_audio_rms(self, infile, channel):
+ sox_mixer_cmd = self.get_sox_mixer_cmd(infile, channel)
+ stat_cmd = '%s -c 1 %s - -n stat 2>&1' % (self.sox_path,
+ self._sox_format)
+ sox_cmd = '%s | %s' % (sox_mixer_cmd, stat_cmd)
+ sox_output = utils.system_output(sox_cmd, retain_output=True)
+ for rms_line in sox_output.split('\n'):
+ m = _SOX_RMS_AMPLITUDE_RE.match(rms_line)
+ if m is not None:
+ return float(m.group(1))
+
+ def get_sox_mixer_cmd(self, infile, channel):
+ # Build up a pan value string for the sox command.
+ if channel == 0:
+ pan_values = '1'
+ else:
+ pan_values = '0'
+ for pan_index in range(1, self._num_channels):
+ if channel == pan_index:
+ pan_values = '%s%s' % (pan_values, ',1')
+ else:
+ pan_values = '%s%s' % (pan_values, ',0')
+
+ return '%s -c 2 %s %s -c 1 %s - mixer %s' % (self.sox_path,
+ self._sox_format, infile, self._sox_format, pan_values)
+
+ def noise_reduce_file(self, in_file, noise_file, out_file):
+ """Runs the sox command to noise-reduce in_file using
+ the noise profile from noise_file.
+
+ Args:
+ in_file: The file to noise reduce.
+ noise_file: The file containing the noise profile.
+ This can be created by recording silence.
+ out_file: The file contains the noise reduced sound.
+
+ Returns:
+ The name of the file containing the noise-reduced data.
+ """
+ prof_cmd = '%s -c 2 %s %s -n noiseprof' % (self.sox_path,
+ _SOX_FORMAT, noise_file)
+ reduce_cmd = ('%s -c 2 %s %s -c 2 %s %s noisered' %
+ (self.sox_path, _SOX_FORMAT, in_file, _SOX_FORMAT, out_file))
+ utils.system('%s | %s' % (prof_cmd, reduce_cmd))
+
+ def record_sample(self, duration, tmpfile):
+ """Records a sample from the default input device.
+
+ Args:
+ duration: How long to record in seconds.
+ tmpfile: The file to record to.
+ """
+ cmd_rec = 'arecord -D %s -d %f -f dat %s' % (self._input_device,
+ duration, tmpfile)
+ logging.info('Command %s recording now (%fs)' % (cmd_rec, duration))
+ utils.system(cmd_rec)