java 构造带复选框的树（CheckTree.java）

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.Graphics;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.util.Enumeration;

import javax.swing.*;

import javax.swing.JPanel;

import javax.swing.JTree;

import javax.swing.plaf.ColorUIResource;

import javax.swing.tree.DefaultMutableTreeNode;

import javax.swing.tree.DefaultTreeModel;

import javax.swing.tree.TreeCellRenderer;

import javax.swing.tree.TreeNode;

import javax.swing.tree.TreePath;

public class CheckTree{

    CheckNode checknode;

    CheckRenderer checkrenderer;

    NodeSelectionListener nodeselectionlistene;

    public CheckTree(JTree tree){

        checknode=new CheckNode(); //构造树

        checkrenderer=new CheckRenderer(); //添加复选框

        nodeselectionlistene=new NodeSelectionListener(tree); //监听复选

    }

}

class CheckNode extends DefaultMutableTreeNode {

    public final static int SINGLE_SELECTION = 0;

    public final static int DIG_IN_SELECTION = 4;

    protected int selectionMode;

    protected boolean isSelected;

    public CheckNode() {

        this(null);

    }

    public CheckNode(Object userObject) {

        this(userObject, true, false);

    }

    public CheckNode(Object userObject, boolean allowsChildren, boolean isSelected) {

        super(userObject, allowsChildren);

        this.isSelected = isSelected;

        setSelectionMode(DIG_IN_SELECTION);

    }

    public void setSelectionMode(int mode) {

        selectionMode = mode;

    }

    public int getSelectionMode() {

        return selectionMode;

    }

    public void setSelected(boolean isSelected) {

        this.isSelected = isSelected;

        if ((selectionMode == DIG_IN_SELECTION) && (children != null)) {

            Enumeration Enum = children.elements();

            while (Enum.hasMoreElements()) {

                CheckNode node = (CheckNode) Enum.nextElement();

                node.setSelected(isSelected);

            }

        }

    }

    public boolean isSelected() {

        return isSelected;

    }

    // If you want to change "isSelected" by CellEditor,

/*

    public void setUserObject(Object obj) {

    if (obj instanceof Boolean) {

    setSelected(((Boolean)obj).booleanValue());

    } else {

    super.setUserObject(obj);

    }

    }

     */

}

class CheckRenderer extends JPanel implements TreeCellRenderer {

    protected JCheckBox check;

    protected TreeLabel label;

    public CheckRenderer() {

        setLayout(null);

        add(check = new JCheckBox());

        add(label = new TreeLabel());

        check.setBackground(UIManager.getColor("Tree.textBackground"));

        label.setForeground(UIManager.getColor("Tree.textForeground"));

    }

    public Component getTreeCellRendererComponent(JTree tree, Object value,

            boolean isSelected, boolean expanded,

            boolean leaf, int row, boolean hasFocus) {

        String stringValue = tree.convertValueToText(value, isSelected,

                expanded, leaf, row, hasFocus);

        setEnabled(tree.isEnabled());

        check.setSelected(((CheckNode) value).isSelected());

        label.setFont(tree.getFont());

        label.setText(stringValue);

        label.setSelected(isSelected);

        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"));

        }

        return this;

    }

    public Dimension getPreferredSize() {

        Dimension d_check = check.getPreferredSize();

        Dimension d_label = label.getPreferredSize();

        return new Dimension(d_check.width + d_label.width,

                (d_check.height < d_label.height ? d_label.height : d_check.height));

    }

    public void doLayout() {

        Dimension d_check = check.getPreferredSize();

        Dimension d_label = label.getPreferredSize();

        int y_check = 0;

        int y_label = 0;

        if (d_check.height < d_label.height) {

            y_check = (d_label.height - d_check.height) / 2;

        } else {

            y_label = (d_check.height - d_label.height) / 2;

        }

        check.setLocation(0, y_check);

        check.setBounds(0, y_check, d_check.width, d_check.height);

        label.setLocation(d_check.width, y_label);

        label.setBounds(d_check.width, y_label, d_label.width, d_label.height);

    }

    public void setBackground(Color color) {

        if (color instanceof ColorUIResource) {

            color = null;

        }

        super.setBackground(color);

    }

    public class TreeLabel extends JLabel {

        boolean isSelected;

        boolean hasFocus;

        public TreeLabel() {

        }

        public void setBackground(Color color) {

            if (color instanceof ColorUIResource) {

                color = null;

            }

            super.setBackground(color);

        }

        public void paint(Graphics g) {

            String str;

            if ((str = getText()) != null) {

                if (0 < str.length()) {

                    if (isSelected) {

                        g.setColor(UIManager.getColor("Tree.selectionBackground"));

                    } else {

                        g.setColor(UIManager.getColor("Tree.textBackground"));

                    }

                    Dimension d = getPreferredSize();

                    int imageOffset = 0;

                    Icon currentI = getIcon();

                    if (currentI != null) {

                        imageOffset = currentI.getIconWidth() + Math.max(0, getIconTextGap() - 1);

                    }

                    g.fillRect(imageOffset, 0, d.width - 1 - imageOffset, d.height);

                    if (hasFocus) {

                        g.setColor(UIManager.getColor("Tree.selectionBorderColor"));

                        g.drawRect(imageOffset, 0, d.width - 1 - imageOffset, d.height - 1);

                    }

                }

            }

            super.paint(g);

        }

        public Dimension getPreferredSize() {

            Dimension retDimension = super.getPreferredSize();

            if (retDimension != null) {

                retDimension = new Dimension(retDimension.width + 3,

                        retDimension.height);

            }

            return retDimension;

        }

        public void setSelected(boolean isSelected) {

            this.isSelected = isSelected;

        }

        public void setFocus(boolean hasFocus) {

            this.hasFocus = hasFocus;

        }

    }

}

class NodeSelectionListener extends MouseAdapter {

    JTree tree;

    NodeSelectionListener(JTree tree) {

        this.tree = tree;

    }

    public void mouseClicked(MouseEvent e) {

        int x = e.getX();

        int y = e.getY();

        int row = tree.getRowForLocation(x, y);

        TreePath path = tree.getPathForRow(row);

        //TreePath path = tree.getSelectionPath();

        if (path != null) {

            CheckNode node = (CheckNode) path.getLastPathComponent();

            boolean isSelected = !(node.isSelected());

            node.setSelected(isSelected);

            if (node.getSelectionMode() == CheckNode.DIG_IN_SELECTION) {

                if (isSelected) {

                    tree.expandPath(path);

                }

            }

            ((DefaultTreeModel) tree.getModel()).nodeChanged(node);

            // I need revalidate if node is root. but why?

//       if (row == 0) {

            tree.revalidate();

            tree.repaint();

//       }

        }

    }

}