X-Git-Url: https://gitweb.fperrin.net/?a=blobdiff_plain;f=tim%2Fprune%2Fundo%2FUndoCompress.java;h=50582215aaf82aeb265c52b3cddac6ca5cb7b7b7;hb=ca9bdb3916f9c39adbbf95d06ac95c21dafbb4e6;hp=843054dc90fd3195b01a3733da099feb0a40c58e;hpb=312fec956e43f5d0a38617da5d0add9c62563e2c;p=GpsPrune.git

diff --git a/tim/prune/undo/UndoCompress.java b/tim/prune/undo/UndoCompress.java
index 843054d..5058221 100644
--- a/tim/prune/undo/UndoCompress.java
+++ b/tim/prune/undo/UndoCompress.java
@@ -12,6 +12,7 @@ public class UndoCompress implements UndoOperation
 {
 	private DataPoint[] _contents = null;
 	protected int _numPointsDeleted = -1;
+	private boolean[] _segmentStarts = null;
 
 
 	/**
@@ -21,6 +22,11 @@ public class UndoCompress implements UndoOperation
 	public UndoCompress(Track inTrack)
 	{
 		_contents = inTrack.cloneContents();
+		// Copy boolean segment start flags
+		_segmentStarts = new boolean[inTrack.getNumPoints()];
+		for (int i=0; i<inTrack.getNumPoints(); i++) {
+			_segmentStarts[i] = inTrack.getPoint(i).getSegmentStart();
+		}
 	}
 
 
@@ -55,6 +61,13 @@ public class UndoCompress implements UndoOperation
 	{
 		// restore track to previous values
 		inTrackInfo.getTrack().replaceContents(_contents);
+		// Copy boolean segment start flags
+		Track track = inTrackInfo.getTrack();
+		if (_segmentStarts.length != track.getNumPoints())
+			throw new UndoException("Cannot undo compress - track length no longer matches");
+		for (int i=0; i<_segmentStarts.length; i++) {
+			track.getPoint(i).setSegmentStart(_segmentStarts[i]);
+		}
 		// clear selection
 		inTrackInfo.getSelection().clearAll();
 	}