dcscope 2.22.1__py3-none-any.whl

dcscope/gui/matrix/plot_matrix.py

@@ -0,0 +1,316 @@
+import copy
+
+from PyQt6 import QtCore, QtWidgets
+
+from ... import pipeline
+
+from .pm_element import MatrixElement
+from .pm_plot import MatrixPlot
+
+
+class PlotMatrix(QtWidgets.QWidget):
+    plot_modify_clicked = QtCore.pyqtSignal(str)
+    matrix_changed = QtCore.pyqtSignal()
+
+    def __init__(self, *args, **kwargs):
+        super(PlotMatrix, self).__init__(*args, **kwargs)
+
+        self.glo = None
+        self._reset_layout(init=True)
+
+        # used for toggling between all active, all inactive and semi state
+        self.semi_states_plot = {}
+
+    def read_pipeline_state(self):
+        """State of the current plot matrix"""
+        # plots
+        plots = []
+        for pw in self.plot_widgets:
+            pw_state = pw.read_pipeline_state()
+            plot = pipeline.Plot._instances[pw_state["identifier"]]
+            plots.append(plot.__getstate__())
+        # elements
+        mestates = {}
+        dm = self.data_matrix
+        for dw in dm.dataset_widgets:
+            idict = {}
+            for pw in self.plot_widgets:
+                me = self.get_matrix_element(dw.identifier, pw.identifier)
+                idict[pw.identifier] = me.read_pipeline_state()["active"]
+            mestates[dw.identifier] = idict
+        state = {"elements": mestates,
+                 "plots": plots}
+        return state
+
+    def write_pipeline_state(self, state):
+        self.setUpdatesEnabled(False)
+        self.blockSignals(True)
+        self.clear()
+        # plot states
+        for jj in range(len(state["plots"])):
+            plot_id = state["plots"][jj]["identifier"]
+            pw_state = {"identifier": plot_id,
+                        "name": state["plots"][jj]["layout"]["name"],
+                        }
+            self.add_plot(state=pw_state)
+        # make sure elements exist
+        self.fill_elements()
+        # element states
+        for slot_id in state["elements"]:
+            ds_d = state["elements"][slot_id]
+            for plot_id in ds_d:
+                el = self.get_matrix_element(slot_id, plot_id)
+                el_state = el.read_pipeline_state()
+                el_state["active"] = ds_d[plot_id]
+                el.write_pipeline_state(el_state)
+        self.blockSignals(False)
+        self.setUpdatesEnabled(True)
+
+    def _reset_layout(self, init=False):
+        if self.glo is not None:
+            # send old layout to Nirvana eventually
+            layout = self.glo
+            self.glo = None
+            self.old_layout = QtWidgets.QWidget()
+            self.old_layout.setLayout(layout)
+            self.old_layout.hide()
+            self.old_layout.deleteLater()
+        # add new layout
+        self.glo = QtWidgets.QGridLayout()
+        self.glo.setAlignment(QtCore.Qt.AlignmentFlag.AlignLeft)
+        self.glo.setSpacing(2)
+        self.glo.setContentsMargins(0, 0, 0, 0)
+        self.setLayout(self.glo)
+
+    @property
+    def data_matrix(self):
+        for ch in self.parent().children():
+            if ch.__class__.__name__ == "DataMatrix":
+                break
+        else:
+            raise KeyError("DataMatrix not found!")
+        return ch
+
+    @property
+    def element_widget_dict(self):
+        els = {}
+        for ii, ws in enumerate(self.data_matrix.dataset_widgets):
+            elsd = {}
+            for jj, wf in enumerate(self.plot_widgets):
+                it = self.glo.itemAtPosition(ii+1, jj)
+                elsd[wf.identifier] = it.widget()
+            els[ws.identifier] = elsd
+        return els
+
+    @property
+    def element_width(self):
+        """Data matrix element width (without 2px spacing)"""
+        return self.data_matrix.element_width
+
+    @property
+    def num_datasets(self):
+        dm = self.data_matrix
+        return dm.num_datasets
+
+    @property
+    def num_plots(self):
+        count = 0
+        for jj in range(self.glo.columnCount()):
+            if self.glo.itemAtPosition(0, jj) is not None:
+                count += 1
+        return count
+
+    @property
+    def plot_widgets(self):
+        plots = []
+        for jj in range(self.glo.columnCount()):
+            item = self.glo.itemAtPosition(0, jj)
+            if item is not None:
+                ps = item.widget()
+                plots.append(ps)
+        return plots
+
+    def add_plot(self, identifier=None, state=None):
+        self.setUpdatesEnabled(False)
+        mp = MatrixPlot(identifier=identifier, state=state)
+        mp.option_action.connect(self.on_option_plot)
+        mp.active_toggled.connect(self.toggle_plot_active)
+        mp.modify_clicked.connect(self.plot_modify_clicked.emit)
+        self.glo.addWidget(mp, 0, self.num_plots)
+        self.fill_elements()
+        self.adjust_size()
+        self.setUpdatesEnabled(True)
+        self.publish_matrix()
+        return mp
+
+    def adjust_size(self):
+        ncols = self.num_plots
+        nrows = self.data_matrix.num_datasets
+        if ncols and nrows:
+            width1 = self.glo.itemAt(0).widget().width()
+            width = (width1 + 2)*ncols - 2
+            height = self.data_matrix.sizeHint().height()
+            self.setMinimumSize(width, height)
+            self.setFixedSize(width, height)
+            QtWidgets.QApplication.processEvents(
+                QtCore.QEventLoop.ProcessEventsFlag.AllEvents, 300)
+            self.setMinimumSize(width, height)
+            self.setFixedSize(width, height)
+
+    @QtCore.pyqtSlot()
+    def changed_element(self):
+        self.publish_matrix()
+
+    def clear(self):
+        """Reset layout"""
+        self._reset_layout()
+        self.semi_states_plot = {}
+
+    def fill_elements(self):
+        # add widgets
+        for ii in range(self.num_datasets):
+            for jj in range(self.num_plots):
+                if self.glo.itemAtPosition(ii+1, jj) is None:
+                    me = MatrixElement()
+                    me.element_changed.connect(self.changed_element)
+                    self.glo.addWidget(me, ii+1, jj)
+        # make sure enabled/disabled is honored
+        dstate = self.data_matrix.read_pipeline_state()
+        pstate = self.read_pipeline_state()
+        for slot_state in dstate["slots"]:
+            slot_id = slot_state["identifier"]
+            if slot_id not in dstate["slots used"]:
+                for plot_state in pstate["plots"]:
+                    plot_id = plot_state["identifier"]
+                    me = self.get_matrix_element(slot_id, plot_id)
+                    mstate = me.read_pipeline_state()
+                    mstate["enabled"] = False
+                    me.write_pipeline_state(mstate)
+
+    def get_matrix_element(self, dataset_id, plot_id):
+        """Return matrix element matching dataset and plot identifiers"""
+        ncols = self.glo.columnCount()
+        nrows = self.glo.rowCount()
+        for ii in range(1, nrows):
+            dw = self.data_matrix.glo.itemAtPosition(ii, 0).widget()
+            if dw.identifier == dataset_id:
+                for jj in range(ncols):
+                    f = self.glo.itemAtPosition(0, jj).widget()
+                    if f.identifier == plot_id:
+                        break
+                else:
+                    raise KeyError("Plot '{}' not found!".format(plot_id))
+                break
+        else:
+            raise KeyError("Dataset '{}' not found!".format(dataset_id))
+        return self.glo.itemAtPosition(ii, jj).widget()
+
+    def get_plot_index(self, plot_id):
+        for ii, pw in enumerate(self.plot_widgets):
+            if pw.identifier == plot_id:
+                break
+        else:
+            raise KeyError("Dataset '{}' not found!".format(plot_id))
+        return ii
+
+    def get_plot_widget_state(self, plot_id, ret_index=False):
+        ii = self.get_plot_index(plot_id)
+        pw = self.plot_widgets[ii]
+        if ret_index:
+            return pw.read_pipeline_state(), ii
+        else:
+            return pw.read_pipeline_state()
+
+    @QtCore.pyqtSlot(str)
+    def on_option_plot(self, option):
+        """Plot option logic (remove, duplicate)"""
+        pw_state = self.sender().read_pipeline_state()
+        plot_id = pw_state["identifier"]
+        plot_index = self.get_plot_index(plot_id)
+        state = self.read_pipeline_state()
+        if option == "remove":
+            state["plots"].pop(plot_index)
+            for ds_key in state["elements"]:
+                state["elements"][ds_key].pop(plot_id)
+        else:  # duplicate
+            plot = pipeline.Plot()
+            new_state = copy.deepcopy(state["plots"][plot_index])
+            new_state["identifier"] = plot.identifier
+            new_state["layout"]["name"] = plot.name
+            state["plots"].insert(plot_index+1, new_state)
+            plot.__setstate__(new_state)
+        self.write_pipeline_state(state)
+        self.publish_matrix()
+
+    def publish_matrix(self):
+        """Publish state via self.matrix_changed signal for Pipeline"""
+        if not self.signalsBlocked():
+            self.matrix_changed.emit()
+
+    @QtCore.pyqtSlot(bool)
+    def toggle_dataset_enable(self, enabled):
+        sender = self.sender()
+        sid = sender.identifier
+        state = self.read_pipeline_state()
+        for p_key in state["elements"][sid]:
+            # update element widget
+            me = self.get_matrix_element(sid, p_key)
+            mstate = me.read_pipeline_state()
+            mstate["enabled"] = enabled
+            me.write_pipeline_state(mstate)
+        self.publish_matrix()
+
+    @QtCore.pyqtSlot()
+    def toggle_plot_active(self):
+        """Switch between all active, all inactive, previous state
+
+        Modifies the matrix elements for a plot/column,
+        which is defined by the signal sender :class:`MatrixPlot`.
+        Cyclic toggling order: semi -> all -> none
+        """
+        sender = self.sender()
+        plot_id = sender.identifier
+
+        states = self.read_pipeline_state()["elements"]
+        state = {}
+        for slot_id in states:
+            state[slot_id] = states[slot_id][plot_id]
+
+        num_actives = sum(list(state.values()))
+
+        # update state according to the scheme in the docstring
+        if num_actives == 0:
+            if plot_id in self.semi_states_plot:
+                # use semi state
+                oldstate = self.semi_states_plot[plot_id]
+                for slot_id in oldstate:
+                    if slot_id in state:
+                        state[slot_id] = oldstate[slot_id]
+            else:
+                # toggle all to active
+                for slot_id in state:
+                    state[slot_id] = True
+        elif num_actives == len(state):
+            # toggle all to inactive
+            for slot_id in state:
+                state[slot_id] = False
+        else:
+            # save semi state
+            self.semi_states_plot[plot_id] = copy.deepcopy(state)
+            # toggle all to active
+            for slot_id in state:
+                state[slot_id] = True
+
+        for slot_id in state:
+            me = self.get_matrix_element(slot_id, plot_id)
+            me.set_active(state[slot_id])
+        self.publish_matrix()
+
+    def update_content(self):
+        ncols = self.glo.columnCount()
+        nrows = self.glo.rowCount()
+        for ii in range(nrows):
+            for jj in range(ncols):
+                item = self.glo.itemAtPosition(ii, jj)
+                if item is not None:
+                    item.widget().update_content()

