bernardo/MARTe2-isttok
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed to decimated integrals

Browse Source 
Signed-off-by: Bernardo Carvalho <bernardo.carvalho@tecnico.ulisboa.pt>
This commit is contained in:
Bernardo Brotas de Carvalho
2024-05-09 12:02:23 +00:00
parent 48d4a0ebb8
commit 01d94b88a5
1 changed files with 40 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
Analysis/pyqtPlotMds.py
View File

@@ -7,31 +7,34 @@ the left/right mouse buttons. Right click on any plot to show a context menu.
import numpy as np import numpy as np
import pyqtgraph as pg import pyqtgraph as pg
from pyqtgraph.Qt import QtCore # from pyqtgraph.Qt import QtCore
from MDSplus import Tree from MDSplus import Tree
import argparse import argparse
app = pg.mkQApp("Plotting MARTe2 AtcaIop Data") app = pg.mkQApp("Plotting MARTe2 AtcaIop Data")
#mw = QtWidgets.QMainWindow() # mw = QtWidgets.QMainWindow()
#mw.resize(800,800) # mw.resize(800,800)
#MAX_SAMPLES = 50000 # MAX_SAMPLES = 50000
ADC_CHANNELS = 14 # channels stored in ISTTOK ADC_CHANNELS = 14 # channels stored in ISTTOK
DECIM_RATE = 200 DECIM_RATE = 200
parser = argparse.ArgumentParser(description = 'Script to support the QA activities') 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='+', help='<Required> Set flag', required=True)
#parser.add_argument('-l','--list', nargs='+') # parser.add_argument('-l','--list', nargs='+')
parser.add_argument('-c', '--crange', nargs='+',type=int, help='Channel plots (1 12)',default=[1, 12]) parser.add_argument('-c', '--crange', nargs='+', type=int,
parser.add_argument('-i', '--irange', nargs='+',type=int,default=[1, 12]) help='Channel plots (1 12)', default=[1, 12])
#parser.add_argument('pulse','-', nargs='+', help='<Required> Set flag', required=True) parser.add_argument('-i', '--irange', nargs='+', type=int, default=[1, 12])
# arser.add_argument('pulse','-', nargs='+', help='<Required> Set flag', required=True)
parser.add_argument('-s', '--shot', type=int, help='Mds+ pulse Number ([1, ...])', default=100) parser.add_argument('-s', '--shot', type=int, help='Mds+ pulse Number ([1, ...])', default=100)
parser.add_argument('-m', '--maxpoints', type=int, help='Max points to plot', default=50000) parser.add_argument('-m', '--maxpoints', type=int, help='Max points to plot', default=50000)
parser.add_argument('-e', '--averages', action='store_true', help='Calc averages') parser.add_argument('-e', '--averages', action='store_true',
#parser.add_argument('-w', '--drift', action='store_true', help='Calc drifts') help='Calc averages')
#, default='') # parser.add_argument('-w', '--drift', action='store_true', help='Calc drifts')
parser.add_argument('-z', '--zero', action='store_true',help='Zero integral Lines') #, default='') parser.add_argument('-z', '--zero', action='store_true',
help='Zero integral Lines')
args = parser.parse_args() args = parser.parse_args()
mdsPulseNumber = args.shot mdsPulseNumber = args.shot
@@ -40,12 +43,12 @@ mdsTreeName = 'rtappisttok'
try: try:
tree = Tree(mdsTreeName, mdsPulseNumber) tree = Tree(mdsTreeName, mdsPulseNumber)
except: except Exception:
print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}') print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}')
exit() exit()
win = pg.GraphicsLayoutWidget(show=True, title="Basic plotting examples") win = pg.GraphicsLayoutWidget(show=True, title="Basic plotting examples")
win.resize(1000,600) win.resize(1000, 600)
win.setWindowTitle('pyqtgraph example: Plotting') win.setWindowTitle('pyqtgraph example: Plotting')
# Enable antialiasing for prettier plots # Enable antialiasing for prettier plots
@@ -53,12 +56,12 @@ pg.setConfigOptions(antialias=True)
p1 = win.addPlot(title="ATCA-IOP ADC raw decimated") p1 = win.addPlot(title="ATCA-IOP ADC raw decimated")
# add plt.addLegend() BEFORE you create the curves. # add plt.addLegend() BEFORE you create the curves.
#mdsNode = tree.getNode("ATCAIOP1.ADC0RAW") # mdsNode = tree.getNode("ATCAIOP1.ADC0RAW")
#dataAdc = mdsNode.getData().data() # dataAdc = mdsNode.getData().data()
#timeData = mdsNode.getDimensionAt(0).data() # timeData = mdsNode.getDimensionAt(0).data()
p1.addLegend() p1.addLegend()
start = args.crange[0] -1; stop = args.crange[1] start = args.crange[0] -1; stop = args.crange[1]
#for i in range(args.crange[0], args.crange[1]): # for i in range(args.crange[0], args.crange[1]):
meanD = np.zeros(ADC_CHANNELS, dtype=int) meanD = np.zeros(ADC_CHANNELS, dtype=int)
driftW = np.zeros(ADC_CHANNELS) driftW = np.zeros(ADC_CHANNELS)
total_samples = 0 total_samples = 0
@@ -69,9 +72,9 @@ for i in range(ADC_CHANNELS):
mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INT") mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INT")
dataAdcInt = mdsNode.getData().data() dataAdcInt = mdsNode.getData().data()
total_samples = DECIM_RATE * len(dataAdcInt[:, 0]) total_samples = DECIM_RATE * len(dataAdcInt[:, 0])
driftW[i] = (dataAdcInt[-1, 0] - dataAdcInt[0, 0]) /total_samples driftW[i] = (dataAdcInt[-1, 0] - dataAdcInt[0, 0]) / total_samples
if(args.averages): if (args.averages):
print(f"EO: {ADC_CHANNELS} ", end='') print(f"EO: {ADC_CHANNELS} ", end='')
for i in range(ADC_CHANNELS): for i in range(ADC_CHANNELS):
print(f"{meanD[i]:d} ", end='') print(f"{meanD[i]:d} ", end='')
@@ -89,26 +92,26 @@ for i in range(start, stop):
timeData = mdsNode.getDimensionAt(0).data() timeData = mdsNode.getDimensionAt(0).data()
y = dataAdc[ :args.maxpoints, 0] y = dataAdc[ :args.maxpoints, 0]
x = DECIM_RATE * np.arange(len(y)) / 2.0e6 x = DECIM_RATE * np.arange(len(y)) / 2.0e6
p1.plot(x,y, pen=pg.mkPen(i, width=2), name=f"Ch {i+1}") p1.plot(x, y, pen=pg.mkPen(i, width=2), name=f"Ch {i+1}")
#p1.setLabel('bottom', "Y Axis", units='s') # p1.setLabel('bottom', "Y Axis", units='s')
win.nextRow() win.nextRow()
p4 = win.addPlot(title="Channel Integrals") p4 = win.addPlot(title="Channel Integrals")
p4.addLegend() p4.addLegend()
#for i in range(8,12): # for i in range(8,12):
start = args.irange[0] -1; stop = args.irange[1] start = args.irange[0] -1; stop = args.irange[1]
#print("WO: ", end='') # print("WO: ", end='')
for i in range(start, stop): for i in range(start, stop):
#mdsNode = tree.getNode(f"ATCAIOP1.ADC8INT") # mdsNode = tree.getNode(f"ATCAIOP1.ADC8INT")
mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INT") mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INTD")
try: try:
dataAdcInt = mdsNode.getData().data() dataAdcInt = mdsNode.getData().data()
timeData = mdsNode.getDimensionAt(0).data() timeData = mdsNode.getDimensionAt(0).data()
total_samples = DECIM_RATE * len(dataAdcInt[:, 0]) total_samples = DECIM_RATE * len(dataAdcInt[:, 0])
y = dataAdcInt[ :args.maxpoints, 0] / 2.0e6 # LSB * sec y = dataAdcInt[ :args.maxpoints, 0] / 2.0e6 # LSB * sec
if(args.zero): if (args.zero):
y = y - dataAdcInt[0, 0] / 2.0e6 # LSB * sec y = y - dataAdcInt[0, 0] / 2.0e6 # LSB * sec
#wo = (dataAdcInt[-1, 0] - dataAdcInt[0, 0]) /total_samples # wo = (dataAdcInt[-1, 0] - dataAdcInt[0, 0]) /total_samples
# print(f"{wo:0.4f} ", end='') # print(f"{wo:0.4f} ", end='')
x = DECIM_RATE * np.arange(len(y)) / 2.0e6 # in sec x = DECIM_RATE * np.arange(len(y)) / 2.0e6 # in sec
p4.plot(x,y, pen=pg.mkPen(i, width=2), name=f"Ch {i+1}") p4.plot(x,y, pen=pg.mkPen(i, width=2), name=f"Ch {i+1}")
@@ -119,13 +122,13 @@ print(" ")
p4.setLabel('bottom', "Time", units='s') p4.setLabel('bottom', "Time", units='s')
#updatePlot() # updatePlot()
if __name__ == '__main__': if __name__ == '__main__':
import sys # import sys
if sys.flags.interactive != 1 or not hasattr(QtCore, 'PYQT_VERSION'): # if sys.flags.interactive != 1 or not hasattr(QtCore, 'PYQT_VERSION'):
pg.QtGui.QApplication.exec_() # pg.QtGui.QApplication.exec_()
#iprint("xwc") # iprint("xwc")
#pg.exec() pg.exec()
# vim: syntax=python ts=4 sw=4 sts=4 sr et # vim: syntax=python ts=4 sw=4 sts=4 sr et