ecmcArrayGui.py:Add tool to plot arrays

2020-10-08 13:51:57 +02:00
parent de04f2b686
commit a43f893336
1 changed files with 131 additions and 0 deletions
@@ -0,0 +1,131 @@
+#*************************************************************************
+# Copyright (c) 2020 European Spallation Source ERIC
+# ecmc is distributed subject to a Software License Agreement found
+# in file LICENSE that is included with this distribution. 
+#
+#   ecmcArrayGui.py
+#
+#  Created on: October 8, 2020
+#      Author: Anders Sandström
+#    
+#  Plots two waveforms (y vs time) updates for each callback on the y-pv
+#  
+#*************************************************************************
+
+import sys
+import epics
+from PyQt5.QtWidgets import *
+from PyQt5 import QtWidgets
+from PyQt5.QtCore import *
+from PyQt5.QtGui import *
+import numpy as np
+import matplotlib
+matplotlib.use("Qt5Agg")
+from matplotlib.figure import Figure
+from matplotlib.animation import TimedAnimation
+from matplotlib.lines import Line2D
+from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
+import matplotlib.pyplot as plt 
+import threading
+
+class comSignal(QObject):
+    data_signal = pyqtSignal(object)
+
+class ecmcArrayGui(QtWidgets.QDialog):
+    def __init__(self,yname=None):        
+        super(ecmcArrayGui, self).__init__()        
+        self.comSignalY = comSignal()        
+        self.comSignalY.data_signal.connect(self.callbackFuncY)
+        self.pause = 0
+        self.spectY = None
+        self.figure = plt.figure()                            
+        self.plotted_line = None
+        self.ax = None
+        self.canvas = FigureCanvas(self.figure)   
+        self.toolbar = NavigationToolbar(self.canvas, self) 
+        self.pauseBtn = QPushButton(text = 'pause')
+        self.pauseBtn.setFixedSize(100, 50)
+        self.pauseBtn.clicked.connect(self.pauseBtnAction)        
+        self.pauseBtn.setStyleSheet("background-color: green")
+        self.pvNameY = yname # "IOC_TEST:Plugin-FFT0-Raw-Data-Act"        
+        self.connectPvs() # Epics
+        self.setGeometry(300, 300, 900, 700)
+        self.setWindowTitle("ecmc Array plot: " + self.pvNameY)
+        layout = QVBoxLayout() 
+        layout.addWidget(self.toolbar) 
+        layout.addWidget(self.canvas) 
+        layout.addWidget(self.pauseBtn) 
+        self.setLayout(layout)                
+        return
+
+    def connectPvs(self):        
+        if self.pvNameY is None:            
+            raise RuntimeError("pvname y must not be 'None'")
+        if len(self.pvNameY)==0:
+            raise RuntimeError("pvname  y must not be ''")
+
+        self.pvY = epics.PV(self.pvNameY)        
+        #print('self.pvY: ' + self.pvY.info)
+        
+        self.pvY.add_callback(self.onChangePvY)        
+        QCoreApplication.processEvents()
+        
+    def onChangePvY(self,pvname=None, value=None, char_value=None,timestamp=None, **kw):
+        self.comSignalY.data_signal.emit(value)        
+
+    def pauseBtnAction(self):        
+        self.pause = not self.pause
+        if self.pause:
+            self.pauseBtn.setStyleSheet("background-color: red");
+        else:
+            self.pauseBtn.setStyleSheet("background-color: green");
+            self.comSignalY.data_signal.emit(self.spectY)
+        return
+
+    def callbackFuncY(self, value):
+        if(np.size(value)) > 0:
+            self.spectY = value
+            self.plotSpect()
+        return
+
+    def plotSpect(self):
+        if self.pause:            
+            return
+
+        if self.spectY is None:
+            return
+        
+        # create an axis 
+        if self.ax is None:
+           self.ax = self.figure.add_subplot(111)
+   
+        # plot data 
+        if self.plotted_line is not None:
+            self.plotted_line.remove()
+
+        self.plotted_line, = self.ax.plot(self.spectY, 'b*-') 
+        self.ax.grid(True)
+
+        plt.xlabel('Time []')
+        plt.ylabel(self.pvNameY +' [' + self.pvY.units + ']')        
+        # refresh canvas 
+        self.canvas.draw()
+
+        self.ax.autoscale(enable=True)
+
+def printOutHelp():
+  print("ecmcArrayGui: Plots waveforms data (updates on data callback). ")
+  print("python ecmcArrayGui.py  <y.pv>")
+  print("example: python ecmcArrayGui.py IOC_TEST:Plugin-FFT0-Raw-Data-Act")
+
+if __name__ == "__main__":
+    import sys    
+    if len(sys.argv)!=2:
+        printOutHelp()
+        sys.exit()    
+    yname=sys.argv[1]
+    app = QtWidgets.QApplication(sys.argv)
+    window=ecmcArrayGui(yname=yname)
+    window.show()
+    sys.exit(app.exec_())