#!/usr/bin/env python3
"""
简单测试Phase 4B配置是否能正常构建模型
"""

import sys
import os
sys.path.insert(0, '/workspace/bevfusion')

def test_config():
    """测试配置能否正常构建模型"""
    try:
        # 设置环境
        os.environ['PYTHONPATH'] = '/workspace/bevfusion'

        # 导入必要的模块
        from mmcv import Config
        from mmdet3d.models import build_model

        print("🔧 加载Phase 4B配置...")

        # 加载配置
        config_file = 'configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/multitask_BEV2X_phase4b_rmtppad_segmentation.yaml'
        cfg = Config.fromfile(config_file)

        print("✅ 配置加载成功")

        # 简化配置用于测试
        cfg.data.samples_per_gpu = 1
        cfg.data.workers_per_gpu = 0

        # 设置训练模式
        cfg.model.train_cfg = {
            'object': {
                'grid_size': [1440, 1440, 41]
            },
            'map': {}
        }

        print("🔧 构建模型...")

        # 构建模型
        model = build_model(cfg.model, train_cfg=cfg.model.train_cfg)

        print("✅ 模型构建成功")
        print(f"   - 模型类型: {type(model).__name__}")

        # 检查heads
        if hasattr(model, 'heads'):
            heads = list(model.heads.keys())
            print(f"   - 模型heads: {heads}")

            if 'map' in model.heads:
                map_head = model.heads['map']
                print(f"   - 分割头类型: {type(map_head).__name__}")

        return True

    except Exception as e:
        print(f"❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    success = test_config()
    if success:
        print("\n🎉 Phase 4B配置测试通过！")
    else:
        print("\n❌ 配置测试失败")