]> Pileus Git - aweather/blobdiff - src/plugin-radar.c
* Fixing some async issue (w/ gthread_init)
[aweather] / src / plugin-radar.c
index c1e122ef4153d1e2672d5c8fe744a05c5188bf54..07161c56af8b762f031b6389db114fd122162fc0 100644 (file)
 #include <gtk/gtkgl.h>
 #include <GL/gl.h>
 #include <math.h>
-
-#include "rsl.h"
+#include <rsl.h>
 
 #include "aweather-gui.h"
 #include "plugin-radar.h"
 #include "data.h"
 
-GtkWidget *drawing;
-GtkWidget *config_body;
-static Sweep *cur_sweep = NULL;  // make this not global
-//static int nred, ngreen, nblue;
-//static char red[256], green[256], blue[256];
-static colormap_t *colormap;
-static guint sweep_tex = 0;
-static Radar *radar = NULL;
-
-static AWeatherGui *gui = NULL;
+/****************
+ * GObject code *
+ ****************/
+static void aweather_radar_plugin_init(AWeatherPluginInterface *iface);
+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)
+{
+       /* Add methods to the interface */
+       iface->expose = _aweather_radar_expose;
+}
+static void aweather_radar_init(AWeatherRadar *radar)
+{
+       /* Set defaults */
+       radar->gui = NULL;
+}
 
 /**************************
  * 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;
@@ -78,24 +90,17 @@ 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))
-                       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(gui);
-       cur_sweep = sweep;
+       aweather_gui_gl_begin(self->gui);
+       self->cur_sweep = sweep;
        int height, width;
        guint8 *data;
-       bscan_sweep(sweep, &data, &width, &height);
-       glDeleteTextures(1, &sweep_tex);
-       glGenTextures(1, &sweep_tex);
-       glBindTexture(GL_TEXTURE_2D, sweep_tex);
+       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);
@@ -103,15 +108,24 @@ static void load_sweep(Sweep *sweep)
        glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, width, height, 0,
                        GL_RGBA, GL_UNSIGNED_BYTE, data);
        g_free(data);
-       aweather_gui_gl_redraw(gui);
-       aweather_gui_gl_end(gui);
+       aweather_gui_gl_redraw(self->gui);
+       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(config_body));
+       GtkWidget *child = gtk_bin_get_child(GTK_BIN(self->config_body));
        if (child)
                gtk_widget_destroy(child);
 
@@ -138,7 +152,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;
 
@@ -168,41 +181,34 @@ 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(config_body), table);
+       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);
        RSL_read_these_sweeps("all", NULL);
-       if (radar) {
-               g_message("Freeing radar");
-               RSL_free_radar(radar);
+       if (self->cur_radar) {
+               g_debug("AWeatherRadar: load_radar - Freeing old radar");
+               RSL_free_radar(self->cur_radar);
        }
-       g_message("Allocating radar");
-       radar = RSL_wsr88d_to_radar(path, site);
+       g_debug("AWeatherRadar: load_radar - Loading new radar");
+       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 */
@@ -214,49 +220,18 @@ 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);
-}
-
-/* decompress a radar file, then chain to the actuall loading function */
-static void load_radar(char *path, gboolean updated, gpointer user_data)
-{
-       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);
-               }
-               g_child_watch_add(pid, load_radar_rsl, raw);
-       }
+       load_radar_gui(self, radar);
 }
 
-static void update_times(char *site, char **last_time)
+static void update_times(AWeatherRadar *self, char *site, char **last_time)
 {
        char *list_uri = g_strdup_printf(
                        "http://mesonet.agron.iastate.edu/data/nexrd2/raw/K%s/dir.list",
@@ -275,7 +250,7 @@ static void update_times(char *site, char **last_time)
                return;
        }
        gchar **lines = g_strsplit(data, "\n", -1);
-       GtkTreeView  *tview  = GTK_TREE_VIEW(aweather_gui_get_widget(gui, "time"));
+       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;
@@ -296,21 +271,149 @@ static void update_times(char *site, char **last_time)
        g_strfreev(lines);
 }
 
+/*****************
+ * ASync helpers *
+ *****************/
+typedef struct {
+       AWeatherRadar *self;
+       gchar *radar_file;
+} decompressed_t;
+
+static void decompressed_cb(GPid pid, gint status, gpointer _self)
+{
+       decompressed_t *udata = _self;
+       if (status != 0) {
+               g_warning("wsr88ddec exited with status %d", status);
+               return;
+       }
+       // TODO: pass cur_file as params? 
+       load_radar(udata->self, udata->radar_file);
+       g_free(udata->radar_file);
+       g_free(udata);
+}
+
+static void cached_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: cached_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);
+}
+
 /*************
  * Callbacks *
  *************/
