caffeinism-utils 0.0.174__tar.gz → 0.0.176__tar.gz

{caffeinism_utils-0.0.174 → caffeinism_utils-0.0.176}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: caffeinism-utils
-Version: 0.0.174
+Version: 0.0.176
 Summary: 
 Author: Kim Minjong
 Author-email: make.dirty.code@gmail.com
@@ -17,6 +17,7 @@ Provides-Extra: av
 Requires-Dist: asyncstdlib (>=3.13.1,<4.0.0)
 Requires-Dist: audioop-lts (>=0.2.1,<0.3.0) ; extra == "av"
 Requires-Dist: av (>=16.0.0,<17.0.0) ; extra == "av"
+Requires-Dist: numba (>=0.62.1,<0.63.0) ; extra == "av"
 Requires-Dist: numpy (<2) ; extra == "av"
 Requires-Dist: pydub (>=0.25.1,<0.26.0) ; extra == "av"
 Description-Content-Type: text/markdown

{caffeinism_utils-0.0.174 → caffeinism_utils-0.0.176}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "caffeinism-utils"
-version = "0.0.174"
+version = "0.0.176"
 description = ""
 authors = ["Kim Minjong <make.dirty.code@gmail.com>"]
 readme = "README.md"
@@ -15,9 +15,10 @@ av = { version = "^16.0.0", optional = true }
 pydub = { version = "^0.25.1", optional = true }
 audioop-lts = { version = "^0.2.1", optional = true }
 numpy = { version = "<2", optional = true }
+numba = { version = "^0.62.1", optional = true }
 
 [tool.poetry.extras]
-av = ["av", "pydub", "numpy", "audioop-lts"]
+av = ["av", "pydub", "numpy", "audioop-lts", "numba"]
 
 [build-system]
 requires = ["poetry-core"]

{caffeinism_utils-0.0.174 → caffeinism_utils-0.0.176}/src/caffeinism_utils/av/filters.py

@@ -208,9 +208,21 @@ class RebuildableFilter(Filter):
 
 
 def to_rgba(reader: Iterable[av.VideoFrame]):
+    reader = iter(reader)
+    first_frame = next(reader)
+
+    graph = Graph()
+    buffer = graph.add_buffer(template=first_frame)
+    buffer >> graph.add("format", pix_fmts="rgba") >> graph.add("buffersink")
+    graph.configure()
+
+    yield graph.pull()
+
     for frame in reader:
-        yield frame.reformat(format="rgba")
+        graph.push(frame)
+        yield graph.pull()
 
+    graph.push(None)
 
 def to_array(iterator: list[av.VideoFrame]):
     for frame in iterator: