servertreeview.java

JGraph扩展应用。自定义Renderer,自定义视图View实现自定义工作流控件

/*
Copyright 2005 Matthew J. Battey

  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

	Unless required by applicable law or agreed to in writing, software distributed
	under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
	CONDITIONS OF ANY KIND, either express or implied. See the License for the
	specific language governing permissions and limitations under the License.

This software implements a Java application to manage a SAFMQ server.
		
Created on May 17, 2005
*/
package flow.graph.gui.tree;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.util.Vector;

import javax.swing.AbstractAction;
import javax.swing.Action;
import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.JOptionPane;
import javax.swing.JPopupMenu;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.KeyStroke;
import javax.swing.ToolTipManager;
import javax.swing.event.TreeExpansionEvent;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.event.TreeWillExpandListener;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.ExpandVetoException;
import javax.swing.tree.TreePath;

import net.antonioshome.swing.treewrapper.TreeWrapper;
import net.antonioshome.swing.treewrapper.example.TreeModelBuilder;

import flow.graph.app.FlowManager;
import flow.graph.gui.tools.SimpleButton;



class ServerTreeCellRenderer extends DefaultTreeCellRenderer {
	public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {
		super.getTreeCellRendererComponent(tree,value,sel,expanded,leaf,row,hasFocus);
		Object val = ((DefaultMutableTreeNode)value).getUserObject();
		if (val instanceof IconItem) {
			Icon icon = ((IconItem)val).getIcon();
			if (icon != null)
				if(!tree.isEnabled())
					setDisabledIcon(icon);
				else
					setIcon(icon);
		}
		if(val instanceof BusinessNode)
			setToolTipText("id="+((BusinessNode)val).getNodeBean().getT_id()+":"+val.toString());
		else
			setToolTipText(val.toString());
		return this;
	}
}

/**
* @author matt
*/
public class ServerTreeView extends JScrollPane implements TreeSelectionListener, TreeWillExpandListener {
	DefaultMutableTreeNode	root;
	JTree					tree;
	DefaultTreeModel		model;

	public Action 	newServiceAction;
	public Action 	modifyServiceAction;
	public Action	newNodeAction;
	public Action	modifyNodeAction;
	
	public SimpleButton toolNewServiceAction;
	public SimpleButton toolModifyServiceAction;
	public SimpleButton toolNewNodeAction;
	public SimpleButton toolModifyNodeAction;
	public Action toolDeleteAction;

	Action	deleteAction;
	
