| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

*

* Copyright (C) 2005 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.cutting;

import java.awt.Component;

import java.awt.geom.AffineTransform;

import java.io.File;

import java.io.IOException;

import javax.swing.JOptionPane;

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

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

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

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

import org.gvsig.raster.buffer.BufferFactory;

import org.gvsig.raster.buffer.BufferInterpolation;

import org.gvsig.raster.buffer.RasterBuffer;

import org.gvsig.raster.dataset.GeoRasterWriter;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.dataset.IRasterDataSource;

import org.gvsig.raster.dataset.InvalidSetViewException;

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.Params;

import org.gvsig.raster.dataset.RasterDriverException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.exceptions.layers.LoadLayerException;

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

import com.iver.cit.gvsig.project.documents.view.gui.View;

/**

 * <code>CuttingProcess</code> es un proceso que usa un <code>Thread</code>

 * para calcular el recorte de una capa.

 *

 * @version 24/04/2007

 * @author BorSanZa - Borja S?nchez Zamorano (borja.sanchez@iver.es)

 */

public class CuttingProcess implements Runnable, IncrementableListener {

        private String                                         fileName = "";

        private IncrementableTask                incrementableTask = null;

        private volatile Thread                        blinker = null;

        private WriterBufferServer                writerBufferServer = null;

        private IRasterDataSource                raster = null;

        private AffineTransform         at = null;

        private boolean                                        oneLayerPerBand = false;

        private int[]                                        drawableBands = null;

        private int[]                                        dValues = null;

        private boolean                                        cancel = false;

        private GeoRasterWriter                 grw = null;

        private int                                                interpMethod = BufferInterpolation.INTERPOLATION_Undefined;

        private boolean                                 loadToc = false;

        private String                                        viewName = "";

        private Params                                        params = null;

        /**

         * Variables de la resoluci?n de salida

         */

        private int                                                                        resolutionWidth = 0, resolutionHeight = 0;

        /**

         * Crea un <code>CuttingProcess</code> para generar un recorte

         * @param loadToc

         * @param dValues

         * @param fileName

         * @param writerBufferServer

         * @param rasterMultiDataset

         * @param extent

         * @param drawableBands

         * @param oneLayerPerBand

         */

        public CuttingProcess(boolean loadToc, String viewName, int[] dValues, String fileName,

                                                        WriterBufferServer writerBufferServer, IRasterDataSource raster,

                                                        AffineTransform at, int[] drawableBands, boolean oneLayerPerBand, int interpMethod) {

                this.fileName = fileName;

                this.writerBufferServer = writerBufferServer;

                this.raster = raster;

                this.at = at;

                this.oneLayerPerBand = oneLayerPerBand;

                this.drawableBands = drawableBands;

                this.loadToc = loadToc;

                this.dValues = dValues;

                this.viewName = viewName;

                this.interpMethod = interpMethod;

        }

        /**

         * Arranca el proceso de recorte de un layer

         */

        public void start() {

                cancel = false;

                blinker = new Thread(this);

                blinker.start();

        }

        /**

         * Establecer la resolucion de salida al crear el recorte

         * @param width

         * @param height

         */

        public void setResolution(int width, int height) {

                resolutionWidth = width;

                resolutionHeight = height;

        }

        /**

         * M?todo donde se ejecutar? el Thread, aqu? se genera el recorte

         */

