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.saveraster.ui.listener;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.util.ArrayList;

import javax.swing.JFrame;

import javax.swing.JOptionPane;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanel;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelEvent;

import org.gvsig.gui.beans.buttonspanel.ButtonsPanelListener;

import org.gvsig.gui.beans.incrementabletask.IncrementableTask;

import org.gvsig.gui.beans.propertiespanel.PropertiesPanel;

import org.gvsig.raster.RasterLibrary;

import org.gvsig.raster.buffer.IQueryableRaster;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.dataset.WriterParams;

import org.gvsig.raster.dataset.WriterParams.Param;

import org.gvsig.raster.util.RasterUtilities;

import org.gvsig.rastertools.saveraster.operations.RasterizerLayer;

import org.gvsig.rastertools.saveraster.operations.SaveRasterProcess;

import org.gvsig.rastertools.saveraster.ui.SaveRasterDialog;

import org.gvsig.rastertools.saveraster.ui.SaveRasterPanel;

import org.gvsig.rastertools.saveraster.ui.properties.WriterPropertiesDialog;

import com.iver.andami.PluginServices;

import com.iver.cit.gvsig.fmap.MapControl;

import com.iver.cit.gvsig.fmap.ViewPort;

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

/**

 * Clase que gestiona los eventos de los botones del dialogo de Salvar a raster.

 *

 * @version 19/04/2007

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

 *

 */

public class SaveRasterDialogListener implements ButtonsPanelListener{

        private SaveRasterDialog         dialog = null;

        private FLayers                                layers = null;

        private MapControl                        mapCtrl = null;

        private GeoRasterWriter         writer = null;

        /**

         * Constructor

         * @param dialog

         */

        public SaveRasterDialogListener(SaveRasterDialog dialog, FLayers layers, MapControl mapCtrl) {

                this.dialog = dialog;

                this.layers = layers;

                this.mapCtrl = mapCtrl;

                dialog.addButtonPressedListener(this);

                //Captura de eventos para el bot?n de propiedades

                ((SaveRasterPanel)dialog.getControlsPanel()).getBProperties().addActionListener(new java.awt.event.ActionListener() {

            public void actionPerformed(java.awt.event.ActionEvent evt) {

                    propsButtonActionPerformed(evt);

             }

        });

        }

        /**

         * Eventos para los botones de aplicar, aceptar y cancelar

         * @param e

         */

        public void actionButtonPressed(ButtonsPanelEvent e) {

                switch (e.getButton()) {

                case ButtonsPanel.BUTTON_ACCEPT:

                        acceptButtonActionPerformed(e);

                        dialog.closeJDialog();

                        break;

                case ButtonsPanel.BUTTON_APPLY:

                        acceptButtonActionPerformed(e);

                        break;

                case ButtonsPanel.BUTTON_CANCEL:

                        dialog.closeJDialog();

                        break;

                }

        }

        /**

         * Acciones realizadas al aceptar.

         * @param e

         */

    private boolean acceptButtonActionPerformed(ButtonsPanelEvent e) {

            SaveRasterPanel controlPanel = dialog.getControlsPanel();

                String fName = dialog.getDataInputListener().getFileName();

                Dimension dimension = new Dimension(dialog.getDataInputListener().getWidthInPixels(),

                                                                                        dialog.getDataInputListener().getHeightInPixels() );

                //Comprobamos que el en la ruta de destino tengamos permisos de escritura

                File f = new File(fName);

                if(!f.exists()) {

                        if(!checkFileExists(fName))

                                return false;

                }

                f = new File(fName.substring(0, fName.lastIndexOf(File.separator)));

                if(f.exists() && f.isDirectory() && !f.canWrite()) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(),

                                        PluginServices.getText(this, "error_file_not_writable"));

