]> Pileus Git - grits/blobdiff - src/plugins/srtm.c
Fix misc bugs and bump version to 0.3
[grits] / src / plugins / srtm.c
index 3beb2bc806acc0e4cd86ba74183457de30a44aa7..e4460fe9d2e2e3338f6dd8f4ca7386670ce6e430 100644 (file)
 
 #include "srtm.h"
 
-/***********
- * Helpers *
- ***********/
-static gboolean rotate(gpointer _self)
+#define MAX_RESOLUTION 500
+#define TILE_WIDTH     1024
+#define TILE_HEIGHT    512
+
+struct _TileData {
+       /* OpenGL has to be first to make gis_opengl_render_tiles happy */
+       guint      opengl;
+       guint16   *bil;
+};
+
+static gdouble _height_func(gdouble lat, gdouble lon, gpointer _self)
 {
        GisPluginSrtm *self = _self;
-       if (gtk_toggle_button_get_active(self->button)) {
-               self->rotation += 1.0;
-               gis_opengl_redraw(self->opengl);
+       if (!self) return 0;
+
+       GisTile *tile = gis_tile_find(self->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;
+
+       /* 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)];
+
+       gdouble elev =
+               px00 * (1-x_rem) * (1-y_rem) +
+               px10 * (  x_rem) * (1-y_rem) +
+               px01 * (1-x_rem) * (  y_rem) +
+               px11 * (  x_rem) * (  y_rem);
+       return elev;
+}
+
+/**********************
+ * Loader and Freeers *
+ **********************/
+#define LOAD_BIL    TRUE
+#define LOAD_OPENGL FALSE
+struct _LoadTileData {
+       GisPluginSrtm    *self;
+       GisTile          *tile;
+       GdkPixbuf        *pixbuf;
+       struct _TileData *data;
+};
+static guint16 *_load_bil(gchar *path)
+{
+       gchar *data;
+       g_file_get_contents(path, &data, NULL, NULL);
+       g_debug("GisPluginSrtm: load_bil %p", data);
+       return (guint16*)data;
+}
+static GdkPixbuf *_load_pixbuf(guint16 *bil)
+{
+       GdkPixbuf *pixbuf = gdk_pixbuf_new(GDK_COLORSPACE_RGB, FALSE, 8, TILE_WIDTH, TILE_HEIGHT);
+       guchar    *pixels = gdk_pixbuf_get_pixels(pixbuf);
+       gint       stride = gdk_pixbuf_get_rowstride(pixbuf);
+       gint       nchan  = gdk_pixbuf_get_n_channels(pixbuf);
+
+       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)(MAX(value,0))/8848 * 255;
+                       guchar color = (float)value/8848 * 255;
+                       pixels[r*stride + c*nchan + 0] = color;
+                       pixels[r*stride + c*nchan + 1] = color;
+                       pixels[r*stride + c*nchan + 2] = color;
+                       if (nchan == 4)
+                               pixels[r*stride + c*nchan + 3] = 128;
+               }
        }
-       return TRUE;
+       g_debug("GisPluginSrtm: load_pixbuf %p", pixbuf);
+       return pixbuf;
+}
+static guint _load_opengl(GdkPixbuf *pixbuf)
+{
+       /* Load image */
+       guchar *pixels = gdk_pixbuf_get_pixels(pixbuf);
+       gint    alpha  = gdk_pixbuf_get_has_alpha(pixbuf);
+       gint    nchan  = 4; // gdk_pixbuf_get_n_channels(pixbuf);
+       gint    width  = gdk_pixbuf_get_width(pixbuf);
+       gint    height = gdk_pixbuf_get_height(pixbuf);
+
+       /* Create Texture */
+       guint opengl;
+       glGenTextures(1, &opengl);
+       glBindTexture(GL_TEXTURE_2D, opengl);
+
+       glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
+       glPixelStorei(GL_PACK_ALIGNMENT, 1);
+       glTexImage2D(GL_TEXTURE_2D, 0, nchan, width, height, 0,
+                       (alpha ? GL_RGBA : GL_RGB), GL_UNSIGNED_BYTE, 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_BORDER);
+       glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_BORDER);
+
+       g_debug("GisPluginSrtm: load_opengl %d", opengl);
+       return opengl;
+}
+static gboolean _load_tile_cb(gpointer _load)
+{
+       struct _LoadTileData *load = _load;
+       GisPluginSrtm    *self   = load->self;
+       GisTile          *tile   = load->tile;
+       GdkPixbuf        *pixbuf = load->pixbuf;
+       struct _TileData *data   = load->data;
+       g_free(load);
+
+       if (LOAD_OPENGL)
+               data->opengl = _load_opengl(pixbuf);
+
+       /* Do necessasairy processing */
+       /* TODO: Lock this and move to thread, can remove self from _load then */
+       if (LOAD_BIL)
+               gis_opengl_set_height_func(self->opengl, tile, _height_func, self, TRUE);
+
+       /* Cleanup unneeded things */
+       if (!LOAD_BIL)
+               g_free(data->bil);
+       if (LOAD_OPENGL)
+               g_object_unref(pixbuf);
+
+       tile->data = data;
+       return FALSE;
 }
