From 7f43c5c31a27c32b633c0d7f9088ae364be4b48f Mon Sep 17 00:00:00 2001 From: Simon Renblad Date: Mon, 5 Feb 2024 11:43:32 +0800 Subject: [PATCH] waveform: add cursor --- artiq/dashboard/waveform.py | 124 +++++++++++++++++++++++++++++++++--- 1 file changed, 115 insertions(+), 9 deletions(-) diff --git a/artiq/dashboard/waveform.py b/artiq/dashboard/waveform.py index 8fef900c0..831efbac6 100644 --- a/artiq/dashboard/waveform.py +++ b/artiq/dashboard/waveform.py @@ -120,6 +120,8 @@ class _BackgroundItem(pg.GraphicsWidgetAnchor, pg.GraphicsWidget): class _BaseWaveform(pg.PlotWidget): + cursorMove = QtCore.pyqtSignal(float) + def __init__(self, name, width, parent=None, pen="r", stepMode="right", connect="finite"): pg.PlotWidget.__init__(self, parent=parent, @@ -136,6 +138,8 @@ class _BaseWaveform(pg.PlotWidget): self.name = name self.width = width + self.x_data = [] + self.y_data = [] self.plot_item = self.getPlotItem() self.plot_item.hideButtons() @@ -160,9 +164,19 @@ class _BaseWaveform(pg.PlotWidget): rect = self.title_label.boundingRect() rect.setHeight(rect.height() * 2) + rect.setWidth(225) self.label_bg = _BackgroundItem(parent=self.plot_item, rect=rect) self.label_bg.anchor(itemPos=(0, 0), parentPos=(0, 0), offset=(0, 0)) + self.cursor = pg.InfiniteLine() + self.cursor_y = None + 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) + def setStoppedX(self, stopped_x): self.stopped_x = stopped_x self.view_box.setLimits(xMax=stopped_x) @@ -171,7 +185,20 @@ class _BaseWaveform(pg.PlotWidget): self.timescale = timescale def onDataChange(self, data): - raise NotImplementedError + try: + self.x_data, self.y_data = zip(*data) + except: + logger.error("Error getting data for waveform: {}".format(self.name), exc_info=True) + + def onCursorMove(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() + self.cursor_y = None + if dr is not None and 0 <= ind < len(self.y_data): + self.cursor_y = self.y_data[ind] def mouseMoveEvent(self, e): if e.buttons() == QtCore.Qt.LeftButton \ @@ -190,6 +217,10 @@ class _BaseWaveform(pg.PlotWidget): if e.modifiers() & QtCore.Qt.ControlModifier: super().wheelEvent(e) + def mouseDoubleClickEvent(self, e): + pos = self.view_box.mapSceneToView(e.pos()) + self.cursorMove.emit(pos.x()) + class BitWaveform(_BaseWaveform): def __init__(self, name, width, parent=None): @@ -197,6 +228,7 @@ class BitWaveform(_BaseWaveform): self._arrows = [] def onDataChange(self, data): + _BaseWaveform.onDataChange(self, data) try: for arw in self._arrows: self.removeItem(arw) @@ -228,29 +260,36 @@ class BitWaveform(_BaseWaveform): self.removeItem(arw) self.plot_data_item.setData(x=[], y=[]) + def onCursorMove(self, x): + _BaseWaveform.onCursorMove(self, x) + self.cursor_label.setText(self.cursor_y) + class AnalogWaveform(_BaseWaveform): def __init__(self, name, width, parent=None): _BaseWaveform.__init__(self, name, width, parent) def onDataChange(self, data): + _BaseWaveform.onDataChange(self, data) try: - x_data, y_data = zip(*data) - self.plot_data_item.setData(x=x_data, y=y_data) - max_y = max(y_data) - min_y = min(y_data) + self.plot_data_item.setData(x=self.x_data, y=self.y_data) + max_y = max(self.y_data) + min_y = min(self.y_data) self.plot_item.setRange(yRange=(min_y, max_y), padding=0.1) except: logger.error( 'Error when displaying waveform: {}'.format(self.name), exc_info=True) self.plot_data_item.setData(x=[], y=[]) + def onCursorMove(self, x): + _BaseWaveform.onCursorMove(self, x) + self.cursor_label.setText(self.cursor_y) + class BitVectorWaveform(_BaseWaveform): def __init__(self, name, width, parent=None): _BaseWaveform.__init__(self, name, width, parent) self._labels = [] - self.x_data = [] self._format_string = "{:0=" + str(math.ceil(width / 4)) + "X}" self.view_box.sigTransformChanged.connect(self._update_labels) @@ -270,11 +309,11 @@ class BitVectorWaveform(_BaseWaveform): self.addItem(lbl) def onDataChange(self, data): + _BaseWaveform.onDataChange(self, data) try: for lbl in self._labels: self.plot_item.removeItem(lbl) self._labels = [] - self.x_data, _ = zip(*data) l = len(data) display_x = np.empty(l * 2) display_y = np.empty(l * 2) @@ -297,6 +336,13 @@ class BitVectorWaveform(_BaseWaveform): self.plot_item.removeItem(lbl) self.plot_data_item.setData(x=[], y=[]) + def onCursorMove(self, x): + _BaseWaveform.onCursorMove(self, x) + t = None + if self.cursor_y is not None: + t = self._format_string.format(int(self.cursor_y, 2)) + self.cursor_label.setText(t) + class LogWaveform(_BaseWaveform): def __init__(self, name, width, parent=None): @@ -306,13 +352,13 @@ class LogWaveform(_BaseWaveform): self._labels = [] def onDataChange(self, data): + _BaseWaveform.onDataChange(self, data) try: for lbl in self._labels: self.plot_item.removeItem(lbl) self._labels = [] - x_data, _ = zip(*data) self.plot_data_item.setData( - x=x_data, y=np.ones(len(x_data))) + x=self.x_data, y=np.ones(len(self.x_data))) old_msg = "" old_x = 0 for x, msg in data: @@ -335,13 +381,19 @@ class LogWaveform(_BaseWaveform): self.plot_item.removeItem(lbl) self.plot_data_item.setData(x=[], y=[]) + def onCursorMove(self, x): + _BaseWaveform.onCursorMove(self, x) + class _WaveformView(QtWidgets.QWidget): + cursorMove = QtCore.pyqtSignal(float) + def __init__(self, parent): QtWidgets.QWidget.__init__(self, parent=parent) self._stopped_x = None self._timescale = 1 + self._cursor_x = 0 layout = QtWidgets.QVBoxLayout() layout.setContentsMargins(0, 0, 0, 0) @@ -376,6 +428,8 @@ class _WaveformView(QtWidgets.QWidget): self._splitter.setHandleWidth(1) scroll_area.setWidget(self._splitter) + self.cursorMove.connect(self.onCursorMove) + def setModel(self, model): self._model = model self._model.dataChanged.connect(self.onDataChange) @@ -421,6 +475,11 @@ class _WaveformView(QtWidgets.QWidget): w = self._splitter.widget(src_start) self._splitter.insertWidget(dest_row, w) + def onCursorMove(self, x): + self._cursor_x = x + for i in range(self._model.rowCount()): + self._splitter.widget(i).onCursorMove(x) + def _create_waveform(self, row): name = self._model.data(self._model.index(row, 0)) ty = self._model.data(self._model.index(row, 1)) @@ -435,6 +494,8 @@ class _WaveformView(QtWidgets.QWidget): w.setXLink(self._ref_vb) w.setStoppedX(self._stopped_x) w.setTimescale(self._timescale) + w.cursorMove.connect(self.cursorMove) + w.onCursorMove(self._cursor_x) action = QtWidgets.QAction("Delete waveform", w) action.triggered.connect(lambda: self._delete_waveform(w)) w.addAction(action) @@ -516,6 +577,45 @@ class _WaveformModel(QtCore.QAbstractTableModel): self.update_data(waveform_data, 0, self.rowCount()) +class _CursorTimeControl(QtWidgets.QLineEdit): + submit = QtCore.pyqtSignal(float) + + def __init__(self, parent): + QtWidgets.QLineEdit.__init__(self, parent=parent) + self._text = "" + self._value = 0 + self._timescale = 1 + self.setDisplayValue(0) + self.textChanged.connect(self._onTextChange) + self.returnPressed.connect(self._onReturnPress) + + def setTimescale(self, timescale): + self._timescale = timescale + + def _onTextChange(self, text): + self._text = text + + def setDisplayValue(self, value): + self._value = value + self._text = pg.siFormat(value * 1e-12 * self._timescale, + suffix="s", + allowUnicode=False, + precision=15) + self.setText(self._text) + + def _setValueFromText(self, text): + try: + self._value = pg.siEval(text) * (1e12 / self._timescale) + except: + logger.error("Error when parsing cursor time input", exc_info=True) + + def _onReturnPress(self): + self._setValueFromText(self._text) + self.setDisplayValue(self._value) + self.submit.emit(self._value) + self.clearFocus() + + class Model(DictSyncTreeSepModel): def __init__(self, init): DictSyncTreeSepModel.__init__(self, "/", ["Channels"], init) @@ -641,6 +741,11 @@ class WaveformDock(QtWidgets.QDockWidget): self._waveform_view.setModel(self._waveform_model) grid.addWidget(self._waveform_view, 1, 0, colspan=12) + self._cursor_control = _CursorTimeControl(self) + self._waveform_view.cursorMove.connect(self._cursor_control.setDisplayValue) + self._cursor_control.submit.connect(self._waveform_view.onCursorMove) + grid.addWidget(self._cursor_control, 0, 3, colspan=6) + def _add_async_action(self, label, coro): action = QtWidgets.QAction(label, self) action.triggered.connect( @@ -674,6 +779,7 @@ class WaveformDock(QtWidgets.QDockWidget): self._waveform_model.update_all(self._waveform_data['data']) self._waveform_view.setStoppedX(self._waveform_data['stopped_x']) self._waveform_view.setTimescale(self._waveform_data['timescale']) + self._cursor_control.setTimescale(self._waveform_data['timescale']) async def load_trace(self): try: