enot-vp 0.1.0__py3-none-any.whl

enot_vp/__init__.py

@@ -0,0 +1 @@
+from enot_vp.video_processor import VideoProcessor as VideoProcessor

enot_vp/video_processor.py

@@ -0,0 +1,75 @@
+import av
+import numpy as np
+
+
+class VideoProcessor:
+    def __init__(self, input_video: str, output_video: str | None = None):
+        self.input_video = input_video
+        self.output_video = output_video
+        self.input_container = None
+        self.output_container = None
+        self.input_stream = None
+        self.output_stream = None
+        self._frame_iterator = None
+
+    def __enter__(self):
+        # Open input video
+        self.input_container = av.open(self.input_video)
+        self.input_stream = self.input_container.streams.video[0]
+
+        # Set up output if specified
+        if self.output_video:
+            self.output_container = av.open(self.output_video, mode="w")
+            self.output_stream = self.output_container.add_stream(
+                codec_name="hevc",
+                rate=self.input_stream.average_rate,
+                width=self.input_stream.width,
+                height=self.input_stream.height,
+            )
+            self.output_stream.pix_fmt = "yuv420p"
+
+        # Create frame iterator with progress bar
+        self._frame_iterator = self.input_container.decode(self.input_stream)
+
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        del exc_tb
+
+        if self.output_container:
+            # Flush any remaining frames in the encoder
+            for packet in self.output_stream.encode():
+                self.output_container.mux(packet)
+            self.output_container.close()
+
+        self.input_container.close()
+
+        if exc_type is not None:
+            print(f"Error occurred: {exc_val}")
+
+        return True
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        if self._frame_iterator is None:
+            raise RuntimeError("VideoProcessor not initialized. Use in a 'with' block.")
+
+        try:
+            return next(self._frame_iterator)
+        except StopIteration:
+            self._frame_iterator.close()
+            raise
+
+    def put(self, processed_frame: av.VideoFrame | np.ndarray):
+        """Add video frame to output video."""
+        if self.output_container is None:
+            raise RuntimeError("No output video specified in constructor")
+
+        # Convert frame to AV VideoFrame and encode
+        if not isinstance(processed_frame, av.VideoFrame):
+            processed_frame = av.VideoFrame.from_ndarray(processed_frame)
+
+        for packet in self.output_stream.encode(processed_frame):
+            self.output_container.mux(packet)

enot_vp-0.1.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: enot-vp
+Version: 0.1.0
+Summary: ENOT video processor package for iterating over video frames easily.
+Author-email: Malofeev Ivan <ruhrozz@bk.ru>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Requires-Dist: av>=14.4.0
+Requires-Dist: numpy>=2.0.2

enot_vp-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+enot_vp/__init__.py,sha256=0jTdY7p0CdAbF2pqIZZpC_3nFvnUgQolDAl_cWfqoc0,69
+enot_vp/video_processor.py,sha256=FsHiBCFQoWpT942I9_L0pVZIzM9j45crceSmLrPeaWA,2494
+enot_vp-0.1.0.dist-info/METADATA,sha256=ZlAuvBXEvvLgDIqj3Z71QgKZupXV3G6JUVUgiyyzx3s,391
+enot_vp-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+enot_vp-0.1.0.dist-info/licenses/LICENSE,sha256=7EI8xVBu6h_7_JlVw-yPhhOZlpY9hP8wal7kHtqKT_E,1074
+enot_vp-0.1.0.dist-info/RECORD,,

enot_vp-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

enot_vp-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2018 The Python Packaging Authority
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.