caffeinism-utils 0.0.182__py3-none-any.whl → 0.0.184__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
@@ -20,6 +21,62 @@ from .filters import AlphaExtractor, AlphaMerger, BaseAlphaMerger, NotAlphaMerge
 
 TIME_BASE = fractions.Fraction(1, 90000)
 
+ffmpeg_color_info = {
+    "color_range": {0: "unknown", 1: "tv", 2: "pc"},
+    "colorspace": {
+        0: "rgb",
+        1: "bt709",
+        2: "unknown",
+        4: "fcc",
+        5: "bt470bg",
+        6: "smpte170m",
+        7: "smpte240m",
+        8: "ycgco",
+        9: "bt2020nc",
+        10: "bt2020c",
+        11: "smpte2085",
+        12: "chroma-derived-nc",
+        13: "chroma-derived-c",
+        14: "ictcp",
+        15: "ipt-c2",
+        16: "ycgco-re",
+        17: "ycgco-ro",
+    },
+    "color_primaries": {
+        1: "bt709",
+        2: "unknown",
+        4: "bt470m",
+        5: "bt470bg",
+        6: "smpte170m",
+        7: "smpte240m",
+        8: "film",
+        9: "bt2020",
+        10: "smpte428",
+        11: "smpte431",
+        12: "smpte432",
+        22: "jedec-p22",
+    },
+    "color_trc": {
+        1: "bt709",
+        2: "unknown",
+        4: "gamma22",
+        5: "gamma28",
+        6: "smpte170m",
+        7: "smpte240m",
+        8: "linear",
+        9: "log100",
+        10: "log316",
+        11: "iec61966-2-4",
+        12: "bt1361e",
+        13: "iec61966-2-1",
+        14: "bt2020-10",
+        15: "bt2020-12",
+        16: "smpte2084",
+        17: "smpte428",
+        18: "arib-std-b67",
+    },
+}
+
 
 class PyAVInterface(ABC):
     _container: av.container.Container
@@ -47,6 +104,22 @@ class PyAVInterface(ABC):
         self._init_container()
         return self._streams
 
+    @property
+    def colorspace(self) -> str:
+        return ffmpeg_color_info["colorspace"][self.streams[0].colorspace]
+
+    @property
+    def color_trc(self) -> str:
+        return ffmpeg_color_info["color_trc"][self.streams[0].color_trc]
+
+    @property
+    def color_primaries(self) -> str:
+        return ffmpeg_color_info["color_primaries"][self.streams[0].color_primaries]
+
+    @property
+    def color_range(self) -> str:
+        return ffmpeg_color_info["color_range"][self.streams[0].color_range]
+
     @property
     def fps(self):
         return self.streams[0].base_rate or self.streams[0].codec_context.framerate
@@ -87,6 +160,7 @@ class BasePyAVReader(PyAVInterface):
         format: str,
         buffer_size: int,
         filter: tuple[type[av.frame.Frame]],
+        hwaccel: av.codec.hwaccel.HWAccel | None = None,
         options={},
     ):
         super().__init__()
@@ -95,6 +169,7 @@ class BasePyAVReader(PyAVInterface):
         self._format = format
         self._buffer_size = buffer_size
         self._filter = filter
+        self._hwaccel = hwaccel
         self._options = options
 
         self._codec_contexts = {}
@@ -110,6 +185,7 @@ class BasePyAVReader(PyAVInterface):
                 "r",
                 format=self._format,
                 buffer_size=self._buffer_size,
+                hwaccel=self._hwaccel,
                 options=self._options,
             )
 
@@ -153,13 +229,19 @@ class PyAVReader(BasePyAVReader):
         start=0,
         end=(2 << 62) - 1,
         *,
+        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,
+            hwaccel=hwaccel,
+            format=format,
+            buffer_size=buffer_size,
+            filter=filter,
+            options=options,
         )
         self.start = start
         self.end = end

{caffeinism_utils-0.0.182.dist-info → caffeinism_utils-0.0.184.dist-info}/METADATA

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

{caffeinism_utils-0.0.182.dist-info → caffeinism_utils-0.0.184.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=YIJS8GpIzH9bWcN4rgXprvn5El4N6JkUGEBXBom3Tis,16874
+caffeinism_utils/av/io.py,sha256=GKY_ZeBsllmIofw5ZoG5TAbLS-DIKZ4V2ryU2lYDBOY,18900
 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.182.dist-info/METADATA,sha256=CWTuiV5oDZJiewhj7WtmHQYSQoCdj8AtctDATQHMC-g,955
-caffeinism_utils-0.0.182.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-caffeinism_utils-0.0.182.dist-info/RECORD,,
+caffeinism_utils-0.0.184.dist-info/METADATA,sha256=FJopy1Q_ETf3qVIDJm12izzLkUFYFO4wHqba_b94KkE,955
+caffeinism_utils-0.0.184.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
+caffeinism_utils-0.0.184.dist-info/RECORD,,

{caffeinism_utils-0.0.182.dist-info → caffeinism_utils-0.0.184.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