from datetime import datetime
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import serial
import numpy as np
import math


SER = serial.Serial('COM14')

CHANNELS = [
    'ax',
    'ay',
    'az',
    'gx',
    'gy',
    'gz',
    'temp',
    'hum',
    'pres',
    'gas',
    'lux',
]
CHANNEL_IDX = {name: i for (i, name) in enumerate(CHANNELS)}

SAMPLE_COUNT = 500


def main():
    fig = plt.figure()
    
    # subfigs = fig.subfigures(3, math.ceil(len(CHANNELS) / 3))
    
    elem_data = {}
    
    
    for chan in CHANNELS:
        idx = CHANNEL_IDX[chan]
        # subfig = subfigs[idx % 3][idx // 3]
        
        ax = fig.add_subplot(3, math.ceil(len(CHANNELS) / 3), idx + 1)
        ax.set_title(chan)
        
        xs = np.arange(SAMPLE_COUNT),
        ys = np.zeros(SAMPLE_COUNT),
        
        line = ax.plot(xs, ys)[0]
        
        elem_data[chan] = {
            'ys': ys,
            # 'fig': subfig,
            'ax': ax,
            'line': line
        }
    
    elem_data['temp']['ax'].set_ylim([15, 40])
    elem_data['hum']['ax'].set_ylim([0, 100])
    elem_data['pres']['ax'].set_ylim([0, 100000])
    
    def animate(i):
        now = datetime.now()
        
        if not SER.in_waiting:
            return []
            
        line = SER.readline().decode('utf-8')
        
        for elem in line.split(','):
            elem = elem.strip()

            if len(elem) == 0:
                continue
                
            split = elem.split(':')

            if len(split) != 2:
                continue

            k = split[0].strip()
            v = float(split[1].strip())
            
            if k in elem_data:
                dat = elem_data[k]
                
                dat['ys'] = np.roll(dat['ys'], -1)
                dat['ys'][-1] = v
                dat['line'].set_ydata(dat['ys'])
        
        yield from (item['line'] for item in elem_data.values())
        
        
    ani = animation.FuncAnimation(fig, animate, interval=50, blit=True)
    plt.show()
    


if __name__ == '__main__':
    main()