numpyimage 2.7.0__tar.gz → 2.9.0__tar.gz

{numpyimage-2.7.0/numpyimage.egg-info → numpyimage-2.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpyimage
-Version: 2.7.0
+Version: 2.9.0
 Summary: Load, save, & manipulate image files as numpy arrays
 Author-email: Jasper Phelps <jasper.s.phelps@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE

{numpyimage-2.7.0 → numpyimage-2.9.0}/npimage/vidio.py

@@ -439,8 +439,16 @@ class VideoWriter:
         self.stream.height = 0
 
     def write(self, frame):
+        if self._closed:
+            raise RuntimeError('VideoWriter is closed, cannot write more frames.')
         if not isinstance(frame, self.av.VideoFrame):
-            if frame.ndim == 3 and frame.shape[-1] == 3:
+            if not isinstance(frame, np.ndarray):
+                frame = np.array(frame)
+            if frame.ndim == 4:
+                for i in range(frame.shape[0]):
+                    self.write(frame[i])
+                return
+            elif frame.ndim == 3 and frame.shape[-1] == 3:
                 frame = self.av.VideoFrame.from_ndarray(frame, format='rgb24')
             elif frame.ndim == 3 and frame.shape[-1] == 4:
                 # While some video codecs support an alpha channel, most don't,
@@ -449,7 +457,8 @@ class VideoWriter:
             elif frame.ndim == 2:
                 frame = self.av.VideoFrame.from_ndarray(frame, format='gray')
             else:
-                raise ValueError(f'Frame must be (H, W) (H, W, 3) or (H, W, 4) but was {frame.shape}')
+                raise ValueError('Frame must have shape (H, W) (H, W, 3) (H, W, 4)'
+                                 f' (t, H, W, 3) or (t, H, W, 4) but was {frame.shape}')
         if self.stream.width == 0:
             self.stream.width = frame.width
         if self.stream.height == 0:
@@ -461,6 +470,9 @@ class VideoWriter:
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def close(self):
         # Flush stream
         for packet in self.stream.encode():
             self.container.mux(packet)

{numpyimage-2.7.0 → numpyimage-2.9.0/numpyimage.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpyimage
-Version: 2.7.0
+Version: 2.9.0
 Summary: Load, save, & manipulate image files as numpy arrays
 Author-email: Jasper Phelps <jasper.s.phelps@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE

{numpyimage-2.7.0 → numpyimage-2.9.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = 'setuptools.build_meta'
 
 [project]
 name = 'numpyimage'
-version = '2.7.0'
+version = '2.9.0'
 description = 'Load, save, & manipulate image files as numpy arrays'
 readme.file = 'README.md'
 readme.content-type = 'text/markdown'