| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2007 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.rastertools.properties.control;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Rectangle2D;

import java.io.File;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.datastruct.Extent;

import org.gvsig.raster.gui.wizards.DriverFileFilter;

import org.gvsig.raster.hierarchy.IRasterDataset;

import org.gvsig.raster.hierarchy.IRasterProperties;

import org.gvsig.raster.hierarchy.IRasterRendering;

import org.gvsig.raster.util.RasterToolsUtil;

import org.gvsig.rastertools.properties.panels.BandSetupPanel;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.cit.gvsig.addlayer.fileopen.FileOpenWizard;

import com.iver.cit.gvsig.fmap.layers.FLayer;

/**

 * Clase que maneja los eventos del panel BandSetupPanel. Gestiona el a?adir o

 * eliminar ficheros de la lista y contiene las acciones a realizar cuando en

 * panel registrable se pulsa aceptar, aplicar o cancelar.

 *

 * @author Nacho Brodin (brodin_ign@gva.es)

 */

public class BandSetupListener implements ActionListener {

        private BandSetupPanel    bandSetupPanel = null;

        private JFileChooser      fileChooser    = null;

        private FLayer            fLayer         = null;

        private IRasterDataset    dataset        = null;

        private IRasterProperties prop           = null;

        private IRasterRendering  render         = null;

        private boolean           enabled        = true;

        /**

         * N?mero de bandas.

         */

        protected int             nbands         = 0;

        /**

         * Lista de geoRasterDataset correspondiente a los ficheros de bandas

         */

        protected RasterDataset[] grd            = null;

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         * @param lyr Capa raster

         */

        public BandSetupListener(BandSetupPanel bs) {

                this.bandSetupPanel = bs;

                bs.getFileList().getJButtonAdd().addActionListener(this);

                bs.getFileList().getJButtonRemove().addActionListener(this);

                bs.getFileList().getJComboBox().addActionListener(this);

        }

        /**

         * Constructor

         * @param bs Panel del selector de bandas

         * @param lyr Capa raster

         */

        public void init(IRasterDataset dset, IRasterProperties prop, FLayer lyr) {

                //TODO: FUNCIONALIDAD: Cancelaci?n para la selecci?n de bandas

                this.dataset = dset;

                this.prop = prop;

                fLayer = lyr;

                if(fLayer instanceof IRasterRendering)

                        render = (IRasterRendering)fLayer;

        }

        /**

         * Listener para la gesti?n de los botones de a?adir y eliminar fichero y

         * el combo de selecci?n de bandas.

         */

