namespace OrpaonVision.Core.Training.Contracts.Commands;

/// <summary>
/// 构建数据集命令。
/// </summary>
public sealed class BuildDatasetCommand
{
    /// <summary>
    /// 数据集名称。
    /// </summary>
    public string Name { get; init; } = string.Empty;

    /// <summary>
    /// 数据集描述。
    /// </summary>
    public string Description { get; init; } = string.Empty;

    /// <summary>
    /// 机种ID。
    /// </summary>
    public Guid ProductTypeId { get; init; }

    /// <summary>
    /// 标注任务ID列表。
    /// </summary>
    public IReadOnlyList<Guid> AnnotationTaskIds { get; init; } = [];

    /// <summary>
    /// 训练集比例（0-1）。
    /// </summary>
    public decimal TrainingRatio { get; init; } = 0.7m;

    /// <summary>
    /// 验证集比例（0-1）。
    /// </summary>
    public decimal ValidationRatio { get; init; } = 0.2m;

    /// <summary>
    /// 测试集比例（0-1）。
    /// </summary>
    public decimal TestRatio { get; init; } = 0.1m;

    /// <summary>
    /// 是否启用数据增强。
    /// </summary>
    public bool EnableDataAugmentation { get; init; } = true;

    /// <summary>
    /// 数据增强参数（JSON）。
    /// </summary>
    public string DataAugmentationJson { get; init; } = string.Empty;

    /// <summary>
    /// 最小图像尺寸。
    /// </summary>
    public int MinImageSize { get; init; } = 640;

    /// <summary>
    /// 最大图像尺寸。
    /// </summary>
    public int MaxImageSize { get; init; } = 640;

    /// <summary>
    /// 创建者。
    /// </summary>
    public string CreatedBy { get; init; } = string.Empty;
}