2 * Copyright (C) 2009-2010 Andy Spencer <andy753421@gmail.com>
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include <glib/gstdio.h>
23 #include <gtk/gtkgl.h>
33 #include "../aweather-location.h"
35 void _gtk_bin_set_child(GtkBin *bin, GtkWidget *new)
37 GtkWidget *old = gtk_bin_get_child(bin);
39 gtk_widget_destroy(old);
40 gtk_container_add(GTK_CONTAINER(bin), new);
41 gtk_widget_show_all(new);
44 static gchar *_find_nearest(time_t time, GList *files,
45 gsize offset, gchar *format)
47 g_debug("GisPluginRadar: _find_nearest");
48 time_t nearest_time = 0;
49 char *nearest_file = NULL;
52 for (GList *cur = files; cur; cur = cur->next) {
53 gchar *file = cur->data;
54 g_message("file=%s", file);
55 strptime(file+offset, format, &tm);
56 if (ABS(time - mktime(&tm)) <
57 ABS(time - nearest_time)) {
59 nearest_time = mktime(&tm);
63 g_debug("GisPluginRadar: _find_nearest = %s", nearest_file);
65 return g_strdup(nearest_file);
81 gchar *code; // Site name. e.g. KLSX
82 gchar *name; // Site name. e.g. St. Louis
83 GisPoint pos; // LLE positions of antena
84 GisMarker *marker; // Map marker for libgis
86 /* Stuff from the parents */
93 RadarSiteStatus status; // Loading status for the site
95 AWeatherLevel2 *level2; // The Level2 structure for the current volume
96 gpointer level2_ref; // GisViewer reference to the added radar
99 time_t time; // Current timestamp of the level2
100 gchar *message; // Error message set while updating
101 guint time_id; // "time-changed" callback ID
102 guint refresh_id; // "refresh" callback ID
103 guint location_id; // "locaiton-changed" callback ID
106 /* format: http://mesonet.agron.iastate.edu/data/nexrd2/raw/KABR/KABR_20090510_0323 */
107 void _site_update_loading(gchar *file, goffset cur,
108 goffset total, gpointer _site)
110 RadarSite *site = _site;
111 GtkWidget *progress_bar = gtk_bin_get_child(GTK_BIN(site->config));
112 double percent = (double)cur/total;
113 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(progress_bar), MIN(percent, 1.0));
114 gchar *msg = g_strdup_printf("Loading... %5.1f%% (%.2f/%.2f MB)",
115 percent*100, (double)cur/1000000, (double)total/1000000);
116 gtk_progress_bar_set_text(GTK_PROGRESS_BAR(progress_bar), msg);
119 gboolean _site_update_end(gpointer _site)
121 RadarSite *site = _site;
123 g_warning("GisPluginRadar: _update_end - %s", site->message);
124 _gtk_bin_set_child(GTK_BIN(site->config), gtk_label_new(site->message));
126 _gtk_bin_set_child(GTK_BIN(site->config),
127 aweather_level2_get_config(site->level2));
129 site->status = STATUS_LOADED;
132 gpointer _site_update_thread(gpointer _site)
134 RadarSite *site = _site;
135 g_debug("GisPluginRadar: _update - %s", site->code);
136 site->status = STATUS_LOADING;
137 site->message = NULL;
139 gboolean offline = gis_viewer_get_offline(site->viewer);
140 gchar *nexrad_url = gis_prefs_get_string(site->prefs,
141 "aweather/nexrad_url", NULL);
143 /* Remove old volume */
144 g_debug("GisPluginRadar: _update - remove - %s", site->code);
145 if (site->level2_ref) {
146 gis_viewer_remove(site->viewer, site->level2_ref);
147 site->level2_ref = NULL;
150 /* Find nearest volume (temporally) */
151 g_debug("GisPluginRadar: _update - find nearest - %s", site->code);
152 gchar *dir_list = g_strconcat(nexrad_url, "/", site->code,
153 "/", "dir.list", NULL);
154 GList *files = gis_http_available(site->http,
155 "^K\\w{3}_\\d{8}_\\d{4}$", site->code,
156 "\\d+ (.*)", (offline ? NULL : dir_list));
158 gchar *nearest = _find_nearest(site->time, files, 5, "%Y%m%d_%H%M");
159 g_list_foreach(files, (GFunc)g_free, NULL);
162 site->message = "No suitable files found";
166 /* Fetch new volume */
167 g_debug("GisPluginRadar: _update - fetch");
168 gchar *local = g_strconcat(site->code, "/", nearest, NULL);
169 gchar *uri = g_strconcat(nexrad_url, "/", local, NULL);
170 gchar *file = gis_http_fetch(site->http, uri, local,
171 offline ? GIS_LOCAL : GIS_UPDATE,
172 _site_update_loading, site);
176 site->message = "Fetch failed";
180 /* Load and add new volume */
181 g_debug("GisPluginRadar: _update - load - %s", site->code);
182 site->level2 = aweather_level2_new_from_file(
183 site->viewer, colormaps, file, site->code);
185 site->message = "Load failed";
188 site->level2_ref = gis_viewer_add(site->viewer,
189 GIS_OBJECT(site->level2), GIS_LEVEL_WORLD, TRUE);
192 g_idle_add(_site_update_end, site);
195 void _site_update(RadarSite *site)
197 site->time = gis_viewer_get_time(site->viewer);
198 g_debug("GisPluginRadar: _on_time_changed %s - %d",
199 site->code, (gint)site->time);
201 /* Add a progress bar */
202 GtkWidget *progress = gtk_progress_bar_new();
203 gtk_progress_bar_set_text(GTK_PROGRESS_BAR(progress), "Loading...");
204 _gtk_bin_set_child(GTK_BIN(site->config), progress);
206 /* Fork loading right away so updating the
207 * list of times doesn't take too long */
208 g_thread_create(_site_update_thread, site, FALSE, NULL);
211 /* RadarSite methods */
212 void radar_site_unload(RadarSite *site)
214 g_debug("GisPluginRadar: radar_site_unload %s", site->code);
216 if (site->status == STATUS_LOADING)
217 return; // Abort if it's still loading
219 g_signal_handler_disconnect(site->viewer, site->time_id);
220 g_signal_handler_disconnect(site->viewer, site->refresh_id);
223 gtk_widget_destroy(site->config);
226 if (site->level2_ref) {
227 gis_viewer_remove(site->viewer, site->level2_ref);
228 site->level2_ref = NULL;
231 site->status = STATUS_UNLOADED;
234 void radar_site_load(RadarSite *site)
236 g_debug("GisPluginRadar: radar_site_load %s", site->code);
237 site->status = STATUS_LOADING;
240 site->config = gtk_alignment_new(0, 0, 1, 1);
241 GtkWidget *tab = gtk_hbox_new(FALSE, 0);
242 GtkWidget *close = gtk_button_new();
243 GtkWidget *label = gtk_label_new(site->name);
244 gtk_container_add(GTK_CONTAINER(close),
245 gtk_image_new_from_stock(GTK_STOCK_CLOSE,
246 GTK_ICON_SIZE_MENU));
247 gtk_button_set_relief(GTK_BUTTON(close), GTK_RELIEF_NONE);
248 g_signal_connect_swapped(close, "clicked",
249 G_CALLBACK(radar_site_unload), site);
250 gtk_box_pack_start(GTK_BOX(tab), label, TRUE, TRUE, 0);
251 gtk_box_pack_end(GTK_BOX(tab), close, FALSE, FALSE, 0);
252 gtk_notebook_append_page(GTK_NOTEBOOK(site->pconfig),
254 gtk_widget_show_all(site->config);
255 gtk_widget_show_all(tab);
257 /* Set up radar loading */
258 site->time_id = g_signal_connect_swapped(site->viewer, "time-changed",
259 G_CALLBACK(_site_update), site);
260 site->refresh_id = g_signal_connect_swapped(site->viewer, "refresh",
261 G_CALLBACK(_site_update), site);
265 void _site_on_location_changed(GisViewer *viewer,
266 gdouble lat, gdouble lon, gdouble elev,
269 static gdouble min_dist = EARTH_R / 20;
270 RadarSite *site = _site;
272 /* Calculate distance, could cache xyz values */
273 gdouble eye_xyz[3], site_xyz[3];
274 lle2xyz(lat, lon, elev, &eye_xyz[0], &eye_xyz[1], &eye_xyz[2]);
275 lle2xyz(site->pos.lat, site->pos.lon, site->pos.elev,
276 &site_xyz[0], &site_xyz[1], &site_xyz[2]);
277 gdouble dist = distd(site_xyz, eye_xyz);
279 /* Load or unload the site if necessasairy */
280 if (dist <= min_dist && dist < elev*1.25 && site->status == STATUS_UNLOADED)
281 radar_site_load(site);
282 else if (dist > 2*min_dist && site->status != STATUS_UNLOADED)
283 radar_site_unload(site);
286 RadarSite *radar_site_new(city_t *city, GtkWidget *pconfig,
287 GisViewer *viewer, GisPrefs *prefs, GisHttp *http)
289 RadarSite *site = g_new0(RadarSite, 1);
290 site->viewer = g_object_ref(viewer);
291 site->prefs = g_object_ref(prefs);
293 site->code = g_strdup(city->code);
294 site->name = g_strdup(city->name);
295 site->pos = city->pos;
296 site->pconfig = pconfig;
299 site->marker = gis_marker_new(city->name);
300 GIS_OBJECT(site->marker)->center = site->pos;
301 GIS_OBJECT(site->marker)->lod = EARTH_R*city->lod;
302 gis_viewer_add(site->viewer, GIS_OBJECT(site->marker),
303 GIS_LEVEL_OVERLAY, FALSE);
305 /* Connect signals */
307 g_signal_connect(viewer, "location-changed",
308 G_CALLBACK(_site_on_location_changed), site);
312 void radar_site_free(RadarSite *site)
314 radar_site_unload(site);
316 g_object_unref(site->viewer);
325 static void _draw_hud(GisCallback *callback, gpointer _self)
328 GisPluginRadar *self = GIS_PLUGIN_RADAR(_self);
332 g_debug("GisPluginRadar: _draw_hud");
333 /* Print the color table */
334 glMatrixMode(GL_MODELVIEW ); glLoadIdentity();
335 glMatrixMode(GL_PROJECTION); glLoadIdentity();
336 glDisable(GL_TEXTURE_2D);
337 glDisable(GL_ALPHA_TEST);
338 glDisable(GL_CULL_FACE);
339 glDisable(GL_LIGHTING);
340 glEnable(GL_COLOR_MATERIAL);
343 for (i = 0; i < 256; i++) {
344 glColor4ubv(self->colormap->data[i]);
345 glVertex3f(-1.0, (float)((i ) - 256/2)/(256/2), 0.0); // bot left
346 glVertex3f(-1.0, (float)((i+1) - 256/2)/(256/2), 0.0); // top left
347 glVertex3f(-0.9, (float)((i+1) - 256/2)/(256/2), 0.0); // top right
348 glVertex3f(-0.9, (float)((i ) - 256/2)/(256/2), 0.0); // bot right
354 GisPluginRadar *gis_plugin_radar_new(GisViewer *viewer, GisPrefs *prefs)
356 /* TODO: move to constructor if possible */
357 g_debug("GisPluginRadar: new");
358 GisPluginRadar *self = g_object_new(GIS_TYPE_PLUGIN_RADAR, NULL);
359 self->viewer = viewer;
363 GisCallback *hud_cb = gis_callback_new(_draw_hud, self);
364 gis_viewer_add(viewer, GIS_OBJECT(hud_cb), GIS_LEVEL_HUD, FALSE);
366 /* Load radar sites */
367 for (city_t *city = cities; city->type; city++) {
368 if (city->type != LOCATION_CITY)
370 RadarSite *site = radar_site_new(city, self->config,
371 self->viewer, self->prefs, self->sites_http);
372 g_hash_table_insert(self->sites, city->code, site);
378 static GtkWidget *gis_plugin_radar_get_config(GisPlugin *_self)
380 GisPluginRadar *self = GIS_PLUGIN_RADAR(_self);
385 static void gis_plugin_radar_plugin_init(GisPluginInterface *iface);
386 G_DEFINE_TYPE_WITH_CODE(GisPluginRadar, gis_plugin_radar, G_TYPE_OBJECT,
387 G_IMPLEMENT_INTERFACE(GIS_TYPE_PLUGIN,
388 gis_plugin_radar_plugin_init));
389 static void gis_plugin_radar_plugin_init(GisPluginInterface *iface)
391 g_debug("GisPluginRadar: plugin_init");
392 /* Add methods to the interface */
393 iface->get_config = gis_plugin_radar_get_config;
395 static void gis_plugin_radar_init(GisPluginRadar *self)
397 g_debug("GisPluginRadar: class_init");
399 self->sites_http = gis_http_new(G_DIR_SEPARATOR_S "nexrad" G_DIR_SEPARATOR_S "level2" G_DIR_SEPARATOR_S);
400 self->sites = g_hash_table_new(g_str_hash, g_str_equal);
401 self->config = gtk_notebook_new();
402 gtk_notebook_set_tab_pos(GTK_NOTEBOOK(self->config), GTK_POS_LEFT);
404 static void gis_plugin_radar_dispose(GObject *gobject)
406 g_debug("GisPluginRadar: dispose");
407 GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject);
408 /* Drop references */
409 G_OBJECT_CLASS(gis_plugin_radar_parent_class)->dispose(gobject);
411 static void gis_plugin_radar_finalize(GObject *gobject)
413 g_debug("GisPluginRadar: finalize");
414 GisPluginRadar *self = GIS_PLUGIN_RADAR(gobject);
416 gis_http_free(self->sites_http);
417 g_hash_table_destroy(self->sites);
418 gtk_widget_destroy(self->config);
419 G_OBJECT_CLASS(gis_plugin_radar_parent_class)->finalize(gobject);
422 static void gis_plugin_radar_class_init(GisPluginRadarClass *klass)
424 g_debug("GisPluginRadar: class_init");
425 GObjectClass *gobject_class = (GObjectClass*)klass;
426 gobject_class->dispose = gis_plugin_radar_dispose;
427 gobject_class->finalize = gis_plugin_radar_finalize;