Add a rewind option to wiggle

neuromorphicsystems · Dec 10, 2023 · 58042c4 · 58042c4
1 parent 0638f55
commit 58042c4
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 21 deletions.
diff --git a/charidotella/__init__.py b/charidotella/__init__.py
@@ -582,6 +582,7 @@ def run_generators(configuration: dict[str, typing.Any]):
                             "forward_duration": utilities.timestamp_to_timecode(
                                 1000000
                             ),
+                            "rewind": True,
                             "tau_to_frametime_ratio": 3.0,
                             "style": "cumulative",
                             "idle_color": "#191919",

diff --git a/charidotella/tasks/wiggle.py b/charidotella/tasks/wiggle.py
@@ -55,27 +55,34 @@ def run(
         )
         sorted_frames = temporary_directory / "sorted_frames"
         sorted_frames.mkdir(exist_ok=True)
-        paths: list[pathlib.Path] = []
-        for path in sorted(base_frames.iterdir()):
-            if path.suffix == ".ppm":
-                paths.append(path)
-        selected_paths: list[pathlib.Path] = []
-        if len(paths) >= round(parameters["tau_to_frametime_ratio"]) + 2:
-            for index, path in enumerate(paths):
-                if (
-                    index >= round(parameters["tau_to_frametime_ratio"])
-                    and index % 2 == 1
-                ):
-                    selected_paths.append(path)
+        if parameters["rewind"]:
+            paths: list[pathlib.Path] = []
+            for path in sorted(base_frames.iterdir()):
+                if path.suffix == ".ppm":
+                    paths.append(path)
+            selected_paths: list[pathlib.Path] = []
+            if len(paths) >= round(parameters["tau_to_frametime_ratio"]) + 2:
+                for index, path in enumerate(paths):
+                    if (
+                        index >= round(parameters["tau_to_frametime_ratio"])
+                        and index % 2 == 1
+                    ):
+                        selected_paths.append(path)
+            else:
+                selected_paths = paths
+            output_index = 0
+            for path in selected_paths:
+                shutil.copyfile(path, sorted_frames / f"{output_index:>06d}.ppm")
+                output_index += 1
+            for path in reversed(selected_paths[1:-1]):
+                path.replace(sorted_frames / f"{output_index:>06d}.ppm")
+                output_index += 1
         else:
-            selected_paths = paths
-        output_index = 0
-        for path in selected_paths:
-            shutil.copyfile(path, sorted_frames / f"{output_index:>06d}.ppm")
-            output_index += 1
-        for path in reversed(selected_paths[1:-1]):
-            path.replace(sorted_frames / f"{output_index:>06d}.ppm")
-            output_index += 1
+            output_index = 0
+            for path in sorted(base_frames.iterdir()):
+                if path.suffix == ".ppm":
+                    shutil.copyfile(path, sorted_frames / f"{output_index:>06d}.ppm")
+                    output_index += 1
         subprocess.run(
             [
                 parameters["ffmpeg"],

diff --git a/charidotella/version.py b/charidotella/version.py
@@ -1 +1 @@
-__version__ = "3.0.0"
+__version__ = "3.1.0"
diff --git a/configuration-schema.json b/configuration-schema.json
@@ -530,6 +530,9 @@
                             "forward_duration": {
                                 "type": ["string", "integer"]
                             },
+                            "rewind": {
+                                "type": ["boolean"]
+                            },
                             "icon": {
                                 "type": "string"
                             },
@@ -576,6 +579,7 @@
                             "type",
                             "icon",
                             "forward_duration",
+                            "rewind",
                             "tau_to_frametime_ratio",
                             "style",
                             "on_color",