-static gboolean on_expose(GtkWidget *da, GdkEventExpose *event, gpointer user_data)
+static void on_sweep_clicked(GtkRadioButton *button, gpointer _self)
 {
-       g_message("radar:expose");
-       if (cur_sweep == NULL)
-               return FALSE;
-       Sweep *sweep = cur_sweep;
+       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, char *time, gpointer _self)
+{
+       AWeatherRadar *self = AWEATHER_RADAR(_self);
+       g_debug("AWeatherRadar: on_time_changed - setting 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);
+
+       self->cur_radar = NULL;
+       self->cur_sweep = NULL; // Clear radar
+       aweather_gui_gl_redraw(self->gui);
+
+       cache_file(base, path, AWEATHER_AUTOMATIC, cached_cb, self);
+       g_free(path);
+}
+
+static void on_site_changed(AWeatherView *view, char *site, gpointer _self)
+{
+       AWeatherRadar *self = AWEATHER_RADAR(_self);
+       g_debug("AWeatherRadar: on_site_changed - Loading wsr88d list for %s", site);
+       char *time = NULL;
+       update_times(self, site, &time);
+       aweather_view_set_time(view, time);
+
+       g_free(time);
+}
+
+static void on_refresh(AWeatherView *view, gpointer user_data, gpointer _self)
+{
+       AWeatherRadar *self = AWEATHER_RADAR(_self);
+       char *site = aweather_view_get_site(view);
+       char *time = NULL;
+       update_times(self, site, &time);
+       aweather_view_set_time(view, time);
+       g_free(time);
+}
+
+/***********
+ * Methods *
+ ***********/
+AWeatherRadar *aweather_radar_new(AWeatherGui *gui)
+{
+       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);
+
+       /* Add configuration tab */
+       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"));
+       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), self);
+       g_signal_connect(view,    "time-changed", G_CALLBACK(on_time_changed), self);
+       g_signal_connect(view,    "refresh",      G_CALLBACK(on_refresh),      self);
+
+       return self;
+}
+
+static void _aweather_radar_expose(AWeatherPlugin *_self)
+{
+       AWeatherRadar *self = AWEATHER_RADAR(_self);
+       g_debug("AWeatherRadar: expose");
+       if (self->cur_sweep == NULL)
+               return;
+       Sweep *sweep = self->cur_sweep;
 
        /* Draw the rays */
 
        glMatrixMode(GL_MODELVIEW);
        glPushMatrix();
-       glBindTexture(GL_TEXTURE_2D, sweep_tex);
+       glBindTexture(GL_TEXTURE_2D, self->cur_sweep_tex);
        glEnable(GL_TEXTURE_2D);
        glDisable(GL_ALPHA_TEST);
        glColor4f(1,1,1,1);
@@ -362,10 +465,10 @@ static gboolean on_expose(GtkWidget *da, GdkEventExpose *event, gpointer user_da
        glBegin(GL_QUADS);
        int i;
        for (i = 0; i < 256; i++) {
-               glColor4ub(colormap->data[i][0],
-                          colormap->data[i][1],
-                          colormap->data[i][2],
-                          colormap->data[i][3]);
+               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
@@ -376,63 +479,4 @@ static gboolean on_expose(GtkWidget *da, GdkEventExpose *event, gpointer user_da
        glEnable(GL_ALPHA_TEST);
         glMatrixMode(GL_PROJECTION); glPopMatrix(); 
        glMatrixMode(GL_MODELVIEW ); glPopMatrix();
-       return FALSE;
-}
-
-static void on_time_changed(AWeatherView *view, char *time, gpointer user_data)
-{
-       g_message("radar:setting 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);
-
-       radar = NULL;
-       cur_sweep = NULL; // Clear radar
-       aweather_gui_gl_redraw(gui);
-
-       cache_file(base, path, AWEATHER_AUTOMATIC, load_radar, NULL);
-       g_free(path);
-}
-
-static void on_site_changed(AWeatherView *view, char *site, gpointer user_data)
-{
-       g_message("Loading wsr88d list for %s", site);
-       char *time = NULL;
-       update_times(site, &time);
-       aweather_view_set_time(view, time);
-
-       g_free(time);
-}
-
-static void on_refresh(AWeatherView *view, gpointer user_data)
-{
-       char *site = aweather_view_get_site(view);
-       char *time = NULL;
-       update_times(site, &time);
-       aweather_view_set_time(view, time);
-       g_free(time);
-}
-
-/* Init */
-gboolean radar_init(AWeatherGui *_gui)
-{
-       gui = _gui;
-       drawing = aweather_gui_get_widget(gui, "drawing");
-       GtkWidget    *config  = aweather_gui_get_widget(gui, "tabs");
-       AWeatherView *view    = aweather_gui_get_view(gui);
-
-       /* Add configuration tab */
-       config_body = gtk_alignment_new(0, 0, 1, 1);
-       gtk_container_set_border_width(GTK_CONTAINER(config_body), 5);
-       gtk_container_add(GTK_CONTAINER(config_body), gtk_label_new("No radar loaded"));
-       gtk_notebook_prepend_page(GTK_NOTEBOOK(config), config_body, gtk_label_new("Radar"));
-
-       /* Set up OpenGL Stuff */
-       g_signal_connect(drawing, "expose-event", G_CALLBACK(on_expose),       NULL);
-       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);
-
-       return TRUE;
 }