带选择框的JTree

CheckNode.java

import java.util.Enumeration;
import javax.swing.tree.DefaultMutableTreeNode;

/**
 * ChcekNodeTree树的结点
 * @version 1.1 01/15/99
 */
public 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 e = children.elements();
   while (e.hasMoreElements()) {
    CheckNode node = (CheckNode) e.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); } }
  */

}

CheckNodeTree.java

import javax.swing.Icon;
import javax.swing.JCheckBox;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.UIManager;
import javax.swing.plaf.ColorUIResource;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;
import javax.swing.tree.TreeModel;

/**
 * 带Chcek的JTree示例
 * @version 1.1 01/15/99
 */
public class CheckNodeTree extends JTree {

 public CheckNodeTree() {
  this(new DefaultTreeModel(null));
 }

 public CheckNodeTree(TreeModel model) {
  super(model);

  setCellRenderer(new CheckRenderer());
  getSelectionModel().setSelectionMode(
    TreeSelectionModel.SINGLE_TREE_SELECTION
  );
  addMouseListener(new NodeSelectionListener(this));
 }

 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);
   if (path != null) {
    DefaultMutableTreeNode node = (DefaultMutableTreeNode)path.getLastPathComponent();
    if (node instanceof CheckNode) {
     CheckNode chNode = (CheckNode) node;
     boolean isSelected = ! (chNode.isSelected());
     chNode.setSelected(isSelected);
     if (chNode.getSelectionMode() == CheckNode.DIG_IN_SELECTION) {
      if ( isSelected) {
       tree.expandPath(path);
      } else {
       tree.collapsePath(path);
      }
     }
     ((DefaultTreeModel) tree.getModel()).nodeChanged(node);
     // I need revalidate if node is root.  but why?
       if (row == 0) {
        tree.revalidate();
        tree.repaint();
       }
    }
   }
  }
 }

 public static void main(String args[]) {
  String[] strs = {"swing",     // 0
    "platf",     // 1
    "basic",     // 2
    "metal",     // 3
  "JTree"};    // 4

  CheckNode[] nodes = new CheckNode[strs.length];
  DefaultMutableTreeNode root = new DefaultMutableTreeNode("root");
  for (int i=0;i<strs.length;i++) {
   nodes[i] = new CheckNode(strs[i]);
  }
  root.add(nodes[0]);
  nodes[0].add(nodes[1]);
  nodes[1].add(nodes[2]);
  nodes[1].add(nodes[3]);
  nodes[0].add(nodes[4]);
  DefaultTreeModel model = new DefaultTreeModel(root);
  CheckNodeTree tree = new CheckNodeTree(model);
  JFrame frame = new JFrame("测试");
  frame.addWindowListener(new WindowAdapter() {
   public void windowClosing(WindowEvent e) {
    System.exit(0);
   }
  });
  JScrollPane jsp = new JScrollPane(tree);
  frame.getContentPane().setLayout(new BorderLayout());
  frame.getContentPane().add(jsp, BorderLayout.CENTER);
  frame.setSize(300, 200);
  frame.setVisible(true);
 }
}

class CheckRenderer extends JPanel implements TreeCellRenderer {
 protected JCheckBox check;

 protected TreeLabel label;
 DefaultTreeCellRenderer dtcr = new DefaultTreeCellRenderer();
 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) {
  if (!(value instanceof CheckNode)) {
   return dtcr.getTreeCellRendererComponent(tree, value, isSelected, expanded, leaf, row, 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;
  }
 }
}

效果图