import numpy as np
from dataclasses import dataclass, field
from typing import List, Optional, Tuple
import time

"""
IMS 座舱空气质量监测系统
整合 CO2/VOC/NOx/PM2.5 多传感器

参考:
- Metis Engineering Air Wise (CAN, CO2/VOC/NOx, 2026-06)
- Senseair K96 (3-channel NDIR, 2026)
- ARAI VOC 排放研究 (GC-MS, 2026-07)
"""

@dataclass
class AirQualityReading:
    """空气质量读数"""
    co2_ppm: float      # CO2 浓度
    voc_ppb: float      # VOC 总量
    nox_ppb: float      # NOx 浓度
    pm25_ugm3: float    # PM2.5
    pm10_ugm3: float    # PM10
    temp_c: float       # 温度
    humidity_pct: float # 湿度
    pressure_hpa: float # 气压
    timestamp: float
    
    @property
    def aqi_score(self) -> float:
        """综合空气质量评分 (0-100, 越高越好)"""
        score = 100
        
        # CO2: 400-1000ppm 正常, >1500ppm 差
        if self.co2_ppm > 1000:
            score -= min(30, (self.co2_ppm - 1000) / 50)
        
        # VOC: <300ppb 好, >1000ppb 差
        if self.voc_ppb > 300:
            score -= min(25, (self.voc_ppb - 300) / 50)
        
        # NOx: <50ppb 好, >200ppb 差
        if self.nox_ppb > 50:
            score -= min(20, (self.nox_ppb - 50) / 15)
        
        # PM2.5: <15ug/m3 好, >35ug/m3 差
        if self.pm25_ugm3 > 15:
            score -= min(25, (self.pm25_ugm3 - 15) / 1)
        
        return max(0, round(score, 1))


@dataclass
class VentilationAction:
    """通风动作"""
    action: str  # 'fresh_air', 'recirculate', 'purge', 'auto'
    fan_speed: int  # 0-7
    ac_on: bool
    reason: str
    urgency: str  # 'info', 'warning', 'critical'


class CabinAirQualityMonitor:
    """座舱空气质量监控器"""
    
    # 健康阈值
    THRESHOLDS = {
        'co2_good': 800,      # ppm
        'co2_warning': 1200,  # ppm
        'co2_critical': 2000, # ppm
        'voc_good': 300,      # ppb
        'voc_warning': 800,   # ppb
        'voc_critical': 1500, # ppb
        'nox_good': 50,       # ppb
        'nox_warning': 150,   # ppb
        'nox_critical': 300,  # ppb
        'pm25_good': 15,      # ug/m3
        'pm25_warning': 35,    # ug/m3
        'pm25_critical': 75,  # ug/m3
    }
    
    # 传感器配置
    SENSORS = {
        'metis_air_wise': {
            'name': 'Metis Air Wise',
            'interface': 'CAN',
            'measures': ['CO2', 'VOC', 'NOx', 'pressure', 'temp', 'humidity', 'shock'],
            'update_hz': 1,
            'power_mw': 150
        },
        'senseair_k96': {
            'name': 'Senseair K96',
            'interface': 'I2C/UART',
            'measures': ['CO2', 'CO', 'HC'],
            'tech': '3-channel NDIR',
            'update_hz': 0.5,
            'power_mw': 80
        },
        'sensirion_sgp40': {
            'name': 'Sensirion SGP40',
            'interface': 'I2C',
            'measures': ['VOC'],
            'tech': 'MOS',
            'update_hz': 1,
            'power_mw': 40
        },
        'pmsa003i': {
            'name': 'Plantower PMSA003I',
            'interface': 'I2C',
            'measures': ['PM2.5', 'PM10'],
            'tech': 'laser scattering',
            'update_hz': 1,
            'power_mw': 100
        }
    }
    
    def __init__(self):
        self.history: List[AirQualityReading] = []
        self.ventilation_actions: List[VentilationAction] = []
    
    def process_reading(self, reading: AirQualityReading) -> VentilationAction:
        """处理读数并决定通风策略"""
        
        # CO2 驱动的新风策略
        if reading.co2_ppm > self.THRESHOLDS['co2_critical']:
            action = 'fresh_air'
            fan_speed = 7
            ac_on = True
            reason = f"CO2 {reading.co2_ppm:.0f}ppm 严重超标"
            urgency = 'critical'
        elif reading.co2_ppm > self.THRESHOLDS['co2_warning']:
            action = 'fresh_air'
            fan_speed = 4
            ac_on = False
            reason = f"CO2 {reading.co2_ppm:.0f}ppm 超标"
            urgency = 'warning'
        elif reading.co2_ppm > self.THRESHOLDS['co2_good']:
            action = 'fresh_air'
            fan_speed = 2
            ac_on = False
            reason = f"CO2 {reading.co2_ppm:.0f}ppm 偏高"
            urgency = 'info'
        else:
            action = 'recirculate'
            fan_speed = 1
            ac_on = True
            reason = "空气质量良好"
            urgency = 'info'
        
        # VOC 驱动的紧急排放
        if reading.voc_ppb > self.THRESHOLDS['voc_critical']:
            action = 'purge'
            fan_speed = 7
            ac_on = True
            reason = f"VOC {reading.voc_ppb:.0f}ppb 严重超标"
            urgency = 'critical'
        elif reading.voc_ppb > self.THRESHOLDS['voc_warning'] and urgency == 'info':
            action = 'fresh_air'
            fan_speed = max(fan_speed, 3)
            reason = f"VOC {reading.voc_ppb:.0f}ppb 超标"
            urgency = 'warning'
        
        # PM2.5 驱动的过滤
        if reading.pm25_ugm3 > self.THRESHOLDS['pm25_critical']:
            action = 'recirculate'
            fan_speed = max(fan_speed, 5)
            ac_on = True
            reason = f"PM2.5 {reading.pm25_ugm3:.0f}ug/m3 严重超标"
            urgency = 'critical'
        elif reading.pm25_ugm3 > self.THRESHOLDS['pm25_warning'] and urgency == 'info':
            action = 'recirculate'
            fan_speed = max(fan_speed, 3)
            reason = f"PM2.5 {reading.pm25_ugm3:.0f}ug/m3 超标"
            urgency = 'warning'
        
        # NOx 驱动（通常来自外部）
        if reading.nox_ppb > self.THRESHOLDS['nox_warning']:
            action = 'recirculate'
            reason = f"NOx {reading.nox_ppb:.0f}ppb 超标(外部?)"
            if urgency == 'info':
                urgency = 'warning'
        
        result = VentilationAction(
            action=action, fan_speed=fan_speed,
            ac_on=ac_on, reason=reason, urgency=urgency
        )
        
        self.history.append(reading)
        self.ventilation_actions.append(result)
        return result


