机器视觉通用平台之缺陷检测(灰度差分EC)算法工具类

using CvBase;
using CWindowTool;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using HalconDotNet;
using System.IO;
using Newtonsoft.Json;

namespace CvImageTool.GrayDiff_EC
{
    public class GrayDiff_ECTool
    {
        public FileINI INI = new FileINI();
        public GrayDiff_ECParam grayDiff_ECParam;    //匹配参数
        private string cProcessSettingFilePath;     //当前流程配置文件的路径
        private string cImageOptSettingFilePath;    //当前算法参数文件路径
        public string imageSelect = "";            //图像输入参数选择项
        public string affinMatrixSelect = "";      //仿射矩阵输入参数选择项
        private int cProcessIndex = -1;             //当前归属流程的索引
        private int cImageOptIndex = -1;            //当前归属算法的索引
        public bool isShowResultRegion = false;
        public bool isShowROI = false;
        public bool isShowResult = false;
        public HObject Image;
        public HTuple AffinMatrix;
        private List<BaseProcess> process;
        private CWindows[] cCWindows;
        public ROITool roiTool;

        public GrayDiff_ECTool(int cProcessIndex, int cImageOptIndex,
           string cProcessSettingFilePath, string cImageOptSettingFilePath, string cDescription,
           List<BaseProcess> processes, CWindows[] cWindows)
        {
            this.cProcessSettingFilePath = cProcessSettingFilePath;
            this.cImageOptSettingFilePath = cImageOptSettingFilePath;
            this.cProcessIndex = cProcessIndex;
            this.cImageOptIndex = cImageOptIndex;
            this.process = processes;
            this.cCWindows = cWindows;
            grayDiff_ECParam = new GrayDiff_ECParam();
            Image = new HObject();
            roiTool = new ROITool(cImageOptSettingFilePath, cDescription + "_ROI");
        }

        /// <summary>
        /// 读取配置文件
        /// </summary>
        /// <param name="cProcessSettingFilePath"></param>
        /// <param name="cImgeOptSettingFilePath"></param>
        /// <returns></returns>
        public bool ReadFile(string cProcessDescription)
        {
            imageSelect = ""; affinMatrixSelect = "";
            if (!File.Exists(cProcessSettingFilePath) || !File.Exists(cImageOptSettingFilePath)) return false;
            imageSelect = FileINI.ReadValueFromIniFile(cProcessDescription, "Image", "", cProcessSettingFilePath);
            affinMatrixSelect = FileINI.ReadValueFromIniFile(cProcessDescription, "Matrix", "", cProcessSettingFilePath);
            isShowResultRegion = Convert.ToBoolean(FileINI.ReadValueFromIniFile(cProcessDescription, "isShowResultRegion", "", cProcessSettingFilePath));
            isShowResult = Convert.ToBoolean(FileINI.ReadValueFromIniFile(cProcessDescription, "isShowResult", "", cProcessSettingFilePath));
            isShowROI = Convert.ToBoolean(FileINI.ReadValueFromIniFile(cProcessDescription, "ShowROI", "", cProcessSettingFilePath));
            grayDiff_ECParam = JsonConvert.DeserializeObject<GrayDiff_ECParam>(File.ReadAllText(cImageOptSettingFilePath));
            return true;
        }

        /// <summary>
        /// 写配置文件
        /// </summary>
        /// <param name="cProcessSettingFilePath"></param>
        /// <param name="cImgeOptSettingFilePath"></param>
        /// <returns></returns>
        public bool WriteFile(string cProcessDescription)
        {
            FileINI.WriteValueFromIniFile(cProcessDescription, "Image", imageSelect, cProcessSettingFilePath);
            FileINI.WriteValueFromIniFile(cProcessDescription, "Matrix", affinMatrixSelect, cProcessSettingFilePath);
            FileINI.WriteValueFromIniFile(cProcessDescription, "isShowResultRegion", isShowResultRegion.ToString(), cProcessSettingFilePath);
            FileINI.WriteValueFromIniFile(cProcessDescription, "isShowResult", isShowResult.ToString(), cProcessSettingFilePath);
            FileINI.WriteValueFromIniFile(cProcessDescription, "ShowROI", isShowROI.ToString(), cProcessSettingFilePath);
            var json = JsonConvert.SerializeObject(grayDiff_ECParam, Formatting.Indented);
            File.WriteAllText(cImageOptSettingFilePath, json);
            return true;
        }


