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.base.core;

import java.util.Iterator;

import java.util.List;

import com.vividsolutions.jts.geom.Geometry;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.NullParameterValueException;

import es.unex.sextante.exceptions.WrongParameterIDException;

import es.unex.sextante.exceptions.WrongParameterTypeException;

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

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

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

import org.gvsig.geoprocess.core.gvVectorLayer;

import org.gvsig.sextante.app.algorithm.base.util.GeometryUtil;

import org.gvsig.sextante.app.algorithm.base.util.JTSFacade;

import org.gvsig.sextante.app.algorithm.base.visitor.exception.VisitorException;

import org.gvsig.tools.dispose.DisposableIterator;

/**

 * Joins all geometries in a layer

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

 */

public class ScalableUnionOperation extends GeometryOperation {

        //geometry type of the layer whose features we are going to fussion

        //(polygon features are optimized in jts with buffer(0) trick, the

        //nor the rest

        private Geometry            geometry         = null;

        /*

         * (non-Javadoc)

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

         */

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

                if(g == null)

                        return null;

                if(g.getGeometryType().getType() != TYPES.SURFACE)

                        return null;

                com.vividsolutions.jts.geom.Geometry actualGeometry = GeometryUtil.geomToJTS(g);

                if(geometry == null){

                        geometry = actualGeometry;

                }else{

                        Geometry[] geoms = new Geometry[2];

                        geoms[0] = geometry;

                        geoms[1] = actualGeometry;

                        geometry = JTSFacade.union(geoms, g.getType());

                }

                return null;

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.sextante.app.algorithm.base.core.IOperation#invoke(org.gvsig.fmap.geom.Geometry, org.gvsig.fmap.dal.feature.EditableFeature)

         */

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

                invoke(g, featureInput);

        }

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

                        throws DataException {

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.sextante.app.algorithm.base.core.IOperation#getResult()

         */

        public Object getResult() {

                return getGeometry();

        }

        /**

         * Returns FMap convex hull geometry.

         * @return

         */

        private org.gvsig.fmap.geom.Geometry getGeometry() {

                return GeometryUtil.jtsToGeom(geometry);

        }

        /**

         * Joins two geometries

         * @param g

         * @param geometryType

         * @return

         */

        private void union(org.gvsig.fmap.geom.Geometry g, int geometryType) {

                if(g == null)

                        return;

                if(g.getGeometryType().getType() != TYPES.SURFACE)

                        return;

                com.vividsolutions.jts.geom.Geometry actualGeometry = GeometryUtil.geomToJTS(g);

                if(geometry == null){

                        geometry = actualGeometry;

                }else{

                        Geometry[] geoms = new Geometry[2];

                        geoms[0] = geometry;

                        geoms[1] = actualGeometry;

                        geometry = JTSFacade.union(geoms, geometryType);

                }

        }

        /**

         * Computes union of all geometries of the clipping layer. The input resources won't 

         * be released.

         *

         * @return one geometry

         * @throws NullParameterValueException 

         * @throws WrongParameterIDException 

         * @throws WrongParameterTypeException 

         * @throws com.iver.cit.gvsig.fmap.DriverException

         * @throws ReadDriverException

         * @throws VisitorException

         * @throws ExpansionFileReadException

         */

        public static org.gvsig.fmap.geom.Geometry joinLayerGeometries(IVectorLayer input) throws WrongParameterTypeException, WrongParameterIDException, NullParameterValueException {

                FeatureStore store = null;

                if(input instanceof gvVectorLayer)

                        store = ((gvVectorLayer)input).getFeatureStore();

                else

                        return null;

                return joinLayerGeometries(store);

        }

        /**

         * Computes union of all geometries of the clipping layer. The input resources won't 

         * be released.

         *

         * @return one geometry

         * @throws NullParameterValueException 

         * @throws WrongParameterIDException 

         * @throws WrongParameterTypeException 

         * @throws com.iver.cit.gvsig.fmap.DriverException

         * @throws ReadDriverException

         * @throws VisitorException

         * @throws ExpansionFileReadException

         */

        @SuppressWarnings("unchecked")

        public static org.gvsig.fmap.geom.Geometry joinLayerGeometries(FeatureStore store) throws WrongParameterTypeException, WrongParameterIDException, NullParameterValueException {

                ScalableUnionOperation operation = new ScalableUnionOperation();

                FeatureSet features = null;

                try {

                        features = store.getFeatureSet();

                        DisposableIterator it = features.iterator();

                        while( it.hasNext() ) {

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

                                List geomList = feature.getGeometries();

                                if(geomList == null) {

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

                                        if(geom != null)

                                                operation.union(geom, geom.getType());

                                        continue;

                                }

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

                                while(itGeom.hasNext()) {

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

                                        operation.union(g, g.getType());

                                }

                        }

                        return operation.getGeometry();

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                return null;

        }

}