ACC_Data_2/data_collection.py at master · ThomasBisset/ACC_Data_2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import datetime
from enums import car_class
from read_shared_memory import read_graphics, read_physics, read_static


def data_collection():
    data = {
        # Basic Info
        "Track": read_static()["track"].lower(),
        "Car": read_static()["carModel"].lower(),
        "CarClass": car_class(read_static()["carModel"]).lower(),
        "Session": read_graphics()["ACC_SESSION_TYPE"],
        "SessionStatus": read_graphics()["ACC_STATUS"],
        "Flag": read_graphics()["flag"],
        "Penalty": read_graphics()["penalty"],
        "InPit": read_graphics()["isInPit"],
        "InPitLane": read_graphics()["isInPitLane"],
        # Laps and Times
        "CurrentSector": read_graphics()["currentSectorIndex"],
        "LapsCompleted": read_graphics()["completedLaps"],
        "BestLapTime": read_graphics()["iBestTime"],
        "LastLapTime": read_graphics()["iLastTime"],
        "LastSplitTime": read_graphics()["iSplit"],
        "SessionTimeLeft": read_graphics()["sessionTimeLeft"],
        "TrackPosition": read_graphics()["normalizedCarPosition"],
        # Weather Info
        "Clock": read_graphics()["Clock"],                             # unit: seconds from midnight
        "AmbientTemperature": round(read_physics()["airTemp"], 3),                         # unit: c
        "TrackTemperature": round(read_physics()["roadTemp"], 3),                          # unit: c
        "GripStatus": read_graphics()["trackGripStatus"],
        "RainLevel": read_graphics()["rainIntensity"],
        "WindSpeed": round(read_graphics()["windSpeed"], 3),                             # unit: m/s
        "WindDirection": round(read_graphics()["windDirection"], 3),                 # unit: radians
        # Tyre Info
        "RainTyres": read_graphics()["rainTyres"],
        "TyreTemperatureFrontLeft": round(read_physics()["tyreCoreTemperature"][0], 3),    # unit: c
        "TyreTemperatureFrontRight": round(read_physics()["tyreCoreTemperature"][1], 3),   # unit: c
        "TyreTemperatureRearLeft": round(read_physics()["tyreCoreTemperature"][2], 3),     # unit: c
        "TyreTemperatureRearRight": round(read_physics()["tyreCoreTemperature"][3], 3),    # unit: c
        "TyrePressureFrontLeft": round(read_physics()["wheelPressure"][0], 3),           # unit: psi
        "TyrePressureFrontRight": round(read_physics()["wheelPressure"][1], 3),          # unit: psi
        "TyrePressureRearLeft": round(read_physics()["wheelPressure"][2], 3),            # unit: psi
        "TyrePressureRearRight": round(read_physics()["wheelPressure"][3], 3),           # unit: psi
        # Electronics Settings Info
        "TractionControl": read_graphics()["TC"],
        "TractionControlCut": read_graphics()["TCCut"],
        "EngineMap": read_graphics()["EngineMap"],
        # Brakes Data
        "BrakeTemperatureFrontLeft": round(read_physics()["brakeTemp"][0], 3),       # unit: celsius
        "BrakeTemperatureFrontRight": round(read_physics()["brakeTemp"][1], 3),      # unit: celsius
        "BrakeTemperatureRearLeft": round(read_physics()["brakeTemp"][2], 3),        # unit: celsius
        "BrakeTemperatureRearRight": round(read_physics()["brakeTemp"][3], 3),       # unit: celsius
        "BrakeBalance": round(read_physics()["brakeBias"], 3),
        "ABS": read_graphics()["ABS"],
        "FrontBrakeCompound": read_physics()["frontBrakeCompound"],
        "RearBrakeCompound": read_physics()["rearBrakeCompound"],
        # Fuel Data
        "CurrentFuel": round(read_physics()["fuel"], 3),                              # unit: litres
        "UsedFuel": round(read_graphics()["usedFuel"], 3),                            # unit: litres
        "EstimatedFuelLaps": round(read_graphics()["fuelEstimatedLaps"], 3),
    }
    return data