Application: gvSIG desktop

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

import java.awt.Component;

import java.awt.FlowLayout;

import javax.swing.Icon;

import javax.swing.ImageIcon;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JTree;

import javax.swing.border.EmptyBorder;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.TreeCellRenderer;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.swing.impl.featuretype.FeatureAttributeTreeModel.Node;

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

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

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

/**

 *

 * @author jjdelcerro

 */

public class FeatureAttributeTreeCellRenderer 

        extends DefaultTreeCellRenderer 

        implements TreeCellRenderer

    {

    private final Icon iconTable;

    private final Icon iconColumn;

    private final Icon iconForeingKey;

//    private final JPanel panel;

//    private final JLabel icon;

//    private final DefaultTreeCellRenderer label;

    private final IconTheme iconTheme;

    private final ImageIcon iconColumnQueryCalculated;

    private final ImageIcon iconColumnTableCalculated;

    public FeatureAttributeTreeCellRenderer() {

        this.iconTheme = ToolsSwingLocator.getIconThemeManager().getCurrent();

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

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

        this.iconColumnTableCalculated = iconTheme.get("featurestore-calculated-column");

        this.iconColumnQueryCalculated = iconTheme.get("featurestore-query-calculated-column");

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

//        this.panel = new JPanel();

//        this.panel.setLayout(new FlowLayout(FlowLayout.LEADING, 3, 2));

//        this.icon = new JLabel();

//        this.label = new DefaultTreeCellRenderer();

//        this.panel.add(this.icon);

//        this.panel.add(this.label);

//        this.panel.setBorder(new EmptyBorder(0, 0, 0, 0));

//        this.panel.setOpaque(false);

    }

    @Override

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

        JLabel theLabel = (JLabel) super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus); 

        if( !(value instanceof Node) ) {

            return theLabel;

        }

        Node node = (Node) value;

        FeatureAttributeDescriptor attrdesc = node.getValue();

        Icon iconElement;

        if( node.isRoot() ) {

           iconElement = this.iconTable;

        } else if( attrdesc == null ) {

           iconElement = leaf? this.iconColumn: this.iconForeingKey;

        } else if( attrdesc.isForeingKey() ) {

           iconElement = this.iconForeingKey;            

        } else if( attrdesc.isComputed() ) {

            iconElement = (attrdesc.getStore()==null)? this.iconColumnQueryCalculated: this.iconColumnTableCalculated;

        } else {

           iconElement = iconColumn;

        }

//        Icon iconElement = node.isRoot()? this.iconTable: leaf? this.iconColumn: this.iconForeingKey;

        Icon iconDataType = null;

        try {

            if( !node.isRoot() ) {

                String iconName = node.getValue().getDataType().getIconName();

                iconDataType = this.iconTheme.get(iconName);

            }

        } catch(Exception ex) {

        }

        String s = node.getLabel();

        if( StringUtils.isBlank(s) ) {

          s = "???";

        }

        theLabel.setText(s);

        if (iconDataType != null) {

            CompoundIcon compoundIcon = ToolsSwingLocator.getToolsSwingManager().createCompoundIcon(iconElement, iconDataType);

            theLabel.setIcon(compoundIcon);

        } else {

            theLabel.setIcon(iconElement);

        }

//        

//        this.label.setText(s);

//        try {

//          this.label.setIcon(iconDataType);

//          this.icon.setIcon(iconElement);

//          this.label.invalidate();

//        } catch(Throwable th) {

//          th.getMessage(); // To allow set a break point

//        }

        return theLabel; //return this.label

    }

}