"""
THOR-50M 对 OMS 开发的影响分析

核心变化：
- 更真实的假人 → 更准确的碰撞数据
- 更多的传感器通道 → 更丰富的数据
- 四位置脊椎 → 更好的 OOP 检测基础

对 IMS 的具体影响：
1. OOP 异常姿态的碰撞验证数据更准确
2. 自适应约束的输入数据维度增加
3. 安全带误用检测的测试场景更丰富
"""

# THOR-50M 传感器通道列表
THOR50M_CHANNELS = {
    '头部': {
        '加速度_X': 'CG accelerometer X',
        '加速度_Y': 'CG accelerometer Y',
        '加速度_Z': 'CG accelerometer Z',
        '旋转_X': 'angular rate X',
        '旋转_Y': 'angular rate Y',
        '旋转_Z': 'angular rate Z',
    },
    '颈部': {
        '力_Fx': 'neck force Fx',
        '力_Fy': 'neck force Fy',
        '力_Fz': 'neck force Fz',
        '矩_Mx': 'neck moment Mx',
        '矩_My': 'neck moment My',
        '矩_Mz': 'neck moment Mz',
    },
    '胸部': {
        '加速度_X': 'chest acc X',
        '加速度_Y': 'chest acc Y',
        '加速度_Z': 'chest acc Z',
        '位移': 'chest deflection (multi-point)',
    },
    '脊椎': {
        '位置1_Fx': 'spine position 1 Fx',
        '位置1_Fy': 'spine position 1 Fy',
        '位置1_Fz': 'spine position 1 Fz',
        '位置1_Mx': 'spine position 1 Mx',
        '位置1_My': 'spine position 1 My',
        '位置1_Mz': 'spine position 1 Mz',
        '位置2_Fx': 'spine position 2 Fx',
        '位置2_Fy': 'spine position 2 Fy',
        # ... 四位置各有六轴力矩
    },
    '面部': {
        '压力_左': 'face pressure left',
        '压力_右': 'face pressure right',
        '压力_上': 'face pressure upper',
        '压力_下': 'face pressure lower',
    },
}

# 统计通道数
total_channels = sum(
    len(sensors) for sensors in THOR50M_CHANNELS.values()
)
print(f"THOR-50M 总传感器通道数: {total_channels}")
print(f"  头部: {len(THOR50M_CHANNELS['头部'])}")
print(f"  颈部: {len(THOR50M_CHANNELS['颈部'])}")
print(f"  胸部: {len(THOR50M_CHANNELS['胸部'])}")
print(f"  脊椎: {len(THOR50M_CHANNELS['脊椎'])}")
print(f"  面部: {len(THOR50M_CHANNELS['面部'])}")

"""
自适应约束策略：从 THOR-50M 数据到实时 OMS 决策

THOR-50M 新增数据维度：
1. 脊椎四位置六轴力矩 → 实时脊椎弯曲估计
2. 面部压力分布 → 安全气囊展开策略
3. 多点胸部位移 → 胸部损伤风险预测
"""

class AdaptiveRestraintStrategy:
    """自适应约束策略"""
    
    def __init__(self):
        # 基于 THOR-50M 数据校准的阈值
        self.injury_thresholds = {
            'chest_deflection': 22,  # mm, 胸部位移阈值
            'neck_moment_y': 190,     # Nm, 颈部弯矩阈值
            'face_pressure': 80,     # kPa, 面部压力阈值
            'spine_force_z': 3500,   # N, 脊椎轴向力阈值
        }
    
    def assess_occupant_state(self, sensor_data):
        """
        基于 OMS 传感器数据评估乘员状态
        
        Args:
            sensor_data: 来自摄像头/雷达的乘员状态
        """
        posture = sensor_data.get('posture', 'normal')
        belt_status = sensor_data.get('belt_status', 'correct')
        
        # 根据乘员状态调整约束参数
        if posture == 'forward_lean':
            # 前倾：降低气囊展开力度，延迟安全带预紧
            return {
                'airbag_force': 0.7,  # 70% 力度
                'belt_pretension': 'delayed',
                'belt_force_limiter': 3000,  # N
                'rationale': '前倾姿态，减小气囊冲击'
            }
        elif posture == 'rear_recline':
            # 后仰：增大气囊展开面积，降低腰带力
            return {
                'airbag_force': 0.9,
                'belt_pretension': 'immediate',
                'belt_force_limiter': 2500,
                'rationale': '后仰姿态，防止 submarining'
            }
        elif belt_status == 'misuse':
            # 安全带误用：最大力度展开气囊
            return {
                'airbag_force': 1.0,
                'belt_pretension': 'max',
                'belt_force_limiter': 4000,
                'rationale': '安全带误用，气囊补偿'
            }
        else:
            # 正常姿态
            return {
                'airbag_force': 0.85,
                'belt_pretension': 'standard',
                'belt_force_limiter': 3500,
                'rationale': '正常姿态'
            }