minor refactor

Author: SBlokhuizen

waveform_editor/settings.py

@@ -63,6 +63,7 @@ def panel(self, nice_mode):
             if p == "name":
                 continue
 
+            # Add warning indicator if required parameter is not filled
             is_inv_required = p == "inv_executable" and nice_mode == self.INVERSE_MODE
             is_dir_required = p == "dir_executable" and nice_mode == self.DIRECT_MODE
             is_base_required = p in self.BASE_REQUIRED

waveform_editor/shape_editor/coil_currents.py

@@ -12,7 +12,7 @@ class CoilCurrents(Viewer):
     coil_ui = param.List(
         doc="List of tuples containing the checkboxes and sliders for the coil currents"
     )
-    nice_mode = param.Parameter(allow_refs=True)
+    nice_mode = param.Selector(allow_refs=True)
 
     def __init__(self, **params):
         super().__init__(**params)

waveform_editor/shape_editor/nice_plotter.py

@@ -26,7 +26,7 @@ class NicePlotter(Viewer):
     pf_active = param.ClassSelector(class_=IDSToplevel, precedence=-1)
     plasma_shape = param.ClassSelector(class_=PlasmaShape, precedence=-1)
     plasma_properties = param.ClassSelector(class_=PlasmaProperties, precedence=-1)
-    nice_mode = param.Parameter(precedence=-1, allow_refs=True)
+    nice_mode = param.Selector(precedence=-1, allow_refs=True)
 
     # Plot parameters
     show_contour = param.Boolean(default=True, label="Show contour lines")
@@ -88,6 +88,15 @@ def __init__(self, **params):
         self.profiles_pane = pn.pane.HoloViews(
             profiles_plot, width=self.PROFILE_WIDTH, height=self.PROFILE_HEIGHT
         )
+        self.panel_layout = pn.Param(
+            self.param,
+            show_name=False,
+            widgets={
+                "show_desired_shape": {
+                    "visible": self.param.nice_mode.rx() == NiceSettings.INVERSE_MODE
+                }
+            },
+        )
 
     @pn.depends("plasma_properties.profile_updated")
     def _plot_profiles(self):
@@ -392,12 +401,4 @@ def _plot_xo_points(self):
         return o_scatter * x_scatter
 
     def __panel__(self):
-        return pn.Param(
-            self.param,
-            show_name=False,
-            widgets={
-                "show_desired_shape": {
-                    "visible": self.param.nice_mode.rx() == NiceSettings.INVERSE_MODE
-                }
-            },
-        )
+        return self.panel_layout