Merge branch 'telemini' into telebt
[fw/altos] / altosui / AltosSiteMap.java
1 /*
2  * Copyright © 2010 Anthony Towns <aj@erisian.com.au>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 package altosui;
19
20 import java.awt.*;
21 import java.awt.image.*;
22 import java.awt.event.*;
23 import javax.swing.*;
24 import javax.swing.event.MouseInputAdapter;
25 import javax.imageio.ImageIO;
26 import javax.swing.table.*;
27 import java.io.*;
28 import java.util.*;
29 import java.text.*;
30 import java.util.prefs.*;
31 import java.lang.Math;
32 import java.awt.geom.Point2D;
33 import java.awt.geom.Line2D;
34
35 public class AltosSiteMap extends JScrollPane implements AltosFlightDisplay {
36         // preferred vertical step in a tile in naut. miles
37         // will actually choose a step size between x and 2x, where this
38         // is 1.5x
39         static final double tile_size_nmi = 0.75;
40
41         static final int px_size = 512;
42
43         static final int MAX_TILE_DELTA = 100;
44
45         private static Point2D.Double translatePoint(Point2D.Double p,
46                         Point2D.Double d)
47         {
48                 return new Point2D.Double(p.x + d.x, p.y + d.y);
49         }
50
51         static class LatLng {
52                 public double lat, lng;
53                 public LatLng(double lat, double lng) {
54                         this.lat = lat;
55                         this.lng = lng;
56                 }
57         }
58
59         // based on google js
60         //  http://maps.gstatic.com/intl/en_us/mapfiles/api-3/2/10/main.js
61         // search for fromLatLngToPoint and fromPointToLatLng
62         private static Point2D.Double pt(LatLng latlng, int zoom) {
63                 double scale_x = 256/360.0 * Math.pow(2, zoom);
64                 double scale_y = 256/(2.0*Math.PI) * Math.pow(2, zoom);
65                 return pt(latlng, scale_x, scale_y);
66         }
67
68         private static Point2D.Double pt(LatLng latlng,
69                                          double scale_x, double scale_y)
70         {
71                 Point2D.Double res = new Point2D.Double();
72                 double e;
73
74                 res.x = latlng.lng * scale_x;
75
76                 e = Math.sin(Math.toRadians(latlng.lat));
77                 e = Math.max(e,-(1-1.0E-15));
78                 e = Math.min(e,  1-1.0E-15 );
79
80                 res.y = 0.5*Math.log((1+e)/(1-e))*-scale_y;
81                 return res;
82         }
83
84         static private LatLng latlng(Point2D.Double pt,
85                                      double scale_x, double scale_y)
86         {
87                 double lat, lng;
88                 double rads;
89
90                 lng = pt.x/scale_x;
91                 rads = 2 * Math.atan(Math.exp(-pt.y/scale_y));
92                 lat = Math.toDegrees(rads - Math.PI/2);
93
94                 return new LatLng(lat,lng);
95         }
96
97         int zoom;
98         double scale_x, scale_y;
99
100         private Point2D.Double pt(double lat, double lng) {
101                 return pt(new LatLng(lat, lng), scale_x, scale_y);
102         }
103
104         private LatLng latlng(double x, double y) {
105                 return latlng(new Point2D.Double(x,y), scale_x, scale_y);
106         }
107         private LatLng latlng(Point2D.Double pt) {
108                 return latlng(pt, scale_x, scale_y);
109         }
110
111         HashMap<Point,AltosSiteMapTile> mapTiles = new HashMap<Point,AltosSiteMapTile>();
112         Point2D.Double centre;
113
114         private Point2D.Double tileCoordOffset(Point p) {
115                 return new Point2D.Double(centre.x - p.x*px_size,
116                                           centre.y - p.y * px_size);
117         }
118
119         private Point tileOffset(Point2D.Double p) {
120                 return new Point((int)Math.floor((centre.x+p.x)/px_size),
121                                  (int)Math.floor((centre.y+p.y)/px_size));
122         }
123
124         private Point2D.Double getBaseLocation(double lat, double lng) {
125                 Point2D.Double locn, north_step;
126
127                 zoom = 2;
128                 // stupid loop structure to please Java's control flow analysis
129                 do {
130                         zoom++;
131                         scale_x = 256/360.0 * Math.pow(2, zoom);
132                         scale_y = 256/(2.0*Math.PI) * Math.pow(2, zoom);
133                         locn = pt(lat, lng);
134                         north_step = pt(lat+tile_size_nmi*4/3/60.0, lng);
135                         if (locn.y - north_step.y > px_size)
136                                 break;
137                 } while (zoom < 22);
138                 locn.x = -px_size * Math.floor(locn.x/px_size);
139                 locn.y = -px_size * Math.floor(locn.y/px_size);
140                 return locn;
141         }
142
143         public void reset() {
144                 // nothing
145         }
146
147         private void bgLoadMap(final AltosSiteMapTile tile,
148                                final File pngfile, final String pngurl)
149         {
150                 //System.out.printf("Loading/fetching map %s\n", pngfile);
151                 Thread thread = new Thread() {
152                         public void run() {
153                                 ImageIcon res;
154                                 res = AltosSiteMapCache.fetchAndLoadMap(pngfile, pngurl);
155                                 if (res != null) {
156                                         tile.loadMap(res);
157                                 } else {
158                                         System.out.printf("# Failed to fetch file %s\n", pngfile);
159                                         System.out.printf(" wget -O '%s' ''\n", pngfile, pngurl);
160                                 }
161                         }
162                 };
163                 thread.start();
164         }
165
166         public static void prefetchMaps(double lat, double lng, int w, int h) {
167                 AltosSiteMap asm = new AltosSiteMap(true);
168                 asm.centre = asm.getBaseLocation(lat, lng);
169
170                 Point2D.Double p = new Point2D.Double();
171                 Point2D.Double p2;
172                 int dx = -w/2, dy = -h/2;
173                 for (int y = dy; y < h+dy; y++) {
174                         for (int x = dx; x < w+dx; x++) {
175                                 LatLng map_latlng = asm.latlng(
176                                                             -asm.centre.x + x*px_size + px_size/2,
177                                                             -asm.centre.y + y*px_size + px_size/2);
178                                 File pngfile = asm.MapFile(map_latlng.lat, map_latlng.lng);
179                                 String pngurl = asm.MapURL(map_latlng.lat, map_latlng.lng);
180                                 if (pngfile.exists()) {
181                                         System.out.printf("Already have %s\n", pngfile);
182                                 } else if (AltosSiteMapCache.fetchMap(pngfile, pngurl)) {
183                                         System.out.printf("Fetched map %s\n", pngfile);
184                                 } else {
185                                         System.out.printf("# Failed to fetch file %s\n", pngfile);
186                                         System.out.printf(" wget -O '%s' ''\n", pngfile, pngurl);
187                                 }
188                         }
189                 }
190         }
191
192         private void initMap(AltosSiteMapTile tile, Point offset) {
193                 Point2D.Double coord = tileCoordOffset(offset);
194
195                 LatLng map_latlng = latlng(px_size/2-coord.x, px_size/2-coord.y);
196
197                 File pngfile = MapFile(map_latlng.lat, map_latlng.lng);
198                 String pngurl = MapURL(map_latlng.lat, map_latlng.lng);
199                 bgLoadMap(tile, pngfile, pngurl);
200         }
201
202         private void initMaps(double lat, double lng) {
203                 centre = getBaseLocation(lat, lng);
204
205                 for (Point k : mapTiles.keySet()) {
206                         initMap(mapTiles.get(k), k);
207                 }
208         }
209
210         private File MapFile(double lat, double lng) {
211                 char chlat = lat < 0 ? 'S' : 'N';
212                 char chlng = lng < 0 ? 'W' : 'E';
213                 if (lat < 0) lat = -lat;
214                 if (lng < 0) lng = -lng;
215                 return new File(AltosPreferences.mapdir(),
216                                 String.format("map-%c%.6f,%c%.6f-%d.png",
217                                               chlat, lat, chlng, lng, zoom));
218         }
219
220         private String MapURL(double lat, double lng) {
221                 return String.format("http://maps.google.com/maps/api/staticmap?center=%.6f,%.6f&zoom=%d&size=%dx%d&sensor=false&maptype=hybrid&format=png32", lat, lng, zoom, px_size, px_size);
222         }
223
224         boolean initialised = false;
225         Point2D.Double last_pt = null;
226         int last_state = -1;
227         public void show(final AltosState state, final int crc_errors) {
228                 // if insufficient gps data, nothing to update
229                 if (!state.gps.locked && state.gps.nsat < 4)
230                         return;
231
232                 if (!initialised) {
233                         if (state.pad_lat != 0 || state.pad_lon != 0) {
234                                 initMaps(state.pad_lat, state.pad_lon);
235                                 initialised = true;
236                         } else if (state.gps.lat != 0 || state.gps.lon != 0) {
237                                 initMaps(state.gps.lat, state.gps.lon);
238                                 initialised = true;
239                         } else {
240                                 return;
241                         }
242                 }
243
244                 final Point2D.Double pt = pt(state.gps.lat, state.gps.lon);
245                 if (last_pt == pt && last_state == state.state)
246                         return;
247
248                 if (last_pt == null) {
249                         last_pt = pt;
250                 }
251                 boolean in_any = false;
252                 for (Point offset : mapTiles.keySet()) {
253                         AltosSiteMapTile tile = mapTiles.get(offset);
254                         Point2D.Double ref, lref;
255                         ref = translatePoint(pt, tileCoordOffset(offset));
256                         lref = translatePoint(last_pt, tileCoordOffset(offset));
257                         tile.show(state, crc_errors, lref, ref);
258                         if (0 <= ref.x && ref.x < px_size)
259                                 if (0 <= ref.y && ref.y < px_size)
260                                         in_any = true;
261                 }
262
263                 Point offset = tileOffset(pt);
264                 if (!in_any) {
265                         Point2D.Double ref, lref;
266                         ref = translatePoint(pt, tileCoordOffset(offset));
267                         lref = translatePoint(last_pt, tileCoordOffset(offset));
268
269                         AltosSiteMapTile tile = createTile(offset);
270                         tile.show(state, crc_errors, lref, ref);
271                         initMap(tile, offset);
272                         finishTileLater(tile, offset);
273                 }
274
275                 scrollRocketToVisible(pt);
276
277                 if (offset != tileOffset(last_pt)) {
278                         ensureTilesAround(offset);
279                 }
280
281                 last_pt = pt;
282                 last_state = state.state;
283         }
284
285         private AltosSiteMapTile createTile(Point offset) {
286                 AltosSiteMapTile tile = new AltosSiteMapTile(px_size);
287                 mapTiles.put(offset, tile);
288                 return tile;
289         }
290         private void finishTileLater(final AltosSiteMapTile tile,
291                                      final Point offset)
292         {
293                 SwingUtilities.invokeLater( new Runnable() {
294                         public void run() {
295                                 addTileAt(tile, offset);
296                         }
297                 } );
298         }
299
300         private void ensureTilesAround(Point base_offset) {
301                 for (int x = -1; x <= 1; x++) {
302                         for (int y = -1; y <= 1; y++) {
303                                 Point offset = new Point(base_offset.x + x, base_offset.y + y);
304                                 if (mapTiles.containsKey(offset))
305                                         continue;
306                                 AltosSiteMapTile tile = createTile(offset);
307                                 initMap(tile, offset);
308                                 finishTileLater(tile, offset);
309                         }
310                 }
311         }
312
313         private Point topleft = new Point(0,0);
314         private void scrollRocketToVisible(Point2D.Double pt) {
315                 Rectangle r = comp.getVisibleRect();
316                 Point2D.Double copt = translatePoint(pt, tileCoordOffset(topleft));
317                 int dx = (int)copt.x - r.width/2 - r.x;
318                 int dy = (int)copt.y - r.height/2 - r.y;
319                 if (Math.abs(dx) > r.width/4 || Math.abs(dy) > r.height/4) {
320                         r.x += dx;
321                         r.y += dy;
322                         comp.scrollRectToVisible(r);
323                 }
324         }
325
326         private void addTileAt(AltosSiteMapTile tile, Point offset) {
327                 if (Math.abs(offset.x) >= MAX_TILE_DELTA ||
328                                 Math.abs(offset.y) >= MAX_TILE_DELTA)
329                 {
330                         System.out.printf("Rocket too far away from pad (tile %d,%d)\n",
331                                           offset.x, offset.y);
332                         return;
333                 }
334
335                 boolean review = false;
336                 Rectangle r = comp.getVisibleRect();
337                 if (offset.x < topleft.x) {
338                         r.x += (topleft.x - offset.x) * px_size;
339                         topleft.x = offset.x;
340                         review = true;
341                 }
342                 if (offset.y < topleft.y) {
343                         r.y += (topleft.y - offset.y) * px_size;
344                         topleft.y = offset.y;
345                         review = true;
346                 }
347                 GridBagConstraints c = new GridBagConstraints();
348                 c.anchor = GridBagConstraints.CENTER;
349                 c.fill = GridBagConstraints.BOTH;
350                 // put some space between the map tiles, debugging only
351                 // c.insets = new Insets(5, 5, 5, 5);
352
353                 c.gridx = offset.x + MAX_TILE_DELTA;
354                 c.gridy = offset.y + MAX_TILE_DELTA;
355                 layout.setConstraints(tile, c);
356
357                 comp.add(tile);
358                 if (review) {
359                         comp.scrollRectToVisible(r);
360                 }
361         }
362
363         private AltosSiteMap(boolean knowWhatYouAreDoing) {
364                 if (!knowWhatYouAreDoing) {
365                         throw new RuntimeException("Arggh.");
366                 }
367         }
368
369         JComponent comp = new JComponent() { };
370         private GridBagLayout layout = new GridBagLayout();
371
372         public AltosSiteMap() {
373                 GrabNDrag scroller = new GrabNDrag(comp);
374
375                 comp.setLayout(layout);
376
377                 for (int x = -1; x <= 1; x++) {
378                         for (int y = -1; y <= 1; y++) {
379                                 Point offset = new Point(x, y);
380                                 AltosSiteMapTile t = createTile(offset);
381                                 addTileAt(t, offset);
382                         }
383                 }
384                 setViewportView(comp);
385                 setPreferredSize(new Dimension(500,200));
386         }
387 }