JAVA之树(创建,遍历,修改,删除,标签,提示信息,展开折叠事件...)

先看一下界面





import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.Graphics;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

import javax.swing.DefaultCellEditor;
import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import javax.swing.JTree;
import javax.swing.ToolTipManager;
import javax.swing.WindowConstants;
import javax.swing.event.TreeExpansionEvent;
import javax.swing.event.TreeExpansionListener;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.event.TreeWillExpandListener;
import javax.swing.plaf.basic.BasicOptionPaneUI.ButtonActionListener;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.ExpandVetoException;
import javax.swing.tree.MutableTreeNode;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;

import org.omg.CORBA.PRIVATE_MEMBER;

public class test extends JFrame{
private JTree tree;
DefaultMutableTreeNode root = new DefaultMutableTreeNode("root");
DefaultTreeModel model = new DefaultTreeModel(root, true);
JButton btnAdd = new JButton("添加");
JButton btnDel = new JButton("删除");
JButton btnFix = new JButton("修改");
DefaultMutableTreeNode[] rootF = new DefaultMutableTreeNode[3];
JPanel btnPanel = new JPanel();
JPanel btnCenter = new JPanel();
JTextField text;
public test(){
init();

addNode();

cellRender();

cellEdit();
}

private void cellEdit() {
// TODO Auto-generated method stub
text = new JTextField();
btnCenter.add(text);

TreeCellEditor cellE = new DefaultCellEditor(text);
tree.setEditable(true);

final JTextArea textArea = new JTextArea();
btnCenter.add(textArea);

tree.addTreeSelectionListener(new TreeSelectionListener() {

@Override
public void valueChanged(TreeSelectionEvent arg0) {
// TODO Auto-generated method stub
textArea.setText("");
TreePath path = tree.getSelectionPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
int count = node.getChildCount();
String info = "";
for(int i = 0;i<count;i++){
textArea.append(node.getChildAt(i).toString() + "\n");
}
}
});
}

private void addNode() {
// TODO Auto-generated method stub

btnAdd.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent arg0) {
// TODO Auto-generated method stub

MutableTreeNode nodeNew = new DefaultMutableTreeNode();
TreePath path = tree.getSelectionPath();
MutableTreeNode nodePar = (MutableTreeNode) path.getLastPathComponent();

int count = nodePar.getChildCount();
nodeNew.setUserObject("child"+(count+1));

model.insertNodeInto(nodeNew, nodePar, count);

TreePath NewPath = path.pathByAddingChild(nodeNew);
if(!tree.isVisible(NewPath)){
tree.makeVisible(NewPath);
}
}
});

btnFix.addActionListener(new ActionListener() {

@Override
public void actionPerformed(ActionEvent arg0) {
// TODO Auto-generated method stub
TreePath path = tree.getSelectionPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
node.setUserObject("fix");
model.nodeChanged(node);
tree.setSelectionPath(path);
repaint();
}
});

btnDel.addActionListener(new ActionListener() {

@Override
public void actionPerformed(ActionEvent arg0) {
// TODO Auto-generated method stub
DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree.getLastSelectedPathComponent();
if(!node.isRoot()){
DefaultMutableTreeNode nodeNext = node.getNextSibling();
if(nodeNext == null){
nodeNext = (DefaultMutableTreeNode) node.getParent();
}
model.removeNodeFromParent(node);
tree.setSelectionPath(new TreePath(nodeNext.getPath()));
}
}
});

tree.addTreeWillExpandListener(new TreeWillExpandListener() {

@Override
public void treeWillExpand(TreeExpansionEvent arg0)
throws ExpandVetoException {
// TODO Auto-generated method stub
TreePath path = arg0.getPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
System.out.println("节点" + node + "将要被展开");
}

@Override
public void treeWillCollapse(TreeExpansionEvent arg0)
throws ExpandVetoException {
// TODO Auto-generated method stub
TreePath path = arg0.getPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
System.out.println("节点" + node + "将要被折叠");
}
});