+static void _load_tile(GisTile *tile, gpointer _self)
+{
+       GisPluginSrtm *self = _self;
 
+       g_debug("GisPluginSrtm: _load_tile");
+       gchar *path = gis_wms_make_local(self->wms, tile);
+       struct _LoadTileData *load = g_new0(struct _LoadTileData, 1);
+       load->self = self;
+       load->tile = tile;
+       load->data = g_new0(struct _TileData, 1);
+       if (LOAD_BIL || LOAD_OPENGL)
+               load->data->bil = _load_bil(path);
+       if (LOAD_OPENGL)
+               load->pixbuf = _load_pixbuf(load->data->bil);
+
+       g_idle_add_full(G_PRIORITY_LOW, _load_tile_cb, load, NULL);
+       g_free(path);
+}
+
+static gboolean _free_tile_cb(gpointer _data)
+{
+       struct _TileData *data = _data;
+       if (LOAD_BIL)
+               g_free(data->bil);
+       if (LOAD_OPENGL)
+               glDeleteTextures(1, &data->opengl);
+       g_free(data);
+       return FALSE;
+}
+static void _free_tile(GisTile *tile, gpointer _self)
+{
+       GisPluginSrtm *self = _self;
+       g_debug("GisPluginSrtm: _free_tile: %p=%d", tile->data, *(guint*)tile->data);
+       g_idle_add_full(G_PRIORITY_LOW, _free_tile_cb, tile->data, NULL);
+}
+
+static gpointer _update_tiles(gpointer _self)
+{
+       GisPluginSrtm *self = _self;
+       g_mutex_lock(self->mutex);
+       gdouble lat, lon, elev;
+       gis_view_get_location(self->view, &lat, &lon, &elev);
+       gis_tile_update(self->tiles,
+                       MAX_RESOLUTION, TILE_WIDTH, TILE_WIDTH,
+                       lat, lon, elev,
+                       _load_tile, self);
+       gis_tile_gc(self->tiles, time(NULL)-10,
+                       _free_tile, self);
+       g_mutex_unlock(self->mutex);
+       return NULL;
+}
+
+/*************
+ * Callbacks *
+ *************/
+static void _on_location_changed(GisView *view, gdouble lat, gdouble lon, gdouble elev,
+               GisPluginSrtm *self)
+{
+       g_thread_create(_update_tiles, self, FALSE, NULL);
+}
 
 /***********
  * Methods *
@@ -43,42 +241,27 @@ GisPluginSrtm *gis_plugin_srtm_new(GisWorld *world, GisView *view, GisOpenGL *op
 {
        g_debug("GisPluginSrtm: new");
        GisPluginSrtm *self = g_object_new(GIS_TYPE_PLUGIN_SRTM, NULL);
+       self->view   = view;
        self->opengl = opengl;
 
-       return self;
-}
+       /* Load initial tiles */
+       _load_tile(self->tiles, self);
+       g_thread_create(_update_tiles, self, FALSE, NULL);
 
