Application: gvSIG desktop » gvSIG raster

org.gvsig.raster.tools.algorithm.reproject.RasterReprojectAlgorithmLibrary

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

/**

 * This exception is thrown if happen problems processing data with this algorithm.

 * 

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

 *

 */

public class ReprojectException extends Exception {

	private static final long serialVersionUID = -3022090543908771484L;

	public ReprojectException(String msg){

		super(msg);

	}

	public ReprojectException(String msg, Throwable e){

		super(msg, e);

	}

}

/* 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.util.HashMap;

import javax.swing.SwingUtilities;

import org.cresques.cts.IProjection;

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

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

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

import org.gvsig.i18n.Messages;

import org.gvsig.raster.tools.algorithm.base.RasterBaseAlgorithmLibrary;

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

/**

 * Process to reproject layers. The size of the new image and the pixel size can

 * be passed by parameter. If some of these values are zero everyone are calculated

 * automatically

 *

 * 10/12/2007

 * @author Nacho Brodin nachobrodin@gmail.com

 */

public class ReprojectProcess extends RasterProcess {

	public static String[]    INTERP_METHODS  = new String[]{"Nearest", "Bilinear", "InverseDistance"};

	public static String      RASTER_STORE    = "RasterStore";

	public static String      PATH            = "Path";

	public static String      DST_PROJECTION  = "DST_Projection";

	public static String      SRC_PROJECTION  = "SRC_Projection";

	public static String      SIZEX           = "SizeX";

	public static String      SIZEY           = "SizeY";

	public static String      CELLSIZE        = "CellSize";

	public static String      FILENAME        = "FileName";

	public static String      TIME            = "Time";

	public static String      INTERPOLATION   = "Interpolation";

	private RasterDataStore   store          = null;

	private String            filename       = null;

	private IProjection       projdst        = null;

	private IProjection       projsrc        = null;

	private Reproject         reproject      = null;

	private long              milis          = 0;

	private int               w              = 0;

	private int               h              = 0;

	private double            cellSize       = 0;

	private int               interpolation  = 0;

	public static void registerParameters() {

		RASTER_STORE = RasterBaseAlgorithmLibrary.registerInputParameter(RASTER_STORE, RasterDataStore.class);

		PATH = RasterBaseAlgorithmLibrary.registerInputParameter(PATH, String.class);

		DST_PROJECTION = RasterBaseAlgorithmLibrary.registerInputParameter(DST_PROJECTION, IProjection.class);

		SRC_PROJECTION = RasterBaseAlgorithmLibrary.registerInputParameter(SRC_PROJECTION, IProjection.class);

		SIZEX = RasterBaseAlgorithmLibrary.registerInputParameter(SIZEX, Integer.class);

		SIZEY = RasterBaseAlgorithmLibrary.registerInputParameter(SIZEY, Integer.class);

		CELLSIZE = RasterBaseAlgorithmLibrary.registerInputParameter(CELLSIZE, Double.class);

		INTERPOLATION = RasterBaseAlgorithmLibrary.registerInputParameter(INTERPOLATION, Integer.class); 

		FILENAME = RasterBaseAlgorithmLibrary.registerOutputParameter(FILENAME, String.class);

		TIME = RasterBaseAlgorithmLibrary.registerOutputParameter(TIME, Long.class);

	}

	/*

	 * (non-Javadoc)

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

	 */

	public void init() {

		store = getParam(RASTER_STORE) != null ? (RasterDataStore)getParam(RASTER_STORE) : null;

		filename = getStringParam(PATH);

		projdst = getParam(DST_PROJECTION) != null ? (IProjection) getParam(DST_PROJECTION) : null;

		projsrc = getParam(SRC_PROJECTION) != null ? (IProjection) getParam(SRC_PROJECTION) : null; 

		w = getIntParam(SIZEX);

		h = getIntParam(SIZEY);

		cellSize = getDoubleParam(CELLSIZE);

		interpolation = getIntParam(INTERPOLATION);

	}