dcscope/gui/matrix/pm_element.py

@@ -0,0 +1,39 @@
+from . import dm_element
+
+
+class MatrixElement(dm_element.MatrixElement):
+    def mousePressEvent(self, event):
+        # toggle selection
+        if not self.invalid:
+            self.active = not self.active
+            self.element_changed.emit()
+            self.update_content()
+            event.accept()
+
+    def update_content(self, quickview=False):
+        if self.invalid:
+            color = "#DCDCDC"  # gray
+            label = "invalid"
+            tooltip = "Incompatible plot settings"
+        elif self.active and self.enabled:
+            color = "#86E7C1"  # turquois
+            label = "active"
+            tooltip = "Click to deactivate"
+        elif self.active and not self.enabled:
+            color = "#C9DAD7"  # gray-turquois
+            label = "active\n(unused)"
+            tooltip = "Click to deactivate"
+        elif not self.active and self.enabled:
+            color = "#EFEFEF"  # light gray
+            label = "inactive"
+            tooltip = "Click to activate"
+        else:
+            color = "#DCDCDC"  # gray
+            label = "inactive\n(unused)"
+            tooltip = "Click to activate"
+
+        self.label.setText(label)
+        self.setToolTip(tooltip)
+        self.label.setToolTip(tooltip)
+        self.setStyleSheet(
+            "background-color:{};color:black".format(color))

