dcscope 2.22.1__py3-none-any.whl

dcscope/gui/matrix/block_matrix.py

@@ -0,0 +1,141 @@
+import copy
+import importlib.resources
+
+from PyQt6 import uic, QtCore, QtWidgets
+
+
+class BlockMatrix(QtWidgets.QWidget):
+    pipeline_changed = QtCore.pyqtSignal(dict)
+    quickviewed = QtCore.pyqtSignal(int, int)
+
+    filter_modify_clicked = QtCore.pyqtSignal(str)
+    plot_modify_clicked = QtCore.pyqtSignal(str)
+    slot_modify_clicked = QtCore.pyqtSignal(str)
+
+    def __init__(self, *args, **kwargs):
+        """Helper class that wraps DataMatrix and PlotMatrix"""
+        super(BlockMatrix, self).__init__(*args, **kwargs)
+        ref = importlib.resources.files(
+            "dcscope.gui.matrix") / "block_matrix.ui"
+        with importlib.resources.as_file(ref) as path_ui:
+            uic.loadUi(path_ui, self)
+
+        self._old_state = {}
+        # Signals
+        # DataMatrix
+        self.data_matrix.matrix_changed.connect(self.on_matrix_changed)
+        self.data_matrix.filter_modify_clicked.connect(
+            self.filter_modify_clicked)
+        self.data_matrix.slot_modify_clicked.connect(self.slot_modify_clicked)
+        self.data_matrix.quickviewed.connect(self.quickviewed)
+        # PlotMatrix
+        self.plot_matrix.matrix_changed.connect(self.on_matrix_changed)
+        self.plot_matrix.plot_modify_clicked.connect(self.plot_modify_clicked)
+
+    def read_pipeline_state(self):
+        state = self.data_matrix.read_pipeline_state()
+        statep = self.plot_matrix.read_pipeline_state()
+        state["plots"] = statep["plots"]
+        for ss in statep["elements"]:
+            for plot in statep["elements"][ss]:
+                state["elements"][ss][plot] = statep["elements"][ss][plot]
+        return state
+
+    def write_pipeline_state(self, state):
+        # DataMatrix
+        stated = copy.deepcopy(state)
+        stated.pop("plots")
+        for slot_state in state["slots"]:
+            slot_id = slot_state["identifier"]
+            for plot_state in state["plots"]:
+                plot_id = plot_state["identifier"]
+                stated["elements"][slot_id].pop(plot_id)
+        self.data_matrix.write_pipeline_state(stated)
+        # PlotMatrix
+        statep = copy.deepcopy(state)
+        statep.pop("filters")
+        statep.pop("filters used")
+        statep.pop("slots used")
+        for slot_state in state["slots"]:
+            slot_id = slot_state["identifier"]
+            for filt_state in state["filters"]:
+                filt_id = filt_state["identifier"]
+                statep["elements"][slot_id].pop(filt_id)
+        self.plot_matrix.write_pipeline_state(statep)
+
+    def add_dataset(self, *args, **kwargs):
+        self.data_matrix.add_dataset(*args, **kwargs)
+
+    def add_filter(self, *args, **kwargs):
+        self.data_matrix.add_filter(*args, **kwargs)
+
+    def add_plot(self, *args, **kwargs):
+        self.plot_matrix.add_plot(*args, **kwargs)
+
+    def adopt_pipeline(self, pipeline_state):
+        self.write_pipeline_state(pipeline_state)
+
+    def enable_quickview(self, view):
+        self.data_matrix.enable_quickview(view)
+
+    def get_quickview_indices(self):
+        return self.data_matrix.get_quickview_indices()
+
+    def get_widget(self, slot_id=None, filt_plot_id=None):
+        """Convenience function for testing"""
+        if slot_id is None and filt_plot_id is not None:
+            # get a filter or a plot
+            w = self.data_matrix.filter_widgets + self.plot_matrix.plot_widgets
+            for wi in w:
+                if wi.identifier == filt_plot_id:
+                    break
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        elif slot_id is not None and filt_plot_id is None:
+            # get a slot
+            for wi in self.data_matrix.dataset_widgets:
+                if wi.identifier == slot_id:
+                    break
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        elif slot_id is not None and filt_plot_id is not None:
+            # get a matrix element
+            wd = self.data_matrix.element_widget_dict
+            wp = self.plot_matrix.element_widget_dict
+            fpd = wd[slot_id]
+            fpp = wp[slot_id]
+            if filt_plot_id in fpp:
+                wi = fpp[filt_plot_id]
+            elif filt_plot_id in fpd:
+                wi = fpd[filt_plot_id]
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        else:
+            raise ValueError(
+                "At least one of `slot_id`, `filt_plot_id` must be specified!")
+
+    def invalidate_elements(self, invalid_dm, invalid_pm):
+        for slot_id, filt_id in invalid_dm:
+            em = self.data_matrix.get_matrix_element(slot_id, filt_id)
+            em.active = False
+            em.invalid = True
+            em.update_content()
+        for slot_id, plot_id in invalid_pm:
+            em = self.plot_matrix.get_matrix_element(slot_id, plot_id)
+            em.active = False
+            em.invalid = True
+            em.update_content()
+
+    def on_matrix_changed(self):
+        state = self.read_pipeline_state()
+        self.pipeline_changed.emit(state)
+
+    def update(self, *args, **kwargs):
+        self.scrollArea_block.update()
+        super(BlockMatrix, self).update(*args, **kwargs)

dcscope/gui/matrix/block_matrix.ui

