Application: gvSIG desktop

/*

 * Created on 10-abr-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: 

* $Log: 

*/

package com.iver.cit.gvsig.cad;

import org.gvsig.fmap.core.FGeometryUtil;

import org.gvsig.jts.JtsUtil;

import org.gvsig.topology.ui.util.GUIUtil;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.Extension;

import com.iver.cit.gvsig.CADExtension;

import com.iver.cit.gvsig.EditionUtilities;

import com.iver.cit.gvsig.drivers.featureiterators.FeatureBitsetIterator;

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

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

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

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

import com.vividsolutions.jts.geom.Geometry;

/**

 * This extension only will active when active layer

 * is a linear (dimension = 1) vectorial layer with

 * selected lines.

 * 

 * It smooths the selected lines.

 * 

 * @author Alvaro Zabala

 *

 */

public class SmoothPolylinesExtension extends Extension {

	private View view;

	public void execute(String actionCommand) {

		try {

			FLyrVect lv = (FLyrVect) CADExtension.

								getEditionManager().

								getActiveLayerEdited().

								getLayer();

			int geometryDimensions = FGeometryUtil.getDimensions(lv.getShapeType());

			FeatureBitsetIterator iterator = new FeatureBitsetIterator(lv.getSource());

			while(iterator.hasNext()){

				IFeature feature = iterator.next();

				IGeometry geometry = feature.getGeometry();

				Geometry jtsGeometry = geometry.toJTSGeometry();

				Geometry generalizedGeometry = null;

				if(geometryDimensions == 1)

					generalizedGeometry = JtsUtil.douglasPeuckerSimplify(jtsGeometry, JtsUtil.GENERALIZATION_FACTOR.doubleValue());

				else 

					generalizedGeometry = JtsUtil.topologyPreservingSimplify(jtsGeometry, JtsUtil.GENERALIZATION_FACTOR.doubleValue());

			}

//			  IGeometry geom1=ShapeFactory.createPolyline2D(newGp1);

//	             IGeometry geom2=ShapeFactory.createPolyline2D(newGp2);

//

//	             DefaultFeature dfLine1 = (DefaultFeature) dre.getLinkedRow().cloneRow();

//	             dfLine1.setGeometry(geom1);

//	    int indexLine1 = addGeometry(geom1, dfLine1

//	      .getAttributes());

//

//	    selectedRowAux.add(new DefaultRowEdited(dfLine1,

//	      IRowEdited.STATUS_ADDED, indexLine1));

//

//	    DefaultFeature dfLine2 = (DefaultFeature) dre.getLinkedRow().cloneRow();

//	          dfLine2.setGeometry(geom2);

//	    int indexLine2 = addGeometry(geom2, dfLine2

//	       .getAttributes());

//

//	    selectedRowAux.add(new DefaultRowEdited(dfLine2,

//	       IRowEdited.STATUS_ADDED, indexLine2));

//			 vea.removeRow(dre.getIndex(), getName(), EditionEvent.GRAPHIC);

//		     vea.endComplexRow(getName());

//		   vle.setSelectionCache(VectorialLayerEdited.NOTSAVEPREVIOUS, selectedRowAux);

		} catch (ReadDriverException e) {

			String title = PluginServices.getText(this, "Error_De_Acceso_Datos");

			String msg = PluginServices.getText(this, "Error_accediendo_a_los_datos");

			GUIUtil.getInstance().messageBox(msg, title);

			e.printStackTrace();

		}

	}

	public void initialize() {

		registerIcons();

	}

	private void registerIcons(){

		PluginServices.getIconTheme().registerDefault("smooth-geometry",

													   this.getClass().

													   getClassLoader().

													   getResource("images/smooth-geometry.gif")

		);

	}

	/**

	 * Returns if this Edit CAD tool is visible. 

	 * For this, there must be an active vectorial editing lyr in the TOC, which geometries'

	 * dimension would must be linear or polygonal, and with at least one selected geometry.

	 *  

	 */

