Application: gvSIG desktop » gvSIG geoprocess

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

import java.awt.geom.Rectangle2D;

import es.unex.sextante.core.GeoAlgorithm;

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;

import com.vividsolutions.jts.geom.Geometry;

public class ClipAlgorithm extends GeoAlgorithm {

        public static final String  RESULT   = "RESULT";

        public static final String  LAYER    = "LAYER";

        /**

         * processes input layer's features to clip its geometries with clipping

         * layer bounding box

         */

        private ClipVisitor visitor;

        /*

         * (non-Javadoc)

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

         */

        public void defineCharacteristics(){

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

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

                setGeneratesUserDefinedRasterOutput(true);

                try {

                        m_Parameters.addInputVectorLayer(LAYER,

                                                                                                Sextante.getText( "Input_layer"),

                                                                                                IVectorLayer.SHAPE_TYPE_MIXED,

                                                                                                true);

                } catch (RepeatedParameterNameException e) {

                        Sextante.addErrorToLog(e);

                }

                addOutputVectorLayer(RESULT,

                                                                Sextante.getText( "Clip"),

                                                                OutputVectorLayer.SHAPE_TYPE_LINE);

        }

        public boolean processAlgorithm() throws GeoAlgorithmExecutionException{

                Geometry clippingGeometry = null;

                try {

                        clippingGeometry = computeJtsClippingPoly();

                } catch (Exception e) {

                        e.printStackTrace();

                }

                visitor = new ClipVisitor(clippingGeometry, resultLayerDefinition,

                                schemaManager, writer);

                Strategy strategy = StrategyManager.getStrategy(firstLayer);

                Rectangle2D clippingRect = FConverter.

                        convertEnvelopeToRectangle2D(clippingGeometry.

                                                                        getEnvelopeInternal());

                try {

                        if (onlyFirstLayerSelection) {

                                visitor.setSelection(firstLayer.getRecordset().getSelection());

                        }

                        strategy.process(visitor,

                                        clippingRect,

                                        cancelMonitor);

                } catch (Exception e) {

                        e.printStackTrace();

                }

                finished = true;

                return true;

        }

        /**

         * Computes union of all geometries of the clipping layer

         *

         * @return

         * @throws com.iver.cit.gvsig.fmap.DriverException

         * @throws ReadDriverException

         * @throws VisitorException

         * @throws ExpansionFileReadException

         */

        //TODO Esto lo vamos a quitar, y lo vamos a hacer para cada

        //feature individual

        private Geometry computeJtsClippingPoly()

                        throws ReadDriverException, ExpansionFileReadException, VisitorException {

                ScalableUnionVisitor visitor = new ScalableUnionVisitor(overlayLayer.getShapeType());

                Strategy strategy = StrategyManager.getStrategy(overlayLayer);

                if (onlyClipLayerSelection) {

                        strategy.process(visitor, overlayLayer.getRecordset()

                                        .getSelection());

                } else {

                        strategy.process(visitor);

                }

                return visitor.getJtsConvexHull();

        }

}