GH-55 Expand All Threads and Collapse All Threads actions for Java Threads view

Author: jisedlac

visualvm/heapviewer/src/com/sun/tools/visualvm/heapviewer/java/impl/JavaThreadsView.java

@@ -59,6 +59,9 @@
 import com.sun.tools.visualvm.heapviewer.ui.HeapViewerNodeAction;
 import com.sun.tools.visualvm.heapviewer.ui.PluggableTreeTableView;
 import com.sun.tools.visualvm.heapviewer.ui.TreeTableViewColumn;
+import javax.swing.AbstractAction;
+import javax.swing.JPopupMenu;
+import javax.swing.SwingUtilities;
 import org.openide.util.NbBundle;
 import org.openide.util.lookup.ServiceProvider;
 
@@ -74,7 +77,9 @@
     "JavaThreadsView_TooltipObjects=Objects",
     "JavaThreadsView_TooltipHTML=HTML",
     "JavaThreadsView_Details=Details:",
-    "JavaThreadsView_SelectAction=Select in Threads"
+    "JavaThreadsView_SelectAction=Select in Threads",
+    "JavaThreadsView_ExpandAction=Expand All Threads",
+    "JavaThreadsView_CollapseAction=Collapse All Threads"
 })
 public class JavaThreadsView extends HeapViewerFeature {
 
@@ -112,6 +117,35 @@ protected void childrenChanged() {
                 CCTNode[] children = getRoot().getChildren();
                 for (CCTNode child : children) expandNode((HeapViewerNode)child);
             }
+            @Override
+            protected void populatePopup(HeapViewerNode node, JPopupMenu popup) {
+                if (popup.getComponentCount() > 0) popup.addSeparator();
+                
+                popup.add(new AbstractAction(Bundle.JavaThreadsView_ExpandAction()) {
+                    @Override
+                    public void actionPerformed(ActionEvent e) {
+                        SwingUtilities.invokeLater(new Runnable() {
+                            @Override
+                            public void run() {
+                                CCTNode[] children = getRoot().getChildren();
+                                for (CCTNode child : children) expandNode((HeapViewerNode)child);
+                            }
+                        });
+                    }
+                });
+                
+                popup.add(new AbstractAction(Bundle.JavaThreadsView_CollapseAction()) {
+                    @Override
+                    public void actionPerformed(ActionEvent e) {
+                        SwingUtilities.invokeLater(new Runnable() {
+                            @Override
+                            public void run() {
+                                collapseChildren((HeapViewerNode)getRoot());
+                            }
+                        });
+                    }
+                });
+            }
         };
         objectsView.setViewName(Bundle.JavaThreadsView_Name());
 

visualvm/heapviewer/src/com/sun/tools/visualvm/heapviewer/ui/TreeTableView.java

@@ -242,6 +242,11 @@ public void expandNode(HeapViewerNode node) {
         treeTable.expandPath(HeapViewerNode.fromNode(node));
     }
 
+    public void collapseChildren(HeapViewerNode node) {
+        if (treeTable == null) return;
+        treeTable.collapseChildren(HeapViewerNode.fromNode(node));
+    }
+    
 
 //    protected void willBeSorted(List<RowSorter.SortKey> sortKeys) {}
 
@@ -380,6 +385,8 @@ protected void populatePopup(JPopupMenu popup, Object value, Object userValue) {
                         HeapViewerNodeAction.Actions.forNode(node, actionProviders, context, actions, new PinAction(node), new ResetPinAction());
                 nodeActions.populatePopup(popup);
 
+                TreeTableView.this.populatePopup(node, popup);
+                
                 if (popup.getComponentCount() > 0) popup.addSeparator();
                 popup.add(treeTable.createCopyMenuItem());
                 TreeTableView.this.populatePopupLast(node, popup);
@@ -499,6 +506,9 @@ public boolean requestFocusInWindow() {
         return comp;
     }
 
+    protected void populatePopup(HeapViewerNode node, JPopupMenu popup) {
+    }
+    
     protected void populatePopupLast(HeapViewerNode node, JPopupMenu popup) {
         if (filterComponent.getComponentCount() > 0) {
             popup.addSeparator();