From 661b235f577a84903d4592d160c5e8f72f328b18 Mon Sep 17 00:00:00 2001
From: kwai <kwai@KWAI-NET>
Date: Thu, 20 Jul 2023 19:12:43 +0800
Subject: [PATCH] block

---
 .../Container/BlockItemsContainerInfo.cs        | 12 ++++++++++++
 .../BlockDesignerItemViewModel.cs               | 17 +++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/AIStudio.Wpf.DiagramDesigner/ViewModels/BaseViewModel/Container/BlockItemsContainerInfo.cs b/AIStudio.Wpf.DiagramDesigner/ViewModels/BaseViewModel/Container/BlockItemsContainerInfo.cs
index b9d4b3e..de11411 100644
--- a/AIStudio.Wpf.DiagramDesigner/ViewModels/BaseViewModel/Container/BlockItemsContainerInfo.cs
+++ b/AIStudio.Wpf.DiagramDesigner/ViewModels/BaseViewModel/Container/BlockItemsContainerInfo.cs
@@ -317,5 +317,17 @@ namespace AIStudio.Wpf.DiagramDesigner
 
             return containBound;
         }
+
+        public object GetResult()
+        {
+            if (OnlyOneChild)
+            {
+                return Children.FirstOrDefault()?.GetResult();
+            }
+            else
+            {
+                return null;
+            }
+        }
     }
 }
diff --git a/AIStudio.Wpf.DiagramDesigner/ViewModels/BlockViewModel/BlockDesignerItemViewModel.cs b/AIStudio.Wpf.DiagramDesigner/ViewModels/BlockViewModel/BlockDesignerItemViewModel.cs
index 4d1cc21..70c3c30 100644
--- a/AIStudio.Wpf.DiagramDesigner/ViewModels/BlockViewModel/BlockDesignerItemViewModel.cs
+++ b/AIStudio.Wpf.DiagramDesigner/ViewModels/BlockViewModel/BlockDesignerItemViewModel.cs
@@ -413,6 +413,23 @@ namespace AIStudio.Wpf.DiagramDesigner
         {
 
         }
+
+        public virtual object GetResult()
+        {
+            if (FirstContainer != null)
+            {
+                return FirstContainer.GetResult();
+            }
+            else if (SecondContainer != null)
+            {
+                return SecondContainer.GetResult();
+            }
+            else if (ThirdContainer != null)
+            {
+                return ThirdContainer.GetResult();
+            }
+            return null;
+        }
         #endregion
     }