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

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.io.File;

import org.cresques.cts.ICoordTrans;

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

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

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

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

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

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

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

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

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

import org.gvsig.raster.tools.algorithm.base.process.RasterProcess;

import org.gvsig.raster.tools.algorithm.base.util.Interpolation;

/**

 * Reprojects a RasterDataStore. 

 *

 * @version 30/04/2008

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class Reproject {

        private RasterDataStore   store                 = null;

        private String            pathDest              = null;

        private int               percent               = 0;

        private int               interpolationMethod   = 0;

        private Interpolation     interpolation         = null; 

        private RasterProcess     process               = null;

        /**

         * Constructor de la clase.

         * @param lyr

         * @param pathDest Ruta de destino

         */

        public Reproject(RasterDataStore store, String pathDest, int inter, RasterProcess process) {

                this.store = store;

                this.pathDest = pathDest;

                this.interpolationMethod = inter;

                this.process = process;

        }

        /**

         * M?todo para la transformaci?n del raster.

         * @param destinationSrs Proyecci?n destino

         */

        public int warp(IProjection destinationSrs, IProjection sourceSrs, int w, int h, double cellSize) throws ReprojectException {

                if (store == null)

                        throw new ReprojectException("Capa no valida.");

                if (!store.isReproyectable())

                        throw new ReprojectException("Esta capa no se puede reproyectar.");

                if (destinationSrs == null || destinationSrs.getAbrev() == null)

                        throw new ReprojectException("Proyecci?n de destino no valida.");

                File file = new File(pathDest);

                if (!file.getParentFile().canWrite())

                        throw new ReprojectException("Ruta de destino no valida.");

                ICoordTrans transf = sourceSrs.getCT(destinationSrs);

                Extent bbox = store.getExtent();

                Point2D ul = new Point2D.Double(bbox.getULX(), bbox.getULY());

                Point2D lr = new Point2D.Double(bbox.getLRX(), bbox.getLRY());

                //Point2D p = new Point2D.Double(lyr.getFullRasterExtent().getULX(), lyr.getFullRasterExtent().getULY());

                ul = transf.convert(ul, ul);

                lr = transf.convert(lr, lr);

                Extent newBbox = RasterLocator.getManager().getDataStructFactory().createExtent(ul, lr);

                if(w <= 0 || h <= 0 ) {

                        double[] size = getSize(bbox, newBbox);

                        w = (int)size[0];

                        h = (int)size[1];

                        cellSize = size[2];

                }

                int dataType = store.getDataType()[0];

                Buffer buf = RasterLocator.getManager().createBuffer(

                                store.getDataType()[0], w, h, store.getBandCount(), true);

                NoData nd = RasterLocator.getManager().getDataStructFactory().createDefaultNoData(

                                store.getBandCount(), dataType);

                RasterQuery query = RasterLocator.getManager().createQuery();

                query.setAllDrawableBands();

                query.setAreaOfInterest();

                query.setReadOnly(true);

                try {

                        ICoordTrans t = destinationSrs.getCT(sourceSrs);

                        Buffer sourceBuffer = store.query(query);

                        if(interpolationMethod < 0) {

                                for (int row = 0; row < buf.getHeight(); row++) {

                                        for (int col = 0; col < buf.getWidth(); col++) {

                                                Point2D p = transformPoint(newBbox, col, row, cellSize, t);

                                                writePixel(dataType, sourceBuffer, buf, p, col, row, nd);

                                        }

                                        process.updatePercent(row, buf.getHeight());

                                }

                        } else {

                                interpolation = new Interpolation(sourceBuffer);

                                for (int row = 0; row < buf.getHeight(); row++) {

                                        for (int col = 0; col < buf.getWidth(); col++) {

                                                Point2D p = transformPoint(newBbox, col, row, cellSize, t);

                                                writePixelInterpolated(dataType, sourceBuffer, buf, p, col, row, nd, interpolationMethod);

                                        }

                                        process.updatePercent(row, buf.getHeight());

                                }

                        }

                        export(pathDest, buf, cellSize, newBbox.getULX(), newBbox.getULY());

                } catch (RasterDriverException e) {

                        new ReprojectException("", e);

                } catch (ProcessInterruptedException e) {

                } catch (InvalidSetViewException e) {

                        new ReprojectException("", e);

                }

                return 0;

        }

        /**

         * Writes one pixel in the destination buffer

         * @param type

         * @param sourceBuffer

         * @param buf

         * @param p

         * @param col

         * @param row

         */

        private void writePixel(int type, Buffer sourceBuffer, Buffer buf, Point2D p, int col, int row, NoData nd) {

                if(type == Buffer.TYPE_BYTE) {

                        if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++) {

                                        //if(iBand == 0)

                                                //System.out.println("Row:" + row + " Col:" + col + "  Y:" + (int)p.getY() + " X:" + (int)p.getX());

                                        buf.setElem(row, col, iBand, sourceBuffer.getElemByte((int)p.getY(), (int)p.getX(), iBand));

                                }

                        else

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, nd.getValue().byteValue());

                } else if(type == Buffer.TYPE_DOUBLE) {

                        if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, sourceBuffer.getElemDouble((int)p.getY(), (int)p.getX(), iBand));

                        else

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, nd.getValue().doubleValue());

                } else if(type == Buffer.TYPE_FLOAT) {

                        if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, sourceBuffer.getElemFloat((int)p.getY(), (int)p.getX(), iBand));

                        else

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, nd.getValue().floatValue());

                } else if(type == Buffer.TYPE_SHORT) {

                        if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, sourceBuffer.getElemShort((int)p.getY(), (int)p.getX(), iBand));

                        else

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, nd.getValue().shortValue());

                } else if(type == Buffer.TYPE_INT) {

                        if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, sourceBuffer.getElemInt((int)p.getY(), (int)p.getX(), iBand));

                        else

                                for (int iBand = 0; iBand < store.getBandCount(); iBand++)

                                        buf.setElem(row, col, iBand, nd.getValue().intValue());

                } 

        }

        /**

         * Writes one pixel in the destination buffer

         * @param type

         * @param sourceBuffer

         * @param buf

         * @param p

         * @param col

         * @param row

         */

        private void writePixelInterpolated(int type, Buffer sourceBuffer, Buffer buf, Point2D p, int col, int row, NoData nd, int interpMethod) {

                double value = 0;

                if(p.getX() > 0 && p.getX() < sourceBuffer.getWidth() && p.getY() > 0 && p.getY() < sourceBuffer.getHeight())

                        for (int iBand = 0; iBand < store.getBandCount(); iBand++) {

                                if(interpMethod == 0) //Nearest neighbor

                                        value = interpolation.getNearestNeighbour(p.getX(), p.getY(), iBand);

                                if(interpMethod == 1) //Bilinear

                                        value = interpolation.getBilinearValue(p.getX(), p.getY(), iBand);

                                if(interpMethod == 2) //Inverse distance

                                        value = interpolation.getInverseDistance(p.getX(), p.getY(), iBand);

                                if(type == Buffer.TYPE_BYTE)

                                        buf.setElem(row, col, iBand, (byte)value);

                                else if(type == Buffer.TYPE_DOUBLE)

                                        buf.setElem(row, col, iBand, (double)value);

                                else if(type == Buffer.TYPE_FLOAT)

                                        buf.setElem(row, col, iBand, (float)value);

                                else if(type == Buffer.TYPE_SHORT)

                                        buf.setElem(row, col, iBand, (short)value);

                                else if(type == Buffer.TYPE_INT)

                                        buf.setElem(row, col, iBand, (int)value);

                        }

                else

                        for (int iBand = 0; iBand < store.getBandCount(); iBand++) {

                                if(type == Buffer.TYPE_BYTE)

                                        buf.setElem(row, col, iBand, nd.getValue().byteValue());

                                else if(type == Buffer.TYPE_DOUBLE)

                                        buf.setElem(row, col, iBand, nd.getValue().doubleValue());

                                else if(type == Buffer.TYPE_FLOAT)

                                        buf.setElem(row, col, iBand, nd.getValue().floatValue());

                                else if(type == Buffer.TYPE_SHORT)

                                        buf.setElem(row, col, iBand, nd.getValue().shortValue());

                                else if(type == Buffer.TYPE_INT)

                                        buf.setElem(row, col, iBand, nd.getValue().intValue());

                        }

        }

        /**

         * Transforms the upper left coordinate of a pixel using the transformation 

         * which is passed by parameter

         * @param p

         * @param newBbox

         * @param col

         * @param row

         * @param cellSize

         * @param t

         */

        private Point2D transformPoint(Extent newBbox, int col, int row, double cellSize, ICoordTrans t) {

                Point2D p = new Point2D.Double(

                                newBbox.getULX() + (col * cellSize), 

                                newBbox.getULY() - (row * cellSize));

                p = t.convert(p, p);

                p = store.worldToRaster(p);

                return p;

        }

        /**

         * Gets the size of the new image

         * @param bbox

         * @param newBbox

         * @return

         */

        private double[] getSize(Extent bbox, Extent newBbox) {

                double sumSideOldBBox = bbox.width() + bbox.height();

                double sumSideNewBBox = newBbox.width() + newBbox.height();

                double d1x = (bbox.width() * 100) / sumSideOldBBox; 

                double d1y = (bbox.height() * 100) / sumSideOldBBox;

                double d2x = (newBbox.width() * 100) / sumSideNewBBox;

                double d2y = (newBbox.height() * 100) / sumSideNewBBox;

                double p2y = (store.getHeight() * d2y) / d1y;

                double p2x = (store.getWidth() * d2x) / d1x;

                double newCellSize = newBbox.width() / p2x;

                return new double[]{Math.round(p2x), Math.round(p2y), newCellSize};

        }

        public boolean export(final String sFilename, Buffer buf, double cellsize, double minX, double minY) {

        try {

            RasterManager manager = RasterLocator.getManager();

            final DataServerWriter writerBufferServer =

                manager.createDataServerWriter();

            writerBufferServer.setBuffer(buf, -1);

            final Params params = manager.createWriterParams(sFilename);

            final AffineTransform affineTransform =

                new AffineTransform(cellsize, 0, 0,

                    -cellsize, minX, minY);

            final RasterWriter writer =

                manager.createWriter(writerBufferServer, sFilename,

                    buf.getBandCount(), affineTransform, buf.getWidth(),

                    buf.getHeight(), buf.getDataType(), params, null);

            writer.dataWrite();

            writer.writeClose();

        } catch (final Exception e) {

                e.printStackTrace();

            return false;

        }

        return true;

    }

        /**

         * Obtiene el porcentaje de progreso de la tarea de reproyecci?n para

         * mostrarlo por pantalla.

         * @return

         */

        public int getPercent() {

                return percent;

        }

}