"""配置加载模块"""
import yaml
from pathlib import Path
from typing import Dict, Any, List


class ConfigLoader:
    """加载和验证YAML配置文件"""

    def __init__(self, config_path: str):
        """
        Args:
            config_path: 配置文件路径
        """
        self.config_path = Path(config_path)
        self.config = None

    def load(self) -> Dict[str, Any]:
        """加载配置文件"""
        if not self.config_path.exists():
            raise FileNotFoundError(f"配置文件不存在: {self.config_path}")

        with open(self.config_path, 'r', encoding='utf-8') as f:
            self.config = yaml.safe_load(f)

        self._validate()
        return self.config

    def _validate(self):
        """验证配置文件格式"""
        if not isinstance(self.config, dict):
            raise ValueError("配置文件格式错误：根节点必须是字典")

        # 检查必需字段
        if 'point_cloud' not in self.config:
            raise ValueError("配置文件缺少 'point_cloud' 字段")

        if 'tasks' not in self.config:
            raise ValueError("配置文件缺少 'tasks' 字段")

        if not isinstance(self.config['tasks'], list):
            raise ValueError("'tasks' 必须是列表")

        # 验证每个任务
        for i, task in enumerate(self.config['tasks']):
            self._validate_task(task, i)

    def _validate_task(self, task: Dict[str, Any], index: int):
        """验证单个任务配置"""
        required_fields = ['name', 'type']
        for field in required_fields:
            if field not in task:
                raise ValueError(f"任务 {index} 缺少必需字段: {field}")

        # 验证任务类型
        valid_types = ['plane_to_plane', 'point_to_plane', 'point_to_point']
        if task['type'] not in valid_types:
            raise ValueError(f"任务 {index} 类型无效: {task['type']}, 必须是 {valid_types} 之一")

    def get_point_cloud_path(self) -> str:
        """获取点云文件路径"""
        return self.config['point_cloud']

    def get_tasks(self) -> List[Dict[str, Any]]:
        """获取所有任务配置"""
        return self.config['tasks']

    def get_global_settings(self) -> Dict[str, Any]:
        """获取全局设置"""
        return self.config.get('global_settings', {})

    def is_visualization_enabled(self) -> bool:
        """是否启用可视化"""
        return self.get_global_settings().get('enable_visualization', False)

    def get_visualization_dir(self) -> str:
        """获取可视化输出目录"""
        return self.get_global_settings().get('visualization_dir', 'visualizations')

    def get_results_dir(self) -> str:
        """获取结果输出目录"""
        return self.get_global_settings().get('results_dir', 'results')