@@ -0,0 +1,207 @@
+<?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>310</width>
+    <height>464</height>
+   </rect>
+  </property>
+  <property name="font">
+   <font>
+    <pointsize>9</pointsize>
+   </font>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <property name="topMargin">
+    <number>0</number>
+   </property>
+   <property name="rightMargin">
+    <number>0</number>
+   </property>
+   <property name="bottomMargin">
+    <number>0</number>
+   </property>
+   <item>
+    <widget class="QScrollArea" name="scrollArea_block">
+     <property name="sizePolicy">
+      <sizepolicy hsizetype="Preferred" vsizetype="Preferred">
+       <horstretch>0</horstretch>
+       <verstretch>0</verstretch>
+      </sizepolicy>
+     </property>
+     <property name="minimumSize">
+      <size>
+       <width>300</width>
+       <height>0</height>
+      </size>
+     </property>
+     <property name="lineWidth">
+      <number>1</number>
+     </property>
+     <property name="verticalScrollBarPolicy">
+      <enum>Qt::ScrollBarAsNeeded</enum>
+     </property>
+     <property name="horizontalScrollBarPolicy">
+      <enum>Qt::ScrollBarAsNeeded</enum>
+     </property>
+     <property name="sizeAdjustPolicy">
+      <enum>QAbstractScrollArea::AdjustToContents</enum>
+     </property>
+     <property name="widgetResizable">
+      <bool>true</bool>
+     </property>
+     <property name="alignment">
+      <set>Qt::AlignLeading|Qt::AlignLeft|Qt::AlignTop</set>
+     </property>
+     <widget class="QWidget" name="scrollAreaWidgetContents">
+      <property name="geometry">
+       <rect>
+        <x>0</x>
+        <y>0</y>
+        <width>299</width>
+        <height>462</height>
+       </rect>
+      </property>
+      <layout class="QGridLayout" name="gridLayout_3">
+       <property name="spacing">
+        <number>2</number>
+       </property>
+       <item row="0" column="2" alignment="Qt::AlignLeft|Qt::AlignTop">
+        <widget class="QToolButton" name="toolButton_new_filter">
+         <property name="toolTip">
+          <string>Add a new filter</string>
+         </property>
+         <property name="text">
+          <string>+</string>
+         </property>
+         <property name="icon">
+          <iconset theme="view-filter">
+           <normaloff>.</normaloff>.</iconset>
+         </property>
+        </widget>
+       </item>
+       <item row="0" column="3">
+        <widget class="PlotMatrix" name="plot_matrix" native="true">
+         <property name="sizePolicy">
+          <sizepolicy hsizetype="Expanding" vsizetype="Expanding">
+           <horstretch>0</horstretch>
+           <verstretch>0</verstretch>
+          </sizepolicy>
+         </property>
+         <property name="minimumSize">
+          <size>
+           <width>91</width>
+           <height>67</height>
+          </size>
+         </property>
+         <property name="maximumSize">
+          <size>
+           <width>95</width>
+           <height>156</height>
+          </size>
+         </property>
+        </widget>
+       </item>
+       <item row="0" column="4" alignment="Qt::AlignLeft|Qt::AlignTop">
+        <widget class="QToolButton" name="toolButton_new_plot">
+         <property name="toolTip">
+          <string>Add a new plot</string>
+         </property>
+         <property name="text">
+          <string>+</string>
+         </property>
+         <property name="icon">
+          <iconset theme="draw-watercolor">
+           <normaloff>.</normaloff>.</iconset>
+         </property>
+        </widget>
+       </item>
+       <item row="0" column="5">
+        <spacer name="horizontalSpacer_2">
+         <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 row="0" column="0" rowspan="2" colspan="2">
+        <widget class="DataMatrix" name="data_matrix" native="true">
+         <property name="sizePolicy">
+          <sizepolicy hsizetype="Expanding" vsizetype="Expanding">
+           <horstretch>0</horstretch>
+           <verstretch>0</verstretch>
+          </sizepolicy>
+         </property>
+         <property name="minimumSize">
+          <size>
+           <width>67</width>
+           <height>99</height>
+          </size>
+         </property>
+        </widget>
+       </item>
+       <item row="3" column="1">
+        <spacer name="verticalSpacer">
+         <property name="orientation">
+          <enum>Qt::Vertical</enum>
+         </property>
+         <property name="sizeHint" stdset="0">
+          <size>
+           <width>20</width>
+           <height>0</height>
+          </size>
+         </property>
+        </spacer>
+       </item>
+       <item row="2" column="0">
+        <widget class="QToolButton" name="toolButton_load_dataset">
+         <property name="toolTip">
+          <string>Import a dataset</string>
+         </property>
+         <property name="text">
+          <string>+</string>
+         </property>
+         <property name="icon">
+          <iconset theme="office-chart-ring">
+           <normaloff>.</normaloff>.</iconset>
+         </property>
+         <property name="autoRaise">
+          <bool>false</bool>
+         </property>
+        </widget>
+       </item>
+      </layout>
+     </widget>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <customwidgets>
+  <customwidget>
+   <class>DataMatrix</class>
+   <extends>QWidget</extends>
+   <header>dcscope.gui.matrix.data_matrix</header>
+   <container>1</container>
+  </customwidget>
+  <customwidget>
+   <class>PlotMatrix</class>
+   <extends>QWidget</extends>
+   <header>dcscope.gui.matrix.plot_matrix</header>
+   <container>1</container>
+  </customwidget>
+ </customwidgets>
+ <resources/>
+ <connections/>
+</ui>