Application: gvSIG desktop » gvSIG geoprocess

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Iterator;

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

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

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.extension.core.gvGeoAlgorithm;

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

import es.unex.sextante.exceptions.UnsupportedOutputChannelException;

public class DissolveAlgorithm extends gvGeoAlgorithm {

	public static final String        RESULT            = "RESULT";

	public static final String        LAYER             = "LAYER";

	public static final String        FIELD             = "FIELD";

	public static final String        FUNCTIONS         = "FUNCTIONS";

	public static final String        SELECTED_GEOM     = "SELECTED_GEOM";

	public static final String        DISSOLV_ADJ       = "DISSOLV_ADJ";

	public static final String        FUNCTION_LIST     = "FUNCTION_LIST";

	public static final String        Summary[]         = {"Min", "Max", "Sum", "Avg"};

	private boolean                   funcList[]        = new boolean[Summary.length];

			m_Parameters.addString(FUNCTION_LIST, Sextante.getText("Function_list"));

		IVectorLayer layer = m_Parameters.getParameterValueAsVectorLayer(LAYER);

		loadSummary(functionList);

		FeatureStore storeLayer = null;

		if(layer instanceof gvVectorLayer)

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

		else

			return false;

		FeatureSet features = null;

		FeatureType featureType = null;

		try {

			features = storeLayer.getFeatureSet();

			featureType = features.getDefaultFeatureType();

		} catch (DataException e) {

			Sextante.addErrorToLog(e);

			return false;

		}

		FeatureStore outFeatStore = buildDissolvedOutPutStore(featureType, indexField, layer.getShapeType(),

														Sextante.getText("Dissolve"), RESULT);

		DissolveOperation op = new DissolveOperation();

		try {

			op.computesOperation(storeLayer, outFeatStore, attrNames, selectedGeom, true);

		} catch (DataException e) {

			Sextante.addErrorToLog(e);

			return false;

		}

	/**

	 * Checks if the parameter is in Summary list

	 * @param it

	 * @return the position in the list

	 */

	private int isInList(String it) {

		for (int i = 0; i < Summary.length; i++) {

			if(Summary[i].compareTo(it) == 0)

				return i;

		}

		return -1;

	}

	/**

	 * Loads the list of functions to use

	 * @param functionList

	 */

	private void loadSummary(String functionList) {

		String[] attrList = functionList.split(";");

		for (int i = 0; i < attrList.length; i++) {

			String[] func = attrList[i].split(",");

			for (int j = 1; j < func.length; j++) {

				int pos = isInList(func[j]);

				if(pos != -1)

					funcList[pos] = true;

			}

		}

	}

	/**

	 * Builds the output FeatureStore 

	 * @param featureType

	 * @return FeatureStore

	 */

	@SuppressWarnings("unchecked")

	protected FeatureStore buildDissolvedOutPutStore(FeatureType featureType1,

											int indexField,

											int shapeType,

											String sextanteLayerName, 

											String sextanteLayerLabel) {

		ArrayList<Class> typesList = new ArrayList<Class>();

		ArrayList<String> attr = new ArrayList<String>();

		attr.add("FID");

		typesList.add(Integer.class);

		FeatureAttributeDescriptor desc = featureType1.getAttributeDescriptor(indexField);

		attr.add(desc.getName());

		typesList.add(desc.getObjectClass());

		for (int i = 0; i < funcList.length; i++) 

			if(funcList[i])

				attr.add(Summary[i]);

		attrNames = new String[attr.size()];

		attr.toArray(attrNames);

		Class[] types = new Class[typesList.size()];

		typesList.toArray(types);

		try {

			IVectorLayer output = getNewVectorLayer(sextanteLayerLabel,

													sextanteLayerName,

													shapeType, types, attrNames);

			return ((gvVectorLayer)output).getFeatureStore();

		} catch (UnsupportedOutputChannelException e) {

			Sextante.addErrorToLog(e);

		}

		return null;

	}