class FatigueRecoveryModel:
    """
    基于 EASA 受控休息理念的疲劳恢复模型
    
    参考：
    - EASA CAT.OP.MPA.210: 30min 睡眠 + 20min 恢复
    - NASA Rosekind: 小睡限制疲劳下降
    """
    
    def __init__(self):
        # 疲劳等级与恢复建议
        self.recovery_protocol = {
            'mild': {
                'perclos_range': (15, 25),
                'recommended_rest_min': 15,
                'recovery_check_sec': 300,  # 5min 后复检
                'action': '短暂休息或开窗/调整',
            },
            'moderate': {
                'perclos_range': (25, 35),
                'recommended_rest_min': 20,
                'recovery_check_sec': 600,
                'action': '建议 15-20min 小睡',
            },
            'severe': {
                'perclos_range': (35, 100),
                'recommended_rest_min': 30,
                'recovery_check_sec': 1200,
                'action': '强制停车+30min 小睡',
            },
        }
    
    def recommend_recovery(self, perclos: float, 
                          microsleep_count: int = 0) -> Dict:
        """
        根据疲劳程度推荐恢复方案
        
        Args:
            perclos: 当前 PERCLOS 值 (%)
            microsleep_count: 过去 10 分钟微睡眠次数
            
        Returns:
            恢复建议字典
        """
        if microsleep_count >= 3:
            level = 'severe'
        elif perclos >= 35:
            level = 'severe'
        elif perclos >= 25:
            level = 'moderate'
        elif perclos >= 15:
            level = 'mild'
        else:
            return {'level': 'normal', 'action': '继续驾驶'}
        
        protocol = self.recovery_protocol[level]
        return {
            'level': level,
            **protocol,
            'current_perclos': perclos,
            'microsleep_count': microsleep_count,
        }


# 测试
if __name__ == "__main__":
    model = FatigueRecoveryModel()
    
    test_cases = [
        (10, 0, "正常"),
        (20, 0, "轻度"),
        (30, 1, "中度"),
        (45, 3, "严重"),
    ]
    
    for perclos, microsleep, label in test_cases:
        result = model.recommend_recovery(perclos, microsleep)
        print(f"\n{label} (PERCLOS={perclos}%, 微睡眠={microsleep}次):")
        print(f"  等级: {result['level']}")
        print(f"  建议休息: {result.get('recommended_rest_min', 0)} 分钟")
        print(f"  动作: {result.get('action', 'N/A')}")