Tree cell disapearing

Question

Hi
I'm using Java 1.5.0.11
I'm trying to make my own tree cell renderer and editor : the goal is to add to the standard tree cell (icon + uneditable text) an icon to the left, to indicate a certain "state". The state can be change by clicking on it, showxing a JPopupMenu to select a new one.
In my proto, it works well, except that when I change one row, it is redrawn completely blank. I need to click on it again to have it appear.
Any idea on why?
Best regards

Answer 1

TreeSelector. java
package treeselectproto;
import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.event.ActionEvent;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.swing.AbstractAction;
import javax.swing.AbstractCellEditor;
import javax.swing.JButton;
import javax.swing.JLabel;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.ScrollPaneConstants;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreePath;
public class TreeSelector {
     protected JTree itsTree;
     protected JScrollPane itsScrollPane;
     protected Map<Object, State> itsStates=new HashMap<Object, State>();
     protected CellRenderer itsRenderer;
     protected JPopupMenu itsPopupMenu = new JPopupMenu();
     public TreeSelector()
          itsTree=new JTree();
          itsRenderer=new CellRenderer();
          itsTree.setCellRenderer(itsRenderer);
          itsTree.setCellEditor(itsRenderer);
          itsTree.setEditable(true);
          buildScrollPane();
          List<State> theStates=State.getPopupStates();
          for(State s:theStates)
               PopupAction a=new PopupAction(s);
               JMenuItem mi=new JMenuItem(a);
               itsPopupMenu.add(mi);
     private void buildScrollPane() {
          itsScrollPane= new JScrollPane(itsTree,
                    ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS,
                    ScrollPaneConstants.HORIZONTAL_SCROLLBAR_ALWAYS);
     public JScrollPane getScrollPane() {
          return itsScrollPane;
     @SuppressWarnings("serial")
     private  class CellRenderer extends AbstractCellEditor implements TreeCellRenderer, TreeCellEditor
          private JPanel itsPanel;
          private JLabel itsLabel;
          private JButton itsStateBtn;
          private BorderLayout itsLayout;
          public  CellRenderer()
               itsPanel=new JPanel();
               itsPanel.setLayout(itsLayout=new BorderLayout());
               itsPanel.setDoubleBuffered(false);
               itsLayout.setVgap(1);
               itsLayout.setHgap(1);
               itsLabel=new JLabel();
               itsLabel.setDoubleBuffered(false);
               itsPanel.add(itsLabel,BorderLayout.CENTER);
               itsStateBtn=new JButton(new AbstractAction(){
                    public void actionPerformed(ActionEvent aE) {
                         showPopup();
               itsStateBtn.setBorderPainted(true);
               itsStateBtn.setContentAreaFilled(false);
               itsStateBtn.setDoubleBuffered(false);
               itsPanel.add(itsStateBtn,BorderLayout.WEST);
          protected void showPopup() {
               itsPopupMenu.show(itsStateBtn, 0, 0);
          public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus)
               TreePath thePath=tree.getPathForRow(row);
               itsPanel.setBackground(tree.getBackground());
               if(thePath != null)
                    State theState=getStateForTreePath(thePath);
                    String ic=theState.getSign();
                    itsStateBtn.setText(ic);
               itsLabel.setText(value.toString());
               itsStateBtn.setPreferredSize(null);
               itsLabel.setPreferredSize(null);
               itsPanel.setBackground( selected? Color.lightGray:tree.getBackground());
               itsLayout.invalidateLayout(itsPanel);
               return itsPanel;
          private Object itsValue;
          public Component getTreeCellEditorComponent(JTree aTree, Object aValue, boolean aIsSelected, boolean aExpanded, boolean aLeaf, int aRow) {
               // if edting, it has the focus
               itsValue=aValue;
               return getTreeCellRendererComponent(aTree,aValue,aIsSelected,aExpanded,aLeaf,aRow,true);
          public Object getCellEditorValue() {
               return itsValue;
     public State getStateForTreePath(TreePath thePath) {
          State ret=itsStates.get(thePath.getLastPathComponent());
          if(ret==null)
               ret=State.defaultState();
          return ret;
     @SuppressWarnings("serial")
     private class PopupAction extends AbstractAction
          protected State itsState;
          public PopupAction(State aState) {
               super(aState.getLabel());
               itsState = aState;
          public void actionPerformed(ActionEvent aE) {
               TreePath thePath=itsTree.getEditingPath();
               Object theObj=thePath.getLastPathComponent();
               itsStates.put(theObj, itsState);
}