Display_D_HP/TeamAAS-VM/Views/Home/ShowVisionRender.xaml.cs

using Cognex.VisionPro;
using Cognex.VisionPro.ImageFile;
using Prism.Events;
using Prism.Ioc;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.IO;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Forms.Integration;
using TeamAAS_VP.Core;
using TeamAAS_VP.Enums;
using TeamAAS_VP.Events;
using TeamAAS_VP.Interfaces;
using TeamAAS_VP.Models;
using TeamAAS_VP.Resources.Languages;
using TeamAAS_VP.ViewModels.Home;

namespace TeamAAS_VP.Views.Home
{
    /// <summary>
    /// Interaction logic for ShowVisionRender
    /// </summary>
    public partial class ShowVisionRender : UserControl
    {
        private ShowVisionRenderViewModel viewModel;

        // key=ShowRender.Id
        private readonly Dictionary<Guid, WindowsFormsHost> vmRenders;
        private readonly Dictionary<Guid, TextBlock> Titles;

        private readonly IEventAggregator _eventAggregator;
        private readonly ISystemDatabaseService _systemDatabaseService;
        private readonly IProductService _productService;

        // ======================= 【新增】保存任务控制：防闪退/防资源泄露 =======================
        private CancellationTokenSource _saveImageCts = new CancellationTokenSource();
        private readonly SemaphoreSlim _saveSemaphore = new SemaphoreSlim(2, 2);

        public ShowVisionRender()
        {
            InitializeComponent();

            vmRenders = new Dictionary<Guid, WindowsFormsHost>();
            Titles = new Dictionary<Guid, TextBlock>();

            viewModel = DataContext as ShowVisionRenderViewModel;

            _eventAggregator = viewModel._eventAggregator;
            _systemDatabaseService = ((Prism.PrismApplicationBase)App.Current).Container.Resolve<ISystemDatabaseService>();
            _productService = ((Prism.PrismApplicationBase)App.Current).Container.Resolve<IProductService>();

            viewModel.UpdateLayout += ViewModel_UpdateLayout;
            _eventAggregator.GetEvent<RenderUpdateNotification>().Subscribe(UpdateRenderModuleSource);
            _eventAggregator.GetEvent<ProductChangedNotification>().Subscribe(ProductChanged);

            // 【可选】控件卸载时取消所有保存任务，避免页面关闭后仍在截图/保存导致闪退
            this.Unloaded += ShowVisionRender_Unloaded;
        }

        private void ShowVisionRender_Unloaded(object sender, RoutedEventArgs e)
        {
            try
            {
                _saveImageCts.Cancel();
                _saveImageCts.Dispose();
            }
            catch { }
        }

        /// <summary>
        /// 更新某个工位/模块的显示内容（图像/图形/Record/保存）
        /// </summary>
        private void UpdateRenderModuleSource(ShowRender render)
        {
            this.gridRender.Dispatcher.BeginInvoke(new Action(() =>
            {
                foreach (var item in vmRenders)
                {
                    if (item.Value?.Child is CogRecordDisplay disp)
                    {
                        // 通过 Tag(产品名) 匹配当前需要更新的窗口
                        if (disp.Tag != null && disp.Tag.ToString() == render.ProceductName)
                        {
                            disp.Record = null;
                            disp.Image = render.Image;

                            disp.StaticGraphics.Clear();
                            if (render.Graphic != null)
                                disp.StaticGraphics.AddList(render.Graphic, "");

                            if (render.Record != null)
                                disp.Record = render.Record;

                            // ======================= 【关键优化】保存图像走安全入口 =======================
                            RequestSaveImage(render, disp);

                            break;
                        }
                    }
                }
            }));
        }

