| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 * 

 * 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., 51 Franklin Street, Fifth Floor, Boston, 

 * MA  02110-1301, USA.

 * 

 */

package org.gvsig.editing.gui.cad.tools;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.operation.tojts.ToJTS;

import org.gvsig.fmap.geom.primitive.GeneralPathX;

import org.gvsig.fmap.geom.util.Converter;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontrol.MapControlDrawer;

import org.gvsig.tools.dispose.DisposableIterator;

public class AutoCompletePolygon extends PolylineCADTool {

    @Override

    /**

     * M?todo para dibujar la lo necesario para el estado en el que nos

     * encontremos.

     *

     * @param g Graphics sobre el que dibujar.

     * @param selectedGeometries BitSet con las geometr?as seleccionadas.

     * @param x par?metro x del punto que se pase para dibujar.

     * @param y par?metro x del punto que se pase para dibujar.

     */

    public void drawOperation(MapControlDrawer renderer, double x, double y) {

        Geometry geom = getGeometry();

        if (geom.getHandlers(Geometry.SELECTHANDLER).length == 0

            && firstPoint != null) {

            GeneralPathX gpx = new GeneralPathX();

            gpx.moveTo(firstPoint.getX(), firstPoint.getY());

            gpx.lineTo(x, y);

            renderer.draw(createCurve(gpx),

                mapControlManager.getGeometrySelectionSymbol());

        } else

            if (geom.getHandlers(Geometry.SELECTHANDLER).length > 1) {

                GeneralPathX gpxGeom = new GeneralPathX();

                gpxGeom.moveTo(x, y);

                gpxGeom

                    .append(

                        geom.getPathIterator(null, geomManager.getFlatness()),

                        true);

                gpxGeom.closePath();

                renderer.draw(createCurve(gpxGeom),

                    mapControlManager.getGeometrySelectionSymbol());

            }

    }

    private Geometry autoComplete(Geometry digitizedGeom)

        throws CreateGeometryException {

        com.vividsolutions.jts.geom.Geometry jtsGeom = null;

        FeatureSet selected = null;

        DisposableIterator iterator = null;

        try {

            jtsGeom =

                (com.vividsolutions.jts.geom.Geometry) digitizedGeom

                    .invokeOperation(ToJTS.CODE, null);

            FLyrVect lyrVect = (FLyrVect) getVLE().getLayer();

            // Se supone que debe ser r?pido, ya que est? indexado

            selected =

                lyrVect.queryByGeometry(digitizedGeom, lyrVect

                    .getFeatureStore().getDefaultFeatureType());

            iterator = selected.iterator();

            while (iterator.hasNext()) {

                Feature feature = (Feature) iterator.next();

                Geometry aux = feature.getDefaultGeometry();

                com.vividsolutions.jts.geom.Geometry jtsAux =

                    (com.vividsolutions.jts.geom.Geometry) aux.invokeOperation(

                        ToJTS.CODE, null);

                jtsGeom = jtsGeom.difference(jtsAux);

            }

        } catch (ReadException e) {

            NotificationManager.showMessageError(

                PluginServices.getText(this,

                    "Error_in_Autocomplete_Polygon_Tool_")

                    + " "

                    + e.getLocalizedMessage(), e);

        } catch (com.vividsolutions.jts.geom.TopologyException e) {

            NotificationManager.showMessageError(

                PluginServices.getText(this,

                    "Error_in_Autocomplete_Polygon_Tool_")

                    + " "

                    + e.getLocalizedMessage(), e);

        } catch (Exception e) {

            NotificationManager.showMessageError(

                PluginServices.getText(this,

                    "Error_in_Autocomplete_Polygon_Tool_")

                    + " "

                    + e.getLocalizedMessage(), e);

        } finally {

            if (iterator != null) {

                iterator.dispose();

            }

            if (selected != null) {

                selected.dispose();

            }

        }

        return Converter.jtsToGeometry(jtsGeom);

    }

    @Override

    public Feature insertGeometry(Geometry geometry) {

        Geometry newGeom;

        try {

            newGeom = autoComplete(geometry);

            return super.insertGeometry(newGeom);

        } catch (CreateGeometryException e) {

            NotificationManager.addError(e);

        }

        return null;

    }

    public boolean isApplicable(int shapeType) {

        switch (shapeType) {

        case Geometry.TYPES.SURFACE:

            return true;

        }

        return false;

    }

}