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 java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.logging.Level;

import java.util.logging.Logger;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.GeometryUtils;

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

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

import org.gvsig.fmap.geom.complex.Complex;

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

import org.gvsig.fmap.geom.handler.Handler;

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

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

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

/**

 * @author fdiaz

 *

 */

public abstract class AbstractMultiPrimitive extends AbstractAggregate implements MultiPrimitive {

    /**

     *

     */

    private static final long serialVersionUID = -5993428771359742467L;

    /**

     * @param type

     * @param subtype

     */

    public AbstractMultiPrimitive(int type, int subtype) {

        super(type, subtype);

    }

    public void addPrimitive(Primitive primitive) {

        Geometry geom = fixPrimitive(primitive);

        if (geom != null) {

            if (geom instanceof MultiPrimitive) {

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

                    addPrimitive(((MultiPrimitive) geom).getPrimitiveAt(i));

                }

            } else {

                primitives.add((Primitive) geom);

            }

        }

    }

    @Override

    public void addPrimitives(Aggregate aggregate) {

        for (Geometry primitive : aggregate) {

            this.addPrimitive((Primitive) primitive);

        }

    }

    public void addPrimitives(Complex complex) {

        for (Geometry primitive : complex) {

            if(GeometryUtils.isSubtype(this.getType(), primitive.getType())){

                this.addPrimitive((Primitive) primitive);

            }

        }

    }

    @Override

    public void addPrimitives(Geometry geometry) {

        if(geometry instanceof Primitive){

            addPrimitive((Primitive) geometry);

        } else if(geometry instanceof Complex){

            addPrimitives((Complex) geometry);

        } else {

            addPrimitives((Aggregate)geometry);

        }

    }

    /**

     * @param primitive

     * @return

     */

    protected abstract Geometry fixPrimitive(Primitive primitive);

    protected Geometry clonePrimitives(AbstractMultiPrimitive clone) {

        clone.ensureCapacity(primitives.size());

        for (Iterator iterator = primitives.iterator(); iterator.hasNext();) {

            Primitive primitive = (Primitive) iterator.next();

            clone.addPrimitive((Primitive)primitive.cloneGeometry());

        }

        return clone;

    }

    /* (non-Javadoc)

     * @see org.gvsig.fmap.geom.Geometry#getHandlers(int)

     */

    public Handler[] getHandlers(int type) {

        List<Handler> handlers = new ArrayList<Handler>();

        for (Iterator<Primitive> iterator = primitives.iterator(); iterator.hasNext();) {

            Primitive primitive = (Primitive) iterator.next();

            Handler[] primitiveHandlers = primitive.getHandlers(type);

            for (int i = 0; i < primitiveHandlers.length; i++) {

                handlers.add(primitiveHandlers[i]);

            }

        }

        return handlers.toArray(new Handler[handlers.size()]);

    }

    /* (non-Javadoc)

     * @see org.gvsig.fmap.geom.aggregate.Aggregate#union()

     */

    public Geometry union() throws GeometryOperationException, GeometryOperationNotSupportedException {

        try {

            Geometry result = null;

            MultiPrimitive invalidResult = null;

            for (Iterator<Primitive> iterator = primitives.iterator(); iterator.hasNext();) {

                Primitive primitive = (Primitive) iterator.next();

                if (primitive.isValid()) {

                    if (result == null) {

                        result = primitive;

                    } else {

                        result = result.union(primitive);

                    }

                } else {

                    if (invalidResult == null) {

                        GeometryManager geomManager = GeometryLocator.getGeometryManager();

                        invalidResult = geomManager.createMultiPrimitive(primitive.getGeometryType());

                    }

                    invalidResult.addPrimitive(primitive);

                }

            }

            if (invalidResult == null) {

                return result;

            }

            invalidResult.addPrimitives(result);

            return invalidResult;

        } catch (CreateGeometryException ex) {

            GeometryManager geomManager = GeometryLocator.getGeometryManager();

            throw new GeometryOperationException(

                    this.getGeometryType().getType(), 

                    geomManager.getGeometryOperationCode(Geometry.OPERATIONS.UNION), 

                    ex);

        }

    }

    /* (non-Javadoc)

     * @see org.gvsig.fmap.geom.aggregate.Aggregate#intersection()

     */

    public Geometry intersection() throws GeometryOperationException, GeometryOperationNotSupportedException {

        Geometry result = null;

        for (Iterator<Primitive> iterator = primitives.iterator(); iterator.hasNext();) {

            Primitive primitive = (Primitive) iterator.next();

            if(result==null){

                result = primitive;

            } else {

                result.intersection(primitive);

            }

        }

        return result;

    }

}