        /// <summary>
        /// 【核心】根据 obj.Count 自动生成 1~16 个显示格子（标题+CogRecordDisplay）
        /// </summary>
        private async void ViewModel_UpdateLayout(System.Collections.ObjectModel.ObservableCollection<Models.ShowRender> obj)
        {
            // ======================= 【关键优化】布局刷新前先取消旧保存任务 =======================
            try
            {
                _saveImageCts.Cancel();
                _saveImageCts.Dispose();
            }
            catch { }
            _saveImageCts = new CancellationTokenSource();

            System.Windows.Media.FontFamily font = Application.Current.Resources["DefaultFont"] as System.Windows.Media.FontFamily;

            // 1) 清理旧控件
            if (vmRenders.Count > 0)
            {
                foreach (var render in vmRenders)
                {
                    // WindowsFormsHost.Dispose() 会销毁 WinForms 句柄，防止资源泄漏
                    render.Value?.Dispose();
                }
            }

            vmRenders.Clear();
            Titles.Clear();
            gridRender.Children.Clear();
            gridRender.RowDefinitions.Clear();
            gridRender.ColumnDefinitions.Clear();

            // 2) 判空
            if (obj == null || obj.Count == 0) return;

            // 3) 限制最大 16（防止越界）
            int count = Math.Min(obj.Count, 16);

            // 4) 计算行列（最大 4*4）
            GetGridSize(count, out int rows, out int cols);

            // 5) 创建主Grid的行列
            for (int r = 0; r < rows; r++)
                gridRender.RowDefinitions.Add(new RowDefinition() { Height = new GridLength(1, GridUnitType.Star) });

            for (int c = 0; c < cols; c++)
                gridRender.ColumnDefinitions.Add(new ColumnDefinition() { Width = new GridLength(1, GridUnitType.Star) });

            gridRender.ShowGridLines = true;

            // 6) 逐个创建子格子（每个子格子两行：标题/画面）
            for (int i = 0; i < count; i++)
            {
                var sr = obj[i];

                // (1) 创建显示控件（WindowsFormsHost + CogRecordDisplay）
                var disp = new CogRecordDisplay()
                {
                    Tag = sr.ProceductName,
                };
                //ConfigureCogDisplay(disp);

                var host = new WindowsFormsHost()
                {
                    Tag = sr.ProceductName,
                    Child = disp
                };
                vmRenders.Add(sr.Id, host);

                // (2) 创建标题
                var title = new TextBlock()
                {
                    Text = sr.ProceductName,
                    HorizontalAlignment = System.Windows.HorizontalAlignment.Center,
                    FontWeight = FontWeights.Bold,
                    FontFamily = font,
                    FontSize = 12,
                    Foreground = System.Windows.Media.Brushes.Black
                };
                Titles.Add(sr.Id, title);

                // (3) 子Grid（标题 + 显示）
                var cell = new Grid();
                cell.RowDefinitions.Add(new RowDefinition() { Height = new GridLength(1, GridUnitType.Auto) });  // 标题
                cell.RowDefinitions.Add(new RowDefinition() { Height = new GridLength(1, GridUnitType.Star) });  // 图像

                cell.Children.Add(title);
                cell.Children.Add(host);

                Grid.SetRow(title, 0);
                Grid.SetColumn(title, 0);

                Grid.SetRow(host, 1);
                Grid.SetColumn(host, 0);

                // (4) 放到主 gridRender 中
                int row = i / cols;
                int col = i % cols;

                gridRender.Children.Add(cell);
                Grid.SetRow(cell, row);
                Grid.SetColumn(cell, col);
            }
        }

        /// <summary>
        /// 根据数量返回最合适的 rows/cols（最大 4*4）
        /// </summary>
        private void GetGridSize(int count, out int rows, out int cols)
        {
            rows = 1;
            cols = 1;

            if (count <= 1) { rows = 1; cols = 1; return; }
            if (count <= 2) { rows = 1; cols = 2; return; }
            if (count <= 4) { rows = 2; cols = 2; return; }
            if (count <= 6) { rows = 2; cols = 3; return; }
            if (count <= 9) { rows = 3; cols = 3; return; }
            if (count <= 12) { rows = 3; cols = 4; return; }

            rows = 4;
            cols = 4;
        }

        /// <summary>
        /// 统一配置 VisionPro 显示属性
        /// </summary>
        private void ConfigureCogDisplay(CogRecordDisplay disp)
        {
            disp.HorizontalScrollBar = false;
            disp.VerticalScrollBar = false;
            disp.AutoFit = true;
            disp.BackColor = System.Drawing.SystemColors.ActiveCaption;
        }

