Application: gvSIG desktop

/* 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.panels;

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

        /**

     * 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) fLayer).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, IRasterDataset dset, IRasterProperties prop, FLayer lyr){

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

                this.bandSetupPanel = bs;

                this.dataset = dset;

                this.prop = prop;

                fLayer = lyr;

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

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

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

        }

        /**

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

                setNewBandsPositionInRendering();

        }

        /**

         * Acciones a ejecutar cuando se aplica

         */

        public void apply(){

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

                }

        }

}