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

import java.util.Iterator;

import java.util.List;

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

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

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

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.geom.Geometry;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

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;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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 AbstractSextanteGeoProcess {

    private static final Logger log = LoggerFactory.getLogger(ConvexHullAlgorithm.class);

    public static final String LAYER = "LAYER";

    public static final String RESULT = "RESULT";

    public static final String CHECK = "CHECK";

    public void defineCharacteristics() {

        setName(getTranslation("ConvexHull"));

        setGroup(getTranslation("basic_vect_algorithms"));

        // setGeneratesUserDefinedRasterOutput(false);

        try {

            m_Parameters.addInputVectorLayer(LAYER,

                    getTranslation("Input_layer"), IVectorLayer.SHAPE_TYPE_WRONG,

                    true);

            addOutputVectorLayer(RESULT, getTranslation("ConvexHull"),

                    OutputVectorLayer.SHAPE_TYPE_POLYGON);

            m_Parameters.addBoolean(CHECK,

                    getTranslation("Selected_geometries_convex_hull"), false);

        } catch (RepeatedParameterNameException e) {

            Sextante.addErrorToLog(e);

        }

    }

    /*

     * (non-Javadoc)

     *

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

     */

    @SuppressWarnings("unchecked")

    public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

        if (existsOutPutFile(ConvexHullAlgorithm.RESULT, 0)) {

            throw new GeoAlgorithmExecutionException(getTranslation("file_exists"));

        }

        IVectorLayer input = m_Parameters.getParameterValueAsVectorLayer(LAYER);

        boolean selectedGeom

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

        FeatureStore store = null;

        if (input instanceof FlyrVectIVectorLayer) {

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

        } else {

            return false;

        }

        ConvexHullOperation convexHullOperation = new ConvexHullOperation();

        FeatureSet features = null;

        try {

            Iterator it = null;

            if (selectedGeom) {

                features = store.getFeatureSet();

                FeatureSelection ds = store.getFeatureSelection();

                it = ds.iterator();

            } else {

                FeatureQuery query = getQueryFromAnalysisExtent(m_AnalysisExtent, store);

                features = store.getFeatureSet(query);

                it = features.iterator();

            }

            int numberOfFeatures = (int) features.getSize();

            int iCount = 0;

            while (it.hasNext()) {

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

                List geomList = feature.getGeometries();

                setProgress(iCount, numberOfFeatures);

                iCount++;

                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, getTranslation("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) {

            log.error("", e);

        } catch (CreateEnvelopeException e) {

            log.error("Error creating envelope", e);

        }

        return !m_Task.isCanceled();

    }

}