Java JTree_5

package com.han;

import java.awt.BorderLayout;
import java.awt.Container;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Enumeration;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextField;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;

public class JTree_5 extends JFrame {

	/**
	 * 
	 */
	private static final long serialVersionUID = -7187852901157241733L;

	public JTree_5() {
		// TODO Auto-generated constructor stub
		DefaultMutableTreeNode root = new DefaultMutableTreeNode("倾城爱家");
		DefaultMutableTreeNode nodeA = new DefaultMutableTreeNode("经理办公室");
		DefaultMutableTreeNode nodeB = new DefaultMutableTreeNode("人事部");
		DefaultMutableTreeNode nodeC = new DefaultMutableTreeNode("技术部");
		DefaultMutableTreeNode nodeD = new DefaultMutableTreeNode("服务部");
		DefaultMutableTreeNode nodeE = new DefaultMutableTreeNode("推广部");
		
		DefaultMutableTreeNode nodeCA = new DefaultMutableTreeNode("网络维护部");
		DefaultMutableTreeNode nodeCB = new DefaultMutableTreeNode("应用开发部");
		
		DefaultMutableTreeNode nodeDA = new DefaultMutableTreeNode("网络服务部");
		DefaultMutableTreeNode nodeDB = new DefaultMutableTreeNode("内勤开发部");
		
		root.add(nodeA);
		root.add(nodeB);
		root.add(nodeC);
		root.add(nodeD);
		root.add(nodeE);
		nodeC.add(nodeCA);
		nodeC.add(nodeCB);
		nodeD.add(nodeDA);
		nodeD.add(nodeDB);
		
		final DefaultTreeModel treeModel = new DefaultTreeModel(root, true);
		final JTree tree =new JTree(treeModel);
		
		Container container = getContentPane();
		container.add(tree, BorderLayout.CENTER);
		
		@SuppressWarnings("rawtypes")
		Enumeration enumeration = root.preorderEnumeration();
		while (enumeration.hasMoreElements()) {
			DefaultMutableTreeNode node = (DefaultMutableTreeNode) enumeration.nextElement();
			tree.expandPath(new TreePath(node.getPath()));
		}
		
		JPanel panel = new JPanel(new GridLayout(1, 5, 10, 10));
		container.add(panel, BorderLayout.SOUTH);
		JLabel label = new JLabel("名称：");
		final JTextField textField = new JTextField("测试");
		JButton button1 = new JButton("添加");
		JButton button2 = new JButton("修改");
		JButton button3 = new JButton("删除");
		panel.add(label);
		panel.add(textField);
		panel.add(button1);
		panel.add(button2);
		panel.add(button3);
		
		// add
		button1.addActionListener(new ActionListener(){

			@Override
			public void actionPerformed(ActionEvent e) {
				// TODO Auto-generated method stub
				DefaultMutableTreeNode node = new DefaultMutableTreeNode(textField.getText());
				TreePath path = tree.getSelectionPath();
				DefaultMutableTreeNode parent = (DefaultMutableTreeNode) path.getLastPathComponent();
				treeModel.insertNodeInto(node, parent, parent.getChildCount());
				TreePath newNodePath = path.pathByAddingChild(node);
				if (!tree.isVisible(newNodePath)) {
					tree.makeVisible(newNodePath);
				}
				tree.setSelectionPath(newNodePath);
			}
			
		});
		
		// modify
		button2.addActionListener(new ActionListener(){

			@Override
			public void actionPerformed(ActionEvent e) {
				// TODO Auto-generated method stub
				TreePath path = tree.getSelectionPath();
				DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
				node.setUserObject(textField.getText());
				treeModel.nodeChanged(node);				
			}
			
		});
		
		// delete
		button3.addActionListener(new ActionListener(){

			@Override
			public void actionPerformed(ActionEvent e) {
				// TODO Auto-generated method stub
				TreePath path = tree.getSelectionPath();
				DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
				if (!node.isRoot()) {
					DefaultMutableTreeNode nextSelectedNode = node.getNextSibling();
					if (nextSelectedNode == null) {
						nextSelectedNode = (DefaultMutableTreeNode) node.getParent();
					}
					tree.setSelectionPath(new TreePath(nextSelectedNode.getPath()));
					treeModel.removeNodeFromParent(node);
				}
			}
			
		});
		
	}

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		JTree_5 instance = new JTree_5();
		instance.setTitle("维护企业架构");
		instance.setVisible(true);
		instance.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
		instance.pack();
	}

}