import matplotlib.pyplot as plt
import numpy as np

# Lecture des donnees du fichier txt
t, x, y = np.loadtxt('donnees_basket.txt', unpack=True, usecols=(0, 1, 2), delimiter = '\t', skiprows = 2)

t0 = np.min(t)
t = t - t0
x0 = np.min(x)
x = x - x0

g = 9.81
h = y[0]

v0 = 1 # Valeur de la vitesse initiale du lancer pour modéliser la courbe
alpha = 10 # Angle de tir à faire varier pour modéliser la courbe

xth = v0*np.cos(alpha*np.pi/180)*t
yth = -0.5*g*t**2+v0*np.sin(alpha*np.pi/180)*t+h

print(yth[0], y[0], t[0])
# Creation du second graphique
plt.title('Trajectoire du ballon')
plt.xlabel('Temps t (s)')
plt.ylabel('Abscisse x et altitude y (m)')
plt.scatter(t, x, color = 'orange', label = 'Abscisse', marker = '+')
plt.scatter(t, y, color = 'red', label = 'Altitude', marker = '+')
plt.plot(t, xth, color = 'orange', label = 'Courbe théorique de x(t)')
plt.plot(t, yth, color = 'red', label = 'Courbe théorique de z(t)')
plt.legend()
plt.show()

# Creation du premier graphique
plt.axis('equal')
plt.title('Trajectoire du ballon')
plt.xlabel('Abscisse x (m)')
plt.ylabel('Altitude y (m)')
plt.scatter(x, y, color = 'blue', label = 'Trajectoire', marker = '+')
plt.plot(xth, yth, color = 'blue', label = 'Trajectoire théorique')
plt.legend()
plt.show()