bec-widgets 0.87.1__py3-none-any.whl → 0.88.0__py3-none-any.whl

bec_widgets/widgets/waveform/waveform_toolbar/curve_dialog/curve_dialog.py

@@ -0,0 +1,341 @@
+from __future__ import annotations
+
+import os
+from typing import Literal
+
+from pydantic import BaseModel
+from qtpy.QtCore import QObject, QSize, Slot
+from qtpy.QtGui import QIcon
+from qtpy.QtWidgets import QComboBox, QLineEdit, QPushButton, QSpinBox, QTableWidget, QVBoxLayout
+
+import bec_widgets
+from bec_widgets.qt_utils.error_popups import WarningPopupUtility
+from bec_widgets.qt_utils.settings_dialog import SettingWidget
+from bec_widgets.utils import Colors, UILoader
+from bec_widgets.widgets.color_button.color_button import ColorButton
+from bec_widgets.widgets.device_line_edit.device_line_edit import DeviceLineEdit
+
+MODULE_PATH = os.path.dirname(bec_widgets.__file__)
+
+
+class CurveSettings(SettingWidget):
+    def __init__(self, parent=None, *args, **kwargs):
+        super().__init__(parent, *args, **kwargs)
+        current_path = os.path.dirname(__file__)
+
+        self.ui = UILoader(self).loader(os.path.join(current_path, "curve_dialog.ui"))
+        self._setup_icons()
+
+        self.warning_util = WarningPopupUtility(self)
+
+        self.layout = QVBoxLayout(self)
+        self.layout.addWidget(self.ui)
+
+        self.ui.add_curve.clicked.connect(self.add_curve)
+        self.ui.add_dap.clicked.connect(self.add_dap)
+        self.ui.x_mode.currentIndexChanged.connect(self.set_x_mode)
+        self.ui.normalize_colors_scan.clicked.connect(lambda: self.change_colormap("scan"))
+        self.ui.normalize_colors_dap.clicked.connect(lambda: self.change_colormap("dap"))
+
+    def _setup_icons(self):
+        add_icon = QIcon()
+        add_icon.addFile(
+            os.path.join(MODULE_PATH, "assets", "toolbar_icons", "add.svg"), size=QSize(20, 20)
+        )
+        self.ui.add_dap.setIcon(add_icon)
+        self.ui.add_dap.setToolTip("Add DAP Curve")
+        self.ui.add_curve.setIcon(add_icon)
+        self.ui.add_curve.setToolTip("Add Scan Curve")
+
+    @Slot(dict)
+    def display_current_settings(self, config: dict | BaseModel):
+
+        # What elements should be enabled
+        x_name = self.target_widget.waveform._x_axis_mode["name"]
+        x_entry = self.target_widget.waveform._x_axis_mode["entry"]
+        self._enable_ui_elements(x_name, x_entry)
+        cm = self.target_widget.config.color_palette
+        self.ui.color_map_selector_scan.combo.setCurrentText(cm)
+
+        # Scan Curve Table
+        for label, curve in config["scan_segment"].items():
+            row_count = self.ui.scan_table.rowCount()
+            self.ui.scan_table.insertRow(row_count)
+            DialogRow(
+                parent=self,
+                table_widget=self.ui.scan_table,
+                client=self.target_widget.client,
+                row=row_count,
+                config=curve.config,
+            ).add_scan_row()
+
+        # Add DAP Curves
+        for label, curve in config["DAP"].items():
+            row_count = self.ui.dap_table.rowCount()
+            self.ui.dap_table.insertRow(row_count)
+            DialogRow(
+                parent=self,
+                table_widget=self.ui.dap_table,
+                client=self.target_widget.client,
+                row=row_count,
+                config=curve.config,
+            ).add_dap_row()
+
+    def _enable_ui_elements(self, name, entry):
+        if name is None:
+            name = "best_effort"
+        if name in ["index", "timestamp", "best_effort"]:
+            self.ui.x_mode.setCurrentText(name)
+            self.set_x_mode()
+        else:
+            self.ui.x_mode.setCurrentText("device")
+            self.set_x_mode()
+            self.ui.x_name.setText(name)
+            self.ui.x_entry.setText(entry)
+
+    @Slot()
+    def set_x_mode(self):
+        x_mode = self.ui.x_mode.currentText()
+        if x_mode in ["index", "timestamp", "best_effort"]:
+            self.ui.x_name.setEnabled(False)
+            self.ui.x_entry.setEnabled(False)
+            self.ui.dap_table.setEnabled(False)
+            self.ui.add_dap.setEnabled(False)
+            if self.ui.dap_table.rowCount() > 0:
+                self.warning_util.show_warning(
+                    title="DAP Warning",
+                    message="DAP is not supported without specific x-axis device. All current DAP curves will be removed.",
+                    detailed_text=f"Affected curves: {[self.ui.dap_table.cellWidget(row, 0).text() for row in range(self.ui.dap_table.rowCount())]}",
+                )
+        else:
+            self.ui.x_name.setEnabled(True)
+            self.ui.x_entry.setEnabled(True)
+            self.ui.dap_table.setEnabled(True)
+            self.ui.add_dap.setEnabled(True)
+
+    @Slot()
+    def change_colormap(self, target: Literal["scan", "dap"]):
+        if target == "scan":
+            cm = self.ui.color_map_selector_scan.combo.currentText()
+            table = self.ui.scan_table
+        if target == "dap":
+            cm = self.ui.color_map_selector_dap.combo.currentText()
+            table = self.ui.dap_table
+        rows = table.rowCount()
+        colors = Colors.golden_angle_color(colormap=cm, num=rows + 1, format="HEX")
+        color_button_col = 2 if target == "scan" else 3
+        for row, color in zip(range(rows), colors):
+            table.cellWidget(row, color_button_col).setColor(color)
+
+    @Slot()
+    def accept_changes(self):
+        self.accept_curve_changes()
+
+    def accept_curve_changes(self):
+        old_curves_scans = list(self.target_widget.waveform._curves_data["scan_segment"].values())
+        old_curves_dap = list(self.target_widget.waveform._curves_data["DAP"].values())
+        for curve in old_curves_scans:
+            curve.remove()
+        for curve in old_curves_dap:
+            curve.remove()
+        self.get_curve_params()
+
+    def get_curve_params(self):
+        x_mode = self.ui.x_mode.currentText()
+
+        if x_mode in ["index", "timestamp", "best_effort"]:
+            x_name = x_mode
+            x_entry = x_mode
+        else:
+            x_name = self.ui.x_name.text()
+            x_entry = self.ui.x_entry.text()
+
+        self.target_widget.set_x(x_name=x_name, x_entry=x_entry)
+
+        for row in range(self.ui.scan_table.rowCount()):
+            y_name = self.ui.scan_table.cellWidget(row, 0).text()
+            y_entry = self.ui.scan_table.cellWidget(row, 1).text()
+            color = self.ui.scan_table.cellWidget(row, 2).get_color()
+            style = self.ui.scan_table.cellWidget(row, 3).currentText()
+            width = self.ui.scan_table.cellWidget(row, 4).value()
+            symbol_size = self.ui.scan_table.cellWidget(row, 5).value()
+            self.target_widget.plot(
+                y_name=y_name,
+                y_entry=y_entry,
+                color=color,
+                pen_style=style,
+                pen_width=width,
+                symbol_size=symbol_size,
+            )
+
+        if x_mode not in ["index", "timestamp", "best_effort"]:
+
+            for row in range(self.ui.dap_table.rowCount()):
+                y_name = self.ui.dap_table.cellWidget(row, 0).text()
+                y_entry = self.ui.dap_table.cellWidget(row, 1).text()
+                dap = self.ui.dap_table.cellWidget(row, 2).currentText()
+                color = self.ui.dap_table.cellWidget(row, 3).get_color()
+                style = self.ui.dap_table.cellWidget(row, 4).currentText()
+                width = self.ui.dap_table.cellWidget(row, 5).value()
+                symbol_size = self.ui.dap_table.cellWidget(row, 6).value()
+
+                self.target_widget.add_dap(
+                    x_name=x_name,
+                    x_entry=x_entry,
+                    y_name=y_name,
+                    y_entry=y_entry,
+                    dap=dap,
+                    color=color,
+                    pen_style=style,
+                    pen_width=width,
+                    symbol_size=symbol_size,
+                )
+        self.target_widget.scan_history(-1)
+
+    def add_curve(self):
+        row_count = self.ui.scan_table.rowCount()
+        self.ui.scan_table.insertRow(row_count)
+        DialogRow(
+            parent=self,
+            table_widget=self.ui.scan_table,
+            client=self.target_widget.client,
+            row=row_count,
+            config=None,
+        ).add_scan_row()
+
+    def add_dap(self):
+        row_count = self.ui.dap_table.rowCount()
+        self.ui.dap_table.insertRow(row_count)
+        DialogRow(
+            parent=self,
+            table_widget=self.ui.dap_table,
+            client=self.target_widget.client,
+            row=row_count,
+            config=None,
+        ).add_dap_row()
+
+
+class DialogRow(QObject):
+    def __init__(
+        self,
+        parent=None,
+        table_widget: QTableWidget = None,
+        row: int = None,
+        config: dict = None,
+        client=None,
+    ):
+
+        super().__init__(parent=parent)
+        self.client = client
+
+        self.table_widget = table_widget
+        self.row = row
+        self.config = config
+        self.init_default_widgets()
+
+    def init_default_widgets(self):
+
+        # Remove Button
+        self.remove_button = RemoveButton()
+
+        # Name and Entry
+        self.device_line_edit = DeviceLineEdit()
+        self.entry_line_edit = QLineEdit()
+
+        self.dap_combo = QComboBox()
+        self.populate_dap_combobox()
+
+        # Styling
+        self.color_button = ColorButton()
+        self.style_combo = StyleComboBox()
+        self.width = QSpinBox()
+        self.width.setMinimum(1)
+        self.width.setMaximum(20)
+        self.width.setValue(2)
+
+        self.symbol_size = QSpinBox()
+        self.symbol_size.setMinimum(1)
+        self.symbol_size.setMaximum(20)
+        self.symbol_size.setValue(5)
+
+        self.remove_button.clicked.connect(
+            lambda: self.remove_row()
+        )  # From some reason do not work without lambda
+
+    def populate_dap_combobox(self):
+        available_models = [
+            attr
+            for attr in dir(self.client.dap)
+            if not attr.startswith("__")
+            and not callable(getattr(self.client.dap, attr))
+            and not attr.startswith("_")
+        ]
+        self.dap_combo.addItems(available_models)
+
+    def add_scan_row(self):
+        if self.config is not None:
+            self.device_line_edit.setText(self.config.signals.y.name)
+            self.entry_line_edit.setText(self.config.signals.y.entry)
+            self.color_button.setColor(self.config.color)
+            self.style_combo.setCurrentText(self.config.pen_style)
+            self.width.setValue(self.config.pen_width)
+            self.symbol_size.setValue(self.config.symbol_size)
+        else:
+            default_color = Colors.golden_angle_color(
+                colormap="magma", num=self.row + 1, format="HEX"
+            )[-1]
+            self.color_button.setColor(default_color)
+
+        self.table_widget.setCellWidget(self.row, 0, self.device_line_edit)
+        self.table_widget.setCellWidget(self.row, 1, self.entry_line_edit)
+        self.table_widget.setCellWidget(self.row, 2, self.color_button)
+        self.table_widget.setCellWidget(self.row, 3, self.style_combo)
+        self.table_widget.setCellWidget(self.row, 4, self.width)
+        self.table_widget.setCellWidget(self.row, 5, self.symbol_size)
+        self.table_widget.setCellWidget(self.row, 6, self.remove_button)
+
+    def add_dap_row(self):
+        if self.config is not None:
+            self.device_line_edit.setText(self.config.signals.y.name)
+            self.entry_line_edit.setText(self.config.signals.y.entry)
+            self.dap_combo.setCurrentText(self.config.signals.dap)
+            self.color_button.setColor(self.config.color)
+            self.style_combo.setCurrentText(self.config.pen_style)
+            self.width.setValue(self.config.pen_width)
+            self.symbol_size.setValue(self.config.symbol_size)
+        else:
+            default_color = Colors.golden_angle_color(
+                colormap="magma", num=self.row + 1, format="HEX"
+            )[-1]
+            self.color_button.setColor(default_color)
+
+        self.table_widget.setCellWidget(self.row, 0, self.device_line_edit)
+        self.table_widget.setCellWidget(self.row, 1, self.entry_line_edit)
+        self.table_widget.setCellWidget(self.row, 2, self.dap_combo)
+        self.table_widget.setCellWidget(self.row, 3, self.color_button)
+        self.table_widget.setCellWidget(self.row, 4, self.style_combo)
+        self.table_widget.setCellWidget(self.row, 5, self.width)
+        self.table_widget.setCellWidget(self.row, 6, self.symbol_size)
+        self.table_widget.setCellWidget(self.row, 7, self.remove_button)
+
+    @Slot()
+    def remove_row(self):
+        row = self.table_widget.indexAt(self.remove_button.pos()).row()
+        self.cleanup()
+        self.table_widget.removeRow(row)
+
+    def cleanup(self):
+        self.device_line_edit.cleanup()
+
+
+class StyleComboBox(QComboBox):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.addItems(["solid", "dash", "dot", "dashdot"])
+
+
+class RemoveButton(QPushButton):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        icon_path = os.path.join(MODULE_PATH, "assets", "toolbar_icons", "remove.svg")
+        self.setIcon(QIcon(icon_path))

