X-Git-Url: http://gitweb.fperrin.net/?a=blobdiff_plain;f=tim%2Fprune%2Fsave%2FKmlExporter.java;h=82b4f1c12ca7cc7f679f8896bbe86555142137a0;hb=52bf9e8686c916be37a26a0b75340393d4478b05;hp=11474a6c8e8fb37f911f53f3f79e51edebc5fcf4;hpb=5625a1abadb5f2ca5f017fe7dbda1d5141cb637b;p=GpsPrune.git

diff --git a/tim/prune/save/KmlExporter.java b/tim/prune/save/KmlExporter.java
index 11474a6..82b4f1c 100644
--- a/tim/prune/save/KmlExporter.java
+++ b/tim/prune/save/KmlExporter.java
@@ -32,14 +32,18 @@ import javax.swing.JPanel;
 import javax.swing.JProgressBar;
 import javax.swing.JTextField;
 import javax.swing.SwingConstants;
-import javax.swing.filechooser.FileFilter;
 
+import tim.prune.Config;
 import tim.prune.I18nManager;
+import tim.prune.UpdateMessageBroker;
+import tim.prune.data.Altitude;
 import tim.prune.data.Coordinate;
 import tim.prune.data.DataPoint;
+import tim.prune.data.Field;
 import tim.prune.data.Track;
 import tim.prune.data.TrackInfo;
 import tim.prune.gui.ImageUtils;
