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.

 *

 */

import java.util.ArrayList;

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

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

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

import es.unex.sextante.additionalInfo.AdditionalInfoMultipleInput;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.outputs.OutputVectorLayer;

/**

 * Merge algorithm

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

 */

public class MergeAlgorithm extends gvGeoAlgorithm {

	public static final String        RESULT            = "RESULT";

	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(Sextante.getText("Merge"));

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

		setGeneratesUserDefinedRasterOutput(false);

		try {

			m_Parameters.addMultipleInput(LAYERS, 

											Sextante.getText("Input_layers"), 

											AdditionalInfoMultipleInput.DATA_TYPE_VECTOR_ANY, 

											true);

			m_Parameters.addInputVectorLayer(FIELDLAYER, 

											Sextante.getText( "Fields"), 

											IVectorLayer.SHAPE_TYPE_WRONG, 

											true);

			addOutputVectorLayer(RESULT,

								Sextante.getText("Merge"),

								OutputVectorLayer.SHAPE_TYPE_UNDEFINED);

		} catch (RepeatedParameterNameException e) {

			Sextante.addErrorToLog(e);

		}

		setExternalParameters(new MergeParametersPanel());

	}

	@SuppressWarnings("unchecked")

	public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

		IVectorLayer layer = m_Parameters.getParameterValueAsVectorLayer(FIELDLAYER);

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

		FeatureStore storeLayer = null;

		if(layer instanceof gvVectorLayer)

			storeLayer = ((gvVectorLayer)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 gvVectorLayer && layer.getShapeType() == lyr.getShapeType())

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

				else

					return false;

			}

			//Builds the output FeatureStore

			FeatureSet features = null;

			features = storeLayer.getFeatureSet();

			FeatureType featureType = features.getDefaultFeatureType();

			FeatureStore outFeatStore = buildOutPutStore(featureType, layer.getShapeType(), Sextante.getText("Merge"), RESULT);

			//Calls the operation

			GeometryOperation operation = new MergeOperation(layers, layer);

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

		} catch (DataException e) {

			Sextante.addErrorToLog(e);

			return false;

		}