| gvSIG desktop 1

package org.gvsig.raster.grid.roi;

import junit.framework.TestCase;

import org.gvsig.raster.buffer.RasterBufferInvalidException;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridExtent;

import org.gvsig.raster.grid.OutOfGridException;

import com.iver.cit.gvsig.fmap.core.GeneralPathX;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.core.ShapeFactory;

/**

 * Test sobre la obtenci?n de estad?sticas de regiones de inter?s.

 * 

 *  @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public class TestRoiStatistics extends TestCase {

	Grid grid = null;

	GridExtent extent = null;

	int cellSize = 1;

	double xMin = 0;

	double xMax = 10;

	double yMin = 0;

	double yMax = 10;

	VectorialROI roi = null;

	GeneralPathX path = null;

	IGeometry geometry = null;

	public void setUp() {

		int bands[] = {0};

		extent = new GridExtent();

		extent.setCellSize(cellSize);

		extent.setXRange(xMin, xMax);

		extent.setYRange(yMin, yMax);

		try {

			grid = new Grid(extent, extent, IBuffer.TYPE_DOUBLE, bands);

			for (int x = 0; x<5; x++)

				for (int y = 0; y<5; y++){

					grid.setCellValue(x, y, 0.0);

				}

			for (int x = 5; x<xMax; x++)

				for (int y = 0; y<5; y++){

					grid.setCellValue(x, y, 1.0);

				}

			for (int x = 0; x<5; x++)

				for (int y = 5; y<10; y++){

					grid.setCellValue(x, y, 2.0);

				}

			for (int x = 5; x<9; x++)

				for (int y = 5; y<10; y++){

					grid.setCellValue(x, y, 3.0);

				}

		} catch (RasterBufferInvalidException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (OutOfGridException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}

	public void start() {

		this.setUp();

		this.testStack();

	}

	public void testStack() {

		path = new GeneralPathX();

		path.moveTo(4,4);

		path.lineTo(4,0);

		path.lineTo(0,0);

		path.lineTo(0,4);

		path.closePath();

		geometry=ShapeFactory.createPolygon2D(path);

		roi = new VectorialROI(grid);

		roi.addGeometry(geometry);

		assertEquals(2.0,roi.getMaxValue(),0.0);

		assertEquals(2.0,roi.getMeanValue(),0.0);

		assertEquals(2.0,roi.getMinValue(),0.0);

		path = new GeneralPathX();

		path.moveTo(9,4);

		path.lineTo(9,2);

		path.lineTo(6,2);

		path.lineTo(6,4);

		path.closePath();

		geometry=ShapeFactory.createPolygon2D(path);

		roi = new VectorialROI(grid);

		roi.addGeometry(geometry);

		assertEquals(3.0,roi.getMaxValue(),0.0);

		assertEquals(3.0,roi.getMeanValue(),0.0);

		assertEquals(3.0,roi.getMinValue(),0.0);

	}

}

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

 *

 * Copyright (C) 2006 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.raster.grid.roi;

import org.gvsig.raster.buffer.RasterBufferInvalidAccessException;

import org.gvsig.raster.dataset.IBuffer;

/**

 * Estad?sticas asociadas a una ROI.

 * 

 * Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public class ROIStatistic {

	private ROI roi = null;

	private double[] 			max = null;

	private double[] 			min = null;

	private double[] 			mean = null;

	private double[] 			variance = null;

	private boolean 			statisticsCalculated = false;

	protected int				bandToOperate = 0;

	public ROIStatistic(ROI roi) {

		this.roi = roi;

	}

	public void calculateStatistics() throws RasterBufferInvalidAccessException{

		int x, y;

		double z;

		int[] iValues = new int[roi.getGrid().getBandCount()];

		mean = new double[roi.getGrid().getBandCount()];

		min = new double[roi.getGrid().getBandCount()];

		max = new double[roi.getGrid().getBandCount()];

		variance = new double[roi.getGrid().getBandCount()];

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

			mean[iBand]	= 0.0;

			variance[iBand]	= 0.0;

			iValues[iBand]	= 0;

		}

		int oldBandToOperate = bandToOperate;

		statisticsCalculated = true;

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

			roi.getGrid().setBandToOperate(iBand);

			for (y = 0; y <= roi.getNY(); y++){ //TODO: Consultar con Nacho. Creo que NY y NX se calculan mal.

				for (x = 0; x <= roi.getNX(); x++){

					if(roi.isInGrid(x, y)){ //TODO: comprobarlo solo una vez, no para cada banda....

						z = getValue(x, y);

						if(!roi.isNoDataValue(z))	{

							if( iValues[iBand] == 0 )

								min[iBand] = max[iBand] = z;

							else if( min[iBand] > z )

								min[iBand]	= z;

							else if( max[iBand] < z )

								max[iBand]	= z;

							mean[iBand]	+= z;

							variance[iBand] += z * z;

							iValues[iBand]++;

						}

					}

				}

			}

			if( iValues[iBand] > 0 ){

				mean[iBand]	/= (double) iValues[iBand];

				variance[iBand] = variance[iBand] / (double) iValues[iBand] - mean[iBand] * mean[iBand];

			}

		}

		bandToOperate = oldBandToOperate;

		statisticsCalculated = true;

	}

	/**

	 * Obtiene un valor de una celda del grid independientemente del tipo de dato.

	 * @param x Posici?n en X

	 * @param y Posici?n en Y

	 * @return Valor de la celda en formato double

	 * @throws RasterBufferInvalidAccessException 

	 * @throws RasterBufferInvalidAccessException

	 */

	private double getValue(int x, int y) throws RasterBufferInvalidAccessException {

		switch(roi.getGrid().getDataType()) {

		case IBuffer.TYPE_BYTE:return (double)(roi.getCellValueAsByte(x, y)  & 0xff);

		case IBuffer.TYPE_SHORT:return (double)(roi.getCellValueAsShort(x, y)  & 0xffff);

		case IBuffer.TYPE_INT:return (double)(roi.getCellValueAsInt(x, y)  & 0xffffffff);

		case IBuffer.TYPE_FLOAT:return (double)roi.getCellValueAsFloat(x, y);

		case IBuffer.TYPE_DOUBLE:return (double)roi.getCellValueAsDouble(x, y);

		}

		return 0;

	}

	/**

	 * Obtiene el valor m?ximo del grid

	 * @return Valor m?ximo

	 */

	public double getMax() {

		return max[bandToOperate];

	}

	/**

	 * Obtiene el valor m?dio del grid

	 * @return Valor medio

	 */

	public double getMean() {

		return mean[bandToOperate];

	}

	/**

	 * Obtiene el valor m?ximo del grid

	 * @return Valor m?nimo

	 */

	public double getMin() {

		return min[bandToOperate];

	}

	/**

	 * Consulta si las estadisticas han sido calculadas o no.

	 * @return true si las estadisticas del grid est?n calculadas y false si no lo est?n 

	 */

	public boolean isStatisticsCalculated() {

		return statisticsCalculated;

	}

	/**

	 * Asigna el flag que informa si las estadisticas est?n calculadas para un grid concreto

	 * @param calc true si est?n calculadas y false si no lo est?n

	 */

	public void setStatisticsCalculated(boolean calc){

		this.statisticsCalculated = calc;

	}

	/**

	 * Obtiene la varianza

	 * @return Varianza

	 */

	public double getVariance() {

		return variance[bandToOperate];

	}

	/**

	 * Asigna la banda sobre la que se realizan las operaciones. Por defecto es la banda 0

	 * con lo que para el uso de MDTs no habr? que modificar este valor.

	 * @param band Banda sobre la que se realizan las operaciones.

	 */

	public void setBandToOperate(int band){

		this.bandToOperate = band;

	}

}

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

 *

 * Copyright (C) 2006 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.raster.grid.roi;

