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()

verd = np.asarray(ver['Distance'])
verv = np.asarray(ver['Speed'])
vert = np.asarray(ver['Time'])
# converting from kilometers per hour to meters per second
verv = verv/3.6
vert = vert/np.timedelta64(1, 's')

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

plt.plot(vert, verv, linewidth=0.8, label='Lap 37')
plt.plot(vert2, verv2, linewidth=0.8, label='Lap 19')
plt.legend(fancybox=False,framealpha=0.6,edgecolor='gray')
plt.xlabel('time [s]')
plt.ylabel('speed [m/s]')
plt.grid(which='both', alpha=0.3, linestyle='--')
plt.tight_layout()
plt.savefig('lap37ver.png',dpi=300)