caffeinism-utils 0.0.183__py3-none-any.whl → 0.0.185__py3-none-any.whl

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

{caffeinism_utils-0.0.183.dist-info → caffeinism_utils-0.0.185.dist-info}/METADATA

@@ -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.dist-info → caffeinism_utils-0.0.185.dist-info}/RECORD

@@ -4,7 +4,7 @@ caffeinism_utils/asyncio.py,sha256=33yJ51vY0JvS6HJiBsEXXxAE3UlpLeudK8tanlu599Y,4
 caffeinism_utils/av/__init__.py,sha256=J7dtpfW9R9BH-HJmlubAaFMqcEOMOu--zH64XJ7A2hI,423
 caffeinism_utils/av/codecs.py,sha256=gkOeazEqJWC1fRTLGpjdQbeorpVZpRHDzN7Ky01AML8,3360
 caffeinism_utils/av/filters.py,sha256=-jb7S96bQuhGAnEaohzfVZbTqXDnl_fHW1XOUfpzTb4,16229
-caffeinism_utils/av/io.py,sha256=WS5NcFWVdUQo8mGh0w0v8dnRkSV0XJMBfZZUeU9WVVQ,18613
+caffeinism_utils/av/io.py,sha256=XyiAscZzpyDStgr3cp_CxZNvav0E7OGwpOEPXhayHiA,19465
 caffeinism_utils/av/utils.py,sha256=bBbmwDO0PCX9rJvfho9GO7hjBZqXudEjAvrgc9dWD94,6196
 caffeinism_utils/http.py,sha256=ZDcdHxaSPzbo6Jy2RfWl6Ib5Hii2170AkXlb-KKuI4I,424
 caffeinism_utils/io.py,sha256=cgA7LwV8zgfEd3Qaubogow3V21IR4yp2drRmOgql6Tw,3771
@@ -12,6 +12,6 @@ caffeinism_utils/iteration.py,sha256=7BddGudGRBmNyJi3cTpmcQ2zWplu5zjxYCN4J_cWe3c
 caffeinism_utils/prefetch.py,sha256=yNTbBaSF-BA5iA1zUfnWVvcMm3br5nlBmPgvBMXHO7I,4144
 caffeinism_utils/utils.py,sha256=-g0DDuiCrck4xjN9T5ZUaklClXqDLCfOB6PKN4D8mlE,927
 caffeinism_utils/zip.py,sha256=L1d8qfphao-iD5G7bDu4LMQNg6ULc4kCqUk5gJbhSdE,1679
-caffeinism_utils-0.0.183.dist-info/METADATA,sha256=T3KzhjPay0auUGkRUoSsSDG3jaMIFn2BR4T7eoKVvkU,955
-caffeinism_utils-0.0.183.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-caffeinism_utils-0.0.183.dist-info/RECORD,,
+caffeinism_utils-0.0.185.dist-info/METADATA,sha256=Oj7cFqWiTGknkKN3Br4SDKpyCVsgWInmZTM0ta8wRUI,955
+caffeinism_utils-0.0.185.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
+caffeinism_utils-0.0.185.dist-info/RECORD,,

{caffeinism_utils-0.0.183.dist-info → caffeinism_utils-0.0.185.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.1
+Generator: poetry-core 2.3.1
 Root-Is-Purelib: true
 Tag: py3-none-any