	public boolean isEnabled() {

		try {

			if (EditionUtilities.getEditionStatus() == 

				EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE) {

					this.view = (View) PluginServices.getMDIManager().getActiveWindow();

					if (CADExtension.getEditionManager().getActiveLayerEdited() == null)

						return false;

					FLyrVect lv = (FLyrVect) CADExtension.

											getEditionManager().

											getActiveLayerEdited().

											getLayer();

					int geometryDimensions = FGeometryUtil.getDimensions(lv.getShapeType());

					if(geometryDimensions <= 0)

						return false;

					return lv.getRecordset().getSelection().cardinality() != 0;

			}

		} catch (ReadDriverException e) {

			NotificationManager.addError(e.getMessage(),e);

			return false;

		}

		return true;

	}

	public boolean isVisible() {

		if (EditionUtilities.getEditionStatus() == EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE)

			return true;

		return false;

	}

}

/*

 * Created on 10-abr-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: 

* $Log: 

*/

package com.iver.cit.gvsig.cad;

import org.gvsig.fmap.core.FGeometryUtil;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.Extension;

import com.iver.cit.gvsig.CADExtension;

import com.iver.cit.gvsig.EditionUtilities;

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

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

import com.iver.cit.gvsig.gui.cad.tools.PolylineCADTool;

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

/**

 * CAD extension to split geometries from a digitized linear geometry.

 * 

 * @author Alvaro Zabala

 *

 */

public class SplitGeometryCADToolExtension extends Extension {

	private View view;

	private MapControl mapControl;

	private SplitGeometryCADTool cadTool;

	public void execute(String actionCommand) {

		CADExtension.initFocus();

		if (actionCommand.equals(SplitGeometryCADTool.SPLIT_GEOMETRY_TOOL_NAME)) {

        	CADExtension.setCADTool(SplitGeometryCADTool.SPLIT_GEOMETRY_TOOL_NAME, true);

        }

		CADExtension.getEditionManager().setMapControl(mapControl);

		CADExtension.getCADToolAdapter().configureMenu();

	}

	public void initialize() {

		cadTool = new SplitGeometryCADTool();

		CADExtension.addCADTool(SplitGeometryCADTool.SPLIT_GEOMETRY_TOOL_NAME, cadTool);

		registerIcons();

	}

	private void registerIcons(){

		PluginServices.getIconTheme().registerDefault("split-geometry",

													   this.getClass().

													   getClassLoader().

													   getResource("images/split-poly.png")

		);

	}

	/**

	 * Returns if this Edit CAD tool is visible. 

	 * For this, there must be an active vectorial editing lyr in the TOC, which geometries'

	 * dimension would must be linear or polygonal, and with at least one selected geometry.

	 *  

	 */

	public boolean isEnabled() {

		try {

			if (EditionUtilities.getEditionStatus() == 

				EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE) {

					this.view = (View) PluginServices.getMDIManager().getActiveWindow();

					mapControl = view.getMapControl();

					if (CADExtension.getEditionManager().getActiveLayerEdited() == null)

						return false;

					FLyrVect lv = (FLyrVect) CADExtension.

											getEditionManager().

											getActiveLayerEdited().

											getLayer();

					int geometryDimensions = FGeometryUtil.getDimensions(lv.getShapeType());

					if(geometryDimensions <= 0)

						return false;

					return lv.getRecordset().getSelection().cardinality() != 0;

			}

		} catch (ReadDriverException e) {

			NotificationManager.addError(e.getMessage(),e);

			return false;

		}

		return true;

	}

	public boolean isVisible() {

		if (EditionUtilities.getEditionStatus() == EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE)

			return true;

		return false;

	}

}

/*

 * Created on 10-abr-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: 

* $Log: 

*/

package com.iver.cit.gvsig.cad;

import java.awt.Graphics;

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.event.InputEvent;

import java.awt.event.MouseEvent;

import java.awt.geom.Point2D;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.jts.JtsUtil;

import statemap.State;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.iver.andami.PluginServices;

import com.iver.cit.gvsig.cad.sm.SplitGeometryCADToolContext;

import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;

import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileWriteException;

import com.iver.cit.gvsig.exceptions.validate.ValidateRowException;

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

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

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

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

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

import com.iver.cit.gvsig.fmap.core.v02.FConverter;

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

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

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

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

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

import com.iver.cit.gvsig.gui.cad.DefaultCADTool;

