Corrected WO setting by epics caput

Signed-off-by: Bernardo Carvalho <bernardo.carvalho@tecnico.ulisboa.pt>
2024-04-29 18:32:00 +01:00
parent 7dfc58fd15
commit d62dc8a082
6 changed files with 280 additions and 185 deletions
--- a/Analysis/calcDriftMds.py
+++ b/Analysis/calcDriftMds.py
@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+"""
+This script plots the MARTe2 ATCAIop samples stored in MDSplus
+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 numpy as np
+
+from MDSplus import Tree
+import argparse
+
+ADC_CHANNELS = 14 # channels stored in ISTTOK
+DECIM_RATE = 200
+
+def main(args):
+    mdsPulseNumber = args.shot
+
+    mdsTreeName = 'rtappisttok'
+    try:
+        tree = Tree(mdsTreeName, mdsPulseNumber)
+    except:
+        print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}')
+        exit()
+
+# add plt.addLegend() BEFORE you create the curves.
+#mdsNode = tree.getNode("ATCAIOP1.ADC0RAW")
+#dataAdc = mdsNode.getData().data()
+#timeData = mdsNode.getDimensionAt(0).data()
+    start = args.crange[0] -1; stop = args.crange[1]
+#for i in range(args.crange[0], args.crange[1]):
+    meanD = np.zeros(ADC_CHANNELS, dtype=int)
+    driftW = np.zeros(ADC_CHANNELS)
+    total_samples = 0
+    for i in range(ADC_CHANNELS):
+        mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}RAW")
+        dataAdc = mdsNode.getData().data()
+        meanD[i] = np.mean(dataAdc[:,0]).astype(int)
+        mdsNode = tree.getNode(f"ATCAIOP1.ADC{i}INT")
+        dataAdcInt = mdsNode.getData().data()
+        total_samples = DECIM_RATE * len(dataAdcInt[:, 0])
+        driftW[i] =  (dataAdcInt[-1, 0] - dataAdcInt[0, 0]) /total_samples
+
+   # if(args.averages):
+    print(f"EO: {ADC_CHANNELS} ", end='')
+    for i in range(ADC_CHANNELS):
+        print(f"{meanD[i]:d} ", end='')
+    print(" ")
+    print(f"WO: {ADC_CHANNELS} ", end='')
+    for i in range(ADC_CHANNELS):
+       print(f"{driftW[i]:0.3f} ", end='')
+    print(" ")
+    print(f"Samples {total_samples}, time {total_samples/2e3:.2f} ms")
+
+
+if __name__ == '__main__':
+    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 plots (1 12)',default=[1, 12])
+    parser.add_argument('-i', '--irange', nargs='+',type=int,default=[1, 12])
+    parser.add_argument('-s', '--shot', type=int, help='Mds+ pulse Number ([1, ...])', default=100)
+    #parser.add_argument('-e', '--averages', action='store_true', help='Calc averages')
+#parser.add_argument('-w', '--drift', action='store_true', help='Calc drifts')
+#, default='')
+
+    args = parser.parse_args()
+    main(args)
+# vim: set syntax=python ts=4 sw=4 sts=4 sr et