1 package tim.prune.function;
3 import java.awt.BorderLayout;
4 import java.awt.Component;
5 import java.awt.FlowLayout;
6 import java.awt.event.ActionEvent;
7 import java.awt.event.ActionListener;
9 import javax.swing.BorderFactory;
10 import javax.swing.JButton;
11 import javax.swing.JDialog;
12 import javax.swing.JLabel;
13 import javax.swing.JList;
14 import javax.swing.JPanel;
15 import javax.swing.JScrollPane;
16 import javax.swing.ListSelectionModel;
19 import tim.prune.GenericFunction;
20 import tim.prune.I18nManager;
21 import tim.prune.data.Altitude;
22 import tim.prune.data.Field;
23 import tim.prune.data.SourceInfo;
24 import tim.prune.data.Track;
25 import tim.prune.load.TrackNameList;
28 * Function to allow the selection of which tracks to load from the file / stream
30 public class SelectTracksFunction extends GenericFunction
32 private Field[] _fieldArray = null;
33 private Object[][] _dataArray = null;
34 private Altitude.Format _altFormat = Altitude.Format.NO_FORMAT;
35 private SourceInfo _sourceInfo = null;
36 private TrackNameList _trackNameList = null;
37 private JDialog _dialog = null;
38 private JList _trackList = null;
42 * @param inApp app object to use for load
43 * @param inFieldArray field array
44 * @param inDataArray data array
45 * @param inAltFormat altitude format
46 * @param inSourceInfo source information
47 * @param inTrackNameList track name list
49 public SelectTracksFunction(App inApp, Field[] inFieldArray, Object[][] inDataArray,
50 Altitude.Format inAltFormat, SourceInfo inSourceInfo, TrackNameList inTrackNameList)
53 _fieldArray = inFieldArray;
54 _dataArray = inDataArray;
55 _altFormat = inAltFormat;
56 _sourceInfo = inSourceInfo;
57 _trackNameList = inTrackNameList;
65 _dialog = new JDialog(_parentFrame, I18nManager.getText("function.open"));
66 _dialog.setLocationRelativeTo(_parentFrame);
67 _dialog.getContentPane().add(makeContents());
69 _dialog.setVisible(true);
74 * @return the contents of the window as a Component
76 private Component makeContents()
78 JPanel mainPanel = new JPanel();
79 mainPanel.setLayout(new BorderLayout());
80 mainPanel.add(new JLabel(I18nManager.getText("dialog.selecttracks.intro")), BorderLayout.NORTH);
82 final int numTracks = _trackNameList.getNumTracks();
83 String[] names = new String[numTracks];
84 for (int i=0; i<numTracks; i++)
86 String name = _trackNameList.getTrackName(i);
87 if (name == null || name.equals("")) {
88 name = I18nManager.getText("dialog.selecttracks.noname");
90 names[i] = name + " (" + _trackNameList.getNumPointsInTrack(i) + ")";
92 _trackList = new JList(names);
93 _trackList.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);
94 mainPanel.add(new JScrollPane(_trackList), BorderLayout.CENTER);
96 JButton selectAllButton = new JButton(I18nManager.getText("button.selectall"));
97 selectAllButton.addActionListener(new ActionListener() {
98 public void actionPerformed(ActionEvent e) {
102 JPanel eastPanel = new JPanel();
103 eastPanel.add(selectAllButton);
104 mainPanel.add(eastPanel, BorderLayout.EAST);
106 // button panel at bottom
107 JPanel buttonPanel = new JPanel();
108 buttonPanel.setLayout(new FlowLayout(FlowLayout.RIGHT));
109 JButton okButton = new JButton(I18nManager.getText("button.ok"));
110 okButton.addActionListener(new ActionListener() {
111 public void actionPerformed(ActionEvent e)
116 buttonPanel.add(okButton);
117 JButton cancelButton = new JButton(I18nManager.getText("button.cancel"));
118 cancelButton.addActionListener(new ActionListener() {
119 public void actionPerformed(ActionEvent e)
122 _app.informNoDataLoaded();
125 buttonPanel.add(cancelButton);
126 mainPanel.add(buttonPanel, BorderLayout.SOUTH);
127 mainPanel.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 15));
131 /** @return name key */
132 public String getNameKey() {
133 return "dialog.selecttracks";
137 * Select all the tracks in the list
139 private void selectAll()
141 _trackList.setSelectionInterval(0, _trackNameList.getNumTracks()-1);
145 * OK pressed, so finish the load
147 private void finish()
150 // Load track as it is, which is expensive but makes interrogating waypoints easier
151 Track loadedTrack = new Track();
152 loadedTrack.load(_fieldArray, _dataArray, _altFormat);
153 int[] tracks = _trackList.getSelectedIndices();
154 // Check if all tracks are selected, then don't have to filter at all
155 if (tracks.length == _trackNameList.getNumTracks()) {
156 _app.informDataLoaded(loadedTrack, _sourceInfo);
160 // build array of which tracks have been selected
161 boolean[] selectedTracks = new boolean[_trackNameList.getNumTracks()];
162 for (int i=0; i<tracks.length; i++) {
163 selectedTracks[tracks[i]] = true;
165 // Loop over all points, counting points which survive filter and making flag array
166 int numPointsSelected = 0;
167 int currentTrack = -1;
168 final int totalPoints = loadedTrack.getNumPoints();
169 boolean[] selectedPoints = new boolean[totalPoints];
170 for (int i=0; i<totalPoints; i++)
172 final int startOfNextTrack = _trackNameList.getStartIndex(currentTrack+1);
173 if (i == startOfNextTrack) {currentTrack++;}
174 if (currentTrack < 0 || selectedTracks[currentTrack] || loadedTrack.getPoint(i).isWaypoint()) {
175 selectedPoints[i] = true;
179 // If none of the points have been selected, then load nothing
180 if (numPointsSelected <= 0) {
181 _app.informNoDataLoaded();
184 // Create new data array of required length
185 Object[][] newDataArray = new String[numPointsSelected][];
186 // Loop over all points again, copying surviving points
188 for (int i=0; i<totalPoints; i++)
190 if (selectedPoints[i]) {
191 newDataArray[currPoint] = _dataArray[i];
195 // Construct Track and call informDataLoaded
196 Track filteredTrack = new Track();
197 filteredTrack.load(_fieldArray, newDataArray, _altFormat);
198 // Tell source info object which points were selected (pass selectedPoints array)
199 _sourceInfo.setPointIndices(selectedPoints);
200 _app.informDataLoaded(filteredTrack, _sourceInfo);