Compressed audio (continuation of #36)

audioread/__init__.py

@@ -112,3 +112,18 @@ def audio_open(path):
 
     # All backends failed!
     raise NoBackendError()
+
+
+def decode(audio):
+    """Given a file-like object containing encoded audio data, create an
+    audio file object that produces its *raw* data.
+    """
+    # FFmpeg.
+    from . import ffdec
+    try:
+        return ffdec.FFmpegAudioFile(audio=audio)
+    except DecodeError:
+        pass
+
+    # All backends failed!
+    raise NoBackendError()

audioread/ffdec.py

@@ -51,6 +51,8 @@ class NotInstalledError(FFmpegError):
 class ReadTimeoutError(FFmpegError):
     """Reading from the ffmpeg command-line tool timed out."""
 
+class NoInputError(FFmpegError):
+    """Reading from the ffmpeg command-line tool timed out."""
 
 class QueueReaderThread(threading.Thread):
     """A thread that consumes data from a filehandle and sends the data
@@ -73,6 +75,18 @@ def run(self):
                 # Stream closed (EOF).
                 break
 
+class WriterThread(threading.Thread):
+    """A thread that writes data to a filehandle
+    """
+    def __init__(self, fh, audio=None):
+        super(WriterThread, self).__init__()
+        self.fh = fh
+        self.audio = audio
+        self.daemon = True
+
+    def run(self):
+        self.fh.write(self.audio.read())
+        self.fh.close()
 
 def popen_multiple(commands, command_args, *args, **kwargs):
     """Like `subprocess.Popen`, but can try multiple commands in case
@@ -100,7 +114,10 @@ def popen_multiple(commands, command_args, *args, **kwargs):
 
 class FFmpegAudioFile(object):
     """An audio file decoded by the ffmpeg command-line utility."""
-    def __init__(self, filename, block_size=4096):
+    def __init__(self, filename=None, audio=None, block_size=4096):
+        self.openFile = True if filename is not None else False
+        self.readAudio = True if audio is not None else False
+
         # On Windows, we need to disable the subprocess's crash dialog
         # in case it dies. Passing SEM_NOGPFAULTERRORBOX to SetErrorMode
         # disables this behavior.
@@ -118,14 +135,26 @@ def __init__(self, filename, block_size=4096):
             )
 
         try:
-            self.devnull = open(os.devnull)
-            self.proc = popen_multiple(
-                COMMANDS,
-                ['-i', filename, '-f', 's16le', '-'],
-                stdout=subprocess.PIPE,
-                stderr=subprocess.PIPE,
-                stdin=self.devnull,
-            )
+            if self.openFile:
+                self.devnull = open(os.devnull)
+                self.proc = popen_multiple(
+                    COMMANDS,
+                    ['-i', filename, '-f', 's16le', '-'],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    stdin=self.devnull,
+                )
+            elif self.readAudio:
+                self.devnull = open(os.devnull)
+                self.proc = popen_multiple(
+                    COMMANDS,
+                    ['-i', '-', '-f', 's16le', '-'],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    stdin=subprocess.PIPE,
+                )
+            else:
+                raise NoInputError()
 
         except OSError:
             raise NotInstalledError()
@@ -141,6 +170,11 @@ def __init__(self, filename, block_size=4096):
                 finally:
                     windows_error_mode_lock.release()
 
+        # Start a thread to write the compressed audio to Popen.stdin
+        if self.readAudio:
+            self.stdin_writer = WriterThread(self.proc.stdin,audio)
+            self.stdin_writer.start()
+
         # Start another thread to consume the standard output of the
         # process, which contains raw audio data.
         self.stdout_reader = QueueReaderThread(self.proc.stdout, block_size)

decode.py

@@ -21,21 +21,31 @@
 import contextlib
 
 
-def decode(filename):
-    filename = os.path.abspath(os.path.expanduser(filename))
-    if not os.path.exists(filename):
-        print("File not found.", file=sys.stderr)
-        sys.exit(1)
+def decode(filename=None):
+    """Decode audio from a file on disk or, if no file is specified,
+    from the standard input.
+    """
+    if filename:
+        filename = os.path.abspath(os.path.expanduser(filename))
+        if not os.path.exists(filename):
+            print("File not found.", file=sys.stderr)
+            sys.exit(1)
 
     try:
-        with audioread.audio_open(filename) as f:
+        if filename:
+            f = audioread.audio_open(filename)
+        else:
+            f = audioread.decode(sys.stdin)
+
+        with f:
             print('Input file: %i channels at %i Hz; %.1f seconds.' %
                   (f.channels, f.samplerate, f.duration),
                   file=sys.stderr)
             print('Backend:', str(type(f).__module__).split('.')[1],
                   file=sys.stderr)
 
-            with contextlib.closing(wave.open(filename + '.wav', 'w')) as of:
+            outname = filename or 'out'
+            with contextlib.closing(wave.open(outname + '.wav', 'w')) as of:
                 of.setnchannels(f.channels)
                 of.setframerate(f.samplerate)
                 of.setsampwidth(2)
@@ -49,4 +59,7 @@ def decode(filename):
 
 
 if __name__ == '__main__':
-    decode(sys.argv[1])
+    if sys.argv[1:]:
+        decode(sys.argv[1])
+    else:
+        decode()