Swing/CheckBoxNodeTree のバックアップ(No.2)

TITLE:JTreeの葉ノードをJCheckBoxにする

Posted by aterai at 2011-03-21

JTreeの葉ノードをJCheckBoxにする

JTreeの葉ノードを編集可能なJCheckBoxにします。

  • &jnlp;
  • &jar;
  • &zip;
CheckBoxNodeTree.png

サンプルコード

class CheckBoxNode {
  public final String text;
  public final boolean selected;
  public CheckBoxNode(String text, boolean selected) {
    this.text = text;
    this.selected = selected;
  }
  @Override public String toString() {
    return text;
  }
}

class CheckBoxNodeRenderer extends JCheckBox implements TreeCellRenderer {
  @Override public Component getTreeCellRendererComponent(JTree tree,
      Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {
    this.tree = tree;
    if(leaf && value != null && value instanceof DefaultMutableTreeNode) {
      this.setEnabled(tree.isEnabled());
      this.setFont(tree.getFont());
      this.setOpaque(false);
      Object userObject = ((DefaultMutableTreeNode)value).getUserObject();
      if(userObject!=null && userObject instanceof CheckBoxNode) {
        CheckBoxNode node = (CheckBoxNode)userObject;
        this.setText(node.text);
        this.setSelected(node.selected);
      }
      return this;
    }
    return renderer.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);
  }
  private JTree tree = null;
  private DefaultTreeCellRenderer renderer = new DefaultTreeCellRenderer();
  @Override public void updateUI() {
    super.updateUI();
    setName("Tree.cellRenderer");
    //1.6.0_24 bug??? @see 1.7.0 DefaultTreeCellRenderer#updateUI()
    renderer = new DefaultTreeCellRenderer();
    if(tree!=null) { //update all node width???
      DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
      DefaultMutableTreeNode root = (DefaultMutableTreeNode)model.getRoot();
      java.util.Enumeration depth = root.depthFirstEnumeration();
      while(depth.hasMoreElements()) {
        model.nodeChanged((TreeNode)depth.nextElement());
      }
    }
  }
}

class CheckBoxNodeEditor extends JCheckBox implements TreeCellEditor {
  private final JTree tree;
  public CheckBoxNodeEditor(JTree tree) {
    super();
    this.tree = tree;
    setFocusable(false);
    setRequestFocusEnabled(false);
    setOpaque(false);
    addActionListener(new ActionListener() {
      @Override public void actionPerformed(ActionEvent e) {
        stopCellEditing();
      }
    });
  }
  @Override public Component getTreeCellEditorComponent(JTree tree,
      Object value, boolean isSelected, boolean expanded, boolean leaf, int row) {
    if(leaf && value != null && value instanceof DefaultMutableTreeNode) {
      Object userObject = ((DefaultMutableTreeNode)value).getUserObject();
      if(userObject!=null && userObject instanceof CheckBoxNode) {
        this.setSelected(((CheckBoxNode)userObject).selected);
      }else{
        this.setSelected(false);
      }
      this.setText(value.toString());
    }
    return this;
  }
  @Override public Object getCellEditorValue() {
    return new CheckBoxNode(getText(), isSelected());
  }
  @Override public boolean isCellEditable(EventObject e) {
    if(e != null && e instanceof MouseEvent) {
      TreePath path = tree.getPathForLocation(
          ((MouseEvent)e).getX(), ((MouseEvent)e).getY());
      Object o = path.getLastPathComponent();
      if(o!=null && o instanceof TreeNode) {
        return ((TreeNode)o).isLeaf();
      }
    }
    return false;
  }
//......

解説

JCheckBoxを継承するTreeCellRenderer、TreeCellEditorを作成して、setEditable(true)としたJTreeに設定し、葉ノードをチェックできるようにしています。ノードがチェックされているかどうかといった状態の保存は、DefaultMutableTreeNode#setUserObject(Object)でタイトルと選択状態をもつオブジェクトを設定することで行っています。

葉ノードでない場合、表示にはDefaultTreeCellRendererを使っています。逆にDefaultTreeCellRendererを継承するTreeCellRendererで、葉ノードの表示をJCheckBoxに委譲する方法でも同様となりますが、この場合、JDK 1.6.0_xx では*1 Look And Feel を変更してもアイコンや選択色が更新されないようです。

葉ノードだけ編集可能に制限するため、TreeCellEditor#isCellEditable(EventObject)でクリックしたノードがTreeNode#isLeaf()かを判断しています。

参考リンク

コメント