using HalconTemplateMatch;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Drawing;

namespace Check.Main.Infer
{
    public class HalconTemplateDetector : IDetector
    {
        private LogoMatcher _matcher; // 使用您现有的 LogoMatcher
        private double _scoreThreshold = 0.5; // 可以从 ModelSettings 中配置

        public void Initialize(string modelPath, object detectionSettings = null)
        {
            _matcher = new LogoMatcher();
            _matcher.LoadTemplates(modelPath); // modelPath 现在是 Halcon 模板目录

            // 如果 detectionSettings 包含阈值，可以在这里解析
            if (detectionSettings is HalconDetectionSettings halconSettings)
            {
                _scoreThreshold = halconSettings.ScoreThreshold;
            }
        }

        public DetectionResult Detect(Bitmap image)
        {
            if (_matcher == null)
                throw new InvalidOperationException("HalconTemplateDetector 未初始化。");

            Bitmap resultImage;
            double score = _matcher.FindLogo(image, out resultImage);
            bool isOk = score >= _scoreThreshold;
            string message = isOk ? "OK" : "NG";

            // 如果需要绘制结果图像，可以在 LogoMatcher 中添加绘制逻辑并返回
            // 假设 LogoMatcher 也可以返回一个带有匹配标记的 Bitmap
            // Bitmap resultImage = _matcher.DrawResults(image, score);

            return new DetectionResult(isOk, message, score); //, resultImage
        }

        public void Dispose()
        {
            // LogoMatcher 如果有需要释放的资源，可以在这里处理
        }
    }

    // 辅助设置类，用于传递给 HalconTemplateDetector
    public class HalconDetectionSettings
    {
        public double ScoreThreshold { get; set; } = 0.5;
        // 可以添加其他 Halcon 匹配参数
    }
}