-static GtkWidget *gis_plugin_srtm_get_config(GisPlugin *_self)
-{
-       GisPluginSrtm *self = GIS_PLUGIN_SRTM(_self);
-       return GTK_WIDGET(self->button);
+       /* Connect signals */
+       self->sigid = g_signal_connect(self->view, "location-changed",
+                       G_CALLBACK(_on_location_changed), self);
+
+       return self;
 }
 
 static void gis_plugin_srtm_expose(GisPlugin *_self)
 {
        GisPluginSrtm *self = GIS_PLUGIN_SRTM(_self);
-       g_debug("GisPluginSrtm: expose");
-
-       glMatrixMode(GL_PROJECTION);
-       glLoadIdentity();
-       glOrtho(1,-1, -1,1, -10,10);
-
-       glMatrixMode(GL_MODELVIEW);
-       glLoadIdentity();
-
-       float light_ambient[]  = {0.1f, 0.1f, 0.0f, 1.0f};
-       float light_diffuse[]  = {0.9f, 0.9f, 0.9f, 1.0f};
-       float light_position[] = {-30.0f, 50.0f, 40.0f, 1.0f};
-       glLightfv(GL_LIGHT0, GL_AMBIENT,  light_ambient);
-       glLightfv(GL_LIGHT0, GL_DIFFUSE,  light_diffuse);
-       glLightfv(GL_LIGHT0, GL_POSITION, light_position);
-       glEnable(GL_COLOR_MATERIAL);
-
-       glTranslatef(0.5, -0.5, -2);
-       glRotatef(self->rotation, 1, 1, 0);
-       glColor4f(0.9, 0.9, 0.7, 1.0);
-       glDisable(GL_CULL_FACE);
-       gdk_gl_draw_teapot(TRUE, 0.25);
+       g_debug("GisPluginSrtm: expose tiles=%p data=%p",
+               self->tiles, self->tiles->data);
+       if (LOAD_OPENGL)
+               gis_opengl_render_tiles(self->opengl, self->tiles);
 }
 
 
@@ -94,25 +277,27 @@ static void gis_plugin_srtm_plugin_init(GisPluginInterface *iface)
 {
        g_debug("GisPluginSrtm: plugin_init");
        /* Add methods to the interface */
-       iface->expose     = gis_plugin_srtm_expose;
-       iface->get_config = gis_plugin_srtm_get_config;
+       iface->expose = gis_plugin_srtm_expose;
 }
 /* Class/Object init */
 static void gis_plugin_srtm_init(GisPluginSrtm *self)
 {
        g_debug("GisPluginSrtm: init");
        /* Set defaults */
-       self->button    = GTK_TOGGLE_BUTTON(gtk_toggle_button_new_with_label("Rotate"));
-       self->rotate_id = g_timeout_add(1000/60, rotate, self);
-       self->rotation  = 30.0;
-       self->opengl    = NULL;
+       self->mutex = g_mutex_new();
+       self->tiles = gis_tile_new(NULL, NORTH, SOUTH, EAST, WEST);
+       self->wms   = gis_wms_new(
+               "http://www.nasa.network.com/elev", "srtm30", "application/bil",
+               "srtm/", "bil", TILE_WIDTH, TILE_HEIGHT);
 }
 static void gis_plugin_srtm_dispose(GObject *gobject)
 {
        g_debug("GisPluginSrtm: dispose");
        GisPluginSrtm *self = GIS_PLUGIN_SRTM(gobject);
-       g_source_remove(self->rotate_id);
        /* Drop references */
+       g_signal_handler_disconnect(self->view, self->sigid);
+       if (LOAD_BIL)
+               gis_opengl_clear_height_func(self->opengl);
        G_OBJECT_CLASS(gis_plugin_srtm_parent_class)->dispose(gobject);
 }
 static void gis_plugin_srtm_finalize(GObject *gobject)
@@ -120,6 +305,9 @@ static void gis_plugin_srtm_finalize(GObject *gobject)
        g_debug("GisPluginSrtm: finalize");
        GisPluginSrtm *self = GIS_PLUGIN_SRTM(gobject);
        /* Free data */
+       gis_tile_free(self->tiles, _free_tile, self);
+       gis_wms_free(self->wms);
+       g_mutex_free(self->mutex);
        G_OBJECT_CLASS(gis_plugin_srtm_parent_class)->finalize(gobject);
 
 }