Track Throttle Map

import tif1
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
import numpy as np

# Load session
session = tif1.get_session(2023, 'Silverstone', 'Q')
fastest_lap = session.laps.pick_fastest()

# Get telemetry
telemetry = fastest_lap.get_car_data()

# Extract position and throttle
x = telemetry['X']
y = telemetry['Y']
throttle = telemetry['Throttle']

# Prepare segments
points = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.concatenate([points[:-1], points[1:]], axis=1)

# Create plot
fig, ax = plt.subplots(figsize=(12, 10))

# Color-coded by throttle
norm = plt.Normalize(0, 100)
lc = LineCollection(segments, cmap='RdYlGn', norm=norm, linewidth=4)
lc.set_array(throttle)
line = ax.add_collection(lc)

# Colorbar
cbar = plt.colorbar(line, ax=ax, label='Throttle (%)')

# Format
ax.set_aspect('equal')
ax.axis('off')
plt.suptitle(f"{session.event['EventName']} {session.event.year} - Throttle Map")
plt.tight_layout()
plt.show()

import tif1

# Custom colormap, saved directly to a file
fig, ax = tif1.plot_track_throttle_map(
    2023, 'Silverstone', 'Q',
    cmap='viridis',
    save_path='track_throttle_map.png',
    dpi=300,
)
plt.show()

import tif1
import matplotlib.pyplot as plt

# One call: loads the session and draws the throttle-colored track map
fig, ax = tif1.plot_track_throttle_map(2023, 'Silverstone', 'Q')
plt.show()

# Or save straight to a file
# tif1.plot_track_throttle_map(2023, 'Silverstone', 'Q', save_path='track_throttle_map.png')