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.algorithm.saveraster;

import java.awt.Dimension;

import java.awt.geom.AffineTransform;

import java.io.File;

import java.io.IOException;

import java.util.Date;

import org.cresques.cts.IProjection;

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.dal.coverage.util.CRSUtils;

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

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.layers.FLayers;

import org.gvsig.raster.algorithm.process.ProcessException;

import org.gvsig.raster.algorithm.process.DataProcess;

/**

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

        public static String      VIEWPORT          = "ViewPort";

        public static String      PATH              = "Path";

        public static String      DIMENSION         = "Dimension";

        public static String      BLOCKSIZE         = "Blocksize";

        public static String      FILENAME          = "FileName";

        public static String      PARAMS            = "WriterParams";

        public static String      FLYRS             = "FLyrs";

        public static String      TIME              = "Time";

        public static String      PROJECTION        = "Projection";

        private IProjection       projection        = null;

        private ViewPort          viewPort          = null;

        private Dimension         dimension         = null;

        private RasterizeFLayers  rasterizerLayer   = null;

        private String            fileName          = "";

        private Params            writerParams      = null;

        //private CopyDataset       jp2Copy           = null;

        private boolean           supportImage      = false;

        private RasterManager     rManager          = RasterLocator.getManager();

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

        public static void registerParameters() {

                registerInputParameter(VIEWPORT, ViewPort.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(PATH, String.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(DIMENSION, Dimension.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(FLYRS, FLayers.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(PARAMS, Params.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerOutputParameter(PROJECTION, IProjection.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerOutputParameter(FILENAME, String.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

                registerOutputParameter(TIME, Long.class, SaveRasterAlgorithmLibrary.PROCESS_LABEL);

        }

        public void init() {

                viewPort = getParam(VIEWPORT) != null ? (ViewPort)getParam(VIEWPORT) : null;

                dimension = getParam(DIMENSION) != null ? (Dimension) getParam(DIMENSION) : null;

                FLayers flyrs = getParam(FLYRS) != null ? (FLayers) getParam(FLYRS) : null;

                int blockSize = getIntParam(BLOCKSIZE);

                rasterizerLayer = new RasterizeFLayers(flyrs, viewPort, blockSize);

                fileName = getStringParam(FILENAME);

                writerParams = getParam(PARAMS) != null ? (Params) getParam(PARAMS) : null;

                projection = getParam(PROJECTION) != null ? (IProjection) getParam(PROJECTION) : null;

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

        }

        /**

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

         */

        public void process() throws ProcessInterruptedException, ProcessException {

                TaskEventManager task = rManager.getRasterTask();

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

                vpData.setDPI((int)viewPort.getDPI());

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

                                        throw new ProcessException("error_processing", 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) {

                        throw new ProcessException("error_processing", ev);

                } catch(OutOfMemoryError ev) {

                        throw new ProcessException("memoria_excedida", ev);

                } finally {

                        if (incrementableTask != null) {

                                incrementableTask.processFinalize();

                                incrementableTask = null;

                        }

                }

        }

        /**

         * 

         * @param name

         * @param at

         * @throws IOException

         * @throws ProcessException 

         * @throws InterruptedException

         */

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

                //TODO: Usar el nuevo API basado en NewRasterStoreParameters y DataServerExplorer. Ejemplo en ClippingProcess

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

                        CRSUtils crsUtil = RasterLocator.getManager().getCRSUtils();

                        geoRasterWriter.setWkt(crsUtil.convertIProjectionToWkt(projection));

                } catch (NotSupportedExtensionException e) {

                        throw new ProcessException("extension_no_soportada", e);

                } catch (RasterDriverException e) {

                        throw new ProcessException("no_driver_escritura", 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) throws ProcessException {

                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)

                                throw new ProcessException("no_driver_escritura");

                        return writer.getParams();

                } catch (NotSupportedExtensionException e1) {

                        throw new ProcessException("no_driver_escritura", e1);

                } catch (RasterDriverException e1) {

                        throw new ProcessException("no_driver_escritura", e1);

                }

        }

        /**

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

        }

}