how can I change this tree with different Icons for Different nodes at Run Time

[Michelle Wang]

Dear sir:
One more question again,
I use following code to set icon for the each node
here I assume customLeafIcon has Icon of "Bird.GIF"

   
    DefaultTreeCellRenderer renderer = new DefaultTreeCellRenderer();
    renderer.setOpenIcon(customOpenIcon);
    renderer.setClosedIcon(customClosedIcon);
    renderer.setLeafIcon(customLeafIcon);
    tree.setCellRenderer(renderer);
 

but for the leaf nodes, I have different Icons for different components, For Ex. at RUN Time, If I drop a leaf Node with TextString called "cat" in this Tree, then I need only Cat Icon for this node dropped, same, If I drop a leaf Node with TextString called "tiger" in this Tree, then I need only tiger Icon for this node dropped, and so on, but above mentioned code will change all leaf nodes with SAME Icon of "Bird.GIF", how can I change this tree with different Icons for Different nodes at Run Time, not Design Time??

Thanks in Advance.
Michelle

[Craig Wood]

In the section Customizing a Tree's Display there is an example named TreeIconDemo2 and short discussion about changing leaf icons according to the value of the leaf node at runtime.

[Michelle Wang]

Hello Sir:
Thanks for your advice, I follow your link and study the part,
it is very good, but I met a new problem.
I used the following code from this example:

   
//...where the tree is initialized:
    //Enable tool tips.
    ToolTipManager.sharedInstance().registerComponent(tree);
    
    ImageIcon tutorialIcon = createImageIcon("images/middle.gif");
    if (tutorialIcon != null) {
        tree.setCellRenderer(new MyRenderer(tutorialIcon));
    }
...
class MyRenderer extends DefaultTreeCellRenderer {
    Icon tutorialIcon;
 
    public MyRenderer(Icon icon) {
        tutorialIcon = icon;
    }
 
    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 (leaf && isTutorialBook(value)) {
            setIcon(tutorialIcon);
            setToolTipText("This book is in the Tutorial series.");
        } else {
            setToolTipText(null); //no tool tip
        } 
 
        return this;
    }
 
    protected boolean isTutorialBook(Object value) {
        DefaultMutableTreeNode node =
                (DefaultMutableTreeNode)value;
        BookInfo nodeInfo =
                (BookInfo)(node.getUserObject());
        String title = nodeInfo.bookName;
        if (title.indexOf("Tutorial") >= 0) {
            return true;
        }
 
        return false;
    }
}
 

 

I rewrite class MyRenderer and extend DefaultTreeCellRenderer,
at initial, it successes to put the Icon I need to the specific nodes/leaves; but when I drag n drop other nodes/leaves with different Icon, all nodes replaced with latest nodes'/leaves' Icon.

What is wrong here??
How to keep all previous nodes'/leaves' icons without updating until I require to do so??
Thanks so much.
Michelle

[Michelle Wang]

Can Somebody throw some light on this issue?? I tried all day yesterady, but failed. Thanks a lot.

[Craig Wood]

when I drag n drop other nodes/leaves with different Icon, all nodes replaced with latest
nodes'/leaves' Icon
Does this mean that after any drag_n_drop other nodes have their icon changed to the icon of
the node that was just dropped? If so the example below might be helpful. If not, then I
don't understand your question.

import java.awt.*;
import java.awt.datatransfer.*;
import java.awt.event.*;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.*;
import javax.imageio.ImageIO;
import javax.swing.*;
import javax.swing.tree.*;
 
public class CustomNodesTest {
    private JScrollPane getContent(BufferedImage[] images, String[] ids) {
        ImageIcon[] icons = getScaledIcons(images);
        JTree tree = new JTree(getData(ids));
        tree.setCellRenderer(new CustomNodesRenderer(icons, ids));
        tree.getSelectionModel().setSelectionMode(
                TreeSelectionModel.SINGLE_TREE_SELECTION);
        tree.setDragEnabled(true);
        tree.setTransferHandler(new CustomNodesHandler());
        return new JScrollPane(tree);
    }
 
