Pileus Git - grits/blob - src/plugins/map.c

   1 /*
   2  * Copyright (C) 2009-2011 Andy Spencer <andy753421@gmail.com>
   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, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 /**
  19  * SECTION:map
  20  * @short_description: Map plugin
  21  *
  22  * #GritsPluginMap provides map overlays. Much of this data is obtained from the
  23  * OpenStreetMap project.
  24  */
  25
  26 #include <time.h>
  27 #include <glib/gstdio.h>
  28
  29 #include <grits.h>
  30
  31 #include "map.h"
  32
  33 #define MAX_RESOLUTION 1
  34 #define TILE_WIDTH     256
  35 #define TILE_HEIGHT    256
  36 //#define MAX_RESOLUTION 100
  37 //#define TILE_WIDTH     1024
  38 //#define TILE_HEIGHT    512
  39
  40 static const guchar colormap[][2][4] = {
  41         //{{0x73, 0x91, 0xad}, {0x73, 0x91, 0xad, 0x00}}, // Oceans
  42         //{{0xf6, 0xee, 0xee}, {0xf6, 0xee, 0xee, 0x00}}, // Ground
  43         //{{0xff, 0xff, 0xff}, {0xff, 0xff, 0xff, 0xff}}, // Borders
  44         //{{0x73, 0x93, 0xad}, {0x73, 0x93, 0xad, 0x40}}, // Lakes
  45         //{{0xff, 0xe1, 0x80}, {0xff, 0xe1, 0x80, 0x60}}, // Cities
  46 };
  47
  48 struct _LoadTileData {
  49         GritsPluginMap *map;
  50         GritsTile      *tile;
  51         guint8         *pixels;
  52         gboolean        alpha;
  53         gint            width;
  54         gint            height;
  55 };
  56 static gboolean _load_tile_cb(gpointer _data)
  57 {
  58         struct _LoadTileData *data = _data;
  59         g_debug("GritsPluginMap: _load_tile_cb start");
  60         if (data->map->aborted) {
  61                 g_debug("GritsPluginMap: _load_tile - aborted");
  62                 return FALSE;
  63         }
  64
  65         guint *tex = g_new0(guint, 1);
  66         glGenTextures(1, tex);
  67         glBindTexture(GL_TEXTURE_2D, *tex);
  68
  69         glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
  70         glPixelStorei(GL_PACK_ALIGNMENT, 1);
  71         glTexImage2D(GL_TEXTURE_2D, 0, 4, data->width, data->height, 0,
  72                         (data->alpha ? GL_RGBA : GL_RGB), GL_UNSIGNED_BYTE, data->pixels);
  73         glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
  74         glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  75         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
  76         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
  77         glFlush();
  78
  79         data->tile->data = tex;
  80         if (data->map->viewer)
  81                 gtk_widget_queue_draw(GTK_WIDGET(data->map->viewer));
  82         g_free(data->pixels);
  83         g_free(data);
  84         return FALSE;
  85 }
  86
  87 static void _load_tile(GritsTile *tile, gpointer _map)
  88 {
  89         GritsPluginMap *map = _map;
  90         g_debug("GritsPluginMap: _load_tile start %p", g_thread_self());
  91         if (map->aborted) {
  92                 g_debug("GritsPluginMap: _load_tile - aborted");
  93                 return;
  94         }
  95
  96         /* Download tile */
  97         gchar *path = grits_tms_fetch(map->tms, tile, GRITS_ONCE, NULL, NULL);
  98         //gchar *path = grits_wms_fetch(map->wms, tile, GRITS_ONCE, NULL, NULL);
  99         if (!path) return; // Canceled/error
 100
 101         /* Load pixbuf */
 102         GdkPixbuf *pixbuf = gdk_pixbuf_new_from_file(path, NULL);
 103         if (!pixbuf) {
 104                 g_warning("GritsPluginMap: _load_tile - Error loading pixbuf %s", path);
 105                 g_remove(path);
 106                 g_free(path);
 107                 return;
 108         }
 109         g_free(path);
 110
 111         /* Copy pixbuf data for callback */
 112         struct _LoadTileData *data = g_new0(struct _LoadTileData, 1);
 113         data->map    = map;
 114         data->tile   = tile;
 115         data->pixels = gdk_pixbuf_get_pixels(pixbuf);
 116         data->alpha  = gdk_pixbuf_get_has_alpha(pixbuf);
 117         data->width  = gdk_pixbuf_get_width(pixbuf);
 118         data->height = gdk_pixbuf_get_height(pixbuf);
 119         data->pixels = g_memdup(data->pixels,
 120                         data->width * data->height * (data->alpha ? 4 : 3));
 121         g_object_unref(pixbuf);
 122
 123         /* Map texture colors, if needed */
 124         for (int i = 0; i < data->width * data->height; i++) {
 125                 for (int j = 0; j < G_N_ELEMENTS(colormap); j++) {
 126                         if (data->pixels[i*4+0] == colormap[j][0][0] &&
 127                             data->pixels[i*4+1] == colormap[j][0][1] &&
 128                             data->pixels[i*4+2] == colormap[j][0][2]) {
 129                                 data->pixels[i*4+0] = colormap[j][1][0];
 130                                 data->pixels[i*4+1] = colormap[j][1][1];
 131                                 data->pixels[i*4+2] = colormap[j][1][2];
 132                                 data->pixels[i*4+3] = colormap[j][1][3];
 133                                 break;
 134                         }
 135                 }
 136         }
 137
 138         /* Load the GL texture from the main thread */
 139         g_idle_add_full(G_PRIORITY_LOW, _load_tile_cb, data, NULL);
 140         g_debug("GritsPluginMap: _load_tile end %p", g_thread_self());
 141 }
 142
 143 static gboolean _free_tile_cb(gpointer data)
 144 {
 145         glDeleteTextures(1, data);
 146         g_free(data);
 147         return FALSE;
 148 }
 149 static void _free_tile(GritsTile *tile, gpointer _map)
 150 {
 151         if (tile->data) {
 152                 g_debug("GritsPluginMap: _free_tile: %p", tile->data);
 153                 g_idle_add_full(G_PRIORITY_LOW, _free_tile_cb, tile->data, NULL);
 154         }
 155 }
 156
 157 static void _update_tiles(gpointer _, gpointer _map)
 158 {
 159         g_debug("GritsPluginMap: _update_tiles");
 160         GritsPluginMap *map = _map;
 161         GritsPoint eye;
 162         grits_viewer_get_location(map->viewer, &eye.lat, &eye.lon, &eye.elev);
 163         grits_tile_update(map->tiles, &eye,
 164                         MAX_RESOLUTION, TILE_WIDTH, TILE_WIDTH,
 165                         _load_tile, map);
 166         grits_tile_gc(map->tiles, time(NULL)-10,
 167                         _free_tile, map);
 168 }
 169
 170 /*************
 171  * Callbacks *
 172  *************/
 173 static void _on_location_changed(GritsViewer *viewer,
 174                 gdouble lat, gdouble lon, gdouble elev, GritsPluginMap *map)
 175 {
 176         g_thread_pool_push(map->threads, NULL+1, NULL);
 177 }
 178
 179 /***********
 180  * Methods *
 181  ***********/
 182 /**
 183  * grits_plugin_map_new:
 184  * @viewer: the #GritsViewer to use for drawing
 185  *
 186  * Create a new instance of the map plugin.
 187  *
 188  * Returns: the new #GritsPluginMap
 189  */
 190 GritsPluginMap *grits_plugin_map_new(GritsViewer *viewer)
 191 {
 192         g_debug("GritsPluginMap: new");
 193         GritsPluginMap *map = g_object_new(GRITS_TYPE_PLUGIN_MAP, NULL);
 194         map->viewer = g_object_ref(viewer);
 195
 196         /* Load initial tiles */
 197         _update_tiles(NULL, map);
 198
 199         /* Connect signals */
 200         map->sigid = g_signal_connect(map->viewer, "location-changed",
 201                         G_CALLBACK(_on_location_changed), map);
 202
 203         /* Add renderers */
 204         grits_viewer_add(viewer, GRITS_OBJECT(map->tiles), GRITS_LEVEL_WORLD, FALSE);
 205
 206         return map;
 207 }
 208
 209
 210 /****************
 211  * GObject code *
 212  ****************/
 213 /* Plugin init */
 214 static void grits_plugin_map_plugin_init(GritsPluginInterface *iface);
 215 G_DEFINE_TYPE_WITH_CODE(GritsPluginMap, grits_plugin_map, G_TYPE_OBJECT,
 216                 G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN,
 217                         grits_plugin_map_plugin_init));
 218 static void grits_plugin_map_plugin_init(GritsPluginInterface *iface)
 219 {
 220         g_debug("GritsPluginMap: plugin_init");
 221         /* Add methods to the interface */
 222 }
 223 /* Class/Object init */
 224 static void grits_plugin_map_init(GritsPluginMap *map)
 225 {
 226         g_debug("GritsPluginMap: init");
 227         /* Set defaults */
 228         map->threads = g_thread_pool_new(_update_tiles, map, 1, FALSE, NULL);
 229         map->tiles = grits_tile_new(NULL, 85.0511, -85.0511, EAST, WEST);
 230         map->tms   = grits_tms_new("http://tile.openstreetmap.org",
 231                 "osmtile/", "png");
 232         map->tiles->proj = GRITS_PROJ_MERCATOR;
 233         //map->tiles = grits_tile_new(NULL, NORTH, SOUTH, EAST, WEST);
 234         //map->wms   = grits_wms_new(
 235         //      "http://vmap0.tiles.osgeo.org/wms/vmap0",
 236         //      "basic,priroad,secroad,depthcontour,clabel,statelabel",
 237         //       "image/png", "osm/", "png", TILE_WIDTH, TILE_HEIGHT);
 238         g_object_ref(map->tiles);
 239 }
 240 static void grits_plugin_map_dispose(GObject *gobject)
 241 {
 242         g_debug("GritsPluginMap: dispose");
 243         GritsPluginMap *map = GRITS_PLUGIN_MAP(gobject);
 244         map->aborted = TRUE;
 245         /* Drop references */
 246         if (map->viewer) {
 247                 GritsViewer *viewer = map->viewer;
 248                 map->viewer = NULL;
 249                 g_signal_handler_disconnect(viewer, map->sigid);
 250                 grits_viewer_remove(viewer, GRITS_OBJECT(map->tiles));
 251                 soup_session_abort(map->tms->http->soup);
 252                 //soup_session_abort(map->wms->http->soup);
 253                 g_thread_pool_free(map->threads, TRUE, TRUE);
 254                 while (gtk_events_pending())
 255                         gtk_main_iteration();
 256                 g_object_unref(viewer);
 257         }
 258         G_OBJECT_CLASS(grits_plugin_map_parent_class)->dispose(gobject);
 259 }
 260 static void grits_plugin_map_finalize(GObject *gobject)
 261 {
 262         g_debug("GritsPluginMap: finalize");
 263         GritsPluginMap *map = GRITS_PLUGIN_MAP(gobject);
 264         /* Free data */
 265         grits_tms_free(map->tms);
 266         //grits_wms_free(map->wms);
 267         grits_tile_free(map->tiles, _free_tile, map);
 268         G_OBJECT_CLASS(grits_plugin_map_parent_class)->finalize(gobject);
 269
 270 }
 271 static void grits_plugin_map_class_init(GritsPluginMapClass *klass)
 272 {
 273         g_debug("GritsPluginMap: class_init");
 274         GObjectClass *gobject_class = (GObjectClass*)klass;
 275         gobject_class->dispose  = grits_plugin_map_dispose;
 276         gobject_class->finalize = grits_plugin_map_finalize;
 277 }