tree.addTreeExpansionListener(new TreeExpansionListener() {

@Override
public void treeExpanded(TreeExpansionEvent arg0) {
// TODO Auto-generated method stub
TreePath path = arg0.getPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
System.out.println("节点" + node + "被展开");
}

@Override
public void treeCollapsed(TreeExpansionEvent arg0) {
// TODO Auto-generated method stub
TreePath path = arg0.getPath();
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
System.out.println("节点" + node + "被折叠");
}
});
}

public void init(){
tree = new JTree(root);
JScrollPane js = new JScrollPane();
js.setViewportView(tree);
getContentPane().add(js,BorderLayout.CENTER);

tree.setShowsRootHandles(false);

for(int i = 0;i<3;i++){
rootF[i] = new DefaultMutableTreeNode(i+1);
DefaultMutableTreeNode[] rootC = new DefaultMutableTreeNode[2];
for(int j = 1;j<=2;j++){
rootC[j-1] = new DefaultMutableTreeNode("Child" + (i*2 + j));
rootF[i].add(rootC[j-1]);
}
root.add(rootF[i]);
}
tree.expandPath(new TreePath(root.getPath()));

DefaultTreeCellRenderer render = (DefaultTreeCellRenderer) tree.getCellRenderer();

//ImageIcon icon = new ImageIcon("ii.jpg");
render.setLeafIcon(null);
render.setClosedIcon(null);
render.setOpenIcon(null);

btnCenter.setLayout(new GridLayout(5,1,0,0));
btnPanel.setLayout(new GridLayout(4, 1,0,0));
JButton[] btn = new JButton[4];
String[] opeName = {"先序","后序","bfs","dfs"};

for(int i = 1;i<=4;i++){
btn[i-1] = new JButton(opeName[i-1]);
btnPanel.add(btn[i-1]);
}
btnCenter.add(btnAdd);
btnCenter.add(btnFix);
btnCenter.add(btnDel);

for(int i = 0;i<4;i++){
btn[i].addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent arg0) {
// TODO Auto-generated method stub
Enumeration enumeration = null;
String btnName = arg0.getActionCommand();
if(btnName.equals("先序")){
enumeration = root.preorderEnumeration();
}else{
if(btnName.equals("后序")){
enumeration = root.postorderEnumeration();
}else{
if(btnName.equals("bfs")){
enumeration = root.breadthFirstEnumeration();
}else{
if(btnName.equals("dfs"))
enumeration = root.depthFirstEnumeration();
}
}
}

while(enumeration.hasMoreElements()){
DefaultMutableTreeNode node = (DefaultMutableTreeNode)enumeration.nextElement();
for(int j = 0;j<node.getLevel();j++){
System.out.print("---");
}
System.out.println(node.getUserObject());
}
}
});
}

getContentPane().add(btnPanel,BorderLayout.EAST);
getContentPane().add(btnCenter,BorderLayout.WEST);
setVisible(true);
setSize(300, 300);
setDefaultCloseOperation(WindowConstants.EXIT_ON_CLOSE);
}

public class ToolTipNode implements TreeCellRenderer{

private DefaultTreeCellRenderer renderer = new DefaultTreeCellRenderer();
private Map<DefaultMutableTreeNode,String> map;
public ToolTipNode(Map<DefaultMutableTreeNode,String> map){
this.map = map;
}

@Override
public Component getTreeCellRendererComponent(JTree tree, Object value,
boolean selected, boolean expanded, boolean leaf, int row,
boolean hasFocus) {
// TODO Auto-generated method stub
renderer.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);
renderer.setToolTipText("<html><font face=黑体 size=16 color=red>" + map.get(value) + "</font></html>");
return renderer;
}

}

public void cellRender(){
ToolTipManager.sharedInstance().registerComponent(tree);
Map<DefaultMutableTreeNode,String> map = new HashMap<DefaultMutableTreeNode,String>();
map.put(root,"有"+root.getChildCount()+"个子文件");
for(int i = 0;i<3;i++){
map.put(rootF[i], "有"+rootF[i].getChildCount()+"个子文件");
}
tree.setCellRenderer(new ToolTipNode(map));
}

public static void main(String[] agrs){
new test();
}
}