import numpy as np

class AviationInspiredFatigue:
    """
    航空启发的多时间尺度疲劳模型
    
    时间尺度:
    1. 即时疲劳 (秒级): 微睡眠、眼皮下沉
    2. 短期疲劳 (分钟级): PERCLOS累积
    3. 中期疲劳 (小时级): 驾驶时长、HRV变化
    4. 长期疲劳 (天级): 睡眠债务、昼夜节律
    
    航空经验: 80%的严重疲劳由长期积累导致
    """
    
    def __init__(self):
        self.instant_fatigue = 0.0    # 0-1
        self.short_term = 0.0         # 0-1
        self.medium_term = 0.0        # 0-1
        self.long_term = 0.0          # 0-1
    
    def update_instant(self, eye_openness: float, perclos: float):
        """秒级更新"""
        # 微睡眠检测
        if eye_openness < 0.2:
            self.instant_fatigue = min(1.0, self.instant_fatigue + 0.3)
        else:
            self.instant_fatigue *= 0.95  # 衰减
    
    def update_short_term(self, perclos_60s: float):
        """分钟级更新"""
        # PERCLOS 60秒窗口
        self.short_term = perclos_60s / 30.0  # 30%为严重
    
    def update_medium_term(self, drive_minutes: float, hrv_rmssd: float):
        """小时级更新"""
        # 驾驶时长 + HRV
        time_factor = min(1.0, drive_minutes / 240)  # 4小时饱和
        hrv_factor = max(0, (30 - hrv_rmssd) / 30)    # RMSSD<30为疲劳
        self.medium_term = 0.5 * time_factor + 0.5 * hrv_factor
    
    def update_long_term(self, sleep_debt_hours: float, circadian_phase: float):
        """天级更新"""
        # 睡眠债务 + 昼夜节律
        debt_factor = min(1.0, sleep_debt_hours / 4.0)  # 4小时债务饱和
        # circadian_phase: 0=最佳(下午), 1=最差(凌晨3-5点)
        self.long_term = 0.6 * debt_factor + 0.4 * circadian_phase
    
    def get_total_fatigue(self) -> dict:
        """综合疲劳评估"""
        # 加权融合（航空经验权重）
        total = (
            0.15 * self.instant_fatigue +
            0.25 * self.short_term +
            0.30 * self.medium_term +
            0.30 * self.long_term
        )
        
        level = 'normal' if total < 0.3 else \
                'mild' if total < 0.5 else \
                'moderate' if total < 0.7 else 'severe'
        
        return {
            'total': total,
            'level': level,
            'breakdown': {
                'instant': self.instant_fatigue,
                'short_term': self.short_term,
                'medium_term': self.medium_term,
                'long_term': self.long_term
            }
        }


# 测试
if __name__ == "__main__":
    fm = AviationInspiredFatigue()
    
    # 模拟驾驶4小时后的状态
    fm.update_instant(eye_openness=0.15, perclos=0.25)  # 眼皮下沉
    fm.update_short_term(perclos_60s=22.0)              # PERCLOS 22%
    fm.update_medium_term(drive_minutes=240, hrv_rmssd=18)  # 4h, HRV下降
    fm.update_long_term(sleep_debt_hours=3.0, circadian_phase=0.8)  # 债务+凌晨
    
    result = fm.get_total_fatigue()
    print("=== 多时间尺度疲劳评估 ===")
    print(f"综合: {result['total']:.2f} ({result['level']})")
    print(f"分解:")
    for scale, val in result['breakdown'].items():
        print(f"  {scale:12s}: {val:.2f}")