| gvSIG desktop 1

/*

 * Created on 23-feb-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: DissolveGeoprocess.java 4571 2006-03-23 21:07:32Z azabala $

* $Log$

* Revision 1.8  2006-03-23 21:03:45  azabala

* *** empty log message ***

*

* Revision 1.7  2006/03/17 19:53:14  azabala

* *** empty log message ***

*

* Revision 1.6  2006/03/15 18:33:24  azabala

* *** empty log message ***

*

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

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

*

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

* *** empty log message ***

*

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

* *** empty log message ***

*

* Revision 1.2  2006/03/05 19:58:20  azabala

* *** empty log message ***

*

* Revision 1.1  2006/02/26 20:53:44  azabala

* *** empty log message ***

*

*

*/

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

import java.util.ArrayList;

import java.util.Iterator;

import java.util.Map;

import com.iver.andami.PluginServices;

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

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

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

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

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

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

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

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

import com.iver.cit.gvsig.fmap.operations.CancellableMonitorable;

import com.iver.cit.gvsig.fmap.operations.DefaultCancellableMonitorable;

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

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

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

import com.iver.gvsig.geoprocessing.impl.AbstractGeoprocess;

import com.iver.gvsig.geoprocessing.impl.jtsprocessors.FeaturePersisterProcessor2;

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

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

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

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

import com.iver.utiles.swing.threads.IMonitorableTask;

/**

 * Processes each geometry of a polygonal vectorial layer, looking for

 * its adjacent polygons. If this adjacent polygons has the same specified 

 * field value as processed geometry, this polygons will be dissolved.

 * 

 * 

 * @author azabala

 *

 */

