numpyimage 3.0.2__tar.gz → 3.1.1__tar.gz

{numpyimage-3.0.2/numpyimage.egg-info → numpyimage-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpyimage
-Version: 3.0.2
+Version: 3.1.1
 Summary: Load, save, & manipulate image files as numpy arrays
 Author-email: Jasper Phelps <jasper.s.phelps@gmail.com>
 License: MIT License

{numpyimage-3.0.2 → numpyimage-3.1.1}/npimage/vidio.py

@@ -161,6 +161,7 @@ class VideoStreamer:
         self.container = av.open(str(self.filename))
         self.stream = self.container.streams.video[0]
         self.time_base = self.stream.time_base
+        self._frame_iterator = self.container.decode(self.stream)
         self._shape = None
         self._first_frame = None
         self._width = None
@@ -201,14 +202,14 @@ class VideoStreamer:
             frames_pts = []
             # Fallback to ffprobe if PyAV didn't work
             cmd = ['ffprobe',
-                    '-select_streams', 'v:0',
-                    '-show_frames',
-                    '-show_entries', 'frame=pts',
-                    '-of', 'default=noprint_wrappers=1:nokey=1',
-                    '-v', 'quiet',
-                    self.filename]
+                   '-select_streams', 'v:0',
+                   '-show_frames',
+                   '-show_entries', 'frame=pts',
+                   '-of', 'default=noprint_wrappers=1:nokey=1',
+                   '-v', 'quiet',
+                   self.filename]
             ffprobe_result = subprocess.run(cmd, stdout=subprocess.PIPE,
-                                    stderr=subprocess.PIPE, text=True)
+                                            stderr=subprocess.PIPE, text=True)
             if ffprobe_result.returncode != 0:
                 raise RuntimeError(f'ffprobe failed: {ffprobe_result.stderr}')
 
@@ -322,6 +323,22 @@ class VideoStreamer:
     def frame_number_to_time(self, frame_number: int) -> float:
         return float(self.frame_number_to_pts(frame_number) * self.time_base)
 
+    def pts_to_frame_number(self, pts: int) -> int:
+        if self._framerate == 'variable':
+            if pts not in self.frames_pts:
+                raise ValueError(f'PTS {pts} not in video index.')
+            return self.frames_pts.index(pts)
+        else:
+            if pts < self.pts0:
+                raise ValueError(f'PTS {pts} is before the start of the'
+                                 f' video (PTS {self.pts0}).')
+            if pts > self.pts_delta * (self.n_frames - 1) + self.pts0:
+                raise ValueError(f'PTS {pts} is after the end of the video (PTS '
+                                 f'{self.pts_delta * (self.n_frames - 1) + self.pts0}).')
+            if (pts - self.pts0) % self.pts_delta != 0:
+                raise ValueError(f'PTS {pts} is between frames for this video.')
+            return (pts - self.pts0) // self.pts_delta
+
     def __getitem__(self, key) -> np.ndarray:
         if isinstance(key, tuple):
             frame_idx = key[0]
@@ -350,9 +367,11 @@ class VideoStreamer:
             to the requested frame number.
             """
             target_pts = self.frame_number_to_pts(frame_number)
-            for frame in self.container.decode(self.stream):
+            for frame in self._frame_iterator:
                 if frame.pts is None:
-                    continue  # Skip frames without PTS
+                    if self.verbose:
+                        print('WARNING: Skipping a frame with no PTS.')
+                    continue
                 if frame.pts == target_pts:
                     frame = frame.to_ndarray(format='rgb24')
                     self._current_frame_number = frame_number
@@ -360,8 +379,8 @@ class VideoStreamer:
                 if frame.pts > target_pts:
                     raise RuntimeError(f'Frame with PTS {target_pts} not found after'
                                        f' seeking – current frame PTS: {frame.pts}')
-            raise RuntimeError(f'Frame with PTS {target_pts} not found after'
-                               f' seeking – current frame PTS: {frame.pts}')
+            raise RuntimeError('Ran out of frames before finding frame with requested'
+                               f' PTS {target_pts} – current frame PTS: {frame.pts}')
 
         if isinstance(frame_number, slice):  # Support slicing
             start, stop, step = frame_number.indices(self.n_frames)
@@ -385,10 +404,14 @@ class VideoStreamer:
 
     def _normalize_frame_number(self, frame_number: int) -> int:
         """
-        Normalize frame number to be within the valid range.
+        Support negative indexing by converting negative frame numbers to
+        positive ones, e.g. -1 becomes n_frames - 1, -2 becomes n_frames - 2, etc.
         """
-        if not isinstance(frame_number, int):
-            raise TypeError(f'Frame number must be an integer, not {type(frame_number)}')
+        try:
+            frame_number = int(frame_number)
+        except ValueError:
+            raise TypeError(f'Frame number must be castable to int but got "{frame_number}"')
+
         if frame_number < -self.n_frames:
             raise IndexError(f'Negative frame {frame_number} not in'
                              f' valid range [-{self.n_frames}, -1]')
@@ -626,6 +649,8 @@ class FFmpegVideoWriter:
         command = [
             'ffmpeg',
             '-hide_banner',
+            '-loglevel', 'error',
+            '-nostats',
             '-y',  # Overwrite output
             '-f', 'rawvideo',
             '-vcodec', 'rawvideo',
@@ -710,12 +735,13 @@ class FFmpegVideoWriter:
 
             # Wait for FFmpeg to finish
             if self._process:
+                stderr_data = self._process.stderr.read()
                 return_code = self._process.wait()
 
                 # Check for errors
                 if return_code != 0:
-                    stderr_output = self._process.stderr.read().decode()
-                    raise RuntimeError(f'FFmpeg failed with return code {return_code}: {stderr_output}')
+                    raise RuntimeError(f'FFmpeg failed with return code {return_code}:'
+                                       f' {stderr_data.decode()}')
         finally:
             # Clean up process references
             self._process = None

{numpyimage-3.0.2 → numpyimage-3.1.1/numpyimage.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpyimage
-Version: 3.0.2
+Version: 3.1.1
 Summary: Load, save, & manipulate image files as numpy arrays
 Author-email: Jasper Phelps <jasper.s.phelps@gmail.com>
 License: MIT License

{numpyimage-3.0.2 → numpyimage-3.1.1}/pyproject.toml

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