holobench 1.25.2__py3-none-any.whl → 1.26.3__py3-none-any.whl

bencher/results/composable_container/composable_container_video.py

@@ -0,0 +1,184 @@
+from __future__ import annotations
+import numpy as np
+from copy import deepcopy
+from pathlib import Path
+from dataclasses import dataclass
+from moviepy.editor import (
+    ImageClip,
+    CompositeVideoClip,
+    clips_array,
+    concatenate_videoclips,
+    VideoClip,
+    VideoFileClip,
+)
+from moviepy.video.fx.margin import margin
+
+from bencher.results.composable_container.composable_container_base import (
+    ComposableContainerBase,
+    ComposeType,
+)
+from bencher.video_writer import VideoWriter
+
+
+@dataclass()
+class RenderCfg:
+    compose_method: ComposeType = ComposeType.sequence
+    var_name: str = None
+    var_value: str = None
+    background_col: tuple[int, int, int] = (255, 255, 255)
+    duration: float = 10.0
+    duration_target: bool = True
+    min_frame_duration: float = 1.0 / 30
+    max_frame_duration: float = 2.0
+    margin: int = 0
+
+
+@dataclass
+class ComposableContainerVideo(ComposableContainerBase):
+    def append(self, obj: VideoClip | ImageClip | str | np.ndarray) -> None:
+        """Appends an image or video to the container
+
+        Args:
+            obj (VideoClip | ImageClip | str | np.ndarray): Any representation of an image or video
+
+        Raises:
+            RuntimeWarning: if file format is not recognised
+        """
+
+        # print(f"append obj: {type(obj)}, {obj}")
+        if obj is not None:
+            if isinstance(obj, VideoClip):
+                self.container.append(obj)
+            elif isinstance(obj, ComposableContainerVideo):
+                self.container.append(obj.render())
+            elif isinstance(obj, np.ndarray):
+                self.container.append(ImageClip(obj))
+            else:
+                path = Path(obj)
+                extension = str.lower(path.suffix)
+                if extension in [".jpg", ".jepg", ".png"]:
+                    self.container.append(ImageClip(obj))
+                elif extension in [".mpeg", ".mpg", ".mp4", ".webm"]:
+                    # print(obj)
+                    self.container.append(VideoFileClip(obj))
+                else:
+                    raise RuntimeWarning(f"unsupported filetype {extension}")
+        else:
+            raise RuntimeWarning("No data passed to ComposableContainerVideo.append()")
+
+    def calculate_duration(self, frames, render_cfg: RenderCfg):
+        if render_cfg.duration_target:
+            # calculate duration based on fps constraints
+            duration = 10.0 if render_cfg.duration is None else render_cfg.duration
+            frame_duration = duration / frames
+            if render_cfg.min_frame_duration is not None:
+                frame_duration = max(frame_duration, render_cfg.min_frame_duration)
+            if render_cfg.max_frame_duration is not None:
+                frame_duration = min(frame_duration, render_cfg.max_frame_duration)
+            duration = frame_duration * frames
+        else:
+            duration = render_cfg.duration
+            frame_duration = duration / float(frames)
+
+        print("max_frame_duration", render_cfg.max_frame_duration)
+        print("DURATION", duration)
+
+        return duration, frame_duration
+
+    def render(self, render_cfg: RenderCfg = None, **kwargs) -> CompositeVideoClip:
+        """Composes the images/videos into a single image/video based on the type of compose method
+
+        Args:
+            compose_method (ComposeType, optional): optionally override the default compose type. Defaults to None.
+
+        Returns:
+            CompositeVideoClip: A composite video clip containing the images/videos added via append()
+        """
+        if render_cfg is None:
+            render_cfg = RenderCfg(**kwargs)
+
+        print("rc", render_cfg)
+        _, frame_duration = self.calculate_duration(float(len(self.container)), render_cfg)
+        out = None
+        print(f"using compose type{render_cfg.compose_method}")
+        max_duration = 0.0
+
+        for i in range(len(self.container)):
+            if self.container[i].duration is None:
+                self.container[i].duration = frame_duration
+            max_duration = max(max_duration, self.container[i].duration)
+        match render_cfg.compose_method:
+            case ComposeType.right | ComposeType.down:
+                for i in range(len(self.container)):
+                    self.container[i] = self.extend_clip(self.container[i], max_duration)
+                    self.container[i] = margin(
+                        self.container[i], top=render_cfg.margin, color=render_cfg.background_col
+                    )
+
+                if render_cfg.compose_method == ComposeType.right:
+                    clips = [self.container]
+                else:
+                    clips = [[c] for c in self.container]
+                out = clips_array(clips, bg_color=render_cfg.background_col)
+                if out.duration is None:
+                    out.duration = max_duration
+            case ComposeType.sequence:
+                out = concatenate_videoclips(
+                    self.container, bg_color=render_cfg.background_col, method="compose"
+                )
+            # case ComposeType.overlay:
+            #     for i in range(len(self.container)):
+            #         self.container[i].alpha = 1./len(self.container)
+            #     out = CompositeVideoClip(self.container, bg_color=render_args.background_col)
+            #     out.duration = fps
+            case _:
+                raise RuntimeError("This compose type is not supported")
+
+        label = self.label_formatter(render_cfg.var_name, render_cfg.var_value)
+        if label is not None:
+            # print("adding label")
+            label = ImageClip(
+                np.array(VideoWriter.create_label(label, color=render_cfg.background_col))
+            )
+            label.duration = out.duration
+            label_compose = ComposeType.down
+            if render_cfg.compose_method == ComposeType.down:
+                label_compose = ComposeType.right
+            con2 = ComposableContainerVideo()
+            con2.append(label)
+            con2.append(out)
+            return con2.render(
+                RenderCfg(
+                    background_col=render_cfg.background_col,
+                    compose_method=label_compose,
+                    duration=out.duration,
+                    duration_target=False,  # want exact duration
+                )
+            )
+        return out
+
+    def to_video(
+        self,
+        render_args: RenderCfg = None,
+    ) -> str:
+        """Returns the composite video clip as a webm file path
+
+        Returns:
+            str: webm filepath
+        """
+        return VideoWriter().write_video_raw(self.render(render_args))
+
+    def deep(self):
+        return deepcopy(self)
+
+    def extend_clip(self, clip: VideoClip, desired_duration: float):
+        if clip.duration < desired_duration:
+            return concatenate_videoclips(
+                [
+                    clip,
+                    ImageClip(
+                        clip.get_frame(clip.duration), duration=desired_duration - clip.duration
+                    ),
+                ]
+            )
+        return clip

