ForceCoordinateSystemFeatureReaderTest.java example

Explorer
geotools-master
/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 * 
 *    (C) 2002-2008, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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
 *    Lesser General Public License for more details.
 */
package org.geotools.data.crs;

import junit.framework.TestCase;

import org.geotools.data.DataUtilities;
import org.geotools.data.FeatureReader;
import org.geotools.data.collection.ListFeatureCollection;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.feature.simple.SimpleFeatureBuilder;
import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
import org.geotools.referencing.crs.DefaultEngineeringCRS;
import org.geotools.referencing.crs.DefaultGeographicCRS;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.referencing.crs.CoordinateReferenceSystem;

import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.Point;

/**
 * 
 *
 * @source $URL$
 */
public class ForceCoordinateSystemFeatureReaderTest extends TestCase {

    private static final String FEATURE_TYPE_NAME = "testType";

    protected void setUp() throws Exception {
        super.setUp();
    }

    /**
     * create a feature collection with 1 feature in it.
     * @param crs the CRS of the FeatureType
     * @param p the point to add, should be same CRS as crs
     * @return
     * @throws Exception
     */
    private SimpleFeatureCollection createTestFeatureCollection(CoordinateReferenceSystem crs,
            Point p) throws Exception {
        
        SimpleFeatureTypeBuilder builder = new SimpleFeatureTypeBuilder();
        builder.setName(FEATURE_TYPE_NAME);
        builder.setCRS(crs);
        builder.add("geom", Point.class );
        
        SimpleFeatureType ft = builder.buildFeatureType();
        
        SimpleFeatureBuilder b = new SimpleFeatureBuilder(ft);
        b.add( p );
        
        ListFeatureCollection features = new ListFeatureCollection(ft);
        features.add( b.buildFeature(null));
        
        return features;
    }
    
    public void testSameCRS() throws Exception {
        CoordinateReferenceSystem crs = DefaultGeographicCRS.WGS84;
        GeometryFactory fac=new GeometryFactory();
        Point p = fac.createPoint(new Coordinate(10,10) );
        
        SimpleFeatureCollection features = createTestFeatureCollection(crs, p);
        FeatureReader<SimpleFeatureType, SimpleFeature> original = DataUtilities.reader( features );
        //FeatureReader<SimpleFeatureType, SimpleFeature> original = new CollectionFeatureReader( features, features.getSchema() );
        
        
        try (ForceCoordinateSystemFeatureReader modified = new ForceCoordinateSystemFeatureReader(
                DataUtilities.reader(features), crs);) {
            SimpleFeature f1=original.next();
            SimpleFeature f2=modified.next();
            
            assertEquals(f1,f2);
            
            assertFalse( original.hasNext() );
            assertFalse( modified.hasNext() );
            
            assertSame(modified.builder.getFeatureType(), original.getFeatureType());
        }
    }
    
    public void testDifferentCRS() throws Exception {
        CoordinateReferenceSystem srcCRS = DefaultGeographicCRS.WGS84;
        GeometryFactory fac=new GeometryFactory();
        Point p = fac.createPoint(new Coordinate(10,10) );
        
        SimpleFeatureCollection features = createTestFeatureCollection(srcCRS, p);
        FeatureReader<SimpleFeatureType, SimpleFeature> original = DataUtilities.reader( features );
        
        CoordinateReferenceSystem destCRS=DefaultEngineeringCRS.CARTESIAN_2D;
        try (ForceCoordinateSystemFeatureReader modified = new ForceCoordinateSystemFeatureReader(
                DataUtilities.reader(features), destCRS);){
            SimpleFeature f1=original.next();
            SimpleFeature f2=modified.next();
            
            assertEquals(((Geometry)f1.getDefaultGeometry()).getCoordinate(),((Geometry)f2.getDefaultGeometry()).getCoordinate());
            SimpleFeatureType f1Type = f1.getFeatureType();
            SimpleFeatureType f2Type = f2.getFeatureType();
            assertFalse( f1Type.getCoordinateReferenceSystem().equals(f2Type.getCoordinateReferenceSystem()) );
            assertEquals( srcCRS, f1Type.getCoordinateReferenceSystem());
            assertEquals( srcCRS, f1Type.getGeometryDescriptor().getCoordinateReferenceSystem());
            assertEquals( destCRS, f2Type.getCoordinateReferenceSystem());
            assertEquals( destCRS, f2Type.getGeometryDescriptor().getCoordinateReferenceSystem());
            
            assertFalse( original.hasNext() );
            assertFalse( modified.hasNext() );
            
            assertNotNull(modified.builder);
        }
    }
    
    public void testNullDestination() throws Exception {
        CoordinateReferenceSystem crs = DefaultGeographicCRS.WGS84;
        GeometryFactory fac=new GeometryFactory();
        Point p = fac.createPoint(new Coordinate(10,10) );
        
        SimpleFeatureCollection features = createTestFeatureCollection(crs, p);
        
        try{
            new ForceCoordinateSystemFeatureReader(
                    DataUtilities.reader( features ), (CoordinateReferenceSystem)null);
            fail(); // should throw a nullpointer exception.
        }catch(NullPointerException e){
            // good
        }
        
    }
    
    public void testNullSource() throws Exception {
        CoordinateReferenceSystem srcCRS = null;
        GeometryFactory fac=new GeometryFactory();
        Point p = fac.createPoint(new Coordinate(10,10) );
        
        SimpleFeatureCollection features = createTestFeatureCollection(srcCRS, p);
        FeatureReader<SimpleFeatureType, SimpleFeature> original = DataUtilities.reader( features );
        
        CoordinateReferenceSystem destCRS=DefaultEngineeringCRS.CARTESIAN_2D;
        try(ForceCoordinateSystemFeatureReader modified = new ForceCoordinateSystemFeatureReader(
                DataUtilities.reader( features ), destCRS);){
            SimpleFeature f1=original.next();
            SimpleFeature f2=modified.next();
            
            assertEquals(((Geometry)f1.getDefaultGeometry()).getCoordinate(),((Geometry)f2.getDefaultGeometry()).getCoordinate());
            SimpleFeatureType f1Type = f1.getFeatureType();
            SimpleFeatureType f2Type = f2.getFeatureType();
            assertFalse( f2Type.getCoordinateReferenceSystem().equals(f1Type.getCoordinateReferenceSystem()) );
            assertEquals( srcCRS, f1Type.getCoordinateReferenceSystem());
            assertEquals( srcCRS, f1Type.getGeometryDescriptor().getCoordinateReferenceSystem());
            assertEquals( destCRS, f2Type.getCoordinateReferenceSystem());
            assertEquals( destCRS, f2Type.getGeometryDescriptor().getCoordinateReferenceSystem());
            
            assertFalse( original.hasNext() );
            assertFalse( modified.hasNext() );
            
            assertNotNull(modified.builder);
        }
    }
}