    private DefaultMutableTreeNode getData(String[] ids) {
        int[][] keys = { { 2 }, { 1, 0 }, { 0 }};
        DefaultMutableTreeNode root = new DefaultMutableTreeNode("root");
        for(int j = 0; j < keys.length; j++) {
            DefaultMutableTreeNode node =
                new DefaultMutableTreeNode("Node " + (j+1));
            root.insert(node, j);
            for(int k = 0; k < keys[j].length; k++) {
                String s = ids[keys[j][k]];
                node.insert(new DefaultMutableTreeNode(s), k);
            }
        }
        return root;
    }
 
    private ImageIcon[] getScaledIcons(BufferedImage[] images) {
        ImageIcon[] icons = new ImageIcon[images.length];
        int w = 30, h = 30, type = BufferedImage.TYPE_INT_RGB;
        for(int j = 0; j < images.length; j++) {
            BufferedImage image = new BufferedImage(w, h, type);
            Graphics2D g2 = image.createGraphics();
            g2.setRenderingHint(RenderingHints.KEY_INTERPOLATION,
                                RenderingHints.VALUE_INTERPOLATION_BICUBIC);
            double xScale = (double)w/images[j].getWidth();
            double yScale = (double)h/images[j].getHeight();
            double scale = Math.max(xScale, yScale);  // fill
            AffineTransform at = AffineTransform.getScaleInstance(scale, scale);
            g2.drawRenderedImage(images[j], at);
            g2.dispose();
            icons[j] = new ImageIcon(image);
        }
        return icons;
    }
 
    public static void main(String[] args) throws IOException {
        String[] ids = { "Dog", "Cat", "Bird" };
        BufferedImage[] images = new BufferedImage[ids.length];
        for(int j = 0; j < images.length; j++) {
            File file = new File("images/" + ids[j] + ".gif");
            images[j] = ImageIO.read(file);
        }
        CustomNodesTest test = new CustomNodesTest();
        JFrame f = new JFrame();
        f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        f.getContentPane().add(test.getContent(images, ids));
        f.setSize(400,400);
        f.setLocation(200,200);
        f.setVisible(true);
    }
}
 
class CustomNodesRenderer extends DefaultTreeCellRenderer {
    ImageIcon[] icons;
    String[] ids;
 
    public CustomNodesRenderer(ImageIcon[] icons, String[] ids) {
        this.icons = icons;
        this.ids = ids;
    }
 
    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(leaf) {
            DefaultMutableTreeNode node = (DefaultMutableTreeNode)value;
            String text = node.getUserObject().toString();
            for(int j = 0; j < ids.length; j++) {
                if(text.equals(ids[j])) {
                    setIcon(icons[j]);
                    break;
                }
            }
        }
        return this;
    }
}
 
class CustomNodesHandler extends TransferHandler {
    DataFlavor flavor = DataFlavor.stringFlavor;
    DefaultMutableTreeNode lastSelection;
 
    public boolean importData(JComponent comp, Transferable t) {
        if(canImport(comp, t.getTransferDataFlavors())) {
            JTree tree = (JTree)comp;
            DefaultMutableTreeNode parent =
                (DefaultMutableTreeNode)tree.getLastSelectedPathComponent();
            try {
                String data = (String)t.getTransferData(flavor);
                DefaultMutableTreeNode child = new DefaultMutableTreeNode(data);
                DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
                model.insertNodeInto(child, parent, parent.getChildCount());
                tree.expandPath(new TreePath(parent.getPath()));
                return true;
            } catch(UnsupportedFlavorException ufe) {
                System.out.println("UnsupportedFlavor: " + ufe.getMessage());
            } catch(IOException ioe) {
                System.out.println("IO Error: " + ioe.getMessage());
            }
        }
        return false;
    }
 
    public boolean canImport(JComponent comp, DataFlavor[] transferFlavors) {
        for(int j = 0; j < transferFlavors.length; j++) {
            if(transferFlavors[j].equals(flavor))
                return true;
        }
        return false;
    }
 
    public int getSourceActions(JComponent c) {
        return COPY_OR_MOVE;
    }
 
    protected void exportDone(JComponent source, Transferable data, int action) {
        if(action == MOVE && lastSelection != null) {
            JTree tree = (JTree)source;
            DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
            model.removeNodeFromParent(lastSelection);
        }
    }
 
    protected Transferable createTransferable(JComponent c) {
        JTree tree = (JTree)c;
        DefaultMutableTreeNode node =
            (DefaultMutableTreeNode)tree.getLastSelectedPathComponent();
        lastSelection = node;
        String text = node.getUserObject().toString();
        return new StringSelection(text);
    }
}