dcscope/gui/matrix/pm_plot.py

@@ -0,0 +1,86 @@
+import importlib.resources
+
+from PyQt6 import uic, QtCore, QtWidgets
+
+from ... import pipeline
+
+
+class MatrixPlot(QtWidgets.QWidget):
+    active_toggled = QtCore.pyqtSignal()
+    option_action = QtCore.pyqtSignal(str)
+    modify_clicked = QtCore.pyqtSignal(str)
+
+    def __init__(self, identifier=None, state=None, *args, **kwargs):
+        super(MatrixPlot, self).__init__(*args, **kwargs)
+        ref = importlib.resources.files("dcscope.gui.matrix") / "pm_plot.ui"
+        with importlib.resources.as_file(ref) as path_ui:
+            uic.loadUi(path_ui, self)
+
+        # options button
+        menu = QtWidgets.QMenu()
+        menu.addAction('duplicate', self.action_duplicate)
+        menu.addAction('remove', self.action_remove)
+        self.toolButton_opt.setMenu(menu)
+
+        # toggle all active, all inactive, semi state
+        self.toolButton_toggle.clicked.connect(self.active_toggled.emit)
+        self.toolButton_modify.clicked.connect(self.on_modify)
+
+        if state is None:
+            plot = pipeline.Plot._instances[identifier]
+            self.identifier = identifier
+            self.name = plot.name
+            # set tooltip/label
+            self.update_content()
+        else:
+            self.write_pipeline_state(state)
+
+    def read_pipeline_state(self):
+        state = {"name": self.name,
+                 "identifier": self.identifier,
+                 }
+        return state
+
+    def write_pipeline_state(self, state):
+        self.identifier = state["identifier"]
+        self.name = state["name"]
+        self.update_content()
+
+    @property
+    def name(self):
+        plot = pipeline.Plot._instances[self.identifier]
+        return plot.name
+
+    @name.setter
+    def name(self, text):
+        plot = pipeline.Plot._instances[self.identifier]
+        plot.name = text
+
+    def action_duplicate(self):
+        self.option_action.emit("duplicate")
+
+    def action_remove(self):
+        self.option_action.emit("remove")
+
+    def on_modify(self):
+        self.modify_clicked.emit(self.identifier)
+
+    def set_label_string(self, string):
+        if self.label.fontMetrics().boundingRect(string).width() < 60:
+            nstring = string
+        else:
+            nstring = string + "..."
+            while True:
+                width = self.label.fontMetrics().boundingRect(nstring).width()
+                if width > 60:
+                    nstring = nstring[:-4] + "..."
+                else:
+                    break
+        self.label.setText(nstring)
+
+    @QtCore.pyqtSlot()
+    def update_content(self):
+        """Reset tool tips and title"""
+        self.label.setToolTip(self.name)
+        self.set_label_string(self.name)
+        self.update()