+import tim.prune.load.GenericFileFilter;
 
 /**
  * Class to export track information
@@ -52,6 +56,7 @@ public class KmlExporter implements Runnable
 	private Track _track = null;
 	private JDialog _dialog = null;
 	private JTextField _descriptionField = null;
+	private JCheckBox _altitudesCheckbox = null;
 	private JCheckBox _kmzCheckbox = null;
 	private JCheckBox _exportImagesCheckbox = null;
 	private JProgressBar _progressBar = null;
@@ -116,6 +121,10 @@ public class KmlExporter implements Runnable
 		descPanel.add(_descriptionField);
 		mainPanel.add(descPanel);
 		dialogPanel.add(mainPanel, BorderLayout.CENTER);
+		// Checkbox for altitude export
+		_altitudesCheckbox = new JCheckBox(I18nManager.getText("dialog.exportkml.altitude"));
+		_altitudesCheckbox.setHorizontalTextPosition(SwingConstants.LEFT);
+		mainPanel.add(_altitudesCheckbox);
 		// Checkboxes for kmz export and image export
 		_kmzCheckbox = new JCheckBox(I18nManager.getText("dialog.exportkml.kmz"));
 		_kmzCheckbox.setHorizontalTextPosition(SwingConstants.LEFT);
@@ -166,6 +175,8 @@ public class KmlExporter implements Runnable
 	 */
 	private void enableCheckboxes()
 	{
+		boolean hasAltitudes = _track.hasData(Field.ALTITUDE);
+		if (!hasAltitudes) {_altitudesCheckbox.setSelected(false);}
 		boolean hasPhotos = _trackInfo.getPhotoList() != null && _trackInfo.getPhotoList().getNumPhotos() > 0;
 		_exportImagesCheckbox.setSelected(hasPhotos && _kmzCheckbox.isSelected());
 		_exportImagesCheckbox.setEnabled(hasPhotos && _kmzCheckbox.isSelected());
@@ -179,19 +190,14 @@ public class KmlExporter implements Runnable
 	{
 		// OK pressed, so choose output file
 		if (_fileChooser == null)
-			{_fileChooser = new JFileChooser();}
-		_fileChooser.setDialogType(JFileChooser.SAVE_DIALOG);
-		_fileChooser.setFileFilter(new FileFilter() {
-			public boolean accept(File f)
-			{
-				return (f != null && (f.isDirectory()
-					|| f.getName().toLowerCase().endsWith(".kml") || f.getName().toLowerCase().endsWith(".kmz")));
-			}
-			public String getDescription()
-			{
-				return I18nManager.getText("dialog.exportkml.filetype");
-			}
-		});
+		{
+			_fileChooser = new JFileChooser();
+			_fileChooser.setDialogType(JFileChooser.SAVE_DIALOG);
+			_fileChooser.setFileFilter(new GenericFileFilter("filetype.kmlkmz", new String[] {"kml", "kmz"}));
+			// start from directory in config which should be set
+			File configDir = Config.getWorkingDirectory();
+			if (configDir != null) {_fileChooser.setCurrentDirectory(configDir);}
+		}
 		String requiredExtension = null, otherExtension = null;
 		if (_kmzCheckbox.isSelected())
 		{
@@ -293,10 +299,12 @@ public class KmlExporter implements Runnable
 
 			// close file
 			writer.close();
-			JOptionPane.showMessageDialog(_parentFrame, I18nManager.getText("dialog.save.ok1")
-				 + " " + numPoints + " " + I18nManager.getText("dialog.save.ok2")
-				 + " " + _exportFile.getAbsolutePath(),
-				I18nManager.getText("dialog.save.oktitle"), JOptionPane.INFORMATION_MESSAGE);
+			// Store directory in config for later
+			Config.setWorkingDirectory(_exportFile.getParentFile());
+			// show confirmation
+			UpdateMessageBroker.informSubscribers(I18nManager.getText("confirm.save.ok1")
+				 + " " + numPoints + " " + I18nManager.getText("confirm.save.ok2")
+				 + " " + _exportFile.getAbsolutePath());
 			// export successful so need to close dialog and return
 			_dialog.dispose();
 			return;
@@ -326,7 +334,6 @@ public class KmlExporter implements Runnable
 	private int exportData(OutputStreamWriter inWriter, boolean inExportImages)
 	throws IOException
 	{
-		// TODO: Look at segments of track, and split into separate lines in Kml if necessary
 		inWriter.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<kml xmlns=\"http://earth.google.com/kml/2.1\">\n<Folder>\n");
 		inWriter.write("\t<name>");
 		if (_descriptionField != null && _descriptionField.getText() != null && !_descriptionField.getText().equals(""))
@@ -339,6 +346,7 @@ public class KmlExporter implements Runnable
 		}
 		inWriter.write("</name>\n");
 
+		boolean exportAltitudes = _altitudesCheckbox.isSelected();
 		int i = 0;
 		DataPoint point = null;
 		boolean hasTrackpoints = false;
@@ -352,7 +360,11 @@ public class KmlExporter implements Runnable
 			// Make a blob for each waypoint
 			if (point.isWaypoint())
 			{
-				exportWaypoint(point, inWriter);
+				exportWaypoint(point, inWriter, exportAltitudes);
+			}
+			else if (point.getPhoto() == null)
+			{
+				hasTrackpoints = true;
 			}
 			// Make a blob with description for each photo
 			if (point.getPhoto() != null)
@@ -363,29 +375,43 @@ public class KmlExporter implements Runnable
 					writtenPhotoHeader = true;
 				}
 				photoNum++;
-				exportPhotoPoint(point, inWriter, inExportImages, photoNum);
-			}
-			else
-			{
-				hasTrackpoints = true;
+				exportPhotoPoint(point, inWriter, inExportImages, photoNum, exportAltitudes);
 			}
 		}
 		// Make a line for the track, if there is one
 		if (hasTrackpoints)
 		{
-			inWriter.write("\t<Placemark>\n\t\t<name>track</name>\n\t\t<Style>\n\t\t\t<LineStyle>\n"
+			// Set up strings for start and end of track segment
+			String trackStart = "\t<Placemark>\n\t\t<name>track</name>\n\t\t<Style>\n\t\t\t<LineStyle>\n"
 				+ "\t\t\t\t<color>cc0000cc</color>\n\t\t\t\t<width>4</width>\n\t\t\t</LineStyle>\n"
-				+ "\t\t</Style>\n\t\t<LineString>\n\t\t\t<coordinates>");
+				+ "\t\t\t<PolyStyle><color>33cc0000</color></PolyStyle>\n"
+				+ "\t\t</Style>\n\t\t<LineString>\n";
+			if (exportAltitudes) {
+				trackStart += "\t\t\t<extrude>1</extrude>\n\t\t\t<altitudeMode>absolute</altitudeMode>\n";
+			}
+			trackStart += "\t\t\t<coordinates>";
+			String trackEnd = "\t\t\t</coordinates>\n\t\t</LineString>\n\t</Placemark>";
+
+			// Start segment
+			inWriter.write(trackStart);
 			// Loop over track points
+			boolean firstTrackpoint = true;
 			for (i=0; i<numPoints; i++)
 			{
 				point = _track.getPoint(i);
-				if (!point.isWaypoint())
+				// start new track segment if necessary
+				if (point.getSegmentStart() && !firstTrackpoint) {
+					inWriter.write(trackEnd);
+					inWriter.write(trackStart);
+				}
+				if (!point.isWaypoint() && point.getPhoto() == null)
 				{
-					exportTrackpoint(point, inWriter);
+					exportTrackpoint(point, inWriter, exportAltitudes);
+					firstTrackpoint = false;
 				}
 			}
-			inWriter.write("\t\t\t</coordinates>\n\t\t</LineString>\n\t</Placemark>");
+			// end segment
+			inWriter.write(trackEnd);
 		}
 		inWriter.write("</Folder>\n</kml>");
 		return numPoints;
