Application: gvSIG desktop

/*

 * Created on 10-oct-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: LineCleanGeoprocess.java 8659 2006-11-09 21:08:32Z azabala $

* $Log$

* Revision 1.4  2006-11-09 21:08:32  azabala

* *** empty log message ***

*

* Revision 1.3  2006/10/19 16:06:48  azabala

* *** empty log message ***

*

* Revision 1.2  2006/10/17 18:25:53  azabala

* *** empty log message ***

*

* Revision 1.1  2006/10/10 18:50:17  azabala

* First version in CVS

*

*

*/

package com.iver.cit.gvsig.topology.lineclean.fmap;

import java.io.File;

import java.util.Map;

import com.iver.andami.PluginServices;

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

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.IWriter;

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

import com.iver.cit.gvsig.fmap.edition.writers.shp.MultiShpWriter;

import com.iver.cit.gvsig.fmap.edition.writers.shp.ShpWriter;

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

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

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

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

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

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

import com.iver.cit.gvsig.geoprocess.core.fmap.AbstractGeoprocess;

import com.iver.cit.gvsig.geoprocess.core.fmap.DefinitionUtils;

import com.iver.cit.gvsig.geoprocess.core.fmap.FeaturePersisterProcessor2;

import com.iver.cit.gvsig.geoprocess.core.fmap.GeoprocessException;

import com.iver.cit.gvsig.geoprocess.core.fmap.XTypes;

import com.iver.cit.gvsig.project.documents.view.gui.View;

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

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

public class LineCleanGeoprocess extends AbstractGeoprocess {

        /**

         * Schema of the result layer

         */

        private ILayerDefinition resultLayerDefinition;

        /**

         * flag to only clip selection of input layer

         */

        private boolean onlyFirstLayerSelection = false;

        /**

         * Processes features (writing them)

         */

        FeaturePersisterProcessor2 processor;

        /**

         * Writer to save in a temporal layer intersections

         */

        private IWriter intersectionsWriter;

        FeaturePersisterProcessor2 intersectsProcessor;

        public LineCleanGeoprocess(FLyrVect inputLayer){

                this.firstLayer = inputLayer;

        }

        public void setParameters(Map params) throws GeoprocessException {

                Boolean firstLayerSelection = (Boolean) params.get("firstlayerselection");

                if (firstLayerSelection != null)

                        this.onlyFirstLayerSelection = 

                                firstLayerSelection.booleanValue();

        }

        public void checkPreconditions() throws GeoprocessException {

                if (firstLayer == null)

                        throw new GeoprocessException("CLEAN: capa de entrada a null");

                if (this.writer == null || this.schemaManager == null) {

                        throw new GeoprocessException(

                                        "Operacion de CLEAN sin especificar capa de resultados");

                }

        }

        public void process() throws GeoprocessException {

                try {

                        createTask().run();

                } catch (Exception e) {

                        throw new GeoprocessException("Error al ejecutar el geoproceso");

                }

        }

        public ILayerDefinition createLayerDefinition() {

                if (resultLayerDefinition == null) {

                        try {

                                resultLayerDefinition = DefinitionUtils.

                                                        createLayerDefinition(firstLayer);

                        } catch (Exception e) {

                                e.printStackTrace();

                        }

                }

                return resultLayerDefinition;

        }

        class LineCleanTask extends AbstractMonitorableTask {

                private LineCleanTask() {

                        setInitialStep(0);

                        try {

                                if (onlyFirstLayerSelection) {

                                        int numSelected = firstLayer.getRecordset().getSelection()

                                                        .cardinality();

                                        setFinalStep(numSelected);

                                } else {

                                        int numShapes = firstLayer.getSource().getShapeCount();

                                        setFinalStep(numShapes);

                                }// else

                        } catch (com.iver.cit.gvsig.fmap.DriverException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        } catch (DriverIOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

                        setDeterminatedProcess(true);

                        setStatusMessage(PluginServices.getText(this,

                                        "LineClean._Progress_Message"));

                }

                public void run() throws Exception {

                        processor =

                                new FeaturePersisterProcessor2(writer);

                        intersectionsWriter = new ShpWriter();

                        String temp = System.getProperty("java.io.tmpdir") + 

                                        "/intersections_" + 

                                        System.currentTimeMillis() +

                                        ".shp";

                        File newFile = new File(temp);

                        ((ShpWriter) intersectionsWriter).setFile(newFile);

                        ILayerDefinition intersectDefinition = new SHPLayerDefinition();

                        intersectDefinition.setShapeType(XTypes.POINT);

                        FieldDescription[] intersectFields = new FieldDescription[2];

                        intersectFields[0] = new FieldDescription();

                        intersectFields[0].setFieldLength(10);

                        intersectFields[0].setFieldDecimalCount(0);

                        intersectFields[0].setFieldName("FID1");

                        intersectFields[0].setFieldType(XTypes.INTEGER);

                        intersectFields[1] = new FieldDescription();

                        intersectFields[1].setFieldLength(10);

                        intersectFields[1].setFieldDecimalCount(0);

                        intersectFields[1].setFieldName("FID2");

                        intersectFields[1].setFieldType(XTypes.INTEGER);

                        intersectDefinition.setFieldsDesc(intersectFields);

                        ((ShpWriter) intersectionsWriter).initialize(

                                        (LayerDefinition) intersectDefinition);

                        ((SHPLayerDefinition) intersectDefinition).setFile(newFile);

                        ShpSchemaManager interSchMg = 

                                new ShpSchemaManager(newFile.getAbsolutePath());

                        interSchMg.createSchema(intersectDefinition);

                        intersectsProcessor = new 

                                FeaturePersisterProcessor2(intersectionsWriter);

                        FBitSet selection = null;

                        LineCleanVisitor visitor = 

                                new LineCleanVisitor(processor,

                                                             intersectsProcessor,

                                                onlyFirstLayerSelection, 

                                                resultLayerDefinition,

                                                intersectDefinition);

                        Strategy strategy = StrategyManager.getStrategy(firstLayer);

                        try {

                                strategy.process(visitor, this);

                        } catch (Exception e) {

                                throw new GeoprocessException("Error al ejecutar el geoproceso line clean ", e);

                        }

                }

                // TODO INTERNACIONALIZAR LOS MENSAJES

                public String getNote() {

                        return "Limpiando l?neas..." + " " + getCurrentStep() + " "

                                        + "de" + " " + getFinishStep();

                }

                public void cancel() {

                        setCanceled(true);

                        LineCleanGeoprocess.this.cancel();

                }

        }

        public IMonitorableTask createTask() {

                return new LineCleanTask();

        }

        public FLayer getResult() throws GeoprocessException {

                FLyrVect cleanedLayer = (FLyrVect) createLayerFrom(this.writer);

                FLyrVect pseudoNodes = (FLyrVect) createLayerFrom(this.intersectionsWriter);

                try {

                        if(pseudoNodes.getSource().getShapeCount() == 0){

                                return cleanedLayer;

                        }else{

                                MapContext map = ((View)PluginServices.

                                                getMDIManager().

                                                getActiveWindow()).

                                                getModel().

                                                getMapContext();

                                FLayers solution = new FLayers(map, null);

                                solution.setName(this.firstLayer.getName()+"_cleaned");

                                solution.addLayer(cleanedLayer);

                                solution.addLayer(pseudoNodes);

                                return solution;

                        }

                } catch (DriverIOException e) {

                        throw new GeoprocessException("Error de lectura de datos");

                }

        }

}