Fix node overlap issue by skipping the original node from AutoLayout repositioning (#11414)

zeusongit · web-flow · commit 6c94345c5005 · 2021-03-02T11:41:18.000-05:00
* prevent overlap
diff --git a/src/DynamoCore/Graph/Workspaces/LayoutExtensions.cs b/src/DynamoCore/Graph/Workspaces/LayoutExtensions.cs
@@ -20,7 +20,7 @@ public static class LayoutExtensions
         /// </summary>
         /// <param name="workspace">Workspace on which graph layout will be performed.</param>
         /// <param name="reuseUndoRedoGroup">If true, skip initializing new undo action group.</param>
-        internal static List<GraphLayout.Graph> DoGraphAutoLayout(this WorkspaceModel workspace, bool reuseUndoRedoGroup = false)
+        internal static List<GraphLayout.Graph> DoGraphAutoLayout(this WorkspaceModel workspace, bool reuseUndoRedoGroup = false, bool isNodeAutoComplete = false, Guid? originalNodeGUID = null)
         {
             if (workspace.Nodes.Count() < 2) return null;
 
@@ -50,7 +50,14 @@ public static class LayoutExtensions
             // Run layout algorithm for each subgraph
             layoutSubgraphs.Skip(1).ToList().ForEach(g => RunLayoutSubgraph(g, isGroupLayout));
             AvoidSubgraphOverlap(layoutSubgraphs);
-            SaveLayoutGraph(workspace, layoutSubgraphs);
+
+            if (isNodeAutoComplete)
+            {
+                SaveLayoutGraphForNodeAutoComplete(workspace, layoutSubgraphs, originalNodeGUID);
+            }
+            else {
+                SaveLayoutGraph(workspace, layoutSubgraphs);
+            }
 
             // Restore the workspace model selection information
             selection.ToList().ForEach(x => x.Select());
@@ -426,6 +433,46 @@ private static void SaveLayoutGraph(this WorkspaceModel workspace, List<GraphLay
                     node.ReportPosition();
                     workspace.HasUnsavedChanges = true;
 
+                    double noteOffset = -n.NotesHeight;
+                    foreach (NoteModel note in n.LinkedNotes)
+                    {
+                        if (note.IsSelected || DynamoSelection.Instance.Selection.Count == 0)
+                        {
+                            note.X = node.X;
+                            note.Y = node.Y + noteOffset;
+                            noteOffset += note.Height + GraphLayout.Graph.VerticalNoteDistance;
+                            note.ReportPosition();
+                        }
+                    }
+                }
+            }
+        }
+        /// <summary>
+        /// This method pushes changes from the GraphLayout.Graph objects
+        /// back to the workspace models, but only for nodes placed by NodeAutocomplete.
+        /// </summary>
+        private static void SaveLayoutGraphForNodeAutoComplete(this WorkspaceModel workspace, List<GraphLayout.Graph> layoutSubgraphs, Guid? originalNodeGUID)
+        {
+            // Assign coordinates to nodes outside groups
+            foreach (var node in workspace.Nodes)
+            {
+                GraphLayout.Graph graph = layoutSubgraphs
+                    .FirstOrDefault(g => g.FindNode(node.GUID) != null);
+
+                if (graph != null)
+                {
+                    GraphLayout.Node n = graph.FindNode(node.GUID);
+                    double offsetY = graph.OffsetY;
+                    //skipping the original node to avoid jumping of node
+                    if (node.GUID != originalNodeGUID )
+                    {
+                            node.X = n.X;
+                            node.Y = n.Y + n.NotesHeight + offsetY;
+                        
+                    }
+                    node.ReportPosition();
+                    workspace.HasUnsavedChanges = true;
+
                     double noteOffset = -n.NotesHeight;
                     foreach (NoteModel note in n.LinkedNotes)
                     {
diff --git a/src/DynamoCoreWpf/ViewModels/Search/NodeSearchElementViewModel.cs b/src/DynamoCoreWpf/ViewModels/Search/NodeSearchElementViewModel.cs
@@ -286,11 +286,12 @@ protected virtual void CreateAndConnectToPort(object parameter)
             dynamoViewModel.ExecuteCommand(new DynamoModel.CreateAndConnectNodeCommand(id, initialNode.GUID,
                 Model.CreationName, 0, portModel.Index, adjustedX, 0, createAsDownStreamNode, false, true));
 
-            // Clear current selections and select all input nodes as we need to perform Auto layout on only the input nodes.
-            DynamoSelection.Instance.ClearSelection();
             var inputNodes = initialNode.InputNodes.Values.Where(x => x != null).Select(y => y.Item2);
 
+            // Clear current selections and select all input nodes as we need to perform Auto layout on only the input nodes.
+            DynamoSelection.Instance.ClearSelection();
             DynamoSelection.Instance.Selection.AddRange(inputNodes);
+            DynamoSelection.Instance.Selection.Add(initialNode);
         }
 
         protected virtual bool CanCreateAndConnectToPort(object parameter)
@@ -305,8 +306,9 @@ private void AutoLayoutNodes(object sender, EventArgs e)
         {
             var nodeView = (NodeView) sender;
             var dynamoViewModel = nodeView.ViewModel.DynamoViewModel;
+            var originalNodeId = nodeView.ViewModel.NodeModel.OutputNodes.Values.SelectMany(s => s.Select(t => t.Item2)).Distinct().FirstOrDefault().GUID;
 
-            dynamoViewModel.CurrentSpace.DoGraphAutoLayout(true);
+            dynamoViewModel.CurrentSpace.DoGraphAutoLayout(true, true, originalNodeId);
 
             DynamoSelection.Instance.ClearSelection();
 
