import fastf1 as ff1
from fastf1 import utils
from fastf1 import plotting
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import rcParams
rcParams['font.family']='serif'

plotting.setup_mpl(color_scheme='None')
ff1.Cache.enable_cache('fastf1cache')

year = 2021
GP = 'Saudi Arabia Grand Prix'
event = 'R'
lap = '37'

race = ff1.get_session(year, GP, event)
laps = race.load_laps(with_telemetry=True)
lap  = laps.loc[laps['LapNumber']==37]
#lap2 = laps.loc[laps['LapNumber']==19]
ver  = lap.pick_driver('VER').get_telemetry()
#ver2 = lap2.pick_driver('VER').get_telemetry()

verv = np.asarray(ver['Speed'])
vert = np.asarray(ver['Time'])
vert = vert/np.timedelta64(1, 's')
verv = verv/3.6

#vert2 = np.asarray(ver2['Time'])
#vert2 = vert2/np.timedelta64(1, 's')
#verv2 = np.asarray(ver2['Speed'])
#verv2 = verv2/3.6

dv = np.diff(verv)
dt = np.diff(vert)
dvdt = dv/dt
dvdt = np.append(0, dvdt)
gs = dvdt/9.8

#dv2 = np.diff(verv2)
#dt2 = np.diff(vert2)
#dvdt2 = dv2/dt2
#dvdt2 = np.append(0, dvdt2)
#gs2 = dvdt2/9.8

plt.plot(vert, gs, linewidth=0.8, label='Lap 37')
#plt.plot(vert2, gs2, linewidth=0.8, label='Lap 19')
plt.legend(fancybox=False,framealpha=0.6,edgecolor='gray')
plt.xlabel('time[s]')
plt.ylabel('acceleration [G]')
plt.grid(which='both', alpha=0.3, linestyle='--')
plt.tight_layout()

plt.savefig('lap19veracceleration.png',dpi=300)