hoppla, forgot sources files

shapeout2/gui/pipeline_plot.py

@@ -0,0 +1,166 @@
+import pkg_resources
+
+import dclab
+import numpy as np
+from PyQt5 import uic, QtWidgets
+import pyqtgraph as pg
+
+from ..pipeline import Plot
+from .. import plot_cache
+
+
+class PipelinePlot(QtWidgets.QWidget):
+    def __init__(self, parent, pipeline, plot_id, *args, **kwargs):
+        QtWidgets.QWidget.__init__(self, parent=parent, *args, **kwargs)
+        path_ui = pkg_resources.resource_filename(
+            "shapeout2.gui", "pipeline_plot.ui")
+        uic.loadUi(path_ui, self)
+        self.pipeline = pipeline
+        self.identifier = plot_id
+        self.update_content()
+
+    def update_content(self):
+        datasets = self.pipeline.get_plot_datasets(self.identifier)
+        plot = Plot.get_instances()[self.identifier]
+        state = plot.__getstate__()
+        # Plot size and title
+        gen = state["general"]
+        parent = self.parent()
+        self.setWindowTitle(gen["name"])
+        self.setMinimumSize(gen["size x"], gen["size y"])
+        self.setMaximumSize(gen["size x"], gen["size y"])
+        size_hint = self.parent().sizeHint()
+        parent.setMinimumSize(size_hint)
+        parent.setMaximumSize(size_hint)
+        self.plot.redraw(datasets, state)
+
+
+class PipelinePlotWidget(pg.PlotWidget):
+    def __init__(self, *args, **kwargs):
+        super(PipelinePlotWidget, self).__init__(*args, **kwargs)
+        self._plot_elements = []
+
+    def redraw(self, datasets, state):
+        # Remove everything
+        for el in self._plot_elements:
+            self.removeItem(el)
+
+        if not datasets:
+            return
+        # General
+        gen = state["general"]
+        self.plotItem.setLabels(
+            left=dclab.dfn.feature_name2label[gen["axis y"]],
+            bottom=dclab.dfn.feature_name2label[gen["axis x"]])
+        # TODO:
+        # - test whether all datasets have same channel width / pixel size
+        # Isoelastics
+        if gen["isoelastics"]:
+            cfg = datasets[0].config
+            els = add_isoelastics(plot_widget=self,
+                                  plot_state=state,
+                                  channel_width=cfg["setup"]["channel width"],
+                                  pixel_size=cfg["imaging"]["pixel size"])
+            self._plot_elements += els
+        # Set Log scale
+        self.plotItem.setLogMode(x=gen["scale x"] == "logarithmic",
+                                 y=gen["scale y"] == "logarithmic")
+        # Scatter data
+        if state["scatter"]["enabled"]:
+            for rtdc_ds in datasets:
+                sct = add_scatter(plot_widget=self,
+                                  rtdc_ds=rtdc_ds,
+                                  plot_state=state,
+                                  )
+                self._plot_elements += sct
+
+
+def add_isoelastics(plot_widget, plot_state, channel_width, pixel_size):
+    elements = []
+    axis_x = plot_state["general"]["axis x"]
+    axis_y = plot_state["general"]["axis y"]
+    isodef = dclab.isoelastics.get_default()
+    # We do not use isodef.get_with_rtdcbase, because then the
+    # isoelastics would be shifted according to flow rate and.
+    # viscosity. We could do it, but for visualization there is
+    # really no need and also, the plots then look the same as
+    # in Shape-Out 1.
+    try:
+        iso = isodef.get(
+            method="numerical",
+            channel_width=channel_width,
+            flow_rate=None,
+            viscosity=None,
+            col1=axis_x,
+            col2=axis_y,
+            add_px_err=True,
+            px_um=pixel_size)
+    except KeyError:
+        pass
+    else:
+        for ss in iso:
+            iline = pg.PlotDataItem(x=ss[:, 0], y=ss[:, 1])
+            plot_widget.addItem(iline)
+            elements.append(iline)
+            # send them to the back
+            iline.setZValue(-1)
+    return elements
+
+
+def add_scatter(plot_widget, plot_state, rtdc_ds):
+    gen = plot_state["general"]
+    sca = plot_state["scatter"]
+    scatter = pg.ScatterPlotItem(size=sca["marker size"],
+                                 pen=pg.mkPen(color=(0, 0, 0, 0)),
+                                 brush=pg.mkBrush("k"),
+                                 symbol="s")
+    plot_widget.addItem(scatter)
+
+    if sca["marker hue"] == "kde":
+        kde_type = gen["kde"]
+    else:
+        kde_type = "none"
+
+    x, y, kde, _idx = plot_cache.get_scatter_data(
+        rtdc_ds=rtdc_ds,
+        downsample=sca["downsample"] * sca["downsampling value"],
+        xax=gen["axis x"],
+        yax=gen["axis y"],
+        xscale=gen["scale x"],
+        yscale=gen["scale y"],
+        kde_type=kde_type,
+    )
+
+    # define colormap
+    # TODO:
+    # - improve speed?
+    # - common code base with QuickView
+    if sca["marker hue"] == "kde":
+        brush = []
+        kde -= kde.min()
+        kde /= kde.max()
+        num_hues = 500
+        for k in kde:
+            color = pg.intColor(int(k*num_hues), num_hues)
+            brush.append(color)
+    elif sca["marker hue"] == "feature":
+        brush = []
+        feat = rtdc_ds[sca["hue feature"]][rtdc_ds.filter.all]
+        feat -= feat.min()
+        feat /= feat.max()
+        num_hues = 500
+        for f in feat:
+            color = pg.intColor(int(f*num_hues), num_hues)
+            brush.append(color)
+    elif sca["marker hue"] == "dataset":
+        raise NotImplementedError("TODO")
+    else:
+        brush = pg.mkBrush("k")
+    # convert to log-scale if applicable
+    if gen["scale x"] == "logarithmic":
+        x = np.log10(x)
+    if gen["scale y"] == "logarithmic":
+        y = np.log10(y)
+
+    scatter.setData(x=x, y=y, brush=brush)
+    return [scatter]

shapeout2/gui/pipeline_plot.ui

@@ -0,0 +1,63 @@
+<?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>739</width>
+    <height>641</height>
+   </rect>
+  </property>
+  <property name="sizePolicy">
+   <sizepolicy hsizetype="Expanding" vsizetype="Expanding">
+    <horstretch>0</horstretch>
+    <verstretch>0</verstretch>
+   </sizepolicy>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <layout class="QHBoxLayout" name="horizontalLayout">
+   <property name="sizeConstraint">
+    <enum>QLayout::SetMinAndMaxSize</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="PipelinePlotWidget" name="plot" native="true">
+     <property name="sizePolicy">
+      <sizepolicy hsizetype="Expanding" vsizetype="Expanding">
+       <horstretch>0</horstretch>
+       <verstretch>0</verstretch>
+      </sizepolicy>
+     </property>
+     <property name="styleSheet">
+      <string notr="true">background-color:#FFF</string>
+     </property>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <customwidgets>
+  <customwidget>
+   <class>PipelinePlotWidget</class>
+   <extends>QWidget</extends>
+   <header>shapeout2.gui.pipeline_plot</header>
+   <container>1</container>
+  </customwidget>
+ </customwidgets>
+ <resources/>
+ <connections/>
+</ui>