TeamAAS-VP-HeShuo/TeamAAS-VM/Core/Cameras/CognexCamera.cs

using Cognex.VisionPro;
using MaterialDesignThemes.Wpf;
using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Diagnostics;
using System.Diagnostics.Contracts;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using TeamAAS_VP.Enums;
using TeamAAS_VP.Interfaces;
using TeamAAS_VP.Models;
using TeamAAS_VP;

namespace TeamAAS_VP.Core.Cameras
{
    public class CognexCamera : ICamera, INotifyPropertyChanged
    {
        #region 字段
        private Thread m_hReceiveThread;
        #endregion

        #region 属性

        public CameraBrand CameraBrand { get => CameraBrand.Cognex; }

        public string Name { get; private set; }
        public Guid ID { get; private set; }
        public string ManufacturerName { get; private set; }

        public string ModelName { get; private set; }

        public string SerialNumber { get; private set; }

        public CameraType CameraType { get; private set; }

        ICogAcqFifo CameraAcq;

        public UInt32 ImageWidth { get; private set; }

        public UInt32 ImageHeight { get; private set; }

        private bool _IsGrabbing;

        /// <summary>
        /// 正在采集
        /// </summary>
        public bool IsGrabbing
        {
            get { return _IsGrabbing; }
            private set { SetProperty(ref _IsGrabbing, value); }
        }

        private ICogImage _Image;

        public ICogImage Image
        {
            get { return _Image; }
            private set { SetProperty(ref _Image, value); }
        }

        public bool IsConnected { get; private set; }

        /// <summary>
        /// 采集用时
        /// </summary>
        public TimeSpan TotalTime { get; private set; }

        /// <summary>
        /// 错误信息
        /// </summary>
        public string ErrorMessage { get; private set; }

        private bool IsHaveCamera = false;
        #endregion

        #region 事件
        public event Action<ICogImage, TimeSpan, string> ImageCallbackEvent;
        public event Action<Guid, bool> CameraConnectChangedEvent;
        #endregion

        public CognexCamera(Guid _ID, string _Name, string _SerialNumber)
        {
            ID = _ID;
            Name = _Name;
            SerialNumber = _SerialNumber;
            //using (CogFrameGrabbers mFrameGrabbers = new CogFrameGrabbers())//识别已安装的帧抓取器的类//帧抓取器:相机
            //{
            //    foreach (ICogFrameGrabber item in mFrameGrabbers)
            //    {
            //        if (item.SerialNumber == SerialNumber)
            //        {
            //            var videoFormat = item.AvailableVideoFormats[0];

            //            foreach (string format in item.AvailableVideoFormats)
            //            {
            //                if (format == "Generic GigEVision (Mono)")
            //                {
            //                    videoFormat = format;
            //                    break;
            //                }

            //                if (format != "Generic GigEVision (Bayer Color)") continue;
            //                videoFormat = format;
            //                break;
            //            }
            //            CameraAcq = item.CreateAcqFifo(videoFormat, CogAcqFifoPixelFormatConstants.Format8Grey, 0,
            //            true);
            //            //CameraAcq.OwnedExposureParams.Exposure = exposure;
            //            //CameraAcq.OwnedBrightnessParams.Brightness = brightness;
            //            //CameraAcq.OwnedContrastParams.Contrast = contrast;
            //            //
            //            CameraAcq.OwnedTriggerParams.TriggerEnabled = true;
            //            CameraAcq.OwnedTriggerParams.TriggerModel = CogAcqTriggerModelConstants.FreeRun;
            //            IsHaveCamera = true;
            //            break;
            //        }

            //    }
            //}

            IsConnected = false;
        }

        #region 方法

        /// <summary>
        /// 获取设备
        /// </summary>
        /// <returns></returns>
        public static CameraInfo[] GetDevices()
        {
            List<CameraInfo> cameras = new List<CameraInfo>();
            if (!CheckSoftwareInstalled().isInstalled)
            {
                LogHelper.WriteLogInfo("未安装Cognex VisionPro");
                return cameras.ToArray();
            }
            using (CogFrameGrabbers mFrameGrabbers = new CogFrameGrabbers())//识别已安装的帧抓取器的类//帧抓取器:相机
            {
                foreach (ICogFrameGrabber item in mFrameGrabbers)
                {
                    string[] bf1 = item.Name.Split(':');
                    string manufacturerName=item.Name;
                    string model = item.Name;
                    CameraType type = CameraType.GIGE;
                    if (bf1.Length==3)
                    {
                        manufacturerName = bf1[1].Trim();
                        model = bf1[2].Trim();
                        if (bf1[0].ToLower().Trim().Contains("gige"))
                        {
                            type = CameraType.GIGE;
                        }
                        else if (bf1[0].ToLower().Trim().Contains("usb"))
                        {
                            type = CameraType.USB;
                        }
                        else
                        {
                            type = CameraType.Unknown;
                        }
                    }
                    cameras.Add(new Models.CameraInfo()
                    {
                        CameraName = "",
                        CameraBrand = CameraBrand.Cognex,
                        CameraType = type,
                        Id = Guid.NewGuid(),
                        ManufacturerName = manufacturerName,
                        Model = model,
                        SerialNumber = item.SerialNumber,
                        CameraIp = "",
                    });
                }
            }
            return cameras.ToArray();
        }

