Application: gvSIG desktop » gvSIG raster

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.mapcontext.raster.swing.impl.transparency;

import java.awt.Color;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.List;

import java.util.Locale;

import javax.swing.DefaultListModel;

import javax.swing.JComponent;

import javax.swing.JPanel;

import javax.swing.colorchooser.AbstractColorChooserPanel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.mapcontext.exceptions.LegendLayerException;

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

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.fmap.mapcontext.raster.swing.transparency.TransparencyPanel;

import org.gvsig.raster.lib.legend.api.RasterLegend;

import org.gvsig.raster.lib.legend.api.RasterLegendLocator;

import org.gvsig.raster.lib.legend.api.RasterLegendManager;

import org.gvsig.raster.lib.legend.api.Transparency;

import org.gvsig.raster.lib.legend.api.TransparencyRange;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class TransparencyPanelController extends TransparencyPanelView implements TransparencyPanel, ActionListener, ChangeListener, ListSelectionListener{

    /**

     *

     */

    private static final long serialVersionUID = -7509955533999332846L;

    private static final Logger LOG = LoggerFactory.getLogger(TransparencyPanelController.class);

    /**

     * @param layer

     */

    public TransparencyPanelController() {

        colorChooser.setPreviewPanel(new JPanel());

        initializeComponents();

        translate();

        this.invalidate();

    }

    /**

     *

     */

    @SuppressWarnings("unchecked")

    private void initializeComponents() {

        initializeColorChooserPanel();

        chkNoDataTransparency.addActionListener(this);

        chkActivateOpacity.addActionListener(this);

        chkActivatePixelTransparency.addActionListener(this);

        sldOpacity.addChangeListener(this);

        spOpacity.addChangeListener(this);

        btnAdd.addActionListener(this);

        btnRemove.addActionListener(this);

        // Botones para poder a?adirles acciones desde fuera

//        btnExternalTool0.addActionListener(this);

//        btnExternalTool1.addActionListener(this);

//        btnExternalTool2.addActionListener(this);

//        lstTransparencyRanges.setModel(new TransparencyRangesListModel());

        lstTransparencyRanges.getParent().setPreferredSize(new Dimension(210, lstTransparencyRanges.getParent().getHeight()));;

        lstTransparencyRanges.setModel(new DefaultListModel<TransparencyRange>());

        lstTransparencyRanges.addListSelectionListener(this);

    }

    /**

     *

     */

    private void initializeColorChooserPanel() {

        AbstractColorChooserPanel[] defaultPanels =

            javax.swing.colorchooser.ColorChooserComponentFactory.getDefaultChooserPanels();

        AbstractColorChooserPanel[] chooserPanelsCleaned = new AbstractColorChooserPanel[1];

        //FIXME: Pi??n fijo, ponemos el RGB. A ver si esto se puede hacer mejor.

        chooserPanelsCleaned[0] = defaultPanels[3];

        colorChooser.setChooserPanels(chooserPanelsCleaned);

    }

    private void translate() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        chkNoDataTransparency.setText(i18nManager.getTranslation(chkNoDataTransparency.getText()));

        chkNoDataTransparency.setToolTipText(i18nManager.getTranslation(chkNoDataTransparency.getToolTipText()));

//        lblNoDataValue.setText(i18nManager.getTranslation(lblNoDataValue.getText()));

//        lblNoDataValue.setToolTipText(i18nManager.getTranslation(lblNoDataValue.getToolTipText()));

//        txtNoDataValue.setToolTipText(i18nManager.getTranslation(txtNoDataValue.getToolTipText()));

//        btnSaveNoData.setText(i18nManager.getTranslation(btnSaveNoData.getText()));

//        btnSaveNoData.setToolTipText(i18nManager.getTranslation(btnSaveNoData.getToolTipText()));

//        btnRemoveNoData.setText(i18nManager.getTranslation(btnRemoveNoData.getText()));

//        btnRemoveNoData.setToolTipText(i18nManager.getTranslation(btnRemoveNoData.getToolTipText()));

//        btnDefaultNoDataValue.setText(i18nManager.getTranslation(btnDefaultNoDataValue.getText()));

//        btnDefaultNoDataValue.setToolTipText(i18nManager.getTranslation(btnDefaultNoDataValue.getToolTipText()));

//        btnRestoreNoData.setText(i18nManager.getTranslation(btnRestoreNoData.getText()));

//        btnRestoreNoData.setToolTipText(i18nManager.getTranslation(btnRestoreNoData.getToolTipText()));

        lblOpacity.setText(i18nManager.getTranslation(lblOpacity.getText()));

        chkActivateOpacity.setText(i18nManager.getTranslation(chkActivateOpacity.getText()));

        chkActivateOpacity.setToolTipText(i18nManager.getTranslation(chkActivateOpacity.getToolTipText()));

        lblPixelTransparency.setText(i18nManager.getTranslation(lblPixelTransparency.getText()));

        chkActivatePixelTransparency.setText(i18nManager.getTranslation(chkActivatePixelTransparency.getText()));

        chkActivatePixelTransparency.setToolTipText(i18nManager.getTranslation(chkActivatePixelTransparency.getToolTipText()));

        btnAdd.setText(i18nManager.getTranslation(btnAdd.getText()));

        btnAdd.setToolTipText(i18nManager.getTranslation(btnAdd.getToolTipText()));

        btnRemove.setText(i18nManager.getTranslation(btnRemove.getText()));

        btnRemove.setToolTipText(i18nManager.getTranslation(btnRemove.getToolTipText()));

        btnExternalTool0.setText(i18nManager.getTranslation(btnExternalTool0.getText()));

        btnExternalTool0.setToolTipText(i18nManager.getTranslation(btnExternalTool0.getToolTipText()));

        btnExternalTool1.setText(i18nManager.getTranslation(btnExternalTool1.getText()));

        btnExternalTool1.setToolTipText(i18nManager.getTranslation(btnExternalTool1.getToolTipText()));

        btnExternalTool2.setText(i18nManager.getTranslation(btnExternalTool2.getText()));

        btnExternalTool2.setToolTipText(i18nManager.getTranslation(btnExternalTool2.getToolTipText()));

        rbtnAnd.setText(i18nManager.getTranslation(rbtnAnd.getText()));

        rbtnAnd.setToolTipText(i18nManager.getTranslation(rbtnAnd.getToolTipText()));

        rbtnOr.setText(i18nManager.getTranslation(rbtnOr.getText()));

        rbtnOr.setToolTipText(i18nManager.getTranslation(rbtnOr.getToolTipText()));

        chkRed.setText(i18nManager.getTranslation(chkRed.getText()));

        chkRed.setToolTipText(i18nManager.getTranslation(chkRed.getToolTipText()));

        chkGreen.setText(i18nManager.getTranslation(chkGreen.getText()));

        chkGreen.setToolTipText(i18nManager.getTranslation(chkGreen.getToolTipText()));

        chkBlue.setText(i18nManager.getTranslation(chkBlue.getText()));

        chkBlue.setToolTipText(i18nManager.getTranslation(chkBlue.getToolTipText()));

        chkAlpha.setText(i18nManager.getTranslation(chkAlpha.getText()));

        chkAlpha.setToolTipText(i18nManager.getTranslation(chkAlpha.getToolTipText()));

    }

    /**

     * @param locale

    *

    */

    public void setLocate(Locale locale) {

        Locale l = super.getLocale();

        if (!l.equals(locale)) {

            translate();

        }

        super.setLocale(locale);

    }

    /**

     * @param layer

     */

    @SuppressWarnings("unchecked")

    public void set(FLayer layer){

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        RasterLayer rasterLayer = (RasterLayer)layer;

        RasterLegend legend = (RasterLegend) rasterLayer.getLegend();

        //FIXME: ?Debemos crear una leyenda cuando no hemos podido obtenerla de la capa, o debe cascar?

        if(legend == null){

            legend = legendManager.createLegend();

        }

        if(legend.areTransparentNoData()){

            chkNoDataTransparency.doClick();

        }

        int alpha = 255;

        Transparency transparency = legend.getTransparency();

        if (transparency != null) {

            chkActivateOpacity.doClick();

            alpha = transparency.getAlpha();

            if(transparency.getTransparencyRanges()!=null && transparency.getTransparencyRanges().size()>0){

                chkActivatePixelTransparency.doClick();

                for (TransparencyRange range : transparency.getTransparencyRanges()) {

                    ((DefaultListModel<TransparencyRange>) lstTransparencyRanges.getModel()).addElement(range);

                }

            }

        }

        sldOpacity.setValue(alpha);

        spOpacity.setValue(alpha);

    }

    /**

     * @param layer

     */

    public boolean fetch(FLayer layer){

        RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

        RasterLegend legend = ((RasterLayer)layer).getLegend();

        if(legend == null){

            legend = legendManager.createLegend();

        }

        legend.setTransparentNoData(chkNoDataTransparency.isSelected());

        if(chkActivateOpacity.isSelected()){

            Transparency transparency = legend.getTransparency();

            if(transparency==null){

                transparency = legendManager.createTransparency();

            }

            transparency.setValue(((Integer)spOpacity.getValue()));

            legend.setTransparency(transparency);

        }

        if(chkActivatePixelTransparency.isSelected()){

            Transparency transparency = legend.getTransparency();

            if(transparency==null){

                transparency = legendManager.createTransparency();

            }

            transparency.setTransparencyRanges(this.getTransparencyRanges());

            legend.setTransparency(transparency);

        }

        layer.getMapContext().invalidate();

        return true;

    }

    private List<TransparencyRange> getTransparencyRanges() {

        ArrayList<TransparencyRange> result = new ArrayList<TransparencyRange>();

        @SuppressWarnings("unchecked")

        DefaultListModel<TransparencyRange> model = ((DefaultListModel<TransparencyRange>)lstTransparencyRanges.getModel());

        for (int i = 0; i < model.getSize(); i++) {

            result.add(model.get(i));

        }

        return result;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @SuppressWarnings("unchecked")

    @Override

    public void actionPerformed(ActionEvent e) {

        if(e.getSource() == chkActivateOpacity){

            boolean selected = chkActivateOpacity.isSelected();

            sldOpacity.setEnabled(selected);

            spOpacity.setEnabled(selected);

        }

        if(e.getSource() == chkActivatePixelTransparency){

            boolean selected = chkActivatePixelTransparency.isSelected();

            colorChooser.setEnabled(selected);

            btnAdd.setEnabled(selected);

            btnRemove.setEnabled(selected);

            //FIXME: Supongo que habr? que comprobar que tienen una acci?n asignada para poder habilitarlos

//            btnExternalTool0.setEnabled(selected);

//            btnExternalTool1.setEnabled(selected);

//            btnExternalTool2.setEnabled(selected);

            rbtnAnd.setEnabled(selected);

            rbtnOr.setEnabled(selected);

            chkRed.setEnabled(selected);

            chkGreen.setEnabled(selected);

            chkBlue.setEnabled(selected);

            chkAlpha.setEnabled(selected);

            lstTransparencyRanges.setEnabled(selected);

        }

        DefaultListModel<TransparencyRange> transparencyListModel = (DefaultListModel<TransparencyRange>) lstTransparencyRanges.getModel();

        if(e.getSource() == btnAdd){

            Color color = this.colorChooser.getColor();

            RasterLegendManager legendManager = RasterLegendLocator.getRasterLegendManager();

            TransparencyRange range = legendManager.createTransparencyRange();

            if(chkRed.isSelected()){

                int[] values = new int[2];

                values[0] = color.getRed();

                values[1] = color.getRed();

                range.setRed(values);

            }

            if(chkGreen.isSelected()){

                int[] values = new int[2];

                values[0] = color.getGreen();

                values[1] = color.getGreen();

                range.setGreen(values);

            }

            if(chkBlue.isSelected()){

                int[] values = new int[2];

                values[0] = color.getBlue();

                values[1] = color.getBlue();

                range.setBlue(values);

            }

            range.setAnd(rbtnAnd.isSelected());

            if(chkAlpha.isSelected()){

                range.setAlpha(color.getAlpha());

            } else {

                range.setAlpha(0);

            }

            int index = getIndex(range);

            if(index < 0){

                transparencyListModel.addElement(range);

                lstTransparencyRanges.setSelectedIndex(transparencyListModel.getSize()-1);

            } else {

                ((DefaultListModel<TransparencyRange>)lstTransparencyRanges.getModel()).setElementAt(range, index);

                lstTransparencyRanges.setSelectedIndex(index);

            }

        }

        if(e.getSource() == btnRemove){

            int selectedIndex = lstTransparencyRanges.getSelectedIndex();

            transparencyListModel.removeElementAt(selectedIndex);

        }

    }

//    private boolean isRangeInList(TransparencyRange range) {

//        if(getIndex(range)>=0){

//            return true;

//        }

//        return false;

//    }

    private int getIndex(TransparencyRange range) {

        for(int i=0; i<lstTransparencyRanges.getModel().getSize(); i++){

            TransparencyRange element = (TransparencyRange) lstTransparencyRanges.getModel().getElementAt(i);

            if(element.isSameRange(range)){

                return i;

            }

        }

        return -1;

    }

    @Override

    public void stateChanged(ChangeEvent e) {

        if (e.getSource() == sldOpacity) {

            Integer value = (Integer) spOpacity.getValue();

            if (value != sldOpacity.getValue()) {

                spOpacity.setValue(new Integer(sldOpacity.getValue()));

            }

        }

        if (e.getSource() == spOpacity) {

            Integer value = (Integer) spOpacity.getValue();

            if (value <= 0) {

                spOpacity.setValue(0);

            }

            ;

            if (value >= 255) {

                spOpacity.setValue(255);

            }

            ;

            if (value != sldOpacity.getValue()) {

                sldOpacity.setValue(value);

            }

        }

    }

    @Override

    public void valueChanged(ListSelectionEvent e) {

        if (e.getSource() == lstTransparencyRanges) {

            int selectedIndex = lstTransparencyRanges.getSelectedIndex();

            if (selectedIndex >= 0) {

                btnRemove.setEnabled(true);

                TransparencyRange range =

                    (TransparencyRange) lstTransparencyRanges.getModel().getElementAt(selectedIndex);

                setRange(range);

            } else {

                btnRemove.setEnabled(false);

            }

        }

        return;

    }

    /**

     * @param TransparencyRange range

     */

    private void setRange(TransparencyRange range) {

        int alpha = range.getAlpha();

        int[] redValues = range.getRed();

        int red = 255;

        if (redValues[0] >= 0) {

            red = redValues[0];

            chkRed.setSelected(true);

        } else {

            chkRed.setSelected(false);

        }

        int[] greenValues = range.getGreen();

        int green = 255;

        if (greenValues[0] >= 0) {

            green = greenValues[0];

            chkGreen.setSelected(true);

        } else {

            chkGreen.setSelected(false);

        }

        int[] blueValues = range.getBlue();

        int blue = 255;

        if (blueValues[0] >= 0) {

            blue = blueValues[0];

            chkBlue.setSelected(true);

        } else {

            chkBlue.setSelected(false);

        }

        if (alpha >= 0) {

            setColor(new Color(red, green, blue, alpha));

            chkAlpha.setSelected(true);

        } else {

            setColor(new Color(red, green, blue));

            chkAlpha.setSelected(false);

        }

        if(range.isAnd()){

            rbtnAnd.setSelected(true);

            rbtnOr.setSelected(false);

        } else {

            rbtnAnd.setSelected(false);

            rbtnOr.setSelected(true);

        }

    }

    @Override

    public void setColor(Color color) {

        colorChooser.setColor(color);

    }

}