import matplotlib.pyplot as plt
import numpy as np

# Lecture des donnees du fichier txt
t, x, y = np.loadtxt('donnees_tyrolienne.txt', unpack=True, usecols=(0, 1, 2), delimiter = '\t', skiprows = 2)
yprime = 1.41*y
vx = np.diff(x)/np.diff(t)
vy = np.diff(y)/np.diff(t)
v = np.sqrt(vx**2+vy**2)
m = 75
g = 9.81
Ec = 0.5*m*v**2
Epp = m*g*yprime[:-1]
Em = Ec+Epp

# Creation du graphique
plt.axis('equal')
plt.title('Trajectoire de la balle')
plt.xlabel('Abscisse x (m)')
plt.ylabel('Altitude y (m)')
plt.plot(x, y, color = 'blue', label = 'Trajectoire', marker = '+')
plt.legend()
plt.show()

# Creation du graphique des energies
plt.title('Trajectoire de la balle')
plt.xlabel('Temps t (s)')
plt.ylabel('Energie (J)')
plt.plot(t[:-1], Ec, color = 'blue', label = 'Energie cinetique', marker = '+')
plt.plot(t[:-1], Epp, color = 'red', label = 'Energie potentielle de pesanteur', marker = '+')
plt.plot(t[:-1], Em, color = 'green', label = 'Energie mecanique', marker = '+')
plt.legend()
plt.show()

# Creation du graphique
plt.title('Trajectoire de la balle')
plt.xlabel('Temps t (s)')
plt.ylabel('Vitesse v (m/s)')
plt.plot(t[:-1], v, color = 'orange', label = 'Vitesse', marker = '+')
plt.legend()
plt.show()