~zhengyabo/SemanticSegmentationVer.git

using HalconDotNet;
using LB_VisionControls;
using System.Data.Common;
using LB_VisionProcesses.Alogrithms;
using System.Windows.Forms;
using OpenCvSharp;
using System;
 
namespace LB_VisionProcesses.Alogrithms
{
    public partial class IntersectionLine2ToolEdit : TAlgorithmEdit
    {
        public IntersectionLine2ToolEdit(IntersectionLine2Tool subject = null)
        {
            if (subject != null && subject is IntersectionLine2Tool)
            {
                Subject = subject;
            }
            else
            {
                Subject = new IntersectionLine2Tool();
            }
 
            this.Dock = DockStyle.Fill;
            InitializeComponent();
        }
 
        /// <summary>
        /// 控件加载事件
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void IntersectionLine2ToolEdit_Load(object sender, EventArgs e)
        {
            pnlInputImage.Controls.Add(inputImageHSmartWindowControl);
            inputImageHSmartWindowControl.Dock = DockStyle.Fill;
 
            pnlRecordImage.Controls.Add(recordImageHSmartWindowControl);
            recordImageHSmartWindowControl.Dock = DockStyle.Fill;
 
            //遍历可以选择的Roi类型枚举
            foreach (var value in Enum.GetValues(typeof(RoiType)))
            {
                cmbTypeRoi.Items.Add(value.ToString());
            }
 
            //遍历可以选择的Fixture枚举
            cmbFixture.Items.Add("");
            foreach (string value in IProcess.dicFixtures.Keys)
            {
                cmbFixture.Items.Add(value.ToString());
            }
 
            cmbTypeRoi.Text = RoiType.None.ToString();
            LoadParas();
 
            if (Subject.Result)
            {
                lblResult.BackColor = Color.Green;
                lblResult.Text = "True";
            }
            else
            {
                lblResult.BackColor = Color.Red;
                lblResult.Text = "False";
            }
 
            lblMsg.Text = Msg.Length > 50 ? Msg.Substring(0, 50) : Msg;
            lblMsgToolTip.SetToolTip(BtmStatusStrip, Msg);
            lblRunTime.Text = $"{Subject.RunTime}ms";
        }
 
        /// <summary>
        /// 更新运行参数
        /// </summary>
        public override void UpdataInputs()
        {
            //设置运行参数
            Subject.Params.Inputs["是否期望有交点"] = ckbNeedInter.Checked;
            Subject.Params.Inputs["要求交点在线段上"] = ckbOnSegment.Checked;
 
            if (cmbFixture.Text == "")
            {
                Subject.Params.Fixture = new Fixture();
            }
            else if (IProcess.dicFixtures.ContainsKey(cmbFixture.Text))
            {
                Subject.Params.Fixture = IProcess.dicFixtures[cmbFixture.Text];
            }
 
            Type type = inputImageHSmartWindowControl.oRoi?.GetType();
            switch (type)
            {
                case Type t when t == typeof(HRectangle2):
                    {
                        HRectangle2 hRectangle2 = (HRectangle2)inputImageHSmartWindowControl.oRoi;
                        Subject.Params.ROI
                            = new HRectangle2(hRectangle2.X - Subject.Params.Fixture.X, hRectangle2.Y - Subject.Params.Fixture.Y
                            , hRectangle2.Phi - Subject.Params.Fixture.Phi, hRectangle2.Width, hRectangle2.Height);
                        break;
                    }
                case Type t when t == typeof(HCircle):
                    {
                        HCircle hCircle = (HCircle)inputImageHSmartWindowControl.oRoi;
                        Subject.Params.ROI
                            = new HCircle(hCircle.X - Subject.Params.Fixture.X, hCircle.Y - Subject.Params.Fixture.Y, hCircle.Radius);
                        break;
                    }
                case Type t when t == typeof(HSegment):
                    {
                        HSegment hSegment = (HSegment)inputImageHSmartWindowControl.oRoi;
                        Subject.Params.ROI
                            = new HSegment(hSegment.StartX - Subject.Params.Fixture.X, hSegment.StartY - Subject.Params.Fixture.Y
                            , hSegment.EndX - Subject.Params.Fixture.X, hSegment.EndY - Subject.Params.Fixture.Y);
                        break;
                    }
                default:
                    {
                        Subject.Params.ROI = new ROI();
                        break;
                    }
            }
        }
 
