| gvSIG desktop 1

/*

 * Created on 17-abr-2007

 *

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

 *

 * Copyright (C) 2004 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.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

*

* $Id$

* $Log$

* Revision 1.1  2007-04-19 18:12:56  azabala

* *** empty log message ***

*

*

*/

package com.iver.cit.gvsig.fmap.featureiterators;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.net.URL;

import junit.framework.TestCase;

import org.cresques.cts.IProjection;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;

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

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

import com.iver.cit.gvsig.fmap.crs.CRSFactory;

import com.iver.cit.gvsig.fmap.drivers.IFeatureIterator;

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

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

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

public class FeatureIteratorTest extends TestCase {

	static final String fwAndamiDriverPath = "../_fwAndami/gvSIG/extensiones/com.iver.cit.gvsig/drivers";

	private File baseDataPath;

	private File baseDriversPath;

	private String SHP_DRIVER_NAME = "gvSIG shp driver";

	private String DXF_DRIVER_NAME = "gvSIG DXF Memory Driver";

	private IProjection PROJECTION_DEFAULT = 

		CRSFactory.getCRS("EPSG:23030");

	private IProjection newProjection = 

		CRSFactory.getCRS("EPSG:23029");

	protected void setUp() throws Exception {

		super.setUp();

		URL url = this.getClass().getResource("testdata");

		if (url == null)

			throw new Exception("No se encuentra el directorio con datos de prueba");

		baseDataPath = new File(url.getFile());

		if (!baseDataPath.exists())

			throw new Exception("No se encuentra el directorio con datos de prueba");

		baseDriversPath = new File(fwAndamiDriverPath);

		if (!baseDriversPath.exists())

			throw new Exception("Can't find drivers path: " + fwAndamiDriverPath);

		LayerFactory.setDriversPath(baseDriversPath.getAbsolutePath());

		if (LayerFactory.getDM().getDriverNames().length < 1)

			throw new Exception("Can't find drivers in path: " + fwAndamiDriverPath);

	}

	private FLayer newLayer(String fileName, 

									   String driverName) 

								throws LoadLayerException {

		File file = new File(baseDataPath, fileName);

		return LayerFactory.createLayer(fileName, 

										driverName, 

										file, PROJECTION_DEFAULT);

	}

	public void test1() {

		try {

			//pruebas de reproyeccion y seleccion de numero de campos

			FLyrVect lyr = (FLyrVect) newLayer("Cantabria.shp", SHP_DRIVER_NAME);

			lyr.setAvailable(true);

			IFeatureIterator iterator = lyr.getSource().getFeatureIterator(new String[]{"ID", "AREA"}, null);

			IFeature feature = iterator.next();

			assertTrue(feature != null);

			iterator = lyr.getSource().getFeatureIterator((String[])null, newProjection);

			IFeature feature2 = iterator.next();

			//test of field restriction

			assertTrue(feature.getAttributes().length == 2);

			assertTrue(feature2.getAttributes().length > 2);

			assertTrue(!feature.getGeometry().toJTSGeometry().equals(feature2.getGeometry().toJTSGeometry()));

			//pruebas de iteracion espacial

			FLyrVect lyr2 = (FLyrVect) newLayer("Edificaciones.shp", SHP_DRIVER_NAME);

			lyr2.setAvailable(true);

			double xmin = 410000;

			double xmax = 415000;

			double ymin = 4790000;

			double ymax = 4793000;

			Rectangle2D rect = new Rectangle2D.Double(xmin, ymin, (xmax-xmin), (ymax-ymin));

			iterator = lyr2.getSource().getFeatureIterator(rect, null, newProjection);

			int numFeatures = 0;

			while(iterator.hasNext()){

				feature = iterator.next();

				numFeatures++;

			}

			int numFeatures2 = 0;

			iterator = lyr2.getSource().getFeatureIterator();

			while(iterator.hasNext()){

				feature = iterator.next();

				numFeatures2++;

			}

			assertTrue(numFeatures2 > numFeatures);

			//pruebas de iteracion en base a criterios alfanumericos

			final String expr = "select * from " + lyr.getRecordset().getName() + " where area < 700;";

			iterator = lyr.getSource().getFeatureIterator(expr, null);

			numFeatures2 = 0;

			while(iterator.hasNext()){

				feature = iterator.next();

				numFeatures2++;

			}

			iterator = lyr.getSource().getFeatureIterator();

			while(iterator.hasNext()){

				feature = iterator.next();

				numFeatures++;

			}

			assertTrue(numFeatures2 > 0);

			assertTrue(numFeatures > numFeatures2);

			//pruebas con el driver dxf en vez del shp

			FLyrVect lyr3 = (FLyrVect) newLayer("es1120003_2_02.dxf", DXF_DRIVER_NAME);

			lyr3.setAvailable(true);

			iterator = lyr3.getSource().getFeatureIterator("select * from "+lyr3.getRecordset().getName()+" where ID > 17;", null);

			numFeatures2 = 0;

			while(iterator.hasNext()){

				feature = iterator.next();

				numFeatures2++;

			}

			assertTrue(numFeatures2 > 0);

			//queda por probar los iteradores contra base de datos

			//en base a criterios alfanumericos y espaciales 

		} catch (LoadLayerException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (ReadDriverException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (ExpansionFileReadException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}	

}