Function-to-ECU Allocation

"""Function-to-ECU allocation matrix with constraint checking."""
from dataclasses import dataclass, field
from typing import List, Dict

@dataclass
class ECU:
    ecu_id:     str
    name:       str
    cpu_mips:   int
    rom_kb:     int
    ram_kb:     int
    max_asil:   str  # maximum ASIL this ECU can host

@dataclass
class Function:
    func_id:    str
    name:       str
    asil:       str
    cpu_mips:   int   # required MIPS
    rom_kb:     int
    ram_kb:     int
    allocated_ecu: str = ""

ASIL_ORDER = {"QM": 0, "ASIL-A": 1, "ASIL-B": 2, "ASIL-C": 3, "ASIL-D": 4}

def check_asil_compatibility(func: Function, ecu: ECU) -> bool:
    return ASIL_ORDER[func.asil] <= ASIL_ORDER[ecu.max_asil]

def check_resource_fit(functions: List[Function], ecu: ECU) -> dict:
    allocated = [f for f in functions if f.allocated_ecu == ecu.ecu_id]
    used_cpu = sum(f.cpu_mips for f in allocated)
    used_rom = sum(f.rom_kb   for f in allocated)
    used_ram = sum(f.ram_kb   for f in allocated)
    return {
        "cpu_ok": used_cpu <= ecu.cpu_mips * 0.70,  # 70% max load
        "rom_ok": used_rom <= ecu.rom_kb  * 0.80,
        "ram_ok": used_ram <= ecu.ram_kb  * 0.80,
        "cpu_used_pct": used_cpu / ecu.cpu_mips * 100,
    }