Set default pix_fmt & sample_fmt

tikuma-lsuhsc · tikuma-lsuhsc · commit db6e19cd74ed · 2022-03-31T08:26:45.000-05:00
diff --git a/src/ffmpegio/audio.py b/src/ffmpegio/audio.py
@@ -1,8 +1,9 @@
 """Audio Read/Write Module
 """
 
-from . import ffmpegprocess, utils, configure, FFmpegError, probe, plugins
+from . import ffmpegprocess, utils, configure, FFmpegError, probe, plugins, caps
 from .utils import filter as filter_utils, log as log_utils
+import logging
 
 __all__ = ["create", "read", "write", "filter"]
 
@@ -187,11 +188,14 @@ def read(url, progress=None, show_log=None, **options):
     sample_fmt = options.get("sample_fmt", None)
     ac_in = ar_in = None
     if sample_fmt is None:
-        # use the same format as the input
-        info = probe.audio_streams_basic(url, 0)[0]
-        sample_fmt = info["sample_fmt"]
-        ac_in = info.get("ac", None)
-        ar_in = info.get("ar", None)
+        try:
+            # use the same format as the input
+            info = probe.audio_streams_basic(url, 0)[0]
+            sample_fmt = info["sample_fmt"]
+            ac_in = info.get("ac", None)
+            ar_in = info.get("ar", None)
+        except:
+            sample_fmt = 's16'
 
     input_options = utils.pop_extra_options(options, "_in")
     url, stdin, input = configure.check_url(
diff --git a/src/ffmpegio/image.py b/src/ffmpegio/image.py
@@ -1,5 +1,6 @@
 from . import ffmpegprocess, utils, configure, FFmpegError, probe, plugins
 from .utils import filter as filter_utils, log as log_utils
+import logging
 
 __all__ = ["create", "read", "write", "filter"]
 
@@ -143,9 +144,12 @@ def read(url, show_log=None, **options):
 
     # get pix_fmt of the input file only if needed
     if "pix_fmt" not in options and "pix_fmt_in" not in options:
-        info = probe.video_streams_basic(url, 0)[0]
-        pix_fmt_in = info["pix_fmt"]
-        s_in = (info["width"], info["height"])
+        try:
+            info = probe.video_streams_basic(url, 0)[0]
+            pix_fmt_in = info["pix_fmt"]
+            s_in = (info["width"], info["height"])
+        except:
+            pix_fmt_in = 'rgb24'
     else:
         pix_fmt_in = s_in = None
 
diff --git a/src/ffmpegio/media.py b/src/ffmpegio/media.py
@@ -34,7 +34,7 @@ def read(*urls, progress=None, show_log=None, **options):
     Unlike :py:mod:`video` and :py:mod:`image`, video pixel formats are not autodetected. If output
     'pix_fmt' option is not explicitly set, 'rgb24' is used.
 
-    For audio streams, if 'sample_fmt' output option is not specified, 's16le'.
+    For audio streams, if 'sample_fmt' output option is not specified, 's16'.
 
 
     streams = ['0:v:0','1:a:3'] # pick 1st file's 1st video stream and 2nd file's 4th audio stream
diff --git a/src/ffmpegio/streams/AviStreams.py b/src/ffmpegio/streams/AviStreams.py
@@ -32,7 +32,7 @@ class AviMediaReader:
     Unlike :py:mod:`video` and :py:mod:`image`, video pixel formats are not autodetected. If output
     'pix_fmt' option is not explicitly set, 'rgb24' is used.
 
-    For audio streams, if 'sample_fmt' output option is not specified, 's16le'.
+    For audio streams, if 'sample_fmt' output option is not specified, 's16'.
 
 
     streams = ['0:v:0','1:a:3'] # pick 1st file's 1st video stream and 2nd file's 4th audio stream
diff --git a/src/ffmpegio/streams/SimpleStreams.py b/src/ffmpegio/streams/SimpleStreams.py
@@ -226,11 +226,14 @@ def _finalize(self, ffmpeg_args):
             and inurl not in ("-", "pipe:", "pipe:0")
             and not inopts.get("pix_fmt", None)
         ):
-            # must assign output rgb/grayscale pixel format
-            info = probe.video_streams_basic(inurl, 0)[0]
-            pix_fmt_in = info["pix_fmt"]
-            s_in = (info["width"], info["height"])
-            r_in = info["frame_rate"]
+            try:
+                # must assign output rgb/grayscale pixel format
+                info = probe.video_streams_basic(inurl, 0)[0]
+                pix_fmt_in = info["pix_fmt"]
+                s_in = (info["width"], info["height"])
+                r_in = info["frame_rate"]
+            except:
+                pix_fmt_in = 'rgb24'
         else:
             pix_fmt_in = s_in = r_in = None
 
@@ -290,7 +293,7 @@ def _finalize(self, ffmpeg_args):
                 ac_in = info.get("channels", None)
                 ar_in = info.get("sample_rate", None)
             except:
-                pass
+                sample_fmt_in = 's16'
 
         (
             self.dtype,
diff --git a/src/ffmpegio/video.py b/src/ffmpegio/video.py
@@ -1,5 +1,6 @@
-from . import ffmpegprocess, utils, configure, FFmpegError, probe, plugins
+from . import ffmpegprocess, utils, configure, FFmpegError, probe, plugins, caps
 from .utils import filter as filter_utils, log as log_utils
+import logging
 
 __all__ = ["create", "read", "write", "filter"]
 
@@ -164,10 +165,13 @@ def read(url, progress=None, show_log=None, **options):
 
     # get pix_fmt of the input file only if needed
     if pix_fmt is None and "pix_fmt_in" not in options:
-        info = probe.video_streams_basic(url, 0)[0]
-        pix_fmt_in = info["pix_fmt"]
-        s_in = (info["width"], info["height"])
-        r_in = info["frame_rate"]
+        try:
+            info = probe.video_streams_basic(url, 0)[0]
+            pix_fmt_in = info["pix_fmt"]
+            s_in = (info["width"], info["height"])
+            r_in = info["frame_rate"]
+        except:
+            pix_fmt_in = 'rgb24'
     else:
         pix_fmt_in = s_in = r_in = None
 
