Java - Arboles en java

Les agradecería si me pueden decir como hacer que un árbol echo con java tenga la imagen de las hojas diferentes según algún atributo de esta.
Intente usar la clase DefaultTreeCellRender pero me cambia la imagen de todos los nodos hojas.
Yo necesitaría que según la rama de donde venga tenga una imagen distinta.
Gracias.

Hola:

Hereda de DefaultTreeCellRender y redefine el método getTreeCellRendererComponent().

A este método le pasan absolutamente toda la información del nodo (valor ,si es hoja o no, si está seleccionado, etc, etc). Construyete ahí un JLabel con icono y texto y lo devuelves. Este JLabel es el que se verá en el árbol.

DefaultTreeCellRenderer hereda de JLabel, así que puedes simplemente usar métodos this.setIcon() y this.setText() y devolver luego return this.

Se bueno.