bec_widgets/widgets/waveform/waveform_toolbar/curve_dialog/curve_dialog.ui

@@ -0,0 +1,358 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>Form</class>
+ <widget class="QWidget" name="Form">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>720</width>
+    <height>806</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout_2">
+   <property name="leftMargin">
+    <number>2</number>
+   </property>
+   <property name="topMargin">
+    <number>2</number>
+   </property>
+   <property name="rightMargin">
+    <number>2</number>
+   </property>
+   <property name="bottomMargin">
+    <number>2</number>
+   </property>
+   <item>
+    <widget class="QGroupBox" name="x_group_box">
+     <property name="title">
+      <string>X Axis</string>
+     </property>
+     <layout class="QHBoxLayout" name="horizontalLayout" stretch="0,0,0,0,1,3,1,3">
+      <item>
+       <widget class="QLabel" name="x_mode_label">
+        <property name="text">
+         <string>X Axis Mode</string>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="QComboBox" name="x_mode">
+        <property name="minimumSize">
+         <size>
+          <width>150</width>
+          <height>26</height>
+         </size>
+        </property>
+        <item>
+         <property name="text">
+          <string>best_effort</string>
+         </property>
+        </item>
+        <item>
+         <property name="text">
+          <string>device</string>
+         </property>
+        </item>
+        <item>
+         <property name="text">
+          <string>index</string>
+         </property>
+        </item>
+        <item>
+         <property name="text">
+          <string>timestamp</string>
+         </property>
+        </item>
+       </widget>
+      </item>
+      <item>
+       <spacer name="horizontalSpacer_3">
+        <property name="orientation">
+         <enum>Qt::Orientation::Horizontal</enum>
+        </property>
+        <property name="sizeHint" stdset="0">
+         <size>
+          <width>40</width>
+          <height>20</height>
+         </size>
+        </property>
+       </spacer>
+      </item>
+      <item>
+       <spacer name="horizontalSpacer_4">
+        <property name="orientation">
+         <enum>Qt::Orientation::Horizontal</enum>
+        </property>
+        <property name="sizeHint" stdset="0">
+         <size>
+          <width>40</width>
+          <height>20</height>
+         </size>
+        </property>
+       </spacer>
+      </item>
+      <item>
+       <widget class="QLabel" name="x_name_label">
+        <property name="text">
+         <string>Name</string>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="DeviceLineEdit" name="x_name"/>
+      </item>
+      <item>
+       <widget class="QLabel" name="x_entry_label">
+        <property name="text">
+         <string>Entry</string>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="QLineEdit" name="x_entry"/>
+      </item>
+     </layout>
+    </widget>
+   </item>
+   <item>
+    <widget class="QGroupBox" name="y_group_box">
+     <property name="title">
+      <string>Y Axis</string>
+     </property>
+     <layout class="QVBoxLayout" name="verticalLayout">
+      <item>
+       <widget class="QTabWidget" name="tabWidget">
+        <property name="currentIndex">
+         <number>0</number>
+        </property>
+        <widget class="QWidget" name="tab_scan">
+         <attribute name="title">
+          <string>Scan</string>
+         </attribute>
+         <layout class="QGridLayout" name="gridLayout">
+          <property name="leftMargin">
+           <number>5</number>
+          </property>
+          <property name="topMargin">
+           <number>5</number>
+          </property>
+          <property name="rightMargin">
+           <number>5</number>
+          </property>
+          <property name="bottomMargin">
+           <number>5</number>
+          </property>
+          <item row="0" column="2">
+           <widget class="QPushButton" name="normalize_colors_scan">
+            <property name="text">
+             <string>Normalize Colors</string>
+            </property>
+           </widget>
+          </item>
+          <item row="1" column="0" colspan="4">
+           <widget class="QTableWidget" name="scan_table">
+            <property name="rowCount">
+             <number>0</number>
+            </property>
+            <attribute name="horizontalHeaderCascadingSectionResizes">
+             <bool>false</bool>
+            </attribute>
+            <attribute name="horizontalHeaderShowSortIndicator" stdset="0">
+             <bool>true</bool>
+            </attribute>
+            <attribute name="horizontalHeaderStretchLastSection">
+             <bool>true</bool>
+            </attribute>
+            <attribute name="verticalHeaderStretchLastSection">
+             <bool>false</bool>
+            </attribute>
+            <column>
+             <property name="text">
+              <string>Name</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Entry</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Color</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Style</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Width</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Symbol Size</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Delete</string>
+             </property>
+            </column>
+           </widget>
+          </item>
+          <item row="0" column="1">
+           <spacer name="horizontalSpacer">
+            <property name="orientation">
+             <enum>Qt::Orientation::Horizontal</enum>
+            </property>
+            <property name="sizeHint" stdset="0">
+             <size>
+              <width>40</width>
+              <height>20</height>
+             </size>
+            </property>
+           </spacer>
+          </item>
+          <item row="0" column="0">
+           <widget class="QPushButton" name="add_curve">
+            <property name="toolTip">
+             <string/>
+            </property>
+            <property name="text">
+             <string/>
+            </property>
+           </widget>
+          </item>
+          <item row="0" column="3">
+           <widget class="ColormapSelector" name="color_map_selector_scan"/>
+          </item>
+         </layout>
+        </widget>
+        <widget class="QWidget" name="tab_DAP">
+         <attribute name="title">
+          <string>DAP</string>
+         </attribute>
+         <layout class="QGridLayout" name="gridLayout_2">
+          <property name="leftMargin">
+           <number>5</number>
+          </property>
+          <property name="topMargin">
+           <number>5</number>
+          </property>
+          <property name="rightMargin">
+           <number>5</number>
+          </property>
+          <property name="bottomMargin">
+           <number>5</number>
+          </property>
+          <item row="1" column="0" colspan="4">
+           <widget class="QTableWidget" name="dap_table">
+            <attribute name="horizontalHeaderShowSortIndicator" stdset="0">
+             <bool>true</bool>
+            </attribute>
+            <attribute name="horizontalHeaderStretchLastSection">
+             <bool>true</bool>
+            </attribute>
+            <column>
+             <property name="text">
+              <string>Name</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Entry</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Model</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Color</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Style</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Width</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Symbol Size</string>
+             </property>
+            </column>
+            <column>
+             <property name="text">
+              <string>Delete</string>
+             </property>
+            </column>
+           </widget>
+          </item>
+          <item row="0" column="2">
+           <widget class="QPushButton" name="normalize_colors_dap">
+            <property name="text">
+             <string>Normalize Colors</string>
+            </property>
+           </widget>
+          </item>
+          <item row="0" column="1">
+           <spacer name="horizontalSpacer_2">
+            <property name="orientation">
+             <enum>Qt::Orientation::Horizontal</enum>
+            </property>
+            <property name="sizeHint" stdset="0">
+             <size>
+              <width>585</width>
+              <height>20</height>
+             </size>
+            </property>
+           </spacer>
+          </item>
+          <item row="0" column="0">
+           <widget class="QPushButton" name="add_dap">
+            <property name="text">
+             <string/>
+            </property>
+           </widget>
+          </item>
+          <item row="0" column="3">
+           <widget class="ColormapSelector" name="color_map_selector_dap"/>
+          </item>
+         </layout>
+        </widget>
+       </widget>
+      </item>
+     </layout>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <customwidgets>
+  <customwidget>
+   <class>DeviceLineEdit</class>
+   <extends>QLineEdit</extends>
+   <header>device_line_edit</header>
+  </customwidget>
+  <customwidget>
+   <class>ColormapSelector</class>
+   <extends>QWidget</extends>
+   <header>colormap_selector</header>
+  </customwidget>
+ </customwidgets>
+ <resources/>
+ <connections/>
+</ui>