DouglasPeuckerProcessorTest.java

package net.refractions.linecleaner.cleansing.test;

import java.io.File;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;

import junit.framework.TestCase;

import net.refractions.linecleaner.cleansing.DouglasPeuckerProcessor;
import net.refractions.linecleaner.cleansing.PseudoNodeProcessor;

import org.geotools.data.FeatureSource;
import org.geotools.data.FeatureStore;
import org.geotools.data.shapefile.indexed.IndexedShapefileDataStore;
import org.geotools.data.shapefile.indexed.IndexedShapefileDataStoreFactory;
import org.geotools.resources.TestData;

public class DouglasPeuckerProcessorTest extends TestCase {
	private IndexedShapefileDataStore ds;
	private String typeName;
	private FeatureSource source;
	private FeatureStore featureStore; 
	
	protected void setUp() throws Exception {
        super.setUp();
        long before = System.currentTimeMillis();

		ds = getDataStore("sam.shp");
		
        typeName = ds.getTypeNames()[0];
        source = ds.getFeatureSource( typeName );
        featureStore = (FeatureStore) ds.getFeatureSource();
        System.out.println(source.getFeatures().getCount());
        long after = System.currentTimeMillis();
        after = after-before;
        System.out.println("Startup time: " + after);
    }
	
	public IndexedShapefileDataStore getDataStore(String filename) throws Exception {

        File shp = TestData.file( this, filename );
		Map params=new HashMap();
		params.put( IndexedShapefileDataStoreFactory.URLP.key, shp.toURL());
		params.put( IndexedShapefileDataStoreFactory.CREATE_SPATIAL_INDEX.key, new Boolean(true));
		params.put( IndexedShapefileDataStoreFactory.MEMORY_MAPPED.key, new Boolean(false));
		params.put( IndexedShapefileDataStoreFactory.SPATIAL_INDEX_TYPE.key, IndexedShapefileDataStoreFactory.TREE_GRX);
//		params.put( IndexedShapefileDataStoreFactory.SPATIAL_INDEX_TYPE.key, IndexedShapefileDataStoreFactory.TREE_QIX);
        
		Logger.getLogger("org.geotools.data").setLevel(Level.WARNING);
		
		IndexedShapefileDataStoreFactory fac=new IndexedShapefileDataStoreFactory();
		return (IndexedShapefileDataStore) fac.createDataStore(params);
	}
	
	public void testDP() throws Exception {
		
		PseudoNodeProcessor pnp = new PseudoNodeProcessor(featureStore);
		pnp.run();
		
		DouglasPeuckerProcessor dpProcessor = new DouglasPeuckerProcessor(featureStore, 5);
		dpProcessor.run();
	}
}