From dbae9c048fa0cd67c2e1161e5b6b693f87064154 Mon Sep 17 00:00:00 2001
From: wells.liu <wells.liu@broconcentric.com>
Date: 星期三, 01 七月 2020 17:53:15 +0800
Subject: [PATCH] Merge branch 'master' of http://gitblit.broconcentric.com:8088/r/M071

---
 src/Bro.Device.Gocator/GocatorDriver.cs |  263 ++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 220 insertions(+), 43 deletions(-)

diff --git a/src/Bro.Device.Gocator/GocatorDriver.cs b/src/Bro.Device.Gocator/GocatorDriver.cs
index bdfac10..39fb4a2 100644
--- a/src/Bro.Device.Gocator/GocatorDriver.cs
+++ b/src/Bro.Device.Gocator/GocatorDriver.cs
@@ -1,16 +1,21 @@
 锘縰sing Bro.Common.Base;
 using Bro.Common.Helper;
 using Bro.Common.Interface;
+using HalconDotNet;
 using Lmi3d.GoSdk;
 using Lmi3d.GoSdk.Messages;
 using Lmi3d.Zen;
 using Lmi3d.Zen.Io;
 using System;
 using System.Collections.Generic;
+using System.Drawing;
+using System.IO;
 using System.Linq;
 using System.Runtime.InteropServices;
 using System.Text;
+using System.Threading;
 using System.Threading.Tasks;
