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

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

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

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

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

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

@@ -9,6 +9,7 @@ from types import TracebackType
 from typing import Generator, Iterable
 
 import av
+import av.codec.hwaccel
 import av.container
 import av.frame
 import av.stream
@@ -156,17 +157,21 @@ class BasePyAVReader(PyAVInterface):
         self,
         path,
         *,
+        codec_name: str | None = None,
         format: str,
         buffer_size: int,
         filter: tuple[type[av.frame.Frame]],
+        hwaccel: av.codec.hwaccel.HWAccel | None = None,
         options={},
     ):
         super().__init__()
         self._container = None
         self._path = path
+        self._codec_name = codec_name
         self._format = format
         self._buffer_size = buffer_size
         self._filter = filter
+        self._hwaccel = hwaccel
         self._options = options
 
         self._codec_contexts = {}
@@ -182,6 +187,7 @@ class BasePyAVReader(PyAVInterface):
                 "r",
                 format=self._format,
                 buffer_size=self._buffer_size,
+                hwaccel=self._hwaccel,
                 options=self._options,
             )
 
@@ -189,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":
@@ -225,13 +240,21 @@ 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,
         filter=(av.VideoFrame, av.AudioFrame),
         options={},
     ):
         super().__init__(
-            path, format=format, buffer_size=buffer_size, filter=filter, options=options
+            path,
+            codec_name=codec_name,
+            hwaccel=hwaccel,
+            format=format,
+            buffer_size=buffer_size,
+            filter=filter,
+            options=options,
         )
         self.start = start
         self.end = end