Initial import

Signed-off-by: Bernardo Carvalho <bernardo.carvalho@tecnico.ulisboa.pt>

Analysis/pyqtPlotMds.py

@@ -0,0 +1,99 @@
+"""
+This example demonstrates many of the 2D plotting capabilities
+in pyqtgraph. All of the plots may be panned/scaled by dragging with 
+the left/right mouse buttons. Right click on any plot to show a context menu.
+"""
+
+#import sys
+import numpy as np
+
+import pyqtgraph as pg
+from pyqtgraph.Qt import QtCore
+from MDSplus import Tree
+import argparse
+
+app = pg.mkQApp("Plotting MARTe2 Data")
+#mw = QtWidgets.QMainWindow()
+#mw.resize(800,800)
+
+MAX_SAMPLES = 50000
+ADC_CHANNELS = 4
+DECIM_RATE = 200
+
+parser = argparse.ArgumentParser(description = 'Script to support the QA activities')
+
+    #parser.add_argument('-l','--list', nargs='+', help='<Required> Set flag', required=True)
+    #parser.add_argument('-l','--list', nargs='+')
+parser.add_argument('-c', '--crange', nargs='+',type=int, help='Channel lines (0 4 )',default=[0, 4])
+parser.add_argument('-i', '--irange', nargs='+',type=int,default=[0, 4])
+#parser.add_argument('pulse','-', nargs='+', help='<Required> Set flag', required=True)
+parser.add_argument('-s', '--shot', type=int, help='Mds+ pulse Number ([0, ...])', default=100)
+parser.add_argument('-m', '--maxsamples', type=int, help='Max samples to plot', default=50000)
+parser.add_argument('-z', '--zero', action='store_true',help='Zero integral Lines') #, default='')
+
+args = parser.parse_args()
+mdsPulseNumber = args.shot
+
+#if len(sys.argv) > 1:
+#    mdsPulseNumber = int(sys.argv[1])
+#else:
+#    mdsPulseNumber = 1
+#    #treename = ''
+
+mdsTreeName = 'rtappisttok'
+
+try:
+    tree = Tree(mdsTreeName, mdsPulseNumber)
+except:
+    print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}')
+    exit()
+
+#time = dataCsv['#Time (uint32)[1]']
+#timeRel = time - time[0]
+#x = DECIM_RATE * np.arange(len(vals))
+
+win = pg.GraphicsLayoutWidget(show=True, title="Basic plotting examples")
+win.resize(1000,600)
+win.setWindowTitle('pyqtgraph example: Plotting')
+
+# Enable antialiasing for prettier plots
+pg.setConfigOptions(antialias=True)
+
+p1 = win.addPlot(title="ATCA ADC raw decim")
+# add plt.addLegend() BEFORE you create the curves.
+mdsNode = tree.getNode("ATCAIOP1.ADC0RAW")
+dataAdc = mdsNode.getData().data()
+timeData = mdsNode.getDimensionAt(0).data()
+p1.addLegend()
+for i in range(args.crange[0], args.crange[1]):
+    mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}RAW")
+    dataAdc = mdsNode.getData().data()
+    timeData = mdsNode.getDimensionAt(0).data()
+    #y = dataAdc[ :args.maxsamples, 0]
+    y = dataAdc[ :MAX_SAMPLES, 0]
+    x = DECIM_RATE * np.arange(len(y)) / 2.0e6
+    p1.plot(x,y, pen=pg.mkPen(i, width=2), name=f"Ch {i}")
+#p1.setLabel('bottom', "Y Axis", units='s')
+
+win.nextRow()
+p4 = win.addPlot(title="ATCA Integral Channels")
+p4.addLegend()
+#for i in range(8,12):
+for i in range(args.irange[0], args.irange[1]):
+    mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INT")
+    dataAdcInt = mdsNode.getData().data()
+    timeData = mdsNode.getDimensionAt(0).data()
+    y = dataAdcInt[ :args.maxsamples, 0]
+    if(args.zero):
+        y = y - dataAdcInt[0, 0]
+    #x = DECIM_RATE * np.arange(len(y))
+    x = DECIM_RATE * np.arange(len(y)) / 2.0e6
+    p4.plot(x,y, pen=pg.mkPen(i, width=2), name=f"Ch {i}")
+
+p4.setLabel('bottom', "Time", units='s')
+
+#updatePlot()
+
+if __name__ == '__main__':
+    pg.exec()
+# vim: syntax=python ts=4 sw=4 sts=4 sr et