#%% importamos las librerias necesarias
import matplotlib.pyplot as plt #libreria para graficar
import numpy as np #libreria para hacer cuentas

#%% definimos variables para ir guardando valores
tiempos=[] # definimos lista para los tiempos
posiciones=[] # definimos lista para las posiciones
velocidades=[]  # definimos lista para las velocidades

dt=0.01 #definimos el paso del tiempo
numero_pasos=200 #definimos la cantidad total de pasos
g=-10 #defino la gravedad
t0=0 #definimos el tiempo inicial
x0=0 #definimos la posicion inicial
v0=10 #definimos la velocidad inicial

#%% generamos los valores de tiempos, posiciones y velocidades de un MRUV
t=t0
v=v0
for i in range(0, numero_pasos):
    t=t+dt #avanzamos el tiempo en cada paso
    x=x0+v0*t+g/2*t**2 #tomamos donde estaba el paso anterior
    v=v0+g*t
    
    tiempos.append(t) #guardamos el tiempo nuevo en la lista de tiempos
    posiciones.append(x) #guardamos la posicion nueva en la lista
    velocidades.append(v) #guardamos la velocidad nueva en la lista
    
    print(str(i)+' de '+str(numero_pasos)) #hacemos que imprima el valor de i en pantalla

#%% graficamos
#Grafico la posición vs tiempo
plt.figure() #abrimos una nueva figura para graficar
plt.plot(tiempos, posiciones) #hago el grafico
plt.xlabel('tiempo (s)') #nombre del eje X
plt.ylabel('posicion (m)') #nombre del eje Y
plt.show() #muestra la figura

#Grafico la posición vs tiempo
plt.figure() #abrimos una nueva figura para graficar
plt.plot(tiempos, velocidades) #hago el grafico
plt.xlabel('tiempo (s)') #nombre del eje X
plt.ylabel('velocidad (m)') #nombre del eje Y
plt.show() #muestra la figura