import serial
import time
import collections
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from matplotlib.lines import Line2D
import numpy as np

Samples = 100  #Muestras / Samples
sampleTime = 100  #Tiempo de muestreo / Sample Time
numData = 4

serialPort = 'COM8' # Puerto serial arduino / Arduino serial port
baudRate = 9600  # Baudios

serial.Serial()
def getSerialData(self,Samples,numData,serialConnection, lines):
    while True:
        with serial.Serial('COM8',9600) as ser:
            while True:
                lista =str(ser.readline()).strip("b'")
                lista = lista.strip("\\r\\n'")
                data = lista.split()
                for i in range(numData):
                    data[i].encode("utf-8").decode("unicode-escape").encode("latin1")
                    value = float(data[i])        #array con datos tipo float
                    data[i].append(value) #Guarda lectura en la ultima posicion / #Save reading in the end position
                    lines[i].set_data(range(Samples),data[i]) # Dibujar nueva linea / Drawn new line

#def getSerialData(self,Samples,numData,serialConnection, lines):
    #for i in range(numData):
        #value  = float(serialConnection.readline().strip())  #Leer sensor / Read sensor
        #data[i].append(value) #Guarda lectura en la ultima posicion / #Save reading in the end position
        #lines[i].set_data(range(Samples),data[i]) # Dibujar nueva linea / Drawn new line
        
try:
  serialConnection = serial.Serial(serialPort, baudRate) # Instanciar objeto Serial / Instance Serial Object
except:
  print('Cannot conect to the port')

# Limites de los ejes / Axis limit
xmin = 0
xmax = Samples
ymin = [-100, -100 , -100 , -100]
ymax = [100, 100 , 100 , 100]
lines = []
data = []

for i in range(numData):
    data.append(collections.deque([0] * Samples, maxlen=Samples)) #Lista donde se almacena las variables del sensor
    lines.append(Line2D([], [], color='blue'))                    #Lineas en 2D para graficar los datos
  
fig = plt.figure()# Crea una nueva figura #Create a new figure.
ax1 = fig.add_subplot(2, 2, 1,xlim=(xmin, xmax), ylim=(ymin[0] , ymax[0]))
ax1.title.set_text('Sensor 1 - X')
ax1.set_xlabel("Samples")
ax1.set_ylabel("Degrees")
ax1.add_line(lines[0])

ax2 = fig.add_subplot(2, 2, 2,xlim=(xmin, xmax), ylim=(ymin[1] , ymax[1]))
ax2.title.set_text('Sensor 1 - Y')
ax2.set_xlabel("Samples")
ax2.set_ylabel("Degrees")
ax2.add_line(lines[1])

ax3 = fig.add_subplot(2, 2, 3,xlim=(xmin, xmax), ylim=(ymin[2] , ymax[2]))
ax3.title.set_text('Sensor 2 - X')
ax3.set_xlabel("Samples")
ax3.set_ylabel("Degrees")
ax3.add_line(lines[2])

ax4 = fig.add_subplot(2, 2, 4,xlim=(xmin, xmax), ylim=(ymin[3] , ymax[3]))
ax4.title.set_text('Sensor 2 - Y')
ax4.set_xlabel("Samples")
ax4.set_ylabel("Degrees")
ax4.add_line(lines[3])
    
anim = animation.FuncAnimation(fig,getSerialData, fargs=(Samples,numData,serialConnection,lines), interval=sampleTime)
plt.show()

serialConnection.close() # cerrar puerto serial/ close serial port