Known/Unknown Safe/Unsafe Scenarios

"""Track SOTIF scenario population across four quadrants."""
from dataclasses import dataclass, field
from enum import Enum
from typing import List

class ScenarioStatus(Enum):
    KNOWN_SAFE   = "Q1_known_safe"
    KNOWN_UNSAFE = "Q2_known_unsafe"
    UNKNOWN_SAFE = "Q3_unknown_safe"    # assumption, not tested
    UNKNOWN_UNSAFE = "Q4_unknown_unsafe"  # not yet discovered

@dataclass
class SOTIFScenario:
    id: str
    description: str
    triggering_condition: str
    status: ScenarioStatus
    evidence: str = ""
    design_measure: str = ""

@dataclass
class ScenarioDatabase:
    feature: str
    scenarios: List[SOTIFScenario] = field(default_factory=list)

    def coverage_q1_pct(self) -> float:
        """Percentage of known-safe scenarios (SOTIF coverage KPI)."""
        known = [s for s in self.scenarios
                 if s.status in (ScenarioStatus.KNOWN_SAFE,
                                 ScenarioStatus.KNOWN_UNSAFE)]
        q1 = [s for s in self.scenarios
              if s.status == ScenarioStatus.KNOWN_SAFE]
        if not known: return 0.0
        return len(q1) / len(known) * 100

    def summary(self):
        counts = {s: 0 for s in ScenarioStatus}
        for sc in self.scenarios:
            counts[sc.status] += 1
        for status, count in counts.items():
            print(f"  {status.value}: {count}")
        print(f"  Q1 coverage: {self.coverage_q1_pct():.1f}%")