holy agent

hpcsim/adapter.py

@@ -0,0 +1,103 @@
+from __future__ import annotations
+
+from typing import Dict, Any
+
+
+def normalize_telemetry(payload: Dict[str, Any]) -> Dict[str, Any]:
+    """Normalize Pi/FastF1-like telemetry payload to Enricher expected schema.
+
+    Accepted aliases:
+    - speed: Speed
+    - throttle: Throttle
+    - brake: Brake, Brakes
+    - tire_compound: Compound, TyreCompound, Tire
+    - fuel_level: Fuel, FuelRel, FuelLevel
+    - ers: ERS, ERSCharge
+    - track_temp: TrackTemp
+    - rain_probability: RainProb, PrecipProb
+    - lap: Lap, LapNumber
+    Values are clamped and defaulted if missing.
+    """
+    aliases = {
+        "lap": ["lap", "Lap", "LapNumber"],
+        "speed": ["speed", "Speed"],
+        "throttle": ["throttle", "Throttle"],
+        "brake": ["brake", "Brake", "Brakes"],
+        "tire_compound": ["tire_compound", "Compound", "TyreCompound", "Tire"],
+        "fuel_level": ["fuel_level", "Fuel", "FuelRel", "FuelLevel"],
+        "ers": ["ers", "ERS", "ERSCharge"],
+        "track_temp": ["track_temp", "TrackTemp"],
+        "rain_probability": ["rain_probability", "RainProb", "PrecipProb"],
+    }
+
+    out: Dict[str, Any] = {}
+
+    def pick(key: str, default=None):
+        for k in aliases.get(key, [key]):
+            if k in payload and payload[k] is not None:
+                return payload[k]
+        return default
+
+    def clamp01(x, default=0.0):
+        try:
+            v = float(x)
+        except (TypeError, ValueError):
+            return default
+        return max(0.0, min(1.0, v))
+
+    # Map values with sensible defaults
+    lap = pick("lap", 0)
+    try:
+        lap = int(lap)
+    except (TypeError, ValueError):
+        lap = 0
+
+    speed = pick("speed", 0.0)
+    try:
+        speed = float(speed)
+    except (TypeError, ValueError):
+        speed = 0.0
+
+    throttle = clamp01(pick("throttle", 0.0), 0.0)
+    brake = clamp01(pick("brake", 0.0), 0.0)
+
+    tire_compound = pick("tire_compound", "medium")
+    if isinstance(tire_compound, str):
+        tire_compound = tire_compound.lower()
+    else:
+        tire_compound = "medium"
+
+    fuel_level = clamp01(pick("fuel_level", 0.5), 0.5)
+
+    ers = pick("ers", None)
+    if ers is not None:
+        ers = clamp01(ers, None)
+
+    track_temp = pick("track_temp", None)
+    try:
+        track_temp = float(track_temp) if track_temp is not None else None
+    except (TypeError, ValueError):
+        track_temp = None
+
+    rain_prob = pick("rain_probability", None)
+    try:
+        rain_prob = clamp01(rain_prob, None) if rain_prob is not None else None
+    except Exception:
+        rain_prob = None
+
+    out.update({
+        "lap": lap,
+        "speed": speed,
+        "throttle": throttle,
+        "brake": brake,
+        "tire_compound": tire_compound,
+        "fuel_level": fuel_level,
+    })
+    if ers is not None:
+        out["ers"] = ers
+    if track_temp is not None:
+        out["track_temp"] = track_temp
+    if rain_prob is not None:
+        out["rain_probability"] = rain_prob
+
+    return out