scope 0.5.3__tar.gz → 0.5.4__tar.gz

{scope-0.5.3 → scope-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scope
-Version: 0.5.3
+Version: 0.5.4
 Summary: Metrics logging and analysis
 Home-page: http://github.com/danijar/scope
 Classifier: Intended Audience :: Science/Research

{scope-0.5.3 → scope-0.5.4}/requirements.txt

@@ -1,6 +1,7 @@
 av
 elements>=3.16.6
 fastapi
+mediapy
 numpy
 pillow
 uvicorn[standard]

{scope-0.5.3 → scope-0.5.4}/scope/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.5.3'
+__version__ = '0.5.4'
 
 from .reader import Reader
 from .writer import Writer

{scope-0.5.3 → scope-0.5.4}/scope/formats.py

@@ -176,6 +176,47 @@ class Video:
     return files_length(path)
 
 
+class MediapyVideo:
+
+  def __init__(self, ext='mp4', fps=10):
+    self.ext = ext
+    self.fps = fps
+
+  @property
+  def extension(self):
+    return self.ext
+
+  def valid(self, x):
+    return (
+        isinstance(x, np.ndarray) and
+        x.dtype == np.uint8 and
+        x.ndim == 4 and
+        x.shape[-1] in (1, 3))
+
+  def convert(self, x):
+    return x
+
+  def create(self, path):
+    path.mkdir(exist_ok=True)
+
+  def write(self, path, steps, values):
+    files_write(path, steps, values, self.encode)
+
+  def read(self, path):
+    return files_read(path)
+
+  def encode(self, value):
+    import mediapy
+    return mediapy.compress_video(value, fps=self.fps)
+
+  def decode(self, buffer):
+    import mediapy
+    return mediapy.decompress_video(buffer)
+
+  def length(self, path):
+    return files_length(path)
+
+
 def table_append(filename, fmt, *cols):
   rows = tuple(zip(*cols))
   size = struct.calcsize(fmt)

{scope-0.5.3 → scope-0.5.4}/scope/writer.py

@@ -7,13 +7,12 @@ import numpy as np
 
 from . import formats
 
-FPS = 10
 
 FORMATS = [
     formats.Text(),
     formats.Float(),
     formats.Image(),
-    formats.Video(fps=FPS),
+    formats.Video(fps=10),
 ]