| 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.io.IOException;

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

import org.gvsig.raster.dataset.NotSupportedExtensionException;

import org.gvsig.raster.dataset.RasterDriverException;

import org.gvsig.raster.dataset.MultiRasterDataset;

import org.gvsig.raster.shared.Extent;

import org.gvsig.rastertools.cutting.ui.listener.CuttingPanelListener;

/**

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

 * para calcular el recorte de una capa.

 * 

 * @version 24/04/2007

 * @author 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 MultiRasterDataset                rasterMultiDataset = null;

        private Extent                                        extent = null;

        private boolean                                        oneLayerPerBand = false;

        private int[]                                        drawableBands = null;

        private CuttingPanelListener        cuttingPanelListener = null;

        private int[]                                        dValues = null;

        private boolean                                        cancel = false;

        private GeoRasterWriter                 grw = null;

        private int                                                interpMethod = BufferInterpolation.INTERPOLATION_Undefined;

        /**

         * Variables de la resoluci?n de salida

         */

        private int                                                resolutionWidth = 0, resolutionHeight = 0;

        /**

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

         * @param cuttingPanelListener

         * @param dValues

         * @param fileName

         * @param writerBufferServer

         * @param rasterMultiDataset

         * @param extent

         * @param drawableBands

         * @param oneLayerPerBand

         */

        public CuttingProcess(CuttingPanelListener cuttingPanelListener, int[] dValues, String fileName, 

                                                        WriterBufferServer writerBufferServer, MultiRasterDataset rasterMultiDataset, 

                                                        Extent extent, int[] drawableBands, boolean oneLayerPerBand, int interpMethod) {

                this.fileName = fileName;

                this.writerBufferServer = writerBufferServer;

                this.rasterMultiDataset = rasterMultiDataset;

                this.extent = extent;

                this.oneLayerPerBand = oneLayerPerBand;

                this.drawableBands = drawableBands;

                this.cuttingPanelListener = cuttingPanelListener;

                this.dValues = dValues;

                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 {

                        BufferFactory bufferFactory = new BufferFactory(rasterMultiDataset);

                        bufferFactory.clearDrawableBand();

                        bufferFactory.addDrawableBands(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], dValues[2] - dValues[0], dValues[1] - dValues[3], resolutionWidth, resolutionHeight);

                                        buffer = bufferFactory.getRasterBuf();

                                } catch (InvalidSetViewException e) {

                                        e.printStackTrace();

                                }

                        }

                        if (oneLayerPerBand) {

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

                                        if(cancel)

                                                return;

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

                                        writerBufferServer.setBuffer(buffer, i);

                                        grw = GeoRasterWriter.getWriter(writerBufferServer, f,

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

                                                        buffer.getDataType(), GeoRasterWriter.getWriter(f).getParams());

                                        grw.dataWrite();

                                        grw.writeClose();

                                        if(!cancel)

                                                cuttingPanelListener.cutFinalize(f);

                                }

                        } else {

                                writerBufferServer.setBuffer(buffer, -1);

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

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

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

                                grw.dataWrite();

                                grw.writeClose();

                                if(!cancel)

                                        cuttingPanelListener.cutFinalize(fileName + ".tif");

                        }

                } catch (NotSupportedExtensionException e) {

                        e.printStackTrace();

                } catch (RasterDriverException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

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

        }

}