vidformer 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

vidformer/__init__.py

@@ -1,5 +1,5 @@
 """A Python library for creating and viewing videos with vidformer."""
 
-__version__ = "0.7.0"
+__version__ = "0.8.0"
 
 from .vf import *

vidformer/cv2/vf_cv2.py

@@ -46,6 +46,9 @@ LINE_8 = 8
 LINE_AA = 16
 
 _inline_mat = vf.Filter("_inline_mat")
+_slice_mat = vf.Filter("_slice_mat")
+_slice_write_mat = vf.Filter("_slice_write_mat")
+
 
 _filter_scale = vf.Filter("Scale")
 _filter_rectangle = vf.Filter("cv2.rectangle")
@@ -119,6 +122,92 @@ class Frame:
         frame = frame[:, :, ::-1]  # convert RGB to BGR
         return frame
 
+    def __getitem__(self, key):
+        if not isinstance(key, tuple):
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        if len(key) != 2:
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        if not all(isinstance(x, slice) for x in key):
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        miny = key[0].start if key[0].start is not None else 0
+        maxy = key[0].stop if key[0].stop is not None else self.shape[0]
+        minx = key[1].start if key[1].start is not None else 0
+        maxx = key[1].stop if key[1].stop is not None else self.shape[1]
+
+        # handle negative indices
+        if miny < 0:
+            miny = self.shape[0] + miny
+        if maxy < 0:
+            maxy = self.shape[0] + maxy
+        if minx < 0:
+            minx = self.shape[1] + minx
+        if maxx < 0:
+            maxx = self.shape[1] + maxx
+
+        if (
+            maxy <= miny
+            or maxx <= minx
+            or miny < 0
+            or minx < 0
+            or maxy > self.shape[0]
+            or maxx > self.shape[1]
+        ):
+            raise NotImplementedError("Invalid slice")
+
+        f = _slice_mat(self._f, miny, maxy, minx, maxx)
+        fmt = self._fmt.copy()
+        fmt["width"] = maxx - minx
+        fmt["height"] = maxy - miny
+        return Frame(f, fmt)
+
+    def __setitem__(self, key, value):
+        value = frameify(value, "value")
+
+        if not isinstance(key, tuple):
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        if len(key) != 2:
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        if not all(isinstance(x, slice) for x in key):
+            raise NotImplementedError("Only 2D slicing is supported")
+
+        miny = key[0].start if key[0].start is not None else 0
+        maxy = key[0].stop if key[0].stop is not None else self.shape[0]
+        minx = key[1].start if key[1].start is not None else 0
+        maxx = key[1].stop if key[1].stop is not None else self.shape[1]
+
+        # handle negative indices
+        if miny < 0:
+            miny = self.shape[0] + miny
+        if maxy < 0:
+            maxy = self.shape[0] + maxy
+        if minx < 0:
+            minx = self.shape[1] + minx
+        if maxx < 0:
+            maxx = self.shape[1] + maxx
+
+        if (
+            maxy <= miny
+            or maxx <= minx
+            or miny < 0
+            or minx < 0
+            or maxy > self.shape[0]
+            or maxx > self.shape[1]
+        ):
+            raise NotImplementedError("Invalid slice")
+
+        if value.shape[0] != maxy - miny or value.shape[1] != maxx - minx:
+            raise NotImplementedError("Shape mismatch")
+
+        self._mut()
+        value._mut()
+
+        self._f = _slice_write_mat(self._f, value._f, miny, maxy, minx, maxx)
+
 
 def _inline_frame(arr):
     assert arr.dtype == np.uint8
@@ -137,20 +226,6 @@ def _inline_frame(arr):
     return Frame(f, fmt)
 
 
-def _framify(obj, field_name=None):
-    if isinstance(obj, Frame):
-        return obj
-    elif isinstance(obj, np.ndarray):
-        return _inline_frame(obj)
-    else:
-        if field_name is not None:
-            raise Exception(
-                f"Unsupported type for field {field_name}, expected Frame or np.ndarray"
-            )
-        else:
-            raise Exception("Unsupported type, expected Frame or np.ndarray")
-
-
 class VideoCapture:
     def __init__(self, path):
         self._path = path
@@ -213,7 +288,7 @@ class VideoWriter:
         self._pix_fmt = "yuv420p"
 
     def write(self, frame):
-        frame = _framify(frame, "frame")
+        frame = frameify(frame, "frame")
 
         if frame._fmt["pix_fmt"] != self._pix_fmt:
             f_obj = _filter_scale(frame._f, pix_fmt=self._pix_fmt)
@@ -245,6 +320,24 @@ class VideoWriter_fourcc:
         self._args = args
 
 
