Added CAInput Datasource

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

Analysis/getMdsData.py

@@ -6,20 +6,11 @@ import sys
 import numpy as np
 import matplotlib.pyplot as plt
 from MDSplus import Tree
+import argparse
+
 
 mdsTreeName = 'rtappisttok'
 
-if len(sys.argv) > 1:
-    mdsPulseNumber = int(sys.argv[1])
-else:
-    mdsPulseNumber = 1
-    #treename = ''
-
-try:
-    tree = Tree(mdsTreeName, mdsPulseNumber)
-except:
-    print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}')
-    exit()
 """
 Available Channels
 
@@ -30,6 +21,7 @@ ATCAIOP1.ADC3RAW
 ATCAIOP1.ADC4RAW
 ATCAIOP1.ADC5RAW
 ATCAIOP1.ADC6RAW
+#for i in range(args.irange[0], args.irange[1]):
 ATCAIOP1.ADC7RAW
 
 ATCAIOP1.ADC0INT
@@ -38,6 +30,23 @@ ATCAIOP1.ADC1INT
 """
 
 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, 8])
+    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 ([1, ...])', default=100)
+
+    args = parser.parse_args()
+    mdsPulseNumber = args.shot
+    try:
+        #tree = Tree(mdsTreeName, args.pulse)
+        tree = Tree(mdsTreeName, mdsPulseNumber)
+    except:
+        print(f'Failed opening {mdsTreeName} for pulse number {mdsPulseNumber:d}')
+        exit()
     mdsNode = tree.getNode("ATCAIOP1.ADC0RAW")
     mdsNode = tree.getNode("ATCAIOP1.ADC1INT")
     dataAdc = mdsNode.getData().data()