polygonhandlertest.java

shape file read and write

/*
 *    Geotools2 - OpenSource mapping toolkit
 *    http://geotools.org
 *    (C) 2002-2006, Geotools Project Managment Committee (PMC)
 *
 *    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.shapefile.shp;

import java.io.IOException;
import java.util.ArrayList;

import org.geotools.data.shapefile.TestCaseSupport;

import com.vividsolutions.jts.geom.Coordinate;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.LinearRing;
import com.vividsolutions.jts.geom.PrecisionModel;

/**
 * 
 * @source $URL:
 *         http://svn.geotools.org/geotools/trunk/gt/modules/plugin/shapefile/src/test/java/org/geotools/data/shapefile/shp/PolygonHandlerTest.java $
 * @version $Id: PolygonHandlerTest.java 28748 2008-01-14 21:52:51Z jeichar $
 * @author Ian Schneider
 */
public class PolygonHandlerTest extends TestCaseSupport {

    public PolygonHandlerTest(String testName) throws IOException {
        super(testName);
    }

    public static void main(java.lang.String[] args) {
        junit.textui.TestRunner.run(suite(PolygonHandlerTest.class));
    }

    public void testPolygonHandler() {
        Coordinate[] c = new Coordinate[3];
        c[0] = new Coordinate(0, 0, 0);
        c[1] = new Coordinate(1, 1, Double.NaN);
        c[2] = new Coordinate(1, 2, 3);
        PolygonHandler handler = new PolygonHandler();
        assertTrue(handler.getShapeType() == ShapeType.POLYGON);
        for (int i = 0, ii = c.length; i < ii; i++) {
            assertTrue(handler.pointInList(c[i], c));
        }
    }

    public void testHoleAssignment() {
        java.awt.Dimension ps = new java.awt.Dimension(500, 500);
        PrecisionModel precision = new PrecisionModel();

        ArrayList shells = new ArrayList();
        ArrayList holes = new ArrayList();

        int x = 10;
        int y = 10;

        shells.add(copyTo(x, y, ps.width - 2 * x, ps.height - 2 * y, rectangle(
                precision, 0)));

        int w = 11;
        int h = 11;
        int s = 10;

        int nx = (ps.width - 2 * x) / (w + s);
        int ny = (ps.height - 2 * y) / (h + s);

        for (int i = 0; i < nx; i++) {
            for (int j = 0; j < ny; j++) {
                holes.add(copyTo(x + s + i * (w + s), y + s + j * (h + s), w,
                        h, rectangle(precision, 0)));
            }
        }

        PolygonHandler ph = new PolygonHandler();
        ArrayList assigned = ph.assignHolesToShells(shells, holes);
        assertEquals(((ArrayList) assigned.get(0)).size(), holes.size());

    }

    public static Geometry rectangle(PrecisionModel pm, int SRID) {
        Coordinate[] coords = new Coordinate[5];
        for (int i = 0; i < coords.length; i++) {
            coords[i] = new Coordinate();
        }
        return new GeometryFactory().createLinearRing(coords);
    }

    public static Geometry copyTo(double x, double y, double w, double h,
            Geometry g) {
        if (g.getNumPoints() != 5)
            throw new IllegalArgumentException("Geometry must have 5 points");
        if (!LinearRing.class.isAssignableFrom(g.getClass()))
            throw new IllegalArgumentException("Geometry must be linear ring");
        Coordinate[] coords = g.getCoordinates();
        coords[0].x = x;
        coords[0].y = y;
        coords[1].x = x + w;
        coords[1].y = y;
        coords[2].x = x + w;
        coords[2].y = y + h;
        coords[3].x = x;
        coords[3].y = y + h;
        coords[4].x = x;
        coords[4].y = y;
        return g;
    }
}