TEST: added new test for the scan_removal_tool. Also added an option for launching the projection viewer where the code does not advance until the window is closed

Author: hruth

src/pyxalign/data_structures/projections.py

@@ -688,8 +688,14 @@ def save_projections_object(
             h5_obj.close()
         print(f"projections saved to {h5_obj.file.filename}{h5_obj.name}")
 
-    def launch_viewer(self, options: Optional[ProjectionViewerOptions] = None):
-        self.gui = launch_projection_viewer(self, options, enable_dropping=True)
+    def launch_viewer(
+        self,
+        options: Optional[ProjectionViewerOptions] = None,
+        wait_until_closed: Optional[bool] = False,
+    ):
+        self.gui = launch_projection_viewer(
+            self, options, enable_dropping=True, wait_until_closed=wait_until_closed
+        )
 
 
 class ComplexProjections(Projections):

src/pyxalign/plotting/interactive/launchers.py

@@ -20,10 +20,13 @@ def launch_projection_viewer(
     projections: "p.Projections",
     options: Optional[ProjectionViewerOptions] = None,
     enable_dropping: bool = True,
+    wait_until_closed: bool = False,
 ) -> ProjectionViewer:
     app = QApplication.instance() or QApplication([])
     gui = ProjectionViewer(projections, options, enable_dropping=enable_dropping)
     gui.show()
+    if wait_until_closed:
+        app.exec_()
     return gui
 
 

tests/gui_tests/scan_removal_tool.py

@@ -0,0 +1,12 @@
+from pyxalign import options as opts
+from pyxalign.test_utils_2 import CITestHelper
+
+# Setup the test
+# ci_options = opts.CITestOptions(test_data_name="cSAXS_e18044_LamNI_201907")
+ci_options = opts.CITestOptions(test_data_name="TP2")
+ci_test_helper = CITestHelper(options=ci_options)
+# load data with phase unwrapped projections
+task = ci_test_helper.load_checkpoint_task(file_name="pre_pma_task.h5")
+# launch viewer
+task.phase_projections.launch_viewer(wait_until_closed=True)
+