Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.searchpanel;

import java.awt.Component;

import javax.swing.ImageIcon;

import javax.swing.JLabel;

import javax.swing.JTree;

import javax.swing.tree.DefaultTreeCellRenderer;

import org.gvsig.fmap.dal.swing.impl.searchpanel.AdvancedAttributeSelectionTreeModel.Node;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.icontheme.IconTheme;

/**

 *

 * @author jjdelcerro

 */

public class AdvancedAttributeSelectionTreeCellRenderer extends DefaultTreeCellRenderer {

    private final ImageIcon iconTable;

    private final ImageIcon iconColumn;

    private final ImageIcon iconForeingKey;

    public AdvancedAttributeSelectionTreeCellRenderer() {

        IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        this.iconTable = theme.get("featurestore-table");

        this.iconColumn = theme.get("featurestore-column");

        this.iconForeingKey = theme.get("featurestore-foreing-key");

    }

    @Override

    public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {

        JLabel label = (JLabel) super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus); //To change body of generated methods, choose Tools | Templates.

        if( !(value instanceof Node) ) {

            return label;

        }

        Node node = (Node) value;

        if( node.isRoot() ) {

            label.setIcon(this.iconTable);

            label.setText(node.getLabel());

            return label;

        }

        if( leaf ) {

            label.setIcon(this.iconColumn);

            label.setText(node.getLabel());

            return label;

        }

        label.setIcon(this.iconForeingKey);

        label.setText(node.getLabel());

        return label;

    }

}