[Michelle Wang]

Does this mean that after any drag_n_drop other nodes have their icon changed to the icon of the node that was just dropped?

Yes, this is what I saw.
thanks so much, it is really exciting. Top code!! can be used for many purpose.
[ May 11, 2007: Message edited by: girl sunny ]

[Michelle Wang]

Dear Sir:
Good weekends, I found another example as follows:

here animalIcon is the Icon of coming node I drag and drop into this JTree,
leafIcon is a common leaf Icon for all leaf nodes;

I use following categories:

[1]. Human -
a. whiteman - whitemanIcon
b. blackman - blackmanIcon
c. redman - redmanIcon

[2]. Fish -
a. goldFish - goldFishIcon
b. swordFish - swordFishIcon
c. shark - sharkIcon
d. catFish - catFishIcon

[3]. Dog -
a. AsianDog - AsianDogIcon
b. USDog - USDogIcon
c. EuroDog - EuroDogIcon

I use following code to implement DnD nodes into JTree, so all existing nodes and their Icons will be kept unchanged after coming node DnD finishes;

  
 
ToolTipManager.sharedInstance().registerComponent(tree);
 
ImageIcon animalIcon = createImageIcon(getObjectIcon());
if (animalIcon != null) {
    tree.setCellRenderer(new MyRenderer(animalIcon));
}
...
class MyRenderer extends DefaultTreeCellRenderer {
Icon animalIcon;
 
public MyRenderer(Icon icon) {
animalIcon = icon;
}
 
super.getTreeCellRendererComponent(
        tree, value, sel,
        expanded, leaf, row,
        hasFocus);
 
 
if (isFish(value)) {
 
setIcon(animalIcon);
setToolTipText("This is in the Fish series.");
} else if(isDog(value)){
 
setIcon(animalIcon);
 
}else if(isHuman(value)){
 
setIcon(animalIcon);

} else 
// Now the default icon
if(leaf){
setIcon(defaulLeaftIcon);
}
else if(expanded){
setIcon(defaultOpenedIcon);
}
else{
setIcon(defaultClosedIcon);
}
}
 
return this;
}
 
}
 