                        return false;

                }

                double lrX = Double.parseDouble(controlPanel.getTInfDerX().getText());

                double lrY = Double.parseDouble(controlPanel.getTInfDerY().getText());

                double ulX = Double.parseDouble(controlPanel.getTSupIzqX().getText());

                double ulY = Double.parseDouble(controlPanel.getTSupIzqY().getText());

                double width = 0;

                if(ulX > lrX) width = (double)(ulX - lrX);

                        else width = (double)(lrX - ulX);

                double height = 0;

                if(ulY > lrY) height = (double)(ulY - lrY);

                        else height = (double)(lrY - ulY);

                Rectangle2D ext = null;

                if(ulY < lrY)

                        ext = new Rectangle2D.Double(ulX, ulY, width, height);

                else

                        ext = new Rectangle2D.Double(ulX, lrY, width, height);

                WriterParams params = getWriterParams(fName);

                int blockSize = getBlockSize(params, layers);

                ViewPort vp = new ViewPort(layers.getProjection());

                vp.setBackColor(mapCtrl.getViewPort().getBackColor());

                vp.setImageSize(dimension);

                vp.setExtent(ext);

                //Creamos el servidor de datos de la vista

                RasterizerLayer rasterizer = new RasterizerLayer(layers, vp, blockSize);

                //Creamos la clase con el proceso y lo lanzamos

                SaveRasterProcess saveRasterProcess = new SaveRasterProcess(vp, dimension, rasterizer, fName, params);

                IncrementableTask incrementableTask = new IncrementableTask(rasterizer);

                saveRasterProcess.setIncrementableTask(incrementableTask);

                incrementableTask.showWindow();

                saveRasterProcess.start();

                incrementableTask.start();

                return true;

    }

    /**

         * Comprueba si existe el fichero pasado por par?metro existe solicitando confirmaci?n

         * de sobreescritura. Si el usuario acepta devolver? true y si no acepta devuelve false.

         * Si se acepta la sobreescritura el fichero es borrado.

         * @param file Cadena con el nombre del fichero a comprobar su existencia.

         * @return Devuelve true si el fichero existe y va a ser sobreescrito

         */

        private boolean checkFileExists(String file){

                File f = new File(file);

                if(f.exists()){

                        String string1 = PluginServices.getText(this, "yes");

                        String string2 = PluginServices.getText(this, "no");

                        Object[] options = {string1, string2};

                        int n = JOptionPane.showOptionDialog((Component)PluginServices.getMainFrame(),

                                        PluginServices.getText(this, "error_file_exists"),

                                        PluginServices.getText(this, "confirmacion"),

                                        JOptionPane.YES_NO_OPTION,

                                        JOptionPane.QUESTION_MESSAGE,

                                        null,

                                        options,

                                        string1);

                        if (n == JOptionPane.YES_OPTION){

                                f.delete();

                                return true;

                        }else

                                return false;

                }

                return true;

        }

    /**

     * Funci?n a ejecutar cuando se pulsa el bot?n de propiedades.

     * @param e ActionEvent

     */

    private void propsButtonActionPerformed(ActionEvent e){

            String name = dialog.getDataInputListener().getFileName();

                //Si no se ha seleccionado ning?n fichero salimos

                if(name == null || name.equals(""))

                        return;

                WriterParams params = getWriterParams(name);

                PropertiesPanel panel = new PropertiesPanel();

                WriterPropertiesDialog dialog = new WriterPropertiesDialog(panel, params);

                PluginServices.getMDIManager().addWindow(dialog);

        }

    /**

     * Obtiene los par?metros del driver de escritura. Si el driver no se ha creado a?n se obtienen

     * unos par?metros con la inicializaci?n por defecto. Si se ha creado ya y se han modificado se

     * devuelven los par?metros con las modificaciones. Si se cambia de driver se devolver? un WriterParams

     * como si fuera la primera vez que se abre.

     * @param name Nombre del fichero sobre el que se salva.

     * @return WriterParams

     */

    private WriterParams getWriterParams(String name) {

            String ext = RasterUtilities.getExtensionFromFileName(name);

            try {

                        if(writer == null) //La primera vez que se obtiene el driver

                                writer = GeoRasterWriter.getWriter(name);

                        else {

                                String newType = GeoRasterWriter.getDriverType(ext);

                                String oldType = writer.getDriverName();

                                if(!newType.equals(oldType))  //Cambio de driver despu?s de haber seleccionado y modificado las propiedades de uno

                                        writer = GeoRasterWriter.getWriter(name);

                        }

                        if(writer == null)

                                JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), PluginServices.getText(this, "no_driver_escritura"));

                        return writer.getParams();

                } catch (NotSupportedExtensionException e1) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), PluginServices.getText(this, "no_driver_escritura"));

                        return null;

                } catch (RasterDriverException e1) {

                        JOptionPane.showMessageDialog((Component)PluginServices.getMainFrame(), PluginServices.getText(this, "no_driver_escritura"));

                        return null;

                }

    }

    /**

     * Calculo del tama?o de bloque. Para ello comprueba si el raster est? siendo tileado. Si est? siendo

     * tileado se aplica el alto del tile menor para no pedir un bloque mayor que lo que un servidor puede

     * devolver. Si no est? siendo tileado se devuelve el tama?o de bloque de los par?metros.

     * @param flyrs Capas

     * @param params Par?metros del driver de escritura

     * @return tama?o de bloque

     */

    private int getBlockSize(WriterParams params, FLayers flyrs) {

            int blockSize = RasterLibrary.blockHeight;

                try {

                        blockSize = Integer.parseInt(params.getParamById("blocksize").defaultValue);

                } catch (NumberFormatException ex) {

                        //Se queda con el valor por defecto

                }

                //Recorremos todas las capas comprobando si alguna de ellas implementa RasterOperations y tilea.

                //En ese caso se obtiene el ancho de bloque. El ancho de bloque total ser? el menor obtenido.

                //Esto lo hacemos para que las capas que tilean WMS, WCS, ... no hagan demasiadas peticiones al servidor

                //por tener un ancho de bloque muy peque?o de modo que el ancho del bloque se ajuste al Tile menor

                //soportado por los servidores que intervienen en el salvado.

                int[] wBlock = null;

                boolean isTiling = false;

                int block = Integer.MAX_VALUE;

                for(int i = 0; i < flyrs.getLayersCount(); i++){

                        if(flyrs.getLayer(i) instanceof IQueryableRaster){

                                if(((IQueryableRaster)flyrs.getLayer(i)).isTiled()){

                                        wBlock = ((IQueryableRaster)flyrs.getLayer(i)).getTileSize();

                                        if((wBlock[0] - 1)< block){

                                                block = wBlock[0] - 1;

                                                isTiling = true;

                                        }

                                }

                        }

                }

                if(isTiling) {

                        params.changeParamValue("blocksize", String.valueOf(block));

                        return block;

                }

                return blockSize;

    }

}