Application: gvSIG desktop » gvSIG geoprocess

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2012 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.geoprocess.algorithm.base.core;

import java.util.ArrayList;

import java.util.Date;

import java.util.Iterator;

import java.util.List;

import com.vividsolutions.jts.geom.Geometry;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.NullParameterValueException;

import es.unex.sextante.exceptions.WrongParameterIDException;

import es.unex.sextante.exceptions.WrongParameterTypeException;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.EditableFeature;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

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

import org.gvsig.geoprocess.algorithm.base.util.GeometryUtil;

import org.gvsig.geoprocess.algorithm.base.util.JTSFacade;

import org.gvsig.geoprocess.algorithm.base.visitor.exception.VisitorException;

import org.gvsig.geoprocess.lib.sextante.AbstractSextanteGeoProcess;

import org.gvsig.geoprocess.lib.sextante.dataObjects.FlyrVectIVectorLayer;

import org.gvsig.tools.dispose.DisposableIterator;

/**

 * Joins all geometries in a layer

 *

 * @author <a href="mailto:nachobrodin@gmail.com">Nacho Brodin</a>

 */

public class ScalableUnionOperation extends GeometryOperation {

    private static final Logger LOG = LoggerFactory

            .getLogger(ScalableUnionOperation.class);

        //geometry type of the layer whose features we are going to fussion

    //(polygon features are optimized in jts with buffer(0) trick, the

    //nor the rest

    private Geometry geometry = null;

    public ScalableUnionOperation() {

        super(null);

    }

    public ScalableUnionOperation(AbstractSextanteGeoProcess p) {

        super(p);

    }

    /*

     * (non-Javadoc)

     * @see org.gvsig.geoprocess.algorithm.base.core.IOperation#invoke(org.gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.Feature)

     */

    public EditableFeature invoke(org.gvsig.fmap.geom.Geometry g, Feature featureInput) {

        if (g == null) {

            return null;

        }

        if (g.getGeometryType().getType() != TYPES.SURFACE) {

            return null;

        }

        com.vividsolutions.jts.geom.Geometry actualGeometry = GeometryUtil.geomToJTS(g);

        if (geometry == null) {

            geometry = actualGeometry;

        } else {

            Geometry[] geoms = new Geometry[2];

            geoms[0] = geometry;

            geoms[1] = actualGeometry;

            geometry = JTSFacade.union(geoms, g.getType());

        }

        return null;

    }

    /*

     * (non-Javadoc)

     * @see org.gvsig.geoprocess.algorithm.base.core.IOperation#invoke(org.gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.EditableFeature)

     */

    public void invoke(org.gvsig.fmap.geom.Geometry g, EditableFeature featureInput) {

        invoke(g, featureInput);

    }

    public void setFeatureStore(FeatureStore out, String[] attrNames)

            throws DataException {

    }

    /*

     * (non-Javadoc)

     * @see org.gvsig.geoprocess.algorithm.base.core.IOperation#getResult()

     */

    public Object getResult() {

        return getGeometry();

    }

    /**

     * Returns FMap convex hull geometry.

     *

     * @return

     */

    private org.gvsig.fmap.geom.Geometry getGeometry() {

        return GeometryUtil.jtsToGeom(geometry);

    }

    /**

     * Joins two geometries

     *

     * @param g

     * @param geometryType

     * @return

     */

    private void union(org.gvsig.fmap.geom.Geometry g, int geometryType) {

        Date time1 = new Date();

        if (g == null) {

            return;

        }

        if (g.getGeometryType().getType() != TYPES.SURFACE) {

            return;

        }

        com.vividsolutions.jts.geom.Geometry actualGeometry = GeometryUtil.geomToJTS(g);

        if (geometry == null) {

            geometry = actualGeometry;

        } else {

            Geometry[] geoms = new Geometry[2];

            geoms[0] = geometry;

            geoms[1] = actualGeometry;

            geometry = JTSFacade.union(geoms, geometryType);

        }

        Date time2 = new Date();

        LOG.info("scalableUnion time = " + (time2.getTime() - time1.getTime() + " ms"));

    }

