Jtree (节点的渲染+资源管理器)

我们的还是自定义的Jtree的类：

package jtree.customNode;

import java.io.File;

import javax.swing.JTree;
import javax.swing.tree.DefaultTreeModel;

import util.ImageManager;
import jtree.checkBoxTree.CheckBoxTreeNodeListender;

public class ZTreeCusNode extends JTree {

    private static final long serialVersionUID = -581164150235777786L;
    private static final String ROOT_NAME = "我的电脑";
    private static final int levelUp = 3;

     public ZTreeCusNode() {
        this.setModel(new DefaultTreeModel(createRootNode()));
        this.setCellRenderer(new CusNodeTreeCellRender());
        this.addCheckSelectListender();
    }

     private void addCheckSelectListender() {
         this.addMouseListener(new CheckBoxTreeNodeListender(this));
    }

    public  CusTreeNode createRootNode(){
        CusTreeNode treeNode = null;
        CusTreeNode rootNode = new CusTreeNode(ROOT_NAME,ImageManager.getImageIconByShortName("home.png"));
          for(int i = 0; i < File.listRoots().length ; i++){
              if(File.listRoots()[i].isDirectory()){
                  String rootPath = File.listRoots()[i].getPath();
                  treeNode  = creatDefaultMutableTreeNode(rootPath,0);
                  rootNode.add(treeNode);
                  treeNode = null;
              }
          } 

            return rootNode;
        }

    private CusTreeNode creatDefaultMutableTreeNode(String nodePath,int level) {
        CusTreeNode node = null;
        if(level == 0){
            node = new CusTreeNode(nodePath,ImageManager.getImageIconByShortName("pan.png"));
        }else if(level == 1){
            node = new CusTreeNode(nodePath,ImageManager.getImageIconByShortName("ff.png"));
        }else{
            node = new CusTreeNode(nodePath);
        }
        CusTreeNode treeNode = null;
        level = level+1;
        File file = new File(nodePath);
        if(file.isDirectory() && file.listFiles() != null){
             for(int i = 0; i < file.listFiles().length && level < levelUp; i++){
                  if(file.listFiles()[i].isDirectory()){
                      String rootPath = file.listFiles()[i].getPath();
                      treeNode  = creatDefaultMutableTreeNode(rootPath,level);
                      node.add(treeNode);
                      treeNode = null;
                  }
              }
        }

        return node;
    }
}

但是因为各个节点的图标，可能比一样，我们把图标的信息集合到节点中去，这样的话，可以根据条件给不同的节点设置不同的节点，显示的时候，直接从节点中取出来即可，根据这种分析，我们需要自定义DefaultMutableTreeNode 。

package jtree.customNode;

import javax.swing.Icon;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreeSelectionModel;

public class CusTreeNode extends DefaultMutableTreeNode {

    private static final long serialVersionUID = 3311434895405929201L;

    private boolean isSelected = false;

    private boolean enabled = false;

    private boolean isVisible = false;

    private Icon icon = null;

    private String iconName = null;

    private int selectionMode = 0;

    public CusTreeNode(Object userObject,Icon icon) {
        this(userObject, true, false, true, true, icon);
    }

    public CusTreeNode(String nodePath) {
        this(nodePath, true, false, true, true, null);
    }

    public CusTreeNode(Object userObject, boolean allowsChildren,

            boolean isSelected, boolean enabled, boolean isVisible, Icon icon) {

            super(userObject, allowsChildren);

            this.isSelected = isSelected;

            this.enabled = enabled;

            this.isVisible = isVisible;

            this.icon = icon;

            setSelectionMode(TreeSelectionModel.DISCONTIGUOUS_TREE_SELECTION);

        }

    public boolean isSelected() {
        return isSelected;
    }

    public void setSelected(boolean isSelected) {
        this.isSelected = isSelected;
    }

    public boolean isEnabled() {
        return enabled;
    }

    public void setEnabled(boolean enabled) {
        this.enabled = enabled;
    }

    public boolean isVisible() {
        return isVisible;
    }

    public void setVisible(boolean isVisible) {
        this.isVisible = isVisible;
    }

    public Icon getIcon() {
        return icon;
    }

    public void setIcon(Icon icon) {
        this.icon = icon;
    }

    public String getIconName() {
        return iconName;
    }

    public void setIconName(String iconName) {
        this.iconName = iconName;
    }

    public int getSelectionMode() {
        return selectionMode;
    }

    public void setSelectionMode(int selectionMode) {
        this.selectionMode = selectionMode;
    }

}

然后是节点的渲染如下：

package jtree.customNode;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;

import javax.swing.Icon;
import javax.swing.JPanel;
import javax.swing.JTree;
import javax.swing.UIManager;
import javax.swing.plaf.ColorUIResource;
import javax.swing.tree.TreeCellRenderer;

import jtree.checkBoxTree.CheckBoxTreeLabel;

public class CusNodeTreeCellRender extends JPanel implements TreeCellRenderer {

    private static final long serialVersionUID = 4676667399191240255L;

    protected CheckBoxTreeLabel label;

    // protected JLabel label;
    public CusNodeTreeCellRender() {
        setLayout(new BorderLayout());
        add(label = new CheckBoxTreeLabel());
        // add(label = new JLabel());
        label.setForeground(UIManager.getColor("Tree.textForeground"));
        this.setPreferredSize(new Dimension(100, 20));
    }

    /*
     * (non-Javadoc)
     *
     * @see
     * javax.swing.tree.TreeCellRenderer#getTreeCellRendererComponent(javax.
     * swing.JTree, java.lang.Object, boolean, boolean, boolean, int, boolean)
     */
    @Override
    public Component getTreeCellRendererComponent(JTree tree, Object value,
            boolean selected, boolean expanded, boolean leaf, int row,
            boolean hasFocus) {
        String stringValue = tree.convertValueToText(value, selected, expanded,
                leaf, row, hasFocus);
        boolean nodeIsEnabled = ((CusTreeNode) value).isEnabled();
        setEnabled(nodeIsEnabled);

        label.setFont(tree.getFont());
        label.setText(stringValue);
        label.setSelected(selected);
        label.setFocus(hasFocus);
        if (leaf)
            label.setIcon(UIManager.getIcon("Tree.leafIcon"));
        else if (expanded)
            label.setIcon(UIManager.getIcon("Tree.openIcon"));
        else{
             label.setIcon(UIManager.getIcon("Tree.closedIcon"));
        }
        Icon icon = ((CusTreeNode) value).getIcon();
        if(icon != null){
             label.setIcon(icon);
        }

        return this;
    }

    @Override
    public void setBackground(Color color) {
        if (color instanceof ColorUIResource)
            color = null;
        super.setBackground(color);
    }
}

测试代码：

public class TestZTreeCusNode {

    /**
     * @param args
     */
    public static void main(String[] args) {
        ZTreeCusNode tree = new ZTreeCusNode();
        JScrollPane scroll = new JScrollPane(tree);
        //测试的环境
        JFrame test = new JFrame();
        test.setSize(450, 330);
        test.setLocation((test.getToolkit().getScreenSize().width - 450) / 2,
                (test.getToolkit().getScreenSize().height - 330) / 2);
        test.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        test.add(scroll);
        test.setVisible(true);

    }

}