update sync videos to skip videos with existing alignment

Author: darsnack

packages/cheese3d/cheese3d/config.py

@@ -318,8 +318,7 @@ def build_regex(regex):
             if "view" not in regex:
                 raise RuntimeError("Regex must contain 'view' key.")
             for key, rstr in regex.items():
-                full_regex = full_regex.replace("{{" + key + "}}", # type: ignore
-                                                fr"(?P<{key}>{rstr})")
+                full_regex = full_regex.replace("{{" + key + "}}", fr"(?P<{key}>{rstr})")
             missing_groups = re.match("({{.*}})", full_regex)
             if missing_groups:
                 raise RuntimeError(f"Regex is missing definitions for groups: {missing_groups.groups}")

packages/cheese3d/cheese3d/synchronize/core.py

@@ -242,12 +242,14 @@ def synchronize_videos(pipeline_cfg: SyncConfig,
                                         threshold=pipeline_cfg.led_threshold,
                                         crop=crop)
         pipeline = SyncPipeline.from_cfg(pipeline_cfg, ref_reader, target_reader)
-        align_param = pipeline.align_recording()
-        pipeline.write_json(align_param)
-        frame_lag = int(round(maybe(align_param.lag, 0), 2) * align_param.sample_rate)
-        align_params[view] = (video,
-                              frame_lag,
-                              align_param.sample_rate)
+        target_path = pipeline.target.root_path()
+        if target_path.with_suffix(".align.json").exists():
+            print(f"alignment file exists, skipping {str(target_path)}...")
+        else:
+            align_param = pipeline.align_recording()
+            pipeline.write_json(align_param)
+            frame_lag = int(round(maybe(align_param.lag, 0), 2) * align_param.sample_rate)
+            align_params[view] = (video, frame_lag, align_param.sample_rate)
     ref_lag = min(-lag for _, lag, _ in align_params.values())
     for video, lag, view_fps in align_params.values():
         shift = -lag - ref_lag

packages/cheese3d/cheese3d/synchronize/readers.py

@@ -36,7 +36,7 @@ def load_signal(self):
                                   "Instead, inherit from this class to implement "
                                   "a specific sync signal source.")
 
-    def root_path(self):
+    def root_path(self) -> Path:
         raise NotImplementedError("Attempt to use SyncSignalReader directly. "
                                   "Instead, inherit from this class to implement "
                                   "a specific sync signal source.")
@@ -171,7 +171,7 @@ def load_signal(self):
         led_df = pd.read_csv(self.source, sep="\t", names=["timestamp", "signal"])
         analog_signal = led_df["signal"].values
         threshold = maybe(self.threshold, 0.1)
-        analog_signal = np.where(analog_signal > threshold, 1, 0) # type: ignore
+        analog_signal = np.where(analog_signal > threshold, 1, 0)
 
         return analog_signal