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

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

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

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

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

import org.gvsig.geoprocess.lib.sextante.AbstractSextanteGeoProcess;

import org.gvsig.geoprocess.lib.sextante.dataObjects.FlyrVectIVectorLayer;

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;

/**

 * Intersection algorithm

 * <UL>

 * <LI>Pol-Pol: 3 layers (polygon, point, line)</LI>

 * <LI>Pol-Line: 2 layers (point, line)</LI>

 * <LI>Pol-Point: 1 layer (point)</LI>

 * <LI>Line-Point: 1 layer (point)</LI>

 * <LI>Line-Line: 2 layers (point, line)</LI>

 * <LI>Point-Point: 1 layer (point)</LI>

 * </UL>

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

 */

public class IntersectionAlgorithm extends AbstractSextanteGeoProcess {

        public static final String  RESULT_POL      = "RESULT_POL";

        public static final String  RESULT_POINT    = "RESULT_POINT";

        public static final String  RESULT_LINE     = "RESULT_LINE";

        public static final String  LAYER           = "LAYER";

        public static final String  INTER           = "INTER";

        public static final String  CHECK           = "CHECK";

        /*

         * (non-Javadoc)

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

         */

        public void defineCharacteristics() {

        setName(getTranslation("Intersection"));

        setGroup(getTranslation("basic_vect_algorithms"));

        // setGeneratesUserDefinedRasterOutput(false);

                try {

                        m_Parameters.addInputVectorLayer(LAYER, 

                getTranslation("Input_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG, 

                                                                                                true);

                        m_Parameters.addInputVectorLayer(INTER, 

                getTranslation("Overlays_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG, 

                                                                                                true);

            m_Parameters.addBoolean(CHECK,

                getTranslation("Selected_geometries"), false);

                } catch (RepeatedParameterNameException e) {

                        Sextante.addErrorToLog(e);

                }

                addOutputVectorLayer(RESULT_POL, getTranslation("Intersection_polygon"),

                                                                OutputVectorLayer.SHAPE_TYPE_POLYGON);

                addOutputVectorLayer(RESULT_LINE, getTranslation("Intersection_line"),

                                                                OutputVectorLayer.SHAPE_TYPE_LINE);

                addOutputVectorLayer(RESULT_POINT, getTranslation("Intersection_point"),

                                                                OutputVectorLayer.SHAPE_TYPE_POINT);

        }

        /*

         * (non-Javadoc)

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

         */

        public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

                if(existsOutPutFile(IntersectionAlgorithm.RESULT_LINE, 0)) {

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

            }

                if(existsOutPutFile(IntersectionAlgorithm.RESULT_POINT, 0)) {

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

            }

                if(existsOutPutFile(IntersectionAlgorithm.RESULT_LINE, 0)) {

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

            }

                IVectorLayer inter = m_Parameters.getParameterValueAsVectorLayer(INTER);

                IVectorLayer layer = m_Parameters.getParameterValueAsVectorLayer(LAYER);

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

                try {

                        computesIntersection(layer, inter, layer.getShapeType(), selectedGeom);

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                        return false;

                }

                return true;

        }

        /**

         * Builds a layer with the intersection between the input layer and the templateGeometry

         * @param layer

         *        Input layer

         * @param templateGeometry

         * @param shapeType

         *        Output shape type

         * @param selectedGeom

         *        If it's true only selected geometries will be computed

         * @throws GeoAlgorithmExecutionException 

         */

        private void computesIntersection(IVectorLayer layer,

                                                                IVectorLayer overlay,

                                                                int shapeType, 

                                                                boolean selectedGeom) throws DataException, GeoAlgorithmExecutionException {

                FeatureStore storeLayer = null;

                FeatureStore storeOverlay = null;

                if(layer instanceof FlyrVectIVectorLayer && 

                        overlay instanceof FlyrVectIVectorLayer) {

                        storeLayer = ((FlyrVectIVectorLayer)layer).getFeatureStore();

                        storeOverlay = ((FlyrVectIVectorLayer)overlay).getFeatureStore();

                } else

                        return;

                FeatureSet features1 = null;

                features1 = storeLayer.getFeatureSet();

                FeatureType featureType1 = features1.getDefaultFeatureType();

                FeatureSet features2 = null;

                features2 = storeOverlay.getFeatureSet();

                FeatureType featureType2 = features2.getDefaultFeatureType();

                FeatureStore outFeatStorePol = null;

                FeatureStore outFeatStoreLine = null;

                FeatureStore outFeatStorePoint = null;

                IntersectionOperation operation = new IntersectionOperation(storeOverlay, this);

                operation.setTaskStatus(getStatus());

                //La de puntos se genera siempre

                outFeatStorePoint =

                        buildOutPutStoreFromUnion(featureType1, featureType2,

                                        IVectorLayer.SHAPE_TYPE_POINT,

                                        getTranslation("Intersection_point"), RESULT_POINT);

                getStatus().setTitle("Point");

                operation.computesGeometryOperation(storeLayer, outFeatStorePoint, attrNames, selectedGeom, true);

                //La de pol?gonos solo si es intersecci?n entre pol?gonos

                if (isPolygon(storeLayer) && isPolygon(storeOverlay)) {

                        outFeatStorePol =

              buildOutPutStoreFromUnion(featureType1, featureType2,

                              IVectorLayer.SHAPE_TYPE_POLYGON, getTranslation("Intersection_polygon"), RESULT_POL);

                        getStatus().setTitle("Polygon");

                        operation.computesGeometryOperation(storeLayer, outFeatStorePol, attrNames, selectedGeom, true);

                }

                //La capa de l?neas se genera cuando ning?na de las dos es de puntos

                if (!isPoint(storeLayer) && !isPoint(storeOverlay)) {

                        outFeatStoreLine =

              buildOutPutStoreFromUnion(featureType1, featureType2,

                              IVectorLayer.SHAPE_TYPE_LINE, getTranslation("Intersection_line"), RESULT_LINE);

                        getStatus().setTitle("Line");

                        operation.computesGeometryOperation(storeLayer, outFeatStoreLine, attrNames, selectedGeom, true);

                }

                if(outFeatStorePol == null) {

                        getNewVectorLayer(RESULT_POL, getTranslation("Null_polygon"),

                    OutputVectorLayer.SHAPE_TYPE_POLYGON, new Class[]{Integer.class}, new String[]{""});

                }

                if(outFeatStoreLine == null) {

                        getNewVectorLayer(RESULT_LINE, getTranslation("Null_line"),

                    OutputVectorLayer.SHAPE_TYPE_LINE, new Class[]{Integer.class}, new String[]{""});

                }

                if(outFeatStorePoint == null) {

                        getNewVectorLayer(RESULT_POINT, getTranslation("Null_point"),

                    OutputVectorLayer.SHAPE_TYPE_POINT, new Class[]{Integer.class}, new String[]{""});

                }

        }

}