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

import java.util.Iterator;

import java.util.List;

import com.vividsolutions.jts.geom.Geometry;

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

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

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.dal.feature.exception.FeatureIndexException;

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.algorithm.base.util.GeometryUtil;

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

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

import org.gvsig.tools.dispose.DisposableIterator;

/**

 *

 * @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 FeatureIndex              index                 = null;

        public SpatiallyIndexedSpatialJoinOperation(FlyrVectIVectorLayer targetLayer, String indexName, AbstractSextanteGeoProcess p) {

                super(p);

                FeatureIndexes indexes = targetLayer.getFeatureStore().getIndexes();

                index = indexes.getFeatureIndex(indexName);

                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(org.gvsig.fmap.geom.Geometry g, Feature featureInput) {

            boolean addedFeature = false;

                if(g == null)

                        return lastEditFeature;

                if(featureSelection == null) {

                        try {

                                featureSelection = storeOverlay.getFeatureSelection();

                        } catch (DataException e1) {

                                //Sin selecci?n tiramos

                        }

                }

                try {

                        Geometry gJts = GeometryUtil.geomToJTS(g);

            Envelope rect = null;

                    rect = g.getEnvelope();

                        FeatureSet featSetMatch = index.getMatchFeatureSet(rect);

                        FeatureSet featSetNearest = index.getNearestFeatureSet(numOfNeighbours, rect);

                        double nearestDistance = Double.MAX_VALUE;

            Iterator it = featSetMatch.iterator();

            //Recorremos el featSetMatch para calcular la nearestDistance

            while(it.hasNext()) {

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

                if( featureSelection != null &&

                    !featureSelection.isEmpty() &&

                    !featureSelection.isSelected(feat))

                    continue;

                List geomList = feat.getGeometries();

                if(geomList == null) {

                    org.gvsig.fmap.geom.Geometry g2 = feat.getDefaultGeometry();

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        nearestDistance = dist;

                    }

                    continue;

                }

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

                while(itGeom.hasNext()) {

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

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        nearestDistance = dist;

                    }

                }

            }

//            it.dispose();

            //Recorremos el featSetNearest para calcular la nearestDistance

            it = featSetNearest.iterator();

            while(it.hasNext()) {

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

                if( featureSelection != null &&

                    !featureSelection.isEmpty() &&

                    !featureSelection.isSelected(feat))

                    continue;

                List geomList = feat.getGeometries();

                if(geomList == null) {

                    org.gvsig.fmap.geom.Geometry g2 = feat.getDefaultGeometry();

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        nearestDistance = dist;

                    }

                    continue;

                }

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

                while(itGeom.hasNext()) {

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

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        nearestDistance = dist;

                    }

                }

            }

//            it.dispose();

            //Recorremos el featSetMatch para obtener las geometr?as que est?n a una distancia igual o menor a la nearestDistance

            it = featSetMatch.iterator();

            while (it.hasNext()) {

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

                if (featureSelection != null && !featureSelection.isEmpty()

                    && !featureSelection.isSelected(feat))

                    continue;

                List geomList = feat.getGeometries();

                if (geomList == null) {

                    org.gvsig.fmap.geom.Geometry g2 = feat.getDefaultGeometry();

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        buildFeature(featureInput, feat, new Double(

                            nearestDistance), g);

                        addedFeature= true;

                    }

                     continue;

                }

                Iterator<org.gvsig.fmap.geom.Geometry> itGeom =

                    geomList.iterator();

                while (itGeom.hasNext()) {

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

                    Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                    double dist = gJts.distance(g2Jts);

                    if (dist <= nearestDistance) {

                        buildFeature(featureInput, feat, new Double(

                            nearestDistance), g);

                        addedFeature= true;

                    }

                }

            }

//            it.dispose();

            //Recorremos el featSetNearest para obtener las geometr?as que est?n a una distancia igual o menor a la nearestDistance

            //si no se ha a?adido todav?a ninguna feature

            if (!addedFeature) {

                it = featSetNearest.iterator();

                while (it.hasNext()) {

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

                    if (featureSelection != null && !featureSelection.isEmpty()

                        && !featureSelection.isSelected(feat))

                        continue;

                    List geomList = feat.getGeometries();

                    if (geomList == null) {

                        org.gvsig.fmap.geom.Geometry g2 =

                            feat.getDefaultGeometry();

                        Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                        double dist = gJts.distance(g2Jts);

                        if (dist <= nearestDistance) {

                            buildFeature(featureInput, feat, new Double(

                                nearestDistance), g);

                            addedFeature = true;

                        }

                        continue;

                    }

                    Iterator<org.gvsig.fmap.geom.Geometry> itGeom =

                        geomList.iterator();

                    while (itGeom.hasNext()) {

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

                        Geometry g2Jts = GeometryUtil.geomToJTS(g2);

                        double dist = gJts.distance(g2Jts);

                        if (dist <= nearestDistance && !addedFeature) {

                            buildFeature(featureInput, feat, new Double(

                                nearestDistance), g);

                            addedFeature = true;

                        }

                    }

                }

//                it.dispose();

            }

                } catch(FeatureIndexException e) {

                        Sextante.addErrorToLog(e);

                } catch (DataException 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();

                int sizeFeat1 = feat1.getType().size() - 1;

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

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

                for (int i = sizeFeat1; i < sizeFeat1 + feat2.getType().size() - 1; i++)

                        outFeat.set(i, feat2.get(i - sizeFeat1));

                outFeat.set(outFeat.getType().size() - 2, 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) {

        }

}