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

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

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

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

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

{caffeinism_utils-0.0.181 → caffeinism_utils-0.0.183}/src/caffeinism_utils/av/filters.py

@@ -232,7 +232,13 @@ def to_yuva420p(reader: Iterable[av.VideoFrame]):
 
     graph = Graph()
     buffer = graph.add_buffer(template=first_frame)
-    buffer >> graph.add("format", pix_fmts="yuva420p", color_ranges="tv", color_spaces="bt709") >> graph.add("buffersink")
+    (
+        buffer
+        >> graph.add(
+            "format", pix_fmts="yuva420p", color_ranges="tv", color_spaces="bt709"
+        )
+        >> graph.add("buffersink")
+    )
     graph.configure()
 
     graph.push(first_frame)
@@ -244,6 +250,7 @@ def to_yuva420p(reader: Iterable[av.VideoFrame]):
 
     graph.push(None)
 
+
 def to_array(iterator: list[av.VideoFrame]):
     for frame in iterator:
         yield frame.to_ndarray()

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

@@ -20,6 +20,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 +103,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
@@ -416,7 +488,7 @@ class PyAVWriter(PyAVInterface):
         cc.qmax = stream_cc.qmax
         cc.qmin = stream_cc.qmin
 
-        cc.options = stream_cc.options
+        cc.options = self._options
 
         return cc
 

{caffeinism_utils-0.0.181 → caffeinism_utils-0.0.183}/src/caffeinism_utils/av/utils.py

@@ -1,6 +1,7 @@
 import av
 import numba
 
+
 @numba.jit(
     (
         numba.types.MemoryView(numba.uint8, 1, "C"),