| gvSIG desktop 1

/**

 * 

 */

package org.gvsig.data.datastores.vectorial;

import java.io.File;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Set;

import org.gvsig.data.CloseException;

import org.gvsig.data.DataManager;

import org.gvsig.data.DataStoreParameters;

import org.gvsig.data.InitializeException;

import org.gvsig.data.OpenException;

import org.gvsig.data.ReadException;

import org.gvsig.data.datastores.vectorial.file.DataStoreTest;

import org.gvsig.data.datastores.vectorial.file.shp.Register;

import org.gvsig.data.datastores.vectorial.file.shp.SHPStore;

import org.gvsig.data.datastores.vectorial.file.shp.SHPStoreParameters;

import org.gvsig.data.datastores.vectorial.file.shp.SHPTest;

import org.gvsig.data.datastores.vectorial.file.shp.ShpFeatureID;

import org.gvsig.data.vectorial.Feature;

import org.gvsig.data.vectorial.FeatureAttributeDescriptor;

import org.gvsig.data.vectorial.FeatureCollection;

import org.gvsig.data.vectorial.FeatureType;

import org.gvsig.data.vectorial.filter.FilterFactory2Impl;

import org.opengis.filter.Filter;

import org.opengis.filter.FilterFactory2;

import org.opengis.filter.expression.Expression;

import org.opengis.filter.sort.SortBy;

import org.opengis.filter.sort.SortOrder;

import junit.framework.TestCase;

/**

 * @author jcarras

 *

 */

public class FilterTest  extends TestCase {

	private File fileNormal = new File(DataStoreTest.class.getResource("data/pruebaFiltros.shp").getFile());

	SHPStore store;

	public static void main(String[] args) {

		junit.textui.TestRunner.run(FilterTest.class);

	}

	protected void setUp() throws Exception {

		super.setUp();

	}

	public void initialize() {

		Register.selfRegister();

		DataManager manager = DataManager.getManager();

		SHPStoreParameters shpParameters=null;

		shpParameters=(SHPStoreParameters)manager.createDataStoreParameters(SHPStore.DATASTORE_NAME);

		shpParameters.setFile(fileNormal);

//		shpParameters.setDBFFile(SHP.getDbfFile(file));

//		shpParameters.setSHXFile(SHP.getShxFile(file));

		try {

			store= (SHPStore)manager.createDataStore((DataStoreParameters)shpParameters);

		} catch (InitializeException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}

	public void testOrder(){

		System.out.println("testOrder");

		initialize();

		SortBy[] order = new SortBy[2];

		order[0]=new FilterFactory2Impl().sort("TIPO", SortOrder.ASCENDING);

		order[1]=new FilterFactory2Impl().sort("AREA", SortOrder.ASCENDING);

		assertEquals(9,testshp(null, order));

	}

	public void testGT(){

		System.out.println("testGT");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(30000000);

		Filter filter = fact.greater(exp1, exp2);

		assertEquals(4,testshp(filter, null));

	}

	public void testGT_string(){

		System.out.println("testGT_string");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.greater(exp1, exp2);

		assertEquals(2,testshp(filter, null));

	}

	public void testGTE(){

		System.out.println("testGTE");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.greaterOrEqual(exp1, exp2);

		assertEquals(3,testshp(filter, null));

	}

	public void testLT(){

		System.out.println("testLT");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(30000000);

		Filter filter = fact.less(exp1, exp2);

		assertEquals(5,testshp(filter, null));

	}

	public void testLTE(){

		System.out.println("testLTE");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.lessOrEqual(exp1, exp2);

		assertEquals(7,testshp(filter, null));

	}	

	public void testEquals(){

		System.out.println("testEquals");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("INE");

		Expression exp2 = fact.literal(3125);

		Filter filter = fact.equals(exp1, exp2);

		assertEquals(1,testshp(filter, null));

	}	

	public void testEquals_String(){

		System.out.println("testEquals");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.equals(exp1, exp2);

		assertEquals(1,testshp(filter, null));

	}	

	public void testNotEquals(){

		System.out.println("testNotEquals");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.notEqual(exp1, exp2,true);

		assertEquals(8,testshp(filter, null));

	}	

	public void testNot(){

		System.out.println("testNot");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Expression exp2 = fact.literal("PEDREGUER");

		Filter filter = fact.equals(exp1, exp2);

		filter = fact.not(filter);

		assertEquals(8,testshp(filter, null));

	}	

	/**

	 * Current shp driver fails when using a shapefile 

	 * with null values 

	 */

	public void testIsNull(){

		System.out.println("testIsNull");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("OTRO_IDIOM");

		Filter filter = fact.not(fact.isNull(exp1));

		assertEquals(7,testshp(filter, null));

	}	

	public void testBetween(){

		System.out.println("Between");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.literal(25000000.0);

		Expression exp3 = fact.literal(40000000);

		Expression exp2 = fact.property("AREA");

		Filter filter = fact.between(exp1, exp2, exp3);

		assertEquals(2,testshp(filter, null));

	}	

	public void testLike(){

		System.out.println("testLike");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("NOMBRE");

		Filter filter = fact.like(exp1, "%I_A%");

		assertEquals(2,testshp(filter, null));

	}	

	public void testFId(){

		System.out.println("testId");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		HashSet ids = new HashSet();

		ids.add(new ShpFeatureID(store,3));

		ids.add(new ShpFeatureID(store,6));

		Filter filter = fact.id(ids);

		assertEquals(2,testshp(filter, null));

	}

	public void testBBOX(){

		System.out.println("testBBOX");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("GEOMETRY");

		Filter filter = fact.bbox(exp1, 769009, 4291393, 772960, 4293701, null);

		assertEquals(4,testshp(filter, null));

	}

	public void testAnd(){

		System.out.println("testAnd");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("GEOMETRY");

		Filter filter1 = fact.bbox(exp1, 769009, 4291393, 772960, 4293701, null);

		Expression exp2 = fact.property("NOMBRE");

		Filter filter2 = fact.like(exp2, "B%");

		Filter filter = fact.and(filter1, filter2);

		assertEquals(1,testshp(filter, null));

	}

	public void testOr(){

		System.out.println("testOr");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("GEOMETRY");

		Filter filter1 = fact.bbox(exp1, 769009, 4291393, 772960, 4293701, null);

		Expression exp2 = fact.property("NOMBRE");

		Filter filter2 = fact.like(exp2, "B%");

		Filter filter = fact.or(filter1, filter2);

		assertEquals(5,testshp(filter, null));

	}

	/**

	 * area + 3e7 > 6e7 --> area > 3e7

	 */

	public void testAdd(){

		System.out.println("testAdd");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(30000000);

		Expression exp3 = fact.literal(60000000);

		Filter filter = fact.greater(fact.add(exp1, exp2),exp3);

		assertEquals(4,testshp(filter, null));

	}

	/**

	 * area - 3e7 > 0 --> area > 3e7

	 */

	public void testSubstract(){

		System.out.println("testSubstract");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(30000000);

		Expression exp3 = fact.literal(0);

		Filter filter = fact.greater(fact.subtract(exp1, exp2),exp3);

		assertEquals(4,testshp(filter, null));

	}

	/**

	 * area * 2 > 6e7 --> area > 3e7

	 */

	public void testMultiply(){

		System.out.println("testMultiply");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(2);

		Expression exp3 = fact.literal(60000000);

		Filter filter = fact.greater(fact.multiply(exp1, exp2),exp3);

		assertEquals(4,testshp(filter, null));

	}

	/**

	 * area / 2 > 1.5e7 --> area > 3e7

	 */

	public void testDivide(){

		System.out.println("testDivide");

		initialize();

		FilterFactory2 fact = new FilterFactory2Impl();

		Expression exp1 = fact.property("AREA");

		Expression exp2 = fact.literal(2);

		Expression exp3 = fact.literal(15000000);

		Filter filter = fact.greater(fact.divide(exp1, exp2),exp3);

		assertEquals(4,testshp(filter, null));

	}

	private int testshp(Filter filter, SortBy[] order){

		try {

			store.open();

		} catch (OpenException e2) {

			// TODO Auto-generated catch block

			e2.printStackTrace();

		}

		FeatureType ft=store.getDefaultFeatureType();

		FeatureCollection featureCollection=null;

		try {

			featureCollection = (FeatureCollection)store.getDataCollection(ft,filter,order);

		} catch (ReadException e1) {

			// TODO Auto-generated catch block

			e1.printStackTrace();

		}

		Iterator iter = featureCollection.iterator();

		Iterator titlesIt=ft.iterator();

		System.out.print("ID");

		while (titlesIt.hasNext()) {

			FeatureAttributeDescriptor descriptor = (FeatureAttributeDescriptor) titlesIt.next();

			System.out.print("\t"+ descriptor.getName() + "(" + descriptor.getDataType().substring(0,3)+")");

		}

		System.out.println("");

		while (iter.hasNext()) {

			Feature feature = (Feature)iter.next();

			System.out.print(feature.getID().toString());

			//System.out.println("SHP Feature ------------------- ");

			Iterator iterator=ft.iterator();

			while (iterator.hasNext()) {

				FeatureAttributeDescriptor descriptor = (FeatureAttributeDescriptor) iterator.next();

				int i=descriptor.ordinal();

				String type=descriptor.getDataType();

				//System.out.println("****** "+ descriptor.getName());

				if (type.equals(FeatureAttributeDescriptor.TYPE_BOOLEAN)){

					System.out.print("\t"+ feature.getBoolean(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_BYTE)){

					System.out.print("\t"+ feature.getByte(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_DATE)){

					try{

						System.out.print("\t"+ feature.getDate(i));

					} catch (Exception e){

					}

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_DOUBLE)){

					System.out.print("\t"+ feature.getDouble(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_FLOAT)){

					System.out.print("\t"+ feature.getFloat(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_INT)){

					System.out.print("\t"+ feature.getInt(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_LONG)){

					System.out.print("\t"+ feature.getLong(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_STRING)){

					System.out.print("\t"+ feature.getString(i));

				}else if (type.equals(FeatureAttributeDescriptor.TYPE_GEOMETRY)){

					System.out.print("\tGeometry");

				}

			}

			System.out.println("");

		}

		iter=null;

		int numFeat = featureCollection.size();

		featureCollection.dispose();

		try {

			store.close();

			store.dispose();

		} catch (CloseException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		return numFeat;

	}

}