File: InterpolateFunction.java

package info (click to toggle)
gpsprune 26.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,824 kB
  • sloc: java: 52,154; sh: 25; makefile: 21; python: 15
file content (129 lines) | stat: -rw-r--r-- 4,126 bytes parent folder | download | duplicates (4)
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package tim.prune.function;

import java.util.ArrayList;

import javax.swing.JOptionPane;

import tim.prune.App;
import tim.prune.I18nManager;
import tim.prune.cmd.InsertVariousPointsCmd;
import tim.prune.data.DataPoint;
import tim.prune.data.Track;

/**
 * Function to interpolate between the points in a range
 */
public class InterpolateFunction extends SingleNumericParameterFunction
{
	/**
	 * Constructor
	 * @param inApp app object
	 */
	public InterpolateFunction(App inApp) {
		super(inApp, 1, 1000);
	}

	/** @return name key */
	public String getNameKey() {
		return "function.interpolate";
	}

	/** @return description key for input parameter */
	public String getDescriptionKey() {
		return "dialog.interpolate.parameter.text";
	}

	/** @return current (or default) parameter value */
	public int getCurrentParamValue() {
		return 0;
	}

	/**
	 * Complete the function after the input parameter has been chosen
	 */
	public void completeFunction(int inParam)
	{
		// Firstly, work out whether the selected range only contains waypoints or not
		final int startIndex = _app.getTrackInfo().getSelection().getStart();
		final int endIndex   = _app.getTrackInfo().getSelection().getEnd();
		if (startIndex < 0 || endIndex < 0 || endIndex <= startIndex || inParam <= 0) {
			return;
		}

		boolean betweenWaypoints = false;
		// if there are only waypoints, then ask whether to interpolate them
		if (!_app.getTrackInfo().getTrack().isTrackPointWithin(startIndex, endIndex))
		{
			int answer = JOptionPane.showConfirmDialog(_parentFrame,
				I18nManager.getText("dialog.interpolate.betweenwaypoints"),
				getName(), JOptionPane.YES_NO_OPTION);
			if (answer != JOptionPane.YES_OPTION) {
				// user said no (or cancel), so nothing to do
				return;
			}
			betweenWaypoints = true;
		}

		final int numToAdd = inParam;
		final Track track = _app.getTrackInfo().getTrack();
		InsertVariousPointsCmd command = makeCommand(track, startIndex, endIndex, numToAdd, betweenWaypoints);
		if (_app.execute(command))
		{
			final int lastIndex = endIndex + command.getNumInserted();
			_app.getTrackInfo().getSelection().selectRange(startIndex, lastIndex);
			_app.getTrackInfo().selectPoint(lastIndex);
		}
	}

	/**
	 * Create the command to do the interpolation
	 * @param inTrack track object
	 * @param inStartIndex start of selection
	 * @param inEndIndex end of selection, inclusive
	 * @param inNumToAdd number of points to interpolate between each pair
	 * @param inBetweenWaypoints true to also interpolate between waypoints
	 * @return command object
	 */
	private InsertVariousPointsCmd makeCommand(Track inTrack, int inStartIndex, int inEndIndex, int inNumToAdd,
		boolean inBetweenWaypoints)
	{
		ArrayList<Integer> indexes = new ArrayList<>();
		ArrayList<DataPoint> insertedPoints = new ArrayList<>();
		// Points before selected range
		for (int i=0; i<=inStartIndex; i++) {
			indexes.add(i);
		}
		// Interpolate the selection
		DataPoint prevPoint = inTrack.getPoint(inStartIndex);
		for (int i=inStartIndex+1; i<=inEndIndex; i++)
		{
			DataPoint currPoint = inTrack.getPoint(i);
			boolean interpolate = (currPoint.isWaypoint() && prevPoint.isWaypoint() && inBetweenWaypoints)
				|| (!currPoint.isWaypoint() && !prevPoint.isWaypoint() && !currPoint.getSegmentStart());
			if (interpolate)
			{
				for (int j=0; j<inNumToAdd; j++)
				{
					DataPoint newPoint = PointUtils.interpolate(prevPoint, currPoint, j, inNumToAdd);
					if (j == 0 && inBetweenWaypoints) {
						newPoint.setSegmentStart(true);
					}
					insertedPoints.add(newPoint);
					indexes.add(inTrack.getNumPoints() + insertedPoints.size() - 1);
				}
			}
			indexes.add(i);
			if (!currPoint.isWaypoint() || inBetweenWaypoints) {
				prevPoint = currPoint;
			}
		}
		// Points after selected range
		for (int i=inEndIndex+1; i<inTrack.getNumPoints(); i++) {
			indexes.add(i);
		}
		InsertVariousPointsCmd command = new InsertVariousPointsCmd(indexes, insertedPoints);
		command.setDescription(getName());
		command.setConfirmText(I18nManager.getTextWithNumber("confirm.pointsadded", insertedPoints.size()));
		return command;
	}
}