public class DissolveGeoprocess extends AbstractGeoprocess

                                                        implements IOneLayerGeoprocess {

        private String dissolveField;

        private boolean dissolveOnlySelection;

        private LayerDefinition resultLayerDefinition;

        private Map fields_functions;

        private DissolveVisitor visitor;

        /**

         * Constructor.

         * 

         * @param inputLayer Layer whose geometries we are 

         * going to dissolve

         */

        public DissolveGeoprocess(FLyrVect inputLayer, String dissolveField) {

                setFirstOperand(inputLayer);

                this.dissolveField = dissolveField;

        }

        public void setFieldsFunctions(Map fieldsFunctions){

                this.fields_functions = fieldsFunctions;

        }

        public void setFirstOperand(FLyrVect inputLayer) {

                this.firstLayer = inputLayer;

        }

        public void setParameters(Map params) throws GeoprocessException {

                Boolean onlySelection =

                        (Boolean) params.get("layer_selection");

                if(onlySelection != null)

                        dissolveOnlySelection = onlySelection.booleanValue();

        }

        public void checkPreconditions() throws GeoprocessException {

                if (firstLayer == null)

                        throw new GeoprocessException("Buffer con capa de entrada a null");

                if(this.writer == null || 

                   this.schemaManager == null){

                        throw new GeoprocessException("Operacion de dissolve sin especificar capa de resultados");

                }

                try {

                        if(firstLayer.getShapeType() != XTypes.POLYGON)

                                throw new GeoprocessException("La capa a disolver debe ser de pol?gonos");

                } catch (DriverException e) {

                        throw new GeoprocessException("Error intentando verificar el tipo de geometria de la capa a disolver");

                } 

                if(this.dissolveField == null)

                        throw new GeoprocessException("No se ha proporcionado el campo para dissolver");

        }

        public void process() throws GeoprocessException {

                try {

                        FeaturePersisterProcessor2 processor =

                                new FeaturePersisterProcessor2(writer);

                        visitor = new DissolveVisitor(this.dissolveField, processor);

                        visitor.setDissolvedAttributesInfo(fields_functions);

                        Strategy strategy =

                                StrategyManager.getStrategy(firstLayer);

                        visitor.setStrategy(strategy);

                        if(this.dissolveOnlySelection){

                                strategy.process(visitor, firstLayer.

                                                                                           getRecordset().

                                                                                           getSelection());

                        }else{

                                strategy.process(visitor);

                        }

                } catch (DriverException e) {

                        throw new GeoprocessException("Error accediendo a los datos durante un dissolve");

                } catch (VisitException e) {

                        throw new GeoprocessException("Error al procesar los elementos de la capa de entrada durante un dissolve");

                }

        }

        public void cancel() {

                try {

                        schemaManager.drop();

                } catch (EditionException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

        }

        public ILayerDefinition createLayerDefinition() {

                if(resultLayerDefinition == null){

                        ArrayList fields = new ArrayList();

                        resultLayerDefinition = new SHPLayerDefinition();

                        resultLayerDefinition.setShapeType(XTypes.POLYGON);

                        Iterator fieldsIt = fields_functions.keySet().iterator();

                        while(fieldsIt.hasNext()){

                                String field = (String) fieldsIt.next();

                                SummarizationFunction[] functions =

                                        (SummarizationFunction[]) fields_functions.get(field);

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

                                        FieldDescription description = 

                                                new FieldDescription();

                                        description.setFieldLength(10);

                                        description.setFieldDecimalCount(4);

                                        //to avoid truncation of field names (f.example shp)

                                        //we only catch three first letters

                                        String shortName = null;

                                        if(field.length() > 3)

                                                shortName = field.substring(0,3);

                                        else

                                                shortName = field;

                                        description.setFieldName(

                                                        shortName + "_" + functions[i].toString());

                                        description.setFieldType(XTypes.DOUBLE);

                                        fields.add(description);

                                }//for

                        }//while

                        FieldDescription[] fieldsDesc = null;

                        if(fields.size() == 0){

                                fieldsDesc = new FieldDescription[0];

                        }else{

                                fieldsDesc = new FieldDescription[fields.size()];

                                fields.toArray(fieldsDesc);

                        }

                        resultLayerDefinition.setFieldsDesc(fieldsDesc);

                }

                return resultLayerDefinition;

        }

        public IMonitorableTask createTask() {

                try {

                        return new DissolveMonitorableTask();

                } catch (DriverIOException e) {

                        return null;

                }

        }

        /**

         * IMonitorableTask that allows to run diff geoprocess in background,

         * with cancelation requests.

         * 

         * @author azabala

         * 

         */

        class DissolveMonitorableTask implements IMonitorableTask {

                private CancellableMonitorable cancelMonitor = null;

                String DISSOLVE_MESSAGE = PluginServices.getText(this, "Mensaje_dissolve");

                String DISSOLVE_NOTE = PluginServices.getText(this, "Mensaje_procesando_dissolves");

                String OF = PluginServices.getText(this, "De");

                private boolean finished = false;

                DissolveMonitorableTask() throws DriverIOException {

                        initialize();

                }

                void initialize() throws DriverIOException{

                        cancelMonitor = createCancelMonitor();

                }

                private CancellableMonitorable createCancelMonitor() throws DriverIOException{

                        DefaultCancellableMonitorable monitor = new 

                                                        DefaultCancellableMonitorable();

                        monitor.setInitialStep(0);

                        //monitor.setDeterminatedProcess(false);

                        monitor.setFinalStep(firstLayer.getSource().getShapeCount());

                        monitor.setDeterminatedProcess(true);

                        return monitor;

                }

                public int getInitialStep() {

                        return cancelMonitor.getInitialStep();

                }

                public int getFinishStep() {

                        return cancelMonitor.getFinalStep();

                }

                public int getCurrentStep() {

                        //return cancelMonitor.getCurrentStep();

                        if(visitor == null)

                                return getInitialStep();

                        else

                                return visitor.getNumProcessedGeometries();

                }

                public String getStatusMessage() {

                        return DISSOLVE_MESSAGE;

                }

                public String getNote() {

                        return DISSOLVE_NOTE + " "+

                        getCurrentStep() + " " +

                        OF + " " +

                        getFinishStep();

                }

                public void cancel() {

                        ((DefaultCancellableMonitorable) cancelMonitor).setCanceled(true);

                }

                public void run() throws GeoprocessException {

                        try {

                                FeaturePersisterProcessor2 processor =

                                        new FeaturePersisterProcessor2(writer);

                                visitor = new DissolveVisitor(dissolveField, processor);

                                visitor.setDissolvedAttributesInfo(fields_functions);

                                Strategy strategy =

                                        StrategyManager.getStrategy(firstLayer);

                                visitor.setStrategy(strategy);

                                if(dissolveOnlySelection){

                                        strategy.process(visitor, firstLayer.

                                                                                                   getRecordset().

                                                                                                   getSelection(),

                                                                                                   cancelMonitor);

                                }else{

                                        strategy.process(visitor, cancelMonitor);

                                }

                        } catch (DriverException e) {

                                throw new GeoprocessException("Error accediendo a los datos durante un dissolve");

                        } catch (VisitException e) {

                                throw new GeoprocessException("Error al procesar los elementos de la capa de entrada durante un dissolve");

                        } finally{

                                finished = true;

                        }

                }

                public boolean isDefined() {

                        return cancelMonitor.isDeterminatedProcess();

                }

                public boolean isCanceled() {

                        return cancelMonitor.isCanceled();

                }

                public boolean isFinished() {

                        return finished;

                }

        }

}