using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using CvBase;
using CWindowTool;
using HalconDotNet;

namespace CvImageTool
{
    public partial class OCRForm : Form
    {
        private int ProcessIndex { set; get; }
        private int ImageOptIndex { set; get; }
        private string Description { set; get; }
        private OCRTool ocrTool;

        private List<BaseProcess> processes;
        private CWindows[] cCWindows;
        public OCRForm()
        {
            InitializeComponent();
        }

        public OCRForm(int processIndex, int imageOptIndex, string cProcessSettingFilePath, string cImageOptSettingFilePath, 
            string trainFilePath, string cDescription, List<BaseProcess> baseProcesses, CWindows[] cWindows)
        {
            this.StartPosition = FormStartPosition.CenterParent;
            InitializeComponent();
            this.ProcessIndex = processIndex;
            this.ImageOptIndex = imageOptIndex;
            this.Description = cDescription;
            this.processes = baseProcesses;
            this.cCWindows = cWindows;
            ocrTool = new OCRTool(processIndex, imageOptIndex, cProcessSettingFilePath, cImageOptSettingFilePath, trainFilePath, processes, cWindows);
        }

        private void OCRForm_Load(object sender, EventArgs e)
        {
            Image_OptNameCMBox.Items.Clear();
            Image_OptNameCMBox.Items.Add("相机");
            Matrix_OptNameCMBox.Items.Clear();
            for (int i = 0; i < ImageOptIndex; i++)
            {
                string cDes = processes[ProcessIndex].BaseImageOpts[i].Description;
                Image_OptNameCMBox.Items.Add(cDes);
                Matrix_OptNameCMBox.Items.Add(cDes);
            }

            if (ocrTool.ReadFile(Description))
            {
                string[] values = ocrTool.imageSelect.Split('_');
                if (values.Length == 2)
                {
                    Image_OptNameCMBox.Text = values[0];
                    Image_ParamNameCMBox.Text = values[1];
                }
                else if (values.Length == 1)
                {
                    Image_OptNameCMBox.Text = values[0];
                }

                values = ocrTool.affinMatrixSelect.Split('_');
                if (values.Length == 2)
                {
                    Matrix_OptNameCMBox.Text = values[0];
                    Matrix_ParamNameCMBox.Text = values[1];
                }
                else if (values.Length == 1)
                {
                    Matrix_OptNameCMBox.Text = values[0];
                }
            }

            propertyGrid1.SelectedObject = ocrTool.ocrParams;
            ShowCharRegionCBox.Checked = ocrTool.isShowCharRegion;
            ShowROICBox.Checked = ocrTool.isShowROI;
            ShowCharCBox.Checked = ocrTool.isShowChar;
            if (ocrTool.ocrParams.ROILength1 != 0 && ocrTool.ocrParams.ROILength2 != 0)
            {
                cWindows1.ADrawROI2d.VR._X = ocrTool.ocrParams.ROICenterColumn;
                cWindows1.ADrawROI2d.VR._Y = ocrTool.ocrParams.ROICenterRow;
                cWindows1.ADrawROI2d.VR._Angle = ocrTool.ocrParams.ROIAngle;
                cWindows1.ADrawROI2d.VR._Length1 = ocrTool.ocrParams.ROILength1;
                cWindows1.ADrawROI2d.VR._Length2 = ocrTool.ocrParams.ROILength2;
            }
            if (cCWindows[ProcessIndex].AImage2d != null)
            {
                //HOperatorSet.CopyObj(cCWindows[ProcessIndex].AImage2d, out HObject objectsSelected, 1, 1);
                //cWindows1.AImage2d = objectsSelected;
                cWindows1.AImage2d = cCWindows[ProcessIndex].AImage2d;
            }
            InitCWImage();
            cWindows1.ADisplay();
        }

        private void InitCWImage()
        {
            #region 界面图像初始化
            string[] imageSelects = ocrTool.imageSelect.Split('_');
            if (imageSelects.Length != 2) return;
            if (imageSelects[0] == "相机")
            {
                if (imageSelects[1] == "灰度图")
                    cWindows1.AImage2d = processes[ProcessIndex].ImageGray;
                else
                    cWindows1.AImage2d = processes[ProcessIndex].ImageHeight;
            }
            else
            {
                ProcessTool.Instance.FindImageOpt(processes[ProcessIndex], imageSelects[0], out BaseImageOpt paramByImageOpt1, out int paramByImageOptIndex1);
                if (paramByImageOptIndex1 >= 0)
                {
                    processes[ProcessIndex].BaseImageOutParams[paramByImageOptIndex1].GetHObjectValue(imageSelects[1], out HObject value);
                    cWindows1.AImage2d = value;
                }
            }
            #endregion
        }

        private void Image_OptNameCMBox_SelectedIndexChanged(object sender, EventArgs e)
        {
            if (Image_OptNameCMBox.SelectedIndex > 0)
            {
                Image_ParamNameCMBox.Items.Clear();
                foreach (string value in processes[ProcessIndex].BaseImageOutParams[Image_OptNameCMBox.SelectedIndex - 1].OutputHObjectNames)
                    Image_ParamNameCMBox.Items.Add(value);
            }
            else if (Image_OptNameCMBox.SelectedIndex == 0)
            {
                Image_ParamNameCMBox.Items.Clear();
                Image_ParamNameCMBox.Items.Add("灰度图");
                Image_ParamNameCMBox.Items.Add("深度图");
            }
            else
            {
                Image_ParamNameCMBox.Items.Clear();
            }
        }

        private void Matrix_OptNameCMBox_SelectedIndexChanged(object sender, EventArgs e)
        {
            if (Matrix_OptNameCMBox.SelectedIndex >= 0)
            {
                Matrix_ParamNameCMBox.Items.Clear();
                foreach (string value in processes[ProcessIndex].BaseImageOutParams[Matrix_OptNameCMBox.SelectedIndex].OutputHTupleNames)
                    Matrix_ParamNameCMBox.Items.Add(value);
            }
        }

        private void SaveROIBtn_Click(object sender, EventArgs e)
        {
            cWindows1.ADrawROI2d.ROIValue();
            ocrTool.ocrParams.ROICenterRow = cWindows1.ADrawROI2d.VR._Y;
            ocrTool.ocrParams.ROICenterColumn = cWindows1.ADrawROI2d.VR._X;
            ocrTool.ocrParams.ROIAngle = cWindows1.ADrawROI2d.VR._Angle;
            ocrTool.ocrParams.ROILength1 = cWindows1.ADrawROI2d.VR._Length1;
            ocrTool.ocrParams.ROILength2 = cWindows1.ADrawROI2d.VR._Length2;
            ocrTool.WriteFile(Description);
        }

        private void SaveBtn_Click(object sender, EventArgs e)
        {

            ocrTool.imageSelect = Image_OptNameCMBox.Text + "_" + Image_ParamNameCMBox.Text;
            ocrTool.affinMatrixSelect = Matrix_OptNameCMBox.Text + "_" + Matrix_ParamNameCMBox.Text;
            ocrTool.isShowChar = ShowCharCBox.Checked;
            ocrTool.isShowCharRegion = ShowCharRegionCBox.Checked;
            ocrTool.isShowROI = ShowROICBox.Checked;
            ocrTool.ocrParams = propertyGrid1.SelectedObject as OCRParams;
            ocrTool.WriteFile(Description);
        }

        private void TestBtn_Click(object sender, EventArgs e)
        {
            InitCWImage();
            ocrTool.Image = cWindows1.AImage2d;
            cWindows1.ADispose();
            dataGridView1.Rows.Clear();
            dataGridView1.Rows.Add(1);
            Stopwatch sw = new Stopwatch();
            sw.Start();
            ocrTool.Running(out HTuple classWord, out HTuple confidence, out HObject rectangle, out HObject wordRegions);
            sw.Stop();
            label3.Text = "总计运行时间：" + sw.ElapsedMilliseconds;
            if (ShowROICBox.Checked)
            {
                CObject cObject = new CObject();
                cObject._Color = "blue";
                cObject._Draw = false;
                cObject._Object = rectangle;
                cWindows1.AObject.Add(cObject);
            }
            if (confidence.Length > 0)
            {
                string code = "";
                if (ShowCharCBox.Checked)
                {
                    CText cText01 = new CText();
                    for (int i = 0; i < classWord.Length; i++)
                    {
                        code += classWord[i].S;
                    }
                    cText01._Text = "Code:" + code.ToString();
                    cText01._X = 100;
                    cText01._Y = 100;
                    cText01._Color = "blue";
                    cWindows1.AText.Add(cText01);
                }
                if (ShowCharRegionCBox.Checked)
                {
                    CObject cObject = new CObject();
                    cObject._Color = "green";
                    cObject._Draw = false;
                    cObject._Object = wordRegions;
                    cWindows1.AObject.Add(cObject);
                }
                dataGridView1.Rows[0].Cells[0].Value = code.ToString();
            }
            cWindows1.ADisplay();
        }

        HObject wordRegions = null;
        HObject rectanges = null;
        private void GetWordBtn_Click(object sender, EventArgs e)
        {
            try
            {
                this.cWindows1.ADispose();
                ocrTool.Image = this.cWindows1.AImage2d;
                this.cWindows1.ADrawROI2d.ROIValue();
                ocrTool.GetWordRegion(out wordRegions, out rectanges);
                CWindowTool.CObject cObject = new CWindowTool.CObject();
                cObject._Color = "green";
                cObject._Object = wordRegions;
                cObject._Draw = false;
                this.cWindows1.AObject.Add(cObject);
                HObject cRect = new HObject();
                for (int i = 1; i <= wordRegions.CountObj(); i++)
                {
                    cRect = wordRegions.SelectObj(i);
                    HOperatorSet.AreaCenter(cRect, out HTuple area, out HTuple row, out HTuple col);
                    CWindowTool.CText cText = new CWindowTool.CText();
                    cText._Color = "blue";
                    cText._Text = i.ToString();
                    cText._X = col;
                    cText._Y = row;
                    this.cWindows1.AText.Add(cText);
                }
                CWindowTool.CObject cObject1 = new CWindowTool.CObject();
                cObject1._Color = "blue";
                cObject1._Object = rectanges;
                cObject1._Draw = false;
                this.cWindows1.AObject.Add(cObject1);
                this.cWindows1.ADisplay();
                this.cWindows1.ADrawROI2d.ROIDelete();
            }
            catch { }
        }

        private void TrainWordBtn_Click(object sender, EventArgs e)
        {
            try
            {
                if (!ocrTool.Train(wordRegions, textBox1.Text.Trim()))
                {
                    MessageBox.Show("训练出错！");
                }

                //if (ocrTool.trainFile == "")
                //    ocrTool.trainFile = test.file_Valuejson.Substring(0, test.file_Valuejson.LastIndexOf('\\') + 1) + System.DateTime.Now.ToString("ocr_yyyyMMddHHmmssff") + ".trf";
                //if (!test.algorithOCR.Train(test.image, test._Value, wordRegion, textBox1.Text.Trim()))
                //{
                //    MessageBox.Show("训练出错！");
                //}
            }
            catch { }
        }

      
    }
}