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

import java.util.Iterator;

import org.apache.commons.lang3.mutable.MutableDouble;

import org.apache.commons.lang3.mutable.MutableObject;

import es.unex.sextante.core.Sextante;

import java.util.List;

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

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

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

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

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

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

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

import org.gvsig.fmap.geom.Geometry;

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

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

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

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

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.namestranslator.NamesTranslator;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

/**

 *

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

 */

public class SpatiallyIndexedSpatialJoinOperation extends GeometryOperation {

    private FeatureStore storeOverlay = null;

    private FeatureSelection featureSelection = null;

    /**

     * Geometry.distance() is a costly operation. Thats the reason for we are

     * only using a nearest neighbor. TODO SpatialIndex works with Rectangle2D,

     * and this may be a simplification that drives to errors. Make additional

     * probes to use a number of default neighbors

     */

    static final int DEFAULT_NUM_NEIGBOURS = 1;

    /**

     * Number of neighbors that nearestFinder must found.

     */

    int numOfNeighbours = DEFAULT_NUM_NEIGBOURS;

    /**

     * Specialized instance in nearest neighbor search.

     */

    private SpatialIndex index = null;

    public SpatiallyIndexedSpatialJoinOperation(FlyrVectIVectorLayer targetLayer, SpatialIndex index, AbstractSextanteGeoProcess p) {

        super(p);

        this.index = index;

        storeOverlay = targetLayer.getFeatureStore();

    }

    /*

         * (non-Javadoc)

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

     */

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

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

        final MutableObject<Feature> foundFeature = new MutableObject<Feature>(null);

        boolean addedFeature = false;

        if (g == null) {

            return lastEditFeature;

        }

        try {

            final MutableDouble nearestDistance = new MutableDouble(Double.POSITIVE_INFINITY);

            index.query(g, new Visitor() {

                @Override

                public void visit(Object obj) throws VisitCanceledException, BaseException {

                    FeatureReference ref = (FeatureReference) obj;

                    Feature feat = ref.getFeature();

                    Geometry g2 = feat.getDefaultGeometry();

                    double dist = g.distance(g2);

                    if (dist <= nearestDistance.getValue()) {

                        nearestDistance.setValue(dist);

                        foundFeature.setValue(feat.getCopy());

                    }

                }

            });

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

            while (iterator.hasNext()) {

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

                Feature feat = ref.getFeature();

                Geometry g2 = feat.getDefaultGeometry();

                double dist = g.distance(g2);

                if (dist <= nearestDistance.getValue()) {

                    nearestDistance.setValue(dist);

                    foundFeature.setValue(feat.getCopy());

                }

            }

            if (foundFeature != null) {

                buildFeature(featureInput, foundFeature.getValue(), new Double(nearestDistance.getValue()), g);

                addedFeature = true;

            }

        } catch (Exception e) {

            Sextante.addErrorToLog(e);

        }

        return lastEditFeature;

    }

    /**

     * Builds a feature and adds it to the output file

     *

     * @param feat1

     * @param feat2

     * @param value

     * @param g

     * @throws DataException

     */

    private void buildFeature(Feature feat1, Feature feat2, Object value, org.gvsig.fmap.geom.Geometry g) throws DataException {

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

        NamesTranslator nameTranslator;

        try {

            nameTranslator = ((SpatialJoinAlgorithm) this.process).namesTranslator;

        } catch (Exception ex) {

            nameTranslator = null;

        }

        if (nameTranslator != null) {

            for (FeatureAttributeDescriptor attr1 : feat1.getType().getAttributeDescriptors()) {

                String attrName = attr1.getName();

                String translation = nameTranslator.getTranslation(attrName);

                Object fValue = feat1.get(attrName);

                if (translation!=null && !translation.isEmpty()) {

                    outFeat.set(translation, fValue);

                }

            }

            for (FeatureAttributeDescriptor attr2 : feat2.getType().getAttributeDescriptors()) {

                String attrName = attr2.getName();

                List<String> translation = nameTranslator.getAllTranslations(attrName);

                Object fValue = feat2.get(attrName);

                if (translation!=null && !translation.isEmpty()) {

                    outFeat.set(translation.get(translation.size() - 1), fValue);

                }

            }

            outFeat.set(nameTranslator.getTranslation("DIST"), (Object) value);

        } else {

            outFeat.copyFrom(feat2);

            outFeat.copyFrom(feat1);

            outFeat.set("DIST", (Object) value);

        }

        try {

            persister.addFeature(outFeat, g);

        } catch (CreateGeometryException e) {

            Sextante.addErrorToLog(e);

        }

    }

    /*

         * (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 featureInput) {

    }

}