"""
AR-HUD + DMS联动自适应策略引擎

根据DMS输出的驾驶员状态，动态调整AR-HUD显示参数

核心维度：
1. 疲劳等级（0-100）
2. 分心等级（0-100）
3. 视线落点（road/instrument/phone/away）
4. 光照条件（day/night/tunnel）

调整参数：
1. AR内容亮度（0-100%）
2. 警示颜色（绿→黄→红）
3. 信息密度（精简/标准/详细）
4. 警告位置（固定/跟随视线/中央抢占）
5. 声音提示（静音/提示音/语音）
"""

from dataclasses import dataclass
from enum import Enum
from typing import Tuple

class FatigueLevel(Enum):
    AWAKE = 0      # 清醒
    MILD = 1       # 轻度疲劳
    MODERATE = 2   # 中度疲劳
    SEVERE = 3     # 严重疲劳

class DistractLevel(Enum):
    FOCUSED = 0    # 专注
    MILD = 1       # 轻度分心
    MODERATE = 2   # 中度分心
    SEVERE = 3     # 严重分心

class GazeZone(Enum):
    ROAD = "road"
    INSTRUMENT = "instrument"
    PHONE = "phone"
    AWAY = "away"

class LightCondition(Enum):
    DAY = "day"
    NIGHT = "night"
    TUNNEL = "tunnel"

@dataclass
class DriverState:
    fatigue: FatigueLevel
    distract: DistractLevel
    gaze: GazeZone
    light: LightCondition
    perclos: float  # 0-100
    gaze_off_road_s: float  # 视线离开道路秒数

@dataclass
class ARHUDConfig:
    brightness_pct: float        # AR内容亮度
    warning_color: str            # green/yellow/orange/red
    info_density: str             # minimal/standard/detailed
    nav_position: str             # lane_center/shifted_center/center
    fcw_intensity: str            # subtle/standard/prominent/urgent
    speed_display: str            # corner/center/large
    audio_alert: str              # silent/chime/voice
    lane_highlight: bool          # 车道线高亮
    eye_contact_prompt: bool      # "eyes on road"提示

class ARHUDAdapter:
    """AR-HUD自适应策略引擎"""
    
    def compute_config(self, state: DriverState) -> ARHUDConfig:
        """根据驾驶员状态计算AR-HUD配置"""
        config = ARHUDConfig(
            brightness_pct=70,
            warning_color="green",
            info_density="standard",
            nav_position="lane_center",
            fcw_intensity="standard",
            speed_display="corner",
            audio_alert="silent",
            lane_highlight=False,
            eye_contact_prompt=False,
        )
        
        # === 疲劳自适应 ===
        if state.fatigue == FatigueLevel.MILD:
            config.brightness_pct = 80  # 略增亮度
            config.warning_color = "yellow"
            config.info_density = "minimal"  # 减少认知负荷
            config.lane_highlight = True    # 增强车道引导
        elif state.fatigue == FatigueLevel.MODERATE:
            config.brightness_pct = 90
            config.warning_color = "orange"
            config.info_density = "minimal"
            config.lane_highlight = True
            config.audio_alert = "chime"     # 声音提示
            config.speed_display = "center"  # 限速移至中央
        elif state.fatigue == FatigueLevel.SEVERE:
            config.brightness_pct = 100
            config.warning_color = "red"
            config.info_density = "minimal"
            config.audio_alert = "voice"     # 语音唤醒
            config.eye_contact_prompt = True # "请注视道路"
            config.fcw_intensity = "urgent"
        
        # === 分心自适应 ===
        if state.distract == DistractLevel.MILD:
            config.fcw_intensity = "prominent"  # 增强前车警告
        elif state.distract == DistractLevel.MODERATE:
            config.nav_position = "shifted_center"  # 导航偏中央
            config.fcw_intensity = "prominent"
            config.lane_highlight = True
        elif state.distract == DistractLevel.SEVERE:
            config.nav_position = "center"  # 完全抢占中央
            config.fcw_intensity = "urgent"
            config.audio_alert = "chime"
            config.eye_contact_prompt = True
        
        # === 视线落点自适应 ===
        if state.gaze == GazeZone.PHONE:
            # 视线在手机上→全屏抢占
            config.nav_position = "center"
            config.brightness_pct = 100
            config.warning_color = "red"
            config.audio_alert = "voice"
            config.eye_contact_prompt = True
        elif state.gaze == GazeZone.AWAY:
            # 视线完全离开→紧急
            config.brightness_pct = 100
            config.warning_color = "red"
            config.fcw_intensity = "urgent"
            config.audio_alert = "voice"
        
        # === 光照自适应 ===
        if state.light == LightCondition.NIGHT:
            config.brightness_pct *= 0.6  # 夜间降低亮度避免眩光
        elif state.light == LightCondition.TUNNEL:
            config.brightness_pct *= 0.8
        
        # 确保范围
        config.brightness_pct = max(20, min(100, config.brightness_pct))
        
        return config


# ==================== 测试场景 ====================
if __name__ == "__main__":
    adapter = ARHUDAdapter()
    
    scenarios = [
        ("正常白天驾驶", DriverState(
            FatigueLevel.AWAKE, DistractLevel.FOCUSED,
            GazeZone.ROAD, LightCondition.DAY, 5, 0
        )),
        ("轻度疲劳+夜间", DriverState(
            FatigueLevel.MILD, DistractLevel.FOCUSED,
            GazeZone.ROAD, LightCondition.NIGHT, 15, 0
        )),
        ("中度疲劳+分心", DriverState(
            FatigueLevel.MODERATE, DistractLevel.MILD,
            GazeZone.INSTRUMENT, LightCondition.DAY, 25, 2
        )),
        ("严重疲劳", DriverState(
            FatigueLevel.SEVERE, DistractLevel.FOCUSED,
            GazeZone.ROAD, LightCondition.NIGHT, 40, 0
        )),
        ("看手机", DriverState(
            FatigueLevel.AWAKE, DistractLevel.SEVERE,
            GazeZone.PHONE, LightCondition.DAY, 8, 5
        )),
        ("隧道中分心", DriverState(
            FatigueLevel.MILD, DistractLevel.MODERATE,
            GazeZone.AWAY, LightCondition.TUNNEL, 12, 3
        )),
    ]
    
    print("=" * 90)
    print("AR-HUD + DMS联动 自适应策略测试")
    print("=" * 90)
    
    for name, state in scenarios:
        cfg = adapter.compute_config(state)
        print(f"\n场景: {name}")
        print(f"  状态: 疲劳={state.fatigue.name}, 分心={state.distract.name}, "
              f"视线={state.gaze.value}, 光照={state.light.value}")
        print(f"  --- AR-HUD配置 ---")
        print(f"  亮度: {cfg.brightness_pct:.0f}%")
        print(f"  警示色: {cfg.warning_color}")
        print(f"  信息密度: {cfg.info_density}")
        print(f"  导航位置: {cfg.nav_position}")
        print(f"  FCW强度: {cfg.fcw_intensity}")
        print(f"  速度显示: {cfg.speed_display}")
        print(f"  声音: {cfg.audio_alert}")
        print(f"  车道线高亮: {'是' if cfg.lane_highlight else '否'}")
        print(f"  眼睛提示: {'是' if cfg.eye_contact_prompt else '否'}")