protected boolean isFish(Object value) {
    DefaultMutableTreeNode node =
            (DefaultMutableTreeNode)value;
    Animal nodeInfo =
            (Animal)(node.getUserObject());
    String title = nodeInfo.animalName;
    if (title.indexOf("Fish") >= 0) {
        return true;
    }
protected boolean isDog(Object value) {
    DefaultMutableTreeNode node =(DefaultMutableTreeNode)value;
    Animal nodeInfo = (Animal)(node.getUserObject());
    String title = nodeInfo.dogName;
    if (title.indexOf("Dog") >= 0) {
         return true;
 }
protected boolean isHuman(Object value) {
    DefaultMutableTreeNode node =(DefaultMutableTreeNode)value;
    Animal nodeInfo =(Animal)(node.getUserObject());
    String title = nodeInfo.peopleName;
    if (title.indexOf("People") >= 0) {
        return true;
 }       
    return false;
}
}
 

I tried this code but found out that the icon of new coming node replaces all other icons of original nodes, I tried to use Craig's code to update code to meet this scenario, but not success.

When I tried this program:
[1]. I Drag and drop whiteman node into this JTree, then coming node and all leaves' icon become whitemanIcon;
[2]. then I Drag and drop swordFish node into this JTree, then coming node and all leaves' icon become swordFishIcon;
[3]. at last I Drag and drop EuroDog node into this JTree, then coming node and all leaves' icon become EuroDogIcon;

because I only drag and drop whiteman,swordFish, EuroDogIcon three nodes, I need these three whitemanIcon,swordFishIcon, EuroDogIcon to attach three diifferent nodes, not one for all, but they all gone, only remain last EuroDogIcon, what is wrong here??

Thanks a lot!!

Michelle

[Michelle Wang]

Hi, Craig:
How to update your code meet my above-mentioned example or update my example to meet your excellent sample?? I tried this weekends, but still get all icons replaced by last DnD node's Icon.
Thanks
Michelle

[Craig Wood]

Some general comments.
The cell renderer for a tree is used to render every node in the tree. So it must have
access to any_icon/all_icons needed for the nodes in the tree. There are two ways you
could approach this. Keep a complete set of icons in the cell renderer and give it the
information it needs to be able to recognize which icon is to be set for each leaf node.
Calling super will take care of setting the default icons for the the tree so you won't
have to be concerned with the rest of the (non-leaf) nodes.
The other way you could do this is to create a UserObject that will include icons so the
cell renderer can get the icon that way, ie, directly from the user object. This would
enable you to drag new/unknown icons into the tree and have the renderer set them. You
would make a new user object class instance for each new/dropped node (which we do anyway)
and include the incoming icon in it.
Since you have a detailed list of possible icons and since the first option is much easier
to set up let's consider it.

class PseudoRenderer extends DefaultTreeCellRenderer {
    Map<String,ImageIcon> iconMap;
 
    public PeeudoRenderer(Map<String,ImageIcon> iconMap) {
        this.iconMap = iconMap;
    }
 
    public Component getTreeCellRendererComponent(JTree tree, etc,
                                                  boolean hasFocus) {
        // Set everything as for a normal tree.
        super.getTreeCellRendererComponent(tree, value, sel, expanded,
                                           leaf, row, hasFocus);
        // Customize some leaf nodes.
        // The default icons have already been set for the leaf nodes.
        // We examine each node value to see if we recongnize it and
        // then assign the proper icon. To do this you could use an
        // array index as shown in CustonNodesRenderer. Or you could
        // use a Map in which you fetch the icon with a string/name key.
        if(leaf) {
            DefaultMutableTreeNode node = (DefaultMutableTreeNode)value;
            String key = node.getUserObject().toString();
            if(iconMap.contains(key)) {
                // Cast not required for generics.
                ImageIcon icon = (ImageIcon)iconMap.get(key);
                setIcon(icon);
            }
        }
        return this;
    }
}

With this idea the user object can be a simple string. And the drag_n_drop code can remain
the same.

// Building the tree.
DefaultMutableTreeNode node = new DefaultMutableTreeNode("cat fish");
root.insert(node, j);
...
// Initialize map with all strings and corresponding icons.
Map<String,ImageIcon> map = new HashMap<String,ImageIcon>();
map.put("sword fish", swordFishIcon);
map.put("cat fish", catFishIcon);
...

If you still need a more complicated user object then we should consider its design and
use.

[Michelle Wang]

Thanks a lot, Craig, really good,
When I input the code, it prpmpts
"Syntax Error,parameterized Types Only available if source level is 5.0";

Map <String, ImageIcon> iconMap

;

My JDK is 6, can we have some way other than JDK5.0??
Thanks

[Craig Wood]

some way other than JDK5.0
Here is the non–generic code to use.

Map map = new HashMap();
map.put("sword fish", swordFishIcon);
...
 
class PseudoRenderer extends DefaultTreeCellRenderer {
    Map iconMap;
 
    public PseudoRenderer(Map iconMap) {
        ...
                // Cast is required for non-generics.
                ImageIcon icon = (ImageIcon)iconMap.get(key);

[Michelle Wang]

Great, Thanks so much, success!!

[Michelle Wang]

Another problem occurs, for any coming DnD CustomImageIcon, generally I use CustomImageIcon.getToolTipText() to get default/original name so it totally separate CustomImageIcon.getname() and customImageIcon.getToolTipText(), so when I change the name of coming DnD CustomImageIcon, I only change the value of CustomImageIcon's TextString, not CustomImageIcon.getToolTipText()'s value, but here If I change the values of each coming DnD's nodes' names, then their Icons all become default icons, because node.getUserObject().toString() change the value and did not contain the identical string needed to make decision.
Any idea to solve this issue??



For example, here:

protected boolean isFish(Object value) {    
DefaultMutableTreeNode node =  (DefaultMutableTreeNode)value;  
  Animal nodeInfo = (Animal)(node.getUserObject());   
 String title = nodeInfo.animalName;  
  if (title.indexOf("Fish") >= 0) {     
  return true;    }

when I change he JTree node which contains string "Fish" to "Myfavorite",
then Icon Fish is gone and replaced by deafult icon.

Actually, I use tree.setEditable(true) to change value of each node with success, but I hope to keep the icon unchanged, only change name. how to do it?
Thanks.
[ May 16, 2007: Message edited by: girl sunny ]

[Craig Wood]

Each time the tree or a leaf is rendered your renderer must be able to find out which icon
(or no icon) goes for each leaf node. Even if you drop in other strings that have no
corresponding icon. So you'll have to build a data structure that can remember the icons for
each leaf node. You can get and set the values (icons) for any given leaf node. You could
consider a Map or making up a small class for this. One question to consider is: "can you
have multiple string "keys" in your leaf nodes?" This would make the Map option difficult.
You could use something else besides the string/userObject as a key, perhaps a TreePath or
index into the TreeModel. This structure will have to be available to your renderer, ie, the
renderer would need a reference to the class so it could access the data, and maybe to your
TransferHandler.

[Michelle Wang]

I spent sveral days to digest and study your excellent comments and tried several ways to implement it.
I tried to use several maps to catch all useful information and tried to setup some links between these maps, but link lost when I change the name of the node so icons also lost.
I also tried to use TreePath and index (it looks like a coordinates of X axis and Y axis to locate a node)to identify the node but it must be put into TreeSelectionListener, so when node's name is changed, It cannot be caught immediately so that cannot be associated with proper icon.

Actually that will be great if there is one more column for Hashmap other than <key, value> pair, i mean if we can construct a set pair such as
<key1,key2, value> then we can put all original node name into key2 and to use key2 to search and key1 to display.

I tried HashTable, but only <key, value> pair;

In Java, is there any class which has more than two elements pair such as <key1,key2, value> or <key1,value1, value2> etc??

good weekends.

Thanks


But

[Craig Wood]

Collections like HashMap and HashTable are the data structures that the designers made for
us to use. If they don't suit your needs you can make up something that does. Anything that
does what you want and is easy to read is generally okay.

class DataStore {
    String searchKey;
    String displayKey;
    ImageIcon icon;
    // other member variables as desired
 
    // Constructor(s) for instantiation.
 
    // Methods as desired.
}

Edit:
You can make one of these for each leaf node.
[ May 19, 2007: Message edited by: Craig Wood ]

[Michelle Wang]

Thanks so much, I creat this DataStore class, and I can refer to the instance of this class in following:

class PseudoRenderer extends DefaultTreeCellRenderer {
Map iconMap;
DataStore datastore;
public PseudoRenderer(Map iconMap) {
...
// Cast is required for non-generics.
// ImageIcon icon = (ImageIcon)iconMap.get(key);

and replace map by DataStore datastore;
and I use a vector v to store all datastore object that was initialized with each coming DnD component,

when I change the name one node such as "GoldFish", fish icon still gone;
because I am not sure how to associate value with this datastore in following code

protected boolean isFish(Object value) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode)value;
Animal nodeInfo = (Animal)(node.getUserObject());
String title = nodeInfo.animalName;
if (title.indexOf("Fish") >= 0) {
return true;
}

here value is a node object, it has one method value.toString(); so once its text string changed, cannot track its original names.
Because I need to change both leaf name and node's name, that is why I use value, not leaf parameter.

Any further idea will welcome
Thanks so much

Michelle
[ May 20, 2007: Message edited by: Michelle Wang ]

[Craig Wood]

Make your generics/non–generics selection here:

class LeafStores {
    List<LeafNodeStore> list = new ArrayList<LeafNodeStore>();
//    List list = new ArrayList();   // non-generic statement
 

You can get any LeafNodeStore from LeafStores by its TreePath in the tree and change the
icon as you like.

import java.awt.*;
import java.awt.datatransfer.*;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import javax.imageio.ImageIO;
import javax.swing.*;
import javax.swing.tree.*;
 
public class CustomNodesTest2 {
    private JScrollPane getContent(BufferedImage[] images, String[] ids) {
        ImageIcon[] icons = getScaledIcons(images, ids);
        LeafStores stores = new LeafStores(ids, icons);
        JTree tree = new JTree(getData(stores, ids, icons));
        tree.setCellRenderer(new CustomNodesRenderer2(stores));
        tree.getSelectionModel().setSelectionMode(
                TreeSelectionModel.SINGLE_TREE_SELECTION);
        tree.setDragEnabled(true);
        tree.setTransferHandler(new CustomNodesHandler2(stores));
        return new JScrollPane(tree);
    }
 
    private DefaultMutableTreeNode getData(LeafStores stores,
                                           String[] ids, ImageIcon[] icons) {
        int[][] keys = { { 2 }, { 1, 0 }, { 0 }};
        DefaultMutableTreeNode root = new DefaultMutableTreeNode("root");
        for(int j = 0; j < keys.length; j++) {
            DefaultMutableTreeNode node =
                new DefaultMutableTreeNode("Node " + (j+1));
            root.insert(node, j);
            for(int k = 0; k < keys[j].length; k++) {
                String s = ids[keys[j][k]];
                DefaultMutableTreeNode child = new DefaultMutableTreeNode(s);
                node.insert(child, k);
                TreePath path = new TreePath(child.getPath());
                stores.add(new LeafNodeStore(path, s, icons[keys[j][k]]));
            }
        }
        return root;
    }
 
    private ImageIcon[] getScaledIcons(BufferedImage[] images, String[] ids) {
        ImageIcon[] icons = new ImageIcon[images.length];
        Icon icon = UIManager.getIcon("Tree.leafIcon");
        int w = icon.getIconWidth();
        int h = icon.getIconHeight();
        int type = //BufferedImage.TYPE_INT_RGB;
                   BufferedImage.TYPE_BYTE_INDEXED;
        for(int j = 0; j < images.length; j++) {
            BufferedImage image = new BufferedImage(w, h, type);
            Graphics2D g2 = image.createGraphics();
            g2.setRenderingHint(RenderingHints.KEY_INTERPOLATION,
                                RenderingHints.VALUE_INTERPOLATION_BICUBIC);
            double xScale = (double)w/images[j].getWidth();
            double yScale = (double)h/images[j].getHeight();
            double scale = Math.max(xScale, yScale);  // fill
            AffineTransform at = AffineTransform.getScaleInstance(scale, scale);
            g2.drawRenderedImage(images[j], at);
            g2.dispose();
            icons[j] = new ImageIcon(image, ids[j]);
        }
        return icons;
    }
 
    public static void main(String[] args) throws IOException {
        String[] ids = { "Dog", "Cat", "Bird" };
        BufferedImage[] images = new BufferedImage[ids.length];
        for(int j = 0; j < images.length; j++) {
            File file = new File("images/" + ids[j] + ".gif");
            images[j] = ImageIO.read(file);
        }
        CustomNodesTest2 test = new CustomNodesTest2();
        JFrame f = new JFrame();
        f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        f.getContentPane().add(test.getContent(images, ids));
        f.setSize(400,400);
        f.setLocation(200,200);
        f.setVisible(true);
    }
}
 
class LeafStores {
    List<LeafNodeStore> list = new ArrayList<LeafNodeStore>();
//    List list = new ArrayList();   // non-generic statement
    String[] ids;
    ImageIcon[] icons;
 
    public LeafStores(String[] ids, ImageIcon[] icons) {
        this.ids = ids;
        this.icons = icons;
    }
 
    public void add(LeafNodeStore store) {
        list.add(store);
    }
 
    public void remove(TreePath path) {
        if(path != null) {
            for(int j = 0; j < list.size(); j++) {
                LeafNodeStore store = (LeafNodeStore)list.get(j);
                if(path.equals(store.path))
                    list.remove(store);
            }
        }
    }
 
    public LeafNodeStore getStore(TreePath path) {
        if(path != null) {
            for(int j = 0; j < list.size(); j++) {
                LeafNodeStore store = (LeafNodeStore)list.get(j);
                if(path.equals(store.path))
                    return store;
            }
        }
        return null;
    }
 
    public ImageIcon getIcon(String text) {
        for(int j = 0; j < ids.length; j++) {
            if(text.equals(ids[j]))
                return icons[j];
        }
        return null;
    }
 
    public String toString() {
        StringBuilder sb = new StringBuilder();
        for(int j = 0; j < list.size(); j++) {
            LeafNodeStore store = (LeafNodeStore)list.get(j);
            sb.append(store.toString());
            if(j < list.size()-1)
                 sb.append("\n");
        }
        return sb.toString();
    }
}
 
class LeafNodeStore {
    TreePath path;
    String text;
    ImageIcon icon;
 
    public LeafNodeStore(TreePath path, String text, ImageIcon icon) {
        this.path = path;
        this.text = text;
        this.icon = icon;
    }
 
    public String toString() {
        String desc = (icon != null) ? icon.getDescription() : "null";
        return "LeafNodeStore[path:" + path +
                           ", text:" + text +
                           ", icon:" + desc + "]";
    }
}
 
class CustomNodesRenderer2 extends DefaultTreeCellRenderer {
    LeafStores stores;
 
    public CustomNodesRenderer2(LeafStores stores) {
        this.stores = stores;
    }
 
    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(leaf) {
            DefaultMutableTreeNode node = (DefaultMutableTreeNode)value;
            TreePath path = new TreePath(node.getPath());
            LeafNodeStore store = stores.getStore(path);
            if(store != null) {
                // If the store icon is null you can set a null icon
                // or you can leave the default leaf icon which was
                // set above in the call to super. For the second
                // option you could guard the setIcon statement with:
                // if(store.icon != null)
                setIcon(store.icon);
            }
        }
        return this;
    }
}
 
class CustomNodesHandler2 extends TransferHandler {
    DataFlavor flavor = DataFlavor.stringFlavor;
    DefaultMutableTreeNode lastSelection;
    LeafStores stores;
 
    public CustomNodesHandler2(LeafStores stores) {
        this.stores = stores;
    }
 
    public boolean importData(JComponent comp, Transferable t) {
        if(canImport(comp, t.getTransferDataFlavors())) {
            JTree tree = (JTree)comp;
            DefaultMutableTreeNode parent =
                (DefaultMutableTreeNode)tree.getLastSelectedPathComponent();
            try {
                String data = (String)t.getTransferData(flavor);
                DefaultMutableTreeNode child = new DefaultMutableTreeNode(data);
                DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
                model.insertNodeInto(child, parent, parent.getChildCount());
                tree.expandPath(new TreePath(parent.getPath()));
                TreePath path = new TreePath(child.getPath());
                stores.add(new LeafNodeStore(path, data, stores.getIcon(data)));
                return true;
            } catch(UnsupportedFlavorException ufe) {
                System.out.println("UnsupportedFlavor: " + ufe.getMessage());
            } catch(IOException ioe) {
                System.out.println("IO Error: " + ioe.getMessage());
            }
        }
        return false;
    }
 
    public boolean canImport(JComponent comp, DataFlavor[] transferFlavors) {
        for(int j = 0; j < transferFlavors.length; j++) {
            if(transferFlavors[j].equals(flavor))
                return true;
        }
        return false;
    }
 
    public int getSourceActions(JComponent c) {
        return COPY_OR_MOVE;
    }
 
    protected void exportDone(JComponent source, Transferable data, int action) {
        if(action == MOVE && lastSelection != null) {
            JTree tree = (JTree)source;
            DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
            if(lastSelection.isLeaf()) {
                TreePath path = new TreePath(lastSelection.getPath());
                stores.remove(path);
            }
            model.removeNodeFromParent(lastSelection);
        }
    }
 
    protected Transferable createTransferable(JComponent c) {
        JTree tree = (JTree)c;
        DefaultMutableTreeNode node =
            (DefaultMutableTreeNode)tree.getLastSelectedPathComponent();
        lastSelection = node;
        String text = node.getUserObject().toString();
        return new StringSelection(text);
    }
}

[Michelle Wang]

Excellent code, thanks so much, I tried to edit and change the nodes' names and leaf's name, but cannot, I add tree.editable(true) etc that code works on my part, but here still not work, can we do that??
great thanks.

[Craig Wood]

Try using a triple–click on the node text to begin editing of the node.
End editing by pressing the enter key.

[Michelle Wang]

Thanks a lot, small question, If I hope to Drag nad Drop a subtree AA (AA has children of subtree CC, leaves DD1, DD2, DD3 etc) under the main tree into another subtree BB(has children of EE1, EE2 etc), I found the all leaves or children and their icons under this AA(such as children of subtree CC, leaves DD1, DD2, DD3 etc) are lost when AA becomes BB's Child,

and also, when node's name changed from CC to Tiger, Tiger icon is same as CC, if move Tiger leaf from subtree AA to BB, then Tiger's icon gone, anyway can remedy??

I tried to update your code, not success.
Thanks so much!!
[ May 22, 2007: Message edited by: Michelle Wang ]

[Craig Wood]

I found the all leaves or children and their icons...are lost
The TransferHandler uses a StringTransferable to transfer a string for a single node. So
this transferHandler would probably transfer only the userObject string of the subtree root
node into the tree. To transfer a subtree you will have to make changes to/re–design the
transferHandler.

[Michelle Wang]

Really thants a lot, let me try again.