properties tab select between manual and eq ids

waveform_editor/gui/shape_editor/plasma_properties.py

@@ -1,140 +1,278 @@
+from pathlib import Path
+
 import imas
 import panel as pn
 import param
 from panel.viewable import Viewer
 
-from waveform_editor.gui.util import (
-    EquilibriumInput,
-    FormattedEditableFloatSlider,
-    WarningIndicator,
-)
 from waveform_editor.shape_editor.plasma_properties_calc import (
     compute_profiles_from_params,
 )
 
+MANUAL = "Manual"
+EQ_IDS = "Eq IDS"
+PARAMETRIC = "Parametric"
 
-class PlasmaPropertiesParams(Viewer):
-    """Helper class containing parameters defining the plasma properties."""
+_CARD_CSS = (Path(__file__).parent.parent / "styles" / "property_card.css").read_text()
 
-    ip = param.Number(
-        default=-1.5e7, softbounds=[-1.7e7, 0], label="Plasma current [A]"
-    )
-    r0 = param.Number(
-        default=6.2, softbounds=[5, 7], label="Reference major radius [m]"
-    )
-    b0 = param.Number(
-        default=-5.3, softbounds=[-10, 10], label="Toroidal field at R0 [T]"
-    )
-    alpha = param.Number(default=0.5, softbounds=[0.5, 2], step=0.01, label="Alpha")
-    beta = param.Number(default=0.5, softbounds=[0.5, 2], step=0.01, label="Beta")
-    gamma = param.Number(default=1.0, softbounds=[0.5, 2], step=0.01, label="Gamma")
+
+class PropertyInput(Viewer):
+    """A single scalar plasma property with its own Manual / Eq IDS mode toggle."""
+
+    mode = param.ObjectSelector(default=MANUAL, objects=[MANUAL, EQ_IDS])
+    value = param.Number(default=0.0)
+    ids_uri = param.String(default="")
+    ids_time = param.Number(default=0.0)
+    changed = param.Event()
+
+    def __init__(self, label, default_value, step=0.01, **params):
+        super().__init__(**params)
+        self.value = default_value
+        self._label = label
+
+        self._mode_toggle = pn.widgets.RadioButtonGroup(
+            options=[MANUAL, EQ_IDS],
+            value=self.mode,
+            button_style="outline",
+            button_type="primary",
+            stylesheets=[_CARD_CSS],
+        )
+        self._mode_toggle.param.watch(self._on_mode_change, "value")
+
+        self._value_input = pn.widgets.FloatInput(
+            value=self.value,
+            step=step,
+            sizing_mode="stretch_width",
+            margin=(4, 0, 0, 0),
+        )
+        self._value_input.param.watch(self._on_value_change, "value")
+
+        self._uri_input = pn.widgets.TextInput.from_param(
+            self.param.ids_uri,
+            name="",
+            placeholder="IDS URI",
+            sizing_mode="stretch_width",
+        )
+        self._time_input = pn.widgets.FloatInput.from_param(
+            self.param.ids_time, name="Time [s]", width=100
+        )
+        self.param.watch(
+            lambda *_: self.param.trigger("changed"), ["ids_uri", "ids_time"]
+        )
+
+    def _on_mode_change(self, event):
+        self.mode = event.new
+        self.param.trigger("changed")
+
+    def _on_value_change(self, event):
+        self.value = event.new
+        self.param.trigger("changed")
 
     def __panel__(self):
