Focus Problem with JTree and Menus

Question

Hi all,
I have a problem with focus when editing a JTree and selecting a menu. The problem occurs when the user single clicks on a node, invoking the countdown to edit. If the user quickly clicks on a menu item, the focus will go to the menu item, but then when the countdown finishes, the node now has keyboard focus.
Below is code to reproduce the problem. Click on the node, hover for a little bit, then quickly click on the menu item.
How would I go about fixing the problem? Thanks!
import java.awt.Container;
import java.awt.GridLayout;
import javax.swing.JFrame;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
public class MyTree extends JTree {
     public MyTree(DefaultTreeModel treemodel) {
          setModel(treemodel);
          setRootVisible(true);
     public static void main(String[] args) {
          JFrame frame = new JFrame();
          JMenuBar bar = new JMenuBar();
          JMenu menu = new JMenu("Test");
          JMenuItem item = new JMenuItem("Item");
          menu.add(item);
          bar.add(menu);
          frame.setJMenuBar(bar);
          Container contentPane = frame.getContentPane();
          contentPane.setLayout(new GridLayout(1, 2));
          DefaultMutableTreeNode root1 = new DefaultMutableTreeNode("Root");
          root1.add(new DefaultMutableTreeNode("Root2"));
          DefaultTreeModel model = new DefaultTreeModel(root1);
          MyTree tree1 = new MyTree(model);
          tree1.setEditable(true);
          tree1.setCellEditor(
               new ComponentTreeCellEditor(
                    tree1,
                    new ComponentTreeCellRenderer()));
          tree1.setRowHeight(0);
          contentPane.add(tree1);
          frame.pack();
          frame.show();
import java.awt.FlowLayout;
import java.util.EventObject;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextField;
import javax.swing.JTree;
import javax.swing.tree.DefaultTreeCellEditor;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreeCellRenderer;
public class ComponentTreeCellEditor
     extends DefaultTreeCellEditor
     implements TreeCellEditor {
     private String m_oldValue;
     private TreeCellRenderer m_renderer = null;
     private DefaultTreeModel m_model = null;
     private JPanel m_display = null;
     private JTextField m_field = null;
     private JTree m_tree = null;
     public ComponentTreeCellEditor(
          JTree tree,
          DefaultTreeCellRenderer renderer) {
          super(tree, renderer);
          m_renderer = renderer;
          m_model = (DefaultTreeModel) tree.getModel();
          m_tree = tree;
          m_display = new JPanel(new FlowLayout(FlowLayout.LEFT));
          m_field = new JTextField();
          m_display.add(new JLabel("My Label "));
          m_display.add(m_field);
     public java.awt.Component getTreeCellEditorComponent(
          JTree tree,
          Object value,
          boolean isSelected,
          boolean expanded,
          boolean leaf,
          int row) {
          m_field.setText(value.toString());
          return m_display;
     public Object getCellEditorValue() {
          return m_field.getText();
      * The edited cell should always be selected.
      * @param anEvent the event that fired this function.
      * @return true, always.
     public boolean shouldSelectCell(EventObject anEvent) {
          return true;
      * Only edit immediately if the event is null.
      * @param event the event being studied
      * @return true if event is null, false otherwise.
     protected boolean canEditImmediately(EventObject event) {
          return (event == null);
}

Answer 1

You can provide a cell renderer to the JTree to paint the checkBox.
The following code checks or unchecks the box with each click also:
_tree.setCellRenderer(new DefaultTreeCellRenderer()
  private JCheckBox checkBox = null;
  public Component getTreeCellRendererComponent(JTree tree,
                                                Object value,
                                                boolean selected,
                                                boolean expanded,
                                                boolean leaf,
                                                int row,
                                                boolean hasFocus)
    // Each node will be drawn as a check box
    if (checkBox == null)
      checkBox  = new JCheckBox();
      checkBox .setBackground(tree.getBackground());
    DefaultMutableTreeNode node = (DefaultMutableTreeNode)value;
    checkBox.setText(node.getUserObject().toString());
    checkBox.setSelected(selected);
    return checkBox;
});