classtreenoderender.java

来自「这是自定义工作流的图形定义部分」· Java 代码 · 共 55 行

import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import java.awt.*;


/**
 * Title:        work-flow
 * Description:
 * Copyright:    Copyright (c) 2002
 * Company:      CSU
 * @author 	     smallbeetle
 * @version 1.0
 */

public class  ClassTreeNodeRender extends DefaultTreeCellRenderer{
  Icon methodIcon = null;
  Icon propertyIcon = null;
  Icon classIcon = null;
  Icon defaultIcon = null;

  public ClassTreeNodeRender (){
    methodIcon = new ImageIcon((Image)csuSystem.getObject("method.gif"));
    propertyIcon = new ImageIcon((Image)csuSystem.getObject("property.gif"));
    classIcon = new ImageIcon((Image)csuSystem.getObject("class.gif"));
    defaultIcon =  new ImageIcon((Image)csuSystem.getObject("default.gif"));
  }

  public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus){
    super.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);
    // get the user object
    try{
      DefaultMutableTreeNode defaultNode = (DefaultMutableTreeNode)value;
      ClassNode node = (ClassNode)defaultNode.getUserObject();
      switch(node.type){
        case ClassNode.CLASS :
          this.setIcon(classIcon) ;
          break;
        case ClassNode.PROPERTY  :
          this.setIcon(propertyIcon) ;
          break;
        case ClassNode.METHOD :
          this.setIcon(methodIcon);
          break;
        default:
          this.setIcon(defaultIcon);
          break;
      }
      this.setText(node.label);
     }catch(Exception ex){}
     return this;
  }
}