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

import es.unex.sextante.core.Sextante;

import org.gvsig.fmap.dal.DataTypes;

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.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.dissolve.Summary;

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 IntersectsSpatialJoinOperation extends GeometryOperation {

        /**

         * Specialized instance in nearest neighbor search.

         */

        private FeatureIndex              index                 = null;

        private Summary                   summary               = null;

        private FeatureStore              storeOverlay          = null;

        private FeatureSelection          featureSelection      = null;

        public IntersectsSpatialJoinOperation(FlyrVectIVectorLayer targetLayer, String indexName, Summary summary, AbstractSextanteGeoProcess p) {

                super(p);

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

                index = indexes.getFeatureIndex(indexName);

                this.summary = summary;

                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("deprecation")

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

                if(g == null)

                        return lastEditFeature;

                if(featureSelection == null) {

                        try {

                                featureSelection = storeOverlay.getFeatureSelection();

                        } catch (DataException e1) {

                                //Sin selecci?n tiramos

                        }

                }

                try {

                        Envelope rect = g.getEnvelope();

                        FeatureSet featSet = index.getMatchFeatureSet(rect);

                        boolean first = true;

                        int numReg = 0;

                        DisposableIterator it = featSet.iterator();

                        while(it.hasNext()) {

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

                                if( featureSelection != null && 

                                        !featureSelection.isEmpty() && 

                                        !featureSelection.isSelected(feat))

                                        continue;

                                if(first) {

                                        summary.loadDefaultSummarizes(feat);

                                        first = false;

                                }

                                summary.updateValues(feat);

                                numReg ++;

                        }

                        buildNewFeature(featureInput, summary, numReg);

                } catch(FeatureIndexException e) {

                        Sextante.addErrorToLog(e);

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                return lastEditFeature;

        }

        /**

         * Builds a new output feature and adds it to the output file

         * @param feat1

         * @param feat2

         * @param value

         * @param g

         * @throws DataException

         */

        @SuppressWarnings("unchecked")

        private void buildNewFeature(Feature feat, Summary summary, int numReg) throws DataException {

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

                FeatureAttributeDescriptor[] attrs = feat.getType().getAttributeDescriptors();

                //Loads the old feature

                int pos = 0;

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

                        if (attrs[i].getDataType().getType() != DataTypes.GEOMETRY) {

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

                                pos ++;

                        }

                }

                //Loads the statistics

                summary.loadEditableFeature(outFeat);

                //Loads the new field

                outFeat.set(SpatialJoinAlgorithm.NEW_FIELD, new Integer(numReg));

                //Saves the geometry

                try {

                        List l = feat.getGeometries();

                        if(l == null) {

                                persister.addFeature(outFeat, feat.getDefaultGeometry());

                        } else {

                                EditableFeature editFeat = null;

                                for (int i = 0; i < l.size(); i++) {

                                        if(editFeat == null) {

                                                editFeat = persister.addFeature(outFeat, (org.gvsig.fmap.geom.Geometry)l.get(i));

                                        } else {

                                                persister.addGeometryToExistingFeature(editFeat, (org.gvsig.fmap.geom.Geometry)l.get(i));

                                        }

                                }

                        }

                } 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) {

        }

}