| gvSIG desktop 1

/*

 * Created on 01-mar-2006

 *

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2004 IVER T.I. and 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.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

 *

 * $Id: NearestSpatialJoinVisitor.java 4447 2006-03-15 18:35:56Z azabala $

 * $Log$

 * Revision 1.4  2006-03-15 18:34:31  azabala

 * *** empty log message ***

 *

 * Revision 1.3  2006/03/14 18:32:46  fjp

 * Cambio con LayerDefinition para que sea compatible con la definici?n de tablas tambi?n.

 *

 * Revision 1.2  2006/03/07 21:01:33  azabala

 * *** empty log message ***

 *

 * Revision 1.1  2006/03/06 19:48:39  azabala

 * *** empty log message ***

 *

 * Revision 1.1  2006/03/05 19:59:47  azabala

 * *** empty log message ***

 *

 *

 */

package com.iver.gvsig.geoprocessing.impl.spatialjoin;

import com.hardcode.gdbms.engine.values.Value;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.core.IFeature;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.drivers.ILayerDefinition;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

import com.iver.cit.gvsig.fmap.operations.strategies.Strategy;

import com.iver.cit.gvsig.fmap.operations.strategies.VisitException;

import com.iver.gvsig.geoprocessing.model.FeatureProcessor;

import com.iver.gvsig.geoprocessing.model.GeoprocessException;

import com.iver.gvsig.geoprocessing.schemabuilder.DefinitionUtils;

import com.iver.gvsig.geoprocessing.schemabuilder.FeatureFactory;

/**

 * This visitor implement Nearest Geometry Spatial Join.

 * 

 * It is a particular case of 1-1 relationship

 * (a feature of layer A always be related to a feature of layer B)

 * 

 * If two features of layer B are at the same distance of a feature of layer A,

 * we will took the first one analized.

 * @author azabala

 *

 */

public class NearestSpatialJoinVisitor implements SpatialJoinVisitor {

        /**

         * Needed to create layer definition

         */

        private FLyrVect sourceLayer;

        /**

         * Reads data of features for the source layer

         */

        private SelectableDataSource sourceRecordset;

        /**

         * Reads data of features for the target layer

         */

        private FLyrVect targetLayer;

        /**

         * Recordset of this layer

         */

        private SelectableDataSource targetRecordset;

        /**

         * Strategy to do querys against target Layer

         */

        private Strategy strategy;

        /**

         * Processes results of dissolve operations (save them in a file, or cache

         * them in memory, etc)

         */

        private FeatureProcessor featureProcessor;

        /**

         * It looks for nearest feature to a given feature

         */

        private NearestFeatureVisitor lookForNearestVisitor;

        private ILayerDefinition layerDefinition;

        private boolean onlySecondLayerSelection;

        /**

         * Constructor. It receives layer with which we want to do a spatial join

         * 

         * @param targetRecordset

         */

        public NearestSpatialJoinVisitor(FLyrVect sourceLayer,

                        FLyrVect targetLayer,

                        FeatureProcessor processor) {

                this.sourceLayer = sourceLayer;

                this.sourceRecordset = sourceLayer.getRecordset();

                this.targetLayer = targetLayer;

                this.targetRecordset = targetLayer.getRecordset();

                this.featureProcessor = processor;

                this.lookForNearestVisitor = new NearestFeatureVisitor();

        }

        /**

         * Processes a Feature of source layer, looking for its nearest feature of

         * target layer and taking attributes from it

         */

        public void visit(IGeometry g, int sourceIndex) throws VisitException {

                lookForNearestVisitor.setQueryGeometry(g.toJTSGeometry());

                try {

                        if(onlySecondLayerSelection)

                                strategy.process(lookForNearestVisitor, targetRecordset.getSelection());

                        else

                                strategy.process(lookForNearestVisitor);

                        int targetIndex = lookForNearestVisitor.getNearestFeatureIndex();

                        IFeature joinedFeature = createFeature(g, sourceIndex, targetIndex);

                        this.featureProcessor.processFeature(joinedFeature);

                } catch (DriverException e) {

                        throw new VisitException(

                                        "Problemas accediendo a los datos durante un nearest spatial join");

                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {

                        throw new VisitException(

                        "Problemas accediendo a los datos durante un nearest spatial join");

                }

        }

        public void stop(FLayer layer) {

                this.featureProcessor.finish();

        }

        public boolean start(FLayer layer) {

                try {

                        this.featureProcessor.start();

                } catch (EditionException e) {

                        return false;

                }

                return true;

        }

        public String getProcessDescription() {

                return "Spatial joining by nearest criteria";

        }

        private IFeature createFeature(IGeometry g, int sourceLayerIndex, int targetLayerIndex) throws DriverException, com.hardcode.gdbms.engine.data.driver.DriverException {

                IFeature solution = null;

                int numFieldsA = sourceRecordset.getFieldCount();

                int numFieldsB = targetRecordset.getFieldCount();

                Value[] featureAttr = new Value[numFieldsA + numFieldsB];

                for (int indexField = 0; indexField < numFieldsA; indexField++) {

                        featureAttr[indexField] = sourceRecordset

                                .getFieldValue(sourceLayerIndex,indexField);

                }

                for (int indexFieldB = 0; indexFieldB < numFieldsB; indexFieldB++) {

                        featureAttr[numFieldsA + indexFieldB] = 

                                targetRecordset.getFieldValue(targetLayerIndex, indexFieldB);

                }

                solution = FeatureFactory.createFeature(featureAttr, g);

                return solution;

        }

        public ILayerDefinition getResultLayerDefinition() throws GeoprocessException{

                if(layerDefinition == null){

                        try {

                                layerDefinition = DefinitionUtils.

                                mergeLayerDefinitions(sourceLayer, targetLayer);

                        } catch (Exception e) {

                                throw new GeoprocessException("Problemas al crear el esquema de la capa solucion de un spatial join");

                        } 

                }

                return layerDefinition;

        }

        public void setFeatureProcessor(FeatureProcessor processor) {

                this.featureProcessor = processor;

        }

        public void setCancelableStrategy(Strategy secondLyrStrategy) {

                this.strategy = secondLyrStrategy;

        }

        public void setOnlySecondLyrSelection(boolean onlySecondLayerSelection) {

                this.onlySecondLayerSelection = onlySecondLayerSelection;

        }

}