Mudando Icone de Jtree com TreeCellRenderer

package jsprint.gui.componentes.renderer;

import java.awt.Color;
import java.awt.Component;

import javax.swing.Icon;
import javax.swing.JLabel;
import javax.swing.JTree;
import javax.swing.tree.TreeCellRenderer;

/**
 * Classe responsável por estilizar uma Jtree para sprint.
 * 
 * @author estevao.costa
 *
 */
public class JTreeSprintRenderer extends JLabel implements TreeCellRenderer {

	/**
	 * Seriakl Id.
	 */
	private static final long serialVersionUID = 2421948525968191554L;

	/**
	 * Icone de fechado.
	 */
	private Icon closedIcon;

	/**
	 * Icone da raiz.
	 */
	private Icon rootIcon;
	
	/**
	 * Icone da folha.
	 */
	private Icon leafIcon;

	/**
	 * Icone de aberto.
	 */
	private Icon openIcon;

	/**
	 * Cor do texto selecionado.
	 */
	private Color corTextoSelecionado;

	/**
	 * Cor dos nós.
	 */
	private Color corTextoNaoSelecionado;

	/**
	 * Cor de fundo.
	 */
	private Color corFundoSelecionado;

	/**
	 * Cor do fundo dos nos selecionados.
	 */
	private Color corFundoNaoSelecionado;

	
	/**
	 * Construtor.
	 */
	public JTreeSprintRenderer() {
	}
	
	/**
	 * Método responsável pela renderização do componete.
	 */
	public Component getTreeCellRendererComponent(JTree tree, Object value,
			boolean selected, boolean expanded, boolean leaf, int row,
			boolean hasFocus) {
		
		String stringValue = tree.convertValueToText(value, selected, expanded, leaf, row, hasFocus);
		
		if (selected) {
			this.setForeground(this.corTextoSelecionado);
			this.setBackground(this.corFundoSelecionado);
		} else {
			this.setForeground(this.corTextoNaoSelecionado);
			this.setBackground(this.corFundoNaoSelecionado);
		}
		
		if (!tree.isEnabled()) {
			this.setEnabled(false);
		} else {
			this.setEnabled(true);
		}
		
        Icon icon = null;
        
        if (isRoot(row)) {
        	icon = getRootIcon();
        } else if (leaf) {
            icon = getLeafIcon();
        } else if (expanded) {
            icon = getOpenIcon();
        } else {
            icon = getClosedIcon();
        }
        
		this.setIcon(icon);
		this.setText(stringValue);
		this.repaint();
		return this;
	}

	/**
	 * Método responsável por verificar se a linha corrente é raiz ou não.
	 * @param row
	 * @return
	 */
	private boolean isRoot(final Integer row){
		return row == 0;
	}

	public Icon getRootIcon() {
		return rootIcon;
	}

	public void setRootIcon(Icon rootIcon) {
		this.rootIcon = rootIcon;
	}

	/**
	 * Sets the icon used to represent non-leaf nodes that are expanded.
	 */
	public void setOpenIcon(Icon newIcon) {
		openIcon = newIcon;
	}

	/**
	 * Returns the icon used to represent non-leaf nodes that are expanded.
	 */
	public Icon getOpenIcon() {
		return openIcon;
	}

	/**
	 * Sets the icon used to represent non-leaf nodes that are not expanded.
	 */
	public void setClosedIcon(Icon newIcon) {
		closedIcon = newIcon;
	}

	/**
	 * Returns the icon used to represent non-leaf nodes that are not expanded.
	 */
	public Icon getClosedIcon() {
		return closedIcon;
	}

	/**
	 * Sets the icon used to represent leaf nodes.
	 */
	public void setLeafIcon(Icon newIcon) {
		leafIcon = newIcon;
	}

	/**
	 * Returns the icon used to represent leaf nodes.
	 */
	public Icon getLeafIcon() {
		return leafIcon;
	}

	public Color getCorTextoSelecionado() {
		return corTextoSelecionado;
	}

	public void setCorTextoSelecionado(Color corTextoSelecionado) {
		this.corTextoSelecionado = corTextoSelecionado;
	}

	public Color getCorTextoNaoSelecionado() {
		return corTextoNaoSelecionado;
	}

	public void setCorTextoNaoSelecionado(Color corTextoNaoSelecionado) {
		this.corTextoNaoSelecionado = corTextoNaoSelecionado;
	}

	public Color getCorFundoSelecionado() {
		return corFundoSelecionado;
	}

	public void setCorFundoSelecionado(Color corFundoSelecionado) {
		this.corFundoSelecionado = corFundoSelecionado;
	}

	public Color getCorFundoNaoSelecionado() {
		return corFundoNaoSelecionado;
	}

	public void setCorFundoNaoSelecionado(Color corFundoNaoSelecionado) {
		this.corFundoNaoSelecionado = corFundoNaoSelecionado;
	}
}