+using static Bro.Common.Helper.EnumHelper;
 
 namespace Bro.Device.Gocator
 {
@@ -30,14 +35,23 @@
 
         public override void Snapshot()
         {
+            GoDataSet dataSet = null;
             if (IIConfig.IsAsyncMode)
             {
+                if (!IIConfig.IsHardwareTrigger)
+                {
+                    _snapFlag = true;
+                    if (!_snapHandle.WaitOne(IIConfig.SnapshotTimeout))
+                    {
+                        LogAsync(DateTime.Now, $"{Name}鑾峰彇鍥惧儚瓒呮椂", "");
+                        return;
+                    }
 
+                    dataSet = _currentData;
+                }
             }
             else
             {
-                GoDataSet dataSet = null;
-
                 try
                 {
                     dataSet = system.ReceiveData(IIConfig.SnapshotTimeout);
@@ -47,33 +61,23 @@
                     LogAsync(DateTime.Now, $"{Name}鑾峰彇鍥惧儚寮傚父", ex.GetExceptionMessage());
                     return;
                 }
-
-                HandleGoData(dataSet);
             }
+
+            HandleGoData(dataSet);
         }
 
-        private void HandleGoData(GoDataSet dataSet)
+        private void HandleGoData(GoDataSet dataSet, GoImageSet imgSet = null)
         {
             if (dataSet == null)
+            {
                 return;
+            }
 
             for (UInt32 i = 0; i < dataSet.Count; i++)
             {
                 GoDataMsg dataObj = (GoDataMsg)dataSet.Get(i);
                 switch (dataObj.MessageType)
                 {
-                    //case GoDataMessageType.Stamp:
-                    //    {
-                    //        GoStampMsg stampMsg = (GoStampMsg)dataObj;
-                    //        for (UInt32 j = 0; j < stampMsg.Count; j++)
-                    //        {
-                    //            GoStamp stamp = stampMsg.Get(j);
-                    //            Console.WriteLine("Frame Index = {0}", stamp.FrameIndex);
-                    //            Console.WriteLine("Time Stamp = {0}", stamp.Timestamp);
-                    //            Console.WriteLine("Encoder Value = {0}", stamp.Encoder);
-                    //        }
-                    //    }
-                    //    break;
                     case GoDataMessageType.Surface:
                         {
                             GoSurfaceMsg surfaceMsg = (GoSurfaceMsg)dataObj;
@@ -82,43 +86,83 @@
                             long bufferSize = width * height;
                             IntPtr bufferPointer = surfaceMsg.Data;
 
-                            //Console.WriteLine("Whole Part Height Map received:");
-                            //Console.WriteLine(" Buffer width: {0}", width);
-                            //Console.WriteLine(" Buffer height: {0}", height);
+                            if (imgSet != null)
+                            {
+                                imgSet.HImage = new HImage();
+                                imgSet.HImage.GenImage1("uint2", (int)width, (int)height, bufferPointer);
 
-                            //short[] ranges = new short[bufferSize];
-                            //Marshal.Copy(bufferPointer, ranges, 0, ranges.Length);
+                                imgSet.HImage_2 = new HImage();
+                                imgSet.HImage_2.GenImage1("uint2", (int)width, (int)height, bufferPointer);
+                            }
 
-                            Generate16GrayImageByPointer((int)(width / 1), (int)(height / 1), bufferPointer, "");
+                            Generate16GrayImageByPointer((int)width, (int)height, bufferPointer, imgSet?.Id);
                         }
                         break;
-                        //case GoDataMessageType.SurfaceIntensity:
-                        //    {
-                        //        GoSurfaceIntensityMsg surfaceMsg = (GoSurfaceIntensityMsg)dataObj;
-                        //        long width = surfaceMsg.Width;
-                        //        long height = surfaceMsg.Length;
-                        //        long bufferSize = width * height;
-                        //        IntPtr bufferPointeri = surfaceMsg.Data;
-
-                        //        //Console.WriteLine("Whole Part Intensity Image received:");
-                        //        //Console.WriteLine(" Buffer width: {0}", width);
-                        //        //Console.WriteLine(" Buffer height: {0}", height);
-                        //        //byte[] ranges = new byte[bufferSize];
-                        //        //Marshal.Copy(bufferPointeri, ranges, 0, ranges.Length);
-                        //        //GenerateGrayImageByPointer((int)width, (int)height, bufferPointeri, "");
-                        //    }
-                        //    break;
                 }
             }
         }
 
-        public override ImageSet Snapshot(IOperationConfig config)
+        public override IImageSet Snapshot(IOperationConfig config)
         {
-            ImageSet imgSet = base.Snapshot(config);
+            if (!IIConfig.IsAsyncMode)
+            {
+                if (config is GocatorOperationConfig opConfig)
+                {
+                    if (opConfig.IsOpenConnection)
+                    {
+                        system.Start();
+                    }
+                    else
+                    {
+                        system.Stop();
+                    }
+
+                    if (!opConfig.IsSnapshotAction)
+                    {
+                        return null;
+                    }
+                }
+            }
+
+            GoImageSet imgSet = new GoImageSet();
+            InitialImageSet(imgSet, (config as CameraOprerationConfigBase).ImageSaveOption);
+
+            GoDataSet dataSet = null;
+            if (IIConfig.IsAsyncMode)
+            {
+                if (!IIConfig.IsHardwareTrigger)
+                {
+                    _snapFlag = true;
+                    if (!_snapHandle.WaitOne(IIConfig.SnapshotTimeout))
+                    {
+                        LogAsync(DateTime.Now, $"{Name}鑾峰彇鍥惧儚瓒呮椂", "");
+                        return null;
+                    }
+
+                    dataSet = _currentData;
+                }
+            }
+            else
+            {
+                try
+                {
+                    dataSet = system.ReceiveData(IIConfig.SnapshotTimeout);
+                }
+                catch (Exception ex)
+                {
+                    LogAsync(DateTime.Now, $"{Name}鑾峰彇鍥惧儚寮傚父", ex.GetExceptionMessage());
+                    return null;
+                }
+            }
+
+            HandleGoData(dataSet, imgSet);
+            if (imgSet.HImage == null)
+            {
+                LogAsync(DateTime.Now, $"{Name}鏈兘鑾峰彇HImage鍥惧儚", "");
+            }
 
             return imgSet;
         }
-
 
         float _currentExposure = 0;
         string _currentJob = "";
@@ -193,7 +237,15 @@
             _currentJob = sensor.DefaultJob;
             if (!string.IsNullOrWhiteSpace(IIConfig.DefaultJob) && _currentJob != IIConfig.DefaultJob)
             {
-                _currentJob = sensor.DefaultJob = IIConfig.DefaultJob;
+                //_currentJob = sensor.DefaultJob = IIConfig.DefaultJob;
+                string currentJob = IIConfig.DefaultJob;
+                bool isChanged = false;
+                sensor.LoadedJob(ref currentJob, ref isChanged);
+
+                if (!isChanged)
+                {
+                    throw new ProcessException($"{Name}鏈垚鍔熷垏鎹㈣嚦浠诲姟{IIConfig.DefaultJob}");
+                }
             }
 
             sensor.Flush();
@@ -240,6 +292,10 @@
             get => InitialConfig as GocatorInitialConfig;
         }
 
+        volatile bool _snapFlag = false;
+        readonly ManualResetEvent _snapHandle = new ManualResetEvent(false);
+
+        GoDataSet _currentData = null;
         /// <summary>
         /// 寮傛妯″紡鑾峰彇鏁版嵁
         /// </summary>
