8295934: IGV: keep node selection when changing view or graph

Reviewed-by: thartmann, rcastanedalo

Author: tobiasholenstein

src/utils/IdealGraphVisualizer/Bytecodes/src/main/java/com/sun/hotspot/igv/bytecodes/SelectBytecodesAction.java

@@ -42,7 +42,8 @@ protected void performAction(Node[] activatedNodes) {
         InputGraphProvider p = LookupHistory.getLast(InputGraphProvider.class);
         if (p != null) {
             p.clearSelectedNodes();
-            p.addSelectedNodes(c.getNodes());
+            p.addSelectedNodes(c.getNodes(), true);
+            p.centerSelectedNodes();
         }
     }
 

src/utils/IdealGraphVisualizer/ControlFlow/src/main/java/com/sun/hotspot/igv/controlflow/ControlFlowScene.java

@@ -99,6 +99,9 @@ public void setGraph(InputGraph g) {
             removeEdge(e);
         }
 
+        edgeLayer.removeChildren();
+        mainLayer.removeChildren();
+
         for (InputBlock b : g.getBlocks()) {
             addNode(b);
         }
@@ -118,33 +121,34 @@ public void setGraph(InputGraph g) {
         validate();
     }
 
-    public void clearSelection() {
+    private void clearSelection() {
         for (BlockWidget w : selection) {
             w.setState(w.getState().deriveSelected(false));
         }
         selection.clear();
         selectionChanged();
     }
 
-    public void selectionChanged() {
+    private void selectionChanged() {
         InputGraphProvider p = LookupHistory.getLast(InputGraphProvider.class);
         if (p != null) {
             Set<InputNode> inputNodes = new HashSet<>();
             for (BlockWidget w : selection) {
                 inputNodes.addAll(w.getBlock().getNodes());
             }
             p.clearSelectedNodes();
-            p.addSelectedNodes(inputNodes);
+            p.addSelectedNodes(inputNodes, true);
+            p.centerSelectedNodes();
         }
     }
 
-    public void addToSelection(BlockWidget widget) {
+    private void addToSelection(BlockWidget widget) {
         widget.setState(widget.getState().deriveSelected(true));
         selection.add(widget);
         selectionChanged();
     }
 
-    public void removeFromSelection(BlockWidget widget) {
+    private void removeFromSelection(BlockWidget widget) {
         widget.setState(widget.getState().deriveSelected(false));
         selection.remove(widget);
         selectionChanged();

src/utils/IdealGraphVisualizer/ControlFlow/src/main/java/com/sun/hotspot/igv/controlflow/ControlFlowTopComponent.java

@@ -29,6 +29,7 @@
 import com.sun.hotspot.igv.util.LookupHistory;
 import java.awt.BorderLayout;
 import javax.swing.JScrollPane;
+import javax.swing.SwingUtilities;
 import org.openide.ErrorManager;
 import org.openide.util.NbBundle;
 import org.openide.windows.TopComponent;
@@ -102,14 +103,16 @@ public void componentClosed() {
 
     @Override
     public void changed(InputGraphProvider lastProvider) {
-        if (lastProvider != null) {
-            InputGraph graph = lastProvider.getGraph();
-            if (graph != null) {
-                scene.setGraph(graph);
-                return;
+        SwingUtilities.invokeLater(() -> {
+            if (lastProvider != null) {
+                InputGraph graph = lastProvider.getGraph();
+                if (graph != null) {
+                    scene.setGraph(graph);
+                    return;
+                }
             }
-        }
-        scene.setGraph(new InputGraph(""));
+            scene.setGraph(new InputGraph(""));
+        });
     }
 
     @Override

src/utils/IdealGraphVisualizer/Data/src/main/java/com/sun/hotspot/igv/data/InputGraph.java

@@ -255,7 +255,12 @@ public String getDisplayName() {
     }
 
     public int getIndex() {
-        return getGroup().getGraphs().indexOf(this);
+        Group group = getGroup();
+        if (group != null) {
+            return group.getGraphs().indexOf(this);
+        } else {
+            return -1;
+        }
     }
 
     public Collection<InputNode> getNodes() {

src/utils/IdealGraphVisualizer/Data/src/main/java/com/sun/hotspot/igv/data/services/InputGraphProvider.java

@@ -36,10 +36,11 @@ public interface InputGraphProvider {
 
     InputGraph getGraph();
 
-    void addSelectedNodes(Collection<InputNode> nodes);
+    void centerSelectedNodes();
 
-    void clearSelectedNodes();
+    void addSelectedNodes(Collection<InputNode> nodes, boolean showIfHidden);
 
+    void clearSelectedNodes();
 
     /**
      * @return an iterator walking forward through the {@link InputGraph}s following the {@link #getGraph()}

src/utils/IdealGraphVisualizer/Graph/src/main/java/com/sun/hotspot/igv/graph/Diagram.java

@@ -23,8 +23,8 @@
  */
 package com.sun.hotspot.igv.graph;
 
-import com.sun.hotspot.igv.data.*;
 import com.sun.hotspot.igv.data.Properties;
+import com.sun.hotspot.igv.data.*;
 import java.awt.Font;
 import java.util.*;
 
@@ -268,4 +268,14 @@ public int compare(Figure a, Figure b) {
 
         System.out.println("=============================================================");
     }
+
+    public Figure getRootFigure() {
+        Properties.PropertySelector<Figure> selector = new Properties.PropertySelector<>(figures);
+        Figure root = selector.selectSingle(new Properties.StringPropertyMatcher("name", "Root"));
+        if (root == null) {
+            root = selector.selectSingle(new Properties.StringPropertyMatcher("name", "Start"));
+        }
+        return root;
+    }
+
 }

src/utils/IdealGraphVisualizer/Util/src/main/java/com/sun/hotspot/igv/util/RangeSliderModel.java

@@ -45,11 +45,11 @@ public class RangeSliderModel implements ChangedEventProvider<RangeSliderModel>
     private int secondPosition;
     private List<Color> colors;
 
-    public RangeSliderModel(List<String> positions) {
-        assert positions.size() > 0;
-        this.changedEvent = new ChangedEvent<>(this);
-        this.colorChangedEvent = new ChangedEvent<>(this);
-        setPositions(positions);
+    public RangeSliderModel() {
+        changedEvent = new ChangedEvent<>(this);
+        colorChangedEvent = new ChangedEvent<>(this);
+        positions = new ArrayList<>();
+        colors = new ArrayList<>();
     }
 
     protected void setPositions(List<String> positions) {
@@ -88,10 +88,12 @@ public int getSecondPosition() {
     public void setPositions(int fp, int sp) {
         assert fp >= 0 && fp < positions.size();
         assert sp >= 0 && sp < positions.size();
-        firstPosition = fp;
-        secondPosition = sp;
-        ensureOrder();
-        changedEvent.fire();
+        if (firstPosition != fp || secondPosition != sp) {
+            firstPosition = fp;
+            secondPosition = sp;
+            ensureOrder();
+            changedEvent.fire();
+        }
     }
 
     private void ensureOrder() {

src/utils/IdealGraphVisualizer/View/src/main/java/com/sun/hotspot/igv/view/BlockQuickSearch.java

@@ -94,6 +94,7 @@ public void evaluate(SearchRequest request, SearchResponse response) {
                             }
                             editor.clearSelectedNodes();
                             editor.addSelectedNodes(b.getNodes(), true);
+                            editor.centerSelectedNodes();
                             editor.requestActive();
                         },
                         "B" + b.getName() + (theGraph != null ? " in " + theGraph.getName() : ""))) {