X-Git-Url: http://pileus.org/git/?a=blobdiff_plain;f=src%2Fplugins%2Fradar.c;h=75fc3783de71e9399ec75d570ae9f40ab7ef8f80;hb=4c964909b5b3c4ea73f93666dde5a2351658464f;hp=26403eaecce57bfd76bddc0426e3e34ebbeba5fc;hpb=7200f71a7275a4ca3faa4d3d8182720fcc0b9496;p=aweather diff --git a/src/plugins/radar.c b/src/plugins/radar.c index 26403ea..75fc378 100644 --- a/src/plugins/radar.c +++ b/src/plugins/radar.c @@ -1,21 +1,24 @@ /* - * Copyright (C) 2009 Andy Spencer - * + * Copyright (C) 2009-2010 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 . */ +#define _XOPEN_SOURCE +#include #include +#include #include #include #include @@ -23,609 +26,616 @@ #include #include -#include +#include #include "radar.h" -#include "marching.h" +#include "level2.h" +#include "../aweather-location.h" -/**************** - * GObject code * - ****************/ -/* Plugin init */ -static void gis_plugin_radar_plugin_init(GisPluginInterface *iface); -static void gis_plugin_radar_expose(GisPlugin *_radar); -static GtkWidget *gis_plugin_radar_get_config(GisPlugin *_self); -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) +void _gtk_bin_set_child(GtkBin *bin, GtkWidget *new) { - g_debug("GisPluginRadar: plugin_init"); - /* Add methods to the interface */ - iface->expose = gis_plugin_radar_expose; - iface->get_config = gis_plugin_radar_get_config; + GtkWidget *old = gtk_bin_get_child(bin); + if (old) + gtk_widget_destroy(old); + gtk_container_add(GTK_CONTAINER(bin), new); + gtk_widget_show_all(new); } -/* Class/Object init */ -static void gis_plugin_radar_init(GisPluginRadar *self) + +static gchar *_find_nearest(time_t time, GList *files, + gsize offset, gchar *format) { - g_debug("GisPluginRadar: class_init"); - /* Set defaults */ - self->soup = NULL; - self->cur_triangles = NULL; - self->cur_num_triangles = 0; + g_debug("GisPluginRadar: _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; + strptime(file+offset, format, &tm); + if (ABS(time - mktime(&tm)) < + ABS(time - nearest_time)) { + nearest_file = file; + nearest_time = mktime(&tm); + } + } - self->config_body = gtk_alignment_new(0, 0, 1, 1); - gtk_container_set_border_width(GTK_CONTAINER(self->config_body), 5); - gtk_container_add(GTK_CONTAINER(self->config_body), gtk_label_new("No radar loaded")); + g_debug("GisPluginRadar: _find_nearest = %s", nearest_file); + if (nearest_file) + return g_strdup(nearest_file); + else + return NULL; } -static void gis_plugin_radar_dispose(GObject *gobject) + + +/************** + * RadarSites * + **************/ +typedef enum { + STATUS_UNLOADED, + STATUS_LOADING, + STATUS_LOADED, +} RadarSiteStatus; +struct _RadarSite { + /* Information */ + city_t *city; + GisMarker *marker; // Map marker for libgis + gpointer *marker_ref; // Reference to maker + + /* Stuff from the parents */ + GisViewer *viewer; + GisHttp *http; + GisPrefs *prefs; + GtkWidget *pconfig; + + /* When loaded */ + 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 + + /* 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 +}; + +/* format: http://mesonet.agron.iastate.edu/data/nexrd2/raw/KABR/KABR_20090510_0323 */ +void _site_update_loading(gchar *file, goffset cur, + goffset total, gpointer _site) { - g_debug("GisPluginRadar: dispose"); - GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); - g_signal_handler_disconnect(self->view, self->time_changed_id); - /* Drop references */ - G_OBJECT_CLASS(gis_plugin_radar_parent_class)->dispose(gobject); + RadarSite *site = _site; + GtkWidget *progress_bar = gtk_bin_get_child(GTK_BIN(site->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); } -static void gis_plugin_radar_finalize(GObject *gobject) +gboolean _site_update_end(gpointer _site) { - g_debug("GisPluginRadar: finalize"); - GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); - /* Free data */ - G_OBJECT_CLASS(gis_plugin_radar_parent_class)->finalize(gobject); + 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)); + } else { + _gtk_bin_set_child(GTK_BIN(site->config), + aweather_level2_get_config(site->level2)); + } + site->status = STATUS_LOADED; + return FALSE; +} +gpointer _site_update_thread(gpointer _site) +{ + RadarSite *site = _site; + g_debug("GisPluginRadar: _update - %s", site->city->code); + site->status = STATUS_LOADING; + site->message = NULL; + + gboolean offline = gis_viewer_get_offline(site->viewer); + gchar *nexrad_url = gis_prefs_get_string(site->prefs, + "aweather/nexrad_url", NULL); + + /* Remove old volume */ + g_debug("GisPluginRadar: _update - remove - %s", site->city->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->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->city->code, + "\\d+ (.*)", (offline ? NULL : dir_list)); + g_free(dir_list); + gchar *nearest = _find_nearest(site->time, files, 5, "%Y%m%d_%H%M"); + g_list_foreach(files, (GFunc)g_free, NULL); + g_list_free(files); + if (!nearest) { + site->message = "No suitable files found"; + goto out; + } + /* Fetch new volume */ + g_debug("GisPluginRadar: _update - 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, + _site_update_loading, site); + g_free(nexrad_url); + g_free(nearest); + g_free(local); + g_free(uri); + if (!file) { + site->message = "Fetch failed"; + goto out; + } + + /* Load and add new volume */ + g_debug("GisPluginRadar: _update - load - %s", site->city->code); + site->level2 = aweather_level2_new_from_file( + site->viewer, colormaps, file, site->city->code); + 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); + +out: + g_idle_add(_site_update_end, site); + return NULL; } -static void gis_plugin_radar_class_init(GisPluginRadarClass *klass) +void _site_update(RadarSite *site) { - g_debug("GisPluginRadar: class_init"); - GObjectClass *gobject_class = (GObjectClass*)klass; - gobject_class->dispose = gis_plugin_radar_dispose; - gobject_class->finalize = gis_plugin_radar_finalize; + site->time = gis_viewer_get_time(site->viewer); + g_debug("GisPluginRadar: _on_time_changed %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); + + /* Fork loading right away so updating the + * list of times doesn't take too long */ + g_thread_create(_site_update_thread, site, FALSE, NULL); } -/************************** - * Data loading functions * - **************************/ -/* Convert a sweep to an 2d array of data points */ -static void bscan_sweep(GisPluginRadar *self, Sweep *sweep, colormap_t *colormap, - guint8 **data, int *width, int *height) +/* RadarSite methods */ +void radar_site_unload(RadarSite *site) { - /* Calculate max number of bins */ - int max_bins = 0; - for (int i = 0; i < sweep->h.nrays; i++) - max_bins = MAX(max_bins, sweep->ray[i]->h.nbins); - - /* Allocate buffer using max number of bins for each ray */ - guint8 *buf = g_malloc0(sweep->h.nrays * max_bins * 4); - - /* Fill the data */ - for (int ri = 0; ri < sweep->h.nrays; ri++) { - Ray *ray = sweep->ray[ri]; - for (int bi = 0; bi < ray->h.nbins; bi++) { - /* copy RGBA into buffer */ - //guint val = dz_f(ray->range[bi]); - guint8 val = (guint8)ray->h.f(ray->range[bi]); - guint buf_i = (ri*max_bins+bi)*4; - buf[buf_i+0] = colormap->data[val][0]; - buf[buf_i+1] = colormap->data[val][1]; - buf[buf_i+2] = colormap->data[val][2]; - buf[buf_i+3] = colormap->data[val][3]; // TESTING - if (val == BADVAL || val == RFVAL || val == APFLAG || - val == NOTFOUND_H || val == NOTFOUND_V || val == NOECHO) { - buf[buf_i+3] = 0x00; // transparent - } - } + if (site->status != STATUS_LOADED) + return; // Abort if it's still loading + + g_debug("GisPluginRadar: radar_site_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); + + /* Remove tab */ + 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; } - /* set output */ - *width = max_bins; - *height = sweep->h.nrays; - *data = buf; + site->status = STATUS_UNLOADED; } -/* Load a sweep as the active texture */ -static void load_sweep(GisPluginRadar *self, Sweep *sweep) +void radar_site_load(RadarSite *site) { - GisOpenGL *opengl = self->opengl; - gis_opengl_begin(opengl); - self->cur_sweep = sweep; - int height, width; - guint8 *data; - bscan_sweep(self, sweep, self->cur_colormap, &data, &width, &height); - glDeleteTextures(1, &self->cur_sweep_tex); - glGenTextures(1, &self->cur_sweep_tex); - glBindTexture(GL_TEXTURE_2D, self->cur_sweep_tex); - glPixelStorei(GL_PACK_ALIGNMENT, 1); - glPixelStorei(GL_UNPACK_ALIGNMENT, 1); - glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR); - glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST); - glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, width, height, 0, - GL_RGBA, GL_UNSIGNED_BYTE, data); - g_free(data); - gis_opengl_redraw(opengl); - gis_opengl_end(opengl); + g_debug("GisPluginRadar: radar_site_load %s", site->city->code); + site->status = STATUS_LOADING; + + /* 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->city->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); + gtk_widget_show_all(site->config); + gtk_widget_show_all(tab); + + /* Set up radar loading */ + site->time_id = g_signal_connect_swapped(site->viewer, "time-changed", + G_CALLBACK(_site_update), site); + site->refresh_id = g_signal_connect_swapped(site->viewer, "refresh", + G_CALLBACK(_site_update), site); + _site_update(site); } -static void load_colormap(GisPluginRadar *self, gchar *table) +void _site_on_location_changed(GisViewer *viewer, + gdouble lat, gdouble lon, gdouble elev, + gpointer _site) { - /* Set colormap so we can draw it on expose */ - for (int i = 0; colormaps[i].name; i++) - if (g_str_equal(colormaps[i].name, table)) - self->cur_colormap = &colormaps[i]; + static gdouble min_dist = EARTH_R / 20; + 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->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); + + /* Load or unload the site if necessasairy */ + if (dist <= min_dist && dist < elev*1.25 && site->status == STATUS_UNLOADED) + radar_site_load(site); + else if (dist > 2*min_dist && site->status != STATUS_UNLOADED) + radar_site_unload(site); } -/* Add selectors to the config area for the sweeps */ -static void on_sweep_clicked(GtkRadioButton *button, gpointer _self); -static void load_radar_gui(GisPluginRadar *self, Radar *radar) +gboolean _site_add_marker(gpointer _site) { - /* Clear existing items */ - GtkWidget *child = gtk_bin_get_child(GTK_BIN(self->config_body)); - if (child) - gtk_widget_destroy(child); - - gdouble elev; - guint rows = 1, cols = 1, cur_cols; - gchar row_label_str[64], col_label_str[64], button_str[64]; - GtkWidget *row_label, *col_label, *button = NULL, *elev_box = NULL; - GtkWidget *table = gtk_table_new(rows, cols, FALSE); - - for (guint vi = 0; vi < radar->h.nvolumes; vi++) { - Volume *vol = radar->v[vi]; - if (vol == NULL) continue; - rows++; cols = 1; elev = 0; - - /* Row label */ - g_snprintf(row_label_str, 64, "%s:", vol->h.type_str); - row_label = gtk_label_new(row_label_str); - gtk_label_set_use_markup(GTK_LABEL(row_label), TRUE); - gtk_misc_set_alignment(GTK_MISC(row_label), 1, 0.5); - gtk_table_attach(GTK_TABLE(table), row_label, - 0,1, rows-1,rows, GTK_FILL,GTK_FILL, 5,0); - - for (guint si = 0; si < vol->h.nsweeps; si++) { - Sweep *sweep = vol->sweep[si]; - if (sweep == NULL || sweep->h.elev == 0) continue; - if (sweep->h.elev != elev) { - cols++; - elev = sweep->h.elev; - - /* Column label */ - g_object_get(table, "n-columns", &cur_cols, NULL); - if (cols > cur_cols) { - g_snprintf(col_label_str, 64, "%.2f°", elev); - col_label = gtk_label_new(col_label_str); - gtk_label_set_use_markup(GTK_LABEL(col_label), TRUE); - gtk_widget_set_size_request(col_label, 50, -1); - gtk_table_attach(GTK_TABLE(table), col_label, - cols-1,cols, 0,1, GTK_FILL,GTK_FILL, 0,0); - } - - elev_box = gtk_hbox_new(TRUE, 0); - gtk_table_attach(GTK_TABLE(table), elev_box, - cols-1,cols, rows-1,rows, GTK_FILL,GTK_FILL, 0,0); - } + RadarSite *site = _site; + site->marker = gis_marker_new(site->city->name); + GIS_OBJECT(site->marker)->center = site->city->pos; + GIS_OBJECT(site->marker)->lod = EARTH_R*site->city->lod; + site->marker_ref = gis_viewer_add(site->viewer, + GIS_OBJECT(site->marker), GIS_LEVEL_OVERLAY, FALSE); + return FALSE; +} +RadarSite *radar_site_new(city_t *city, GtkWidget *pconfig, + GisViewer *viewer, GisPrefs *prefs, GisHttp *http) +{ + RadarSite *site = g_new0(RadarSite, 1); + site->viewer = g_object_ref(viewer); + site->prefs = g_object_ref(prefs); + site->http = http; + site->city = city; + site->pconfig = pconfig; + + /* Add marker */ + g_idle_add(_site_add_marker, site); + + /* Connect signals */ + site->location_id = g_signal_connect(viewer, "location-changed", + G_CALLBACK(_site_on_location_changed), site); + return site; +} +void radar_site_free(RadarSite *site) +{ + radar_site_unload(site); + gis_viewer_remove(site->viewer, site->marker_ref); + if (site->location_id) + g_signal_handler_disconnect(site->viewer, site->location_id); + g_object_unref(site->viewer); + g_object_unref(site->prefs); + g_free(site); +} - /* Button */ - g_snprintf(button_str, 64, "%3.2f", elev); - button = gtk_radio_button_new_with_label_from_widget( - GTK_RADIO_BUTTON(button), button_str); - gtk_widget_set_size_request(button, -1, 26); - //button = gtk_radio_button_new_from_widget(GTK_RADIO_BUTTON(button)); - //gtk_widget_set_size_request(button, -1, 22); - g_object_set(button, "draw-indicator", FALSE, NULL); - gtk_box_pack_end(GTK_BOX(elev_box), button, TRUE, TRUE, 0); - g_object_set_data(G_OBJECT(button), "type", vol->h.type_str); - g_object_set_data(G_OBJECT(button), "sweep", sweep); - g_signal_connect(button, "clicked", G_CALLBACK(on_sweep_clicked), self); - } - } - gtk_container_add(GTK_CONTAINER(self->config_body), table); - gtk_widget_show_all(table); +/************** + * RadarConus * + **************/ +struct _RadarConus { + GisViewer *viewer; + GisHttp *http; + GtkWidget *config; + time_t time; + GisTile *tile; + gpointer *tile_ref; + GdkPixbuf *pixbuf; + const gchar *message; + gchar *nearest; + + guint time_id; // "time-changed" callback ID + guint refresh_id; // "refresh" callback ID +}; + +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); } -static void _gis_plugin_radar_grid_set(GRIDCELL *grid, int gi, Ray *ray, int bi) +gboolean _conus_update_end(gpointer _conus) { - Range range = ray->range[bi]; - - double angle = d2r(ray->h.azimuth); - double tilt = d2r(ray->h.elev); - - double lx = sin(angle); - double ly = cos(angle); - double lz = sin(tilt); - - double dist = bi*ray->h.gate_size + ray->h.range_bin1; - - grid->p[gi].x = lx*dist; - grid->p[gi].y = ly*dist; - grid->p[gi].z = lz*dist; - - guint8 val = (guint8)ray->h.f(ray->range[bi]); - if (val == BADVAL || val == RFVAL || val == APFLAG || - val == NOTFOUND_H || val == NOTFOUND_V || val == NOECHO || - val > 80) - val = 0; - grid->val[gi] = (float)val; - //g_debug("(%.2f,%.2f,%.2f) - (%.0f,%.0f,%.0f) = %d", - // angle, tilt, dist, - // grid->p[gi].x, - // grid->p[gi].y, - // grid->p[gi].z, - // val); + RadarConus *conus = _conus; + + guchar *pixels = gdk_pixbuf_get_pixels(conus->pixbuf); + gboolean alpha = gdk_pixbuf_get_has_alpha(conus->pixbuf); + gint width = gdk_pixbuf_get_width(conus->pixbuf); + gint height = gdk_pixbuf_get_height(conus->pixbuf); + + if (!conus->tile->data) { + conus->tile->data = g_new0(guint, 1); + glGenTextures(1, conus->tile->data); + } + + guint *tex = conus->tile->data; + glBindTexture(GL_TEXTURE_2D, *tex); + + glPixelStorei(GL_UNPACK_ALIGNMENT, 1); + glPixelStorei(GL_PACK_ALIGNMENT, 1); + glTexImage2D(GL_TEXTURE_2D, 0, 4, width, height, 0, + (alpha ? GL_RGBA : GL_RGB), GL_UNSIGNED_BYTE, pixels); + 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); + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP); + glFlush(); + + /* finish */ + _gtk_bin_set_child(GTK_BIN(conus->config), + gtk_label_new(conus->nearest)); + gtk_widget_queue_draw(GTK_WIDGET(conus->viewer)); + + /* free data */ + g_object_unref(conus->pixbuf); + g_free(conus->nearest); + + return FALSE; } -/* Load a radar from a decompressed file */ -static void load_radar(GisPluginRadar *self, gchar *radar_file) +gpointer _conus_update_thread(gpointer _conus) { - char *dir = g_path_get_dirname(radar_file); - char *site = g_path_get_basename(dir); - g_free(dir); - g_debug("GisPluginRadar: load_radar - Loading new radar"); - RSL_read_these_sweeps("all", NULL); - Radar *radar = self->cur_radar = RSL_wsr88d_to_radar(radar_file, site); - if (radar == NULL) { - g_warning("fail to load radar: path=%s, site=%s", radar_file, site); - g_free(site); - return; + RadarConus *conus = _conus; + + /* Find nearest */ + gboolean offline = gis_viewer_get_offline(conus->viewer); + gchar *conus_url = "http://radar.weather.gov/Conus/RadarImg/"; + GList *files = gis_http_available(conus->http, + "^Conus_[^\"]*_N0Ronly.gif$", "", + NULL, (offline ? NULL : conus_url)); + conus->nearest = _find_nearest(conus->time, files, 6, "%Y%m%d_%H%M"); + g_list_foreach(files, (GFunc)g_free, NULL); + g_list_free(files); + if (!conus->nearest) { + conus->message = "No suitable files"; + goto out; } - g_free(site); -#ifdef MARCHING - /* Load the surface */ - if (self->cur_triangles) { - g_free(self->cur_triangles); - self->cur_triangles = NULL; + /* Fetch the image */ + gchar *uri = g_strconcat(conus_url, conus->nearest, NULL); + gchar *path = gis_http_fetch(conus->http, uri, conus->nearest, GIS_ONCE, + _conus_update_loading, conus); + g_free(uri); + if (!path) { + conus->message = "Fetch failed"; + goto out; } - self->cur_num_triangles = 0; - int x = 1; - for (guint vi = 0; vi < radar->h.nvolumes; vi++) { - if (radar->v[vi] == NULL) continue; - - for (guint si = 0; si+1 < radar->v[vi]->h.nsweeps; si++) { - Sweep *sweep0 = radar->v[vi]->sweep[si+0]; - Sweep *sweep1 = radar->v[vi]->sweep[si+1]; - - //g_debug("_gis_plugin_radar_expose: sweep[%3d-%3d] -- nrays = %d, %d", - // si, si+1,sweep0->h.nrays, sweep1->h.nrays); - - /* Skip super->regular resolution switch for now */ - if (sweep0 == NULL || sweep0->h.elev == 0 || - sweep1 == NULL || sweep1->h.elev == 0 || - sweep0->h.nrays != sweep1->h.nrays) - continue; - - /* We repack the arrays so that raysX[0] is always north, etc */ - Ray **rays0 = g_malloc0(sizeof(Ray*)*sweep0->h.nrays); - Ray **rays1 = g_malloc0(sizeof(Ray*)*sweep1->h.nrays); - - for (guint ri = 0; ri < sweep0->h.nrays; ri++) - rays0[(guint)(sweep0->ray[ri]->h.azimuth * sweep0->h.nrays / 360)] = - sweep0->ray[ri]; - for (guint ri = 0; ri < sweep1->h.nrays; ri++) - rays1[(guint)(sweep1->ray[ri]->h.azimuth * sweep1->h.nrays / 360)] = - sweep1->ray[ri]; - - for (guint ri = 0; ri+x < sweep0->h.nrays; ri+=x) { - //g_debug("_gis_plugin_radar_expose: ray[%3d-%3d] -- nbins = %d, %d, %d, %d", - // ri, ri+x, - // rays0[ri ]->h.nbins, - // rays0[ri+1]->h.nbins, - // rays1[ri ]->h.nbins, - // rays1[ri+1]->h.nbins); - - for (guint bi = 0; bi+x < rays1[ri]->h.nbins; bi+=x) { - GRIDCELL grid = {}; - _gis_plugin_radar_grid_set(&grid, 7, rays0[(ri )%sweep0->h.nrays], bi+x); - _gis_plugin_radar_grid_set(&grid, 6, rays0[(ri+x)%sweep0->h.nrays], bi+x); - _gis_plugin_radar_grid_set(&grid, 5, rays0[(ri+x)%sweep0->h.nrays], bi ); - _gis_plugin_radar_grid_set(&grid, 4, rays0[(ri )%sweep0->h.nrays], bi ); - _gis_plugin_radar_grid_set(&grid, 3, rays1[(ri )%sweep0->h.nrays], bi+x); - _gis_plugin_radar_grid_set(&grid, 2, rays1[(ri+x)%sweep0->h.nrays], bi+x); - _gis_plugin_radar_grid_set(&grid, 1, rays1[(ri+x)%sweep0->h.nrays], bi ); - _gis_plugin_radar_grid_set(&grid, 0, rays1[(ri )%sweep0->h.nrays], bi ); - - TRIANGLE tris[10]; - int n = march_one_cube(grid, 40, tris); - - self->cur_triangles = g_realloc(self->cur_triangles, - (self->cur_num_triangles+n)*sizeof(TRIANGLE)); - for (int i = 0; i < n; i++) { - //g_debug("triangle: "); - //g_debug("\t(%f,%f,%f)", tris[i].p[0].x, tris[i].p[0].y, tris[i].p[0].z); - //g_debug("\t(%f,%f,%f)", tris[i].p[1].x, tris[i].p[1].y, tris[i].p[1].z); - //g_debug("\t(%f,%f,%f)", tris[i].p[2].x, tris[i].p[2].y, tris[i].p[2].z); - self->cur_triangles[self->cur_num_triangles+i] = tris[i]; - } - self->cur_num_triangles += n; - //g_debug(" "); - } - } - } - break; // Exit after first volume (reflectivity) + + /* Load the image to a pixbuf */ + GError *error = NULL; + conus->pixbuf = gdk_pixbuf_new_from_file(path, &error); + g_free(path); + if (!gdk_pixbuf_get_has_alpha(conus->pixbuf)) { + GdkPixbuf *tmp = gdk_pixbuf_add_alpha(conus->pixbuf, TRUE, 0xff, 0xff, 0xff); + g_object_unref(conus->pixbuf); + conus->pixbuf = tmp; } -#endif - /* Load the first sweep by default */ - if (radar->h.nvolumes < 1 || radar->v[0]->h.nsweeps < 1) { - g_warning("No sweeps found\n"); - } else { - /* load first available sweep */ - for (int vi = 0; vi < radar->h.nvolumes; vi++) { - if (radar->v[vi]== NULL) continue; - for (int si = 0; si < radar->v[vi]->h.nsweeps; si++) { - if (radar->v[vi]->sweep[si]== NULL) continue; - load_colormap(self, radar->v[vi]->h.type_str); - load_sweep(self, radar->v[vi]->sweep[si]); + /* Map the pixbuf's alpha values */ + const guchar colormap[][2][4] = { + {{0x04, 0xe9, 0xe7}, {0x04, 0xe9, 0xe7, 0x30}}, + {{0x01, 0x9f, 0xf4}, {0x01, 0x9f, 0xf4, 0x60}}, + {{0x03, 0x00, 0xf4}, {0x03, 0x00, 0xf4, 0x90}}, + }; + guchar *pixels = gdk_pixbuf_get_pixels(conus->pixbuf); + gint height = gdk_pixbuf_get_height(conus->pixbuf); + gint width = gdk_pixbuf_get_width(conus->pixbuf); + for (int i = 0; i < width*height; i++) { + for (int j = 0; j < G_N_ELEMENTS(colormap); j++) { + if (pixels[i*4+0] > 0xe0 && + pixels[i*4+1] > 0xe0 && + pixels[i*4+2] > 0xe0) { + pixels[i*4+3] = 0x00; + break; + } + if (pixels[i*4+0] == colormap[j][0][0] && + pixels[i*4+1] == colormap[j][0][1] && + pixels[i*4+2] == colormap[j][0][2]) { + pixels[i*4+0] = colormap[j][1][0]; + pixels[i*4+1] = colormap[j][1][1]; + pixels[i*4+2] = colormap[j][1][2]; + pixels[i*4+3] = colormap[j][1][3]; break; } - break; } } - load_radar_gui(self, radar); -} - -/***************** - * ASync helpers * - *****************/ -typedef struct { - GisPluginRadar *self; - gchar *radar_file; -} decompressed_t; - -static void decompressed_cb(GPid pid, gint status, gpointer _udata) -{ - g_debug("GisPluginRadar: decompressed_cb"); - decompressed_t *udata = _udata; - if (status != 0) { - g_warning("wsr88ddec exited with status %d", status); - return; - } - load_radar(udata->self, udata->radar_file); - g_spawn_close_pid(pid); - g_free(udata->radar_file); - g_free(udata); +out: + g_idle_add(_conus_update_end, conus); + return NULL; } -static void cache_chunk_cb(char *path, goffset cur, goffset total, gpointer _self) +void _conus_update(RadarConus *conus) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - double percent = (double)cur/total; + conus->time = gis_viewer_get_time(conus->viewer); + g_debug("GisPluginRadar: _conus_update - %d", + (gint)conus->time); - //g_message("GisPluginRadar: cache_chunk_cb - %lld/%lld = %.2f%%", - // cur, total, percent*100); + /* 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(conus->config), progress); - gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(self->progress_bar), MIN(percent, 1.0)); + g_thread_create(_conus_update_thread, conus, FALSE, NULL); +} - gchar *msg = g_strdup_printf("Loading radar... %5.1f%% (%.2f/%.2f MB)", - percent*100, (double)cur/1000000, (double)total/1000000); - gtk_label_set_text(GTK_LABEL(self->progress_label), msg); - g_free(msg); +RadarConus *radar_conus_new(GtkWidget *pconfig, + GisViewer *viewer, GisHttp *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); + conus->tile = gis_tile_new(NULL, + 50.406626367301044, 50.406626367301044-0.017971305190311*1600, + -127.620375523875420+0.017971305190311*3400, -127.620375523875420); + conus->tile->zindex = 1; + conus->tile_ref = gis_viewer_add(viewer, + GIS_OBJECT(conus->tile), GIS_LEVEL_WORLD, TRUE); + + 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); + + gtk_notebook_append_page(GTK_NOTEBOOK(pconfig), conus->config, gtk_label_new("Conus")); + _conus_update(conus); + return conus; } -static void cache_done_cb(char *path, gboolean updated, gpointer _self) +void radar_conus_free(RadarConus *conus) { - g_debug("GisPluginRadar: cache_done_cb - updated = %d", updated); - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - char *decompressed = g_strconcat(path, ".raw", NULL); - if (!updated && g_file_test(decompressed, G_FILE_TEST_EXISTS)) { - load_radar(self, decompressed); - return; - } + g_signal_handler_disconnect(conus->viewer, conus->time_id); + g_signal_handler_disconnect(conus->viewer, conus->refresh_id); - decompressed_t *udata = g_malloc(sizeof(decompressed_t)); - udata->self = self; - udata->radar_file = decompressed; - g_debug("GisPluginRadar: cache_done_cb - File updated, decompressing.."); - char *argv[] = {"wsr88ddec", path, decompressed, NULL}; - GPid pid; - GError *error = NULL; - g_spawn_async( - NULL, // const gchar *working_directory, - argv, // gchar **argv, - NULL, // gchar **envp, - G_SPAWN_SEARCH_PATH| - G_SPAWN_DO_NOT_REAP_CHILD, - // GSpawnFlags flags, - NULL, // GSpawnChildSetupFunc child_setup, - NULL, // gpointer user_data, - &pid, // GPid *child_pid, - &error); // GError **error - if (error) { - g_warning("failed to decompress WSR88D data: %s", - error->message); - g_error_free(error); + if (conus->tile->data) { + glDeleteTextures(1, conus->tile->data); + g_free(conus->tile->data); } - g_child_watch_add(pid, decompressed_cb, udata); - self->soup = NULL; -} + gis_viewer_remove(conus->viewer, conus->tile_ref); -/************* - * Callbacks * - *************/ -static void on_sweep_clicked(GtkRadioButton *button, gpointer _self) -{ - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - load_colormap(self, g_object_get_data(G_OBJECT(button), "type" )); - load_sweep (self, g_object_get_data(G_OBJECT(button), "sweep")); + g_object_unref(conus->viewer); + g_free(conus); } -static void on_time_changed(GisView *view, const char *time, gpointer _self) + +/****************** + * GisPluginRadar * + ******************/ +static void _draw_hud(GisCallback *callback, gpointer _self) { + /* TODO */ GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - g_debug("GisPluginRadar: on_time_changed - setting time=%s", time); - // format: http://mesonet.agron.iastate.edu/data/nexrd2/raw/KABR/KABR_20090510_0323 - char *site = gis_view_get_site(view); - char *path = g_strdup_printf("nexrd2/raw/%s/%s_%s", site, site, time); - - /* Set up progress bar */ - GtkWidget *child = gtk_bin_get_child(GTK_BIN(self->config_body)); - if (child) gtk_widget_destroy(child); - - GtkWidget *vbox = gtk_vbox_new(FALSE, 10); - gtk_container_set_border_width(GTK_CONTAINER(vbox), 10); - self->progress_bar = gtk_progress_bar_new(); - self->progress_label = gtk_label_new("Loading radar..."); - gtk_box_pack_start(GTK_BOX(vbox), self->progress_bar, FALSE, FALSE, 0); - gtk_box_pack_start(GTK_BOX(vbox), self->progress_label, FALSE, FALSE, 0); - gtk_container_add(GTK_CONTAINER(self->config_body), vbox); - gtk_widget_show_all(self->config_body); - - /* Clear radar */ - if (self->cur_radar) - RSL_free_radar(self->cur_radar); - self->cur_radar = NULL; - self->cur_sweep = NULL; - gis_opengl_redraw(self->opengl); - - /* Start loading the new radar */ - if (self->soup) { - soup_session_abort(self->soup); - self->soup = NULL; + if (!self->colormap) + return; + + g_debug("GisPluginRadar: _draw_hud"); + /* Print the color table */ + glMatrixMode(GL_MODELVIEW ); glLoadIdentity(); + glMatrixMode(GL_PROJECTION); glLoadIdentity(); + glDisable(GL_TEXTURE_2D); + glDisable(GL_ALPHA_TEST); + 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 } - gchar *base = gis_prefs_get_string(self->prefs, "aweather/nexrad_url"); - if (gis_world_get_offline(self->world)) - self->soup = cache_file(base, path, GIS_ONCE, - cache_chunk_cb, cache_done_cb, self); - else - self->soup = cache_file(base, path, GIS_UPDATE, - cache_chunk_cb, cache_done_cb, self); - g_free(path); + glEnd(); } -/*********** - * Methods * - ***********/ -GisPluginRadar *gis_plugin_radar_new(GisWorld *world, GisView *view, GisOpenGL *opengl, GisPrefs *prefs) +/* Methods */ +GisPluginRadar *gis_plugin_radar_new(GisViewer *viewer, GisPrefs *prefs) { /* TODO: move to constructor if possible */ g_debug("GisPluginRadar: new"); GisPluginRadar *self = g_object_new(GIS_TYPE_PLUGIN_RADAR, NULL); - self->world = world; - self->view = view; - self->opengl = opengl; + self->viewer = viewer; self->prefs = prefs; - self->time_changed_id = g_signal_connect(view, "time-changed", - G_CALLBACK(on_time_changed), self); + + /* Load HUD */ + GisCallback *hud_cb = gis_callback_new(_draw_hud, self); + self->hud_ref = gis_viewer_add(viewer, GIS_OBJECT(hud_cb), GIS_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++) { + if (city->type != LOCATION_CITY) + continue; + RadarSite *site = radar_site_new(city, self->config, + self->viewer, self->prefs, self->sites_http); + g_hash_table_insert(self->sites, city->code, site); + } + return self; } static GtkWidget *gis_plugin_radar_get_config(GisPlugin *_self) { GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - return self->config_body; + return self->config; } -static void gis_plugin_radar_expose(GisPlugin *_self) +/* 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) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - g_debug("GisPluginRadar: expose"); - if (self->cur_sweep == NULL) - return; - Sweep *sweep = self->cur_sweep; - -#ifdef MARCHING - /* Draw the surface */ - glMatrixMode(GL_MODELVIEW); - glPushMatrix(); - glDisable(GL_TEXTURE_2D); - float light_ambient[] = {0.1f, 0.1f, 0.0f}; - float light_diffuse[] = {0.9f, 0.9f, 0.9f}; - float light_position[] = {-300000.0f, 500000.0f, 400000.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_LIGHT0); - glEnable(GL_LIGHTING); - glEnable(GL_COLOR_MATERIAL); - glColor4f(1,1,1,0.75); - g_debug("ntri=%d", self->cur_num_triangles); - glBegin(GL_TRIANGLES); - for (int i = 0; i < self->cur_num_triangles; i++) { - TRIANGLE t = self->cur_triangles[i]; - do_normal(t.p[0].x, t.p[0].y, t.p[0].z, - t.p[1].x, t.p[1].y, t.p[1].z, - t.p[2].x, t.p[2].y, t.p[2].z); - glVertex3f(t.p[0].x, t.p[0].y, t.p[0].z); - glVertex3f(t.p[1].x, t.p[1].y, t.p[1].z); - glVertex3f(t.p[2].x, t.p[2].y, t.p[2].z); - } - glEnd(); - glPopMatrix(); -#endif - - /* Draw the rays */ - glDisable(GL_LIGHTING); - glDisable(GL_COLOR_MATERIAL); - glMatrixMode(GL_MODELVIEW); - glPushMatrix(); - glBindTexture(GL_TEXTURE_2D, self->cur_sweep_tex); - glEnable(GL_TEXTURE_2D); - glDisable(GL_ALPHA_TEST); - glColor4f(1,1,1,1); - glBegin(GL_QUAD_STRIP); - for (int ri = 0; ri <= sweep->h.nrays; ri++) { - Ray *ray = NULL; - double angle = 0; - if (ri < sweep->h.nrays) { - ray = sweep->ray[ri]; - angle = d2r(ray->h.azimuth - ((double)ray->h.beam_width/2.)); - } else { - /* Do the right side of the last sweep */ - ray = sweep->ray[ri-1]; - angle = d2r(ray->h.azimuth + ((double)ray->h.beam_width/2.)); - } - - double lx = sin(angle); - double ly = cos(angle); - - double near_dist = ray->h.range_bin1; - double far_dist = ray->h.nbins*ray->h.gate_size + ray->h.range_bin1; - - /* (find middle of bin) / scale for opengl */ - // near left - glTexCoord2f(0.0, (double)ri/sweep->h.nrays-0.01); - glVertex3f(lx*near_dist, ly*near_dist, 2.0); - - // far left - // todo: correct range-height function - double height = sin(d2r(ray->h.elev)) * far_dist; - glTexCoord2f(1.0, (double)ri/sweep->h.nrays-0.01); - glVertex3f(lx*far_dist, ly*far_dist, height); - } - //g_print("ri=%d, nr=%d, bw=%f\n", _ri, sweep->h.nrays, sweep->h.beam_width); - glEnd(); - glPopMatrix(); - - /* Texture debug */ - //glBegin(GL_QUADS); - //glTexCoord2d( 0., 0.); glVertex3f(-500., 0., 0.); // bot left - //glTexCoord2d( 0., 1.); glVertex3f(-500., 500., 0.); // top left - //glTexCoord2d( 1., 1.); glVertex3f( 0., 500., 3.); // top right - //glTexCoord2d( 1., 0.); glVertex3f( 0., 0., 3.); // bot right - //glEnd(); + g_debug("GisPluginRadar: plugin_init"); + /* Add methods to the interface */ + iface->get_config = gis_plugin_radar_get_config; +} +static void gis_plugin_radar_init(GisPluginRadar *self) +{ + g_debug("GisPluginRadar: 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->conus_http = gis_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 = gtk_notebook_new(); + gtk_notebook_set_tab_pos(GTK_NOTEBOOK(self->config), GTK_POS_LEFT); +} +static void gis_plugin_radar_dispose(GObject *gobject) +{ + g_debug("GisPluginRadar: dispose"); + GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); + gis_viewer_remove(self->viewer, self->hud_ref); + radar_conus_free(self->conus); + /* Drop references */ + G_OBJECT_CLASS(gis_plugin_radar_parent_class)->dispose(gobject); +} +static void gis_plugin_radar_finalize(GObject *gobject) +{ + g_debug("GisPluginRadar: finalize"); + GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject); + /* Free data */ + gis_http_free(self->conus_http); + gis_http_free(self->sites_http); + g_hash_table_destroy(self->sites); + gtk_widget_destroy(self->config); + G_OBJECT_CLASS(gis_plugin_radar_parent_class)->finalize(gobject); - /* Print the color table */ - glDisable(GL_TEXTURE_2D); - glDisable(GL_DEPTH_TEST); - glMatrixMode(GL_MODELVIEW ); glPushMatrix(); glLoadIdentity(); - glMatrixMode(GL_PROJECTION); glPushMatrix(); glLoadIdentity(); - glBegin(GL_QUADS); - int i; - for (i = 0; i < 256; i++) { - glColor4ub(self->cur_colormap->data[i][0], - self->cur_colormap->data[i][1], - self->cur_colormap->data[i][2], - self->cur_colormap->data[i][3]); - 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 - } - glEnd(); - glEnable(GL_DEPTH_TEST); - glEnable(GL_ALPHA_TEST); - glMatrixMode(GL_PROJECTION); glPopMatrix(); - glMatrixMode(GL_MODELVIEW ); glPopMatrix(); +} +static void gis_plugin_radar_class_init(GisPluginRadarClass *klass) +{ + g_debug("GisPluginRadar: class_init"); + GObjectClass *gobject_class = (GObjectClass*)klass; + gobject_class->dispose = gis_plugin_radar_dispose; + gobject_class->finalize = gis_plugin_radar_finalize; }