Icon was changed to default icon when editting a node on JTree

Question

I have a tree with icon on nodes. However, when I edit the node, the icon is changed to default icon.
I don't known how to write the treeCellEditor to fix that one.
The following is my code:
package description.ui;
import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import javax.swing.ImageIcon;
import javax.swing.JFrame;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.ToolTipManager;
import javax.swing.WindowConstants;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellEditor;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;
public class Tree extends javax.swing.JPanel {
     private JTree tree;
     private JScrollPane jScrollPane1;
     public static void main(String[] args) {
          JFrame frame = new JFrame();
          frame.getContentPane().add(new Tree());
          frame.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
          frame.pack();
          frame.show();
     public Tree() {
          super();
          initGUI();
     private void initGUI() {
          try {
               BorderLayout thisLayout = new BorderLayout();
               this.setLayout(thisLayout);
               setPreferredSize(new Dimension(400, 300));
                jScrollPane1 = new JScrollPane();
                this.add(jScrollPane1, BorderLayout.CENTER);
                    DefaultMutableTreeNode rootNode = createNode();
                    tree = new JTree(rootNode);
                    tree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
                    jScrollPane1.setViewportView(tree);
                    ToolTipManager.sharedInstance().registerComponent(tree);
                    MyCellRenderer cellRenderer = new MyCellRenderer();
                    tree.setCellRenderer(cellRenderer);
                    tree.setEditable(true);
                    tree.setCellEditor(new DefaultTreeCellEditor(tree, cellRenderer));
                    //tree.setCellEditor(new MyCellEditor(tree, cellRenderer));
          } catch (Exception e) {
               e.printStackTrace();
     private void btRemoveActionPerformed(ActionEvent evt) {
         TreePath path = tree.getSelectionPath();
         DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode)path.getLastPathComponent();
         ((DefaultTreeModel)tree.getModel()).removeNodeFromParent(selectedNode);
     private DefaultMutableTreeNode createNode() {
         DefaultMutableTreeNode rootNode = new DefaultMutableTreeNode("Doc");
         DefaultMutableTreeNode ch1 = createChuongNode(rootNode, "Ch1");
         DefaultMutableTreeNode ch2 = createChuongNode(rootNode, "Ch2");
         createTextLeafNode(ch1, "title");
         return rootNode;
     private DefaultMutableTreeNode createChuongNode(DefaultMutableTreeNode parent, String name) {
         DefaultMutableTreeNode node = new DefaultMutableTreeNode(new ChapterNodeData(name));
         parent.add(node);
         return node;
     private DefaultMutableTreeNode createTextLeafNode(DefaultMutableTreeNode parent, String name) {
         DefaultMutableTreeNode node = new DefaultMutableTreeNode(new TitleNodeData(name));
         parent.add(node);
         return node;
      private class MyCellRenderer extends DefaultTreeCellRenderer {
             ImageIcon titleIcon;
             ImageIcon chapterIcon;
             public MyCellRenderer() {
                 titleIcon = new ImageIcon(getClass().getClassLoader()
                        .getResource("description/ui/icons/Text16.gif"));
                 chapterIcon = new ImageIcon(getClass().getClassLoader()
                        .getResource("description/ui/icons/Element16.gif"));
             public Component getTreeCellRendererComponent(
                                 JTree tree,
                                 Object value,
                                 boolean sel,
                                 boolean expanded,
                                 boolean leaf,
                                 int row,
                                 boolean hasFocus) {
                 super.getTreeCellRendererComponent(
                                 tree, value, sel,
                                 expanded, leaf, row,
                                 hasFocus);
                 if (isChapterNode(value)) {
                     setIcon(chapterIcon);
                     setToolTipText("chapter");
                 } else if (isTextLeafNode(value)) {
                     setIcon(titleIcon);
                     setToolTipText("title");
                 return this;
             protected boolean isChapterNode(Object node) {
                 return ((DefaultMutableTreeNode)node).getUserObject() instanceof ChapterNodeData;
             protected boolean isTextLeafNode(Object node) {
                 return ((DefaultMutableTreeNode)node).getUserObject() instanceof TitleNodeData;
      private class MyCellEditor extends DefaultTreeCellEditor {
             ImageIcon titleIcon;
             ImageIcon chapterIcon;
          public MyCellEditor(JTree tree, DefaultTreeCellRenderer renderer) {
              super(tree, renderer);
              titleIcon = new ImageIcon(getClass().getClassLoader()
                     .getResource("description/ui/icons/Text16.gif"));
              titleIcon = new ImageIcon(getClass().getClassLoader()
                     .getResource("description/ui/icons/Element16.gif"));
          public Component getTreeCellEditorComponent(
                       JTree tree,
                       Object value,
                       boolean isSelected,
                       boolean expanded,
                       boolean leaf,
                       int row) {
              super.getTreeCellEditorComponent(tree, value, isSelected, expanded, leaf, row);
              return this.editingComponent;
      abstract class NodeData{
          String name;
          public NodeData(String name) {
              this.name = name;
          public String getName() {
              return name;
          public void setName(String name) {
              this.name = name;
          public String toString() {
              return name;
      class ChapterNodeData extends NodeData {
          public ChapterNodeData(String s) {
              super(s);
      class TitleNodeData extends NodeData {
          public TitleNodeData(String attr) {
              super(attr);
}

Answer 1

Arungeeth wrote:
I know the name of the node... but i cant able to find that nodeHere is some sample code for searching and selecting a node:
    TreeModel model = jtemp.getModel();
    if (model != null) {
        Object root = model.getRoot();
        search(model, root, "Peter");//search for the name 'Peter'
        System.out.println(jtemp.getSelectionPath().getLastPathComponent());
    } else {
        System.out.println("Tree is empty.");
private void search(TreeModel model, Object o, String argSearch) {
    int cc;
    cc = model.getChildCount(o);
    for (int i = 0; i < cc; i++) {
        DefaultMutableTreeNode child = (DefaultMutableTreeNode) model.getChild(o, i);
        if (model.isLeaf(child)) {
            TreeNode[] ar = child.getPath();
            String currentValue = Arrays.toString(ar);
            if (currentValue.contains(argSearch)) {
                jtemp.setSelectionPath(new TreePath(ar));
        } else {
            search(model, child, argSearch);
}