-        widgets = {}
-        for name in self.param:
-            if isinstance(self.param[name], param.Number):
-                widgets[name] = FormattedEditableFloatSlider
-        return pn.Param(self.param, widgets=widgets, show_name=False)
+        is_manual = pn.bind(lambda m: m == MANUAL, self.param.mode)
+        is_ids = pn.bind(lambda m: m == EQ_IDS, self.param.mode)
+        header = pn.Row(
+            pn.pane.HTML(self._label, sizing_mode="stretch_width", margin=(5, 0)),
+            self._mode_toggle,
+            sizing_mode="stretch_width",
+            margin=0,
+        )
+        return pn.Column(
+            header,
+            pn.Column(self._value_input, visible=is_manual, margin=(4, 0, 0, 0)),
+            pn.Column(
+                self._uri_input, self._time_input, visible=is_ids, margin=(4, 0, 0, 0)
+            ),
+            css_classes=["property-card"],
+            stylesheets=[_CARD_CSS],
+            max_width=600,
+            margin=(0, 0, 8, 0),
+        )
 
 
-class PlasmaProperties(Viewer):
-    MANUAL_INPUT = "Manual"
-    EQUILIBRIUM_INPUT = "Equilibrium IDS"
-    input_mode = param.ObjectSelector(
-        default=EQUILIBRIUM_INPUT,
-        objects=[EQUILIBRIUM_INPUT, MANUAL_INPUT],
-        label="Plasma properties input mode",
-    )
+class PlasmaProfiles(Viewer):
+    """Widget for selecting the plasma profile source: parametric or from IDS.
 
-    input = param.ClassSelector(class_=EquilibriumInput, default=EquilibriumInput())
-    properties_params = param.ClassSelector(
-        class_=PlasmaPropertiesParams, default=PlasmaPropertiesParams()
-    )
+    In Parametric mode, dpressure_dpsi and f_df_dpsi are computed analytically
+    from alpha/beta/gamma shape parameters. In Eq IDS mode, they are read
+    directly from an equilibrium IDS file.
+    """
+
+    mode = param.ObjectSelector(default=PARAMETRIC, objects=[PARAMETRIC, EQ_IDS])
+    alpha = param.Number(default=0.5, step=0.01)
+    beta = param.Number(default=0.5, step=0.01)
+    gamma = param.Number(default=1.0, step=0.01)
+    ids_uri = param.String(default="")
+    ids_time = param.Number(default=0.0)
+    changed = param.Event()
+
+    def __init__(self, **params):
+        super().__init__(**params)
+
+        self._mode_toggle = pn.widgets.RadioButtonGroup(
+            options=[PARAMETRIC, EQ_IDS],
+            value=self.mode,
+            button_style="outline",
+            button_type="primary",
+            stylesheets=[_CARD_CSS],
+        )
+        self._mode_toggle.param.watch(self._on_mode_change, "value")
+
+        self._alpha_input = pn.widgets.FloatInput.from_param(
+            self.param.alpha, name="Alpha", sizing_mode="stretch_width"
+        )
+        self._beta_input = pn.widgets.FloatInput.from_param(
+            self.param.beta, name="Beta", sizing_mode="stretch_width"
+        )
+        self._gamma_input = pn.widgets.FloatInput.from_param(
+            self.param.gamma, name="Gamma", sizing_mode="stretch_width"
+        )
+        self.param.watch(
+            lambda *_: self.param.trigger("changed"), ["alpha", "beta", "gamma"]
+        )
+
+        self._uri_input = pn.widgets.TextInput.from_param(
+            self.param.ids_uri,
+            name="",
+            placeholder="IDS URI",
+            sizing_mode="stretch_width",
+        )
+        self._time_input = pn.widgets.FloatInput.from_param(
+            self.param.ids_time, name="Time [s]", width=100
+        )
+        self.param.watch(
+            lambda *_: self.param.trigger("changed"), ["ids_uri", "ids_time"]
+        )
+
+    def _on_mode_change(self, event):
+        self.mode = event.new
+        self.param.trigger("changed")
+
+    def __panel__(self):
+        is_parametric = pn.bind(lambda m: m == PARAMETRIC, self.param.mode)
+        is_ids = pn.bind(lambda m: m == EQ_IDS, self.param.mode)
+        header = pn.Row(
+            pn.pane.HTML("Profile source", sizing_mode="stretch_width", margin=(5, 0)),
+            self._mode_toggle,
+            sizing_mode="stretch_width",
+            margin=0,
+        )
+        return pn.Column(
+            header,
+            pn.Column(
+                self._alpha_input,
+                self._beta_input,
+                self._gamma_input,
+                visible=is_parametric,
+                margin=(4, 0, 0, 0),
+            ),
+            pn.Column(
+                self._uri_input,
+                self._time_input,
+                visible=is_ids,
+                margin=(4, 0, 0, 0),
+            ),
+            css_classes=["property-card"],
+            stylesheets=[_CARD_CSS],
+            max_width=600,
+            margin=(0, 0, 8, 0),
+        )
 