import com.iver.cit.gvsig.gui.cad.exception.CommandException;

import com.iver.cit.gvsig.layers.VectorialLayerEdited;

import com.vividsolutions.jts.geom.Coordinate;

import com.vividsolutions.jts.geom.Geometry;

import com.vividsolutions.jts.geom.GeometryCollection;

import com.vividsolutions.jts.geom.LineString;

/**

 * CAD Tool which splits the selected geometries of a vectorial editing

 * layer with a digitized polyline.

 * 

 * 

 * @author Alvaro Zabala

 *

 */

public class SplitGeometryCADTool extends DefaultCADTool {

	/**

	 * String representation of this tool (used for example to active the tool

	 * in mapcontrol)

	 */

	public static final String SPLIT_GEOMETRY_TOOL_NAME = "_split_geometry";

	/**

	 * finite state machine for this CAD tool

	 */

	private SplitGeometryCADToolContext _fsm;

	/**

	 * Flag to mark if the digitized line has been finished.

	 */

	private boolean digitizingFinished = false;

	/**

	 * Collection of digitized geometries

	 */

	private List<Point2D> clickedPoints;

	/**

	 * Default Constructor

	 */

	public SplitGeometryCADTool(){

	}

	/**

	 * Initialization method.

	 */

	public void init() {

		digitizingFinished = false;

		_fsm = new SplitGeometryCADToolContext(this);

		setNextTool(SplitGeometryCADTool.SPLIT_GEOMETRY_TOOL_NAME);

	}

	public boolean isDigitizingFinished() {

		return digitizingFinished;

	}

	public String toString() {

		return SplitGeometryCADTool.SPLIT_GEOMETRY_TOOL_NAME;

	}

	public void finishDigitizedLine(){

	}

	public ArrayList getSelectedRows(){

		ArrayList selection = super.getSelectedRows();

		if(selection != null ){

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

				VectorialEditableAdapter vea = getVLE().getVEA();

				try {

					FBitSet bitset = vea.getSelection();

					for (int j = bitset.nextSetBit(0);j >= 0;j = bitset.nextSetBit(j + 1)){

						IRowEdited rowEd =vea.getRow(j);

						selection.add(rowEd);

					}

				} catch (ExpansionFileReadException e) {

					e.printStackTrace();

				} catch (ReadDriverException e) {

					e.printStackTrace();

				}

			}//selection size

		}

