Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.geom.jts.aggregate;

import com.vividsolutions.jts.geom.LineString;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryException;

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

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

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

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

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

import org.gvsig.fmap.geom.jts.GeometryJTS;

import org.gvsig.fmap.geom.jts.util.JTSUtils;

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

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

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

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

import org.gvsig.fmap.geom.type.GeometryType;

/**

 * @author fdiaz

 *

 */

public class DefaultMultiCurve extends AbstractMultiCurve {

    /**

     * @param geometryType

     */

    public DefaultMultiCurve(GeometryType geometryType) {

        super(geometryType.getSubType());

    }

    /**

     * @param subtype

     */

    public DefaultMultiCurve(int subtype) {

        super(subtype);

    }

    @Override

    public Geometry cloneGeometry() {

        DefaultMultiCurve clone = new DefaultMultiCurve(getGeometryType());

        for(int i=0; i<primitives.size(); i++){

            clone.addPrimitive((Primitive)primitives.get(i).cloneGeometry());

        }

        clone.setProjection(this.getProjection());

        return clone;

    }

    @Override

    public Geometry force2D() throws GeometryOperationNotSupportedException, GeometryOperationException {

        DefaultMultiCurve other = new DefaultMultiCurve(getGeometryType());

        other.setProjection(this.getProjection());

        other.ensureCapacity(primitives.size());

        for (Primitive primitive : primitives) {

            other.addPrimitive((Primitive)primitive.forceSubtype(Geometry.SUBTYPES.GEOM2D));

        }

        return other;

    }

    @Override

    public Geometry force2DM() throws GeometryOperationNotSupportedException, GeometryOperationException {

        DefaultMultiCurve other = new DefaultMultiCurve(getGeometryType());

        other.setProjection(this.getProjection());

        other.ensureCapacity(primitives.size());

        for (Primitive primitive : primitives) {

            other.addPrimitive((Primitive)primitive.forceSubtype(Geometry.SUBTYPES.GEOM2DM));

        }

        return other;

    }

    @Override

    public Geometry force3D() throws GeometryOperationNotSupportedException, GeometryOperationException {

        DefaultMultiCurve other = new DefaultMultiCurve(getGeometryType());

        other.setProjection(this.getProjection());

        other.ensureCapacity(primitives.size());

        for (Primitive primitive : primitives) {

            other.addPrimitive((Primitive)primitive.forceSubtype(Geometry.SUBTYPES.GEOM3D));

        }

        return other;

    }

    @Override

    public Geometry force3DM() throws GeometryOperationNotSupportedException, GeometryOperationException {

        DefaultMultiCurve other = new DefaultMultiCurve(getGeometryType());

        other.setProjection(this.getProjection());

        other.ensureCapacity(primitives.size());

        for (Primitive primitive : primitives) {

            other.addPrimitive((Primitive)primitive.forceSubtype(Geometry.SUBTYPES.GEOM3DM));

        }

        return other;

    }

    @Override

    public int getDimension() {

        switch (getGeometryType().getSubType()) {

        case Geometry.SUBTYPES.GEOM2D:

            return 2;

        case Geometry.SUBTYPES.GEOM2DM:

            return 3;

        case Geometry.SUBTYPES.GEOM3D:

            return 3;

        case Geometry.SUBTYPES.GEOM3DM:

            return 4;

        default:

            return 0;

        }

    }

    @Override

    public MultiPoint toPoints() throws GeometryException {

        MultiPoint multipoint = null;

        for (int i = 0; i < this.getPrimitivesNumber(); i++) {

            if(multipoint==null){

                multipoint = this.getPrimitiveAt(i).toPoints();

            } else {

                MultiPoint points = (MultiPoint)this.getPrimitiveAt(i).toPoints();

                for (int j = 0; j < points.getPrimitivesNumber(); j++) {

                    multipoint.addPrimitive(points.getPrimitiveAt(j));

                }

            }

        }

        return multipoint;

    }

    @Override

    public MultiLine toLines() throws GeometryException {

        MultiLine multiline = null;

        for (int i = 0; i < this.getPrimitivesNumber(); i++) {

            if(multiline==null){

                multiline = this.getPrimitiveAt(i).toLines();

            } else {

                MultiLine lines = (MultiLine)this.getPrimitiveAt(i).toLines();

                for (int j = 0; j < lines.getPrimitivesNumber(); j++) {

                    multiline.addPrimitive(lines.getPrimitiveAt(j));

                }

            }

        }

        return multiline;

    }

