# getcdata.py - fetch the calibrated data from the AS7265x module
# this program is 3-clause BSD license
import serial as ser
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import time
#global variables
uartpath = '/dev/ttyUSB0'
uartbaud = 115200
uarttout = 5

wl = [410, 435, 460, 485, 510, 535, 560, 585, 610, 645, 680, 705, 730, 760, 810, 860, 900, 940]
responseorder = [i for i in 'RSTUVWGHIJKLABCDEF'] # works, do NOT touch!
realorder = [i for i in 'ABCDEFGHRISJTUVWKL']


print('getcdata')
print('This utility is part of the TeraHz project')

wavelens = pd.Series(realorder)


plt.ion()
win = plt.figure()
spectrum=win.add_subplot(111)


with ser.Serial(uartpath, uartbaud, timeout=uarttout) as sensor:
    while True:
        sensor.write(b'ATCDATA\n')
        rawresp = sensor.readline().decode()
        # parses, calculates and saves the data
        response = pd.Series([float(i)/35.0 for i in rawresp[:-3].split(',')], index=responseorder)
        data = pd.DataFrame(response, index=realorder, columns = ['uW/cm^2']) # puts data into a DataFrame
        data.insert(0, 'wavelenght', wl) #inserts a legend
        print(data)
        spectrum.cla()
        spectrum.plot(data['wavelenght'], data['uW/cm^2'])
        spectrum.set_xlabel('Valovna dolžina')
        spectrum.set_ylabel('uW/cm2')
        win.canvas.draw()

        time.sleep(0.1)