箭头分离到独立的model中，方便自定义path

2026-03-17 15:06:36 +08:00 · 2023-01-26 18:27:17 +08:00
parent f11a4170d8
commit 9bb6c6d321
42 changed files with 993 additions and 299 deletions
--- a/AIStudio.Wpf.DiagramDesigner/Algorithms/LinksReconnectionAlgorithms.cs
+++ b/AIStudio.Wpf.DiagramDesigner/Algorithms/LinksReconnectionAlgorithms.cs
@@ -0,0 +1,61 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace AIStudio.Wpf.DiagramDesigner.Algorithms
+{
+    public static class LinksReconnectionAlgorithms
+    {
+        public static void ReconnectLinksToClosestPorts(this IDiagramViewModel diagram)
+        {
+            // Only refresh ports once
+            //var portsToRefresh = new HashSet<FullyCreatedConnectorInfo>();
+
+            foreach (var link in diagram.Items.OfType<ConnectionViewModel>())
+            {
+                if (link.IsFullConnection == false)
+                    continue;
+
+                var sourcePorts = link.SourceConnectorInfo.DataItem.Connectors;
+                var targetPorts = link.SinkConnectorInfoFully.DataItem.Connectors;
+
+                // Find the ports with minimal distance
+                var minDistance = double.MaxValue;
+                var minSourcePort = link.SourceConnectorInfo;
+                var minTargetPort = link.SinkConnectorInfoFully;
+                foreach (var sourcePort in sourcePorts)
+                {
+                    foreach (var targetPort in targetPorts)
+                    {
+                        var distance = sourcePort.Position.DistanceTo(targetPort.Position);
+                        if (distance < minDistance)
+                        {
+                            minDistance = distance;
+                            minSourcePort = sourcePort;
+                            minTargetPort = targetPort;
+                        }
+                    }
+                }
+
+                // Reconnect
+                if (link.SourceConnectorInfo != minSourcePort)
+                {
+                    //portsToRefresh.Add(link.SourceConnectorInfo);
+                    //portsToRefresh.Add(minSourcePort);
+                    link.SetSourcePort(minSourcePort);
+                }
+
+                if (link.SinkConnectorInfo != minTargetPort)
+                {
+                    //portsToRefresh.Add(link.SinkConnectorInfoFully);
+                    //portsToRefresh.Add(minTargetPort);
+                    link.SetSinkPort(minTargetPort);
+                }
+            }
+
+            //foreach (var port in portsToRefresh)
+            //    port.Refresh();
+        }
+    }
+}