caffeinism-utils 0.0.184__tar.gz → 0.0.185__tar.gz

{caffeinism_utils-0.0.184 → caffeinism_utils-0.0.185}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: caffeinism-utils
-Version: 0.0.184
+Version: 0.0.185
 Summary: 
 Author: Kim Minjong
 Author-email: make.dirty.code@gmail.com

{caffeinism_utils-0.0.184 → caffeinism_utils-0.0.185}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "caffeinism-utils"
-version = "0.0.184"
+version = "0.0.185"
 description = ""
 authors = ["Kim Minjong <make.dirty.code@gmail.com>"]
 readme = "README.md"

{caffeinism_utils-0.0.184 → caffeinism_utils-0.0.185}/src/caffeinism_utils/av/io.py

@@ -157,6 +157,7 @@ class BasePyAVReader(PyAVInterface):
         self,
         path,
         *,
+        codec_name: str | None = None,
         format: str,
         buffer_size: int,
         filter: tuple[type[av.frame.Frame]],
@@ -166,6 +167,7 @@ class BasePyAVReader(PyAVInterface):
         super().__init__()
         self._container = None
         self._path = path
+        self._codec_name = codec_name
         self._format = format
         self._buffer_size = buffer_size
         self._filter = filter
@@ -193,7 +195,16 @@ class BasePyAVReader(PyAVInterface):
             if av.VideoFrame in self._filter:
                 self._streams = container.streams.video
                 for stream in self._streams:
-                    if stream.codec_context.name in ("vp8", "vp9"):
+                    if (
+                        self._codec_name is not None
+                        and stream.codec_context.name != self._codec_name
+                    ):
+                        codec = av.codec.Codec(self._codec_name, "r")
+                        self._codec_contexts[stream] = codec.create()
+                    elif self._hwaccel is None and stream.codec_context.name in (
+                        "vp8",
+                        "vp9",
+                    ):
                         if stream.codec_context.name == "vp8":
                             codec_name = "libvpx"
                         elif stream.codec_context.name == "vp9":
@@ -229,6 +240,7 @@ class PyAVReader(BasePyAVReader):
         start=0,
         end=(2 << 62) - 1,
         *,
+        codec_name: str | None = None,
         hwaccel: av.codec.hwaccel.HWAccel | None = None,
         format=None,
         buffer_size=32768,
@@ -237,6 +249,7 @@ class PyAVReader(BasePyAVReader):
     ):
         super().__init__(
             path,
+            codec_name=codec_name,
             hwaccel=hwaccel,
             format=format,
             buffer_size=buffer_size,