@@ -247,6 +303,127 @@
         private void onData(KObject data)
         {
             GoDataSet dataSet = (GoDataSet)data;
+
+            if (IIConfig.IsHardwareTrigger)
+            {
+                GoImageSet imgSet = new GoImageSet();
+                InitialImageSet(imgSet, IConfig.ImageSaveOption);
+
+                HandleGoData(dataSet, imgSet);
+
+                if (imgSet.HImage != null)
+                {
+                    OnHImageOutput?.BeginInvoke(this, imgSet.HImage, imgSet.Id, null, null);
+                }
+                else
+                {
+                    LogAsync(DateTime.Now, $"{Name}鑾峰彇HImage澶辫触", "");
+                }
+            }
+            else
+            {
+                if (_snapFlag)
+                {
+                    _snapFlag = false;
+
+                    _currentData = dataSet.Clone<GoDataSet>();
+                    _snapHandle.Set();
+                }
+            }
+        }
+
+        #region 閲嶅啓鍥剧墖淇濆瓨鎿嶄綔
+        public override async void SaveOriginImage(Bitmap map, Bitmap showImage, string imgSetId)
+        {
+            Task.Run(() =>
+            {
+                OnImageUpdated?.Invoke(this, showImage, imgSetId);
+            });
+
+            GoImageSet set = _imageSetList.FirstOrDefault(u => u.Id == imgSetId) as GoImageSet;
+
+            if (set == null)
+                return;
+
+            set.Image = map;
+
+            await Task.Run(() =>
+            {
+                lock (set.SaveLock)
+                {
+                    if (set.ImageSaveOption.IsSaveOriginImage && !set.IsOriginSaved)
+                    {
+                        string imgDir = CheckImageDirectory(set.ImageSaveOption.ImageSaveSubDirectory, "Origin");
+
+                        try
+                        {
+                            //SaveImageByNameAndType(map, set.Id, set.ImageSaveOption.ImageFormat, imgDir);
+                            string filePath = Path.Combine(imgDir, $"{set.Id}.tif");
+                            set.HImage_2.WriteImage("tiff", 0, filePath);
+                        }
+                        catch (Exception ex)
+                        {
+                            LogAsync(DateTime.Now, $"{set.Id}鍘熷浘淇濆瓨澶辫触", ex.GetExceptionMessage());
+                        }
+                    }
+                    set.IsOriginSaved = true;
+                    ClearImageSet(set);
+                }
+            });
+        }
+
+        /// <summary>
+        /// 淇濆瓨鐗瑰畾绫诲瀷鍥剧墖锛屼緥濡侼G鍥剧墖
+        /// </summary>
+        /// <param name="prefix">鍥剧墖绫诲瀷璇存槑 渚嬪鈥淣G鈥�</param>
+        /// <param name="imgSetId">鍥剧墖ID</param>
+        public override async void SaveSelectedImage(string prefix, string imgSetId)
+        {
+            await Task.Run(() =>
+            {
+                GoImageSet set = _imageSetList.FirstOrDefault(u => u.Id == imgSetId) as GoImageSet;
+                if (set == null)
+                    return;
+
+                lock (set.SaveLock)
+                {
+                    if (set.ImageSaveOption.AddtionalSaveType.Contains(prefix) && !set.IsAddtionalSaved)
+                    {
+                        string imgDir = CheckImageDirectory(set.ImageSaveOption.ImageSaveSubDirectory, prefix);
+
+                        while (set.Image == null)
+                        {
+                            Thread.Sleep(50);
+                        }
+
+                        try
+                        {
+                            string filePath = Path.Combine(imgDir, $"{set.Id}.tif");
+                            set.HImage_2.WriteImage("tiff", 0, filePath);
+                        }
+                        catch (Exception ex)
+                        {
+                            LogAsync(DateTime.Now, $"{set.Id}{prefix}鍥剧墖淇濆瓨澶辫触", ex.GetExceptionMessage());
+                        }
+                    }
+                    set.IsAddtionalSaved = true;
+                    ClearImageSet(set);
+                }
+            });
+        }
+        #endregion
+    }
+
+    public class GoImageSet : ImageSet
+    {
+        public HImage HImage_2 { get; set; }
+
+        public override void Dispose()
+        {
+            base.Dispose();
+
+            HImage_2?.Dispose();
+            HImage_2 = null;
         }
     }
 }

--
Gitblit v1.8.0