	public ServerTreeView(Object rootObject) {
		tree = new JTree();
		//TreeWrapper wrapper1 = new TreeWrapper( tree );
		
		//tree.setModel( TreeModelBuilder.createModel() );
		
		keyListener();
		setViewportView(tree);
		root = new DefaultMutableTreeNode(rootObject);
		model = (DefaultTreeModel)tree.getModel();
		model.setRoot(root);
		root.add(new DefaultMutableTreeNode("No Service Node"));
		tree.setRootVisible(true);
		tree.setShowsRootHandles(true);
		tree.addTreeSelectionListener(this);
		tree.addTreeWillExpandListener(this);
		tree.setEditable(true);

		ToolTipManager.sharedInstance().registerComponent(tree);
		tree.setCellRenderer(new ServerTreeCellRenderer());
	    tree.setCellEditor(new WideCellEditor(tree, (DefaultTreeCellRenderer) tree.getCellRenderer()));
		
		modifyServiceAction = new SimpleAction("修改业务...", new Integer(KeyEvent.VK_U), false) {
			public void actionPerformed(ActionEvent e)	{ modifyService(); } 
		};
		newNodeAction = new SimpleAction("新增节点...", new Integer(KeyEvent.VK_G), false) {
			public void actionPerformed(ActionEvent e)	{ addNode(); }
		};
		modifyNodeAction = new SimpleAction("修改节点...", new Integer(KeyEvent.VK_G), false) {
			public void actionPerformed(ActionEvent e)	{ modifyNode(); }
		};
		deleteAction = new SimpleAction("删除", new Integer(KeyEvent.VK_D), false) {
			public void actionPerformed(ActionEvent e)	{ deleteSelected(); }	
		};
		
		tree.getInputMap().put(KeyStroke.getKeyStroke(KeyEvent.VK_DELETE,0),deleteAction);

		toolModifyServiceAction = new SimpleButton(new ImageIcon(ServerTreeView.class.getResource("images/edit_ser.gif")));
		toolModifyServiceAction.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e)	{ modifyService(); } 
		});
		toolNewNodeAction = new SimpleButton(new ImageIcon(ServerTreeView.class.getResource("images/new_bus.gif")));
		toolNewNodeAction.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e)	{ addNode(); } 
		});
		toolModifyNodeAction = new SimpleButton(new ImageIcon(ServerTreeView.class.getResource("images/edit_bus.gif")));
		toolModifyNodeAction.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent e)	{ modifyNode(); } 
		});
		
		/*
		toolModifyServiceAction = new SimpleAction("修改业务...", new ImageIcon(ServerTreeView.class.getResource("images/edit_ser.gif")), false) {
			public void actionPerformed(ActionEvent e)	{ modifyService(); } 
		};
		toolNewNodeAction = new SimpleAction("新增节点...", new ImageIcon(ServerTreeView.class.getResource("images/new_bus.gif")), false) {
			public void actionPerformed(ActionEvent e)	{ addNode(); } 
		};
		toolModifyNodeAction = new SimpleAction("修改节点...", new ImageIcon(ServerTreeView.class.getResource("images/edit_bus" +
				".gif")), false) {
			public void actionPerformed(ActionEvent e)	{ modifyNode(); } 
		};
		*/
		
		toolDeleteAction = new SimpleAction("删除", new ImageIcon(ServerTreeView.class.getResource("images/delete.gif")), false) {
			public void actionPerformed(ActionEvent e)	{ deleteSelected(); }	
		};
		
		tree.getInputMap().put(KeyStroke.getKeyStroke(KeyEvent.VK_DELETE,0),toolDeleteAction);
		addMouseListener();
	}
	
	/**
	 * Locates the node containing <code>userobj</code> and removes all the children
	 * of that node and replaces them with a new node with <code>child</code> as the 
	 * user data.  This method is typically used to reset the contents of a node containing
	 * a referece to a <code>Server</code>.  All the children of that node are replaced
	 * with the new object passed as <code>child</code>.
	 * 
	 * @param userobj 	The data from the node to be located
	 * @param child 	The new data for the single child of the node to be located.
	 */
	public void resetChildren(Object userobj, Object child) {
		DefaultMutableTreeNode n;
		for(int x=0;x<root.getChildCount();x++){
			n = (DefaultMutableTreeNode)root.getChildAt(x);
			if (n.getUserObject().equals(userobj)) {
				n.removeAllChildren();
				//n.add(new DefaultMutableTreeNode(child));
				model.nodeStructureChanged(n);
				break;
			}
		}
	}
	
	public void resetRoot() {
		/*
		DefaultMutableTreeNode n;
		for(int x=0;x<root.getChildCount();x++){
			n = (DefaultMutableTreeNode)root.getChildAt(x);
			if (n.getUserObject().equals(userobj)) {
				n.removeAllChildren();
				//n.add(new DefaultMutableTreeNode(child));
				model.nodeStructureChanged(n);
				break;
			}
		}*/
		root.removeAllChildren();
		root.add(new DefaultMutableTreeNode("No Child Node"));
	}
	
	private void keyListener(){
		tree.addKeyListener(new KeyListener(){

			public void keyTyped(KeyEvent e) {
				// TODO Auto-generated method stub
				System.out.println("typed...");
			}

			public void keyPressed(KeyEvent e) {
				// TODO Auto-generated method stub
				System.out.println("pressed...");
				System.out.println(e.getSource());
			}

			public void keyReleased(KeyEvent e) {
				System.out.println(e.getComponent());
				// TODO Auto-generated method stub
				TreePath path = ((JTree)e.getSource()).getSelectionPath();
				if (path != null) {
					Object objPath[] = path.getPath();
					
					if(objPath.length == 2){
						//业务列表增加节点
						DefaultMutableTreeNode	node = (DefaultMutableTreeNode)objPath[0];
						if(((DefaultMutableTreeNode)objPath[1]).getUserObject() instanceof ServiceNode){
							System.out.println("1-------------");
						}
						else{
							System.out.println("2-------------");
							
						}
						//StartNode start = (StartNode)node.getUserObject();
					}
					else if(objPath.length == 3){
						//业务节点增加节点
						DefaultMutableTreeNode	node = (DefaultMutableTreeNode)objPath[1];
						//ServiceNode service = (ServiceNode)node.getUserObject();
						System.out.println("3-------------");
					}
				}

			}		
		});
	}
	
	/**
	 * Refereshes the root node's list of children.
	 */
	public void refreshRoot() {
		StartNode start = (StartNode)root.getUserObject();
		start.setNodeChildren(tree, root);
		tree.setSelectionPath(null);
		
		TreePath tp = new TreePath(root.getPath());
		
		tree.expandPath(tp);
		tree.setSelectionPath(tp);
	}
	
	public void activatePath(TreePath path) {
		tree.setSelectionPath(null);
		tree.expandPath(path);
		tree.setSelectionPath(path);
	}
	
	public void explandPath(ServiceNode service) {
		/*
		tree.setSelectionPath(null);
		tree.expandPath(path);
		tree.setSelectionPath(path);
		
		DefaultMutableTreeNode node;	
		for(int x=0;x<parent.getChildCount();x++) {
			node = (DefaultMutableTreeNode)parent.getChildAt(x);
			if (node.getUserObject().getClass() == c) {
				tree.setSelectionPath(null);
				tree.setSelectionPath(new TreePath(node.getPath()));				
			}
		}
		*/
		
		for(int i=0;i<root.getChildCount();i++){
			DefaultMutableTreeNode node = (DefaultMutableTreeNode)root.getChildAt(i);
			//System.out.println("i="+i+","+root.getChildAt(0));
			if(service.equals(node.getUserObject())){
				tree.setSelectionPath(null);
				tree.expandPath(new TreePath(node.getPath()));
				//System.out.println("service="+service+",userobject="+node.getUserObject());
				tree.setSelectionPath(new TreePath(node.getPath()));				
			}
		}
	}
	
	public void addMouseListener(){
		tree.addMouseListener(new MouseListener(){

			public void mouseClicked(MouseEvent e) {
				// TODO Auto-generated method stub
				JPopupMenu menu = new JPopupMenu();
				TreePath tp = tree.getSelectionPath();
				if (tp != null) {
					Object opath[] = tp.getPath();
					//System.out.println("valueChanged...1");
					DefaultMutableTreeNode	n = (DefaultMutableTreeNode)opath[opath.length-1];
					if (n.getUserObject() instanceof ServerTreeViewSelectionListener) {
						//System.out.println("valueChanged...2");
						((ServerTreeViewSelectionListener)n.getUserObject()).doSelection(tree, n);
						tree.expandPath(tp);
						tree.setSelectionPath(tp);
					}
					//System.out.println("valueChanged...3");

					if(opath.length == 1){
						//选中业务列表
						newServiceAction.setEnabled(true);			
						modifyServiceAction.setEnabled(false);
						newNodeAction.setEnabled(true);
						modifyNodeAction.setEnabled(false);

						toolNewServiceAction.setEnabled(true);			
						toolModifyServiceAction.setEnabled(false);
						toolNewNodeAction.setEnabled(true);
						toolModifyNodeAction.setEnabled(false);
					}
					else if (opath.length == 2) {
						if((((DefaultMutableTreeNode)opath[1]).getUserObject() instanceof ServiceNode)){
							//选中业务节点
							newServiceAction.setEnabled(false);			
							modifyServiceAction.setEnabled(true);
							newNodeAction.setEnabled(true);
							modifyNodeAction.setEnabled(false);

							toolNewServiceAction.setEnabled(false);			
							toolModifyServiceAction.setEnabled(true);
							toolNewNodeAction.setEnabled(true);
							toolModifyNodeAction.setEnabled(false);
						}
						else{
							//选中逻辑节点
							newServiceAction.setEnabled(false);			
							modifyServiceAction.setEnabled(false);
							newNodeAction.setEnabled(false);
							
							toolNewServiceAction.setEnabled(false);			
							toolModifyServiceAction.setEnabled(false);
							toolNewNodeAction.setEnabled(false);
							if (n.getUserObject() instanceof BusinessNode){
								modifyNodeAction.setEnabled(true);
								toolModifyNodeAction.setEnabled(true);
							}
						}
					}
					else{
						//选中逻辑节点
						newServiceAction.setEnabled(false);			
						modifyServiceAction.setEnabled(false);
						newNodeAction.setEnabled(false);

						toolNewServiceAction.setEnabled(false);			
						toolModifyServiceAction.setEnabled(false);
						toolNewNodeAction.setEnabled(false);

						if (n.getUserObject() instanceof BusinessNode){
							modifyNodeAction.setEnabled(true);
							toolModifyNodeAction.setEnabled(true);
						}
					}
					
					Object userObject = n.getUserObject();
					if (userObject != null && userObject instanceof Deleteable) {
						deleteAction.setEnabled(true);
						deleteAction.putValue(Action.NAME,"删除 "+((Deleteable)userObject).getDescription());
						tree.getActionMap().put(deleteAction,deleteAction);

						toolDeleteAction.setEnabled(true);
						toolDeleteAction.putValue(Action.NAME,"删除 "+((Deleteable)userObject).getDescription());
						tree.getActionMap().put(toolDeleteAction,toolDeleteAction);
					} else {
						deleteAction.setEnabled(false);
						deleteAction.putValue(Action.NAME,"删除");	
						tree.getActionMap().remove(deleteAction);

						toolDeleteAction.setEnabled(false);
						toolDeleteAction.putValue(Action.NAME,"删除");	
						tree.getActionMap().remove(toolDeleteAction);
					}
					
					deleteAction.setEnabled(n.getUserObject() instanceof Deleteable);
					toolDeleteAction.setEnabled(n.getUserObject() instanceof Deleteable);
				} else {
					newServiceAction.setEnabled(false);			
					modifyServiceAction.setEnabled(false);			
					newNodeAction.setEnabled(false);
					modifyNodeAction.setEnabled(false);

					toolNewServiceAction.setEnabled(false);			
					toolModifyServiceAction.setEnabled(false);			
					toolNewNodeAction.setEnabled(false);
					toolModifyNodeAction.setEnabled(false);

					deleteAction.setEnabled(false);	
					deleteAction.putValue(Action.NAME,"删除");	
					tree.getActionMap().remove(deleteAction);

					toolDeleteAction.setEnabled(false);	
					toolDeleteAction.putValue(Action.NAME,"删除");	
					tree.getActionMap().remove(toolDeleteAction);
				}
				if(e.getButton() == MouseEvent.BUTTON3){
					menu.add(newServiceAction);
					menu.add(modifyServiceAction);
					menu.addSeparator();
					menu.add(getNewNodeAction());
					menu.add(getModifyNodeAction());
					menu.addSeparator();
					menu.add(getDeleteAction());
					menu.show(tree, e.getX(), e.getY());
				}
			}

			public void mousePressed(MouseEvent e) {
				// TODO Auto-generated method stub
				
			}

			public void mouseReleased(MouseEvent e) {
				// TODO Auto-generated method stub
				
			}

			public void mouseEntered(MouseEvent e) {