X-Git-Url: http://pileus.org/git/?p=aweather;a=blobdiff_plain;f=src%2Fplugins%2Fradar.c;h=2dcea83986dc6a0ad45400998a8c0e85d8789b6b;hp=3b566a5a78f3f7f77685ce32d2b4b5b4be71f0c6;hb=cc82e9478bb47cccb085fb2738053b0c5f6b8298;hpb=3da12e4f55084f4faccbc49d71cbd3c1e926707f diff --git a/src/plugins/radar.c b/src/plugins/radar.c index 3b566a5..2dcea83 100644 --- a/src/plugins/radar.c +++ b/src/plugins/radar.c @@ -1,5 +1,5 @@ /* - * Copyright (C) 2009-2010 Andy Spencer + * Copyright (C) 2009-2012 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 @@ -20,19 +20,19 @@ #include #include #include -#include #include -#include #include #include -#include +#include #include "radar.h" #include "level2.h" #include "../aweather-location.h" -void _gtk_bin_set_child(GtkBin *bin, GtkWidget *new) +#include "compat.h" + +static void aweather_bin_set_child(GtkBin *bin, GtkWidget *new) { GtkWidget *old = gtk_bin_get_child(bin); if (old) @@ -42,17 +42,20 @@ void _gtk_bin_set_child(GtkBin *bin, GtkWidget *new) } static gchar *_find_nearest(time_t time, GList *files, - gsize offset, gchar *format) + gsize offset) { - g_debug("GisPluginRadar: _find_nearest"); + g_debug("RadarSite: find_nearest ..."); time_t nearest_time = 0; char *nearest_file = NULL; struct tm tm = {}; for (GList *cur = files; cur; cur = cur->next) { gchar *file = cur->data; - g_message("file=%s", file); - strptime(file+offset, format, &tm); + sscanf(file+offset, "%4d%2d%2d_%2d%2d", + &tm.tm_year, &tm.tm_mon, &tm.tm_mday, + &tm.tm_hour, &tm.tm_min); + tm.tm_year -= 1900; + tm.tm_mon -= 1; if (ABS(time - mktime(&tm)) < ABS(time - nearest_time)) { nearest_file = file; @@ -60,7 +63,7 @@ static gchar *_find_nearest(time_t time, GList *files, } } - g_debug("GisPluginRadar: _find_nearest = %s", nearest_file); + g_debug("RadarSite: find_nearest = %s", nearest_file); if (nearest_file) return g_strdup(nearest_file); else @@ -78,29 +81,28 @@ typedef enum { } RadarSiteStatus; struct _RadarSite { /* Information */ - gchar *code; // Site name. e.g. KLSX - gchar *name; // Site name. e.g. St. Louis - GisPoint pos; // LLE positions of antena - GisMarker *marker; // Map marker for libgis + city_t *city; + GritsMarker *marker; // Map marker for grits /* Stuff from the parents */ - GisViewer *viewer; - GisHttp *http; - GisPrefs *prefs; - GtkWidget *pconfig; + GritsViewer *viewer; + GritsHttp *http; + GritsPrefs *prefs; + GtkWidget *pconfig; /* When loaded */ - RadarSiteStatus status; // Loading status for the site + gboolean hidden; + RadarSiteStatus status; // Loading status for the site GtkWidget *config; - AWeatherLevel2 *level2; // The Level2 structure for the current volume - gpointer level2_ref; // GisViewer reference to the added radar + AWeatherLevel2 *level2; // The Level2 structure for the current volume /* Internal data */ - time_t time; // Current timestamp of the level2 - gchar *message; // Error message set while updating - guint time_id; // "time-changed" callback ID - guint refresh_id; // "refresh" callback ID - guint location_id; // "locaiton-changed" callback ID + time_t time; // Current timestamp of the level2 + gchar *message; // Error message set while updating + guint time_id; // "time-changed" callback ID + guint refresh_id; // "refresh" callback ID + guint location_id; // "locaiton-changed" callback ID + guint idle_source; // _site_update_end idle source }; /* format: http://mesonet.agron.iastate.edu/data/nexrd2/raw/KABR/KABR_20090510_0323 */ @@ -120,42 +122,44 @@ gboolean _site_update_end(gpointer _site) { RadarSite *site = _site; if (site->message) { - g_warning("GisPluginRadar: _update_end - %s", site->message); - _gtk_bin_set_child(GTK_BIN(site->config), gtk_label_new(site->message)); + g_warning("RadarSite: update_end - %s", site->message); + const char *fmt = "http://forecast.weather.gov/product.php?site=NWS&product=FTM&format=TXT&issuedby=%s"; + char *uri = g_strdup_printf(fmt, site->city->code+1); + GtkWidget *box = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0); + GtkWidget *msg = gtk_label_new(site->message); + GtkWidget *btn = gtk_link_button_new_with_label(uri, "View Radar Status"); + gtk_box_set_homogeneous(GTK_BOX(box), TRUE); + gtk_box_pack_start(GTK_BOX(box), msg, TRUE, TRUE, 0); + gtk_box_pack_start(GTK_BOX(box), btn, TRUE, TRUE, 0); + aweather_bin_set_child(GTK_BIN(site->config), box); + g_free(uri); } else { - _gtk_bin_set_child(GTK_BIN(site->config), + aweather_bin_set_child(GTK_BIN(site->config), aweather_level2_get_config(site->level2)); } site->status = STATUS_LOADED; + site->idle_source = 0; return FALSE; } gpointer _site_update_thread(gpointer _site) { RadarSite *site = _site; - g_debug("GisPluginRadar: _update - %s", site->code); - site->status = STATUS_LOADING; + g_debug("RadarSite: update_thread - %s", site->city->code); site->message = NULL; - gboolean offline = gis_viewer_get_offline(site->viewer); - gchar *nexrad_url = gis_prefs_get_string(site->prefs, + gboolean offline = grits_viewer_get_offline(site->viewer); + gchar *nexrad_url = grits_prefs_get_string(site->prefs, "aweather/nexrad_url", NULL); - /* Remove old volume */ - g_debug("GisPluginRadar: _update - remove - %s", site->code); - if (site->level2_ref) { - gis_viewer_remove(site->viewer, site->level2_ref); - site->level2_ref = NULL; - } - /* Find nearest volume (temporally) */ - g_debug("GisPluginRadar: _update - find nearest - %s", site->code); - gchar *dir_list = g_strconcat(nexrad_url, "/", site->code, + g_debug("RadarSite: update_thread - find nearest - %s", site->city->code); + gchar *dir_list = g_strconcat(nexrad_url, "/", site->city->code, "/", "dir.list", NULL); - GList *files = gis_http_available(site->http, - "^K\\w{3}_\\d{8}_\\d{4}$", site->code, + GList *files = grits_http_available(site->http, + "^\\w{4}_\\d{8}_\\d{4}$", site->city->code, "\\d+ (.*)", (offline ? NULL : dir_list)); g_free(dir_list); - gchar *nearest = _find_nearest(site->time, files, 5, "%Y%m%d_%H%M"); + gchar *nearest = _find_nearest(site->time, files, 5); g_list_foreach(files, (GFunc)g_free, NULL); g_list_free(files); if (!nearest) { @@ -164,12 +168,14 @@ gpointer _site_update_thread(gpointer _site) } /* Fetch new volume */ - g_debug("GisPluginRadar: _update - fetch"); - gchar *local = g_strconcat(site->code, "/", nearest, NULL); + g_debug("RadarSite: update_thread - fetch"); + gchar *local = g_strconcat(site->city->code, "/", nearest, NULL); gchar *uri = g_strconcat(nexrad_url, "/", local, NULL); - gchar *file = gis_http_fetch(site->http, uri, local, - offline ? GIS_LOCAL : GIS_UPDATE, + gchar *file = grits_http_fetch(site->http, uri, local, + offline ? GRITS_LOCAL : GRITS_UPDATE, _site_update_loading, site); + g_free(nexrad_url); + g_free(nearest); g_free(local); g_free(uri); if (!file) { @@ -178,54 +184,74 @@ gpointer _site_update_thread(gpointer _site) } /* Load and add new volume */ - g_debug("GisPluginRadar: _update - load - %s", site->code); + g_debug("RadarSite: update_thread - load - %s", site->city->code); site->level2 = aweather_level2_new_from_file( - site->viewer, colormaps, file, site->code); + file, site->city->code, colormaps); + g_free(file); if (!site->level2) { site->message = "Load failed"; goto out; } - site->level2_ref = gis_viewer_add(site->viewer, - GIS_OBJECT(site->level2), GIS_LEVEL_WORLD, TRUE); + grits_object_hide(GRITS_OBJECT(site->level2), site->hidden); + grits_viewer_add(site->viewer, GRITS_OBJECT(site->level2), + GRITS_LEVEL_WORLD+3, TRUE); out: - g_idle_add(_site_update_end, site); + if (!site->idle_source) + site->idle_source = g_idle_add(_site_update_end, site); return NULL; } void _site_update(RadarSite *site) { - site->time = gis_viewer_get_time(site->viewer); - g_debug("GisPluginRadar: _on_time_changed %s - %d", - site->code, (gint)site->time); + if (site->status == STATUS_LOADING) + return; + site->status = STATUS_LOADING; + + site->time = grits_viewer_get_time(site->viewer); + g_debug("RadarSite: update %s - %d", + site->city->code, (gint)site->time); /* Add a progress bar */ GtkWidget *progress = gtk_progress_bar_new(); gtk_progress_bar_set_text(GTK_PROGRESS_BAR(progress), "Loading..."); - _gtk_bin_set_child(GTK_BIN(site->config), progress); + aweather_bin_set_child(GTK_BIN(site->config), progress); + + /* Remove old volume */ + g_debug("RadarSite: update - remove - %s", site->city->code); + if (site->level2) { + grits_viewer_remove(site->viewer, GRITS_OBJECT(site->level2)); + site->level2 = NULL; + } /* Fork loading right away so updating the * list of times doesn't take too long */ - g_thread_create(_site_update_thread, site, FALSE, NULL); + g_thread_new("site-update-thread", _site_update_thread, site); } /* RadarSite methods */ void radar_site_unload(RadarSite *site) { - g_debug("GisPluginRadar: radar_site_unload %s", site->code); - - if (site->status == STATUS_LOADING) + if (site->status != STATUS_LOADED) return; // Abort if it's still loading - g_signal_handler_disconnect(site->viewer, site->time_id); - g_signal_handler_disconnect(site->viewer, site->refresh_id); + g_debug("RadarSite: unload %s", site->city->code); + + if (site->time_id) + g_signal_handler_disconnect(site->viewer, site->time_id); + if (site->refresh_id) + g_signal_handler_disconnect(site->viewer, site->refresh_id); + if (site->idle_source) + g_source_remove(site->idle_source); + site->idle_source = 0; /* Remove tab */ - gtk_widget_destroy(site->config); + if (site->config) + gtk_widget_destroy(site->config); /* Remove radar */ - if (site->level2_ref) { - gis_viewer_remove(site->viewer, site->level2_ref); - site->level2_ref = NULL; + if (site->level2) { + grits_viewer_remove(site->viewer, GRITS_OBJECT(site->level2)); + site->level2 = NULL; } site->status = STATUS_UNLOADED; @@ -233,26 +259,16 @@ void radar_site_unload(RadarSite *site) void radar_site_load(RadarSite *site) { - g_debug("GisPluginRadar: radar_site_load %s", site->code); - site->status = STATUS_LOADING; + g_debug("RadarSite: load %s", site->city->code); /* Add tab page */ site->config = gtk_alignment_new(0, 0, 1, 1); - GtkWidget *tab = gtk_hbox_new(FALSE, 0); - GtkWidget *close = gtk_button_new(); - GtkWidget *label = gtk_label_new(site->name); - gtk_container_add(GTK_CONTAINER(close), - gtk_image_new_from_stock(GTK_STOCK_CLOSE, - GTK_ICON_SIZE_MENU)); - gtk_button_set_relief(GTK_BUTTON(close), GTK_RELIEF_NONE); - g_signal_connect_swapped(close, "clicked", - G_CALLBACK(radar_site_unload), site); - gtk_box_pack_start(GTK_BOX(tab), label, TRUE, TRUE, 0); - gtk_box_pack_end(GTK_BOX(tab), close, FALSE, FALSE, 0); - gtk_notebook_append_page(GTK_NOTEBOOK(site->pconfig), - site->config, tab); + g_object_set_data(G_OBJECT(site->config), "site", site); + gtk_notebook_append_page(GTK_NOTEBOOK(site->pconfig), site->config, + gtk_label_new(site->city->name)); gtk_widget_show_all(site->config); - gtk_widget_show_all(tab); + if (gtk_notebook_get_current_page(GTK_NOTEBOOK(site->pconfig)) == 0) + gtk_notebook_set_current_page(GTK_NOTEBOOK(site->pconfig), -1); /* Set up radar loading */ site->time_id = g_signal_connect_swapped(site->viewer, "time-changed", @@ -262,17 +278,17 @@ void radar_site_load(RadarSite *site) _site_update(site); } -void _site_on_location_changed(GisViewer *viewer, +void _site_on_location_changed(GritsViewer *viewer, gdouble lat, gdouble lon, gdouble elev, gpointer _site) { - static gdouble min_dist = EARTH_R / 20; + static gdouble min_dist = EARTH_R / 30; RadarSite *site = _site; /* Calculate distance, could cache xyz values */ gdouble eye_xyz[3], site_xyz[3]; lle2xyz(lat, lon, elev, &eye_xyz[0], &eye_xyz[1], &eye_xyz[2]); - lle2xyz(site->pos.lat, site->pos.lon, site->pos.elev, + lle2xyz(site->city->pos.lat, site->city->pos.lon, site->city->pos.elev, &site_xyz[0], &site_xyz[1], &site_xyz[2]); gdouble dist = distd(site_xyz, eye_xyz); @@ -283,28 +299,55 @@ void _site_on_location_changed(GisViewer *viewer, radar_site_unload(site); } +static gboolean on_marker_clicked(GritsObject *marker, GdkEvent *event, RadarSite *site) +{ + GritsViewer *viewer = site->viewer; + GritsPoint center = marker->center; + grits_viewer_set_location(viewer, center.lat, center.lon, EARTH_R/35); + grits_viewer_set_rotation(viewer, 0, 0, 0); + /* Recursivly set notebook tabs */ + GtkWidget *widget, *parent; + for (widget = site->config; widget; widget = parent) { + parent = gtk_widget_get_parent(widget); + if (GTK_IS_NOTEBOOK(parent)) { + gint i = gtk_notebook_page_num(GTK_NOTEBOOK(parent), widget); + gtk_notebook_set_current_page(GTK_NOTEBOOK(parent), i); + } + } + return TRUE; +} + RadarSite *radar_site_new(city_t *city, GtkWidget *pconfig, - GisViewer *viewer, GisPrefs *prefs, GisHttp *http) + GritsViewer *viewer, GritsPrefs *prefs, GritsHttp *http) { RadarSite *site = g_new0(RadarSite, 1); site->viewer = g_object_ref(viewer); site->prefs = g_object_ref(prefs); - site->http = http; - site->code = g_strdup(city->code); - site->name = g_strdup(city->name); - site->pos = city->pos; + //site->http = http; + site->http = grits_http_new(G_DIR_SEPARATOR_S + "nexrad" G_DIR_SEPARATOR_S + "level2" G_DIR_SEPARATOR_S); + site->city = city; site->pconfig = pconfig; + site->hidden = TRUE; + + /* Set initial location */ + gdouble lat, lon, elev; + grits_viewer_get_location(viewer, &lat, &lon, &elev); + _site_on_location_changed(viewer, lat, lon, elev, site); /* Add marker */ - site->marker = gis_marker_new(city->name); - GIS_OBJECT(site->marker)->center = site->pos; - GIS_OBJECT(site->marker)->lod = EARTH_R*city->lod; - gis_viewer_add(site->viewer, GIS_OBJECT(site->marker), - GIS_LEVEL_OVERLAY, FALSE); + site->marker = grits_marker_new(site->city->name); + GRITS_OBJECT(site->marker)->center = site->city->pos; + GRITS_OBJECT(site->marker)->lod = EARTH_R*0.75*site->city->lod; + grits_viewer_add(site->viewer, GRITS_OBJECT(site->marker), + GRITS_LEVEL_HUD, FALSE); + g_signal_connect(site->marker, "clicked", + G_CALLBACK(on_marker_clicked), site); + grits_object_set_cursor(GRITS_OBJECT(site->marker), GDK_HAND2); /* Connect signals */ - site->location_id = - g_signal_connect(viewer, "location-changed", + site->location_id = g_signal_connect(viewer, "location-changed", G_CALLBACK(_site_on_location_changed), site); return site; } @@ -312,25 +355,295 @@ RadarSite *radar_site_new(city_t *city, GtkWidget *pconfig, void radar_site_free(RadarSite *site) { radar_site_unload(site); - /* Stuff? */ + grits_viewer_remove(site->viewer, GRITS_OBJECT(site->marker)); + if (site->location_id) + g_signal_handler_disconnect(site->viewer, site->location_id); + grits_http_free(site->http); g_object_unref(site->viewer); - g_free(site->code); + g_object_unref(site->prefs); g_free(site); } -/****************** - * GisPluginRadar * - ******************/ -static void _draw_hud(GisCallback *callback, gpointer _self) +/************** + * RadarConus * + **************/ +#define CONUS_NORTH 50.406626367301044 +#define CONUS_WEST -127.620375523875420 +#define CONUS_WIDTH 3400.0 +#define CONUS_HEIGHT 1600.0 +#define CONUS_DEG_PER_PX 0.017971305190311 + +struct _RadarConus { + GritsViewer *viewer; + GritsHttp *http; + GtkWidget *config; + time_t time; + const gchar *message; + GMutex loading; + + gchar *path; + GritsTile *tile[2]; + + guint time_id; // "time-changed" callback ID + guint refresh_id; // "refresh" callback ID + guint idle_source; // _conus_update_end idle source +}; + +void _conus_update_loading(gchar *file, goffset cur, + goffset total, gpointer _conus) +{ + RadarConus *conus = _conus; + GtkWidget *progress_bar = gtk_bin_get_child(GTK_BIN(conus->config)); + double percent = (double)cur/total; + gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(progress_bar), MIN(percent, 1.0)); + gchar *msg = g_strdup_printf("Loading... %5.1f%% (%.2f/%.2f MB)", + percent*100, (double)cur/1000000, (double)total/1000000); + gtk_progress_bar_set_text(GTK_PROGRESS_BAR(progress_bar), msg); + g_free(msg); +} + +/* Copy images to graphics memory */ +static void _conus_update_end_copy(GritsTile *tile, guchar *pixels) +{ + if (!tile->tex) + glGenTextures(1, &tile->tex); + + gchar *clear = g_malloc0(2048*2048*4); + glBindTexture(GL_TEXTURE_2D, tile->tex); + + glPixelStorei(GL_UNPACK_ALIGNMENT, 1); + glPixelStorei(GL_PACK_ALIGNMENT, 1); + glTexImage2D(GL_TEXTURE_2D, 0, 4, 2048, 2048, 0, + GL_RGBA, GL_UNSIGNED_BYTE, clear); + glTexSubImage2D(GL_TEXTURE_2D, 0, 1,1, CONUS_WIDTH/2,CONUS_HEIGHT, + GL_RGBA, GL_UNSIGNED_BYTE, pixels); + tile->coords.n = 1.0/(CONUS_WIDTH/2); + tile->coords.w = 1.0/ CONUS_HEIGHT; + tile->coords.s = tile->coords.n + CONUS_HEIGHT / 2048.0; + tile->coords.e = tile->coords.w + (CONUS_WIDTH/2) / 2048.0; + glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR); + glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR); + 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(); + g_free(clear); +} + +/* Split the pixbuf into east and west halves (with 2K sides) + * Also map the pixbuf's alpha values */ +static void _conus_update_end_split(guchar *pixels, guchar *west, guchar *east, + gint width, gint height, gint pxsize) +{ + g_debug("Conus: update_end_split"); + guchar *out[] = {west,east}; + const guchar alphamap[][4] = { + {0x04, 0xe9, 0xe7, 0x30}, + {0x01, 0x9f, 0xf4, 0x60}, + {0x03, 0x00, 0xf4, 0x90}, + }; + for (int y = 0; y < height; y++) + for (int x = 0; x < width; x++) { + gint subx = x % (width/2); + gint idx = x / (width/2); + guchar *src = &pixels[(y*width+x)*pxsize]; + guchar *dst = &out[idx][(y*(width/2)+subx)*4]; + if (src[0] > 0xe0 && + src[1] > 0xe0 && + src[2] > 0xe0) { + dst[3] = 0x00; + } else { + dst[0] = src[0]; + dst[1] = src[1]; + dst[2] = src[2]; + dst[3] = 0xff * 0.75; + for (int j = 0; j < G_N_ELEMENTS(alphamap); j++) + if (src[0] == alphamap[j][0] && + src[1] == alphamap[j][1] && + src[2] == alphamap[j][2]) + dst[3] = alphamap[j][3]; + } + } +} + +gboolean _conus_update_end(gpointer _conus) +{ + RadarConus *conus = _conus; + g_debug("Conus: update_end"); + + /* Check error status */ + if (conus->message) { + g_warning("Conus: update_end - %s", conus->message); + aweather_bin_set_child(GTK_BIN(conus->config), gtk_label_new(conus->message)); + goto out; + } + + /* Load and pixbuf */ + GError *error = NULL; + GdkPixbuf *pixbuf = gdk_pixbuf_new_from_file(conus->path, &error); + if (!pixbuf || error) { + g_warning("Conus: update_end - error loading pixbuf: %s", conus->path); + aweather_bin_set_child(GTK_BIN(conus->config), gtk_label_new("Error loading pixbuf")); + g_remove(conus->path); + goto out; + } + + /* Split pixels into east/west parts */ + guchar *pixels = gdk_pixbuf_get_pixels(pixbuf); + gint width = gdk_pixbuf_get_width(pixbuf); + gint height = gdk_pixbuf_get_height(pixbuf); + gint pxsize = gdk_pixbuf_get_has_alpha(pixbuf) ? 4 : 3; + guchar *pixels_west = g_malloc(4*(width/2)*height); + guchar *pixels_east = g_malloc(4*(width/2)*height); + _conus_update_end_split(pixels, pixels_west, pixels_east, + width, height, pxsize); + g_object_unref(pixbuf); + + /* Copy pixels to graphics memory */ + _conus_update_end_copy(conus->tile[0], pixels_west); + _conus_update_end_copy(conus->tile[1], pixels_east); + g_free(pixels_west); + g_free(pixels_east); + + /* Update GUI */ + gchar *label = g_path_get_basename(conus->path); + aweather_bin_set_child(GTK_BIN(conus->config), gtk_label_new(label)); + grits_viewer_queue_draw(conus->viewer); + g_free(label); + +out: + conus->idle_source = 0; + g_free(conus->path); + g_mutex_unlock(&conus->loading); + return FALSE; +} + +gpointer _conus_update_thread(gpointer _conus) +{ + RadarConus *conus = _conus; + conus->message = NULL; + + /* Find nearest */ + g_debug("Conus: update_thread - nearest"); + gboolean offline = grits_viewer_get_offline(conus->viewer); + gchar *conus_url = "http://radar.weather.gov/Conus/RadarImg/"; + gchar *nearest; + if (time(NULL) - conus->time < 60*60*5 && !offline) { + /* radar.weather.gov is full of lies. + * the index pages get cached and out of date */ + /* gmtime is not thread safe, but it's not used very often so + * hopefully it'll be alright for now... :-( */ + struct tm *tm = gmtime(&conus->time); + time_t onthe8 = conus->time - 60*((tm->tm_min+1)%10+1); + tm = gmtime(&onthe8); + nearest = g_strdup_printf("Conus_%04d%02d%02d_%02d%02d_N0Ronly.gif", + tm->tm_year+1900, tm->tm_mon+1, tm->tm_mday, + tm->tm_hour, tm->tm_min); + } else { + GList *files = grits_http_available(conus->http, + "^Conus_[^\"]*_N0Ronly.gif$", "", NULL, NULL); + nearest = _find_nearest(conus->time, files, 6); + g_list_foreach(files, (GFunc)g_free, NULL); + g_list_free(files); + if (!nearest) { + conus->message = "No suitable files"; + goto out; + } + } + + /* Fetch the image */ + g_debug("Conus: update_thread - fetch"); + gchar *uri = g_strconcat(conus_url, nearest, NULL); + conus->path = grits_http_fetch(conus->http, uri, nearest, + offline ? GRITS_LOCAL : GRITS_ONCE, + _conus_update_loading, conus); + g_free(nearest); + g_free(uri); + if (!conus->path) { + conus->message = "Fetch failed"; + goto out; + } + +out: + g_debug("Conus: update_thread - done"); + if (!conus->idle_source) + conus->idle_source = g_idle_add(_conus_update_end, conus); + return NULL; +} + +void _conus_update(RadarConus *conus) { - /* TODO */ - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - if (!self->colormap) + if (!g_mutex_trylock(&conus->loading)) return; + conus->time = grits_viewer_get_time(conus->viewer); + g_debug("Conus: update - %d", + (gint)conus->time); + + /* Add a progress bar */ + GtkWidget *progress = gtk_progress_bar_new(); + gtk_progress_bar_set_text(GTK_PROGRESS_BAR(progress), "Loading..."); + aweather_bin_set_child(GTK_BIN(conus->config), progress); + + g_thread_new("conus-update-thread", _conus_update_thread, conus); +} + +RadarConus *radar_conus_new(GtkWidget *pconfig, + GritsViewer *viewer, GritsHttp *http) +{ + RadarConus *conus = g_new0(RadarConus, 1); + conus->viewer = g_object_ref(viewer); + conus->http = http; + conus->config = gtk_alignment_new(0, 0, 1, 1); + g_mutex_init(&conus->loading); + + gdouble south = CONUS_NORTH - CONUS_DEG_PER_PX*CONUS_HEIGHT; + gdouble east = CONUS_WEST + CONUS_DEG_PER_PX*CONUS_WIDTH; + gdouble mid = CONUS_WEST + CONUS_DEG_PER_PX*CONUS_WIDTH/2; + conus->tile[0] = grits_tile_new(NULL, CONUS_NORTH, south, mid, CONUS_WEST); + conus->tile[1] = grits_tile_new(NULL, CONUS_NORTH, south, east, mid); + conus->tile[0]->zindex = 2; + conus->tile[1]->zindex = 1; + grits_viewer_add(viewer, GRITS_OBJECT(conus->tile[0]), GRITS_LEVEL_WORLD+2, FALSE); + grits_viewer_add(viewer, GRITS_OBJECT(conus->tile[1]), GRITS_LEVEL_WORLD+2, FALSE); + + conus->time_id = g_signal_connect_swapped(viewer, "time-changed", + G_CALLBACK(_conus_update), conus); + conus->refresh_id = g_signal_connect_swapped(viewer, "refresh", + G_CALLBACK(_conus_update), conus); + + g_object_set_data(G_OBJECT(conus->config), "conus", conus); + gtk_notebook_append_page(GTK_NOTEBOOK(pconfig), conus->config, + gtk_label_new("Conus")); + + _conus_update(conus); + return conus; +} - g_debug("GisPluginRadar: _draw_hud"); - /* Print the color table */ +void radar_conus_free(RadarConus *conus) +{ + g_signal_handler_disconnect(conus->viewer, conus->time_id); + g_signal_handler_disconnect(conus->viewer, conus->refresh_id); + if (conus->idle_source) + g_source_remove(conus->idle_source); + + for (int i = 0; i < 2; i++) { + GritsTile *tile = conus->tile[i]; + grits_viewer_remove(conus->viewer, GRITS_OBJECT(tile)); + g_object_unref(tile); + } + + g_object_unref(conus->viewer); + g_free(conus); +} + + +/******************** + * GritsPluginRadar * + ********************/ +static void _draw_hud(GritsCallback *callback, GritsOpenGL *opengl, gpointer _self) +{ + g_debug("GritsPluginRadar: _draw_hud"); + /* Setup OpenGL */ glMatrixMode(GL_MODELVIEW ); glLoadIdentity(); glMatrixMode(GL_PROJECTION); glLoadIdentity(); glDisable(GL_TEXTURE_2D); @@ -338,30 +651,104 @@ static void _draw_hud(GisCallback *callback, gpointer _self) glDisable(GL_CULL_FACE); glDisable(GL_LIGHTING); glEnable(GL_COLOR_MATERIAL); - glBegin(GL_QUADS); - int i; - for (i = 0; i < 256; i++) { - glColor4ubv(self->colormap->data[i]); - glVertex3f(-1.0, (float)((i ) - 256/2)/(256/2), 0.0); // bot left - glVertex3f(-1.0, (float)((i+1) - 256/2)/(256/2), 0.0); // top left - glVertex3f(-0.9, (float)((i+1) - 256/2)/(256/2), 0.0); // top right - glVertex3f(-0.9, (float)((i ) - 256/2)/(256/2), 0.0); // bot right + + GHashTableIter iter; + gpointer name, _site; + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(_self); + g_hash_table_iter_init(&iter, self->sites); + while (g_hash_table_iter_next(&iter, &name, &_site)) { + /* Pick correct colormaps */ + RadarSite *site = _site; + if (site->hidden || !site->level2) + continue; + AWeatherColormap *colormap = site->level2->sweep_colors; + + /* Print the color table */ + glBegin(GL_QUADS); + int len = colormap->len; + for (int i = 0; i < len; i++) { + glColor4ubv(colormap->data[i]); + glVertex3f(-1.0, (float)((i ) - len/2)/(len/2), 0.0); // bot left + glVertex3f(-1.0, (float)((i+1) - len/2)/(len/2), 0.0); // top left + glVertex3f(-0.9, (float)((i+1) - len/2)/(len/2), 0.0); // top right + glVertex3f(-0.9, (float)((i ) - len/2)/(len/2), 0.0); // bot right + } + glEnd(); + } +} + +static void _load_colormap(gchar *filename, AWeatherColormap *cm) +{ + g_debug("GritsPluginRadar: _load_colormap - %s", filename); + FILE *file = fopen(filename, "r"); + if (!file) + g_error("GritsPluginRadar: open failed"); + guint8 color[4]; + GArray *array = g_array_sized_new(FALSE, TRUE, sizeof(color), 256); + if (!fgets(cm->name, sizeof(cm->name), file)) goto out; + if (!fscanf(file, "%f\n", &cm->scale)) goto out; + if (!fscanf(file, "%f\n", &cm->shift)) goto out; + int r, g, b, a; + while (fscanf(file, "%d %d %d %d\n", &r, &g, &b, &a) == 4) { + color[0] = r; + color[1] = g; + color[2] = b; + color[3] = a; + g_array_append_val(array, color); + } + cm->len = (gint )array->len; + cm->data = (void*)array->data; +out: + g_array_free(array, FALSE); + fclose(file); +} + +static void _update_hidden(GtkNotebook *notebook, + gpointer _, guint page_num, gpointer viewer) +{ + g_debug("GritsPluginRadar: _update_hidden - 0..%d = %d", + gtk_notebook_get_n_pages(notebook), page_num); + + for (gint i = 0; i < gtk_notebook_get_n_pages(notebook); i++) { + gboolean is_hidden = (i != page_num); + GtkWidget *config = gtk_notebook_get_nth_page(notebook, i); + RadarConus *conus = g_object_get_data(G_OBJECT(config), "conus"); + RadarSite *site = g_object_get_data(G_OBJECT(config), "site"); + + /* Conus */ + if (conus) { + grits_object_hide(GRITS_OBJECT(conus->tile[0]), is_hidden); + grits_object_hide(GRITS_OBJECT(conus->tile[1]), is_hidden); + } else if (site) { + site->hidden = is_hidden; + if (site->level2) + grits_object_hide(GRITS_OBJECT(site->level2), is_hidden); + } else { + g_warning("GritsPluginRadar: _update_hidden - no site or counus found"); + } } - glEnd(); + grits_viewer_queue_draw(viewer); } /* Methods */ -GisPluginRadar *gis_plugin_radar_new(GisViewer *viewer, GisPrefs *prefs) +GritsPluginRadar *grits_plugin_radar_new(GritsViewer *viewer, GritsPrefs *prefs) { /* TODO: move to constructor if possible */ - g_debug("GisPluginRadar: new"); - GisPluginRadar *self = g_object_new(GIS_TYPE_PLUGIN_RADAR, NULL); - self->viewer = viewer; - self->prefs = prefs; + g_debug("GritsPluginRadar: new"); + GritsPluginRadar *self = g_object_new(GRITS_TYPE_PLUGIN_RADAR, NULL); + self->viewer = g_object_ref(viewer); + self->prefs = g_object_ref(prefs); + + /* Setup page switching */ + self->tab_id = g_signal_connect(self->config, "switch-page", + G_CALLBACK(_update_hidden), viewer); /* Load HUD */ - GisCallback *hud_cb = gis_callback_new(_draw_hud, self); - gis_viewer_add(viewer, GIS_OBJECT(hud_cb), GIS_LEVEL_HUD, FALSE); + self->hud = grits_callback_new(_draw_hud, self); + grits_viewer_add(viewer, GRITS_OBJECT(self->hud), GRITS_LEVEL_HUD, FALSE); + + /* Load Conus */ + self->conus = radar_conus_new(self->config, self->viewer, self->conus_http); /* Load radar sites */ for (city_t *city = cities; city->type; city++) { @@ -375,55 +762,82 @@ GisPluginRadar *gis_plugin_radar_new(GisViewer *viewer, GisPrefs *prefs) return self; } -static GtkWidget *gis_plugin_radar_get_config(GisPlugin *_self) +static GtkWidget *grits_plugin_radar_get_config(GritsPlugin *_self) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(_self); return self->config; } /* GObject code */ -static void gis_plugin_radar_plugin_init(GisPluginInterface *iface); -G_DEFINE_TYPE_WITH_CODE(GisPluginRadar, gis_plugin_radar, G_TYPE_OBJECT, - G_IMPLEMENT_INTERFACE(GIS_TYPE_PLUGIN, - gis_plugin_radar_plugin_init)); -static void gis_plugin_radar_plugin_init(GisPluginInterface *iface) +static void grits_plugin_radar_plugin_init(GritsPluginInterface *iface); +G_DEFINE_TYPE_WITH_CODE(GritsPluginRadar, grits_plugin_radar, G_TYPE_OBJECT, + G_IMPLEMENT_INTERFACE(GRITS_TYPE_PLUGIN, + grits_plugin_radar_plugin_init)); +static void grits_plugin_radar_plugin_init(GritsPluginInterface *iface) { - g_debug("GisPluginRadar: plugin_init"); + g_debug("GritsPluginRadar: plugin_init"); /* Add methods to the interface */ - iface->get_config = gis_plugin_radar_get_config; + iface->get_config = grits_plugin_radar_get_config; } -static void gis_plugin_radar_init(GisPluginRadar *self) +static void grits_plugin_radar_init(GritsPluginRadar *self) { - g_debug("GisPluginRadar: class_init"); + g_debug("GritsPluginRadar: class_init"); /* Set defaults */ - self->sites_http = gis_http_new(G_DIR_SEPARATOR_S "nexrad" G_DIR_SEPARATOR_S "level2" G_DIR_SEPARATOR_S); - self->sites = g_hash_table_new(g_str_hash, g_str_equal); - self->config = gtk_notebook_new(); + self->sites_http = grits_http_new(G_DIR_SEPARATOR_S + "nexrad" G_DIR_SEPARATOR_S + "level2" G_DIR_SEPARATOR_S); + self->conus_http = grits_http_new(G_DIR_SEPARATOR_S + "nexrad" G_DIR_SEPARATOR_S + "conus" G_DIR_SEPARATOR_S); + self->sites = g_hash_table_new_full(g_str_hash, g_str_equal, + NULL, (GDestroyNotify)radar_site_free); + self->config = g_object_ref(gtk_notebook_new()); + + /* Load colormaps */ + for (int i = 0; colormaps[i].file; i++) { + gchar *file = g_build_filename(PKGDATADIR, + "colors", colormaps[i].file, NULL); + _load_colormap(file, &colormaps[i]); + g_free(file); + } + + /* Need to position on the top because of Win32 bug */ gtk_notebook_set_tab_pos(GTK_NOTEBOOK(self->config), GTK_POS_LEFT); } -static void gis_plugin_radar_dispose(GObject *gobject) +static void grits_plugin_radar_dispose(GObject *gobject) { - g_debug("GisPluginRadar: dispose"); - GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); + g_debug("GritsPluginRadar: dispose"); + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(gobject); + if (self->viewer) { + GritsViewer *viewer = self->viewer; + self->viewer = NULL; + g_signal_handler_disconnect(self->config, self->tab_id); + grits_viewer_remove(viewer, GRITS_OBJECT(self->hud)); + radar_conus_free(self->conus); + g_hash_table_destroy(self->sites); + g_object_unref(self->config); + g_object_unref(self->hud); + g_object_unref(self->prefs); + g_object_unref(viewer); + } /* Drop references */ - G_OBJECT_CLASS(gis_plugin_radar_parent_class)->dispose(gobject); + G_OBJECT_CLASS(grits_plugin_radar_parent_class)->dispose(gobject); } -static void gis_plugin_radar_finalize(GObject *gobject) +static void grits_plugin_radar_finalize(GObject *gobject) { - g_debug("GisPluginRadar: finalize"); - GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); + g_debug("GritsPluginRadar: finalize"); + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(gobject); /* Free data */ - gis_http_free(self->sites_http); - g_hash_table_destroy(self->sites); + grits_http_free(self->conus_http); + grits_http_free(self->sites_http); gtk_widget_destroy(self->config); - G_OBJECT_CLASS(gis_plugin_radar_parent_class)->finalize(gobject); + G_OBJECT_CLASS(grits_plugin_radar_parent_class)->finalize(gobject); } -static void gis_plugin_radar_class_init(GisPluginRadarClass *klass) +static void grits_plugin_radar_class_init(GritsPluginRadarClass *klass) { - g_debug("GisPluginRadar: class_init"); + g_debug("GritsPluginRadar: class_init"); GObjectClass *gobject_class = (GObjectClass*)klass; - gobject_class->dispose = gis_plugin_radar_dispose; - gobject_class->finalize = gis_plugin_radar_finalize; + gobject_class->dispose = grits_plugin_radar_dispose; + gobject_class->finalize = grits_plugin_radar_finalize; } -