python-ffmpegio
diff --git a/‎.vscode/settings.json‎
Lines changed: 0 additions & 4 deletions b/‎.vscode/settings.json‎
Lines changed: 0 additions & 4 deletions
diff --git a/‎src/ffmpegio/audio.py‎
Lines changed: 27 additions & 33 deletions b/‎src/ffmpegio/audio.py‎
Lines changed: 27 additions & 33 deletions
@@ -1,11 +1,8 @@
-import sys
 import numpy as np
-from . import ffmpeg, probe, utils
+from . import ffmpeg, utils, configure
 
 
-
-
-def read(filename, **inopts):
+def read(input_url, stream_id=0, **options):
     """Open an audio file.
 
     :param filename: Input media file.
@@ -14,49 +11,46 @@ def read(filename, **inopts):
     :return: sample rate and audio data matrix (column=time,row=channel)
     :rtype: (float, numpy.ndarray)
     """
-    info = probe.audio_streams_basic(
-        filename, index=0, entries=("sample_rate", "sample_fmt", "channels")
-    )[0]
 
-    acodec, dtype = utils.get_audio_format(info["sample_fmt"])
+    args = configure.input_timing(input_url, astream_id=stream_id, **options)
 
-    args = dict(
-        inputs=[(filename, None,)],
-        outputs=[("-", dict(vn=None, acodec=acodec, f="rawvideo", map="a:0"))],
+    args, reader_cfg = configure.audio_io(
+        input_url,
+        stream_id,
+        output_url="-",
+        format="rawvideo",
+        ffmpeg_args=args,
+        **options,
     )
 
+    dtype, nch, rate = reader_cfg[0]
     stdout = ffmpeg.run_sync(args)
-    return (
-        info["sample_rate"],
-        np.frombuffer(stdout, dtype=dtype).reshape(-1, info["channels"]),
-    )
+    return rate, np.frombuffer(stdout, dtype=dtype).reshape(-1, nch)
 
 
-def write(filename, rate, data, **outopts):
+def write(url, rate, data, **options):
     """Write a NumPy array as an audio file.
 
-    :param filename: Output media file.
-    :type filename: str
+    :param url: Output media file.
+    :type url: str
     :param rate: The sample rate (in samples/sec).
     :type rate: int
     :param data: A 1-D or 2-D NumPy array of either integer or float data-type.
     :type data: numpy.ndarray
     :raises Exception: FFmpeg error
     """
-    acodec, _ = utils.get_audio_format(data.dtype)
-    args = dict(
-        inputs=[
-            (
-                "-",
-                dict(
-                    vn=None,
-                    f=acodec[4:],
-                    ar=rate,
-                    channels=data.shape[1] if data.ndim > 1 else 1,
-                ),
-            )
-        ],
-        outputs=[(filename, None,)],
+    args = configure.input_timing(
+        "-",
+        astream_id=0,
+        excludes=("start", "end", "duration"),
+        **{"input_sample_rate": rate, **options},
+    )
+
+    configure.audio_io(
+        utils.array_to_audio_input(data, format=True),
+        output_url=url,
+        ffmpeg_args=args,
+        **options,
     )
 
     ffmpeg.run_sync(args, input=data.tobytes())