        /// <summary>
        /// 检查相机软件是否安装
        /// </summary>
        /// <returns></returns>
        public static (bool isInstalled, string version) CheckSoftwareInstalled()
        {
            string softwareName = "Cognex VisionPro";
            string softwareVersion = "90.0.5";
            string[] registryPaths = new[]
            {
                @"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall",
                @"SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall"
            };
            foreach (var path in registryPaths)
            {
                using (RegistryKey key = Registry.LocalMachine.OpenSubKey(path))
                {
                    if (key == null) continue;

                    foreach (string subKeyName in key.GetSubKeyNames())
                    {
                        using (RegistryKey subKey = key.OpenSubKey(subKeyName))
                        {
                            string displayName = subKey.GetValue("DisplayName")?.ToString();
                            if (displayName != null && displayName.Contains(softwareName))
                            {
                                string version = subKey.GetValue("DisplayVersion")?.ToString();
                                Version v1 = new Version(version);
                                Version v2 = new Version(softwareVersion);
                                if (v1 >= v2)
                                {
                                    return (true, version);
                                }
                                else
                                {
                                    return (false, version);
                                }
                            }
                        }
                    }
                }
            }
            return (false, "");
        }

        /// <summary>
        /// 打开相机
        /// </summary>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public bool OpenDevice()
        {
            IsConnected = false;
            if (!IsHaveCamera || CameraAcq==null)
            {
                using (CogFrameGrabbers mFrameGrabbers = new CogFrameGrabbers())//识别已安装的帧抓取器的类//帧抓取器:相机
                {
                    foreach (ICogFrameGrabber item in mFrameGrabbers)
                    {
                        if (item.SerialNumber== SerialNumber)
                        {
                            var videoFormat = item.AvailableVideoFormats[0];

                            foreach (string format in item.AvailableVideoFormats)
                            {
                                if (format == "Generic GigEVision (Mono)")
                                {
                                    videoFormat = format;
                                    break;
                                }

                                if (format != "Generic GigEVision (Bayer Color)") continue;
                                videoFormat = format;
                                break;
                            }
                            CameraAcq = item.CreateAcqFifo(videoFormat, CogAcqFifoPixelFormatConstants.Format8Grey, 0,
                            true);
                            //CameraAcq.OwnedExposureParams.Exposure = exposure;
                            //CameraAcq.OwnedBrightnessParams.Brightness = brightness;
                            //CameraAcq.OwnedContrastParams.Contrast = contrast;
                            //
                            CameraAcq.OwnedTriggerParams.TriggerEnabled = true;
                            CameraAcq.OwnedTriggerParams.TriggerModel = CogAcqTriggerModelConstants.FreeRun;
                            string[] bf1 = item.Name.Split(':');
                            string manufacturerName = item.Name;
                            string model = item.Name;
                            CameraType type = CameraType.GIGE;
                            if (bf1.Length == 3)
                            {
                                manufacturerName = bf1[1].Trim();
                                model = bf1[2].Trim();
                                if (bf1[0].ToLower().Trim().Contains("gige"))
                                {
                                    type = CameraType.GIGE;
                                }
                                else if (bf1[0].ToLower().Trim().Contains("usb"))
                                {
                                    type = CameraType.USB;
                                }
                                else
                                {
                                    type = CameraType.Unknown;
                                }
                            }
                            ManufacturerName = manufacturerName;
                            ModelName = model;
                            IsHaveCamera = true;
                            break;
                        }
                        
                    }
                }
                if (!IsHaveCamera)
                {
                    return false;
                }
            }
            if (!IsConnected)
            {
                CameraConnectChangedEvent?.Invoke(ID, true);
            }

            IsConnected = true;
            return IsConnected;
        }

        /// <summary>
        /// 关闭相机
        /// </summary>
        public void CloseDevice()
        {
            if (CameraAcq!=null)
            {
                CameraAcq = null;

            }
            IsConnected = false;
        }

        /// <summary>
        /// 采集图像
        /// </summary>
        /// <returns></returns>
        public ICogImage Grab()
        {
            Stopwatch sw = Stopwatch.StartNew();
            for (int i = 0; i < 5; i++)
            {
                try
                {
                    if (CameraAcq == null)
                    {
                        OpenDevice();
                    }

                    CameraAcq.Flush();
                    Image= CameraAcq.CompleteAcquireEx(new CogAcqInfo());
                    ImageWidth = (uint)Image.Width;
                    ImageHeight = (uint)Image.Height;
                    TotalTime = sw.Elapsed;
                    ErrorMessage = "";
                    return Image;
                }
                catch (Exception ex)
                {
                    ErrorMessage = ex.Message;
                    CloseDevice();
                    Thread.Sleep(100);
                    OpenDevice();
                }
            }
            if (IsConnected)
            {
                IsConnected = false;
                CameraConnectChangedEvent?.Invoke(ID, false);
            }
            return null;
        }

