Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

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

* 

*/

package org.gvsig.raster.tools.app.basic.tool.saveraster.operation;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.geom.AffineTransform;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Date;

import javax.swing.JOptionPane;

import org.gvsig.andami.PluginServices;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.RasterManager;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.datastruct.Extent;

import org.gvsig.fmap.dal.coverage.datastruct.Params;

import org.gvsig.fmap.dal.coverage.datastruct.ViewPortData;

import org.gvsig.fmap.dal.coverage.exception.NotSupportedExtensionException;

import org.gvsig.fmap.dal.coverage.exception.ProcessInterruptedException;

import org.gvsig.fmap.dal.coverage.exception.RasterDriverException;

import org.gvsig.fmap.dal.coverage.process.TaskEventManager;

import org.gvsig.fmap.dal.coverage.store.DataServerWriter;

import org.gvsig.fmap.dal.coverage.store.RasterWriter;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.ViewPort;

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

import org.gvsig.i18n.Messages;

import org.gvsig.raster.fmap.layers.FLyrRaster;

import org.gvsig.raster.swing.RasterSwingLibrary;

import org.gvsig.raster.tools.app.basic.RasterToolsUtil;

import org.gvsig.raster.tools.app.basic.raster.process.RasterProcess;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Thread que se encarga de llamar a los writer para realizar la tarea de

 * salvado y/p compresi?n

 *

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

 */

public class SaveRasterProcess extends RasterProcess {

        private ViewPort          viewPort          = null;

        private Dimension         dimension         = null;

        private RasterizerLayer   rasterizerLayer   = null;

        private String            fileName          = "";

        private Params            writerParams      = null;

        private CopyDataset       jp2Copy           = null;

        private boolean           supportImage      = false;

        private ArrayList<FLyrRaster>

                              layers            = null;

        private RasterManager     rManager          = RasterLocator.getManager();

        private Logger            logger            = LoggerFactory.getLogger(RasterProcess.class.toString());

        @SuppressWarnings("unchecked")

        public void init() {

                viewPort = (ViewPort) getParam("viewport");

                dimension = (Dimension) getParam("dimension");

                rasterizerLayer = (RasterizerLayer) getParam("rasterizerlayer");

                fileName = getStringParam("filename");

                writerParams = (Params) getParam("writerparams");

                supportImage = getBooleanParam("remotelayers") && fileName.endsWith(".jp2");

                layers = (ArrayList<FLyrRaster>) getParam("layers");

        }

        /**

         * Procesos de escritura de una porci?n de la vista.

         */

        public void process() throws ProcessInterruptedException {

                TaskEventManager task = rManager.getRasterTask();

                if(layers != null) {

                        for (int i = 0; i < layers.size(); i++) 

                                layers.get(i).setReadingData(Thread.currentThread().getId() + "");

                }

                jp2Copy = null;

                long t2;

                long t1 = new java.util.Date().getTime();

                //Creamos el driver

                Envelope env = viewPort.getAdjustedEnvelope();

                Extent ex = rManager.getDataStructFactory().createExtent(env.getMinimum(0), env.getMaximum(1), env.getMaximum(0), env.getMinimum(1));

                Dimension imgSz = viewPort.getImageSize();

                ViewPortData vpData = rManager.getDataStructFactory().createViewPortData(viewPort.getProjection(), ex, imgSz );

                AffineTransform at = new AffineTransform(vpData.getExtent().width() / imgSz.width,

                                                                                                 0, 0,

                                                                                                 -(vpData.getExtent().height() / imgSz.height),

                                                                                                 vpData.getExtent().getULX(),

                                                                                                 vpData.getExtent().getULY());

                String oldFileName = fileName;

                if(supportImage) {

                        fileName = fileName.substring(0, Math.min(fileName.lastIndexOf(File.separator) + 1, fileName.length() - 1));

                        fileName += rManager.getFileUtils().usesOnlyLayerName() + ".tif";

                        writerParams = getWriterParams(fileName);

                }

                //Ejecutamos el driver con los datos pasados

                try {

                        write(fileName, at, writerParams, rasterizerLayer);

                        if(task.getEvent() != null)

                                task.manageEvent(task.getEvent());

                        if(supportImage) {

                                try {

                                        insertLineLog(Messages.getText("saving_jp2"));

                                        jp2Copy = new CopyDataset(fileName, oldFileName, incrementableTask);

                                        jp2Copy.copy();

                                        new File(fileName).delete();

                                        new File(rManager.getFileUtils().getRMFNameFromFileName(fileName)).delete();

                                } catch (LoadLayerException e) {

                                } catch (org.gvsig.raster.algorithm.process.ProcessException e) {

                                        logger.debug(RasterLocator.getManager().getRasterUtils().getTrace(e));

                                        RasterSwingLibrary.messageBoxError("error_processing", this, e);

                                }

                                jp2Copy = null;

                        } 

                        t2 = new Date().getTime();

                        try {

                                saveRasterFinalize(oldFileName, (t2 - t1));

                        } catch(ArrayIndexOutOfBoundsException exc) {

                                //Si la ventana se ha cerrado ya es porque ha sido cancelada por lo que

                                //producir? esta excepci?n. En este caso no se lanza la ventana de informaci?n

                                //de finalizaci?n.

                        }

                } catch(IOException ev) {

                        if (incrementableTask != null) {

                                incrementableTask.processFinalize();

                                incrementableTask = null;

                        }

                        RasterSwingLibrary.messageBoxError("error_processing", this, ev);

                        return;

                } catch(OutOfMemoryError ev) {

                        if (incrementableTask != null) {

                                incrementableTask.processFinalize();

                                incrementableTask = null;

                        }

                        RasterSwingLibrary.messageBoxError("memoria_excedida", this, new IOException("Out of memory en salvar a raster"));

                } finally {

                        if(layers != null) {

                                for (int i = 0; i < layers.size(); i++) 

                                        layers.get(i).setReadingData(null);

                        }

                }

        }

