1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
package tim.prune.function.compress;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import tim.prune.App;
import tim.prune.GenericFunction;
import tim.prune.cmd.Command;
import tim.prune.cmd.PointFlag;
import tim.prune.cmd.ShuffleAndCropCmd;
import tim.prune.data.DataPoint;
import tim.prune.data.Track;
import tim.prune.data.TrackInfo;
/**
* Function to delete the marked points in the track
*/
public class DeleteMarkedPointsFunction extends GenericFunction
{
private boolean _splitSegments = false;
private String _parentFunctionKey = null;
/** Constructor */
public DeleteMarkedPointsFunction(App inApp) {
super(inApp);
}
@Override
public String getNameKey() {
return "function.deletemarked";
}
/**
* Get notification about parent function
* @param inKey parent function name key
* @param inSplitSegments true to split segment, false to not
*/
public void setParentFunction(String inKey, boolean inSplitSegments)
{
_parentFunctionKey = inKey;
_splitSegments = inSplitSegments;
}
public void begin()
{
ArrayList<Integer> indexesToKeep = new ArrayList<>();
ArrayList<Integer> indexesToDelete = new ArrayList<>();
ArrayList<Integer> indexesOfSegments = new ArrayList<>();
TrackInfo trackInfo = _app.getTrackInfo();
for (int i=0; i<trackInfo.getTrack().getNumPoints(); i++)
{
if (trackInfo.isPointMarkedForDeletion(i))
{
indexesToDelete.add(i);
if (trackInfo.isPointMarkedForSegmentBreak(i) || _splitSegments) {
indexesOfSegments.add(i);
}
}
else {
indexesToKeep.add(i);
}
}
if (indexesToDelete.isEmpty())
{
String titleKey = (_parentFunctionKey == null ? getNameKey() : _parentFunctionKey);
_app.showErrorMessage(titleKey, "dialog.deletemarked.nonefound");
}
else
{
Command command = new ShuffleAndCropCmd(indexesToKeep, indexesToDelete,
makeSegmentFlags(trackInfo.getTrack(), indexesToDelete, indexesOfSegments));
_app.execute(command);
}
}
private List<PointFlag> makeSegmentFlags(Track inTrack, List<Integer> inIndexesToDelete,
List<Integer> inIndexesOfSegments)
{
ArrayList<PointFlag> flags = new ArrayList<>();
HashSet<Integer> deleteSet = new HashSet<>(inIndexesToDelete);
HashSet<Integer> segmentSet = new HashSet<>(inIndexesOfSegments);
final int numPoints = inTrack.getNumPoints();
boolean setSegmentBreak = false;
for (int i=0; i<numPoints; i++)
{
DataPoint point = inTrack.getPoint(i);
if (point.isWaypoint()) {
continue;
}
setSegmentBreak = setSegmentBreak || segmentSet.contains(i);
if (!deleteSet.contains(i) && setSegmentBreak)
{
flags.add(new PointFlag(point, true));
setSegmentBreak = false;
}
}
return flags;
}
}
|