#pragma once

#include <memory>
#include <string>
#include <vector>
#include <unordered_map>
#include <map>
#include <opencv2/core/types.hpp>

namespace pipeline {

// 输入源配置
struct InputSourceConfig {
    std::string type;        // rtsp/video
    std::string name;        // 输入源名称
    std::string url;         // RTSP URL
    int buffer_size{30};     // 缓冲区大小
    std::vector<std::string> outputs;  // 输出目标名称列表
};

// 输入配置
struct InputConfig {
    std::vector<InputSourceConfig> sources;  // 输入源列表
    int max_batch_size{4};                   // 最大批处理大小
};

// 模型配置
struct ModelConfig {
    std::string engine_path;           // 模型文件路径
    std::vector<int> input_shape{3, 640, 640}; // 输入尺寸
    std::string precision{"FP16"};     // 精度模式
    struct {
        float conf{0.5f};              // 置信度阈值
        float nms{0.45f};              // NMS阈值
    } threshold;
    int gpu_id{0};                     // GPU设备ID
    std::vector<std::string> labels;   // 可选：模型标签列表
    std::string version;               // 可选：YOLO版本信息
};

// 渲染配置
struct RenderConfig {
    bool enable{true};  // 控制是否启用渲染功能，默认启用

    // 窗口配置
    struct {
        std::string name{"Detection Results"};
        int width{1280};
        int height{720};
        bool fullscreen{false};
    } window;

    // 类别样式配置
    struct ClassStyle {
        cv::Scalar box_color{0, 255, 0};     // BGR格式，默认绿色
        cv::Scalar text_color{255, 255, 255}; // BGR格式，默认白色
        float transparency{0.0f};             // 0.0-1.0，0表示不透明
        int box_thickness{2};                 // 默认线宽
        double font_scale{0.5};              // 默认字体大小
        int font_thickness{1};               // 默认字体粗细
    };

    // 每个类别的样式映射
    std::map<std::string, ClassStyle> class_styles;
    
    // 默认样式
    ClassStyle default_style;

    // 性能指标显示设置
    struct {
        bool show_fps{true};
        bool show_inference_time{true};
        bool show_gpu_usage{true};
        int update_interval_ms{1000};
    } metrics;
};

// 输出目标配置
struct OutputTargetConfig {
    std::string type;        // video/rtsp
    std::string name;        // 输出名称
    std::string path;        // 输出路径
    int fps{30};            // 帧率
    std::string codec{"h264"}; // 编码器
    int bitrate{4000000};   // 比特率
};

// 输出配置
struct OutputConfig {
    std::vector<OutputTargetConfig> targets;  // 输出目标列表
};

// 日志配置
struct LogConfig {
    std::string level{"info"};         // 日志级别
    std::string save_path{"logs/"};    // 保存路径
};

// 完整Pipeline配置
struct PipelineConfig {
    InputConfig input;       // 输入配置
    ModelConfig inference;   // 推理配置
    RenderConfig render;     // 渲染配置
    OutputConfig output;     // 输出配置
    LogConfig log;          // 日志配置
};

class ConfigParser {
public:
    virtual ~ConfigParser() = default;
    virtual bool parse(const std::string& config_file) = 0;
    virtual bool validate() = 0;
    virtual const PipelineConfig& getConfig() const = 0;
};

// 创建配置解析器
std::unique_ptr<ConfigParser> createConfigParser();

} // namespace pipeline