        /**

         * 

         * @param name

         * @param at

         * @throws IOException

         * @throws InterruptedException

         */

        private void write(String name, AffineTransform at, Params writerParams, DataServerWriter rasterizerLayer) throws IOException, ProcessInterruptedException {

                TaskEventManager task = rManager.getRasterTask();

                RasterWriter geoRasterWriter = null;

                try {

                        //TODO: FUNCIONALIDAD: Poner los gerWriter con la proyecci?n de la vista

                        geoRasterWriter = rManager.createWriter(rasterizerLayer, name,

                                                                                                3, at, dimension.width,

                                                                                                dimension.height, Buffer.TYPE_IMAGE, writerParams, null);

                } catch (NotSupportedExtensionException e) {

                        RasterSwingLibrary.messageBoxError("extension_no_soportada", (Component)PluginServices.getMainFrame(), e);

                } catch (RasterDriverException e) {

                        RasterSwingLibrary.messageBoxError("no_driver_escritura", (Component)PluginServices.getMainFrame(), e);

                }

                if(task.getEvent() != null)

                        task.manageEvent(task.getEvent());

                geoRasterWriter.setCancellableRasterDriver(new ExternalCancellable(incrementableTask));

                geoRasterWriter.dataWrite();

                geoRasterWriter.writeClose();

        }

        /**

         * 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 Params getWriterParams(String name) {

                RasterWriter writer = null;

                String ext = rManager.getFileUtils().getExtensionFromFileName(name);

                try {

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

                                writer = rManager.createWriter(name);

                        else {

                                String newType = rManager.getProviderServices().getWriteDriverType(ext);

                                String oldType = writer.getDriverName();

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

                                        writer = rManager.createWriter(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;

                }

        }

        /**

         * Acciones que se realizan al finalizar de salvar a raster.

         * @param fileName Nombre del fichero

         * @param milis Tiempo que ha tardado en ejecutarse

         */

        private void saveRasterFinalize(String fileName, long milis) {

                if (incrementableTask != null)

                        incrementableTask.hideWindow();

                externalActions.end(new Object[]{fileName, new Long(milis)});

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getPercent()

         */

        public int getPercent() {

                if(rasterizerLayer != null)

                        return rasterizerLayer.getPercent();

                return 0;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getTitle()

         */

        public String getTitle() {

                if(rasterizerLayer != null)

                        return rasterizerLayer.getTitle();

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.rastertools.RasterProcess#getLog()

         */

        public String getLog() {

                if(rasterizerLayer != null)

                        return rasterizerLayer.getLog();

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IIncrementable#getLabel()

         */

        public String getLabel() {

                if(rasterizerLayer != null)

                        return rasterizerLayer.getLabel();

                return null;

        }

}