        public void run() {

                try {

                        IRasterDataSource d = raster.copy();

                        BufferFactory bufferFactory = new BufferFactory(d);

                        bufferFactory.setDrawableBands(drawableBands);

                        if(cancel)

                                return;

                        IBuffer buffer = null;

                        if(        interpMethod != BufferInterpolation.INTERPOLATION_Undefined &&

                                interpMethod != BufferInterpolation.INTERPOLATION_NearestNeighbour) {

                                bufferFactory.setAreaOfInterest(dValues[0], dValues[3], dValues[2] - dValues[0], dValues[1] - dValues[3]);

                                buffer = bufferFactory.getRasterBuf();

                                buffer = ((RasterBuffer)buffer).getAdjustedWindow(resolutionWidth, resolutionHeight, interpMethod);

                        } else {

                                try {

                                        bufferFactory.setAreaOfInterest(dValues[0], dValues[3], Math.abs(dValues[2] - dValues[0]), Math.abs(dValues[1] - dValues[3]), resolutionWidth, resolutionHeight);

                                        buffer = bufferFactory.getRasterBuf();

                                } catch (InvalidSetViewException e) {

                                        NotificationManager.addError("No se ha podido asignar la vista al inicial el proceso de recorte.", e);

                                }

                        }

                        //TODO: FUNCIONALIDAD: Poner los gerWriter con la proyecci?n del fichero fuente

                        if (oneLayerPerBand) {

                                for (int i = 0; i < drawableBands.length; i++) {

                                        if(cancel)

                                                return;

                                        String f = fileName + "_B" + drawableBands[i] + ".tif";

                                        writerBufferServer.setBuffer(buffer, i);

                                        Params p = null;

                                        if(params == null)

                                                p = GeoRasterWriter.getWriter(f).getParams();

                                        else

                                                p = params;

                                        grw = GeoRasterWriter.getWriter(writerBufferServer, f,

                                                        1, at, buffer.getWidth(), buffer.getHeight(),

                                                        buffer.getDataType(), p, null);

                                        grw.setWkt(d.getWktProjection());

                                        grw.dataWrite();

                                        grw.writeClose();

                                        if(!cancel)

                                                cutFinalize(f);

                                }

                        } else {

                                writerBufferServer.setBuffer(buffer, -1);

                                if(params == null) {

                                        grw = GeoRasterWriter.getWriter(writerBufferServer, fileName + ".tif",

                                                        buffer.getBandCount(), at, buffer.getWidth(), buffer.getHeight(),

                                                        buffer.getDataType(), GeoRasterWriter.getWriter(fileName + ".tif").getParams(), null);

                                } else

                                        grw = GeoRasterWriter.getWriter(writerBufferServer, fileName,

                                                        buffer.getBandCount(), at, buffer.getWidth(), buffer.getHeight(),

                                                        buffer.getDataType(), params, null);

                                grw.setWkt(d.getWktProjection());

                                grw.dataWrite();

                                grw.writeClose();

                                if(!cancel) {

                                        if(params == null)

                                                cutFinalize(fileName + ".tif");

                                        else

                                                cutFinalize(fileName);

                                }

                        }

                        d.close();

                } catch (NotSupportedExtensionException e) {

                        NotificationManager.addError("No se ha podido obtener el writer. Extensi?n no soportada", e);

                } catch (RasterDriverException e) {

                        NotificationManager.addError("No se ha podido obtener el writer.", e);

                } catch (IOException e) {

                        NotificationManager.addError("Error en la escritura en GeoRasterWriter.", e);

                }

                if (incrementableTask != null)

                        incrementableTask.processFinalize();

        }

        /**

         * Establece el <code>IncrementableTask</code>

         * @param value

         */

        public void setIncrementableTask(IncrementableTask value) {

                incrementableTask = value;

                incrementableTask.addIncrementableListener(this);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IncrementableListener#actionCanceled(org.gvsig.gui.beans.incrementabletask.IncrementableEvent)

         */

        public void actionCanceled(IncrementableEvent e) {

                if(writerBufferServer != null)

                        writerBufferServer.setCanceled(true, 0);

                if(grw != null)

                        grw.writeCancel();

                cancel = true;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IncrementableListener#actionResumed(org.gvsig.gui.beans.incrementabletask.IncrementableEvent)

         */

        public void actionResumed(IncrementableEvent e) {

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.gui.beans.incrementabletask.IncrementableListener#actionSuspended(org.gvsig.gui.beans.incrementabletask.IncrementableEvent)

         */

        public void actionSuspended(IncrementableEvent e) {

        }

        /**

         * Acciones que se realizan al finalizar de crear los recortes de imagen.

         * Este m?todo es llamado por el thread TailRasterProcess al finalizar.

         */

        private void cutFinalize(String fileName) {

                if (!loadToc)

                        return;

                if(!new File(fileName).exists())

                        return;

                // seleccionamos la vista de gvSIG

                com.iver.cit.gvsig.project.documents.view.gui.View theView = null;

                try {

                        IWindow[] allViews = PluginServices.getMDIManager().getAllWindows();

                        for (int i = 0; i < allViews.length; i++) {

                                if (allViews[i] instanceof com.iver.cit.gvsig.project.documents.view.gui.View

                                        && PluginServices.getMDIManager().getWindowInfo((View) allViews[i]).getTitle().equals(viewName))

                                        theView = (com.iver.cit.gvsig.project.documents.view.gui.View) allViews[i];

                        }

                        if (theView == null)

                                return;

                } catch (ClassCastException ex) {

                        // logger.error(PluginServices.getText(this,"cant_get_view "), ex);

                        return;

                }

                // Cargamos las capas

                theView.getMapControl().getMapContext().beginAtomicEvent();

                //Driver driver = LayerFactory.getDM().getDriver("gvSIG Raster Driver");

                int endIndex = fileName.lastIndexOf(".");

                if (endIndex < 0)

                        endIndex = fileName.length();

                FLayer lyr = null;

                try {

                        lyr = FLyrRasterSE.createLayer(fileName.substring(

                                        fileName.lastIndexOf(File.separator) + 1, endIndex),

                                        new File(fileName), theView.getMapControl().getProjection());

                } catch (LoadLayerException e) {

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

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

                }

                theView.getMapControl().getMapContext().getLayers().addLayer(lyr);

                theView.getMapControl().getMapContext().invalidate();

                theView.getMapControl().getMapContext().endAtomicEvent();

//                        grid.free();

        }

        /**

         * Asigna los par?metros si estos han de ser cambiados. En caso de

         * que no se asignen se usaran los que tiene el driver por defecto.

         * @param params Par?metros del driver de escritura.

         */

        public void setParams(Params params) {

                this.params = params;

        }

}