using OpenCvSharp;
using SHH.CameraSdk;

public class ImageEnhanceCluster : BaseFrameProcessor<EnhanceWorker>
{
    public ImageEnhanceCluster(int count, ProcessingConfigManager configManager)
        : base(count, "EnhanceCluster", configManager)
    {
    }

    protected override EnhanceWorker CreateWorker(int id) => new EnhanceWorker(this, _configManager);
}

public class EnhanceWorker : BaseWorker
{
    private readonly ImageEnhanceCluster _parent;
    private readonly ProcessingConfigManager _configManager;

    public EnhanceWorker(ImageEnhanceCluster parent, ProcessingConfigManager configManager)
    {
        _parent = parent;
        _configManager = configManager;
    }

    protected override void PerformAction(long deviceId, SmartFrame frame, FrameDecision decision)
    {
        // 1. 获取配置
        var options = _configManager.GetOptions(deviceId);

        // 2. 检查开关：如果没开启增强，直接跳过
        if (!options.EnableEnhance) return;

        // 3. 确定操作对象
        // 策略：如果上一站生成了 TargetMat (缩放图)，我们处理缩放图；
        // 如果没有 (例如缩放被禁用)，我们是否要处理原图？
        // 通常 UI 预览场景下，如果不缩放，直接处理 4K 原图会非常卡。
        // 建议：仅当 TargetMat 存在时处理，或者强制 clone 一份原图作为 TargetMat

        Mat srcMat = frame.TargetMat;
        bool createdNew = false;

        // 如果没有 TargetMat (上一站透传了)，但开启了增亮
        // 我们必须基于原图生成一个 TargetMat，否则下游 UI 拿不到处理结果
        if (srcMat == null || srcMat.IsDisposed)
        {
            // 注意：处理 4K 原图非常耗时，生产环境建议这里做个限制
            srcMat = frame.InternalMat;
            createdNew = true; // 标记我们需要 Attach 新的
        }

        // 4. 执行增亮
        Mat brightMat = new Mat();
        // Alpha=1.0, Beta=配置值
        srcMat.ConvertTo(brightMat, -1, 1.0, options.BrightnessLevel);

        // 5. 挂载结果
        // 这会自动释放上一站生成的旧 TargetMat (如果存在)
        frame.AttachTarget(brightMat, frame.ScaleType);
    }

    protected override void NotifyFinished(long did, SmartFrame frame, FrameDecision dec)
    {
        _parent.PassToNext(did, frame, dec);
    }
}