bencher/results/float_formatter.py

@@ -0,0 +1,44 @@
+# from https://stackoverflow.com/questions/22989372/how-to-format-a-floating-number-to-maximum-fixed-width-in-python
+
+
+class FormatFloat:
+    def __init__(self, width=8):
+        self.width = width
+        self.maxnum = int("9" * (width - 1))  # 9999999
+        self.minnum = -int("9" * (width - 2))  # -999999
+
+    def __call__(self, x):
+        # for small numbers
+        # if -999,999 < given < 9,999,999:
+        if self.minnum < x < self.maxnum:
+            # o = f'{x:7}'
+            o = f"{x:{self.width - 1}}"
+
+            # converting int to float without adding zero
+            if "." not in o:
+                o += "."
+
+            # float longer than 8 will need rounding to fit width
+            elif len(o) > self.width:
+                # output = str(round(x, 7 - str(x).index(".")))
+                o = str(round(x, self.width - 1 - str(x).index(".")))
+                if len(o) < self.width:
+                    o += (self.width - len(o)) * "0"
+
+        else:
+            # for exponents
+            # added a loop for super large numbers or negative as "-" is another char
+            # Added max(max_char, 5) to account for max length of less
+            #     than 5, was having too much fun
+            # TODO can i come up with a threshold value for these up front,
+            #     so that i dont have to do this calc for every value??
+            for n in range(max(self.width, 5) - 5, 0, -1):
+                fill = f".{n}e"
+                o = f"{x:{fill}}".replace("+0", "+")
+
+                # if all good stop looping
+                if len(o) == self.width:
+                    break
+            else:
+                raise ValueError(f"Number is too large to fit in {self.width} characters", x)
+        return o