X-Git-Url: https://gitweb.fperrin.net/?p=GpsPrune.git;a=blobdiff_plain;f=tim%2Fprune%2Fundo%2FUndoLookupSrtm.java;h=d82714ae1e6444ce18fda96d1d2cddbc1a0fcff9;hp=2672bd55d95cf10a3576d30b6f90870a71fa4a23;hb=140e9d165f85c3d4f0435a311e091209313faa2a;hpb=c0387c124840c9407e040600fda88f3c3e8f6aa6

diff --git a/tim/prune/undo/UndoLookupSrtm.java b/tim/prune/undo/UndoLookupSrtm.java
index 2672bd5..d82714a 100644
--- a/tim/prune/undo/UndoLookupSrtm.java
+++ b/tim/prune/undo/UndoLookupSrtm.java
@@ -14,6 +14,8 @@ public class UndoLookupSrtm implements UndoOperation
 {
 	/** DataPoint objects which didn't have altitudes before */
 	private DataPoint[] _points;
+	/** Altitude strings if present */
+	private String[] _altitudes;
 
 
 	/**
@@ -24,12 +26,16 @@ public class UndoLookupSrtm implements UndoOperation
 	{
 		Track track = inTrackInfo.getTrack();
 		int numPoints = track.getNumPoints();
-		// Make array of points without altitudes
+		// Make arrays of points and altitudes
 		_points = new DataPoint[numPoints];
+		_altitudes = new String[numPoints];
 		for (int i=0; i<numPoints; i++) {
 			DataPoint point = track.getPoint(i);
-			if (!point.hasAltitude()) {
+			if (!point.hasAltitude() || point.getAltitude().getValue() == 0) {
 				_points[i] = point;
+				if (point.hasAltitude()) {
+					_altitudes[i] = point.getFieldValue(Field.ALTITUDE);
+				}
 			}
 		}
 	}
@@ -55,7 +61,12 @@ public class UndoLookupSrtm implements UndoOperation
 		for (int i=0; i<numPoints; i++) {
 			DataPoint point = _points[i];
 			if (point != null && point.hasAltitude()) {
-				point.setFieldValue(Field.ALTITUDE, null, true);
+				if (_altitudes[i] == null) {
+					point.setFieldValue(Field.ALTITUDE, null, true);
+				}
+				else {
+					point.setFieldValue(Field.ALTITUDE, _altitudes[i], true);
+				}
 			}
 		}
 		_points = null;