From 2f12bfa82913656dd546858937159cf3fc4b6ead Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=89=BE=E7=AB=B9?= <akwkevin@126.com>
Date: Sun, 3 Sep 2023 16:48:56 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=87=A0=E4=B8=AA=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs  | 11 +++++++++++
 .../Services/IUIVisualizerService.cs                  |  2 +-
 .../Services/WPFUIVisualizerService.cs                |  2 +-
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs b/AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs
index 5d361b4..92d7c75 100644
--- a/AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs
+++ b/AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections;
 using System.Drawing;
 using System.Drawing.Imaging;
 using System.IO;
@@ -316,5 +317,15 @@ namespace AIStudio.Wpf.DiagramDesigner
                 return null;
             }
         }
+
+        public static Stream CopyStream(this Stream stream)
+        {
+            byte[] bytearray = new byte[stream.Length];
+            stream.Seek(0, SeekOrigin.Begin);
+            stream.Read(bytearray, 0, bytearray.Length);
+
+            var destination = new MemoryStream(bytearray);
+            return destination;
+        }
     }
 }
diff --git a/AIStudio.Wpf.DiagramDesigner/Services/IUIVisualizerService.cs b/AIStudio.Wpf.DiagramDesigner/Services/IUIVisualizerService.cs
index 375dc04..5025f5f 100644
--- a/AIStudio.Wpf.DiagramDesigner/Services/IUIVisualizerService.cs
+++ b/AIStudio.Wpf.DiagramDesigner/Services/IUIVisualizerService.cs
@@ -17,7 +17,7 @@ namespace AIStudio.Wpf.DiagramDesigner.Services
         /// </summary>
         /// <param name="dataContextForPopup">Object state to associate with the dialog</param>
         /// <returns>True/False if UI is displayed.</returns>
-        bool? ShowDialog(object dataContextForPopup);
+        bool? ShowDialog(object dataContextForPopup, bool buttonVisable = true);
 
         /// <summary>
         /// This method displays a modal dialog associated with the given key.
diff --git a/AIStudio.Wpf.DiagramDesigner/Services/WPFUIVisualizerService.cs b/AIStudio.Wpf.DiagramDesigner/Services/WPFUIVisualizerService.cs
index b5c9058..5f209e0 100644
--- a/AIStudio.Wpf.DiagramDesigner/Services/WPFUIVisualizerService.cs
+++ b/AIStudio.Wpf.DiagramDesigner/Services/WPFUIVisualizerService.cs
@@ -12,7 +12,7 @@ namespace AIStudio.Wpf.DiagramDesigner.Services
         /// </summary>
         /// <param name="dataContextForPopup">Object state to associate with the dialog</param>
         /// <returns>True/False if UI is displayed.</returns>
-        public bool? ShowDialog(object dataContextForPopup)
+        public bool? ShowDialog(object dataContextForPopup, bool buttonVisable = true)
         {
             Window win = new PopupWindow();
             win.DataContext = dataContextForPopup;