AltosSiteMap: better gps check, lower zoom
[fw/altos] / ao-tools / altosui / AltosSiteMap.java
index 25b7779243f4fda1bea9b4be51a2e0380e11007f..72a65b1529721d57f389c01b70914bcfb73319a1 100644 (file)
@@ -21,6 +21,7 @@ import java.awt.*;
 import java.awt.image.*;
 import java.awt.event.*;
 import javax.swing.*;
+import javax.swing.event.MouseInputAdapter;
 import javax.imageio.ImageIO;
 import javax.swing.table.*;
 import java.io.*;
@@ -31,147 +32,238 @@ import java.lang.Math;
 import java.awt.geom.Point2D;
 import java.awt.geom.Line2D;
 
-public class AltosSiteMap extends JComponent implements AltosFlightDisplay {
-    double lat, lng;
-    int zoom;
-    double scale_x, scale_y;
-    Point2D.Double coord_pt;
-    Point2D.Double last_pt;
-
-    Graphics2D g2d;
-
-    private boolean setLocation(double new_lat, double new_lng) {
-        int new_zoom = 15;
-        lat = new_lat;
-        lng = new_lng;
-        zoom = new_zoom;
-        scale_x = 256/360.0 * Math.pow(2, zoom);
-        scale_y = 256/(2.0*Math.PI) * Math.pow(2, zoom);
-        coord_pt = pt(lat, lng, new Point2D.Double(0,0));
-        coord_pt.x = 320-coord_pt.x;
-        coord_pt.y = 320-coord_pt.y;
-        last_pt = null;
-
-        try {
-            File pngfile = new File(AltosPreferences.logdir(), 
-                                    FileCoord(lat, lng, zoom));
-            System.out.printf("Trying file %s\n", pngfile);
-            BufferedImage myPicture = ImageIO.read(pngfile);
-            picLabel.setIcon(new ImageIcon( myPicture ));
-            g2d = myPicture.createGraphics();
-        } catch (Exception e) { 
-            // throw new RuntimeException(e);
-            return false;
-        }
-        return true;
-    }
-
-    private static double limit(double v, double lo, double hi) {
-        if (v < lo)
-            return lo;
-        if (hi < v)
-            return hi;
-        return v;
-    }
-
-    private static String FileCoord(double lat, double lng, int zoom) {
-        char chlat = lat < 0 ? 'S' : 'N';
-        char chlng = lng < 0 ? 'E' : 'W';
-        if (lat < 0) lat = -lat;
-        if (lng < 0) lng = -lng;
-        return String.format("map-%c%.3f,%c%.3f-%d.png",
-                chlat, lat, chlng, lng, zoom);
-    }
-
-
-    // based on google js
-    //  http://maps.gstatic.com/intl/en_us/mapfiles/api-3/2/10/main.js
-    // search for fromLatLngToPoint and fromPointToLatLng
-    private Point2D.Double pt(double lat, double lng) {
-        return pt(lat, lng, coord_pt);
-    }
-
-    private Point2D.Double pt(double lat, double lng, Point2D.Double centre) {
-        Point2D.Double res = new Point2D.Double();
-        double e;
-
-        res.x = centre.x + lng*scale_x;
-        e = limit(Math.sin(Math.toRadians(lat)),-(1-1.0E-15),1-1.0E-15);
-        res.y = centre.y + 0.5*Math.log((1+e)/(1-e))*-scale_y;
-        return res;
-    }
-
-
-    public void reset() {
-        // ?
-    }
-
-    static Color stateColors[] = {
-        Color.WHITE,  // startup
-        Color.WHITE,  // idle
-        Color.WHITE,  // pad
-        Color.RED,    // boost
-        Color.PINK,   // fast
-        Color.YELLOW, // coast
-        Color.CYAN,   // drogue
-        Color.BLUE,   // main
-        Color.BLACK   // landed
-    };
-
-    boolean drawn_landed_circle = false;
-    boolean nomaps = false;
-    public void show(AltosState state, int crc_errors) {
-        if (nomaps)
-            return;
-        if (!state.gps_ready && state.pad_lat == 0 && state.pad_lon == 0)
-            return;
-        double plat = (int)(state.pad_lat*200)/200.0;
-        double plon = (int)(state.pad_lon*200)/200.0;
-
-        if (last_pt == null) {
-            if (!setLocation(plat, plon)) {
-                nomaps = true;
-                return;
-            }
-        }
-
-        Point2D.Double pt = pt(state.gps.lat, state.gps.lon);
-        if (last_pt != null && pt != last_pt) {
-            if (0 <= state.state && state.state < stateColors.length) {
-                g2d.setColor(stateColors[state.state]);
-            }
-            g2d.draw(new Line2D.Double(last_pt, pt));
-        }
-
-        if (state.state == 8 && !drawn_landed_circle) {
-            drawn_landed_circle = true;
-            g2d.setColor(Color.RED);
-            g2d.drawOval((int)pt.x-5, (int)pt.y-5, 10, 10);
-            g2d.drawOval((int)pt.x-20, (int)pt.y-20, 40, 40);
-            g2d.drawOval((int)pt.x-35, (int)pt.y-35, 70, 70);
-        }
-
-        last_pt = pt;
-        repaint();
-    }
-   
-    JLabel picLabel = new JLabel();
-
-    public AltosSiteMap() {
-        GridBagLayout layout = new GridBagLayout();
-        setLayout(layout);
-
-        GridBagConstraints c = new GridBagConstraints();
-
-        c.gridx = 0; c.gridy = 0;
-        c.weightx = 1; c.weighty = 1;
-        c.anchor = GridBagConstraints.CENTER;
-        c.fill = GridBagConstraints.BOTH;
-        picLabel = new JLabel();
-        JScrollPane scrollPane = new JScrollPane(picLabel);
-        layout.setConstraints(scrollPane, c);
-        add(scrollPane);
-
-    }
+public class AltosSiteMap extends JScrollPane implements AltosFlightDisplay {
+       // max vertical step in a tile in naut. miles
+       static final double tile_size_nmi = 2.0;
+
+       static final int px_size = 512;
+
+       private static Point2D.Double translatePoint(Point2D.Double p,
+                       Point2D.Double d)
+       {
+               return new Point2D.Double(p.x + d.x, p.y + d.y);
+       }
+
+       static class LatLng {
+               public double lat, lng;
+               public LatLng(double lat, double lng) {
+                       this.lat = lat;
+                       this.lng = lng;
+               }
+       }
+
+       // based on google js
+       //  http://maps.gstatic.com/intl/en_us/mapfiles/api-3/2/10/main.js
+       // search for fromLatLngToPoint and fromPointToLatLng
+       private static Point2D.Double pt(LatLng latlng, int zoom) {
+               double scale_x = 256/360.0 * Math.pow(2, zoom);
+               double scale_y = 256/(2.0*Math.PI) * Math.pow(2, zoom);
+               return pt(latlng, scale_x, scale_y);
+       }
+
+       private static Point2D.Double pt(LatLng latlng,
+                                        double scale_x, double scale_y)
+       {
+               Point2D.Double res = new Point2D.Double();
+               double e;
+
+               res.x = latlng.lng * scale_x;
+
+               e = Math.sin(Math.toRadians(latlng.lat));
+               e = Math.max(e,-(1-1.0E-15));
+               e = Math.min(e,  1-1.0E-15 );
+
+               res.y = 0.5*Math.log((1+e)/(1-e))*-scale_y;
+               return res;
+       }
+
+       static private LatLng latlng(Point2D.Double pt,
+                                    double scale_x, double scale_y)
+       {
+               double lat, lng;
+               double rads;
+
+               lng = pt.x/scale_x;
+               rads = 2 * Math.atan(Math.exp(-pt.y/scale_y));
+               lat = Math.toDegrees(rads - Math.PI/2);
+
+               return new LatLng(lat,lng);
+       }
+
+       int zoom;
+       double scale_x, scale_y;
+
+       private Point2D.Double pt(double lat, double lng) {
+               return pt(new LatLng(lat, lng), scale_x, scale_y);
+       }
+
+       private LatLng latlng(double x, double y) {
+               return latlng(new Point2D.Double(x,y), scale_x, scale_y);
+       }
+       private LatLng latlng(Point2D.Double pt) {
+               return latlng(pt, scale_x, scale_y);
+       }
+
+       AltosSiteMapTile [] mapTiles = new AltosSiteMapTile[9];
+       Point2D.Double [] tileOffset = new Point2D.Double[9];
+
+       private Point2D.Double getBaseLocation(double lat, double lng) {
+               Point2D.Double locn, north_step;
+
+               zoom = 2;
+               // stupid loop structure to please Java's control flow analysis
+               do {
+                       zoom++;
+                       scale_x = 256/360.0 * Math.pow(2, zoom);
+                       scale_y = 256/(2.0*Math.PI) * Math.pow(2, zoom);
+                       locn = pt(lat, lng);
+                       north_step = pt(lat+tile_size_nmi/60.0, lng);
+                       if (locn.y - north_step.y > px_size)
+                               break;
+               } while (zoom < 22);
+               locn.x = -px_size * Math.floor(locn.x/px_size);
+               locn.y = -px_size * Math.floor(locn.y/px_size);
+               return locn;
+       }
+
+       public void reset() {
+               // nothing
+       }
+
+       private void bgLoadMap(final int i,
+                              final File pngfile, final String pngurl)
+       {
+               Thread thread = new Thread() {
+                       public void run() {
+                               ImageIcon res;
+                               res = AltosSiteMapCache.fetchAndLoadMap(pngfile, pngurl);
+                               if (res != null) {
+                                       mapTiles[i].loadMap(res);
+                               } else {
+                                       System.out.printf("# Failed to fetch file %s\n", pngfile);
+                                       System.out.printf(" wget -O '%s' ''\n", pngfile, pngurl);
+                               }
+                       }
+               };
+               thread.start();
+       }
+
+       public static void prefetchMaps(double lat, double lng, int w, int h) {
+               AltosPreferences.init(null);
+
+               AltosSiteMap asm = new AltosSiteMap(true);
+               Point2D.Double c = asm.getBaseLocation(lat, lng);
+               Point2D.Double p = new Point2D.Double();
+               Point2D.Double p2;
+               int dx = -w/2, dy = -h/2;
+               for (int y = dy; y < h+dy; y++) {
+                       for (int x = dx; x < w+dx; x++) {
+                               LatLng map_latlng = asm.latlng(
+                                                           -c.x + x*px_size + px_size/2,
+                                                           -c.y + y*px_size + px_size/2);
+                               File pngfile = asm.MapFile(map_latlng.lat, map_latlng.lng);
+                               String pngurl = asm.MapURL(map_latlng.lat, map_latlng.lng);
+                               if (pngfile.exists()) {
+                                       System.out.printf("Already have %s\n", pngfile);
+                               } else if (AltosSiteMapCache.fetchMap(pngfile, pngurl)) {
+                                       System.out.printf("Fetched map %s\n", pngfile);
+                               } else {
+                                       System.out.printf("# Failed to fetch file %s\n", pngfile);
+                                       System.out.printf(" wget -O '%s' ''\n", pngfile, pngurl);
+                               }
+                       }
+               }
+       }
+
+       private void initMaps(double lat, double lng) {
+               Point2D.Double c = getBaseLocation(lat, lng);
+               Point2D.Double p = new Point2D.Double();
+
+               for (int i = 0; i < 9; i++) {
+                       int x = i%3 - 1, y = i/3 - 1;
+
+                       tileOffset[i] = new Point2D.Double(
+                               c.x - x*px_size, p.y = c.y - y*px_size);
+                       LatLng map_latlng = latlng(
+                                                   -tileOffset[i].x+px_size/2,
+                                                   -tileOffset[i].y+px_size/2);
+
+                       File pngfile = MapFile(map_latlng.lat, map_latlng.lng);
+                       String pngurl = MapURL(map_latlng.lat, map_latlng.lng);
+                       bgLoadMap(i, pngfile, pngurl);
+               }
+       }
+
+       private File MapFile(double lat, double lng) {
+               char chlat = lat < 0 ? 'S' : 'N';
+               char chlng = lng < 0 ? 'E' : 'W';
+               if (lat < 0) lat = -lat;
+               if (lng < 0) lng = -lng;
+               return new File(AltosPreferences.logdir(),
+                               String.format("map-%c%.6f,%c%.6f-%d.png",
+                                             chlat, lat, chlng, lng, zoom));
+       }
+
+       private String MapURL(double lat, double lng) {
+               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);
+       }
+
+       boolean initialised = false;
+       public void show(AltosState state, int crc_errors) {
+               // if insufficient gps data, nothing to update
+               if (!state.gps.locked) {
+                       if (state.pad_lat == 0 && state.pad_lon == 0)
+                               return;
+                       if (state.gps.nsat < 4)
+                               return;
+               }
+
+               if (!initialised) {
+                       initMaps(state.pad_lat, state.pad_lon);
+                       initialised = true;
+               }
+
+               Point2D.Double pt = pt(state.gps.lat, state.gps.lon);
+               for (int x = 0; x < mapTiles.length; x++) {
+                       mapTiles[x].show(state, crc_errors,
+                                        translatePoint(pt, tileOffset[x]));
+               }
+       }
+
+       private AltosSiteMap(boolean knowWhatYouAreDoing) {
+               if (!knowWhatYouAreDoing) {
+                       throw new RuntimeException("Arggh.");
+               }
+       }
+
+       public AltosSiteMap() {
+               JComponent comp = new JComponent() {
+                       GrabNDrag scroller = new GrabNDrag(this);
+                       {
+                               addMouseMotionListener(scroller);
+                               addMouseListener(scroller);
+                               setAutoscrolls(true);
+                       }
+               };
+
+               GridBagLayout layout = new GridBagLayout();
+               comp.setLayout(layout);
+
+               GridBagConstraints c = new GridBagConstraints();
+               c.anchor = GridBagConstraints.CENTER;
+               c.fill = GridBagConstraints.BOTH;
+
+               // put some space between the map tiles, debugging only
+               // c.insets = new Insets(5, 5, 5, 5);
+               for (int x = 0; x < 9; x++) {
+                       c.gridx = x % 3;
+                       c.gridy = x / 3;
+                       mapTiles[x] = new AltosSiteMapTile(px_size);
+                       layout.setConstraints(mapTiles[x], c);
+                       comp.add(mapTiles[x]);
+               }
+               setViewportView(comp);
+               setPreferredSize(new Dimension(500,200));
+       }
 }