+
+class PlasmaProperties(Viewer):
     profile_updated = param.Event(
         doc="Triggered whenever the dpressure_dpsi and f_df_dpsi are updated."
     )
     has_properties = param.Boolean(doc="Whether the plasma properties are loaded.")
 
     def __init__(self):
         super().__init__()
-        self.indicator = WarningIndicator(visible=self.param.has_properties.rx.not_())
-        self.radio_box = pn.widgets.RadioBoxGroup.from_param(
-            self.param.input_mode, inline=True, margin=(15, 20, 0, 20)
-        )
-        self.panel = pn.Column(self.radio_box, self._panel_property_options)
+        self._ip = PropertyInput("Plasma current [A]", default_value=-1.5e7)
+        self._r0 = PropertyInput("Reference major radius [m]", default_value=6.2)
+        self._b0 = PropertyInput("Toroidal magnetic field [T]", default_value=-5.3)
+        self._profiles = PlasmaProfiles()
+
         self.dpressure_dpsi = None
         self.f_df_dpsi = None
         self.psi_norm = None
         self.ip = None
         self.r0 = None
         self.b0 = None
 
-    @param.depends(
-        "properties_params.param", "input.param", "input_mode", watch=True, on_init=True
-    )
+        for widget in [self._ip, self._r0, self._b0, self._profiles]:
+            widget.param.watch(lambda *_: self._load_plasma_properties(), "changed")
+
+        self._load_plasma_properties()
+
+    def _load_scalar(self, prop: PropertyInput, extractor):
+        """Return a scalar value from manual input or extracted from an IDS."""
+        if prop.mode == MANUAL:
+            return prop.value
+        if not prop.ids_uri:
+            return None
+        try:
+            with imas.DBEntry(prop.ids_uri, "r") as entry:
+                eq = entry.get_slice(
+                    "equilibrium", prop.ids_time, imas.ids_defs.CLOSEST_INTERP
+                )
+            return extractor(eq)
+        except Exception as e:
+            pn.state.notifications.error(f"Could not load from {prop.ids_uri}: {e}")
+            return None
+
     def _load_plasma_properties(self):
-        """Update plasma properties based on input mode."""
+        self.ip = self._load_scalar(
+            self._ip, lambda eq: eq.time_slice[0].global_quantities.ip
+        )
+        self.r0 = self._load_scalar(self._r0, lambda eq: eq.vacuum_toroidal_field.r0)
+        self.b0 = self._load_scalar(self._b0, lambda eq: eq.vacuum_toroidal_field.b0[0])
 
-        if self.input_mode == self.EQUILIBRIUM_INPUT:
-            self._load_properties_from_ids()
-        elif self.input_mode == self.MANUAL_INPUT:
-            self._load_properties_from_params()
+        if self._profiles.mode == EQ_IDS:
+            self._load_profiles_from_ids(
+                self._profiles.ids_uri, self._profiles.ids_time
+            )
+        elif self.r0 is not None:
+            try:
+                self.psi_norm, self.dpressure_dpsi, self.f_df_dpsi = (
+                    compute_profiles_from_params(
+                        r0=self.r0,
+                        alpha=self._profiles.alpha,
+                        beta=self._profiles.beta,
+                        gamma=self._profiles.gamma,
+                    )
+                )
+            except Exception as e:
+                pn.state.notifications.error(f"Could not compute profiles: {e}")
+                self.dpressure_dpsi = self.f_df_dpsi = self.psi_norm = None
+        else:
+            self.dpressure_dpsi = self.f_df_dpsi = self.psi_norm = None
 
