| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.fmap.geom.operation.writegml2;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.OutputStream;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.Geometry.TYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.aggregate.MultiCurve;

import org.gvsig.fmap.geom.aggregate.MultiPoint;

import org.gvsig.fmap.geom.aggregate.MultiSurface;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

import org.gvsig.fmap.geom.primitive.Curve;

import org.gvsig.fmap.geom.primitive.GeneralPathX;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.fmap.geom.primitive.Surface;

import org.gvsig.gpe.lib.api.GPELocator;

import org.gvsig.gpe.lib.api.GPEManager;

import org.gvsig.gpe.lib.api.exceptions.WriterHandlerCreationException;

import org.gvsig.gpe.lib.api.writer.IGPEWriterHandler;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

/**

 * @author gvSIG Team

 * @version $Id$

 *

 */

public class WriteGmlTest extends AbstractLibraryAutoInitTestCase{

    private static final Logger LOG = LoggerFactory.getLogger(WriteGmlTest.class);

    private GeometryManager geometryManager;

    private GPEManager gpeManager;

    protected void doSetUp() throws Exception {

        geometryManager = GeometryLocator.getGeometryManager();   

        gpeManager = GPELocator.getGPEManager();

    }

    private String write(Geometry geometry) throws WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException{      

        OutputStream os = new ByteArrayOutputStream();   

        IGPEWriterHandler writerHandler = gpeManager.createWriterByMimeType("text/xml; subtype=gml/3.1.2");  

        writerHandler.setOutputStream(os);

        writerHandler.initialize(); 

        WriteGml2OperationContext operationContext = new WriteGml2OperationContext();

        operationContext.setWriterHandler(writerHandler);

        operationContext.setId("0");

        operationContext.setSrs("EPSG:23030");

        geometry.invokeOperation(WriteGml2Operation.NAME, operationContext);

        writerHandler.close();

        LOG.info("Parsed geometry: " + os.toString());

        assertTrue(os.toString().startsWith("<?xml"));

        return os.toString();

    }

    public void testPoint2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Point point = (Point)geometryManager.create(TYPES.POINT, SUBTYPES.GEOM2D);

        point.setCoordinateAt(0, 1.0);

        point.setCoordinateAt(1, 2.0);

        write(point);        

    }    

    public void testCurve2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Curve curve = (Curve)geometryManager.create(TYPES.CURVE, SUBTYPES.GEOM2D);

        GeneralPathX generalPathX = new GeneralPathX();

        generalPathX.moveTo(0, 0);

        generalPathX.lineTo(1, 1);

        generalPathX.lineTo(2, 1);      

        curve.setGeneralPath(generalPathX); 

        write(curve);

    }  

    public void testSurface2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Surface surface = (Surface)geometryManager.create(TYPES.SURFACE, SUBTYPES.GEOM2D);

        GeneralPathX generalPathX = new GeneralPathX();

        generalPathX.moveTo(0, 0);

        generalPathX.lineTo(1, 1);

        generalPathX.lineTo(2, 1);

        surface.setGeneralPath(generalPathX);   

        write(surface);

    } 

    public void testMultiPoint2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Point point1 = (Point)geometryManager.create(TYPES.POINT, SUBTYPES.GEOM3D);

        point1.setCoordinateAt(0, 1.0);

        point1.setCoordinateAt(1, 2.0);

        point1.setCoordinateAt(2, 3.0);

        Point point2 = (Point)geometryManager.create(TYPES.POINT, SUBTYPES.GEOM3D);

        point2.setCoordinateAt(0, 4.0);

        point2.setCoordinateAt(1, 5.0);

        point2.setCoordinateAt(2, 6.0);

        MultiPoint multiPoint = (MultiPoint)geometryManager.create(TYPES.MULTIPOINT, SUBTYPES.GEOM2D);

        multiPoint.addPoint(point1);

        multiPoint.addPoint(point2);

        write(multiPoint);        

    }    

    public void testMultiCurve2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Curve curve1 = (Curve)geometryManager.create(TYPES.CURVE, SUBTYPES.GEOM2D);

        GeneralPathX generalPathX1 = new GeneralPathX();

        generalPathX1.moveTo(0, 0);

        generalPathX1.lineTo(1, 1);

        generalPathX1.lineTo(2, 1);     

        curve1.setGeneralPath(generalPathX1);   

        Curve curve2 = (Curve)geometryManager.create(TYPES.CURVE, SUBTYPES.GEOM2D);

        GeneralPathX generalPathX2 = new GeneralPathX();

        generalPathX2.moveTo(0, 0);

        generalPathX2.lineTo(1, 1);

        generalPathX2.lineTo(2, 1);     

        curve2.setGeneralPath(generalPathX2);   

        MultiCurve multiCurve = (MultiCurve)geometryManager.create(TYPES.MULTICURVE, SUBTYPES.GEOM2D);

        multiCurve.addCurve(curve1);

        multiCurve.addCurve(curve2);

        write(multiCurve);

    }  

    public void testMultiSurface2D() throws CreateGeometryException, WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException, IOException{

        Surface surface1 = (Surface)geometryManager.create(TYPES.SURFACE, SUBTYPES.GEOM2D);

        GeneralPathX generalPathX1 = new GeneralPathX();

        generalPathX1.moveTo(0, 0);

        generalPathX1.lineTo(1, 1);

        generalPathX1.lineTo(2, 1);     

        surface1.setGeneralPath(generalPathX1); 

        Surface surface2 = (Surface)geometryManager.create(TYPES.SURFACE, SUBTYPES.GEOM2D);     

        GeneralPathX generalPathX2 = new GeneralPathX();

        generalPathX2.moveTo(0, 0);

        generalPathX2.lineTo(1, 1);

        generalPathX2.lineTo(2, 1);     

        surface2.setGeneralPath(generalPathX2); 

        MultiSurface multiSurface = (MultiSurface)geometryManager.create(TYPES.MULTISURFACE, SUBTYPES.GEOM2D);

        multiSurface.addSurface(surface1);

        multiSurface.addSurface(surface2); 

        write(multiSurface);

    } 

}