Added copy coordinates and messages to pyqtgraph

Author: marcosfrenkel

plottr/plot/pyqtgraph/autoplot.py

@@ -271,6 +271,16 @@ def __init__(self, parent: Optional[PlotWidgetContainer]) -> None:
         layout.setContentsMargins(0, 0, 0, 0)
         layout.setSpacing(0)
 
+        self.statusLabel = QtWidgets.QLabel("Click on plot to copy coordinates")
+        self.statusLabel.setAlignment(QtCore.Qt.AlignCenter)
+        self.statusLabel.setStyleSheet("color: gray;")
+
+        self.clipboardMessageActive = False
+        self.hintMessage = "Click on plot to copy coordinates"
+        self.clipboardResetTimer = QtCore.QTimer(self)
+        self.clipboardResetTimer.setSingleShot(True)
+        self.clipboardResetTimer.timeout.connect(self.clearClipboardMessage)
+
         self.setLayout(layout)
         self.setMinimumSize(*getcfg('main', 'pyqtgraph', 'minimum_plot_size',
                                     default=(400, 400)))
@@ -327,6 +337,11 @@ def _plotData(self, **kwargs: Any) -> None:
             self.figConfig.optionsChanged.connect(self._refreshPlot)
             self.figConfig.figCopied.connect(self.onfigCopied)
             self.figConfig.figSaved.connect(self.onfigSaved)
+            self.layout().addWidget(self.statusLabel)
+
+        if kwargs.get('clearWidget', True):
+            for subplot in self.fmWidget.subPlots:
+                self._connectPlotEvents(subplot)
 
         if self.data.has_meta('title'):
             self.fmWidget.setTitle(self.data.meta_val('title'))
@@ -356,6 +371,45 @@ def _plotData(self, **kwargs: Any) -> None:
     def _refreshPlot(self) -> None:
         self._plotData()
 
+    def showClipboardMessage(self, message: str) -> None:
+        self.clipboardMessageActive = True
+        self.statusLabel.setText(message)
+        self.clipboardResetTimer.start(1500)
+
+    def clearClipboardMessage(self) -> None:
+        self.clipboardMessageActive = False
+        self.statusLabel.setText(self.hintMessage)
+
+    def setStatusMessage(self, message: str) -> None:
+        if not self.clipboardMessageActive:
+            self.statusLabel.setText(message if message else self.hintMessage)
+
+    def _connectPlotEvents(self, subplot: PlotBase) -> None:
+        subplot.plot.scene().sigMouseMoved.connect(
+            lambda pos, sp=subplot: self._onMouseMoved(sp, pos)
+        )
+        subplot.plot.scene().sigMouseClicked.connect(
+            lambda event, sp=subplot: self._onPlotClicked(sp, event)
+        )
+
+    def _onMouseMoved(self, subplot: PlotBase, pos: Any) -> None:
+        vb = subplot.plot.getViewBox()
+        local = vb.mapFromScene(pos)
+        if vb.boundingRect().contains(local):
+            pt = vb.mapSceneToView(pos)
+            self.setStatusMessage(f'({pt.x():.8g}, {pt.y():.8g})')
+        else:
+            self.setStatusMessage('')
+
+    def _onPlotClicked(self, subplot: PlotBase, event: Any) -> None:
+        vb = subplot.plot.getViewBox()
+        local = vb.mapFromScene(event.scenePos())
+        if vb.boundingRect().contains(local):
+            pt = vb.mapSceneToView(event.scenePos())
+            coord_str = '({:.8g}, {:.8g})'.format(pt.x(), pt.y())
+            QtWidgets.QApplication.clipboard().setText(coord_str)
+            self.showClipboardMessage(f"Copied {coord_str} to clipboard")
+
     @Slot()
     def onfigCopied(self) -> None:
         """
@@ -366,6 +420,7 @@ def onfigCopied(self) -> None:
         screenshot = self.fmWidget.grab(rectangle=QtCore.QRect(QtCore.QPoint(0, 0), QtCore.QSize(-1, -1)))
         clipboard = QtWidgets.QApplication.clipboard()
         clipboard.setImage(screenshot.toImage())
+        self.showClipboardMessage("Figure copied to clipboard")
 
     @Slot()
     def onfigSaved(self) -> None:
@@ -449,7 +504,6 @@ def __init__(self, options: FigureOptions,
         self.copyFig = self.addAction('Copy Figure', self._copyFig)
         self.saveFig = self.addAction('Save Figure', self._saveFig)
 
-
     def _setOption(self, option: str, value: Any) -> None:
         setattr(self.options, option, value)
         self.optionsChanged.emit()