cv-frames 0.0.0__py3-none-any.whl

cv_frames-0.0.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: cv-frames
+Version: 0.0.0
+Summary: Read frames from OpenCV like humans
+Requires-Dist: opencv-python
+Requires-Dist: numpy

cv_frames-0.0.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+cvframes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cvframes/iterate.py,sha256=XA5HOTkF-6y-5bkd8ExZn4H9_S_CT1WaJ05byZ-3h2Q,2688
+cv_frames-0.0.0.dist-info/METADATA,sha256=MriWGUqYdMSzEoDCbZKpj1MlZdMS_f8fsvxmW70WWe8,148
+cv_frames-0.0.0.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+cv_frames-0.0.0.dist-info/top_level.txt,sha256=GJnY7NlV_RqhT_FG8XHpzTfSdF4GlYJR6pWj8iJ0BSk,9
+cv_frames-0.0.0.dist-info/RECORD,,

cv_frames-0.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (77.0.3)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

cv_frames-0.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+cvframes

cvframes/iterate.py

@@ -0,0 +1,106 @@
+from pathlib import Path
+from typing import Callable, Generator, Optional, Tuple, TypeVar
+
+import cv2
+import numpy as np
+
+T = TypeVar("T")
+
+
+class IOCapture:
+    def __init__(self, source: str | Path, oname: str | Path = ""):
+        self.icap = cv2.VideoCapture(str(source))
+        self.ocap = (
+            cv2.VideoWriter(
+                str(oname),
+                cv2.VideoWriter_fourcc(*"mp4v"),
+                self.icap.get(cv2.CAP_PROP_FPS),
+                (
+                    int(self.icap.get(cv2.CAP_PROP_FRAME_WIDTH)),
+                    int(self.icap.get(cv2.CAP_PROP_FRAME_HEIGHT)),
+                ),
+            )
+            if oname
+            else None
+        )
+
+    def is_opened(self) -> bool:
+        return self.icap.isOpened()
+
+    def read(self) -> Tuple[bool, np.ndarray]:
+        return self.icap.read()
+
+    def write(self, frame: np.ndarray) -> None:
+        if self.ocap is not None:
+            self.ocap.write(frame)
+
+    def release(self) -> None:
+        self.icap.release()
+        if self.ocap is not None:
+            self.ocap.release()
+
+    def set(self, prop_id: int, value: float) -> None:
+        self.icap.set(prop_id, value)
+
+
+def iterate_generic(
+    ipath: Path,
+    opath: Optional[Path],
+    start_frame: int,
+    stop_frame: int,
+    process_frames: Callable[[np.ndarray], T],
+) -> Generator[tuple[IOCapture, T], None, None]:
+    capture = IOCapture(str(ipath), oname=opath or "")
+    capture.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
+    count = start_frame
+
+    if not capture.is_opened():
+        return
+
+    try:
+        while True:
+            ret, frame = capture.read()
+            count += 1
+            if not ret:
+                break
+            if stop_frame > 0 and count >= stop_frame:
+                break
+
+            yield capture, process_frames(frame)
+    finally:
+        capture.release()
+
+
+def iterate(
+    ipath: Path,
+    opath: Optional[Path] = None,
+    start_frame: int = -1,
+    stop_frame: int = -1,
+) -> Generator[tuple[IOCapture, np.ndarray], None, None]:
+    return iterate_generic(
+        ipath,
+        opath,
+        start_frame,
+        stop_frame,
+        lambda frame: frame,
+    )
+
+
+def iterate_sbs(
+    ipath: Path,
+    opath: Optional[Path] = None,
+    start_frame: int = -1,
+    stop_frame: int = -1,
+) -> Generator[Tuple[IOCapture, Tuple[np.ndarray, np.ndarray]], None, None]:
+    def processor(frame: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
+        _, width, _ = frame.shape
+        mid = width // 2
+        return frame[:, :mid, :], frame[:, mid:, :]
+
+    return iterate_generic(
+        ipath,
+        opath,
+        start_frame,
+        stop_frame,
+        processor,
+    )