        public void Running(out HObject ROI, out HObject resultRegion, out HTuple x, out HTuple y, out HTuple area)
        {
            ROI = new HObject(); resultRegion = new HObject(); x = new HTuple(); y = new HTuple(); area = new HTuple();
            if (Image == null) return;
            string[] imageOptAndParamName = affinMatrixSelect.Split('_');
            ProcessTool.Instance.FindImageOpt(process[cProcessIndex], imageOptAndParamName[0], out BaseImageOpt byImageOpt, out int byIndex);
            if (byIndex >= 0)
                process[cProcessIndex].BaseImageOutParams[byIndex].GetHTupleValue(imageOptAndParamName[1], out AffinMatrix);
            else
                AffinMatrix = null;
            roiTool.ReadROI();
            ROI = roiTool.ROI;
            if (grayDiff_ECParam.isAffin && AffinMatrix != null && AffinMatrix.Length == 6)
            {
                HOperatorSet.AffineTransRegion(ROI, out HObject NROI, AffinMatrix, "nearest_neighbor");
                HObject hObject = null;
                if (ROI != null) hObject = ROI;
                ROI = NROI;
                if (hObject != null) hObject.Dispose();
            }
            HOperatorSet.ReduceDomain(Image, ROI, out HObject ReduceImage);

            HObject DarkImage = null;
            if (grayDiff_ECParam.GrayDarkType == EnumOrStruct.EGrayDarkType.Opening)
                HOperatorSet.GrayOpeningShape(ReduceImage, out DarkImage, grayDiff_ECParam.GrayDarkHeight, 
                    grayDiff_ECParam.GrayDarkWidth, "octagon");
            else
                HOperatorSet.GrayErosionShape(ReduceImage, out DarkImage, grayDiff_ECParam.GrayDarkHeight,
                    grayDiff_ECParam.GrayDarkWidth, "octagon");

            HObject LightImage = null;
            if (grayDiff_ECParam.GrayLightType == EnumOrStruct.EGrayLightType.Closing)
                HOperatorSet.GrayClosingShape(ReduceImage, out LightImage, grayDiff_ECParam.GrayLightHeight,
                    grayDiff_ECParam.GrayLightWidth, "octagon");
            else
                HOperatorSet.GrayDilationShape(ReduceImage, out LightImage, grayDiff_ECParam.GrayLightHeight,
                    grayDiff_ECParam.GrayLightWidth, "octagon");
            ReduceImage.Dispose();
            HOperatorSet.DynThreshold(DarkImage, LightImage, out HObject DiffRegion, grayDiff_ECParam.Constract,
                Enum.GetName(typeof(EnumOrStruct.EComparisonMethod), grayDiff_ECParam.ComparisonMethod));
            DarkImage.Dispose(); LightImage.Dispose();
            HOperatorSet.Connection(DiffRegion, out HObject connectionRegion);
            DiffRegion.Dispose();
            HTuple names = new HTuple();
            names.Append("area");
            names.Append("height");
            names.Append("width");
            names.Append("circularity");
            HTuple minValues = new HTuple();
            minValues.Append(grayDiff_ECParam.MinArea);
            minValues.Append(grayDiff_ECParam.MinHeight);
            minValues.Append(grayDiff_ECParam.MinWidth);
            minValues.Append(grayDiff_ECParam.MinCircularity);
            HTuple maxValues = new HTuple();
            maxValues.Append(grayDiff_ECParam.MaxArea);
            maxValues.Append(grayDiff_ECParam.MaxHeight);
            maxValues.Append(grayDiff_ECParam.MaxWidth);
            maxValues.Append(grayDiff_ECParam.MaxCircularity);
            resultRegion.Dispose();
            HOperatorSet.SelectShape(connectionRegion, out resultRegion, names, "and", minValues, maxValues);
            area.Dispose(); x.Dispose(); y.Dispose();
            HOperatorSet.AreaCenter(resultRegion, out area, out y, out x);
            connectionRegion.Dispose();
        }
    }
}
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值