    /**

     * Joins an array of surface geometries

     *

     * @param geoms

     * @param geometryType

     * @return

     */

    private void union(org.gvsig.fmap.geom.Geometry[] geoms) {

        Date time1 = new Date();

        if (geoms == null) {

            return;

        }

        ArrayList<Geometry> geometries = new ArrayList<Geometry>();

        if (geometry != null) {

            geometries.add(geometry);

        }

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

            if (geoms[i].getGeometryType().getType() == TYPES.SURFACE) {

                geometries.add(GeometryUtil.geomToJTS(geoms[i]));

            }

        }

        geometry = JTSFacade.union((Geometry[]) geometries.toArray(new Geometry[0]), TYPES.SURFACE);

        Date time2 = new Date();

        LOG.info("scalableUnion time = " + (time2.getTime() - time1.getTime() + " ms"));

    }

    /**

     * Computes union of all geometries of the clipping layer. The input

     * resources won't be released.

     *

     * @return one geometry

     * @throws NullParameterValueException

     * @throws WrongParameterIDException

     * @throws WrongParameterTypeException

     * @throws com.iver.cit.gvsig.fmap.DriverException

     * @throws ReadDriverException

     * @throws VisitorException

     * @throws ExpansionFileReadException

     */

    public static org.gvsig.fmap.geom.Geometry joinLayerGeometries(IVectorLayer input, boolean selectedGeom) throws WrongParameterTypeException, WrongParameterIDException, NullParameterValueException {

        FeatureStore store = null;

        if (input instanceof FlyrVectIVectorLayer) {

            store = ((FlyrVectIVectorLayer) input).getFeatureStore();

        } else {

            return null;

        }

        return joinLayerGeometries(store, selectedGeom);

    }

    /**

     * Computes union of all geometries of the clipping layer. The input

     * resources won't be released.

     *

     * @return one geometry

     * @throws NullParameterValueException

     * @throws WrongParameterIDException

     * @throws WrongParameterTypeException

     * @throws com.iver.cit.gvsig.fmap.DriverException

     * @throws ReadDriverException

     * @throws VisitorException

     * @throws ExpansionFileReadException

     */

    @SuppressWarnings({"unchecked", "deprecation"})

    public static org.gvsig.fmap.geom.Geometry joinLayerGeometries(FeatureStore store, boolean selectedGeom) throws WrongParameterTypeException, WrongParameterIDException, NullParameterValueException {

        ScalableUnionOperation operation = new ScalableUnionOperation();

        FeatureSet features = null;

        try {

            if (selectedGeom) {

                features = (FeatureSet) store.getSelection();

            } else {

                features = store.getFeatureSet();

            }

            DisposableIterator it = features.iterator();

            int i = 0;

            List geometries = new ArrayList<org.gvsig.fmap.geom.Geometry>();

            List types = new ArrayList();

            while (it.hasNext()) {

                i++;

                Feature feature = (Feature) it.next();

                List geomList = feature.getGeometries();

                if (geomList == null) {

                    org.gvsig.fmap.geom.Geometry geom = feature.getDefaultGeometry();

                    if (geom != null) {

                        geometries.add(geom);

                        types.add(geom.getType());

                    }

                    continue;

                }

                Iterator<org.gvsig.fmap.geom.Geometry> itGeom = geomList.iterator();

                while (itGeom.hasNext()) {

                    org.gvsig.fmap.geom.Geometry g = itGeom.next();

                    geometries.add(g);

                    types.add(g.getType());

                }

            }

            operation.union((org.gvsig.fmap.geom.Geometry[]) geometries.toArray(new org.gvsig.fmap.geom.Geometry[0]));

            return operation.getGeometry();

        } catch (DataException e) {

            Sextante.addErrorToLog(e);

        }

        return null;

    }

}