# Chalmers/Volvo 研究结果数据化
recline_study = {
    '25_deg_upright': {
        'brain_injury_HIC': 100,  # 基准
        'abdominal_compression': 100,  # 基准
        'chest_compression': 100,  # 基准
        'lap_belt_contact': '正常',
        'submarining_risk': '低',
    },
    '43_deg_semi_recline': {
        'brain_injury_HIC': 145,  # +45%
        'abdominal_compression': 220,  # +120%
        'chest_compression': 75,  # -25%
        'lap_belt_contact': '早期丧失',
        'submarining_risk': '中等',
    },
    '60_deg_full_recline': {
        'brain_injury_HIC': 195,  # +95% — 接近 US 碰撞评级阈值
        'abdominal_compression': 350,  # +250%
        'chest_compression': 50,  # -50%
        'lap_belt_contact': '立即丧失',
        'submarining_risk': '极高',
    }
}

print("座椅 recline 角度与损伤风险关系:")
print(f"{'角度':<8} {'HIC':>8} {'腹部':>8} {'胸部':>8} {'submarining':>12}")
for angle, data in recline_study.items():
    print(f"{angle:<8} {data['brain_injury_HIC']:>8} {data['abdominal_compression']:>8} {data['chest_compression']:>8} {data['submaring_risk']:>12}")

class ReclineSafetyLogic:
    """
    自动驾驶 recline 安全逻辑
    
    OOP 检测 → 姿态分类 → 约束策略选择
    """
    
    def __init__(self):
        self.safety_strategies = {
            'upright_25deg': {
                'restraint_mode': 'standard',
                'belt_pretension': 'normal',
                'airbag_deploy': 'full',
                'risk_level': 'low'
            },
            'semi_recline_43deg': {
                'restraint_mode': 'enhanced',
                'belt_pretension': 'aggressive',  # 提前拉紧
                'anti_submarining_ramp': 'active',
                'lumbar_airbag': 'deploy',
                'airbag_deploy': 'full',
                'risk_level': 'medium'
            },
            'full_recline_60deg': {
                'restraint_mode': 'maximum',
                'belt_pretension': 'max',
                'anti_submarining_ramp': 'max',
                'lumbar_airbag': 'deploy',
                'seat_return': 'auto_return_to_25deg',  # 碰撞前自动回正
                'airbag_deploy': 'full + side',
                'risk_level': 'high',
                'warning': '建议乘员不要完全 recline'
            },
            'rear_facing': {
                'restraint_mode': 'special',
                'belt_pretension': 'max',
                'head_restraint': 'extended',
                'airbag_deploy': 'disable_front',  # 禁用前气囊
                'risk_level': 'very_high'
            }
        }
    
    def evaluate_posture(self, oop_detection_result):
        """根据 OOP 检测结果选择约束策略"""
        angle = oop_detection_result.get('recline_angle', 25)
        facing = oop_detection_result.get('facing', 'forward')
        
        if angle <= 25 and facing == 'forward':
            return self.safety_strategies['upright_25deg']
        elif angle <= 43 and facing == 'forward':
            return self.safety_strategies['semi_recline_43deg']
        elif angle > 43 and facing == 'forward':
            return self.safety_strategies['full_recline_60deg']
        elif facing == 'rear':
            return self.safety_strategies['rear_facing']

class ReclineSafetyGuard:
    """
    座椅 recline 安全守卫
    
    功能：
    1. 实时检测乘员 recline 角度
    2. 评估当前约束系统有效性
    3. 预碰撞时自动调整座椅到安全位置
    4. 对不安全姿态发出警告
    """
    
    def __init__(self):
        self.thresholds = {
            'safe_angle': 25,          # ≤25° 安全
            'caution_angle': 35,       # 25-35° 需要增强约束
            'warning_angle': 45,       # 35-45° 发出警告
            'danger_angle': 55,        # >45° 不建议乘坐
        }
    
    def check_safety(self, occupant_state):
        """
        Args:
            occupant_state: {
                'recline_angle': float,  # 度
                'facing': str,           # forward/rear/side
                'belt_fastened': bool,
                'feet_on_dashboard': bool,
                'seat_occupied': bool
            }
        Returns:
            assessment: {
                'risk_level': str,
                'warning_message': str,
                'restraint_strategy': dict,
                'pre_crash_action': str
            }
        """
        angle = occupant_state['recline_angle']
        
        if not occupant_state['seat_occupied']:
            return {'risk_level': 'none', 'warning_message': '', 'restraint_strategy': {}, 'pre_crash_action': ''}
        
        if occupant_state.get('feet_on_dashboard'):
            return {
                'risk_level': 'extreme',
                'warning_message': '请将脚从仪表台移开——气囊可能导致严重伤害',
                'restraint_strategy': {'airbag_disable': 'front_passenger'},
                'pre_crash_action': 'disable_front_airbag'
            }
        
        if angle <= self.thresholds['safe_angle']:
            return {
                'risk_level': 'low',
                'warning_message': '',
                'restraint_strategy': {'mode': 'standard'},
                'pre_crash_action': 'none'
            }
        elif angle <= self.thresholds['caution_angle']:
            return {
                'risk_level': 'medium',
                'warning_message': '',
                'restraint_strategy': {
                    'mode': 'enhanced',
                    'belt_pretension': 'aggressive',
                    'anti_submarining': 'active'
                },
                'pre_crash_action': 'pre_tension_belt'
            }
        elif angle <= self.thresholds['warning_angle']:
            return {
                'risk_level': 'high',
                'warning_message': '建议调整座椅到更直立的姿势以确保安全',
                'restraint_strategy': {
                    'mode': 'maximum',
                    'belt_pretension': 'max',
                    'anti_submarining': 'max',
                    'lumbar_airbag': 'armed'
                },
                'pre_crash_action': 'auto_return_seat'
            }
        else:  # > 55°
            return {
                'risk_level': 'extreme',
                'warning_message': '当前座椅角度不安全，请调整至更直立的位置',
                'restraint_strategy': {
                    'mode': 'maximum',
                    'belt_pretension': 'max',
                    'anti_submarining': 'max',
                    'lumbar_airbag': 'armed',
                    'seat_auto_return': 'required'
                },
                'pre_crash_action': 'auto_return_seat + tighten_belt'
            }