X-Git-Url: http://pileus.org/git/?p=aweather;a=blobdiff_plain;f=src%2Fplugins%2Fradar.c;h=2dcea83986dc6a0ad45400998a8c0e85d8789b6b;hp=26403eaecce57bfd76bddc0426e3e34ebbeba5fc;hb=cc82e9478bb47cccb085fb2738053b0c5f6b8298;hpb=7200f71a7275a4ca3faa4d3d8182720fcc0b9496 diff --git a/src/plugins/radar.c b/src/plugins/radar.c index 26403ea..2dcea83 100644 --- a/src/plugins/radar.c +++ b/src/plugins/radar.c @@ -1,631 +1,843 @@ /* - * Copyright (C) 2009 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 * 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 -#include #include #include -#include +#include #include "radar.h" -#include "marching.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) +#include "level2.h" +#include "../aweather-location.h" + +#include "compat.h" + +static void aweather_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) { - g_debug("GisPluginRadar: class_init"); - /* Set defaults */ - self->soup = NULL; - self->cur_triangles = NULL; - self->cur_num_triangles = 0; + 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; + 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; + 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("RadarSite: 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; + GritsMarker *marker; // Map marker for grits + + /* Stuff from the parents */ + GritsViewer *viewer; + GritsHttp *http; + GritsPrefs *prefs; + GtkWidget *pconfig; + + /* When loaded */ + gboolean hidden; + RadarSiteStatus status; // Loading status for the site + GtkWidget *config; + 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 + guint idle_source; // _site_update_end idle source +}; + +/* 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("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 { + 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("RadarSite: update_thread - %s", site->city->code); + site->message = NULL; + + gboolean offline = grits_viewer_get_offline(site->viewer); + gchar *nexrad_url = grits_prefs_get_string(site->prefs, + "aweather/nexrad_url", NULL); + + /* Find nearest volume (temporally) */ + 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 = 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); + 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("RadarSite: update_thread - fetch"); + gchar *local = g_strconcat(site->city->code, "/", nearest, NULL); + gchar *uri = g_strconcat(nexrad_url, "/", local, NULL); + 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) { + site->message = "Fetch failed"; + goto out; + } + + /* Load and add new volume */ + g_debug("RadarSite: update_thread - load - %s", site->city->code); + site->level2 = aweather_level2_new_from_file( + file, site->city->code, colormaps); + g_free(file); + if (!site->level2) { + site->message = "Load failed"; + goto out; + } + grits_object_hide(GRITS_OBJECT(site->level2), site->hidden); + grits_viewer_add(site->viewer, GRITS_OBJECT(site->level2), + GRITS_LEVEL_WORLD+3, TRUE); + +out: + if (!site->idle_source) + site->idle_source = 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; + 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..."); + 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_new("site-update-thread", _site_update_thread, site); } -/************************** - * 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("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 */ + if (site->config) + gtk_widget_destroy(site->config); + + /* Remove radar */ + if (site->level2) { + grits_viewer_remove(site->viewer, GRITS_OBJECT(site->level2)); + site->level2 = 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("RadarSite: load %s", site->city->code); + + /* Add tab page */ + site->config = gtk_alignment_new(0, 0, 1, 1); + 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); + 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", + 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(GritsViewer *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 / 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->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) +static gboolean on_marker_clicked(GritsObject *marker, GdkEvent *event, RadarSite *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); - } - - - /* 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); + 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); } } - gtk_container_add(GTK_CONTAINER(self->config_body), table); - gtk_widget_show_all(table); + return TRUE; } -static void _gis_plugin_radar_grid_set(GRIDCELL *grid, int gi, Ray *ray, int bi) +RadarSite *radar_site_new(city_t *city, GtkWidget *pconfig, + GritsViewer *viewer, GritsPrefs *prefs, GritsHttp *http) { - 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); + RadarSite *site = g_new0(RadarSite, 1); + site->viewer = g_object_ref(viewer); + site->prefs = g_object_ref(prefs); + //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 = 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", + G_CALLBACK(_site_on_location_changed), site); + return site; } -/* Load a radar from a decompressed file */ -static void load_radar(GisPluginRadar *self, gchar *radar_file) +void radar_site_free(RadarSite *site) { - 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; - } + radar_site_unload(site); + 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_object_unref(site->prefs); g_free(site); - -#ifdef MARCHING - /* Load the surface */ - if (self->cur_triangles) { - g_free(self->cur_triangles); - self->cur_triangles = NULL; - } - 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) - } -#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]); - 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) +/************** + * 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) { - 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); + 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 cache_chunk_cb(char *path, goffset cur, goffset total, gpointer _self) +/* Copy images to graphics memory */ +static void _conus_update_end_copy(GritsTile *tile, guchar *pixels) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - double percent = (double)cur/total; + if (!tile->tex) + glGenTextures(1, &tile->tex); - //g_message("GisPluginRadar: cache_chunk_cb - %lld/%lld = %.2f%%", - // cur, total, percent*100); + gchar *clear = g_malloc0(2048*2048*4); + glBindTexture(GL_TEXTURE_2D, tile->tex); - gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(self->progress_bar), MIN(percent, 1.0)); + 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); +} - 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); +/* 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]; + } + } } -static void cache_done_cb(char *path, gboolean updated, gpointer _self) +gboolean _conus_update_end(gpointer _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; + 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; } - 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; + /* Load and pixbuf */ 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); + 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; } - g_child_watch_add(pid, decompressed_cb, udata); - self->soup = NULL; -} -/************* - * 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")); + /* 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; } -static void on_time_changed(GisView *view, const char *time, gpointer _self) +gpointer _conus_update_thread(gpointer _conus) { - 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; + 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; } - 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); + +out: + g_debug("Conus: update_thread - done"); + if (!conus->idle_source) + conus->idle_source = g_idle_add(_conus_update_end, conus); + return NULL; } -/*********** - * Methods * - ***********/ -GisPluginRadar *gis_plugin_radar_new(GisWorld *world, GisView *view, GisOpenGL *opengl, GisPrefs *prefs) +void _conus_update(RadarConus *conus) { - /* 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->prefs = prefs; - self->time_changed_id = g_signal_connect(view, "time-changed", - G_CALLBACK(on_time_changed), self); - return self; + 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); } -static GtkWidget *gis_plugin_radar_get_config(GisPlugin *_self) +RadarConus *radar_conus_new(GtkWidget *pconfig, + GritsViewer *viewer, GritsHttp *http) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - return self->config_body; + 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; } -static void gis_plugin_radar_expose(GisPlugin *_self) +void radar_conus_free(RadarConus *conus) { - GisPluginRadar *self = GIS_PLUGIN_RADAR(_self); - g_debug("GisPluginRadar: expose"); - if (self->cur_sweep == NULL) - return; - Sweep *sweep = self->cur_sweep; + 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); + } -#ifdef MARCHING - /* Draw the surface */ - glMatrixMode(GL_MODELVIEW); - glPushMatrix(); + 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); - 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); + glDisable(GL_ALPHA_TEST); + glDisable(GL_CULL_FACE); + glDisable(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); + + 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(); } - 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.)); +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 { - /* Do the right side of the last sweep */ - ray = sweep->ray[ri-1]; - angle = d2r(ray->h.azimuth + ((double)ray->h.beam_width/2.)); + g_warning("GritsPluginRadar: _update_hidden - no site or counus found"); } + } + grits_viewer_queue_draw(viewer); +} - double lx = sin(angle); - double ly = cos(angle); +/* Methods */ +GritsPluginRadar *grits_plugin_radar_new(GritsViewer *viewer, GritsPrefs *prefs) +{ + /* TODO: move to constructor if possible */ + 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 */ + 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++) { + 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); + } - double near_dist = ray->h.range_bin1; - double far_dist = ray->h.nbins*ray->h.gate_size + ray->h.range_bin1; + return self; +} - /* (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); +static GtkWidget *grits_plugin_radar_get_config(GritsPlugin *_self) +{ + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(_self); + return self->config; +} - // 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); +/* GObject code */ +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("GritsPluginRadar: plugin_init"); + /* Add methods to the interface */ + iface->get_config = grits_plugin_radar_get_config; +} +static void grits_plugin_radar_init(GritsPluginRadar *self) +{ + g_debug("GritsPluginRadar: class_init"); + /* Set defaults */ + 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); } - //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(); - - /* 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 + + /* Need to position on the top because of Win32 bug */ + gtk_notebook_set_tab_pos(GTK_NOTEBOOK(self->config), GTK_POS_LEFT); +} +static void grits_plugin_radar_dispose(GObject *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); } - glEnd(); - glEnable(GL_DEPTH_TEST); - glEnable(GL_ALPHA_TEST); - glMatrixMode(GL_PROJECTION); glPopMatrix(); - glMatrixMode(GL_MODELVIEW ); glPopMatrix(); + /* Drop references */ + G_OBJECT_CLASS(grits_plugin_radar_parent_class)->dispose(gobject); +} +static void grits_plugin_radar_finalize(GObject *gobject) +{ + g_debug("GritsPluginRadar: finalize"); + GritsPluginRadar *self = GRITS_PLUGIN_RADAR(gobject); + /* Free data */ + grits_http_free(self->conus_http); + grits_http_free(self->sites_http); + gtk_widget_destroy(self->config); + G_OBJECT_CLASS(grits_plugin_radar_parent_class)->finalize(gobject); + +} +static void grits_plugin_radar_class_init(GritsPluginRadarClass *klass) +{ + g_debug("GritsPluginRadar: class_init"); + GObjectClass *gobject_class = (GObjectClass*)klass; + gobject_class->dispose = grits_plugin_radar_dispose; + gobject_class->finalize = grits_plugin_radar_finalize; }