        /// <summary>
        /// 加载运行参数
        /// </summary>
        public override void LoadParas()
        {
            this.BeginInvoke(new Action(() =>
            {
                ckbNeedInter.Checked = Convert.ToBoolean(Subject.Params.Inputs["是否期望有交点"]);
                ckbOnSegment.Checked = Convert.ToBoolean(Subject.Params.Inputs["要求交点在线段上"]);
 
                if (InputImage != null && InputImage is HObject ho_image && ho_image.IsInitialized())
                {
                    inputImageHSmartWindowControl.ShowHoImage(ho_image);
                }
                else if (InputImage != null && InputImage is Mat)
                {
                    TAlgorithm.Mat2HObject((Mat)InputImage, out HObject image);
                    inputImageHSmartWindowControl.ShowHoImage(image);
                }
                else if (InputImage != null && InputImage is Bitmap)
                {
                    TAlgorithm.Bitmap2HObject((Bitmap)InputImage, out HObject image);
                    inputImageHSmartWindowControl.ShowHoImage(image);
                }
 
                var oBaseSegment = Subject.Params.Inputs["基准线段"];
                var oInterSegment = Subject.Params.Inputs["相交线段"];
 
                if (oBaseSegment is HSegment && oInterSegment is HSegment)
                {
                    HOperatorSet.GenRegionLine(out HObject baseLine, ((HSegment)oBaseSegment).BeginRow, ((HSegment)oBaseSegment).BeginColumn
                        , ((HSegment)oBaseSegment).EndRow, ((HSegment)oBaseSegment).EndColumn);
                    HOperatorSet.GenRegionLine(out HObject interLine, ((HSegment)oInterSegment).BeginRow, ((HSegment)oInterSegment).BeginColumn
                        , ((HSegment)oInterSegment).EndRow, ((HSegment)oInterSegment).EndColumn);
 
                    inputImageHSmartWindowControl.DispObj(baseLine);
                    inputImageHSmartWindowControl.DispObj(interLine);
                }
 
                Type type = Subject.Params.ROI?.GetType();
                if (Subject.Params.ROI != null)
                {
                    switch (type)
                    {
                        case Type t when t == typeof(HRectangle2):
                            {
                                cmbTypeRoi.Text = RoiType.Rectangle2.ToString();
                                break;
                            }
                        case Type t when t == typeof(HCircle):
                            {
                                cmbTypeRoi.Text = RoiType.Circle.ToString();
                                break;
                            }
                        case Type t when t == typeof(HSegment):
                            {
                                cmbTypeRoi.Text = RoiType.Segment.ToString();
                                break;
                            }
                        default:
                            {
                                cmbTypeRoi.Text = RoiType.None.ToString();
                                break;
                            }
                    }
                    if (cmbTypeRoi.Text.ToString() != "None")
                    {
                        ckbDrawRoi.Checked = true;
                    }
                    else
                    {
                        ckbDrawRoi.Checked = false;
                    }
 
                    inputImageHSmartWindowControl.oRoi = Subject.Params.ROI;
                }
 
                if (Subject.Params.Fixture != null)
                {
                    cmbFixture.Text = Subject.Params.Fixture.strName;
                }
                else
                {
                    cmbFixture.Text = "";
                }
 
                switch (type)
                {
                    case Type t when t == typeof(HRectangle2):
                        {
                            inputImageHSmartWindowControl.oRoi
                             = new HRectangle2(Subject.Params.ROI.X + Subject.Params.Fixture.X, Subject.Params.ROI.Y + Subject.Params.Fixture.Y
                             , Subject.Params.ROI.Phi + Subject.Params.Fixture.Phi, ((HRectangle2)Subject.Params.ROI).Width, ((HRectangle2)Subject.Params.ROI).Height);
                            break;
                        }
                    case Type t when t == typeof(HCircle):
                        {
                            inputImageHSmartWindowControl.oRoi
                             = new HCircle(Subject.Params.ROI.X + Subject.Params.Fixture.X, Subject.Params.ROI.Y + Subject.Params.Fixture.Y
                             , ((HCircle)Subject.Params.ROI).Radius);
                            break;
                        }
                    case Type t when t == typeof(HSegment):
                        {
                            inputImageHSmartWindowControl.oRoi
                            = new HSegment(((HSegment)Subject.Params.ROI).StartX + Subject.Params.Fixture.X, ((HSegment)Subject.Params.ROI).StartY + Subject.Params.Fixture.Y
                            , ((HSegment)Subject.Params.ROI).EndX + Subject.Params.Fixture.X, ((HSegment)Subject.Params.ROI).EndY + Subject.Params.Fixture.Y);
                            break;
                        }
                    default:
                        {
                            inputImageHSmartWindowControl.oRoi = null;
                            break;
                        }
                }
            }));
        }
 
