Application: gvSIG desktop » gvSIG geoprocess

/*

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2010 Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.sextante.app.algorithm.intersection;

import java.util.Iterator;

import java.util.List;

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

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

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.exception.CreateGeometryException;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

import org.gvsig.fmap.geom.operation.tojts.ToJTS;

import org.gvsig.sextante.app.algorithm.base.core.DALFeaturePersister;

import com.vividsolutions.jts.geom.Geometry;

import com.vividsolutions.jts.precision.EnhancedPrecisionOp;

import es.unex.sextante.core.Sextante;

/**

 * 

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class IntersectionOperation {

        private FeatureStore                     storeOverlay     = null;

        private DALFeaturePersister              persister        = null;

        private EditableFeature                  lastEditFeature  = null;

        public IntersectionOperation(FeatureStore overlayLayer) {

                this.storeOverlay = overlayLayer;

        }

        /**

         * Sets the output FeatureType

         * @param out

         * @throws DataException 

         */

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

                persister = new DALFeaturePersister(out, attrNames);

        }

        /**

         * Computes intersection between the geometry and the overlay layer. The fields of the

         * intersected features will be added.

         * @param g

         * @param featureInput

         * @return

         */

        @SuppressWarnings("unchecked")

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

                if(g == null)

                        return lastEditFeature;

                Geometry jtsGeom = null;

                try {

                        jtsGeom = (Geometry)g.invokeOperation(ToJTS.CODE, null);

                } catch (GeometryOperationNotSupportedException e1) {

                        Sextante.addErrorToLog(e1);

                } catch (GeometryOperationException e1) {

                        Sextante.addErrorToLog(e1);

                }

                FeatureSet features = null;

                DisposableIterator it = null;

                try {

                        features = storeOverlay.getFeatureSet();

                        it = features.iterator();

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                        return lastEditFeature;

                }

                while( it.hasNext() ) {

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

                        List geomList = featureOverlay.getGeometries();

                        if(geomList == null) {

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

                                lastEditFeature = intersection(jtsGeom, geom, featureInput, featureOverlay);

                                continue;

                        }

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

                        while(itGeom.hasNext()) {

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

                                lastEditFeature = intersection(jtsGeom, geom, featureInput, featureOverlay);

                        }

                }

                it.dispose();

                return lastEditFeature;

        }

        private EditableFeature intersection(Geometry jtsGeom, 

                                                                                        org.gvsig.fmap.geom.Geometry overGeom, 

                                                                                        Feature featureInput, 

                                                                                        Feature featureOverlay) {

                Geometry overlaysGeom = null;

                try {

                        overlaysGeom = (Geometry)overGeom.invokeOperation(ToJTS.CODE, null);

                } catch (GeometryOperationNotSupportedException e1) {

                        Sextante.addErrorToLog(e1);

                        return null;

                } catch (GeometryOperationException e1) {

                        Sextante.addErrorToLog(e1);

                        return null;

                }

                try {

                        if(!jtsGeom.getEnvelope().intersects(overlaysGeom.getEnvelope())) 

                                return lastEditFeature;

                        if(jtsGeom.intersects(overlaysGeom)) {

                                Geometry newGeom = EnhancedPrecisionOp.intersection(jtsGeom, overlaysGeom);

                                if(!newGeom.isEmpty())

                                        lastEditFeature = persister.addFeature(featureInput, featureOverlay, newGeom);

                        }

                } catch (CreateGeometryException e) {

                        Sextante.addErrorToLog(e);

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                return lastEditFeature;

        }

        /**

         * clips feature's geometry with the clipping geometry, preserving

         * feature's original attributes.

         * If feature's geometry doesn't touch clipping geometry, it will be

         * ignored.

         */

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

                /*if(g == null)

                        return;

                Geometry jtsGeom = Converter.geometryToJts(g);

                if(!jtsGeom.getEnvelope().intersects(overlaysGeom.getEnvelope()))

                        return;

                if(jtsGeom.intersects(overlaysGeom)) {

                        try {

                                Geometry newGeom = EnhancedPrecisionOp.difference(jtsGeom, overlaysGeom);

                                persister.addFeature(feature, newGeom);

                        } catch (CreateGeometryException e) {

                                Sextante.addErrorToLog(e);

                        } catch (DataException e) {

                                Sextante.addErrorToLog(e);

                        }

                } */

        }

        /**

         * Ends the edition and closes the FeatureStore

         */

        public void end() {

                persister.end();

        }

        public String getProcessDescription() {

                return "Clipping features agaisnt a clip geometry";

        }

}