import numpy as np
import matplotlib.pyplot as plt
##

fichier = r'C:\Users\bapt-\Desktop\BOULOT\PCSI_22\Mecanique\2_Dynamique\Simulation\traction2.txt'

f = open(fichier,'r')
message = f.read()
lst = message.split("\n")
theta = []
force_sur_capteur = []

for i in range(0, len(lst)-1, 2): 
    theta.append(int(lst[i+1])/180*np.pi)
    force_sur_capteur.append(int(lst[i]))

theta = np.array(theta)
force_sur_capteur = np.array(force_sur_capteur)
force_sur_capteur = (force_sur_capteur/np.max(force_sur_capteur))*0.1*9.81
tension_cable = 0.1*9.81 -force_sur_capteur


d = 7.5
R = 1.8
delta = 1
l0 = 9
epsilon = -0.1

l = np.sqrt((d-R*np.cos(theta))**2 + (delta + R*np.sin(theta))**2  ) + epsilon

plt.close('all')
plt.figure(figsize = [8, 8])
plt.plot(l0-l, tension_cable, 'rs', markeredgecolor = 'k')
plt.xlabel('élongation (cm)', fontsize = 20)
plt.ylabel("tension de l'élastique (N)",  fontsize = 20)
plt.xlim([0, 3])
# plt.plot([0, 2], [0, 0.8], 'k') 
# plt.plot([1.5, 1.5], [0, 1], 'k--')
# plt.text(0.5, 0.7, "Zone \n d'élasticité \n linéaire", fontsize = 20)
plt.show()