diff --git a/main.py b/main.py index 0a4707c397604e5f218812b518d18f07952a20c2..01b6f8363d359091da49a75f6e011c9d4b29cf95 100644 --- a/main.py +++ b/main.py @@ -20,10 +20,10 @@ class Main: self.listWetWeatherRacesSince(2015) racesToAnalyse = [ - SessionIdentifier(2022, 4, "R"), # Imola 2022 (DWR) - SessionIdentifier(2024, 7, "R"), # Imola 2024 (SWR) - SessionIdentifier(2024, 9, "R"), # Canada 2024 (DWR) - SessionIdentifier(2023, 8, "R") # Canada 2023 (SWR) + SessionIdentifier(2022, "Imola", "R"), # Imola 2022 (DWR) + SessionIdentifier(2024, "Imola", "R"), # Imola 2024 (SWR) + SessionIdentifier(2024, "Montreal", "R"), # Canada 2024 (DWR) + SessionIdentifier(2023, "Montreal", "R") # Canada 2023 (SWR) ] self.overtakeAnalysis(racesToAnalyse) @@ -44,7 +44,7 @@ class Main: # Import raceSession = dataHandler.importer.importSession(raceSessionIdentifier) - raceName: str = f"{raceSession.event['EventName']} {raceSession.event.year}" + grandPrixName: str = f"{raceSession.event['EventName']} {raceSession.event.year}" print("Import done") @@ -59,12 +59,13 @@ class Main: # Plot - dataHandler.plotter.plotOvertakesWithTireChangeWindow(overtakesInRaces, earliestTireChange, latestTireChange, raceName) + dataHandler.plotter.plotOvertakesWithTireChangeWindow(overtakesInRaces, earliestTireChange, latestTireChange, grandPrixName) print("Plot done") # Print data - print(f"\n\n===== Data for race {raceSessionIdentifier.event}/{raceSessionIdentifier.year} =====") + print(f"\n\n===== Data for {grandPrixName} =====") + print("Lap\tOvertakes") currentLap = 0 for overtakesInLap in overtakesInRaces: