using System;
using System.Collections.Generic;
using System.Drawing;
using System.Drawing.Imaging;
using Asa.CameraFactory;

namespace Asa.Camera
{
    /// <summary>
    /// 视觉库
    /// </summary>
    public class VisionLib : IDisposable
    {
        private ICamera[] cameraAll;
        private Barcode.Identify identify;
        private List<Region.Feature> feature;
        private int featureIdx = -1;
        private int typeIdx = -1;
        private int cameraIdx = -1;

        /// <summary>
        /// 视觉库
        /// </summary>
        /// <param name="configPath"></param>
        /// <param name="logName"></param>
        public VisionLib(string configPath, string logName = "VisionLib")
        {
            if (!string.IsNullOrWhiteSpace(logName))
                Log.Load(logName);

            cameraAll = new ICamera[3];
            cameraAll[0] = new HIK.IPCamera(configPath, "");
            cameraAll[1] = new HIK.VisionCamera("");
            cameraAll[2] = new Basler.VisionCamera("");
            identify = new Barcode.Identify(configPath, "");
            feature = new();

            string s = "Load Camera";
            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].Load();
                Count += cameraAll[i].Count;
                string[] name = cameraAll[i].Name;
                for (int j = 0; j < name.Length; j++)
                    feature.Add(new Region.Feature(configPath, name[j], ""));
                s += ", " + cameraAll[i].GetType().FullName + ":" + rtn.ToString();
            }
            Log.Info(s);
        }

        /// <summary>
        /// 释放
        /// </summary>
        public void Dispose()
        {
            for (int i = 0; i < cameraAll.Length; i++)
                cameraAll[i].Dispose();
            Log.Info("Dispose Camera");
        }

        /// <summary>
        /// 相机数量
        /// </summary>
        public int Count { private set; get; } = 0;

        /// <summary>
        /// 获取相机名称
        /// </summary>
        /// <returns></returns>
        public string[] GetCameraName()
        {
            List<string> name = new();
            for (int i = 0; i < cameraAll.Length; i++)
                name.AddRange(cameraAll[i].Name);
            return name.ToArray();
        }

        /// <summary>
        /// 打开
        /// </summary>
        public void Open()
        {
            string s = "Open Camera";
            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].OpenAll();
                s += ", " + cameraAll[i].GetType().FullName + ":" + rtn.ToString();
            }
            Log.Info(s);
        }

        /// <summary>
        /// 关闭
        /// </summary>
        public void Close()
        {
            for (int i = 0; i < cameraAll.Length; i++)
                cameraAll[i].CloseAll();
            Log.Info("Close Camera");
        }

        /// <summary>
        /// 获取当前图像
        /// </summary>
        /// <returns></returns>
        public Bitmap[] GetImage()
        {
            List<Bitmap> result = new();
            for (int i = 0; i < cameraAll.Length; i++)
            {
                if (cameraAll[i].Count == 0) continue;
                bool rtn = cameraAll[i].GrabOne(out Bitmap[] bmp);
                if (rtn) result.AddRange(bmp);
            }
            return result.ToArray();
        }

        /// <summary>
        /// 获取当前图像
        /// </summary>
        /// <param name="cameraName"></param>
        /// <returns></returns>
        public Bitmap GetImage(string cameraName)
        {
            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].GrabOne(cameraName, out Bitmap bmp);
                if (rtn) return bmp;
            }
            return null;
        }

        /// <summary>
        /// 保存当前图像
        /// </summary>
        /// <param name="path">文件夹路径</param>
        /// <param name="fileName">文件名，不带扩展名</param>
        /// <param name="format">图像类型</param>
        public void SaveImage(string path, string fileName, ImageFormat format)
        {
            if (!System.IO.Directory.Exists(path)) return;
            string filePath = path.EndsWith("\\") ? path : path + "\\";
            filePath += fileName + "_{0}." + format.ToString();

            for (int i = 0; i < cameraAll.Length; i++)
            {
                if (cameraAll[i].Count == 0) continue;
                bool rtn = cameraAll[i].GrabOne(out Bitmap[] bmp);
                if (!rtn) continue;
                for (int j = 0; j < bmp.Length; j++)
                {
                    string filename = string.Format(filePath, j + 1);
                    bmp[j].Save(filename, format);
                }
            }
        }

        /// <summary>
        /// 保存当前图像
        /// </summary>
        /// <param name="cameraName">相机名称</param>
        /// <param name="path">文件夹路径</param>
        /// <param name="fileName">文件名，不带扩展名</param>
        /// <param name="format">图像类型</param>
        public void SaveImage(string cameraName, string path, string fileName, ImageFormat format)
        {
            if (!System.IO.Directory.Exists(path)) return;
            string filePath = path.EndsWith("\\") ? path : path + "\\";
            filePath += fileName + "." + format.ToString();

            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].GrabOne(cameraName, out Bitmap bmp);
                if (rtn)
                {
                    bmp.Save(filePath, format);
                    break;
                }
            }
        }






        /// <summary>
        /// 从相机图像获取条码
        /// </summary>
        /// <param name="image"></param>
        /// <returns></returns>
        public string[] GetBarCodeText(out List<Bitmap> image)
        {
            List<Barcode.CodeInfo> info = new();
            image = new();

            for (int i = 0; i < cameraAll.Length; i++)
            {
                if (cameraAll[i].Count == 0) continue;
                bool rtn = cameraAll[i].GrabOne(out Bitmap[] bmp);
                if (!rtn) continue;
                image.AddRange(bmp);

                for (int j = 0; j < bmp.Length; j++)
                    info.AddRange(identify.GetCode(bmp[j]));
            }

            List<string> code = new();
            for (int i = 0; i < info.Count; i++)
                code.Add(info[i].Text);

            return code.ToArray();
        }

        /// <summary>
        /// 从相机图像获取条码
        /// </summary>
        /// <param name="cameraName"></param>
        /// <param name="image"></param>
        /// <returns></returns>
        public string[] GetBarCodeText(string cameraName, out Bitmap image)
        {
            List<Barcode.CodeInfo> info = new();
            image = null;

            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].GrabOne(cameraName, out image);
                if (rtn)
                {
                    info.AddRange(identify.GetCode(image));
                    break;
                }
            }

            List<string> code = new();
            for (int i = 0; i < info.Count; i++)
                code.Add(info[i].Text);

            return code.ToArray();
        }

        /// <summary>
        /// 从给定的图像中获取条码文本
        /// </summary>
        /// <param name="bmp"></param>
        /// <returns></returns>
        public string[] GetBarCodeText(Bitmap bmp)
        {
            List<Barcode.CodeInfo> info = identify.GetCode(bmp);
            List<string> code = new();

            for (int i = 0; i < info.Count; i++)
                code.Add(info[i].Text);

            return code.ToArray();
        }

        /// <summary>
        /// 从给定的条码中获取条码文本
        /// </summary>
        /// <param name="info"></param>
        /// <returns></returns>
        public string[] GetBarCodeText(List<Barcode.CodeInfo> info)
        {
            List<string> code = new();
            for (int i = 0; i < info.Count; i++)
                code.Add(info[i].Text);
            return code.ToArray();
        }

        /// <summary>
        /// 从相机图像获取条码
        /// </summary>
        /// <param name="image"></param>
        /// <returns></returns>
        public List<Barcode.CodeInfo> GetBarCode(out List<Bitmap> image)
        {
            List<Barcode.CodeInfo> info = new();
            image = new();

            for (int i = 0; i < cameraAll.Length; i++)
            {
                if (cameraAll[i].Count == 0) continue;
                bool rtn = cameraAll[i].GrabOne(out Bitmap[] bmp);
                if (!rtn) continue;
                image.AddRange(bmp);

                for (int j = 0; j < bmp.Length; j++)
                    info.AddRange(identify.GetCode(bmp[j]));
            }

            return info;
        }

        /// <summary>
        /// 从相机图像获取条码
        /// </summary>
        /// <param name="cameraName"></param>
        /// <param name="image"></param>
        /// <returns></returns>
        public List<Barcode.CodeInfo> GetBarCode(string cameraName, out Bitmap image)
        {
            List<Barcode.CodeInfo> info = new();
            image = null;

            for (int i = 0; i < cameraAll.Length; i++)
            {
                bool rtn = cameraAll[i].GrabOne(cameraName, out image);
                if (rtn)
                {
                    info.AddRange(identify.GetCode(image));
                    break;
                }
            }

            return info;
        }

        /// <summary>
        /// 从给定的图像中获取条码
        /// </summary>
        /// <param name="bmp"></param>
        /// <returns></returns>
        public List<Barcode.CodeInfo> GetBarCode(Bitmap bmp)
        {
            List<Barcode.CodeInfo> info = identify.GetCode(bmp);
            return info;
        }

        /// <summary>
        /// 特征原始图
        /// </summary>
        /// <returns></returns>
        public bool FeatureOriginal()
        {
            bool rtn = false;
            int idx = -1;

            for (int i = 0; i < cameraAll.Length; i++)
            {
                for (int j = 0; j < cameraAll[i].Count; j++)
                {
                    idx++;
                    rtn = cameraAll[i].GrabOne(j, out Bitmap bmp);
                    if (!rtn) continue;
                    rtn = feature[idx].OriginalImage(bmp);
                    if (rtn)
                    {
                        typeIdx = rtn ? i : -1;
                        cameraIdx = rtn ? j : -1;
                        featureIdx = idx;
                        break;
                    }
                }
                if (rtn) break;
            }
            return rtn;
        }

        /// <summary>
        /// 特征原始图
        /// </summary>
        /// <param name="cameraName"></param>
        /// <returns></returns>
        public bool FeatureOriginal(string cameraName)
        {
            bool rtn = false;

            for (int i = 0; i < cameraAll.Length; i++)
            {
                rtn = cameraAll[i].GrabOne(cameraName, out Bitmap bmp);
                if (rtn)
                {
                    int index = feature.FindIndex(match => match.CameraName == cameraName);
                    if (index == -1) return false;
                    rtn = feature[index].OriginalImage(bmp);
                    return rtn;
                }
            }
            return rtn;
        }

        /// <summary>
        /// 特征比较图
        /// </summary>
        /// <param name="exist"></param>
        /// <returns></returns>
        public bool FeatureCompare(out Dictionary<string, bool> exist)
        {
            exist = null;
            if (typeIdx == -1 || cameraIdx == -1) return false;
            bool rtn = cameraAll[typeIdx].GrabOne(cameraIdx, out Bitmap bmp);
            if (rtn) rtn = feature[featureIdx].CompareImage(bmp, out exist);
            return rtn;
        }

        /// <summary>
        /// 特征比较图
        /// </summary>
        /// <param name="cameraName"></param>
        /// <param name="exist"></param>
        /// <returns></returns>
        public bool FeatureCompare(string cameraName, out Dictionary<string, bool> exist)
        {
            exist = null;
            bool rtn = false;

            for (int i = 0; i < cameraAll.Length; i++)
            {
                rtn = cameraAll[i].GrabOne(cameraName, out Bitmap bmp);
                if (rtn)
                {
                    int index = feature.FindIndex(match => match.CameraName == cameraName);
                    if (index == -1) return false;
                    rtn = feature[index].CompareImage(bmp, out exist);
                    return rtn;
                }
            }
            return rtn;
        }

    }
}