@@ -396,18 +422,30 @@ public class KmlExporter implements Runnable
 	 * Export the specified waypoint into the file
 	 * @param inPoint waypoint to export
 	 * @param inWriter writer object
+	 * @param inExportAltitude true to include altitude
 	 * @throws IOException on write failure
 	 */
-	private void exportWaypoint(DataPoint inPoint, Writer inWriter) throws IOException
+	private void exportWaypoint(DataPoint inPoint, Writer inWriter, boolean inExportAltitude) throws IOException
 	{
 		inWriter.write("\t<Placemark>\n\t\t<name>");
 		inWriter.write(inPoint.getWaypointName().trim());
 		inWriter.write("</name>\n");
-		inWriter.write("\t\t<Point>\n\t\t\t<coordinates>");
+		inWriter.write("\t\t<Point>\n");
+		if (inExportAltitude && inPoint.hasAltitude()) {
+			inWriter.write("\t\t\t<altitudeMode>absolute</altitudeMode>\n");
+		}
+		inWriter.write("\t\t\t<coordinates>");
 		inWriter.write(inPoint.getLongitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
 		inWriter.write(',');
 		inWriter.write(inPoint.getLatitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
-		inWriter.write(",0</coordinates>\n\t\t</Point>\n\t</Placemark>\n");
+		inWriter.write(",");
+		if (inExportAltitude && inPoint.hasAltitude()) {
+			inWriter.write("" + inPoint.getAltitude().getStringValue(Altitude.FORMAT_METRES));
+		}
+		else {
+			inWriter.write("0");
+		}
+		inWriter.write("</coordinates>\n\t\t</Point>\n\t</Placemark>\n");
 	}
 
 
@@ -417,9 +455,11 @@ public class KmlExporter implements Runnable
 	 * @param inWriter writer object
 	 * @param inImageLink flag to set whether to export image links or not
 	 * @param inImageNumber number of image for filename
+	 * @param inExportAltitude true to include altitude
 	 * @throws IOException on write failure
 	 */
-	private void exportPhotoPoint(DataPoint inPoint, Writer inWriter, boolean inImageLink, int inImageNumber)
+	private void exportPhotoPoint(DataPoint inPoint, Writer inWriter, boolean inImageLink,
+		int inImageNumber, boolean inExportAltitude)
 	throws IOException
 	{
 		inWriter.write("\t<Placemark>\n\t\t<name>");
@@ -436,11 +476,22 @@ public class KmlExporter implements Runnable
 				+ "<tr><td><center>Caption for the photo</center></td></tr></table>]]></description>");
 		}
 		inWriter.write("<styleUrl>#camera_icon</styleUrl>\n");
-		inWriter.write("\t\t<Point>\n\t\t\t<coordinates>");
+		inWriter.write("\t\t<Point>\n");
+		if (inExportAltitude && inPoint.hasAltitude()) {
+			inWriter.write("\t\t\t<altitudeMode>absolute</altitudeMode>\n");
+		}
+		inWriter.write("\t\t\t<coordinates>");
 		inWriter.write(inPoint.getLongitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
 		inWriter.write(',');
 		inWriter.write(inPoint.getLatitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
-		inWriter.write(",0</coordinates>\n\t\t</Point>\n\t</Placemark>\n");
+		inWriter.write(",");
+		if (inExportAltitude && inPoint.hasAltitude()) {
+			inWriter.write("" + inPoint.getAltitude().getStringValue(Altitude.FORMAT_METRES));
+		}
+		else {
+			inWriter.write("0");
+		}
+		inWriter.write("</coordinates>\n\t\t</Point>\n\t</Placemark>\n");
 	}
 
 
@@ -448,14 +499,22 @@ public class KmlExporter implements Runnable
 	 * Export the specified trackpoint into the file
 	 * @param inPoint trackpoint to export
 	 * @param inWriter writer object
+	 * @param inExportAltitude true to include altitude
 	 */
-	private void exportTrackpoint(DataPoint inPoint, Writer inWriter) throws IOException
+	private void exportTrackpoint(DataPoint inPoint, Writer inWriter, boolean inExportAltitude) throws IOException
 	{
 		inWriter.write(inPoint.getLongitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
 		inWriter.write(',');
 		inWriter.write(inPoint.getLatitude().output(Coordinate.FORMAT_DEG_WITHOUT_CARDINAL));
-		// Altitude not exported, locked to ground by Google Earth
-		inWriter.write(",0\n");
+		// Altitude either absolute or locked to ground by Google Earth
+		inWriter.write(",");
+		if (inExportAltitude && inPoint.hasAltitude()) {
+			inWriter.write("" + inPoint.getAltitude().getStringValue(Altitude.FORMAT_METRES));
+		}
+		else {
+			inWriter.write("0");
+		}
+		inWriter.write("\n");
 	}