Running file read

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

Analysis/SdasImport2File.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Python3 App to Import data from ISTTOK SDAS to MDSPlus
+author:  B. Carvalho / IPFN-IST
+email: bernardo.carvalho@tecnico.ulisboa.pt
+
+Install sdas:
+http://metis.ipfn.tecnico.ulisboa.pt/CODAC/IPFN_Software/SDAS/Access/Python
+"""
+
+import numpy as np
+import argparse
+# import matplotlib.pyplot as plt
+from MdsImportSdas import StartSdas, LoadSdasData
+ISTTOK_RT_PERIOD = 0.0001
+
+def build_import_table_langmuir():
+    SDAS_NODEFMT = 'MARTE_NODE_IVO3.DataCollection.Channel_{}'
+    LANGMUIR_OFF = 24
+    NUM_PROBES = 4
+    """
+    ADC_electric_top_near: MARTE_NODE_IVO3.DataCollection.Channel_024
+    ADC_electric_outer_near: MARTE_NODE_IVO3.DataCollection.Channel_025
+    ADC_electric_bottom_near: MARTE_NODE_IVO3.DataCollection.Channel_026
+    ADC_electric_inner_near: MARTE_NODE_IVO3.DataCollection.Channel_027
+    """
+    table = []
+    for n in range(NUM_PROBES):
+        sdas_str = SDAS_NODEFMT.format(str(LANGMUIR_OFF + n).zfill(3))
+        nd = {'sdas': sdas_str, 'name': f'Langmuir Probe no. {n+1}'}
+        table.append(nd)
+    return table
+
+"""
+def plot_signals(pulse, nodeTable):
+    try:
+        tree = Tree(MDSTREENAME, pulse)
+    except mdsExceptions.TreeNOPATH:
+        print(f'Failed opening {MDSTREENAME}')
+    except mdsExceptions.TreeFOPENR:
+        print(f'Failed opening {MDSTREENAME} for pulse number {pulse:d}')
+        exit()
+    NoOfColors = len(nodeTable)
+    cm = plt.get_cmap('jet')
+    fig = plt.figure(figsize=(5, 7), tight_layout=True)
+    ax = fig.add_subplot(111)
+    ax.set_prop_cycle(color=[cm(1.*i/NoOfColors) for i in range(NoOfColors)])
+    plt.title(f'#{pulse}: Mirnov coil signals')
+
+    for nd in nodeTable:
+        nd_mds = tree.getNode(nd['mds'])
+        mdsData = nd_mds.getData()
+        signal = mdsData.data()
+        if len(signal) > 0:
+            times = mdsData.dims[0]
+            plt.plot(times, signal, label=nd['name'])
+
+    plt.xlabel('Time / s ')
+    plt.legend()
+    plt.grid(True)
+    plt.show()
+"""
+
+
+def get_arguments():
+    parser = argparse.ArgumentParser(description='ISTTOK ')
+    parser.add_argument('-p', '--pulse',
+                        help='pulse (shot) number', default='46241', type=int)
+    # parser.add_argument('-s', '--shot',
+    #                    type=int, help='Mds+ pulse Number ([1, ...])',
+    #                  default=100)
+    parser.add_argument('-f', '--file', type=str,
+                        help='filename device to save', default='LangmuirSdas')
+    parser.add_argument('-e', '--exportData',
+                        action='store_true', help='Export to MDSPlus')
+    parser.add_argument('-m', '--mirnov',
+                        action='store_true', help='Import Mirnov')
+    parser.add_argument('-l', '--langmuir',
+                        action='store_true', help='Import Langmuir')
+    parser.add_argument('-t', '--ploT',
+                        action='store_true', help='Plot Signals')
+    parser.add_argument('-n', '--names',
+                        action='store_true', help='Print Node Table')
+    return parser.parse_args()
+
+
+if (__name__ == "__main__"):
+    args = get_arguments()
+    pulseNo = args.pulse
+
+    nodeTable = build_import_table_langmuir()
+    client = StartSdas()
+    langmuirData = []
+    for nd in nodeTable:
+        print(nd['sdas'])
+        data, tzero_us, period = LoadSdasData(client, nd['sdas'], pulseNo)
+        langmuirData.append(data)
+    time = np.arange(len(data), dtype='uint32') * int(period)
+    langmuirNp = np.array(langmuirData).T
+    data2file = np.insert(langmuirNp, 0, time,  axis=1)
+    filename = f"{args.file:s}_{pulseNo}.csv"
+    head = ('#Time (uint32)[1],Langmuir0 (float32)[1],Langmuir1 (float32)[1],'
+            'Langmuir2 (float32)[1],Langmuir3 (float32)[1]')
+    # formt = ['%d', '%.6f', '%.6f', '%.6f', '%.6f']
+    formt = '%d,%.6f,%.6f,%.6f,%.6f'
+#    formt = '%d,{%.6f,%.6f,%.6f,%.6f}'
+#    head = '#TimeSdas (uint32)[1],LangmuirSignals (float32)[4]'
+    np.savetxt(filename, data2file, fmt=formt,
+               header=head, comments='')  # , delimiter=',')
+
+#    plot_signals(46241, table)