        /// <summary>
        /// 更新输出结果
        /// </summary>
        public override void UpdataOutputs()
        {
            this.BeginInvoke(new Action(() =>
            {
                txtResultX.Text = Subject.Params.Outputs["X"].ToString();
                txtResultY.Text = Subject.Params.Outputs["Y"].ToString();
                txtResultAngle.Text = Subject.Params.Outputs["Angle"].ToString();
 
                var oBaseSegment = Subject.Params.Inputs["基准线段"];
                var oInterSegment = Subject.Params.Inputs["相交线段"];
 
                if (oBaseSegment is HSegment && oInterSegment is HSegment)
                {
                    HOperatorSet.GenRegionLine(out HObject baseLine, ((HSegment)oBaseSegment).BeginRow, ((HSegment)oBaseSegment).BeginColumn
                        , ((HSegment)oBaseSegment).EndRow, ((HSegment)oBaseSegment).EndColumn);
                    HOperatorSet.GenRegionLine(out HObject interLine, ((HSegment)oInterSegment).BeginRow, ((HSegment)oInterSegment).BeginColumn
                        , ((HSegment)oInterSegment).EndRow, ((HSegment)oInterSegment).EndColumn);
 
                    recordImageHSmartWindowControl.DispObj(baseLine);
                    recordImageHSmartWindowControl.DispObj(interLine);
                }
            }));
        }
 
        /// <summary>
        /// 点击运行
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        public override void btnRun_Click(object sender, EventArgs e)
        {
            if (Subject.InputImage != null)
            {
                InputImage = Subject.InputImage;
            }
 
            DateTime StartTime = DateTime.Now;
            Run();
 
            //更新日志与结果
            this.BeginInvoke(new Action(() =>
            {
                if (Subject.Result)
                {
                    lblResult.BackColor = Color.Green;
                    lblResult.Text = "True";
                    recordImageHSmartWindowControl.SetColor("green");
                }
                else
                {
                    lblResult.BackColor = Color.Red;
                    lblResult.Text = "False";
                    recordImageHSmartWindowControl.SetColor("red");
                }
 
                lblMsg.Text = Msg.Length > 50 ? Msg.Substring(0, 50) : Msg;
                lblMsgToolTip.SetToolTip(BtmStatusStrip, Msg);
                lblRunTime.Text = $"{(DateTime.Now - StartTime).TotalMilliseconds}ms";
 
                UpdataOutputs();
                imgTabControl.SelectedTab = tabPageRecordImage;
 
                if (Subject.InputImage != null)
                {
                    if (Subject.InputImage is HObject)
                    {
                        recordImageHSmartWindowControl.ShowHoImage((HObject)Subject.InputImage);
                    }
                    else if (Subject.InputImage is Bitmap)
                    {
                        TAlgorithm.Bitmap2HObject((Bitmap)Subject.InputImage, out HObject image);
                        recordImageHSmartWindowControl.ShowHoImage(image);
                    }
                    else if (Subject.InputImage is Mat)
                    {
                        TAlgorithm.Mat2HObject((Mat)Subject.InputImage, out HObject image);
                        recordImageHSmartWindowControl.ShowHoImage(image);
                    }
                }
 
                //先判断子类再判断父类
                if (Subject.Record != null && Subject.Record is MsgRecord msgRecord)
                {
                    recordImageHSmartWindowControl.DispObj(msgRecord.RecordObject_OK, true);
                    recordImageHSmartWindowControl.DispObj(msgRecord.RecordObject_NG, false);
 
                    for (int i = 0; i < msgRecord.Msg.Length; i++)
                    {
                        recordImageHSmartWindowControl.ShowMsg(msgRecord.Msg[i]
                            , 1 == msgRecord.Result[i] ? true : false, msgRecord.Column[i], msgRecord.Row[i]);
                    }
                }
                else if (Subject.Record != null && Subject.Record is ObjectRecord objRecord)
                {
                    recordImageHSmartWindowControl.DispObj(objRecord.RecordObject_OK, true);
                    recordImageHSmartWindowControl.DispObj(objRecord.RecordObject_NG, false);
                }
 
                GC.Collect();
            }));
        }
 
