copy measurement

Author: JoschD

omc3_gui/segment_by_segment/main_controller.py

@@ -81,6 +81,7 @@ def connect_signals(self):
         view.button_load_measurement.clicked.connect(self.open_measurements)
         view.button_edit_measurement.clicked.connect(self.edit_measurement)
         view.button_remove_measurement.clicked.connect(self.remove_measurement)
+        view.button_copy_measurement.clicked.connect(self.copy_measurement)
 
         view.button_run_matcher.clicked.connect(self.run_matcher)
         view.button_edit_corrections.clicked.connect(self.edit_corrections)
@@ -158,6 +159,7 @@ def set_measurement_interaction_buttons_enabled(self, enabled: bool = True):
             view.button_edit_measurement,
             view.button_run_matcher,
             view.button_edit_corrections,
+            view.button_copy_measurement,
         )
         for button in measurement_interaction_buttons:
             button.setEnabled(enabled)
@@ -261,6 +263,44 @@ def edit_measurement(self, measurement: OpticsMeasurement | None = None):
         if dialog.exec_() == dialog.Accepted:
             LOGGER.debug("Edit dialog closed. Updating measurement.")
 
+    @Slot()
+    def copy_measurement(self, measurement: OpticsMeasurement | None = None):
+        """ Create a copy of the given measurement and add it to the GUI. 
+        If no measurement is given, the currently selected measurement is copied.
+
+        Args:
+            measurement (OpticsMeasurement | None, optional): The measurement to copy. Defaults to None.
+        """
+        if measurement is None:
+            try:
+                measurement = self.get_single_measurement()
+            except ValueError as e:
+                LOGGER.warning(str(e))
+                return
+
+        if measurement.output_dir is None:  # should not be possible, but better to catch
+            LOGGER.error("Cannot copy measurement without output directory.")
+            return
+
+        LOGGER.debug(f"Copying {measurement.display()}.")
+        new_measurement = measurement.copy()
+        
+        # might already have a counter from previous copy
+        name =  re.sub(r"_\d+$", "", measurement.output_dir.name)  
+
+        for count in range(1, 1000):  # limit to avoid infinite loop
+            new_measurement.output_dir = measurement.output_dir.with_name(f"{name}_{count:d}")
+            try:
+                self.add_measurement(new_measurement)
+            except ValueError:
+                continue
+            break
+        else:
+            LOGGER.error(
+                "Could not copy measurement. Counter limit exeeded. Not sure what went wrong."
+            )
+            return
+
     @Slot()
     def remove_measurement(self, measurements: Sequence[OpticsMeasurement] | None = None):
         """ Remove measurements from the GUI.
@@ -298,6 +338,10 @@ def measurement_selection_changed(self, measurements: Sequence[OpticsMeasurement
             return
 
         self.set_measurement_interaction_buttons_enabled(True)
+        if len(measurements) > 1:
+            view.button_edit_measurement.setEnabled(False)
+            view.button_copy_measurement.setEnabled(False)
+
         self.set_all_segment_buttons_enabled(True)
 
         # Group the segments for the measurements into table-items when they have the same defintion ---

omc3_gui/segment_by_segment/main_view.py

@@ -72,6 +72,7 @@ def __init__(self, parent=None):
         self.button_load_measurement: QtWidgets.QPushButton = None
         self.button_remove_measurement: QtWidgets.QPushButton = None
         self.button_edit_measurement: QtWidgets.QPushButton = None
+        self.button_copy_measurement: QtWidgets.QPushButton = None
         self.button_edit_corrections: QtWidgets.QPushButton = None
         self.button_run_matcher: QtWidgets.QPushButton = None
 
@@ -237,9 +238,14 @@ def build_measurement_buttons():
                     grid_buttons_filler.add(remove)
                     self.button_remove_measurement = remove
 
+                    copy = DefaultButton("Copy")
+                    copy.setToolTip("Create a virtual copy of the measurement, with a different output dir.")
+                    grid_buttons_filler.add(copy)
+                    self.button_copy_measurement = copy
+
                     matcher = RunButton("Run Matcher")
                     matcher.setToolTip("Run the Segment-by-Segment Matcher.")
-                    grid_buttons_filler.add(matcher, col_span=2)
+                    grid_buttons_filler.add(matcher)
                     self.button_run_matcher = matcher
 
                     edit_corrections = ChangeButton("Corrections")

omc3_gui/segment_by_segment/measurement_model.py

@@ -8,7 +8,7 @@
 from __future__ import annotations
 
 import logging
-from dataclasses import dataclass, field, fields
+from dataclasses import dataclass, field, fields, asdict
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, ClassVar
 
@@ -175,6 +175,20 @@ def get_sbs_parameters(self) -> dict[str, Any]:
         return parameters
 
     # Builder ------------------------------------------------------------------
+    def copy(self) -> OpticsMeasurement:
+        """ Creates a copy of the measurement. """
+        new_measurement = OpticsMeasurement(
+            **{
+                f.name: getattr(self, f.name) for f in fields(self) 
+                if not f.name.startswith("_")
+            },
+        )
+        for segment in self.segments:
+            new_segment = segment.copy()
+            new_segment.measurement = new_measurement
+            new_measurement.add_segment(new_segment)
+        return new_measurement
+
     @classmethod
     def from_path(cls, path: Path) -> OpticsMeasurement:
         """ Creates an OpticsMeasurement from a folder, by trying 
@@ -300,4 +314,4 @@ def map_lhc_year(year: str | None) -> str:
         return "2018"
 
     return year
-    
+