# ==================== 测试 ====================
if __name__ == "__main__":
    np.random.seed(42)
    monitor = CabinAirQualityMonitor()
    
    print("=" * 75)
    print("IMS 座舱空气质量监测系统测试")
    print("=" * 75)
    
    # 模拟驾驶场景
    scenarios = [
        ('刚上车(停车场)', 450, 120, 15, 8, 35, 60, 1010),
        ('城市驾驶(隧道)', 850, 200, 180, 25, 40, 32, 1010),
        ('高速驾驶(1h后)', 1400, 350, 40, 12, 28, 50, 1005),
        ('长时间关窗', 2100, 900, 30, 18, 30, 55, 1005),
        ('新车VOC释放', 600, 1600, 20, 10, 45, 70, 1010),
        ('雾霾天外部', 500, 100, 80, 85, 50, 25, 1010),
    ]
    
    print(f"\n{'场景':<20} {'CO2':>6} {'VOC':>6} {'NOx':>6} {'PM2.5':>7} "
          f"{'AQI':>6} {'动作':>12} {'档位':>4} {'AC':>4} {'紧急':>8}")
    print("-" * 90)
    
    for name, co2, voc, nox, pm25, temp, hum, pres in scenarios:
        reading = AirQualityReading(
            co2_ppm=co2, voc_ppb=voc, nox_ppb=nox,
            pm25_ugm3=pm25, pm10_ugm3=pm25*1.5,
            temp_c=temp, humidity_pct=hum, pressure_hpa=pres,
            timestamp=time.time()
        )
        
        action = monitor.process_reading(reading)
        
        print(f"{name:<20} {co2:>6.0f} {voc:>6.0f} {nox:>6.0f} {pm25:>7.0f} "
              f"{reading.aqi_score:>6.1f} {action.action:>12} "
              f"{action.fan_speed:>4} {'❄️' if action.ac_on else '—':>4} "
              f"{action.urgency:>8}")
        
        print(f"  原因: {action.reason}")
    
    print(f"\n{'='*75}")
    print("传感器配置:")
    print(f"{'传感器':<20} {'接口':>10} {'测量':>30} {'功耗':>8}")
    print("-" * 70)
    for sid, s in CabinAirQualityMonitor.SENSORS.items():
        measures = ','.join(s['measures'])
        print(f"{s['name']:<20} {s['interface']:>10} {measures:>30} "
              f"{s['power_mw']:>7}mW")
    
    print(f"\n{'='*75}")
    print("健康阈值:")
    print(f"  {'指标':<10} {'良好':>10} {'警告':>10} {'严重':>10} {'单位':>8}")
    print(f"  {'CO2':<10} {'<800':>10} {'>1200':>10} {'>2000':>10} {'ppm':>8}")
    print(f"  {'VOC':<10} {'<300':>10} {'>800':>10} {'>1500':>10} {'ppb':>8}")
    print(f"  {'NOx':<10} {'<50':>10} {'>150':>10} {'>300':>10} {'ppb':>8}")
    print(f"  {'PM2.5':<10} {'<15':>10} {'>35':>10} {'>75':>10} {'ug/m3':>8}")