Application: gvSIG desktop » gvSIG geoprocess

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 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.dispersepoints;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.ListIterator;

import java.util.Set;

import es.unex.sextante.core.Sextante;

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.FeatureReference;

import org.gvsig.fmap.geom.Geometry;

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

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.SpatialIndex;

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

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

import org.gvsig.geoprocess.algorithm.base.core.GeometryOperation;

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

/**

 * Disperse points operation

 *

 * @author fdiaz</a>

 */

public class DispersePointsOperation extends GeometryOperation {

    private SpatialIndex index;

    private double scatterRadius;

    private double matchDistance;

    private List<Set<FeatureReference>> groups;

    /**

     * Builds an instance of this operation.

     *

     * @param index

     * @param scatterRadius

     * @param matchDistance

     * @param p

     *

     */

    public DispersePointsOperation(SpatialIndex index, double scatterRadius, double matchDistance,

        AbstractSextanteGeoProcess p) {

        super(p);

        this.index = index;

        this.scatterRadius = scatterRadius;

        this.matchDistance = matchDistance;

        this.groups = new ArrayList<Set<FeatureReference>>();

    }

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

        // final MutableObject<Feature> foundFeature = new

        // MutableObject<Feature>(null);

        boolean addedFeature = false;

        if (g == null)

            return lastEditFeature;

        try {

            FeatureReference reference = feature.getReference();

            if (!isVisited(reference)) {

                Set<FeatureReference> group = new HashSet<FeatureReference>();

                // Iterator<?> iterator = index.queryNearest(g);

                Iterator<?> iterator = index.query(g.buffer(matchDistance));

                while (iterator.hasNext()) {

                    FeatureReference ref = (FeatureReference) iterator.next();

                    if (!isVisited(ref)) {

                        Feature feat = ref.getFeature().getCopy();

                        Geometry g2 = feat.getDefaultGeometry();

                        double dist = g.distance(g2);

                        if (dist <= matchDistance) {

                            group.add(ref);

                        }

                    }

                }

                groups.add(group);

                if (group.size() == 1) {

                    buildFeature(feature, g);

                } else {

                    buildFeatures(group);

                }

            }

        } catch (Exception e) {

            Sextante.addErrorToLog(e);

        }

        return lastEditFeature;

    }

    /**

     * @param group

     * @throws DataException

     * @throws CreateGeometryException

     */

    private void buildFeatures(Set<FeatureReference> group) throws CreateGeometryException, DataException {

        if (group.isEmpty()) {

            return;

        }

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        int points = group.size();

        Point center = getCenter(group);

        double step = 2 * Math.PI / points;

        int i = 0;

        for (FeatureReference ref : group) {

            Feature feature = ref.getFeature().getCopy();

            buildFeature(feature,

                geomManager.createPoint(

                    center.getX() + (Math.cos(i * step) * this.scatterRadius),

                    center.getY() + (Math.sin(i * step) * this.scatterRadius),

                    SUBTYPES.GEOM2D));

            i++;

        }

    }

    /**

     * @param group

     * @return

     * @throws DataException

     * @throws CreateGeometryException

     */

    private Point getCenter(Set<FeatureReference> group) throws DataException, CreateGeometryException {

        double x = 0;

        double y = 0;

        GeometryManager geomManager = GeometryLocator.getGeometryManager();

        for (FeatureReference ref : group) {

            Point point = (Point) ref.getFeature().getDefaultGeometry();

            x += point.getX();

            y += point.getY();

        }

        int size = group.size();

        return geomManager.createPoint(x / size, y / size, SUBTYPES.GEOM2D);

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.gvsig.geoprocess.algorithm.base.core.GeometryOperation#invoke(org

     * .gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.EditableFeature)

     */

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

        invoke(g, (Feature) feature);

    }

    /**

     * Builds a feature and adds it to the output file

     *

     * @param feat

     * @param value

     * @param g

     * @throws DataException

     */

    private void buildFeature(Feature feat, org.gvsig.fmap.geom.Geometry g) throws DataException {

        EditableFeature outFeat = persister.getOutputFeatureStore().createNewFeature();

        int sizeFeat = feat.getType().size() - 1;

        for (int i = 0; i < sizeFeat; i++)

            outFeat.set(i, feat.get(i));

        try {

            persister.addFeature(outFeat, g);

        } catch (CreateGeometryException e) {

            Sextante.addErrorToLog(e);

        }

    }

    private boolean isVisited(FeatureReference ref) {

        for (Set<FeatureReference> group : groups) {

            if (group.contains(ref)) {

                return true;

            }

        }

        return false;

    }

}