/*
* Copyright (c) 1998, 2011, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
/*
* This source code is provided to illustrate the usage of a given feature
* or technique and has been deliberately simplified. Additional steps
* required for a production-quality application, such as security checks,
* input validation and proper error handling, might not be present in
* this sample code.
*/
package com.sun.tools.example.debug.gui;
import java.util.*;
import javax.swing.*;
import javax.swing.tree.*;
import java.awt.*;
import java.awt.event.*;
import com.sun.jdi.*;
import com.sun.tools.example.debug.event.*;
import com.sun.tools.example.debug.bdi.*;
public class ClassTreeTool extends JPanel {
private static final long serialVersionUID = 526178912591739259L;
private Environment env;
private ExecutionManager runtime;
private SourceManager sourceManager;
private ClassManager classManager;
private JTree tree;
private DefaultTreeModel treeModel;
private ClassTreeNode root;
// private SearchPath sourcePath;
private CommandInterpreter interpreter;
private static String HEADING = "CLASSES";
public ClassTreeTool(Environment env) {
super(new BorderLayout());
this.env = env;
this.runtime = env.getExecutionManager();
this.sourceManager = env.getSourceManager();
this.interpreter = new CommandInterpreter(env);
root = createClassTree(HEADING);
treeModel = new DefaultTreeModel(root);
// Create a tree that allows one selection at a time.
tree = new JTree(treeModel);
tree.setSelectionModel(new SingleLeafTreeSelectionModel());
/******
// Listen for when the selection changes.
tree.addTreeSelectionListener(new TreeSelectionListener() {
public void valueChanged(TreeSelectionEvent e) {
ClassTreeNode node = (ClassTreeNode)
(e.getPath().getLastPathComponent());
if (node != null) {
interpreter.executeCommand("view " + node.getReferenceTypeName());
}
}
});
******/
MouseListener ml = new MouseAdapter() {
@Override
public void mouseClicked(MouseEvent e) {
int selRow = tree.getRowForLocation(e.getX(), e.getY());
TreePath selPath = tree.getPathForLocation(e.getX(), e.getY());
if(selRow != -1) {
if(e.getClickCount() == 1) {
ClassTreeNode node =
(ClassTreeNode)selPath.getLastPathComponent();
// If user clicks on leaf, select it, and issue 'view' command.
if (node.isLeaf()) {
tree.setSelectionPath(selPath);
interpreter.executeCommand("view " + node.getReferenceTypeName());
}
}
}
}
};
tree.addMouseListener(ml);
JScrollPane treeView = new JScrollPane(tree);
add(treeView);
// Create listener.
ClassTreeToolListener listener = new ClassTreeToolListener();
runtime.addJDIListener(listener);
runtime.addSessionListener(listener);
//### remove listeners on exit!
}
private class ClassTreeToolListener extends JDIAdapter
implements JDIListener, SessionListener {
// SessionListener
@Override
public void sessionStart(EventObject e) {
// Get system classes and any others loaded before attaching.
try {
for (ReferenceType type : runtime.allClasses()) {
root.addClass(type);
}
} catch (VMDisconnectedException ee) {
// VM terminated unexpectedly.
} catch (NoSessionException ee) {
// Ignore. Should not happen.
}
}
@Override
public void sessionInterrupt(EventObject e) {}
@Override
public void sessionContinue(EventObject e) {}
// JDIListener
@Override
public void classPrepare(ClassPrepareEventSet e) {
root.addClass(e.getReferenceType());
}
@Override
public void classUnload(ClassUnloadEventSet e) {
root.removeClass(e.getClassName());
}
@Override
public void vmDisconnect(VMDisconnectEventSet e) {
// Clear contents of this view.
root = createClassTree(HEADING);
treeModel = new DefaultTreeModel(root);
tree.setModel(treeModel);
}
}
ClassTreeNode createClassTree(String label) {
return new ClassTreeNode(label, null);
}
class ClassTreeNode extends DefaultMutableTreeNode {
private String name;
private ReferenceType refTy; // null for package
ClassTreeNode(String name, ReferenceType refTy) {
this.name = name;
this.refTy = refTy;
}
@Override
public String toString() {
return name;
}
public ReferenceType getReferenceType() {
return refTy;
}
public String getReferenceTypeName() {
return refTy.name();
}
private boolean isPackage() {
return (refTy == null);
}
@Override
public boolean isLeaf() {
return !isPackage();
}
public void addClass(ReferenceType refTy) {
addClass(refTy.name(), refTy);
}
private void addClass(String className, ReferenceType refTy) {
if (className.equals("")) {
return;
}
int pos = className.indexOf('.');
if (pos < 0) {
insertNode(className, refTy);
} else {
String head = className.substring(0, pos);
String tail = className.substring(pos + 1);
ClassTreeNode child = insertNode(head, null);
child.addClass(tail, refTy);
}
}
private ClassTreeNode insertNode(String name, ReferenceType refTy) {
for (int i = 0; i < getChildCount(); i++) {
ClassTreeNode child = (ClassTreeNode)getChildAt(i);
int cmp = name.compareTo(child.toString());
if (cmp == 0) {
// like-named node already exists
return child;
} else if (cmp < 0) {
// insert new node before the child
ClassTreeNode newChild = new ClassTreeNode(name, refTy);
treeModel.insertNodeInto(newChild, this, i);
return newChild;
}
}
// insert new node after last child
ClassTreeNode newChild = new ClassTreeNode(name, refTy);
treeModel.insertNodeInto(newChild, this, getChildCount());
return newChild;
}
public void removeClass(String className) {
if (className.equals("")) {
return;
}
int pos = className.indexOf('.');
if (pos < 0) {
ClassTreeNode child = findNode(className);
if (!isPackage()) {
treeModel.removeNodeFromParent(child);
}
} else {
String head = className.substring(0, pos);
String tail = className.substring(pos + 1);
ClassTreeNode child = findNode(head);
child.removeClass(tail);
if (isPackage() && child.getChildCount() < 1) {
// Prune non-leaf nodes with no children.
treeModel.removeNodeFromParent(child);
}
}
}
private ClassTreeNode findNode(String name) {
for (int i = 0; i < getChildCount(); i++) {
ClassTreeNode child = (ClassTreeNode)getChildAt(i);
int cmp = name.compareTo(child.toString());
if (cmp == 0) {
return child;
} else if (cmp > 0) {
// not found, since children are sorted
return null;
}
}
return null;
}
}
}