servicetreeview.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.KeyEvent;
import java.util.Vector;

import javax.swing.Action;
import javax.swing.Icon;
import javax.swing.JOptionPane;
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 flow.graph.app.WorkFlow;
import flow.graph.db.bean.NodeBean;
import flow.graph.db.bean.ServiceBean;

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);
		}
		//节点提示
		setToolTipText(val.toString());
		return this;
	}
}

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

	Action newServiceAction;
	Action modifyServiceAction;
	Action newNodeAction;
	Action modifyNodeAction;

	Action deleteAction;
	
	public ServiceTreeView(Object rootObject) {
		tree = new JTree();
		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);
		
		ToolTipManager.sharedInstance().registerComponent(tree);
		tree.setCellRenderer(new ServerTreeCellRenderer());
		
		newServiceAction = new SimpleAction("新增业务", new Integer(KeyEvent.VK_A), false) {
			public void actionPerformed(ActionEvent e)	{ insertService(); } 
		};
		modifyServiceAction = new SimpleAction("修改业务", new Integer(KeyEvent.VK_B), false) {
			public void actionPerformed(ActionEvent e)	{ modifyService(); } 
		};

		newNodeAction = new SimpleAction("新增节点", new Integer(KeyEvent.VK_D), false) {
			public void actionPerformed(ActionEvent e)	{ insertNode(); } 
		};
		modifyNodeAction = new SimpleAction("修改节点", new Integer(KeyEvent.VK_E), false) {
			public void actionPerformed(ActionEvent e)	{ modifyNode(); } 
		};
		
		/*
		newQueueAction = new SimpleAction("New Queue", new Integer(KeyEvent.VK_Q), false) {
			public void actionPerformed(ActionEvent e)	{ addQueue(); } 
		};
		newUserAction = new SimpleAction("New User", new Integer(KeyEvent.VK_U), false) {
			public void actionPerformed(ActionEvent e)	{ addUser(); } 
		};
		newGroupAction = new SimpleAction("New Group", new Integer(KeyEvent.VK_G), false) {
			public void actionPerformed(ActionEvent e)	{ addGroup(); }
		};
		*/
		
		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);
	}
	
	/**
	 * 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;
			}
			else{
				n.removeAllChildren();
				n.add(new DefaultMutableTreeNode("No Service Node"));
				model.nodeStructureChanged(n);
				break;
			}
		}
	}
	
	/**
	 * Refereshes the root node's list of children.
	 */
	public void refreshRoot() {
		StartList sl = (StartList)root.getUserObject();
		sl.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);
	}
	
	/**
	 * Called when a node in the <code>tree</code> is selected.
	 * 
	 * @param event Used to determine which node has been selected.
	 * 
	 * @see javax.swing.event.TreeSelectionListener#valueChanged(javax.swing.event.TreeSelectionEvent)
	 */
	public void valueChanged(TreeSelectionEvent event) {
		System.out.println("valueChanged ...");
		TreePath tp = event.getPath();
		if (tp != null) {
			Object opath[] = tp.getPath();
			DefaultMutableTreeNode	n = (DefaultMutableTreeNode)opath[opath.length-1];
			if (n.getUserObject() instanceof TreeViewSelectionListener) {
				((TreeViewSelectionListener)n.getUserObject()).doSelection(tree, n);
			}

			newServiceAction.setEnabled(false);			
			modifyServiceAction.setEnabled(false);
			newNodeAction.setEnabled(false);
			modifyNodeAction.setEnabled(false);
			deleteAction.setEnabled(false);
			System.out.println("*******************length="+opath.length);
			//System.out.println(((DefaultMutableTreeNode)opath[0]).getUserObject());
			if(opath.length == 1){
				newServiceAction.setEnabled(true);			
				modifyServiceAction.setEnabled(false);
				newNodeAction.setEnabled(true);
				modifyNodeAction.setEnabled(false);
			}
			else if (opath.length == 2) {
				System.out.println("valueChanged...1");
				if((((DefaultMutableTreeNode)opath[1]).getUserObject() instanceof ServiceList)){
					newServiceAction.setEnabled(false);			
					modifyServiceAction.setEnabled(true);
					newNodeAction.setEnabled(true);
					modifyNodeAction.setEnabled(false);
				}
				else{
					newServiceAction.setEnabled(false);			
					modifyServiceAction.setEnabled(false);
					newNodeAction.setEnabled(false);
					modifyNodeAction.setEnabled(true);
				}
			}
			else{
				if((((DefaultMutableTreeNode)opath[2]).getUserObject() instanceof Node)){
					newServiceAction.setEnabled(false);			
					modifyServiceAction.setEnabled(false);
					newNodeAction.setEnabled(false);
					modifyNodeAction.setEnabled(true);
				}
			}
			
			Object userObject = n.getUserObject();
			if (userObject != null && userObject instanceof Deleteable) {
				deleteAction.setEnabled(true);