waveform: add Waveform

artiq/dashboard/waveform.py

@@ -3,10 +3,16 @@ from PyQt5.QtCore import Qt
 
 from artiq.gui.models import DictSyncTreeSepModel, LocalModelManager
 
+import bisect
+import pyqtgraph as pg
 import logging
 
 logger = logging.getLogger(__name__)
 
+DISPLAY_LOW = 0
+DISPLAY_HIGH = 1
+DISPLAY_MID = 0.5
+
 
 class Model(DictSyncTreeSepModel):
     def __init__(self, init):
@@ -61,3 +67,136 @@ class _AddChannelDialog(QtWidgets.QDialog):
                 channels.append((key, width))
         self.accepted.emit(channels)
         self.close()
+
+
+class BackgroundItem(pg.GraphicsWidgetAnchor, pg.GraphicsWidget):
+    def __init__(self, parent, rect):
+        pg.GraphicsWidget.__init__(self, parent)
+        pg.GraphicsWidgetAnchor.__init__(self)
+        self.item = QtWidgets.QGraphicsRectItem(rect, self)
+        brush = QtGui.QBrush(QtGui.QColor(10, 10, 10, 140))
+        self.item.setBrush(brush)
+
+
+class Waveform(pg.PlotWidget):
+    MIN_HEIGHT = 50
+    MAX_HEIGHT = 200
+    PREF_HEIGHT = 75
+
+    cursorMoved = QtCore.pyqtSignal(float)
+
+    def __init__(self, channel, state, parent=None):
+        pg.PlotWidget.__init__(self,
+                               parent=parent,
+                               x=None,
+                               y=None,
+                               pen="r",
+                               stepMode="right",
+                               connect="finite")
+
+        self.setMinimumHeight(Waveform.MIN_HEIGHT)
+        self.setMaximumHeight(Waveform.MAX_HEIGHT)
+        self.setMenuEnabled(False)
+        self.setContextMenuPolicy(QtCore.Qt.ActionsContextMenu)
+
+        self.channel = channel
+        self.name = channel[0]
+        self.width = channel[1][0]
+
+        self.state = state
+        self.x_data = []
+        self.y_data = []
+
+        self.plot_item = self.getPlotItem()
+        self.plot_item.hideButtons()
+        self.plot_item.getAxis("bottom").setStyle(showValues=False, tickLength=0)
+        self.plot_item.hideAxis("top")
+        self.plot_item.getAxis("left").setStyle(showValues=False, tickLength=0)
+        self.plot_item.setRange(yRange=(DISPLAY_LOW, DISPLAY_HIGH), padding=0.1)
+        self.plot_item.showGrid(x=True, y=True)
+
+        self.plot_data_item = self.plot_item.listDataItems()[0]
+        self.plot_data_item.setClipToView(True)
+
+        self.view_box = self.plot_item.getViewBox()
+        self.view_box.setMouseEnabled(x=True, y=False)
+        self.view_box.disableAutoRange(axis=pg.ViewBox.YAxis)
+        self.view_box.setLimits(xMin=0, minXRange=20)
+
+        self.cursor = pg.InfiniteLine()
+        self.cursor_y = 0
+        self.addItem(self.cursor)
+
+        self.cursor_label = pg.LabelItem('', parent=self.plot_item)
+        self.cursor_label.anchor(itemPos=(0, 0), parentPos=(0, 0), offset=(0, 20))
+        self.cursor_label.setAttr('justify', 'left')
+        self.cursor_label.setZValue(10)
+
+        self.title_label = pg.LabelItem(self.name, parent=self.plot_item)
+        self.title_label.anchor(itemPos=(0, 0), parentPos=(0, 0), offset=(0, 0))
+        self.title_label.setAttr('justify', 'left')
+        self.title_label.setZValue(10)
+
+        rect = self.title_label.boundingRect()
+        rect.setHeight(rect.height() * 2)
+        self.label_bg = BackgroundItem(parent=self.plot_item, rect=rect)
+        self.label_bg.anchor(itemPos=(0, 0), parentPos=(0, 0), offset=(0, 0))
+
+    def update_x_max(self):
+        self.view_box.setLimits(xMax=self.state["stopped_x"])
+
+    def set_cursor_visible(self, visible):
+        if visible:
+            self.addItem(self.cursor)
+        else:
+            self.removeItem(self.cursor)
+
+    def on_cursor_move(self, x):
+        self.cursor.setValue(x)
+        if len(self.x_data) < 1:
+            return
+        ind = bisect.bisect_left(self.x_data, x) - 1
+        dr = self.plot_data_item.dataRect()
+        if dr is None:
+            self.cursor_y = None
+        elif dr.left() <= x \
+                and 0 <= ind < len(self.y_data):
+            self.cursor_y = self.y_data[ind]
+        elif x >= dr.right():
+            self.cursor_y = self.y_data[-1]
+        else:
+            self.cursor_y = None
+        self.format_cursor_label()
+
+    def extract_data_from_state(self):
+        raise NotImplementedError
+
+    def display(self):
+        raise NotImplementedError
+
+    def format_cursor_label(self):
+        raise NotImplementedError
+
+    # override
+    def mouseMoveEvent(self, e):
+        if e.buttons() == QtCore.Qt.LeftButton \
+           and e.modifiers() == QtCore.Qt.ShiftModifier:
+            drag = QtGui.QDrag(self)
+            mime = QtCore.QMimeData()
+            drag.setMimeData(mime)
+            pixmapi = QtWidgets.QApplication.style().standardIcon(
+                QtWidgets.QStyle.SP_FileIcon)
+            drag.setPixmap(pixmapi.pixmap(32))
+            drag.exec_(QtCore.Qt.MoveAction)
+        else:
+            super().mouseMoveEvent(e)
+
+    # override
+    def mouseDoubleClickEvent(self, e):
+        pos = self.view_box.mapSceneToView(e.pos())
+        self.cursorMoved.emit(pos.x())
+
+    # override
+    def wheelEvent(self, e):
+        if e.modifiers() & QtCore.Qt.ControlModifier:
+            super().wheelEvent(e)