+        self.has_properties = all(
+            v is not None for v in [self.ip, self.r0, self.b0, self.dpressure_dpsi]
+        )
         self.param.trigger("profile_updated")
 
-    def _load_properties_from_params(self):
-        """Load the plasma properties from the properties parameters. Calculate
-        dpressure_dpsi and f_df_dpsi from the parametric alpha, beta, and gamma
-        parameters."""
-        self.ip = self.properties_params.ip
-        self.r0 = self.properties_params.r0
-        self.b0 = self.properties_params.b0
-        (
-            self.psi_norm,
-            self.dpressure_dpsi,
-            self.f_df_dpsi,
-        ) = compute_profiles_from_params(
-            r0=self.properties_params.r0,
-            alpha=self.properties_params.alpha,
-            beta=self.properties_params.beta,
-            gamma=self.properties_params.gamma,
-        )
-        self.has_properties = True
-
-    def _load_properties_from_ids(self):
-        """Load plasma properties from IDS equilibrium input."""
-        if not self.input.uri:
-            self.has_properties = False
+    def _load_profiles_from_ids(self, uri: str, time: float):
+        if not uri:
+            self.dpressure_dpsi = self.f_df_dpsi = self.psi_norm = None
             return
         try:
-            with imas.DBEntry(self.input.uri, "r") as entry:
-                equilibrium = entry.get_slice(
-                    "equilibrium", self.input.time, imas.ids_defs.CLOSEST_INTERP
-                )
-            self.ip = equilibrium.time_slice[0].global_quantities.ip
-            self.r0 = equilibrium.vacuum_toroidal_field.r0
-            self.b0 = equilibrium.vacuum_toroidal_field.b0[0]
-
-            self.dpressure_dpsi = equilibrium.time_slice[0].profiles_1d.dpressure_dpsi
-            self.f_df_dpsi = equilibrium.time_slice[0].profiles_1d.f_df_dpsi
-            psi = equilibrium.time_slice[0].profiles_1d.psi
+            with imas.DBEntry(uri, "r") as entry:
+                eq = entry.get_slice("equilibrium", time, imas.ids_defs.CLOSEST_INTERP)
+            self.dpressure_dpsi = eq.time_slice[0].profiles_1d.dpressure_dpsi
+            self.f_df_dpsi = eq.time_slice[0].profiles_1d.f_df_dpsi
+            psi = eq.time_slice[0].profiles_1d.psi
             self.psi_norm = (psi - psi[0]) / (psi[-1] - psi[0])
-
-            self.has_properties = True
         except Exception as e:
-            pn.state.notifications.error(
-                f"Could not load plasma property outline from {self.input.uri}:"
-                f" {str(e)}"
-            )
-            self.has_properties = False
-
-    @param.depends("input_mode")
-    def _panel_property_options(self):
-        if self.input_mode == self.MANUAL_INPUT:
-            return self.properties_params
-        elif self.input_mode == self.EQUILIBRIUM_INPUT:
-            return pn.Row(pn.Param(self.input, show_name=False), self.indicator)
+            pn.state.notifications.error(f"Could not load profiles from {uri}: {e}")
+            self.dpressure_dpsi = self.f_df_dpsi = self.psi_norm = None
 
     def __panel__(self):
-        return self.panel
+        return pn.Column(
+            self._ip,
+            self._r0,
+            self._b0,
+            self._profiles,
+            sizing_mode="stretch_width",
+            margin=(20, 20),
+        )

waveform_editor/gui/styles/property_card.css

@@ -0,0 +1,10 @@
+:host(.property-card) {
+    border: 1px solid #dee2e6;
+    border-radius: 8px;
+    padding: 8px 12px 12px 12px;
+    display: block;
+}
+
+.bk-btn.bk-btn-primary.bk-active {
+    color: white !important;
+}