    @Override

    public MultiPolygon toPolygons() throws GeometryException {

        MultiPolygon multipolygon = null;

        for (int i = 0; i < this.getPrimitivesNumber(); i++) {

            if(multipolygon==null){

                multipolygon = this.getPrimitiveAt(i).toPolygons();

            } else {

                MultiPolygon polygons = (MultiPolygon)this.getPrimitiveAt(i).toPolygons();

                for (int j = 0; j < polygons.getPrimitivesNumber(); j++) {

                    multipolygon.addPrimitive(polygons.getPrimitiveAt(j));

                }

            }

        }

        return multipolygon;

    }

    @Override

    public void flip() throws GeometryOperationNotSupportedException, GeometryOperationException {

        for (int i = 0; i < this.getPrimitivesNumber(); i++) {

            this.getPrimitiveAt(i).flip();

        }

        Collections.reverse(primitives);

    }

    @Override

    protected Geometry fixPrimitive(Primitive primitive) {

        int primitiveSubType = primitive.getGeometryType().getSubType();

        int subType = getGeometryType().getSubType();

        if(primitiveSubType == subType){

            return primitive;

        }

        String message = StringUtils.replace("This MultiCurve only accept subtype %(subtype)s primitives", "%(subtype)s", String.valueOf(subType));

        notifyDeprecated(message);

        throw new UnsupportedOperationException(message);

    }

    @Override

    public com.vividsolutions.jts.geom.Geometry getJTS() {

        List<LineString> lineStrings = new ArrayList<>(primitives.size());

        LineString line;

        for (Primitive primitive : primitives) {

            if (primitive instanceof Line) {

                line = (LineString) ((GeometryJTS) primitive).getJTS();

                lineStrings.add(line);

            } else {

                MultiLine multiline;

                try {

                    multiline = primitive.toLines();

                    for (int j = 0; j < multiline.getPrimitivesNumber(); j++) {

                        line =

                                (LineString) ((GeometryJTS) multiline.getPrimitiveAt(j)).getJTS();

                        lineStrings.add(line);

                    }

                } catch (GeometryException e) {

                    GeometryType geomType = primitive.getGeometryType();

                    LOGGER.warn(StringUtils.replaceEach(

                            "Can't convert primitive type=%(type)s, %(subtype)s to MultiLine",

                            new String[] {"%(type)s", "%(subtype)s" },

                            new String[] { String.valueOf(geomType.getType()), String.valueOf(geomType.getSubType()) }));

                }

            }

        }

        return JTSUtils.createJTSMultiLineString(lineStrings.toArray(new LineString[lineStrings.size()]));

    }

    @Override

    public Geometry offset(double distance) throws GeometryOperationNotSupportedException, GeometryOperationException {

        MultiCurve result = new DefaultMultiCurve(getGeometryType());

        for (Primitive primitive : primitives) {

            Geometry offset = primitive.offset(distance);

            if(offset instanceof MultiPrimitive){

                MultiPrimitive multiOffset = (MultiPrimitive)offset;

                for(int i=0; i<multiOffset.getPrimitivesNumber(); i++){

                    result.addPrimitive(multiOffset.getPrimitiveAt(i));

                }

            } else {

                result.addPrimitive((Primitive)primitive.offset(distance));

            }

        }

        return result;

    }

    @Override

    public Geometry offset(int joinStyle, double distance) throws GeometryOperationNotSupportedException, GeometryOperationException {

        MultiCurve result = new DefaultMultiCurve(getGeometryType());

        for (Primitive primitive : primitives) {

            Geometry offset = primitive.offset(joinStyle, distance);

            if(offset instanceof MultiPrimitive){

                MultiPrimitive multiOffset = (MultiPrimitive)offset;

                for(int i=0; i<multiOffset.getPrimitivesNumber(); i++){

                    result.addPrimitive(multiOffset.getPrimitiveAt(i));

                }

            } else {

                result.addPrimitive((Primitive)primitive.offset(joinStyle, distance));

            }

        }

        return result;

    }

}