        // ======================= 【新增】安全保存入口：解决闪退 =======================
        private void RequestSaveImage(ShowRender render, CogRecordDisplay disp)
        {
            if (render == null) return;
            if (disp == null) return;
            if (!render.IsSaveImage) return;

            // Original 模式不需要 CreateContentBitmap（你的 SaveImage 里 Original 只写 ICogImage）
            bool needRecordedBitmap =
                render.SaveImageModel == ProcedureSaveImageModel.Recorded ||
                render.SaveImageModel == ProcedureSaveImageModel.OriginalAndRecorded;
            _saveImageCts = new CancellationTokenSource();
            var token = _saveImageCts.Token;

            // 必须在控件的 UI 线程上执行 CreateContentBitmap
            disp.BeginInvoke(new Action(() =>
            {
                Bitmap clonedBitmap = null;

                try
                {
                    token.ThrowIfCancellationRequested();

                    if (needRecordedBitmap)
                    {
                        // UI线程创建
                        Bitmap uiBitmap = (Bitmap)disp.CreateContentBitmap(
                            Cognex.VisionPro.Display.CogDisplayContentBitmapConstants.Custom);

                        // 立刻 Clone，后台线程只用 Clone（防止控件刷新/释放导致崩溃）
                        clonedBitmap = (Bitmap)uiBitmap.Clone();

                        // UI线程立刻释放
                        uiBitmap.Dispose();
                    }

                    // 后台保存（限并发，防止GDI/内存瞬时飙升）
                    Task.Run(async () =>
                    {
                        await _saveSemaphore.WaitAsync(token);
                        try
                        {
                            token.ThrowIfCancellationRequested();

                            SaveImage(
                                render.SaveImageModel,
                                render.SaveImagePathModel,
                                render.SavePath,
                                render.Image,
                                clonedBitmap,
                                render.Result,
                                render.IsCompress
                            );
                        }
                        catch (OperationCanceledException)
                        {
                            // 取消属于正常情况，不记录
                        }
                        catch (Exception ex)
                        {
                            LogHelper.WriteLogError("后台保存图像任务出错！", ex);
                        }
                        finally
                        {
                            // 确保释放 Clone 的 Bitmap，避免 GDI 泄漏导致闪退
                            try { clonedBitmap?.Dispose(); } catch { }
                            _saveSemaphore.Release();
                        }
                    }, token);
                }
                catch (OperationCanceledException)
                {
                    try { clonedBitmap?.Dispose(); } catch { }
                }
                catch (Exception ex)
                {
                    try { clonedBitmap?.Dispose(); } catch { }
                    LogHelper.WriteLogError("RequestSaveImage(UI截图阶段) 出错！", ex);
                }
            }));
        }

