Enable bufsize control on readers

tikuma-lsuhsc · tikuma-lsuhsc · commit eee964bebd67 · 2022-03-06T09:02:40.000-06:00
diff --git a/src/ffmpegio/streams/SimpleStreams.py b/src/ffmpegio/streams/SimpleStreams.py
@@ -21,6 +21,7 @@ def __init__(
         show_log=None,
         progress=None,
         blocksize=None,
+        bufsize=-1,
         **options,
     ) -> None:
 
@@ -53,7 +54,11 @@ def __init__(
 
         # start FFmpeg
         self._proc = ffmpegprocess.Popen(
-            ffmpeg_args, stdin=stdin, progress=progress, capture_log=True
+            ffmpeg_args,
+            stdin=stdin,
+            progress=progress,
+            capture_log=True,
+            bufsize=bufsize,
         )
 
         # set the log source and start the logger
@@ -215,8 +220,9 @@ def _finalize(self, ffmpeg_args):
         has_fg = configure.has_filtergraph(ffmpeg_args, "video")
 
         pix_fmt = outopts.get("pix_fmt", None)
-        if pix_fmt is None or (
-            not has_fg
+        if (
+            pix_fmt is None
+            and not has_fg
             and inurl not in ("-", "pipe:", "pipe:0")
             and not inopts.get("pix_fmt", None)
         ):
@@ -228,6 +234,9 @@ def _finalize(self, ffmpeg_args):
         else:
             pix_fmt_in = s_in = r_in = None
 
+        if pix_fmt_in is None and pix_fmt is None:
+            raise ValueError("pix_fmt must be specified.")
+
         (
             self.dtype,
             self.shape,