		return selection;

	}

	public void splitSelectedGeometryWithDigitizedLine(){

		Point2D[] clickedPts = new Point2D[this.clickedPoints.size()];

		clickedPoints.toArray(clickedPts);

		Coordinate[] digitizedCoords = JtsUtil.getPoint2DAsCoordinates(clickedPts);

		LineString splittingLs = JtsUtil.GEOMETRY_FACTORY.createLineString(digitizedCoords);

		ArrayList selectedRows = getSelectedRows();

		ArrayList<IRowEdited> splittedSelectionGeoms = new

			ArrayList<IRowEdited>();

		IRowEdited editedRow = null;

		VectorialLayerEdited vle = getVLE();

		VectorialEditableAdapter vea = vle.getVEA();

//		vea.startComplexRow();

		for (int i = 0; i < selectedRows.size(); i++) {

			editedRow = (IRowEdited) selectedRows.get(i);

			IFeature feat = (IFeature) editedRow.getLinkedRow().cloneRow();

			IGeometry ig = feat.getGeometry();

			Geometry jtsGeo = ig.toJTSGeometry();

			Geometry splittedGeo = JtsUtil.split(jtsGeo, splittingLs);

			if(splittedGeo instanceof GeometryCollection){

				//first of all we delete the existing geometry

				try {

					vea.removeRow(editedRow.getIndex(), 

							PluginServices.getText(this,"deleted_feature"),

							EditionEvent.GRAPHIC);

					//and then, we add new features for each splitted geometry

					GeometryCollection gc = (GeometryCollection)splittedGeo;

					for(int j = 0; j < gc.getNumGeometries(); j++){

						Geometry g = gc.getGeometryN(j);

						System.out.println("g"+j+":"+g.toText());

						IGeometry fmapGeo = FConverter.jts_to_igeometry(g);

						DefaultFeature newFeature = new DefaultFeature(fmapGeo, feat.getAttributes(), vea.getNewFID());

						int calculatedIndex = vea.addRow(newFeature, getName(), EditionEvent.GRAPHIC);

						DefaultRowEdited newRowEdited = new DefaultRowEdited(newFeature,

														IRowEdited.STATUS_ADDED, 

															calculatedIndex);

						splittedSelectionGeoms.add(newRowEdited);

					}//for j

				} catch (ExpansionFileReadException e) {

					e.printStackTrace();

				} catch (ExpansionFileWriteException e) {

					e.printStackTrace();

				} catch (ReadDriverException e) {

					e.printStackTrace();

				} catch (ValidateRowException e) {

					e.printStackTrace();

				}

			}//if splittedGeo

		}

//		vea.endComplexRow(description);

//		vle.setSelectionCache(VectorialLayerEdited.SAVEPREVIOUS,

//				splittedSelectionGeoms);

	}

	public void end(){

		getCadToolAdapter().refreshEditedLayer();

		init();

	}

	public void addOption(String s) {

		State actualState = _fsm.getPreviousState();

		String status = actualState.getName();

		if (s.equals(PluginServices.getText(this, "cancel"))) {

			init();

			return;

		}

		if (status.equals("TopologicalEdition.FirstPoint")) {

			return;

		}

		init();

	}

	public void addPoint(double x, double y, InputEvent event) {

		State actualState = _fsm.getPreviousState();

		String status = actualState.getName();

		if (status.equals("SplitGeometry.FirstPoint")) {

			clickedPoints = new ArrayList<Point2D>();

			clickedPoints.add(new Point2D.Double(x,y));

		} else if (status.equals("SplitGeometry.DigitizingLine")) {

			clickedPoints.add(new Point2D.Double(x,y));

			if (event != null && ((MouseEvent) event).getClickCount() == 2) {

				digitizingFinished = true;

				finishDigitizedLine();

				splitSelectedGeometryWithDigitizedLine();

				end();

			}

		}

	}

	public void addValue(double d) {

	}

	/**

	 * Draws a polyline with the clicked digitized points in the specified graphics.

	 * 

	 * @param g2 graphics on to draw the polyline

	 * @param x last x mouse pointer position

	 * @param y last y mouse pointer position

	 */

	protected void drawPolyLine(Graphics2D g, double x, double y) {

		GeneralPathX gpx = 

			new GeneralPathX(GeneralPathX.WIND_EVEN_ODD, 

									clickedPoints.size());

		Point2D firstPoint = clickedPoints.get(0);

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

		for (int i = 1; i < clickedPoints.size(); i++) {

			Point2D clickedPoint = clickedPoints.get(i);

			gpx.lineTo(clickedPoint.getX(), clickedPoint.getY());

		}

		gpx.lineTo(x, y);

		ShapeFactory.createPolyline2D(gpx).draw((Graphics2D) g,

					getCadToolAdapter().getMapControl().getViewPort(),

					DefaultCADTool.geometrySelectSymbol);

	}

	public void drawOperation(Graphics g, double x, double y) {

		State actualState = _fsm.getState();

        String status = actualState.getName();

        if (status.equals("SplitGeometry.FirstPoint") ) {

        	return;

        } else if ((status.equals("SplitGeometry.DigitizingLine"))) {

        	drawPolyLine((Graphics2D) g, x, y);

         }

//        else{

//			try {

//				Image imgSel = getVLE().getSelectionImage();

//				if (imgSel != null)

//					g.drawImage(imgSel, 0, 0, null);

//			} catch (Exception e) {

//				e.printStackTrace();

//			}

//         }

	}

	public String getName() {

		return PluginServices.getText(this, SPLIT_GEOMETRY_TOOL_NAME);

	}

	public void transition(double x, double y, InputEvent event) {

		try {

			_fsm.addPoint(x, y, event);

		} catch (Exception e) {

			init();

		}

	}

	public void transition(double d) {

		_fsm.addValue(d);

	}

	public void transition(String s) throws CommandException {

		if (!super.changeCommand(s)) {

			_fsm.addOption(s);

		}

	}

}