import java.util.ArrayList;

import org.gvsig.raster.grid.Grid;

import com.iver.cit.gvsig.fmap.core.IGeometry;

/**

 * Clase que representa una regi?n de interes conformada por

 * elementos vectoriales. 

 * 

 * @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public class VectorialROI extends ROI {

	private ArrayList				geometries   = null; // Array de geometr?as (IGeometry) que definen el ROI.

	public VectorialROI(Grid grid) {

		super(grid);

		geometries = new ArrayList();

	}

	public boolean isInGrid(int x, int y) {

		boolean inRoi = false;

		if (x>=0 && x<getNX() && y >=0 && y<getNY()){

			double mapX = this.getGridExtent().getMin().getX() + x*getCellSize();

			double mapY = this.getGridExtent().getMax().getY() - y*getCellSize();

			for (int i = 0; i < geometries.size(); i++)

				if (((IGeometry)geometries.get(i)).contains(mapX,mapY)){ //TODO:Ver como est? funcionando contains y si me vale.

					inRoi=true;

					break;

				}

		}

		return inRoi;

	}

	public void addGeometry(IGeometry geometry){

		geometries.add(geometry);

		double minX = Math.min(geometry.getBounds2D().getMinX(),getGridExtent().getMin().getX());

		double minY = Math.min(geometry.getBounds2D().getMinY(),getGridExtent().getMin().getY());

		double maxX = Math.max(geometry.getBounds2D().getMaxX(),getGridExtent().getMax().getX());

		double maxY = Math.max(geometry.getBounds2D().getMaxY(),getGridExtent().getMax().getY());

		getGridExtent().setXRange(minX, maxX);

		getGridExtent().setYRange(minY, maxY);

		xOffset = (int)((minX - getGrid().getGridExtent().getMin().getX())/getCellSize());

		yOffset = (int)((getGrid().getGridExtent().getMax().getY()- maxY)/getCellSize());

	}

	public void deleteGeometry(int index){

		geometries.remove(index);

		//TODO: Reajustar el Extent.

	}

}

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

 *

 * Copyright (C) 2006 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.raster.grid.roi;

import java.awt.Color;

import java.util.Arrays;

import org.gvsig.raster.buffer.RasterBufferInvalidAccessException;

import org.gvsig.raster.dataset.IBuffer;

import org.gvsig.raster.grid.Grid;

import org.gvsig.raster.grid.GridCell;

import org.gvsig.raster.grid.GridExtent;

import org.gvsig.raster.grid.IQueryableGrid;

/**

 * Clase que representa las Regiones de Interes. 

 * 

 * @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public abstract class ROI implements IQueryableGrid {

	private String 					name		   = "";

	private Color					color		   = null;

	private Grid 					grid 		   = null; // Grid asociado a la ROI y que le proporciona los datos.

	private ROIStatistic			statistic 	   = null;

	private GridExtent 				roiExtent      = null; // M?nimo extent que contiene al ROI.

	protected int 					xOffset		   = 0;	// Coord. x del grid donde est? la esquina sup. iz. del roi (coords. pixel)

	protected int					yOffset		   = 0; // Coord. y del grid donde est? la esquina sup. iz. del roi (coords. pixel)

	public ROI(Grid grid) {

		this.grid = grid;

		roiExtent = new GridExtent();

		roiExtent.setCellSize(grid.getCellSize());

		statistic = new ROIStatistic(this);

	}

	public ROI add(ROI roi){

		return null;

	}

	public ROI intersect(ROI roi){

		return null;

	}

	public ROI subtract(ROI roi){

		return null;

	}

	public ROI exclusiveOr(ROI roi){

		return null;

	}

	public double getAspect(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getAspect(x, y);

		else return getNoDataValue();

	}

	public double getCellSize() {

		return grid.getCellSize();

	}

	public byte getCellValueAsByte(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getCellValueAsByte(xOffset+x, yOffset+y);

		else 

			return (byte) grid.getNoDataValue();

	}

	public double getCellValueAsDouble(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getCellValueAsDouble(xOffset+x, yOffset+y);

		else 

			return grid.getNoDataValue();

	}

	public float getCellValueAsFloat(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getCellValueAsFloat(xOffset+x, yOffset+y);

		else 

			return (float) grid.getNoDataValue();

	}

	public int getCellValueAsInt(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getCellValueAsInt(xOffset+x, yOffset+y);

		else 

			return (int) grid.getNoDataValue();

	}

	public short getCellValueAsShort(int x, int y)

			throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getCellValueAsShort(xOffset+x, yOffset+y);

		else 

			return (short) grid.getNoDataValue();

	}

	public int getDirToNextDownslopeCell(int x, int y) throws RasterBufferInvalidAccessException {

		return getDirToNextDownslopeCell(x, y, true);

	}

	public int getDirToNextDownslopeCell(int x, int y,boolean bForceDirToNoDataCell)

													throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getDirToNextDownslopeCell(x, y, bForceDirToNoDataCell);

		else

			return -1;

	}

	public double getDistToNeighborInDir(int iDir) {

		return grid.getDistToNeighborInDir(iDir);

	}

	public GridExtent getGridExtent() {

		return roiExtent;

	}

	public int getLayerNX() {

		return grid.getLayerNX();

	}

	public int getLayerNY() {

		return grid.getLayerNY();

	}

	public double getMaxValue() {

		if (!statistic.isStatisticsCalculated())

			try {

				statistic.calculateStatistics();

			} catch (RasterBufferInvalidAccessException e) {

				//No se calculan las estad?sticas. No hacemos nada

				e.printStackTrace();

			}

		return statistic.getMax();

	}

	public double getMeanValue() {

		if (!statistic.isStatisticsCalculated())

			try {

				statistic.calculateStatistics();

			} catch (RasterBufferInvalidAccessException e) {

				//No se calculan las estad?sticas. No hacemos nada

				e.printStackTrace();

			}

		return statistic.getMean();

	}

	public double getMinValue() {

		if (!statistic.isStatisticsCalculated())

			try {

				statistic.calculateStatistics();

			} catch (RasterBufferInvalidAccessException e) {

				//No se calculan las estad?sticas. No hacemos nada

				e.printStackTrace();

			}

		return statistic.getMin();

	}

	public int getNX() {

		return roiExtent.getNX();

	}

	public int getNY() {

		return roiExtent.getNY();

	}

	public double getNoDataValue() {

		//		 TODO Ver si usar un nodatavalue especifico del roi.

		return grid.getNoDataValue();

	}

	public double getSlope(int x, int y) throws RasterBufferInvalidAccessException {

		if (isInGrid(x, y))

			return grid.getSlope(x, y);

		else 

			return getNoDataValue();

	}

	public GridCell[] getSortedArrayOfCells() throws RasterBufferInvalidAccessException {

		int i;

		int iX,iY;

		int iNX =  getNX();

		int iCells = getNX() * getNY();

		GridCell [] cells = null;

		GridCell cell = null;

		cells = new GridCell[iCells];

		for (i = 0; i < iCells; i++){

			iX = i % iNX;

			iY = i / iNX;

			switch(grid.getDataType()){

			case IBuffer.TYPE_BYTE: cell = new GridCell(iX, iY, getCellValueAsByte(iX, iY)); break;

			case IBuffer.TYPE_SHORT: cell = new GridCell(iX, iY, getCellValueAsShort(iX, iY)); break;

			case IBuffer.TYPE_INT: cell = new GridCell(iX, iY, getCellValueAsInt(iX, iY)); break;

			case IBuffer.TYPE_FLOAT: cell = new GridCell(iX, iY, getCellValueAsFloat(iX, iY)); break;

			case IBuffer.TYPE_DOUBLE: cell = new GridCell(iX, iY, getCellValueAsDouble(iX, iY)); break;

			}

			cells[i] = cell;

		}

		Arrays.sort(cells);

		return cells;

	}

	public double getVariance() {

		if (!statistic.isStatisticsCalculated())

			try {

				statistic.calculateStatistics();

			} catch (RasterBufferInvalidAccessException e) {

				//No se calculan las estad?sticas. No hacemos nada

				e.printStackTrace();

			}

		return statistic.getVariance();

	}

	public abstract boolean isInGrid(int x, int y);

	public boolean isNoDataValue(double dValue) {

		return grid.isNoDataValue(dValue);

	}

	public void setInterpolationMethod(int iMethod) {

		grid.setInterpolationMethod(iMethod);

	}

	public Grid getGrid() {

		return grid;

	}

	public Color getColor() {

		return color;

	}

	public void setColor(Color color) {

		this.color = color;

	}

	public String getName() {

		return name;

	}

	public void setName(String name) {

		this.name = name;

	}

}

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

 *

 * Copyright (C) 2006 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.raster.grid.roi;

import org.gvsig.raster.grid.Grid;

/**

 * Clase que representa una regi?n de interes conformada por

 * un raster. 

 * 

 * @author Diego Guerrero Sevilla (diego.guerrero@uclm.es)

 *

 */

public class RasterROI extends ROI {

	public RasterROI(Grid grid) {

		super(grid);

	}

	public boolean isInGrid(int x, int y) {

		// TODO Auto-generated method stub

		return false;

	}

}