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

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.dal.DataSet;

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

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

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.dal.feature.impl.DefaultFeatureSelection;

import org.gvsig.fmap.geom.Geometry;

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

import org.gvsig.sextante.app.extension.core.gvVectorLayer;

import es.unex.sextante.core.GeoAlgorithm;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.GeoAlgorithmExecutionException;

import es.unex.sextante.exceptions.RepeatedParameterNameException;

import es.unex.sextante.outputs.OutputVectorLayer;

/**

 * Convex Hull Algorithm

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

 */

public class ConvexHullAlgorithm extends GeoAlgorithm {

        public static final String  LAYER     = "LAYER";

        public static final String  RESULT    = "RESULT";

        public static final String  CHECK     = "CHECK";

        public void defineCharacteristics(){

                setName(Sextante.getText("Convex Hull"));

                setGroup(Sextante.getText("gvSIG_Algorithms"));

                setGeneratesUserDefinedRasterOutput(false);

                try {

                        m_Parameters.addInputVectorLayer(LAYER, 

                                                                                                Sextante.getText("Input_layer"), 

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG, 

                                                                                                true);

                        addOutputVectorLayer(RESULT,

                                                                        Sextante.getText("ConvexHull"),

                                                                        OutputVectorLayer.SHAPE_TYPE_POLYGON);

                        m_Parameters.addBoolean(CHECK, Sextante.getText("Selected_geometries"), false);

                } catch (RepeatedParameterNameException e) {

                        Sextante.addErrorToLog(e);

                }

        }

        /*

         * (non-Javadoc)

         * @see es.unex.sextante.core.GeoAlgorithm#processAlgorithm()

         */

        @SuppressWarnings("unchecked")

        public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

                IVectorLayer input = m_Parameters.getParameterValueAsVectorLayer(LAYER);

                boolean selectedGeom = m_Parameters.getParameter(CHECK).getParameterValueAsBoolean();

                FeatureStore store = null;

                if(input instanceof gvVectorLayer)

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

                else

                        return false;

                ConvexHullOperation convexHullOperation = new ConvexHullOperation();

                FeatureSet features = null;

                try {

                        features = store.getFeatureSet();

                        DisposableIterator it = null;

                        if(selectedGeom) {

                                DataSet ds = store.getSelection();

                                it = ((DefaultFeatureSelection)ds).iterator();

                                //if(!it.hasNext())Mensaje de no hay geometrias seleccionadas

                        } else

                                it = features.iterator();

                        while( it.hasNext() ) {

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

                                List geomList = feature.getGeometries();

                                if(geomList == null) {

                                        Geometry geom = feature.getDefaultGeometry();

                                        if(geom != null)

                                                convexHullOperation.invoke(geom);

                                        continue;

                                }

                                Iterator<Geometry> itGeom = geomList.iterator();

                                while(itGeom.hasNext()) {

                                        Geometry g = itGeom.next();

                                        convexHullOperation.invoke(g);

                                }

                        }

                        Geometry g = convexHullOperation.getGeometry();

                        if(g == null)

                                return false;

                        String[] sNames = {"ID"};

                        Class [] types = {Integer.class};

                        IVectorLayer output = getNewVectorLayer(RESULT,

                                                                                                        Sextante.getText("ConvexHull"),

                                                                                                        OutputVectorLayer.SHAPE_TYPE_POLYGON, types, sNames);

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

                        output.addFeature(jtsGeom, new Object[]{new Integer(0)});

                        it.dispose();

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                return !m_Task.isCanceled();

        }

}