#!/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.

To change EPICS EO, WO use. e.g.
caput -a ISTTOK:central:ATCAIOP1-WO 14 0.191 0.174 -0.036 -0.044 0.183 0.126 0.020 0.140 -0.461 -0.572 0.022 -0.262 0.475 0.353

"""
import numpy as np

import pyqtgraph as pg
# from pyqtgraph.Qt import QtCore
import argparse

app = pg.mkQApp("Plotting MARTe2 AtcaIop Data")
# mw = QtWidgets.QMainWindow()
# mw.resize(800,800)

# MAX_SAMPLES = 50000
# ADC_CHANNELS = 14  # channels stored in ISTTOK
MAG_OFFSET = 6  # channels stored in ISTTOK
ADC_DECIM_RATE = 200

parser = argparse.ArgumentParser(description=
        """Script to plot AtcaIop MDSplus data and calc drifts
        To change EPICS EO, WO use. e.g.
        caput -a ISTTOK:central:ATCAIOP1-WO 14 0.191 0.174 -0.036 -0.044 0.183 0.126 0.020 0.140 -0.461 -0.572 0.022 -0.262 0.475 0.353""")

# 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])
# 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('-m', '--maxpoints', type=int,
                    help='Max points to plot', default=1000)
parser.add_argument('-e', '--averages', action='store_true',
                    help='Calc averages')
parser.add_argument('-d', '--decimated', action='store_true',
                    help='Use decimated data')
# parser.add_argument('-w', '--drift', action='store_true', help='Calc drifts')
parser.add_argument('-z', '--zero', action='store_true',
                    help='Zero integral Lines')
parser.add_argument('-f', '--file', type=str,
                        help='filename device to open', default='LangmuirSdas')

args = parser.parse_args()


win = pg.GraphicsLayoutWidget(show=True, title="Basic plotting examples")
win.resize(1000, 600)
win.setWindowTitle('ISTTOK Plotting')

# Enable antialiasing for prettier plots
pg.setConfigOptions(antialias=True)

p1 = win.addPlot(title="SDAS Languir Probe data")
# add plt.addLegend() BEFORE you create the curves.
# mdsNode = tree.getNode("ATCAIOP1.ADC0RAW")
# dataAdc = mdsNode.getData().data()
# timeData = mdsNode.getDimensionAt(0).data()
p1.addLegend()
start = args.crange[0] - 1
stop = args.crange[1]

dataSdas = np.load(args.file)
x = dataSdas[:args.maxpoints, 0] / 1e3  # us -> ms

# for i in range(1, 2):

lineName = ['Time', 'Trigger',
            'ElectricTop', 'ElectricInner', 'ElectricOuter', 'ElectricBottom']
for i in [2, 3, 4, 5]:
    y = dataSdas[:args.maxpoints, i]
    # p1.plot(y, pen=pg.mkPen(i, width=2), name=f"Ch {i}")
    p1.plot(y, pen=pg.mkPen(i, width=2), name=lineName[i])

p1.showGrid(x=True, y=True, alpha=0.5)
# p1.setLabel('bottom', "Time", units='ms')
# p1.setLabel('bottom', "Samp", units='')

win.nextRow()

p2 = win.addPlot(title="MARTE2 output Langmuir data")
dataOut = np.genfromtxt('../Startup/IsttokOutput.csv', delimiter=',')
p2.addLegend()
x = dataOut[:args.maxpoints, 0] / 1e3  # us -> ms
#y = dataOut[:args.maxpoints, 1]
# for i in range(1, 5):
lineName = ['Time', 'OutLang0', 'OutLang1', 'OutLangR', 'OutLangZ']
for i in [1, 2, 3, 4,]:
    y = dataOut[:args.maxpoints, i]
    p2.plot(y, pen=pg.mkPen(i, width=2), name=lineName[i])

p2.showGrid(x=True, y=True, alpha=0.5)

# p2.setLabel('bottom', "Time", units='ms')
# print("WO: ", end='')
# updatePlot()

win.nextRow()
p3 = win.addPlot(title="SDAS Mag Probe data")
lineName = ['Time', 'Trigger',
            'ElectricTop', 'ElectricInner', 'ElectricOuter', 'ElectricBottom']
for i in [0, 1, 2, 3]:
    y = dataSdas[:args.maxpoints, i + MAG_OFFSET]
    p3.plot(y, pen=pg.mkPen(i, width=2), name=lineName[i])

p3.showGrid(x=True, y=True, alpha=0.5)

win.nextRow()

p4 = win.addPlot(title="MARTE2 output Magnetic data")
# dataOut = np.genfromtxt('../Startup/IsttokOutput.csv', delimiter=',')
p4.addLegend()
x = dataOut[:args.maxpoints, 0] / 1e3  # us -> ms
lineName = ['Time', 'OutLang0', 'OutLang1', 'OutLangR', 'OutLangZ',
            'MagOutR', 'MagOutZ']
for i in [5]:
    y = dataOut[:args.maxpoints, i]
    p4.plot(y, pen=pg.mkPen(i, width=2), name=lineName[i])

p4.showGrid(x=True, y=True, alpha=0.5)

p4.setLabel('bottom', "Sample")  # , units='ms')

if __name__ == '__main__':
    # import sys
    # if sys.flags.interactive != 1 or not hasattr(QtCore, 'PYQT_VERSION'):
    #    pg.QtGui.QApplication.exec_()

    # iprint("xwc")
    pg.exec()
# vim: syntax=python ts=4 sw=4 sts=4 sr et