	/**

	 * M?todo donde se ejecutar? el Thread, aqu? se reproyecta el raster.

	 */

	public void process() throws ProcessInterruptedException {

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

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

		try {

			store = store.cloneDataStore();

		} catch (CloneException e1) {

			messageBoxError("error_reprojecting", this, e1);

		}

		reproject = new Reproject(store, filename, interpolation, this);

		try {

			int result = reproject.warp(projdst, projsrc, w, h, cellSize);

			if(result != 0) {

				if (incrementableTask != null) {

					incrementableTask.processFinalize();

					setProgressActive(false);

				}

				messageBoxError("transformation_not_possible", this);

				return;

			}

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

			milis = t2 - t1;

			SwingUtilities.invokeLater(new Runnable() {

				public void run() {

					if (externalActions != null) {

						HashMap<String, Object> map = new HashMap<String, Object>();

						map.put(FILENAME, filename);

						map.put(TIME, new Long(milis));

						externalActions.end(map);

					}

				}

			});

		} catch (ReprojectException e) {

			if (incrementableTask != null)

				incrementableTask.processFinalize();

			messageBoxError("error_reprojecting", this, e);

		}

	}

	/*

	 * (non-Javadoc)

	 * @see org.gvsig.raster.tools.app.basic.raster.process.RasterProcess#getResult()

	 */

	public Object getResult() {

		HashMap<String, Object> map = new HashMap<String, Object>();

		map.put(FILENAME, filename);

		map.put(TIME, new Long(milis));

		return map;

	}

	/*

	 * (non-Javadoc)

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

	 */

	public int getPercent() {

		return percent;

	}

	/*

	 * (non-Javadoc)

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

	 */

	public String getTitle() {

		return Messages.getText("reprojecting");

	}

}

/* 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.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.dataset.Buffer;

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

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

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

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

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

				}

			}

			process.exportRaster(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};

	}

}

/*

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2012 gvSIG Association.

 *

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

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.tools.algorithm.reproject;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.tools.algorithm.base.RasterBaseAlgorithmLibrary;

import org.gvsig.tools.library.AbstractLibrary;

import org.gvsig.tools.library.LibraryException;

/**

 * Initialization of RasterReprojectAlgorithmLibrary library.

 * 

 * @author <a href="mailto:nachobrodin@gmail.com">Nacho Brodin</a>

 */

public class RasterReprojectAlgorithmLibrary extends AbstractLibrary {

	public static final String         PROCESS_LABEL   = "RasterReprojectionProcess";

    @Override

    protected void doInitialize() throws LibraryException {

        // Nothing to do

    }

    @Override

    protected void doPostInitialize() throws LibraryException {

    	//Registers the process and its parameters

    	RasterBaseAlgorithmLibrary.register(PROCESS_LABEL, ReprojectProcess.class);

    	ReprojectProcess.registerParameters();

        Messages.addResourceFamily(

            "org.gvsig.raster.tools.algorithm.reproject", 

            RasterReprojectAlgorithmLibrary.class.getClassLoader(), 

            RasterReprojectAlgorithmLibrary.class.getClass().getName());

        //registerGeoProcess(new RasterReprojectAlgorithmLibrary());

    }

}

<?xml version="1.0" encoding="ISO-8859-1"?>

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">

	<modelVersion>4.0.0</modelVersion>

	<artifactId>org.gvsig.raster.tools.algorithm.reproject</artifactId>

	<packaging>jar</packaging>

	<name>org.gvsig.raster.tools.algorithm.reproject</name>

	<parent>

		<groupId>org.gvsig</groupId>

		<artifactId>org.gvsig.raster.tools.algorithm</artifactId>

		<version>2.0.1-SNAPSHOT</version>

	</parent>

    <properties>

        <build-dir>${basedir}/../../../build</build-dir>

    </properties>

    <dependencies>

		<dependency>

			<groupId>org.gvsig</groupId>

			<artifactId>org.gvsig.raster.tools.algorithm.base</artifactId>

            <scope>compile</scope>

		</dependency>

	</dependencies>

</project>