def get_channel_sox_stat(
input_audio, channel_index, channels=2, bits=16, rate=48000):
"""Gets the sox stat info of the selected channel in the input audio file.
@param input_audio: The input audio file to be analyzed.
@param channel_index: The index of the channel to be analyzed.
(1 for the first channel).
@param channels: The number of channels in the input audio.
@param bits: The number of bits of each audio sample.
@param rate: The sampling rate.
"""
if channel_index <= 0 or channel_index > channels:
raise ValueError('incorrect channel_indexi: %d' % channel_index)
if channels == 1:
return sox_utils.get_stat(
input_audio, channels=channels, bits=bits, rate=rate)
p1 = cmd_utils.popen(
sox_utils.extract_channel_cmd(
input_audio, '-', channel_index,
channels=channels, bits=bits, rate=rate),
stdout=subprocess.PIPE)
p2 = cmd_utils.popen(
sox_utils.stat_cmd('-', channels=1, bits=bits, rate=rate),
stdin=p1.stdout, stderr=subprocess.PIPE)
stat_output = p2.stderr.read()
cmd_utils.wait_and_check_returncode(p1, p2)
return sox_utils.parse_stat_output(stat_output)
def get_one_channel_stat(data, data_format):
"""Gets statistic information of data.
@param data: A list containing one channel data.
@param data_format: A dict containing data format of data.
@return: The sox stat parsed result. An object containing
sameple_count: An int. Samples read.
length: A float. Length in seconds.
rms: A float. RMS amplitude.
rough_frequency: A float. Rough frequency.
"""
if not data:
raise ValueError('Data is empty. Can not get stat')
raw_data = audio_data.AudioRawData(
binary=None, channel=1, sample_format=data_format['sample_format'])
raw_data.copy_channel_data([data])
with tempfile.NamedTemporaryFile() as raw_data_file:
raw_data_path = raw_data_file.name
raw_data.write_to_file(raw_data_path)
bits = 8 * (audio_data.SAMPLE_FORMATS[data_format['sample_format']]
['size_bytes'])
stat = sox_utils.get_stat(raw_data_path,
channels=1,
bits=bits,
rate=data_format['rate'])
return stat
def _rms_of_next_audio_chunk(self):
"""Finds the sox_stats values of the next chunk of audio."""
cras_utils.loopback(self._loopback_file,
channels=1,
duration=self._audio_chunk_size)
stat_output = sox_utils.get_stat(self._loopback_file)
logging.info(stat_output)
return vars(stat_output)['rms']
def loopback(self, noise_profile, primary, secondary):
"""Gets the rms value of the recorded audio of playing two different
tones (the 440 and 523 Hz sine wave) at the specified sampling rate.
@param noise_profile: The noise profile which is used to reduce the
noise of the recored audio.
@param primary: The first sample rate, HW will be set to this.
@param secondary: The second sample rate, will be SRC'd to the first.
"""
popens = []
record_file = os.path.join(self.resultsdir,
'record-%s-%s.wav' % (primary, secondary))
# There should be no other active streams.
self.wait_for_active_stream_count(0)
# Start with the primary sample rate, then add the secondary. This
# causes the secondary to be SRC'd to the primary rate.
try:
# Play the first audio stream and make sure it has been played
popens += self.play_sine_tone(_TEST_TONE_ONE, primary)
self.wait_for_active_stream_count(1)
# Play the second audio stream and make sure it has been played
popens += self.play_sine_tone(_TEST_TONE_TWO, secondary)
self.wait_for_active_stream_count(2)
cras_utils.capture(record_file, duration=1, rate=44100)
# Make sure the playback is still in good shape
if any(p.poll() is not None for p in popens):
# We will log more details later in finally.
raise error.TestFail('process unexpectly stopped')
reduced_file = tempfile.NamedTemporaryFile()
sox_utils.noise_reduce(record_file,
reduced_file.name,
noise_profile,
rate=44100)
sox_stat = sox_utils.get_stat(reduced_file.name, rate=44100)
logging.info('The sox stat of (%d, %d) is %s', primary, secondary,
str(sox_stat))
return sox_stat.rms
finally:
cmd_utils.kill_or_log_returncode(*popens)