        public void actionPerformed(ActionEvent e) {

                if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonAdd()))

                        addFileBand();

                if (e.getSource().equals(bandSetupPanel.getFileList().getJButtonRemove()))

                        delFileBand();

                if (e.getSource().equals(bandSetupPanel.getFileList().getJComboBox()))

                        numberBandSelection();

                bandSetupPanel.onlyApply();

        }

        /**

         * Evento que activado cuando se modifica el n?mero de bandas que se desean

         * visualizar. Se obtiene este n?mero y se llama a la funci?n que controla el

         * cambio de n?mero de bandas visualizadas.

         */

        public void numberBandSelection() {

                String vBands = (String) bandSetupPanel.getFileList().getJComboBox().getSelectedItem();

                if (vBands != null) {

                        if (vBands.compareTo("3") == 0)

                                bandSetupPanel.assignMode((byte) 3);

                        if (vBands.compareTo("2") == 0)

                                bandSetupPanel.assignMode((byte) 2);

                        if (vBands.compareTo("1") == 0)

                                bandSetupPanel.assignMode((byte) 1);

                        bandSetupPanel.resetMode(bandSetupPanel.getMode());

                }

        }

        /**

         * A?ade una banda al raster

         * @param e

         */

        private void addFileBand() {

                // String[] driverNames = null;

                // Creaci?n del dialogo para selecci?n de ficheros

                fileChooser = new JFileChooser(FileOpenWizard.getLastPath());

                fileChooser.setMultiSelectionEnabled(true);

                fileChooser.setAcceptAllFileFilterUsed(false);

                fileChooser.addChoosableFileFilter(new DriverFileFilter());

                int result = fileChooser.showOpenDialog(bandSetupPanel);

                if (result == JFileChooser.APPROVE_OPTION) {

                        IRasterDataSource ds = dataset.getDataSource();

                        File[] files = fileChooser.getSelectedFiles();

                        FileOpenWizard.setLastPath(files[0].getPath());

                        // Lo a?adimos a la capa si no esta

                        for (int i = 0; i < files.length; i++) {

                                // Comprobamos que el fichero no est?

                                boolean exist = false;

                                for (int j = 0; j < ds.getDatasetCount(); j++) {

                                        if (dataset.getDataSource().getDataset(j)[0].getFName().endsWith(files[i].getName()))

                                                exist = true;

                                }

                                if (!exist) {

                                        try {

                                                Rectangle2D extentOrigin = prop.getFullRasterExtent().toRectangle2D();

                                                RasterDataset geoRasterDataset = RasterDataset.open(prop.getProjection(), files[i].getAbsolutePath());

                                                Extent extentNewFile = geoRasterDataset.getExtent();

                                                nbands += geoRasterDataset.getBandCount();

                                                // Comprobamos que el extent y tama?o del fichero a?adido sea igual al

                                                // fichero original. Si no es as? no abrimos la capa y mostramos un aviso

                                                double widthNewFile = (extentNewFile.getMax().getX() - extentNewFile.getMin().getX());

                                                double heightNewFile = (extentNewFile.getMax().getY() - extentNewFile.getMin().getY());

                                                if ((widthNewFile - extentOrigin.getWidth()) > 1.0 || (widthNewFile - extentOrigin.getWidth()) < -1.0 || (heightNewFile - extentOrigin.getHeight()) > 1.0

                                                                || (heightNewFile - extentOrigin.getHeight()) < -1.0) {

                                                        JOptionPane.showMessageDialog(null, PluginServices.getText(this, "extents_no_coincidentes"), "", JOptionPane.ERROR_MESSAGE);

                                                        continue;

                                                }

                                                if ((extentNewFile.getMax().getX() - extentNewFile.getMin().getX()) != extentOrigin.getWidth()

                                                                || (extentNewFile.getMax().getY() - extentNewFile.getMin().getY()) != extentOrigin.getHeight()) {

                                                        JOptionPane.showMessageDialog(null, PluginServices.getText(this, "extents_no_coincidentes"), "", JOptionPane.ERROR_MESSAGE);

                                                        continue;

                                                }

                                                geoRasterDataset.close();

                                        } catch (Exception exc) {

                                                RasterToolsUtil.messageBoxError("addband_error", bandSetupPanel, exc);

                                        }

                                        // Lo a?adimos a la capa

                                        try {

                                                dataset.addFile(files[i].getAbsolutePath());

                                        } catch (NotSupportedExtensionException e) {

                                                RasterToolsUtil.messageBoxError("addband_error", bandSetupPanel, e);

                                        } catch (RasterDriverException e) {

                                                RasterToolsUtil.messageBoxError("addband_error", bandSetupPanel, e);

                                        }

                                } else {

                                        RasterToolsUtil.messageBoxError("fichero_existe" + " " + files[i].getAbsolutePath(), bandSetupPanel);

                                        return;

                                }

                        }

                        // A?adimos los georasterfile a la tabla del Panel

                        bandSetupPanel.addFiles(ds);

                }

        }

        /**

         * Elimina una banda del raster. Si queda solo un fichero o no se ha

         * seleccionado ninguna banda no hace nada.

         *

         * @param e

         */

        private void delFileBand() {

                Object[] objects = bandSetupPanel.getFileList().getJList().getSelectedValues();

                for (int i = objects.length - 1; i >=0; i--) {

                        if (bandSetupPanel.getFileList().getNFiles() > 1) {

                                String pathName = objects[i].toString();

                                RasterDataset geoRasterFile = null;

                                try {

                                        geoRasterFile = RasterDataset.open(prop.getProjection(), pathName);

                                } catch (NotSupportedExtensionException e) {

                                        NotificationManager.addError(e.getMessage(), e);

                                } catch (RasterDriverException e1) {

                                        NotificationManager.addError(e1.getMessage(), e1);

                                }

                                nbands -= geoRasterFile.getBandCount();

                                dataset.delFile(pathName);

                                String file = pathName.substring(pathName.lastIndexOf("/") + 1);

                                file = file.substring(file.lastIndexOf("\\") + 1);

                                bandSetupPanel.removeFile(file);

                        }

                }

        }

        /**

         * Acciones a ejecutar cuando se aplica

         */

        public void apply() {

                if (enabled)

                        setNewBandsPositionInRendering();

        }

        /**

         * Asigna la posici?n de las bandas en el rederizado basandose en la selecci?n

         * hecho en la tabla de bandas.

         */

        public void setNewBandsPositionInRendering() {

                if (prop != null && prop.getRender() != null) {

                        if(render != null) {

                                render.getRenderBands()[0] = bandSetupPanel.getAssignedBand(RasterDataset.RED_BAND);

                                render.getRenderBands()[1] = bandSetupPanel.getAssignedBand(RasterDataset.GREEN_BAND);

                                render.getRenderBands()[2] = bandSetupPanel.getAssignedBand(RasterDataset.BLUE_BAND);

                        }

                        fLayer.getMapContext().invalidate();

                }

        }

        /**

         * Activa o desactiva la acci?n del panel

         * @param enabled true para activa y false para desactivar.

         */

        public void setEnabledPanelAction(boolean enabled) {

                this.enabled = enabled;

        }

}