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 javax.swing.JOptionPane;

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

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;

import org.gvsig.tools.namestranslator.NamesTranslator;

/**

 * 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 SELECTGEOM_INPUT = "SELECTGEOM_INPUT";

    public static final String SELECTGEOM_OVERLAY = "SELECTGEOM_OVERLAY";

    /*

         * (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(SELECTGEOM_INPUT,

                    getTranslation("Selected_geometries_input_layer_inters"), false);

            m_Parameters.addBoolean(SELECTGEOM_OVERLAY,

                    getTranslation("Selected_geometries_overlay_layer_inters"), 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 selectedGeomInput = m_Parameters.getParameter(SELECTGEOM_INPUT).getParameterValueAsBoolean();

        boolean selectedGeomOverlay = m_Parameters.getParameter(SELECTGEOM_OVERLAY).getParameterValueAsBoolean();

        boolean error = false;

        try {

            error = computesIntersection(layer, inter, layer.getShapeType(), selectedGeomInput, selectedGeomOverlay);

        } catch (DataException e) {

            Sextante.addErrorToLog(e);

            return false;

        }

        if (getTaskMonitor().isCanceled()) {

            return false;

        }

        if (error) {

            JOptionPane.showMessageDialog(null,

                    getTranslation("problems_with_some_geometries"), "Error",

                    JOptionPane.WARNING_MESSAGE);

        }

        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 boolean computesIntersection(IVectorLayer layer,

            IVectorLayer overlay,

            int shapeType,

            boolean selectedGeomInput,

            boolean selectedGeomOverlay) 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 false;

        }

        FeatureType featureType1 = storeLayer.getDefaultFeatureType();

        FeatureType featureType2 = storeOverlay.getDefaultFeatureType();

        FeatureStore outFeatStorePol = null;

        FeatureStore outFeatStoreLine = null;

        FeatureStore outFeatStorePoint = null;

        IntersectionOperation operation = new IntersectionOperation(storeOverlay, this);

        operation.setTaskStatus(getStatus());

        this.namesTranslator = NamesTranslator.createTrimTranslator(11);

        //La de puntos se genera siempre

        outFeatStorePoint

                = buildOutPutStoreFromUnion(featureType1, featureType2,

                        IVectorLayer.SHAPE_TYPE_POINT,

                        getTranslation("Intersection_point"), RESULT_POINT);

        if (outFeatStorePoint != null) {

            getStatus().setTitle("Point");

            operation.computesGeometryOperation(storeLayer, outFeatStorePoint,

                    attrNames, selectedGeomInput, selectedGeomOverlay, 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);

            if (outFeatStorePol != null) {

                getStatus().setTitle("Polygon");

                operation.computesGeometryOperation(storeLayer,

                        outFeatStorePol, attrNames, selectedGeomInput,

                        selectedGeomOverlay, 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);

            if (outFeatStoreLine != null) {

                getStatus().setTitle("Line");

                operation.computesGeometryOperation(storeLayer,

                        outFeatStoreLine, attrNames, selectedGeomInput,

                        selectedGeomOverlay, 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[]{""});

        }

        return operation.getErrorInfo();

    }

}