import matplotlib.pyplot as plt

def trouveVitesse() :
    print("Tu sautes en chute libre à l'instant t = 0 ")
    n = input("Entre le temps en dixièmes de secondes pour lequel tu veux connaitre la vitesse : ")
    n = int(n)
    print("Tu veux connaitre la vitesse du parachutiste au bout de",n,"dixièmes de secondes ")
    v = 0
    X=[]
    Y=[]
    for i in range(1,n+1) :
        v = -0.25/800*v**2 + v + 0.981
        print(f"Après {i/10} secondes: vitesse {round(v*3.6,2)} km/h")
        X.append(i/10)
        Y.append(v*3.6)
    return X,Y


def trace(X,Y,titre="",texteOrdonnee="") :
    plt.plot (X, Y, linestyle='-', color='b', label=texteOrdonnee)
    # Ajouter des labels et un titre
    plt.xlabel("Temps (s)")
    plt.ylabel(texteOrdonnee)
    plt.title(titre)
    # Afficher la courbe
    plt.grid(True)
    plt.show()


# Main
X,Y = trouveVitesse()
trace(X,Y,texteOrdonnee = "vitesse (km/h)")