widecelleditor.java

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

/**
 * 
 */
package flow.graph.gui.tree;

import java.awt.Component;
import java.awt.Dimension;
import java.awt.Rectangle;

import javax.swing.JTree;
import javax.swing.tree.DefaultTreeCellEditor;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreePath;

/**
 * @author Administrator
 *
 */
public class WideCellEditor extends DefaultTreeCellEditor {

	  public WideCellEditor(JTree tree, DefaultTreeCellRenderer renderer) {
	    this(tree, renderer, null);
	  }

	  public WideCellEditor(final JTree tree,
	      final DefaultTreeCellRenderer renderer, TreeCellEditor editor) {
	    super(tree, renderer, editor);
	    editingContainer = new WideEditorContainer();
	  }

	  public Component getTreeCellEditorComponent(JTree tree, Object value,
	      boolean isSelected, boolean expanded, boolean leaf, int row) {
	    Component c = super.getTreeCellEditorComponent(tree, value, isSelected,
	        expanded, leaf, row);
	    ((WideEditorContainer) editingContainer).setLocalCopy(tree, lastPath,
	        offset, editingComponent);
	    return c;
	  }

	  class WideEditorContainer extends DefaultTreeCellEditor.EditorContainer {
	    JTree tree;

	    TreePath lastPath;

	    int offset;

	    Component editingComponent;

	    public void doLayout() {
	      if (editingComponent != null) {
	        Dimension cSize = getSize();
	        Dimension eSize = editingComponent.getPreferredSize();
	        int n = lastPath.getPathCount();
	        Rectangle r = new Rectangle();
	        r = tree.getBounds(r);
	        eSize.width = r.width - (offset * n);
	        editingComponent.setSize(eSize);
	        editingComponent.setLocation(offset, 0);
	        editingComponent
	            .setBounds(offset, 0, eSize.width, cSize.height);
	        setSize(new Dimension(eSize.width + offset, cSize.height));
	      }
	    }

	    void setLocalCopy(JTree tree, TreePath lastPath, int offset,
	        Component editingComponent) {
	      this.tree = tree;
	      this.lastPath = lastPath;
	      this.offset = offset;
	      this.editingComponent = editingComponent;
	    }
	  }
	}