        public override void btnLoadImage_Click(object sender, EventArgs e)
        {
            OpenFileDialog openFileDialog = new OpenFileDialog();
 
            // 设置文件对话框的属性
            openFileDialog.Multiselect = false; // 不允许多选
            // 设置文件过滤器，支持多种文件类型
            openFileDialog.Filter = "Image Files (*.png;*.jpg;*.jpeg;*.bmp)|*.png;*.jpg;*.jpeg;*.bmp|All Files (*.*)|*.*";
            // 显示文件对话框
            DialogResult result = openFileDialog.ShowDialog();
 
            // 处理对话框返回结果
            if (result == DialogResult.OK)
            {
                // 获取用户选择的文件名
                string[] selectedFiles = openFileDialog.FileNames;
                if (selectedFiles.Length > 0)
                {
                    HOperatorSet.ReadImage(out HObject ho_Image, selectedFiles[0]);
                    if (ho_Image == null)
                    {
                        lblResult.BackColor = Color.Red;
                        lblResult.Text = "False";
                        lblMsg.Text = $"图片导入失败,路径为{selectedFiles[0]}";
                    }
                    InputImage = ho_Image;
                    imgTabControl.SelectedTab = tabPageInputImage;
                }
            }
        }
 
        public override void btnSaveParas_Click(object sender, EventArgs e) { base.btnSaveParas_Click(sender, e); }
 
        public override void ckbDrawRoi_CheckedChanged(object sender, EventArgs e)
        {
            if (ckbDrawRoi.Checked)
            {
                inputImageHSmartWindowControl.bAollowDraw = true;
                imgTabControl.SelectedTab = tabPageInputImage;
            }
            else
            {
                inputImageHSmartWindowControl.bAollowDraw = false;
                Subject.Params.ROI = new ROI();
            }
        }
 
        public override void cmbTypeRoi_SelectedIndexChanged(object sender, EventArgs e)
        {
            try
            {
                if (Enum.TryParse(cmbTypeRoi.Text.ToString(), out RoiType type))
                {
                    HTuple hv_imageWidth = 0;
                    HTuple hv_imageHeight = 0;
                    if (InputImage != null && InputImage is HObject)
                    {
                        HOperatorSet.GetImageSize((HObject)InputImage, out hv_imageWidth, out hv_imageHeight);
                    }
                    switch (type)
                    {
                        case RoiType.Rectangle2:
                            {
                                inputImageHSmartWindowControl.oRoi
                                = new HRectangle2(hv_imageWidth.TupleReal() / 2, hv_imageHeight.TupleReal() / 2, 0
                                , hv_imageWidth.TupleReal() / 4, hv_imageHeight.TupleReal() / 4);
                                break;
                            }
                        case RoiType.Circle:
                            {
                                inputImageHSmartWindowControl.oRoi
                                = new HCircle(hv_imageWidth.TupleReal() / 2, hv_imageHeight.TupleReal() / 2, hv_imageWidth.TupleReal() / 4);
                                break;
                            }
                        case RoiType.Segment:
                            {
                                inputImageHSmartWindowControl.oRoi
                                = new HSegment(0, 0, hv_imageWidth.TupleReal() / 4, hv_imageHeight.TupleReal() / 4);
                                break;
                            }
                        case RoiType.None:
                        default:
                            {
                                inputImageHSmartWindowControl.oRoi = null;
                                break;
                            }
                    }
                }
            }
            catch { }
        }
 
        public override void cmbFixture_SelectedIndexChanged(object sender, EventArgs e)
        {
            try
            {
                if (IProcess.dicFixtures.ContainsKey(cmbFixture.Text))
                {
                    Subject.Params.Fixture = IProcess.dicFixtures[cmbFixture.Text];
                }
                else
                {
                    Subject.Params.Fixture = new Fixture();
                }
            }
            catch { }
        }
    }
}