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

import java.util.ArrayList;

import javax.swing.JOptionPane;

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

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

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

import es.unex.sextante.additionalInfo.AdditionalInfoMultipleInput;

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.gui.algorithm.GeoAlgorithmParametersPanel;

import es.unex.sextante.outputs.OutputVectorLayer;

/**

 * Merge algorithm

 *

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

 */

public class MergeAlgorithm 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 LAYERS = "LAYERS";

    public static final String FIELDLAYER = "LAYER";

    /*

         * (non-Javadoc)

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

     */

    public void defineCharacteristics() {

        setName(getTranslation("Merge"));

        setGroup(getTranslation("basic_vect_algorithms"));

        // setGeneratesUserDefinedRasterOutput(false);

        try {

            m_Parameters.addMultipleInput(LAYERS,

                    getTranslation("Input_layers"),

                    AdditionalInfoMultipleInput.DATA_TYPE_VECTOR_ANY,

                    true);

            m_Parameters.addInputVectorLayer(FIELDLAYER,

                    getTranslation("Fields"),

                    IVectorLayer.SHAPE_TYPE_WRONG,

                    true);

            addOutputVectorLayer(RESULT_POL, getTranslation("Merge_polygon"),

                    OutputVectorLayer.SHAPE_TYPE_POLYGON);

            addOutputVectorLayer(RESULT_LINE, getTranslation("Merge_line"),

                    OutputVectorLayer.SHAPE_TYPE_LINE);

            addOutputVectorLayer(RESULT_POINT, getTranslation("Merge_point"),

                    OutputVectorLayer.SHAPE_TYPE_POINT);

        } catch (RepeatedParameterNameException e) {

            Sextante.addErrorToLog(e);

        }

    }

    /*

         * (non-Javadoc)

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

     */

    @SuppressWarnings("unchecked")

    public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

        if (existsOutPutFile(RESULT_POL, 0)

                || existsOutPutFile(RESULT_LINE, 0)

                || existsOutPutFile(RESULT_POINT, 0)) {

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

        }

        IVectorLayer layer = m_Parameters.getParameterValueAsVectorLayer(FIELDLAYER);

        ArrayList<IVectorLayer> layers = m_Parameters.getParameterValueAsArrayList(LAYERS);

        FeatureStore storeLayer = null;

        if (layer instanceof FlyrVectIVectorLayer) {

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

        } else {

            return false;

        }

        try {

            //Gets the list of FeatureStore

            ArrayList<FeatureStore> featureStoreList = new ArrayList<FeatureStore>();

            for (int i = 0; i < layers.size(); i++) {

                IVectorLayer lyr = layers.get(i);

                if (lyr instanceof FlyrVectIVectorLayer && layer.getShapeType() == lyr.getShapeType()) {

                    featureStoreList.add(((FlyrVectIVectorLayer) lyr).getFeatureStore());

                } else {

                    JOptionPane.showMessageDialog(null,

                            getTranslation("layers_type_are_different"),

                            "Error",

                            JOptionPane.WARNING_MESSAGE);

                    return false;

                }

            }

            //Builds the output FeatureStore

            FeatureSet features = null;

            features = storeLayer.getFeatureSet();

            FeatureType featureType = features.getDefaultFeatureType();

            /* FeatureStore outFeatStore =

                buildOutPutStore(featureType, layer.getShapeType(),

                    getTranslation("Merge"), RESULT);

                        //Calls the operation

                        GeometryOperation operation = new MergeOperation(layer, this);

            operation.setTaskStatus(getStatus());

                        operation.computesGeometryOperationInAList(featureStoreList, outFeatStore, attrNames, false, true);

             */

            GeometryOperation operation = new MergeOperation(layer, this);

            operation.setTaskStatus(getStatus());

            if (isPolygon(storeLayer) || isUndefined(storeLayer)) {

                FeatureStore outFeatStore

                        = buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_POLYGON,

                                getTranslation("Merge_polygon"), RESULT_POL);

                operation.computesGeometryOperationInAList(featureStoreList, outFeatStore, attrNames,

                        false, false, true);

            } else {

                buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_POLYGON,

                        getTranslation("Null_polygon"), RESULT_POL);

            }

            if (isLine(storeLayer) || isUndefined(storeLayer)) {

                FeatureStore outFeatStore

                        = buildOutPutStore(featureType, IVectorLayer.SHAPE_TYPE_LINE,

                                getTranslation("Merge_line"), RESULT_LINE);

                operation.computesGeometryOperationInAList(featureStoreList, outFeatStore, attrNames,

                        false, false, true);

            } else {

                buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_LINE,

                        getTranslation("Null_line"), RESULT_LINE);

            }

            if (isPoint(storeLayer) || isUndefined(storeLayer)) {

                FeatureStore outFeatStore

                        = buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_POINT,

                                getTranslation("Merge_point"), RESULT_POINT);

                operation.computesGeometryOperationInAList(featureStoreList, outFeatStore, attrNames,

                        false, false, true);

            } else {

                buildOutPutStore(featureType.getCopy(), IVectorLayer.SHAPE_TYPE_POINT,

                        getTranslation("Null_point"), RESULT_POINT);

            }

        } catch (DataException e) {

            Sextante.addErrorToLog(e);

            return false;

        }

        if (getTaskMonitor().isCanceled()) {

            return false;

        }

        return true;

    }

    @Override

    public Class<? extends GeoAlgorithmParametersPanel> getCustomParametersPanelClass() {

        return MergeParametersPanel.class;

    }

}