X-Git-Url: http://pileus.org/git/?p=grits;a=blobdiff_plain;f=src%2Fplugin-radar.c;h=18e92796c0e363cd4fbd579a50e2422011edb3d9;hp=46e73d4a0ee73b4e1f9ed35235a749be2a85f845;hb=1019fcfe2d612df24355f2597e661c28621b01c3;hpb=e778b406a21e31d79ca1cd599469bb4ddc3d6a1c diff --git a/src/plugin-radar.c b/src/plugin-radar.c index 46e73d4..18e9279 100644 --- a/src/plugin-radar.c +++ b/src/plugin-radar.c @@ -18,6 +18,7 @@ #include #include #include +#include #include #include #include @@ -26,37 +27,60 @@ #include "plugin-radar.h" #include "data.h" +static char *nexrad_base = "http://mesonet.agron.iastate.edu/data/"; + /**************** * GObject code * ****************/ +/* Plugin init */ static void aweather_radar_plugin_init(AWeatherPluginInterface *iface); -static void aweather_radar_expose(AWeatherPlugin *_radar); +static void _aweather_radar_expose(AWeatherPlugin *_radar); G_DEFINE_TYPE_WITH_CODE(AWeatherRadar, aweather_radar, G_TYPE_OBJECT, G_IMPLEMENT_INTERFACE(AWEATHER_TYPE_PLUGIN, aweather_radar_plugin_init)); -static void aweather_radar_class_init(AWeatherRadarClass *klass) -{ - GObjectClass *object_class = (GObjectClass*)klass; -} static void aweather_radar_plugin_init(AWeatherPluginInterface *iface) { + g_debug("AWeatherRadar: plugin_init"); /* Add methods to the interface */ - iface->expose = aweather_radar_expose; + iface->expose = _aweather_radar_expose; } +/* Class/Object init */ static void aweather_radar_init(AWeatherRadar *radar) { + g_debug("AWeatherRadar: class_init"); /* Set defaults */ - radar->gui = NULL; + radar->gui = NULL; + radar->soup = NULL; } +static void aweather_radar_dispose(GObject *gobject) +{ + g_debug("AWeatherRadar: dispose"); + AWeatherRadar *self = AWEATHER_RADAR(gobject); + /* Drop references */ + G_OBJECT_CLASS(aweather_radar_parent_class)->dispose(gobject); +} +static void aweather_radar_finalize(GObject *gobject) +{ + g_debug("AWeatherRadar: finalize"); + AWeatherRadar *self = AWEATHER_RADAR(gobject); + /* Free data */ + G_OBJECT_CLASS(aweather_radar_parent_class)->finalize(gobject); -/* TODO: User parameters or user data or something */ -static AWeatherRadar *self = NULL; +} +static void aweather_radar_class_init(AWeatherRadarClass *klass) +{ + g_debug("AWeatherRadar: class_init"); + GObjectClass *gobject_class = (GObjectClass*)klass; + gobject_class->dispose = aweather_radar_dispose; + gobject_class->finalize = aweather_radar_finalize; +} /************************** * Data loading functions * **************************/ /* Convert a sweep to an 2d array of data points */ -static void bscan_sweep(Sweep *sweep, guint8 **data, int *width, int *height) +static void bscan_sweep(AWeatherRadar *self, Sweep *sweep, colormap_t *colormap, + guint8 **data, int *width, int *height) { /* Calculate max number of bins */ int i, max_bins = 0; @@ -75,10 +99,10 @@ static void bscan_sweep(Sweep *sweep, guint8 **data, int *width, int *height) //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] = self->cur_colormap->data[val][0]; - buf[buf_i+1] = self->cur_colormap->data[val][1]; - buf[buf_i+2] = self->cur_colormap->data[val][2]; - buf[buf_i+3] = self->cur_colormap->data[val][3]; + 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]; if (val == BADVAL || val == RFVAL || val == APFLAG || val == NOTFOUND_H || val == NOTFOUND_V || val == NOECHO) { buf[buf_i+3] = 0x00; // transparent @@ -92,21 +116,14 @@ static void bscan_sweep(Sweep *sweep, guint8 **data, int *width, int *height) *data = buf; } -static void load_color_table(char *table) -{ - for (int i = 0; colormaps[i].name; i++) - if (g_str_equal(colormaps[i].name, table)) - self->cur_colormap = &colormaps[i]; -} - /* Load a sweep as the active texture */ -static void load_sweep(Sweep *sweep) +static void load_sweep(AWeatherRadar *self, Sweep *sweep) { aweather_gui_gl_begin(self->gui); self->cur_sweep = sweep; int height, width; guint8 *data; - bscan_sweep(sweep, &data, &width, &height); + 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); @@ -121,8 +138,17 @@ static void load_sweep(Sweep *sweep) aweather_gui_gl_end(self->gui); } +static void load_colormap(AWeatherRadar *self, gchar *table) +{ + /* 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]; +} + /* Add selectors to the config area for the sweeps */ -static void load_radar_gui(Radar *radar) +static void on_sweep_clicked(GtkRadioButton *button, gpointer _self); +static void load_radar_gui(AWeatherRadar *self, Radar *radar) { /* Clear existing items */ GtkWidget *child = gtk_bin_get_child(GTK_BIN(self->config_body)); @@ -132,7 +158,7 @@ static void load_radar_gui(Radar *radar) 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; + 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++) { @@ -152,7 +178,6 @@ static void load_radar_gui(Radar *radar) Sweep *sweep = vol->sweep[si]; if (sweep == NULL || sweep->h.elev == 0) continue; if (sweep->h.elev != elev) { - g_message("adding elev"); cols++; elev = sweep->h.elev; @@ -182,41 +207,30 @@ static void load_radar_gui(Radar *radar) //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_signal_connect_swapped(button, "clicked", - G_CALLBACK(load_color_table), vol->h.type_str); - g_signal_connect_swapped(button, "clicked", - G_CALLBACK(load_sweep), sweep); + + 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); } -/* Load a radar from a file */ -static void load_radar_rsl(GPid pid, gint status, gpointer _path) +/* Load a radar from a decompressed file */ +static void load_radar(AWeatherRadar *self, gchar *radar_file) { - gchar *path = _path; - if (status != 0) { - g_warning("wsr88ddec exited with status %d", status); - return; - } - char *dir = g_path_get_dirname(path); + char *dir = g_path_get_dirname(radar_file); char *site = g_path_get_basename(dir); g_free(dir); + g_debug("AWeatherRadar: load_radar - Loading new radar"); RSL_read_these_sweeps("all", NULL); - if (self->cur_radar) { - g_message("Freeing radar"); - RSL_free_radar(self->cur_radar); - } - g_message("Allocating radar"); - Radar *radar = self->cur_radar = RSL_wsr88d_to_radar(path, site); + 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", path, site); - g_free(path); + g_warning("fail to load radar: path=%s, site=%s", radar_file, site); g_free(site); return; } - g_free(path); g_free(site); /* Load the first sweep by default */ @@ -228,138 +242,233 @@ static void load_radar_rsl(GPid pid, gint status, gpointer _path) 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_color_table(radar->v[vi]->h.type_str); - load_sweep(radar->v[vi]->sweep[si]); + load_colormap(self, radar->v[vi]->h.type_str); + load_sweep(self, radar->v[vi]->sweep[si]); break; } break; } } - load_radar_gui(radar); + load_radar_gui(self, radar); } -/* decompress a radar file, then chain to the actuall loading function */ -static void load_radar(char *path, gboolean updated, gpointer user_data) +/* TODO: These update times functions are getting ugly... */ +static void update_times_gtk(AWeatherRadar *self, GList *times) { - char *raw = g_strconcat(path, ".raw", NULL); - if (!updated) { - load_radar_rsl(0, 0, raw); - } else { - g_message("File updated, decompressing.."); - char *argv[] = {"wsr88ddec", path, raw, 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); + gchar *last_time = NULL; + GRegex *regex = g_regex_new("^[A-Z]{4}_([0-9]{8}_[0-9]{4})$", 0, 0, NULL); // KLSX_20090622_2113 + GMatchInfo *info; + + GtkTreeView *tview = GTK_TREE_VIEW(aweather_gui_get_widget(self->gui, "time")); + GtkListStore *lstore = GTK_LIST_STORE(gtk_tree_view_get_model(tview)); + gtk_list_store_clear(lstore); + GtkTreeIter iter; + times = g_list_reverse(times); + for (GList *cur = times; cur; cur = cur->next) { + g_message("trying time %s", (gchar*)cur->data); + if (g_regex_match(regex, cur->data, 0, &info)) { + gchar *time = g_match_info_fetch(info, 1); + g_message("adding time %s", (gchar*)cur->data); + gtk_list_store_insert(lstore, &iter, 0); + gtk_list_store_set(lstore, &iter, 0, time, -1); + last_time = time; } - g_child_watch_add(pid, load_radar_rsl, raw); } -} -static void update_times(char *site, char **last_time) -{ - char *list_uri = g_strdup_printf( - "http://mesonet.agron.iastate.edu/data/nexrd2/raw/K%s/dir.list", - site); - GFile *list = g_file_new_for_uri(list_uri); - g_free(list_uri); + AWeatherView *view = aweather_gui_get_view(self->gui); + aweather_view_set_time(view, last_time); + g_regex_unref(regex); + g_list_foreach(times, (GFunc)g_free, NULL); + g_list_free(times); +} +static void update_times_online_cb(char *path, gboolean updated, gpointer _self) +{ + GList *times = NULL; gchar *data; gsize length; - GError *error = NULL; - g_file_load_contents(list, NULL, &data, &length, NULL, &error); - g_object_unref(list); - if (error) { - g_warning("Error loading list for %s: %s", site, error->message); - g_error_free(error); - return; - } + g_file_get_contents(path, &data, &length, NULL); gchar **lines = g_strsplit(data, "\n", -1); - GtkTreeView *tview = GTK_TREE_VIEW(aweather_gui_get_widget(self->gui, "time")); - GtkListStore *lstore = GTK_LIST_STORE(gtk_tree_view_get_model(tview)); - gtk_list_store_clear(lstore); - GtkTreeIter iter; for (int i = 0; lines[i] && lines[i][0]; i++) { - // format: `841907 KABR_20090510_0159' - //g_message("\tadding %p [%s]", lines[i], lines[i]); char **parts = g_strsplit(lines[i], " ", 2); - char *time = parts[1]+5; - gtk_list_store_insert(lstore, &iter, 0); - gtk_list_store_set(lstore, &iter, 0, time, -1); + times = g_list_prepend(times, g_strdup(parts[1])); g_strfreev(parts); } + g_strfreev(lines); + g_free(data); + + update_times_gtk(_self, times); +} +static void update_times(AWeatherRadar *self, AWeatherView *view, char *site) +{ + if (aweather_view_get_offline(view)) { + GList *times = NULL; + gchar *path = g_build_filename(g_get_user_cache_dir(), PACKAGE, "nexrd2", "raw", site, NULL); + GDir *dir = g_dir_open(path, 0, NULL); + if (dir) { + const gchar *name; + while ((name = g_dir_read_name(dir))) { + times = g_list_prepend(times, g_strdup(name)); + } + g_dir_close(dir); + } + g_free(path); + update_times_gtk(self, times); + } else { + gchar *path = g_strdup_printf("nexrd2/raw/%s/dir.list", site); + cache_file(nexrad_base, path, AWEATHER_REFRESH, NULL, update_times_online_cb, self); + /* update_times_gtk from update_times_online_cb */ + } +} - if (last_time) - gtk_tree_model_get(GTK_TREE_MODEL(lstore), &iter, 0, last_time, -1); +/***************** + * ASync helpers * + *****************/ +typedef struct { + AWeatherRadar *self; + gchar *radar_file; +} decompressed_t; - g_free(data); - g_strfreev(lines); +static void decompressed_cb(GPid pid, gint status, gpointer _udata) +{ + 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); +} + +static void cache_chunk_cb(char *path, goffset cur, goffset total, gpointer _self) +{ + AWeatherRadar *self = AWEATHER_RADAR(_self); + double percent = (double)cur/total; + + g_message("AWeatherRadar: cache_chunk_cb - %lld/%lld = %.2f%%", + cur, total, percent*100); + + gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(self->progress_bar), MIN(percent, 1.0)); + + 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); +} + +static void cache_done_cb(char *path, gboolean updated, gpointer _self) +{ + AWeatherRadar *self = AWEATHER_RADAR(_self); + char *decompressed = g_strconcat(path, ".raw", NULL); + if (!updated) { + load_radar(self, decompressed); + return; + } + + decompressed_t *udata = g_malloc(sizeof(decompressed_t)); + udata->self = self; + udata->radar_file = decompressed; + g_debug("AWeatherRadar: 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); + } + g_child_watch_add(pid, decompressed_cb, udata); + self->soup = NULL; } /************* * Callbacks * *************/ -static void on_time_changed(AWeatherView *view, char *time, gpointer user_data) +static void on_sweep_clicked(GtkRadioButton *button, gpointer _self) +{ + AWeatherRadar *self = AWEATHER_RADAR(_self); + load_colormap(self, g_object_get_data(G_OBJECT(button), "type" )); + load_sweep (self, g_object_get_data(G_OBJECT(button), "sweep")); +} + +static void on_time_changed(AWeatherView *view, const char *time, gpointer _self) { - g_message("radar:setting time"); + AWeatherRadar *self = AWEATHER_RADAR(_self); + g_debug("AWeatherRadar: on_time_changed - setting time=%s", time); // format: http://mesonet.agron.iastate.edu/data/nexrd2/raw/KABR/KABR_20090510_0323 char *site = aweather_view_get_site(view); - char *base = "http://mesonet.agron.iastate.edu/data/"; - char *path = g_strdup_printf("nexrd2/raw/K%s/K%s_%s", site, site, time); + 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; // Clear radar + self->cur_sweep = NULL; aweather_gui_gl_redraw(self->gui); - cache_file(base, path, AWEATHER_AUTOMATIC, load_radar, NULL); + /* Start loading the new radar */ + if (self->soup) { + soup_session_abort(self->soup); + self->soup = NULL; + } + if (aweather_view_get_offline(view)) + self->soup = cache_file(nexrad_base, path, AWEATHER_ONCE, + cache_chunk_cb, cache_done_cb, self); + else + self->soup = cache_file(nexrad_base, path, AWEATHER_UPDATE, + cache_chunk_cb, cache_done_cb, self); g_free(path); } -static void on_site_changed(AWeatherView *view, char *site, gpointer user_data) +static void on_site_changed(AWeatherView *view, char *site, gpointer _self) { - g_message("Loading wsr88d list for %s", site); - char *time = NULL; - update_times(site, &time); - aweather_view_set_time(view, time); - - g_free(time); + AWeatherRadar *self = AWEATHER_RADAR(_self); + g_debug("AWeatherRadar: on_site_changed - Loading wsr88d list for %s", site); + update_times(self, view, site), &time; } -static void on_refresh(AWeatherView *view, gpointer user_data) +static void on_refresh(AWeatherView *view, gpointer _self) { + AWeatherRadar *self = AWEATHER_RADAR(_self); char *site = aweather_view_get_site(view); - char *time = NULL; - update_times(site, &time); - aweather_view_set_time(view, time); - g_free(time); + update_times(self, view, site); } - - /*********** * Methods * ***********/ AWeatherRadar *aweather_radar_new(AWeatherGui *gui) { - //g_message("aweather_view_new"); - AWeatherRadar *radar = g_object_new(AWEATHER_TYPE_RADAR, NULL); - radar->gui = gui; - - self = radar; + g_debug("AWeatherRadar: new"); + AWeatherRadar *self = g_object_new(AWEATHER_TYPE_RADAR, NULL); + self->gui = gui; GtkWidget *config = aweather_gui_get_widget(gui, "tabs"); AWeatherView *view = aweather_gui_get_view(gui); @@ -371,17 +480,17 @@ AWeatherRadar *aweather_radar_new(AWeatherGui *gui) gtk_notebook_prepend_page(GTK_NOTEBOOK(config), self->config_body, gtk_label_new("Radar")); /* Set up OpenGL Stuff */ - g_signal_connect(view, "site-changed", G_CALLBACK(on_site_changed), NULL); - g_signal_connect(view, "time-changed", G_CALLBACK(on_time_changed), NULL); - g_signal_connect(view, "refresh", G_CALLBACK(on_refresh), NULL); + g_signal_connect(view, "site-changed", G_CALLBACK(on_site_changed), self); + g_signal_connect(view, "time-changed", G_CALLBACK(on_time_changed), self); + g_signal_connect(view, "refresh", G_CALLBACK(on_refresh), self); - return radar; + return self; } -static void aweather_radar_expose(AWeatherPlugin *_radar) +static void _aweather_radar_expose(AWeatherPlugin *_self) { - AWeatherRadar *radar = AWEATHER_RADAR(_radar); - g_message("radar:expose"); + AWeatherRadar *self = AWEATHER_RADAR(_self); + g_debug("AWeatherRadar: expose"); if (self->cur_sweep == NULL) return; Sweep *sweep = self->cur_sweep;