        /// <summary>
        /// 开始采集图像
        /// </summary>
        public void StartGrabbing()
        {
            if (IsGrabbing)
                return;
            IsGrabbing = true;
            m_hReceiveThread = new Thread(GetStreamThreadProc) { IsBackground = true };
            m_hReceiveThread.Start();
        }

        /// <summary>
        /// 停止采集图像
        /// </summary>
        public void StopGrabbing()
        {
            try
            {
                IsGrabbing = false;
                Thread.Sleep(1000);
                if (m_hReceiveThread != null)
                {
                    m_hReceiveThread.Abort();
                    m_hReceiveThread = null;
                }
            }
            catch (Exception)
            {

            }
        }

        /// <summary>
        /// 设置曝光时间
        /// </summary>
        /// <param name="ExposureTime"></param>
        /// <returns></returns>
        public bool SetExposureTime(float ExposureTime)
        {
            if (CameraAcq==null)
                return false;
            CameraAcq.OwnedExposureParams.Exposure = (double)ExposureTime;
            return true;
        }

        /// <summary>
        /// 获取曝光时间
        /// </summary>
        /// <returns></returns>
        public float GetExposureTime()
        {
            if (CameraAcq == null)
                return 0;
            return (float)CameraAcq.OwnedExposureParams.Exposure;
        }

        /// <summary>
        /// 设置增益
        /// </summary>
        /// <param name="Gain"></param>
        /// <returns></returns>
        public bool SetGain(float Gain)
        {
            if (CameraAcq==null)
                return false;
            // 设置增益 
            CameraAcq.OwnedBrightnessParams.Brightness = (double)Gain;
            return true;
        }

        /// <summary>
        /// 获取增益
        /// </summary>
        /// <returns></returns>
        public float GetGain()
        {
            if (CameraAcq == null)
                return 0;
            return (float)(CameraAcq.OwnedBrightnessParams.Brightness);
        }

        private void GetStreamThreadProc()
        {
            while (IsGrabbing)
            {
                try
                {
                    Grab();
                    
                }
                catch (Exception)
                {

                }
                ImageCallbackEvent?.Invoke(Image, TotalTime, ErrorMessage);
                Thread.Sleep(10);
            }
            IsGrabbing = false;
        }
        #endregion

        #region 属性通知
        /// <summary>
        /// Occurs when a property value changes.
        /// </summary>
        public event PropertyChangedEventHandler PropertyChanged;

        /// <summary>
        /// Checks if a property already matches a desired value. Sets the property and
        /// notifies listeners only when necessary.
        /// </summary>
        /// <typeparam name="T">Type of the property.</typeparam>
        /// <param name="storage">Reference to a property with both getter and setter.</param>
        /// <param name="value">Desired value for the property.</param>
        /// <param name="propertyName">Name of the property used to notify listeners. This
        /// value is optional and can be provided automatically when invoked from compilers that
        /// support CallerMemberName.</param>
        /// <returns>True if the value was changed, false if the existing value matched the
        /// desired value.</returns>
        protected virtual bool SetProperty<T>(ref T storage, T value, [CallerMemberName] string propertyName = null)
        {
            if (EqualityComparer<T>.Default.Equals(storage, value)) return false;

            storage = value;
            RaisePropertyChanged(propertyName);

            return true;
        }

        /// <summary>
        /// Checks if a property already matches a desired value. Sets the property and
        /// notifies listeners only when necessary.
        /// </summary>
        /// <typeparam name="T">Type of the property.</typeparam>
        /// <param name="storage">Reference to a property with both getter and setter.</param>
        /// <param name="value">Desired value for the property.</param>
        /// <param name="propertyName">Name of the property used to notify listeners. This
        /// value is optional and can be provided automatically when invoked from compilers that
        /// support CallerMemberName.</param>
        /// <param name="onChanged">Action that is called after the property value has been changed.</param>
        /// <returns>True if the value was changed, false if the existing value matched the
        /// desired value.</returns>
        protected virtual bool SetProperty<T>(ref T storage, T value, Action onChanged, [CallerMemberName] string propertyName = null)
        {
            if (EqualityComparer<T>.Default.Equals(storage, value)) return false;

            storage = value;
            onChanged?.Invoke();
            RaisePropertyChanged(propertyName);

            return true;
        }

        /// <summary>
        /// Raises this object's PropertyChanged event.
        /// </summary>
        /// <param name="propertyName">Name of the property used to notify listeners. This
        /// value is optional and can be provided automatically when invoked from compilers
        /// that support <see cref="CallerMemberNameAttribute"/>.</param>
        protected void RaisePropertyChanged([CallerMemberName] string propertyName = null)
        {
            OnPropertyChanged(new PropertyChangedEventArgs(propertyName));
        }

        /// <summary>
        /// Raises this object's PropertyChanged event.
        /// </summary>
        /// <param name="args">The PropertyChangedEventArgs</param>
        protected virtual void OnPropertyChanged(PropertyChangedEventArgs args)
        {
            PropertyChanged?.Invoke(this, args);
        }
        #endregion
    }
}