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 100
  34 #define TILE_WIDTH     1024
  35 #define TILE_HEIGHT    512
  36
  37 static const guchar colormap[][2][4] = {
  38         {{0x73, 0x91, 0xad}, {0x73, 0x91, 0xad, 0x20}}, // Oceans
  39         {{0xf6, 0xee, 0xee}, {0xf6, 0xee, 0xee, 0x00}}, // Ground
  40         {{0xff, 0xff, 0xff}, {0xff, 0xff, 0xff, 0xff}}, // Borders
  41         {{0x73, 0x93, 0xad}, {0x73, 0x93, 0xad, 0x40}}, // Lakes
  42         {{0xff, 0xe1, 0x80}, {0xff, 0xe1, 0x80, 0x60}}, // Cities
  43 };
  44
  45 struct _LoadTileData {
  46         GritsPluginMap *map;
  47         GritsTile      *tile;
  48         guint8         *pixels;
  49         gboolean        alpha;
  50         gint            width;
  51         gint            height;
  52 };
  53 static gboolean _load_tile_cb(gpointer _data)
  54 {
  55         struct _LoadTileData *data = _data;
  56         g_debug("GritsPluginMap: _load_tile_cb start");
  57
  58         guint *tex = g_new0(guint, 1);
  59         glGenTextures(1, tex);
  60         glBindTexture(GL_TEXTURE_2D, *tex);
  61
  62         glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
  63         glPixelStorei(GL_PACK_ALIGNMENT, 1);
  64         glTexImage2D(GL_TEXTURE_2D, 0, 4, data->width, data->height, 0,
  65                         (data->alpha ? GL_RGBA : GL_RGB), GL_UNSIGNED_BYTE, data->pixels);
  66         glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
  67         glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  68         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP);
  69         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP);
  70         glFlush();
  71
  72         data->tile->data = tex;
  73         gtk_widget_queue_draw(GTK_WIDGET(data->map->viewer));
  74         g_free(data->pixels);
  75         g_free(data);
  76         return FALSE;
  77 }
  78
  79 static void _load_tile(GritsTile *tile, gpointer _map)
  80 {
  81         GritsPluginMap *map = _map;
  82         g_debug("GritsPluginMap: _load_tile start %p", g_thread_self());
  83         if (map->aborted) {
  84                 g_debug("GritsPluginMap: _load_tile - aborted");
  85                 return;
  86         }
  87
  88         /* Download tile */
  89         gchar *path = grits_wms_fetch(map->wms, tile, GRITS_ONCE, NULL, NULL);
  90         if (!path) return; // Canceled/error
  91
  92         /* Load pixbuf */
  93         GdkPixbuf *pixbuf = gdk_pixbuf_new_from_file(path, NULL);
  94         if (!pixbuf) {
  95                 g_warning("GritsPluginMap: _load_tile - Error loading pixbuf %s", path);
  96                 g_remove(path);
  97                 g_free(path);
  98                 return;
  99         }
 100         g_free(path);
 101
 102         /* Copy pixbuf data for callback */
 103         struct _LoadTileData *data = g_new0(struct _LoadTileData, 1);
 104         data->map    = map;
 105         data->tile   = tile;
 106         data->pixels = gdk_pixbuf_get_pixels(pixbuf);
 107         data->alpha  = gdk_pixbuf_get_has_alpha(pixbuf);
 108         data->width  = gdk_pixbuf_get_width(pixbuf);
 109         data->height = gdk_pixbuf_get_height(pixbuf);
 110         data->pixels = g_memdup(data->pixels,
 111                         data->width * data->height * (data->alpha ? 4 : 3));
 112         g_object_unref(pixbuf);
 113
 114         /* Map texture colors, if needed */
 115         for (int i = 0; i < data->width * data->height; i++) {
 116                 for (int j = 0; j < G_N_ELEMENTS(colormap); j++) {
 117                         if (data->pixels[i*4+0] == colormap[j][0][0] &&
 118                             data->pixels[i*4+1] == colormap[j][0][1] &&
 119                             data->pixels[i*4+2] == colormap[j][0][2]) {
 120                                 data->pixels[i*4+0] = colormap[j][1][0];
 121                                 data->pixels[i*4+1] = colormap[j][1][1];
 122                                 data->pixels[i*4+2] = colormap[j][1][2];
 123                                 data->pixels[i*4+3] = colormap[j][1][3];
 124                                 break;
 125                         }
 126                 }
 127         }
 128
 129         /* Load the GL texture from the main thread */
 130         g_idle_add_full(G_PRIORITY_LOW, _load_tile_cb, data, NULL);
 131         g_debug("GritsPluginMap: _load_tile end %p", g_thread_self());
 132 }
 133
 134 static gboolean _free_tile_cb(gpointer data)
 135 {
 136         glDeleteTextures(1, data);
 137         g_free(data);
 138         return FALSE;
 139 }
 140 static void _free_tile(GritsTile *tile, gpointer _map)
 141 {
 142         g_debug("GritsPluginMap: _free_tile: %p", tile->data);
 143         if (tile->data)
 144                 g_idle_add_full(G_PRIORITY_LOW, _free_tile_cb, tile->data, NULL);
 145 }
 146
 147 static void _update_tiles(gpointer _, gpointer _map)
 148 {
 149         g_debug("GritsPluginMap: _update_tiles");
 150         GritsPluginMap *map = _map;
 151         GritsPoint eye;
 152         grits_viewer_get_location(map->viewer, &eye.lat, &eye.lon, &eye.elev);
 153         grits_tile_update(map->tiles, &eye,
 154                         MAX_RESOLUTION, TILE_WIDTH, TILE_WIDTH,
 155                         _load_tile, map);
 156         grits_tile_gc(map->tiles, time(NULL)-10,
 157                         _free_tile, map);
 158 }
 159
 160 /*************
 161  * Callbacks *
 162  *************/
 163 static void _on_location_changed(GritsViewer *viewer,
 164                 gdouble lat, gdouble lon, gdouble elev, GritsPluginMap *map)
 165 {
 166         g_thread_pool_push(map->threads, NULL+1, NULL);
 167 }
 168
 169 /***********
 170  * Methods *
 171  ***********/
 172 /**
 173  * grits_plugin_map_new:
 174  * @viewer: the #GritsViewer to use for drawing
 175  *
 176  * Create a new instance of the map plugin.
 177  *
 178  * Returns: the new #GritsPluginMap
 179  */
 180 GritsPluginMap *grits_plugin_map_new(GritsViewer *viewer)
 181 {
 182         g_debug("GritsPluginMap: new");
 183         GritsPluginMap *map = g_object_new(GRITS_TYPE_PLUGIN_MAP, NULL);
 184         map->viewer = g_object_ref(viewer);
 185
 186         /* Load initial tiles */
 187         _load_tile(map->tiles, map);
 188         _update_tiles(NULL, map);
 189
 190         /* Connect signals */
 191         map->sigid = g_signal_connect(map->viewer, "location-changed",
 192                         G_CALLBACK(_on_location_changed), map);
 193
 194         /* Add renderers */
 195         grits_viewer_add(viewer, GRITS_OBJECT(map->tiles), GRITS_LEVEL_OVERLAY-1, 0);
 196
 197         return map;
 198 }
 199
 200
 201 /****************
 202  * GObject code *
 203  ****************/
 204 /* Plugin init */
 205 static void grits_plugin_map_plugin_init(GritsPluginInterface *iface);
 206 G_DEFINE_TYPE_WITH_CODE(GritsPluginMap, grits_plugin_map, G_TYPE_OBJECT,
 207                 G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN,
 208                         grits_plugin_map_plugin_init));
 209 static void grits_plugin_map_plugin_init(GritsPluginInterface *iface)
 210 {
 211         g_debug("GritsPluginMap: plugin_init");
 212         /* Add methods to the interface */
 213 }
 214 /* Class/Object init */
 215 static void grits_plugin_map_init(GritsPluginMap *map)
 216 {
 217         g_debug("GritsPluginMap: init");
 218         /* Set defaults */
 219         map->threads = g_thread_pool_new(_update_tiles, map, 1, FALSE, NULL);
 220         map->tiles = grits_tile_new(NULL, NORTH, SOUTH, EAST, WEST);
 221         map->wms   = grits_wms_new(
 222                 "http://vmap0.tiles.osgeo.org/wms/vmap0",
 223                 "basic,priroad,secroad,depthcontour,clabel,statelabel",
 224                  "image/png", "osm/", "png", TILE_WIDTH, TILE_HEIGHT);
 225         g_object_ref(map->tiles);
 226 }
 227 static void grits_plugin_map_dispose(GObject *gobject)
 228 {
 229         g_debug("GritsPluginMap: dispose");
 230         GritsPluginMap *map = GRITS_PLUGIN_MAP(gobject);
 231         map->aborted = TRUE;
 232         /* Drop references */
 233         if (map->viewer) {
 234                 g_signal_handler_disconnect(map->viewer, map->sigid);
 235                 grits_viewer_remove(map->viewer, GRITS_OBJECT(map->tiles));
 236                 soup_session_abort(map->wms->http->soup);
 237                 g_thread_pool_free(map->threads, TRUE, TRUE);
 238                 while (gtk_events_pending())
 239                         gtk_main_iteration();
 240                 g_object_unref(map->viewer);
 241                 map->viewer = NULL;
 242         }
 243         G_OBJECT_CLASS(grits_plugin_map_parent_class)->dispose(gobject);
 244 }
 245 static void grits_plugin_map_finalize(GObject *gobject)
 246 {
 247         g_debug("GritsPluginMap: finalize");
 248         GritsPluginMap *map = GRITS_PLUGIN_MAP(gobject);
 249         /* Free data */
 250         grits_wms_free(map->wms);
 251         grits_tile_free(map->tiles, _free_tile, map);
 252         G_OBJECT_CLASS(grits_plugin_map_parent_class)->finalize(gobject);
 253
 254 }
 255 static void grits_plugin_map_class_init(GritsPluginMapClass *klass)
 256 {
 257         g_debug("GritsPluginMap: class_init");
 258         GObjectClass *gobject_class = (GObjectClass*)klass;
 259         gobject_class->dispose  = grits_plugin_map_dispose;
 260         gobject_class->finalize = grits_plugin_map_finalize;
 261 }