| 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 javax.swing.filechooser.FileFilter;

import org.gvsig.fmap.layers.FLyrRasterSE;

import org.gvsig.raster.dataset.MultiRasterDataset;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDataset;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.grid.render.Rendering;

import org.gvsig.raster.shared.Extent;

import org.gvsig.raster.shared.IRasterDataset;

import org.gvsig.raster.shared.IRasterProperties;

import org.gvsig.rastertools.RasterModule;

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

import com.hardcode.driverManager.DriverLoadException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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 boolean                                         enabled = true;

        /**

         * N?mero de bandas.

         */

        protected int nbands = 0;

        /**

         * Lista de geoRasterDataset correspondiente a los ficheros de bandas

         */

        protected RasterDataset[] grd = null;

        /**

         * Filtro para la selecci?n de ficheros raster

         * 

         * @author Nacho Brodin (nachobrodin@gmail.com)

         */

        public class DriverFileFilter extends FileFilter {

                //private String driverName = null;

                public DriverFileFilter(String driverName) throws DriverLoadException {

                        //this.driverName = driverName;

                }

                /**

                 * @see javax.swing.filechooser.FileFilter#accept(java.io.File)

                 */

                public boolean accept(File f) {

                        if (f.isDirectory())

                                return true;

                        if (fLayer instanceof FLyrRasterSE)

                                return FLyrRasterSE.isFileAccepted(f);

                        else

                                throw new RuntimeException("Tipo no reconocido");

                }

                /**

                 * @see javax.swing.filechooser.FileFilter#getDescription()

                 */

                public String getDescription() {

                        return (String)dataset.getInfo("DriverName");

                }

        }

        /**

         * 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;

        }

        /**

         * 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();

        }

        /**

         * 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(RasterModule.getPath());

                fileChooser.setMultiSelectionEnabled(true);

                fileChooser.setAcceptAllFileFilterUsed(false);

                try {

                        FileFilter auxF = new DriverFileFilter((String) dataset.getInfo("DriverName"));

                        fileChooser.addChoosableFileFilter(auxF);

                } catch (DriverLoadException e1) {

                        NotificationManager.addError("No se pudo acceder a los drivers", e1);

                }

                int result = fileChooser.showOpenDialog(bandSetupPanel);

                if (result == JFileChooser.APPROVE_OPTION) {

                        MultiRasterDataset grmd = dataset.getGeoRasterMultiDataset();

                        File[] files = fileChooser.getSelectedFiles();

                        RasterModule.setPath(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 < grmd.getDatasetCount(); j++) {

                                        if (dataset.getGeoRasterMultiDataset().getDataset(j).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);

                                                        return;

                                                }

                                                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);

                                                        return;

                                                }

                                                dataset.getGeoRasterMultiDataset().addDataset(geoRasterDataset);

                                        } catch (Exception exc) {

                                                exc.printStackTrace();

                                        }

                                        // Lo a?adimos a la capa

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

                                } else {

                                        JOptionPane.showMessageDialog(null, PluginServices.getText(this, "fichero_existe") + " " + files[i].getAbsolutePath(), "", JOptionPane.ERROR_MESSAGE);

                                        return;

                                }

                        }

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

                        bandSetupPanel.addFiles(grmd);

                }

        }

        /**

         * 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() {

                if (bandSetupPanel.getFileList().getNFiles() > 1 && bandSetupPanel.getFileList().getJList().getSelectedValue() != null) {

                        String pathName = bandSetupPanel.getFileList().getJList().getSelectedValue().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 acepta

         */

        public void accept() {

                if (enabled)

                        setNewBandsPositionInRendering();

        }

        /**

         * Acciones a ejecutar cuando se aplica

         */

        public void apply() {

                if (enabled)

                        setNewBandsPositionInRendering();

        }

        /**

         * Acciones a ejecutar cuando se cancela

         */

        public void cancel() {

        }

        /**

         * 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) {

                        Rendering render = prop.getRender();

                        int[] renderBands = new int[] { bandSetupPanel.getAssignedBand(RasterDataset.RED_BAND), bandSetupPanel.getAssignedBand(RasterDataset.GREEN_BAND),

                                        bandSetupPanel.getAssignedBand(RasterDataset.BLUE_BAND) };

                        render.setRenderBands(renderBands);

                        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;

        }

}