        /// <summary>
        /// 保存图像（保持你原有逻辑不变，补充 Dispose 兜底，防止压缩分支不释放导致闪退）
        /// </summary>
        private void SaveImage(ProcedureSaveImageModel saveImageModel,
                               ProcedureSaveImagePathModel saveImagePathModel,
                               string path,
                               ICogImage image,
                               Bitmap reimage,
                               bool result,
                               bool isCompress)
        {
            if (string.IsNullOrEmpty(path) || string.IsNullOrWhiteSpace(path) || !Path.IsPathRooted(path))
            {
                _eventAggregator.GetEvent<TaskMessageNotification>()
                    .Publish(new Models.MessageStruct() { Message = "保存图像路径为无效的路径！", level = MessageLevel.Alarm });
                return;
            }

            Task.Run(() =>
            {
                try
                {
                    DateTime now = DateTime.Now;

                    if (saveImageModel == ProcedureSaveImageModel.Original)
                    {
                        string Originalpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\{now:yyyy-MM-dd HH_mm_ss}.bmp";

                        if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_OkNG_Time)
                        {
                            Originalpath = result
                                ? $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp"
                                : $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ok_Time)
                        {
                            if (result)
                                Originalpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            else
                                return;
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ng_Time)
                        {
                            if (!result)
                                Originalpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            else
                                return;
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Formula_ProductSN_Time)
                        {
                            // 当前产品配方名（过滤非法字符）
                            string formulaName = _productService.GetCurrentProduct().Name;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                formulaName = formulaName.Replace(c, '_');

                            // 产品SN（过滤非法字符）
                            string productSN = _productService.CurrentProductCode;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                productSN = productSN.Replace(c, '_');

                            string resultStr = result ? "OK" : "NG";
                            Originalpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\{formulaName}\\{productSN}\\{productSN}-{now:yyyy-MM-dd HH_mm_ss}-{resultStr}.bmp";
                        }

                        if (!Directory.Exists(Path.GetDirectoryName(Originalpath)))
                            Directory.CreateDirectory(Path.GetDirectoryName(Originalpath));

                        using (CogImageFileBMP _cogImageFileTool = new CogImageFileBMP())
                        {
                            _cogImageFileTool.Open(Originalpath, CogImageFileModeConstants.Write);
                            _cogImageFileTool.Append(image);
                            _cogImageFileTool.Close();
                        }
                    }
                    else if (saveImageModel == ProcedureSaveImageModel.Recorded)
                    {
                        string Recordedpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\{now:yyyy-MM-dd HH_mm_ss}.bmp";

                        if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_OkNG_Time)
                        {
                            Recordedpath = result
                                ? $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp"
                                : $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ok_Time)
                        {
                            if (result)
                                Recordedpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            else
                                return;
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ng_Time)
                        {
                            if (!result)
                                Recordedpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            else
                                return;
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Formula_ProductSN_Time)
                        {
                            string formulaName = _productService.GetCurrentProduct().Name;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                formulaName = formulaName.Replace(c, '_');

                            string productSN = _productService.CurrentProductCode;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                productSN = productSN.Replace(c, '_');

                            string resultStr = result ? "OK" : "NG";
                            Recordedpath = $"{path}\\{now:yyyy-MM}\\{now:MM-dd}\\{formulaName}\\{productSN}\\{productSN}-{now:yyyy-MM-dd HH_mm_ss}-{resultStr}.bmp";
                        }

                        if (!Directory.Exists(Path.GetDirectoryName(Recordedpath)))
                            Directory.CreateDirectory(Path.GetDirectoryName(Recordedpath));

                        if (isCompress)
                        {
                            ImageHelper.CompressImage(reimage, Recordedpath, 100);
                        }
                        else
                        {
                            reimage.Save(Recordedpath);
                        }
                    }
                    else if (saveImageModel == ProcedureSaveImageModel.OriginalAndRecorded)
                    {
                        string Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                        string Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\{now:yyyy-MM-dd HH_mm_ss}.bmp";

                        if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_OkNG_Time)
                        {
                            if (result)
                            {
                                Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                                Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            }
                            else
                            {
                                Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                                Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            }
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ok_Time)
                        {
                            if (result)
                            {
                                Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                                Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\OK\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            }
                            else
                            {
                                return;
                            }
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Ng_Time)
                        {
                            if (!result)
                            {
                                Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                                Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\NG\\{now:yyyy-MM-dd HH_mm_ss}.bmp";
                            }
                            else
                            {
                                return;
                            }
                        }
                        else if (saveImagePathModel == ProcedureSaveImagePathModel.Month_Day_Formula_ProductSN_Time)
                        {
                            string formulaName = _productService.GetCurrentProduct().Name;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                formulaName = formulaName.Replace(c, '_');

                            string productSN = _productService.CurrentProductCode;
                            foreach (char c in Path.GetInvalidFileNameChars())
                                productSN = productSN.Replace(c, '_');

                            string resultStr = result ? "OK" : "NG";
                            Originalpath = $"{path}\\Original\\{now:yyyy-MM}\\{now:MM-dd}\\{formulaName}\\{productSN}\\{productSN}-{now:yyyy-MM-dd HH_mm_ss}-{resultStr}.bmp";
                            Recordedpath = $"{path}\\Recored\\{now:yyyy-MM}\\{now:MM-dd}\\{formulaName}\\{productSN}\\{productSN}-{now:yyyy-MM-dd HH_mm_ss}-{resultStr}.bmp";
                        }

                        if (!Directory.Exists(Path.GetDirectoryName(Originalpath)))
                            Directory.CreateDirectory(Path.GetDirectoryName(Originalpath));

                        using (CogImageFileBMP _cogImageFileTool = new CogImageFileBMP())
                        {
                            _cogImageFileTool.Open(Originalpath, CogImageFileModeConstants.Write);
                            _cogImageFileTool.Append(image);
                            _cogImageFileTool.Close();
                        }

                        if (!Directory.Exists(Path.GetDirectoryName(Recordedpath)))
                            Directory.CreateDirectory(Path.GetDirectoryName(Recordedpath));

                        if (isCompress)
                        {
                            ImageHelper.CompressImage(reimage, Recordedpath, 100);
                        }
                        else
                        {
                            reimage.Save(Recordedpath);
                        }
                    }
                }
                catch (Exception ex)
                {
                    LogHelper.WriteLogError("保存流程运行结果的图片时出错！", ex);
                }
                finally
                {
                    // ======================= 【关键修复】无论是否压缩都释放Bitmap =======================
                    try { reimage?.Dispose(); } catch { }
                }
            });
        }

        /// <summary>
        /// 切换产品时清空主页显示（保持你原有逻辑）
        /// </summary>
        private void ProductChanged(ProductModel product)
        {
            this.gridRender.Dispatcher.BeginInvoke(new Action(() =>
            {
                foreach (var item in vmRenders)
                {
                    try
                    {
                        if (item.Value?.Child is CogRecordDisplay disp)
                        {
                            if (disp.Tag != null)
                            {
                                disp.Image = null;
                                disp.StaticGraphics.Clear();
                                // 你原来 break 只清一次；这里保留你的行为
                                break;
                            }
                        }
                    }
                    catch (Exception ex)
                    {
                        LogHelper.WriteLogError(Lang.切换产品清除主页图像时出错, ex);
                    }
                }
            }));
        }
    }
}