import matplotlib.pyplot as plt
from statistics import mean
import importation

x, y, t = importation.Latispro('fichier.txt')
N=len(x)
deltat=t[1]-t[0]

Vx=[0]*N
Vy=[0]*N
ax=[0]*N
ay=[0]*N
a=[0]*N

plt.subplot(121)

for i in range (0,N-1):
    Vx[i]=(x[i+1]-x[i])/deltat
    Vy[i]=(y[i+1]-y[i])/deltat
    plt.quiver(x[i],y[i],Vx[i],Vy[i], scale=15,color="b",angles="xy",units="xy")

for i in range (0,N-2):
    ax[i]=(Vx[i+1]-Vx[i])/deltat
    ay[i]=(Vy[i+1]-Vy[i])/deltat
    a[i]=(ax[i]*ax[i]+ay[i]*ay[i])**0.5
    plt.quiver(x[i],y[i],ax[i],ay[i], scale=75,color="r",angles="xy",units="xy")
    plt.quiver(x[i],y[i],0,-9.8, scale=75,color="b",angles="xy",units="xy")

plt.plot(x,y,"b+") # tracé de la courbe c = f(t) (bleu, marque +, points non reliés)
plt.xlabel("x (m)") # légende abscisses
plt.ylabel("y (m)") # légende ordonnées
plt.axis("equal") # repère orthonormé
plt.title("Trajectoire y=f(x)") # titre

plt.subplot(122)
plt.xlabel("t (s)") # légende abscisses
plt.ylabel("a (m/s²)") # légende ordonnées
plt.plot(t,a,"r+")

plt.show() # affichage des courbes

del a[N-2:N]
print(a)
amoy=mean(a)
print(amoy)