From f7576dcb7fa43c705c5a304b437d79fe6902e588 Mon Sep 17 00:00:00 2001 From: Fabio Manganiello Date: Sun, 16 Dec 2018 16:15:45 +0100 Subject: [PATCH] Added support for audio pass-through to sound plugin --- platypush/plugins/sound.py | 107 ++++++++++++++++++++++++++++++++++++- 1 file changed, 105 insertions(+), 2 deletions(-) diff --git a/platypush/plugins/sound.py b/platypush/plugins/sound.py index 7eb1654a0..579b46023 100644 --- a/platypush/plugins/sound.py +++ b/platypush/plugins/sound.py @@ -126,7 +126,10 @@ class SoundPlugin(Plugin): :type bufsize: int """ - self.stop_playback() + if self._get_playback_state() != PlaybackState.STOPPED: + self.stop_playback() + time.sleep(2) + self.playback_paused_changed.clear() q = queue.Queue(maxsize=bufsize) @@ -243,7 +246,10 @@ class SoundPlugin(Plugin): :type subtype: str """ - self.stop_recording() + if self._get_recording_state() != RecordingState.STOPPED: + self.stop_recording() + time.sleep(2) + self.recording_paused_changed.clear() if file: @@ -311,6 +317,103 @@ class SoundPlugin(Plugin): finally: self.stop_recording() + + @action + def recordplay(self, duration=None, input_device=None, output_device=None, + sample_rate=None, blocksize=_DEFAULT_PLAY_BLOCKSIZE, + latency=0, channels=1, dtype=None): + """ + Records audio and plays it on an output sound device (audio pass-through) + + :param duration: Recording duration in seconds (default: record until stop event) + :type duration: float + + :param input_device: Input device (default: default configured device or system default audio input if not configured) + :type input_device: int or str + + :param output_device: Output device (default: default configured device or system default audio output if not configured) + :type output_device: int or str + + :param sample_rate: Recording sample rate (default: device default rate) + :type sample_rate: int + + :param blocksize: Audio block size (default: 2048) + :type blocksize: int + + :param latency: Device latency in seconds (default: 0) + :type latency: float + + :param channels: Number of channels (default: 1) + :type channels: int + + :param dtype: Data type for the recording - see `soundfile docs `_ for available types (default: input device default) + :type dtype: str + """ + + if self._get_playback_state() != PlaybackState.STOPPED: + self.stop_playback() + time.sleep(2) + + if self._get_recording_state() != RecordingState.STOPPED: + self.stop_recording() + time.sleep(2) + + self.playback_paused_changed.clear() + self.recording_paused_changed.clear() + + if not input_device: + input_device = self.input_device + + if not output_device: + output_device = self.output_device + + if sample_rate is None: + dev_info = sd.query_devices(input_device, 'input') + sample_rate = int(dev_info['default_samplerate']) + + def audio_callback(indata, outdata, frames, time, status): + while self._get_recording_state() == RecordingState.PAUSED: + self.recording_paused_changed.wait() + + if status: + self.logger.warning('Recording callback status: {}'.format( + str(status))) + + outdata[:] = indata + + + try: + import soundfile as sf + import numpy + + with sd.Stream(samplerate=sample_rate, channels=channels, + blocksize=blocksize, latency=latency, + device=(input_device, output_device), + dtype=dtype, callback=audio_callback): + self.start_recording() + self.start_playback() + + self.logger.info('Started recording pass-through from device ' + + '[{}] to device [{}]'. + format(input_device, output_device)) + + recording_started_time = time.time() + + while self._get_recording_state() != RecordingState.STOPPED \ + and (duration is None or + time.time() - recording_started_time < duration): + while self._get_recording_state() == RecordingState.PAUSED: + self.recording_paused_changed.wait() + + time.sleep(0.1) + + except queue.Empty as e: + self.logger.warning('Recording timeout: audio callback failed?') + finally: + self.stop_playback() + self.stop_recording() + + def start_playback(self): with self.playback_state_lock: self.playback_state = PlaybackState.PLAYING