updated pyptv_gui to work with mask_centers.tiffs

Author: alexlib

pyptv/pyptv_gui.py

@@ -478,7 +478,55 @@ def exit_action(self, info):
     def saveas_action(self, info):
         print("not implemented")
 
+    def init_action(self, info):
+        """init_action - clears existing plots from the camera windows,
+        initializes C image arrays with mainGui.orig_image and
+        calls appropriate start_proc_c
+         by using ptv.py_start_proc_c()
+        """
+        mainGui = info.object
+        # synchronize the active run params dir with the temp params dir
+        mainGui.exp1.syncActiveDir()
 
+        for i in range(len(mainGui.camera_list)):
+            try:
+                im = imread(
+                    getattr(
+                        mainGui.exp1.active_params.m_params,
+                        f"Name_{i+1}_Image",
+                    )
+                )
+                if im.ndim > 2:
+                    im = rgb2gray(im)
+
+                mainGui.orig_image[i] = img_as_ubyte(im)
+            except IOError:
+                print("Error reading image, setting zero image")
+                h_img = mainGui.exp1.active_params.m_params.imx
+                v_img = mainGui.exp1.active_params.m_params.imy
+                temp_img = img_as_ubyte(np.zeros((v_img, h_img)))
+                # print(f"setting images of size {temp_img.shape}")
+                exec(f"mainGui.orig_image[{i}] = temp_img")
+
+            if hasattr(mainGui.camera_list[i], "img_plot"):
+                del mainGui.camera_list[i].img_plot
+        mainGui.clear_plots()
+        print("\n Init action \n")
+        # mainGui.update_plots(mainGui.orig_image, is_float=False)
+        mainGui.create_plots(mainGui.orig_image, is_float=False)
+        # mainGui.set_images(mainGui.orig_image)
+
+        (
+            info.object.cpar,
+            info.object.spar,
+            info.object.vpar,
+            info.object.track_par,
+            info.object.tpar,
+            info.object.cals,
+            info.object.epar,
+        ) = ptv.py_start_proc_c(info.object.n_cams)
+        mainGui.pass_init = True
+        print("Read all the parameters and calibrations successfully ")
 
     def draw_mask_action(self, info):
         """ drawing masks GUI """
@@ -608,55 +656,7 @@ def corresp_action(self, info):
         # info.object.drawcross("pair_x", "pair_y", x, y, "yellow", 3)
         # info.object.drawcross("unused_x","unused_y",unused[:,0],unused[:,1],"blue",3)
 
-    def init_action(self, info):
-        """init_action - clears existing plots from the camera windows,
-        initializes C image arrays with mainGui.orig_image and
-        calls appropriate start_proc_c
-         by using ptv.py_start_proc_c()
-        """
-        mainGui = info.object
-        # synchronize the active run params dir with the temp params dir
-        mainGui.exp1.syncActiveDir()
-
-        for i in range(len(mainGui.camera_list)):
-            try:
-                im = imread(
-                    getattr(
-                        mainGui.exp1.active_params.m_params,
-                        f"Name_{i+1}_Image",
-                    )
-                )
-                if im.ndim > 2:
-                    im = rgb2gray(im)
-
-                mainGui.orig_image[i] = img_as_ubyte(im)
-            except IOError:
-                print("Error reading image, setting zero image")
-                h_img = mainGui.exp1.active_params.m_params.imx
-                v_img = mainGui.exp1.active_params.m_params.imy
-                temp_img = img_as_ubyte(np.zeros((v_img, h_img)))
-                # print(f"setting images of size {temp_img.shape}")
-                exec(f"mainGui.orig_image[{i}] = temp_img")
-
-            if hasattr(mainGui.camera_list[i], "img_plot"):
-                del mainGui.camera_list[i].img_plot
-        mainGui.clear_plots()
-        print("\n Init action \n")
-        # mainGui.update_plots(mainGui.orig_image, is_float=False)
-        mainGui.create_plots(mainGui.orig_image, is_float=False)
-        # mainGui.set_images(mainGui.orig_image)
 
-        (
-            info.object.cpar,
-            info.object.spar,
-            info.object.vpar,
-            info.object.track_par,
-            info.object.tpar,
-            info.object.cals,
-            info.object.epar,
-        ) = ptv.py_start_proc_c(info.object.n_cams)
-        mainGui.pass_init = True
-        print("Read all the parameters and calibrations successfully ")
 
     def calib_action(self, info):
         """calib_action - initializes calib class with appropriate number of
@@ -1466,6 +1466,8 @@ def overlay_set_images(self, seq_first: int, seq_last:int):
                 temp_img = []
                 for seq in range(seq_first, seq_last):
                     _ = imread(self.base_name[cam_id] % seq)
+                    if _.ndim > 2:
+                        _ = rgb2gray(_)
                     temp_img.append(img_as_ubyte(_))
 
                 temp_img = np.array(temp_img)
@@ -1489,7 +1491,11 @@ def load_disp_image(self, img_name: str, j: int, display_only: bool = False):
         """
         # print(f"Setting image: {img_name}")
         try:
-            temp_img = img_as_ubyte(imread(img_name))
+            temp_img = imread(img_name)
+            if temp_img.ndim > 2:
+                temp_img = rgb2gray(temp_img)
+
+            temp_img = img_as_ubyte(temp_img)
         except IOError:
             print("Error reading file, setting zero image")
             h_img = self.exp1.active_params.m_params.imx