/* * Copyright (C) 2009-2011 Andy Spencer * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ /** * SECTION:elev * @short_description: Elevation plugin * * #GritsPluginElev provides access to ground elevation. It does this in two ways: * First, it provides a height function used by the viewer when drawing the * world. Second, it can load the elevation data into an image and draw a * greyscale elevation overlay on the planets surface. */ #include #include #include #include "elev.h" /* Configuration */ #define LOAD_BIL TRUE #define LOAD_TEX FALSE /* Tile size constnats */ #define MAX_RESOLUTION 50 #define TILE_WIDTH 1024 #define TILE_HEIGHT 512 #define TILE_CHANNELS 4 #define TILE_SIZE (TILE_WIDTH*TILE_HEIGHT*sizeof(guint16)) struct _TileData { /* OpenGL has to be first to make grits_tile_draw happy */ guint tex; guint16 *bil; }; static gdouble _height_func(gdouble lat, gdouble lon, gpointer _elev) { GritsPluginElev *elev = _elev; if (!elev) return 0; GritsTile *tile = grits_tile_find(elev->tiles, lat, lon); if (!tile) return 0; struct _TileData *data = tile->data; if (!data) return 0; guint16 *bil = data->bil; if (!bil) return 0; gint w = TILE_WIDTH; gint h = TILE_HEIGHT; gdouble ymin = tile->edge.s; gdouble ymax = tile->edge.n; gdouble xmin = tile->edge.w; gdouble xmax = tile->edge.e; gdouble xdist = xmax - xmin; gdouble ydist = ymax - ymin; gdouble x = (lon-xmin)/xdist * w; gdouble y = (1-(lat-ymin)/ydist) * h; gdouble x_rem = x - (int)x; gdouble y_rem = y - (int)y; guint x_flr = (int)x; guint y_flr = (int)y; //if (lon == 180 || lon == -180) // g_message("lon=%f w=%d min=%f max=%f dist=%f x=%f rem=%f flr=%d", // lon, w, xmin, xmax, xdist, x, x_rem, x_flr); /* TODO: Fix interpolation at edges: * - Pad these at the edges instead of wrapping/truncating * - Figure out which pixels to index (is 0,0 edge, center, etc) */ gint16 px00 = bil[MIN((y_flr ),h-1)*w + MIN((x_flr ),w-1)]; gint16 px10 = bil[MIN((y_flr ),h-1)*w + MIN((x_flr+1),w-1)]; gint16 px01 = bil[MIN((y_flr+1),h-1)*w + MIN((x_flr ),w-1)]; gint16 px11 = bil[MIN((y_flr+1),h-1)*w + MIN((x_flr+1),w-1)]; return px00 * (1-x_rem) * (1-y_rem) + px10 * ( x_rem) * (1-y_rem) + px01 * (1-x_rem) * ( y_rem) + px11 * ( x_rem) * ( y_rem); } /********************** * Loader and Freeers * **********************/ struct _LoadTileData { GritsPluginElev *elev; GritsTile *tile; guint8 *pixels; struct _TileData *tdata; }; static guint16 *_load_bil(gchar *path) { gsize len; gchar *data = NULL; g_file_get_contents(path, &data, &len, NULL); g_debug("GritsPluginElev: load_bil %p", data); if (len != TILE_SIZE) { g_warning("GritsPluginElev: _load_bil - unexpected tile size %ld, != %ld", (glong)len, (glong)TILE_SIZE); g_free(data); return NULL; } return (guint16*)data; } static guchar *_load_pixels(guint16 *bil) { g_assert(TILE_CHANNELS == 4); guchar (*pixels)[TILE_WIDTH][TILE_CHANNELS] = g_malloc0(TILE_HEIGHT * TILE_WIDTH * TILE_CHANNELS); for (int r = 0; r < TILE_HEIGHT; r++) { for (int c = 0; c < TILE_WIDTH; c++) { gint16 value = bil[r*TILE_WIDTH + c]; guchar color = (float)value/8848 * 255; //guchar color = (float)(MAX(value,0))/8848 * 255; pixels[r][c][0] = color; pixels[r][c][1] = color; pixels[r][c][2] = color; pixels[r][c][3] = 0xff; } } g_debug("GritsPluginElev: load_pixels %p", pixels); return (guchar*)pixels; } static gboolean _load_tile_cb(gpointer _data) { struct _LoadTileData *data = _data; struct _TileData *tdata = data->tdata; g_debug("GritsPluginElev: _load_tile_cb start"); if (data->elev->aborted) { g_debug("GritsPluginElev: _load_tile - aborted"); return FALSE; } /* Load OpenGL texture (from main thread) */ if (data->pixels) { glGenTextures(1, &tdata->tex); glBindTexture(GL_TEXTURE_2D, tdata->tex); glPixelStorei(GL_UNPACK_ALIGNMENT, 1); glPixelStorei(GL_PACK_ALIGNMENT, 1); glTexImage2D(GL_TEXTURE_2D, 0, TILE_CHANNELS, TILE_WIDTH, TILE_HEIGHT, 0, GL_RGBA, GL_UNSIGNED_BYTE, data->pixels); glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST); glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST); glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE); glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE); glFlush(); } /* Set hight function (from main thread) */ if (tdata->bil) { grits_viewer_set_height_func(data->elev->viewer, &data->tile->edge, _height_func, data->elev, TRUE); } /* Queue tiles for drawing */ data->tile->data = tdata; gtk_widget_queue_draw(GTK_WIDGET(data->elev->viewer)); /* Cleanup */ g_free(data->pixels); g_free(data); return FALSE; } static void _load_tile(GritsTile *tile, gpointer _elev) { GritsPluginElev *elev = _elev; guint16 *bil = NULL; guchar *pixels = NULL; g_debug("GritsPluginElev: _load_tile start %p", g_thread_self()); if (elev->aborted) { g_debug("GritsPluginElev: _load_tile - aborted"); return; } /* Download tile */ gchar *path = grits_wms_fetch(elev->wms, tile, GRITS_ONCE, NULL, NULL); if (!path) return; /* Load bil */ bil = _load_bil(path); g_free(path); if (!bil) return; /* Load pixels */ if (LOAD_TEX) pixels = _load_pixels(bil); if (!LOAD_BIL) g_free(bil); /* Copy pixbuf data for callback */ struct _LoadTileData *data = g_new0(struct _LoadTileData, 1); struct _TileData *tdata = g_new0(struct _TileData, 1); data->elev = elev; data->tile = tile; data->pixels = pixels; data->tdata = tdata; tdata->tex = 0; tdata->bil = bil; /* Load the GL texture from the main thread */ g_idle_add_full(G_PRIORITY_LOW, _load_tile_cb, data, NULL); g_debug("GritsPluginElev: _load_tile end %p", g_thread_self()); } static gboolean _free_tile_cb(gpointer _data) { struct _TileData *data = _data; if (LOAD_BIL) g_free(data->bil); if (LOAD_TEX) glDeleteTextures(1, &data->tex); g_free(data); return FALSE; } static void _free_tile(GritsTile *tile, gpointer _elev) { g_debug("GritsPluginElev: _free_tile: %p", tile->data); if (tile->data) g_idle_add_full(G_PRIORITY_LOW, _free_tile_cb, tile->data, NULL); } static void _update_tiles(gpointer _, gpointer _elev) { g_debug("GritsPluginElev: _update_tiles"); GritsPluginElev *elev = _elev; GritsPoint eye; grits_viewer_get_location(elev->viewer, &eye.lat, &eye.lon, &eye.elev); grits_tile_update(elev->tiles, &eye, MAX_RESOLUTION, TILE_WIDTH, TILE_WIDTH, _load_tile, elev); grits_tile_gc(elev->tiles, time(NULL)-10, _free_tile, elev); } /************* * Callbacks * *************/ static void _on_location_changed(GritsViewer *viewer, gdouble lat, gdouble lon, gdouble elevation, GritsPluginElev *elev) { g_thread_pool_push(elev->threads, NULL+1, NULL); } /*********** * Methods * ***********/ /** * grits_plugin_elev_new: * @viewer: the #GritsViewer to use for drawing * * Create a new instance of the elevation plugin. * * Returns: the new #GritsPluginElev */ GritsPluginElev *grits_plugin_elev_new(GritsViewer *viewer) { g_debug("GritsPluginElev: new"); GritsPluginElev *elev = g_object_new(GRITS_TYPE_PLUGIN_ELEV, NULL); elev->viewer = g_object_ref(viewer); /* Load initial tiles */ _update_tiles(NULL, elev); /* Connect signals */ elev->sigid = g_signal_connect(elev->viewer, "location-changed", G_CALLBACK(_on_location_changed), elev); /* Add renderers */ if (LOAD_TEX) grits_viewer_add(viewer, GRITS_OBJECT(elev->tiles), GRITS_LEVEL_WORLD, FALSE); return elev; } /**************** * GObject code * ****************/ /* Plugin init */ static void grits_plugin_elev_plugin_init(GritsPluginInterface *iface); G_DEFINE_TYPE_WITH_CODE(GritsPluginElev, grits_plugin_elev, G_TYPE_OBJECT, G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN, grits_plugin_elev_plugin_init)); static void grits_plugin_elev_plugin_init(GritsPluginInterface *iface) { g_debug("GritsPluginElev: plugin_init"); /* Add methods to the interface */ } /* Class/Object init */ static void grits_plugin_elev_init(GritsPluginElev *elev) { g_debug("GritsPluginElev: init"); /* Set defaults */ elev->threads = g_thread_pool_new(_update_tiles, elev, 1, FALSE, NULL); elev->tiles = grits_tile_new(NULL, NORTH, SOUTH, EAST, WEST); elev->wms = grits_wms_new( "http://www.nasa.network.com/elev", "mergedSrtm", "application/bil", "srtm/", "bil", TILE_WIDTH, TILE_HEIGHT); g_object_ref(elev->tiles); } static void grits_plugin_elev_dispose(GObject *gobject) { g_debug("GritsPluginElev: dispose"); GritsPluginElev *elev = GRITS_PLUGIN_ELEV(gobject); elev->aborted = TRUE; /* Drop references */ if (elev->viewer) { GritsViewer *viewer = elev->viewer; elev->viewer = NULL; g_signal_handler_disconnect(viewer, elev->sigid); if (LOAD_BIL) grits_viewer_clear_height_func(viewer); if (LOAD_TEX) grits_viewer_remove(viewer, GRITS_OBJECT(elev->tiles)); else g_object_unref(elev->tiles); soup_session_abort(elev->wms->http->soup); g_thread_pool_free(elev->threads, TRUE, TRUE); while (gtk_events_pending()) gtk_main_iteration(); g_object_unref(viewer); } G_OBJECT_CLASS(grits_plugin_elev_parent_class)->dispose(gobject); } static void grits_plugin_elev_finalize(GObject *gobject) { g_debug("GritsPluginElev: finalize"); GritsPluginElev *elev = GRITS_PLUGIN_ELEV(gobject); /* Free data */ grits_wms_free(elev->wms); grits_tile_free(elev->tiles, _free_tile, elev); G_OBJECT_CLASS(grits_plugin_elev_parent_class)->finalize(gobject); } static void grits_plugin_elev_class_init(GritsPluginElevClass *klass) { g_debug("GritsPluginElev: class_init"); GObjectClass *gobject_class = (GObjectClass*)klass; gobject_class->dispose = grits_plugin_elev_dispose; gobject_class->finalize = grits_plugin_elev_finalize; }