+def frameify(obj, field_name=None):
+    """
+    Turn an object (e.g., ndarray) into a Frame.
+    """
+
+    if isinstance(obj, Frame):
+        return obj
+    elif isinstance(obj, np.ndarray):
+        return _inline_frame(obj)
+    else:
+        if field_name is not None:
+            raise Exception(
+                f"Unsupported type for field {field_name}, expected Frame or np.ndarray"
+            )
+        else:
+            raise Exception("Unsupported type, expected Frame or np.ndarray")
+
+
 def imread(path, *args):
     if len(args) > 0:
         raise NotImplementedError("imread does not support additional arguments")
@@ -260,7 +353,7 @@ def imwrite(path, img, *args):
     if len(args) > 0:
         raise NotImplementedError("imwrite does not support additional arguments")
 
-    img = _framify(img)
+    img = frameify(img)
 
     fmt = img._fmt.copy()
     width = fmt["width"]
@@ -316,7 +409,7 @@ def rectangle(img, pt1, pt2, color, thickness=None, lineType=None, shift=None):
     cv.rectangle(	img, pt1, pt2, color[, thickness[, lineType[, shift]]]	)
     """
 
-    img = _framify(img)
+    img = frameify(img)
     img._mut()
 
     assert len(pt1) == 2
@@ -360,7 +453,7 @@ def putText(
     cv.putText(	img, text, org, fontFace, fontScale, color[, thickness[, lineType[, bottomLeftOrigin]]]	)
     """
 
-    img = _framify(img)
+    img = frameify(img)
     img._mut()
 
     assert isinstance(text, str)
@@ -399,7 +492,7 @@ def arrowedLine(
     """
     cv.arrowedLine(	img, pt1, pt2, color[, thickness[, line_type[, shift[, tipLength]]]]	)
     """
-    img = _framify(img)
+    img = frameify(img)
     img._mut()
 
     assert len(pt1) == 2
@@ -433,7 +526,7 @@ def arrowedLine(
 
 
 def line(img, pt1, pt2, color, thickness=None, lineType=None, shift=None):
-    img = _framify(img)
+    img = frameify(img)
     img._mut()
 
     assert len(pt1) == 2
@@ -463,7 +556,7 @@ def line(img, pt1, pt2, color, thickness=None, lineType=None, shift=None):
 
 
 def circle(img, center, radius, color, thickness=None, lineType=None, shift=None):
-    img = _framify(img)
+    img = frameify(img)
     img._mut()
 
     assert len(center) == 2
@@ -514,8 +607,8 @@ def addWeighted(src1, alpha, src2, beta, gamma, dst=None, dtype=-1):
     """
     cv.addWeighted(	src1, alpha, src2, beta, gamma[, dst[, dtype]]	) -> 	dst
     """
-    src1 = _framify(src1, "src1")
-    src2 = _framify(src2, "src2")
+    src1 = frameify(src1, "src1")
+    src2 = frameify(src2, "src2")
     src1._mut()
     src2._mut()
 

{vidformer-0.7.0.dist-info → vidformer-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: vidformer
-Version: 0.7.0
+Version: 0.8.0
 Summary: A Python library for creating and viewing videos with vidformer.
 Author-email: Dominik Winecki <dominikwinecki@gmail.com>
 Requires-Python: >=3.8

vidformer-0.8.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+vidformer/__init__.py,sha256=scMlj2LLMBZXdv9ob_Dhb4ObK7RZLzxVgELyaR650xU,113
+vidformer/vf.py,sha256=GjgaKnbk3o75lwrXP3Ue0LDyTxSf-l0njL5X-6QYZvs,31496
+vidformer/cv2/__init__.py,sha256=wOjDsYyUKlP_Hye8-tyz-msu9xwaPMpN2sGMu3Lh3-w,22
+vidformer/cv2/vf_cv2.py,sha256=8shv614pmNHQyHMqjJSZf08j3eNxE_XtfdLd_kyh5no,19396
+vidformer-0.8.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+vidformer-0.8.0.dist-info/METADATA,sha256=aXFYRxtNx4hfWlqW7Zx_RW-X2CuN3-adYwRrCUkLtI4,1487
+vidformer-0.8.0.dist-info/RECORD,,

vidformer-0.7.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-vidformer/__init__.py,sha256=uCe1BJKzIv5LzLK4TZLlOtw6ru8vCUsf7vJkov5-oCE,113
-vidformer/vf.py,sha256=GjgaKnbk3o75lwrXP3Ue0LDyTxSf-l0njL5X-6QYZvs,31496
-vidformer/cv2/__init__.py,sha256=wOjDsYyUKlP_Hye8-tyz-msu9xwaPMpN2sGMu3Lh3-w,22
-vidformer/cv2/vf_cv2.py,sha256=Rta6wCIVaOAnkGXlpqE06qVjFmnn9FnuP7LUD9xrRHU,16413
-vidformer-0.7.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-vidformer-0.7.0.dist-info/METADATA,sha256=4YKlG7kViJIWONuWeJEkxdIy2iZg4wZJ4NWzaPWATBw,1487
-vidformer-0.7.0.dist-info/RECORD,,