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
from matplotlib.animation import FuncAnimation
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)
# convert from m/s^2 to G 
#gs = dvdt/9.8

dv2 = np.diff(verv2)
dt2 = np.diff(vert2)
dvdt2 = dv2/dt2
dvdt2 = np.append(0, dvdt2)
# convert from m/s^2 to G 
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('lap19hamacceleration.png',dpi=300)