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.

 */

package org.gvsig.sextante.app.algorithm.spatialjoin;

import java.util.HashMap;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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.algorithm.spatialjoin.SpatiallyIndexedSpatialJoinOperation;

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

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

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;

/**

 * Spatial join algorithm

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

 */

public class SpatialJoinAlgorithm extends gvGeoAlgorithm {

        public static final String        RESULT            = "RESULT";

        public static final String        LAYER1            = "LAYER1";

        public static final String        LAYER2            = "LAYER2";

        public static final String        SELECTED_GEOM     = "SELECTED_GEOM";

        public static final String        NEAREST           = "NEAREST";

        public static final String        FUNCTION_LIST     = "FUNCTION_LIST";

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

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

        private HashMap<String, String>   funcMap           = new HashMap<String, String>();

        /*

         * (non-Javadoc)

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

         */

        public void defineCharacteristics(){

                setName(Sextante.getText("Spatialjoin"));

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

                setGeneratesUserDefinedRasterOutput(false);

                try {

                        m_Parameters.addInputVectorLayer(LAYER1,

                                                                                                Sextante.getText("Input_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG,

                                                                                                true);

                        m_Parameters.addInputVectorLayer(LAYER2,

                                                                                                Sextante.getText( "Input_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_WRONG,

                                                                                                true);

                        m_Parameters.addBoolean(SELECTED_GEOM, Sextante.getText("Selected_geometries"), false);

                        m_Parameters.addBoolean(NEAREST, Sextante.getText("use_the_nearest"), false);

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

                } catch (RepeatedParameterNameException e) {

                        Sextante.addErrorToLog(e);

                }

                addOutputVectorLayer(RESULT,

                                                                Sextante.getText("Spatialjoin"),

                                                                OutputVectorLayer.SHAPE_TYPE_UNDEFINED);

        }

        /*

         * (non-Javadoc)

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

         */

        public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

                IVectorLayer layer1 = m_Parameters.getParameterValueAsVectorLayer(LAYER1);

                IVectorLayer layer2 = m_Parameters.getParameterValueAsVectorLayer(LAYER2);

                boolean selectedGeom = m_Parameters.getParameterValueAsBoolean(SELECTED_GEOM);

                boolean nearest = m_Parameters.getParameterValueAsBoolean(NEAREST);

                String functionList = m_Parameters.getParameterValueAsString(FUNCTION_LIST);

                loadSummary(functionList);

                gvVectorLayer lyr1 = null;

                gvVectorLayer lyr2 = null;

                if(layer2 instanceof gvVectorLayer && layer1 instanceof gvVectorLayer) {

                        lyr2 = ((gvVectorLayer)layer2);

                        lyr1 = ((gvVectorLayer)layer1);

                } else

                        return false;

                DataManager dataManager = DALLocator.getDataManager();

                //Builds a new JSIRTree index. To do that we have to replace the default index and

                //when the operation has finished then it restores the original default index

                String indexName = "GEOMETRY";

                try {

                        indexName = lyr2.getFeatureStore().getDefaultFeatureType().getDefaultGeometryAttributeName();

                        FeatureAttributeDescriptor fat = lyr2.getFeatureStore().getDefaultFeatureType().getAttributeDescriptor(indexName);

                        String defaultIndex = dataManager.getDefaultFeatureIndexProviderName(fat.getDataType());

                        dataManager.setDefaultFeatureIndexProviderName(fat.getDataType(), "JSIRTree");

                        lyr2.getFeatureStore().createIndex(lyr2.getFeatureStore().getDefaultFeatureType(), indexName, indexName + "_idx");

                        dataManager.setDefaultFeatureIndexProviderName(fat.getDataType(), defaultIndex);

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                //Builds the output and computes the operation

                try {

                        if(nearest) {

                                FeatureSet features = null;

                                features = lyr1.getFeatureStore().getFeatureSet();

                                FeatureType featureType1 = features.getDefaultFeatureType();

                                features = lyr2.getFeatureStore().getFeatureSet();

                                FeatureType featureType2 = features.getDefaultFeatureType();

                                FeatureStore outFeatStore = buildOutPutStoreFromUnion(featureType1,

                                                featureType2,

                                                lyr1.getShapeType(),

                                                Sextante.getText("SpatialJoin"),

                                                RESULT,

                                                "DIST",

                                                Double.class);

                                SpatiallyIndexedSpatialJoinOperation operation = new SpatiallyIndexedSpatialJoinOperation(lyr2, indexName + "_idx");

                                operation.setProgressModel(this);

                                operation.computesGeometryOperation(lyr1.getFeatureStore(),

                                                outFeatStore,

                                                attrNames,

                                                selectedGeom,

                                                true);

                        }

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                return true;

        }

        /**

         * 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;

                                        funcMap.put(Summary[pos], func[0]);

                                }

                        }

                }

        }

}