Application: gvSIG desktop » gvSIG legends: base legends

package org.gvsig.legend.heatmap.swing.impl;

import java.awt.Color;

import javax.swing.ImageIcon;

import javax.swing.JFormattedTextField;

import javax.swing.JPanel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.app.gui.styling.ColorSchemeSelector;

import org.gvsig.app.project.documents.view.legend.gui.Features;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.rendering.legend.ILegend;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegend;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegendLocator;

import org.gvsig.legend.heatmap.swing.api.HeatmapLegendEditor;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

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

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

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

public class DefaultHeatmapLegendEditor extends DefaultHeatmapLegendEditorView implements HeatmapLegendEditor {

    private static final long serialVersionUID = 4307714097793669675L;

    private HeatmapLegend legend;

    private FLayer layer;

    private final Class<? extends HeatmapLegend> legendClass;

    private ColorChooserController hotColorChooser;

    private ColorChooserController coldColorChooser;

    private ColorSchemeSelector colorSchemaSelector;

    public DefaultHeatmapLegendEditor() {

        this.legendClass = HeatmapLegendLocator.getHeatmapLegendManager().getLegendClass();

    }

    @Override

    public void setData(FLayer layer, ILegend legend) {

        if( layer == this.layer && legend == this.legend ) {

            return;

        }

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        this.layer = layer;

        if( this.getLegendClass().isInstance(legend) ) {

            this.legend = (HeatmapLegend) legend;

        } else {

            this.legend = HeatmapLegendLocator.getHeatmapLegendManager().create();

        }

        this.hotColorChooser = toolsSwingManager.createColorChooserController(txtHotColorLabel, btnHotColor, sldHotColorAlpha);

        this.coldColorChooser = toolsSwingManager.createColorChooserController(txtColdColorLabel, btnColdColor, sldColdColorAlpha);

        this.colorSchemaSelector = new ColorSchemeSelector(cboColorTable, false);

        this.rdbUseColorRamp.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                useColorModeChanged();

            }

        });

        this.chkUseField.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                useFieldChanged();

            }

        });

        this.chkColorTableUseAlpha.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                useAlphaInColorTableChanged();

            }

        });

        this.spnDistance.setValue(this.legend.getDistance());

        if( this.legend.useRamp() ) {

            this.rdbUseColorRamp.setSelected(true);

        } else {

            this.rdbUseColorTable.setSelected(true);

        }

        if( this.legend.useAlphaInColorTable() ) {

            this.chkColorTableUseAlpha.setSelected(true);

        } else {

            this.chkColorTableUseAlpha.setSelected(false);

        }

        this.sldColorTableColdAlpha.setValue(this.legend.getColorTableColdColorAlpha());

        this.sldColorTableHotAlpha.setValue(this.legend.getColorTableHotColorAlpha());

        this.chkUseField.setEnabled(false);

        this.chkUseField.setSelected(false);

        this.cboFields.setEnabled(false);

        this.cboFields.removeAllItems();

        if( layer instanceof FLyrVect ) {

            FeatureStore store = ((FLyrVect) layer).getFeatureStore();

            if( store != null ) {

                try {

                    FeatureType ftype = store.getDefaultFeatureType();

                    for( FeatureAttributeDescriptor attr : ftype ) {

                        this.cboFields.addItem(attr.getName());

                    }

                    if( this.cboFields.getItemCount()>0 ) {

                        this.chkUseField.setEnabled(true);

                        if( !StringUtils.isEmpty(this.legend.getFieldName()) ) {

                            this.chkUseField.setSelected(true);

                            this.cboFields.setSelectedItem(this.legend.getFieldName());

                            this.cboFields.setEnabled(true);

                        }

                    }

                } catch (Exception ex) {

                }

            }

        }

    }

    private void useColorModeChanged() {

        if( this.rdbUseColorRamp.isSelected() ) {

            this.hotColorChooser.setEnabled(true);

            this.coldColorChooser.setEnabled(true);

            this.spnNumberOfColors.setEnabled(true);

            this.cboColorTable.setEnabled(false);

            this.chkColorTableUseAlpha.setEnabled(false);

            this.sldColorTableColdAlpha.setEnabled(false);

            this.sldColorTableHotAlpha.setEnabled(false);

            Color[] table = this.legend.getSourceColorTable();

            this.spnNumberOfColors.setValue(table.length);

            this.coldColorChooser.set(table[0]);

            this.hotColorChooser.set(table[table.length-1]);

        } else {

            this.hotColorChooser.setEnabled(false);

            this.coldColorChooser.setEnabled(false);

            this.spnNumberOfColors.setEnabled(false);

            this.cboColorTable.setEnabled(true);

            this.chkColorTableUseAlpha.setEnabled(true);

            this.sldColorTableColdAlpha.setEnabled(true);

            this.sldColorTableHotAlpha.setEnabled(true);

            this.colorSchemaSelector.setSelectedColors(this.legend.getSourceColorTable());

        }

    }

    private void useFieldChanged() {

        this.cboFields.setEnabled(this.chkUseField.isSelected());

    }

    private void useAlphaInColorTableChanged() {

        boolean x = this.chkColorTableUseAlpha.isSelected();

        this.sldColorTableColdAlpha.setEnabled(x);

        this.sldColorTableHotAlpha.setEnabled(x);

    }

    @Override

    public ILegend getLegend() {

        legend.setDistance((int) spnDistance.getValue());

        if( this.rdbUseColorRamp.isSelected() ) {

            legend.setColorTable(

                (int) spnNumberOfColors.getValue(),

                coldColorChooser.get(),

                hotColorChooser.get()

            );

        } else {

            legend.setColorTable( this.colorSchemaSelector.getSelectedColors());

        }

        if( this.chkUseField.isSelected() ) {

            String fieldName = (String) this.cboFields.getSelectedItem();

            if( fieldName != null ) {

                this.legend.setFieldName(fieldName);

            }

        }

        this.legend.setColorTableColdColorAlpha(this.sldColorTableColdAlpha.getValue());

        this.legend.setColorTableHotColorAlpha(this.sldColorTableHotAlpha.getValue());

        this.legend.setUseAlphaInColorTable(this.chkColorTableUseAlpha.isSelected());

        return legend;

    }

    @Override

    public String getDescription() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        return i18n.getTranslation( "_Heatmap_legend_description");

    }

    @Override

    public ImageIcon getIcon() {

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

        return theme.get("legend-overview-vectorial-heatmap");

    }

    @Override

        public Class getParentClass() {

                return Features.class;

        }

    @Override

    public String getTitle() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        return i18n.getTranslation( "_Heatmap_legend");

    }

    @Override

    public JPanel getPanel() {

        return this;

    }

    @Override

    public Class getLegendClass() {

        return this.legendClass;

    }

    @Override

    public boolean isSuitableFor(FLayer layer) {

        return true;

    }

    private int getIntValue(JFormattedTextField text, int defaultValue) {

        String s = text.getText();

        if( StringUtils.isEmpty(s) ) {

            return defaultValue;

        }

        int value = defaultValue;

        try {

            value = Integer.parseInt(s);

        } catch(Throwable th) {

        }

        return value;

    }

}