dcscope/gui/matrix/pm_plot.ui

@@ -0,0 +1,249 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>Form</class>
+ <widget class="QWidget" name="Form">
+  <property name="enabled">
+   <bool>true</bool>
+  </property>
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>94</width>
+    <height>110</height>
+   </rect>
+  </property>
+  <property name="sizePolicy">
+   <sizepolicy hsizetype="Minimum" vsizetype="Minimum">
+    <horstretch>0</horstretch>
+    <verstretch>0</verstretch>
+   </sizepolicy>
+  </property>
+  <property name="maximumSize">
+   <size>
+    <width>120</width>
+    <height>120</height>
+   </size>
+  </property>
+  <property name="font">
+   <font>
+    <pointsize>9</pointsize>
+   </font>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <property name="autoFillBackground">
+   <bool>false</bool>
+  </property>
+  <property name="styleSheet">
+   <string notr="true">background-color:#86B5E7; color:black</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <property name="spacing">
+    <number>0</number>
+   </property>
+   <property name="sizeConstraint">
+    <enum>QLayout::SetFixedSize</enum>
+   </property>
+   <property name="leftMargin">
+    <number>0</number>
+   </property>
+   <property name="topMargin">
+    <number>0</number>
+   </property>
+   <property name="rightMargin">
+    <number>0</number>
+   </property>
+   <property name="bottomMargin">
+    <number>0</number>
+   </property>
+   <item>
+    <widget class="QWidget" name="widget" native="true">
+     <property name="sizePolicy">
+      <sizepolicy hsizetype="Preferred" vsizetype="Fixed">
+       <horstretch>0</horstretch>
+       <verstretch>0</verstretch>
+      </sizepolicy>
+     </property>
+     <property name="minimumSize">
+      <size>
+       <width>80</width>
+       <height>80</height>
+      </size>
+     </property>
+     <property name="maximumSize">
+      <size>
+       <width>16777215</width>
+       <height>16777215</height>
+      </size>
+     </property>
+     <property name="layoutDirection">
+      <enum>Qt::LeftToRight</enum>
+     </property>
+     <layout class="QVBoxLayout" name="verticalLayout_2">
+      <property name="spacing">
+       <number>2</number>
+      </property>
+      <property name="sizeConstraint">
+       <enum>QLayout::SetFixedSize</enum>
+      </property>
+      <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 alignment="Qt::AlignHCenter">
+       <widget class="QLabel" name="label">
+        <property name="sizePolicy">
+         <sizepolicy hsizetype="Minimum" vsizetype="Minimum">
+          <horstretch>0</horstretch>
+          <verstretch>0</verstretch>
+         </sizepolicy>
+        </property>
+        <property name="font">
+         <font>
+          <pointsize>9</pointsize>
+         </font>
+        </property>
+        <property name="text">
+         <string>Name</string>
+        </property>
+        <property name="alignment">
+         <set>Qt::AlignLeading|Qt::AlignLeft|Qt::AlignVCenter</set>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="QToolButton" name="toolButton_modify">
+        <property name="sizePolicy">
+         <sizepolicy hsizetype="Minimum" vsizetype="Minimum">
+          <horstretch>0</horstretch>
+          <verstretch>0</verstretch>
+         </sizepolicy>
+        </property>
+        <property name="font">
+         <font>
+          <pointsize>9</pointsize>
+         </font>
+        </property>
+        <property name="toolTip">
+         <string>Modify this plot in Analysis View</string>
+        </property>
+        <property name="text">
+         <string>edit</string>
+        </property>
+        <property name="icon">
+         <iconset theme="tools-wizard">
+          <normaloff>.</normaloff>.</iconset>
+        </property>
+        <property name="toolButtonStyle">
+         <enum>Qt::ToolButtonTextBesideIcon</enum>
+        </property>
+        <property name="autoRaise">
+         <bool>true</bool>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <layout class="QHBoxLayout" name="horizontalLayout">
+        <property name="spacing">
+         <number>0</number>
+        </property>
+        <property name="leftMargin">
+         <number>0</number>
+        </property>
+        <property name="topMargin">
+         <number>0</number>
+        </property>
+        <property name="rightMargin">
+         <number>0</number>
+        </property>
+        <item>
+         <widget class="QToolButton" name="toolButton_toggle">
+          <property name="sizePolicy">
+           <sizepolicy hsizetype="Minimum" vsizetype="Minimum">
+            <horstretch>0</horstretch>
+            <verstretch>0</verstretch>
+           </sizepolicy>
+          </property>
+          <property name="toolTip">
+           <string>Include/Exclude all Datasets for this plot</string>
+          </property>
+          <property name="text">
+           <string>c</string>
+          </property>
+          <property name="icon">
+           <iconset theme="object-columns">
+            <normaloff>.</normaloff>.</iconset>
+          </property>
+          <property name="autoRaise">
+           <bool>true</bool>
+          </property>
+         </widget>
+        </item>
+        <item>
+         <spacer name="horizontalSpacer">
+          <property name="orientation">
+           <enum>Qt::Horizontal</enum>
+          </property>
+          <property name="sizeHint" stdset="0">
+           <size>
+            <width>0</width>
+            <height>20</height>
+           </size>
+          </property>
+         </spacer>
+        </item>
+        <item>
+         <widget class="QToolButton" name="toolButton_opt">
+          <property name="sizePolicy">
+           <sizepolicy hsizetype="Minimum" vsizetype="Fixed">
+            <horstretch>0</horstretch>
+            <verstretch>0</verstretch>
+           </sizepolicy>
+          </property>
+          <property name="maximumSize">
+           <size>
+            <width>50</width>
+            <height>16777215</height>
+           </size>
+          </property>
+          <property name="text">
+           <string>...</string>
+          </property>
+          <property name="icon">
+           <iconset theme="preferences-activities">
+            <normaloff>.</normaloff>.</iconset>
+          </property>
+          <property name="popupMode">
+           <enum>QToolButton::InstantPopup</enum>
+          </property>
+          <property name="toolButtonStyle">
+           <enum>Qt::ToolButtonIconOnly</enum>
+          </property>
+          <property name="autoRaise">
+           <bool>true</bool>
+          </property>
+          <property name="arrowType">
+           <enum>Qt::NoArrow</enum>
+          </property>
+         </widget>
+        </item>
+       </layout>
+      </item>
+     </layout>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>