src/geonamessearch.c

   1 /*
   2  * viking -- GPS Data and Topo Analyzer, Explorer, and Manager
   3  *
   4  * Copyright (C) 2009, Hein Ragas
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24 #include <stdlib.h>
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <glib.h>
  28 #include <glib/gstdio.h>
  29 #include <glib/gprintf.h>
  30 #include <glib/gi18n.h>
  31
  32 #include "viking.h"
  33 #include "util.h"
  34 #include "curl_download.h"
  35
  36 #define GEONAMES_WIKIPEDIA_URL_FMT "http://ws.geonames.org/wikipediaBoundingBoxJSON?formatted=true&north=%s&south=%s&east=%s&west=%s"
  37 #define GEONAMES_COUNTRY_PATTERN "\"countryName\": \""
  38 #define GEONAMES_LONGITUDE_PATTERN "\"lng\": "
  39 #define GEONAMES_NAME_PATTERN "\"name\": \""
  40 #define GEONAMES_LATITUDE_PATTERN "\"lat\": "
  41 #define GEONAMES_TITLE_PATTERN "\"title\": \""
  42 #define GEONAMES_WIKIPEDIAURL_PATTERN "\"wikipediaUrl\": \""
  43 #define GEONAMES_THUMBNAILIMG_PATTERN "\"thumbnailImg\": \""
  44 #define GEONAMES_SEARCH_NOT_FOUND "not understand the location"
  45
  46 /* found_geoname: Type to contain data returned from GeoNames.org */
  47
  48 typedef struct {
  49   gchar *name;
  50   gchar *country;
  51   struct LatLon ll;
  52   gchar *desc;
  53 } found_geoname;
  54
  55 found_geoname *new_found_geoname()
  56 {
  57   found_geoname *ret;
  58
  59   ret = (found_geoname *)g_malloc(sizeof(found_geoname));
  60   ret->name = NULL;
  61   ret->country = NULL;
  62   ret->desc = NULL;
  63   ret->ll.lat = 0.0;
  64   ret->ll.lon = 0.0;
  65   return(ret);
  66 }
  67
  68 found_geoname *copy_found_geoname(found_geoname *src)
  69 {
  70   found_geoname *dest = new_found_geoname();
  71   dest->name = g_strdup(src->name);
  72   dest->country = g_strdup(src->country);
  73   dest->ll.lat = src->ll.lat;
  74   dest->ll.lon = src->ll.lon;
  75   dest->desc = g_strdup(src->desc);
  76   return(dest);
  77 }
  78
  79 static void free_list_geonames(found_geoname *geoname, gpointer userdata)
  80 {
  81   g_free(geoname->name);
  82   g_free(geoname->country);
  83   g_free(geoname->desc);
  84 }
  85
  86 void free_geoname_list(GList *found_places)
  87 {
  88   g_list_foreach(found_places, (GFunc)free_list_geonames, NULL);
  89   g_list_free(found_places);
  90 }
  91
  92 static void none_found(VikWindow *vw)
  93 {
  94   GtkWidget *dialog = NULL;
  95
  96   dialog = gtk_dialog_new_with_buttons ( "", GTK_WINDOW(vw), 0, GTK_STOCK_OK, GTK_RESPONSE_ACCEPT, NULL );
  97   gtk_window_set_title(GTK_WINDOW(dialog), _("Search"));
  98
  99   GtkWidget *search_label = gtk_label_new(_("No entries found!"));
 100   gtk_box_pack_start ( GTK_BOX(GTK_DIALOG(dialog)->vbox), search_label, FALSE, FALSE, 5 );
 101   gtk_widget_show_all(dialog);
 102
 103   gtk_dialog_run ( GTK_DIALOG(dialog) );
 104   gtk_widget_destroy(dialog);
 105 }
 106
 107 void buttonToggled(GtkCellRendererToggle* renderer, gchar* pathStr, gpointer data)
 108 {
 109    GtkTreeIter iter;
 110    gboolean enabled;
 111    GtkTreePath* path = gtk_tree_path_new_from_string(pathStr);
 112    gtk_tree_model_get_iter(GTK_TREE_MODEL (data), &iter, path);
 113    gtk_tree_model_get(GTK_TREE_MODEL (data), &iter, 0, &enabled, -1);
 114    enabled = !enabled;
 115    gtk_tree_store_set(GTK_TREE_STORE (data), &iter, 0, enabled, -1);
 116 }
 117
 118 GList *a_select_geoname_from_list(GtkWindow *parent, GList *geonames, gboolean multiple_selection_allowed, const gchar *title, const gchar *msg)
 119 {
 120   GtkTreeIter iter;
 121   GtkCellRenderer *renderer;
 122   GtkCellRenderer *toggle_render;
 123   GtkWidget *view;
 124   found_geoname *geoname;
 125   gchar *latlon_string;
 126   int column_runner;
 127   gboolean checked;
 128   gboolean to_copy;
 129
 130   GtkWidget *dialog = gtk_dialog_new_with_buttons (title,
 131                                                   parent,
 132                                                   GTK_DIALOG_MODAL | GTK_DIALOG_DESTROY_WITH_PARENT,
 133                                                   GTK_STOCK_CANCEL,
 134                                                   GTK_RESPONSE_REJECT,
 135                                                   GTK_STOCK_OK,
 136                                                   GTK_RESPONSE_ACCEPT,
 137                                                   NULL);
 138   GtkWidget *label = gtk_label_new ( msg );
 139   GtkTreeStore *store;
 140   if (multiple_selection_allowed)
 141   {
 142     store = gtk_tree_store_new(4, G_TYPE_BOOLEAN, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_STRING);
 143   }
 144   else
 145   {
 146     store = gtk_tree_store_new(3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_STRING);
 147   }
 148   GList *geoname_runner = geonames;
 149   while (geoname_runner)
 150   {
 151     geoname = (found_geoname *)geoname_runner->data;
 152     latlon_string = g_strdup_printf("(%f,%f)", geoname->ll.lat, geoname->ll.lon);
 153     gtk_tree_store_append(store, &iter, NULL);
 154     if (multiple_selection_allowed)
 155     {
 156       gtk_tree_store_set(store, &iter, 0, FALSE, 1, geoname->name, 2, geoname->country, 3, latlon_string, -1);
 157     }
 158     else
 159     {
 160       gtk_tree_store_set(store, &iter, 0, geoname->name, 1, geoname->country, 2, latlon_string, -1);
 161     }
 162     geoname_runner = g_list_next(geoname_runner);
 163     g_free(latlon_string);
 164   }
 165   view = gtk_tree_view_new();
 166   renderer = gtk_cell_renderer_text_new();
 167   column_runner = 0;
 168   if (multiple_selection_allowed)
 169   {
 170     toggle_render = gtk_cell_renderer_toggle_new();
 171     g_object_set(toggle_render, "activatable", TRUE, NULL);
 172     g_signal_connect(toggle_render, "toggled", (GCallback) buttonToggled, GTK_TREE_MODEL(store));
 173     gtk_tree_view_insert_column_with_attributes( GTK_TREE_VIEW(view), -1, "Select", toggle_render, "active", column_runner, NULL);
 174     column_runner++;
 175   }
 176   gtk_tree_view_insert_column_with_attributes( GTK_TREE_VIEW(view), -1, "Name", renderer, "text", column_runner, NULL);
 177   column_runner++;
 178   gtk_tree_view_insert_column_with_attributes( GTK_TREE_VIEW(view), -1, "Country", renderer, "text", column_runner, NULL);
 179   column_runner++;
 180   gtk_tree_view_insert_column_with_attributes( GTK_TREE_VIEW(view), -1, "Lat/Lon", renderer, "text", column_runner, NULL);
 181   gtk_tree_view_set_headers_visible( GTK_TREE_VIEW(view), TRUE);
 182   gtk_tree_view_set_model(GTK_TREE_VIEW(view), GTK_TREE_MODEL(store));
 183   gtk_tree_selection_set_mode( gtk_tree_view_get_selection(GTK_TREE_VIEW(view)),
 184       multiple_selection_allowed ? GTK_SELECTION_MULTIPLE : GTK_SELECTION_BROWSE );
 185   g_object_unref(store);
 186
 187   gtk_box_pack_start (GTK_BOX(GTK_DIALOG(dialog)->vbox), label, FALSE, FALSE, 0);
 188   gtk_widget_show ( label );
 189   gtk_box_pack_start (GTK_BOX(GTK_DIALOG(dialog)->vbox), view, FALSE, FALSE, 0);
 190   gtk_widget_show ( view );
 191   while ( gtk_dialog_run (GTK_DIALOG (dialog)) == GTK_RESPONSE_ACCEPT )
 192   {
 193     GtkTreeSelection *selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(view));
 194     GList *selected_geonames = NULL;
 195
 196     gtk_tree_model_get_iter_first( GTK_TREE_MODEL(store), &iter);
 197     geoname_runner = geonames;
 198     while (geoname_runner)
 199     {
 200       to_copy = FALSE;
 201       if (multiple_selection_allowed)
 202       {
 203         gtk_tree_model_get(GTK_TREE_MODEL(store), &iter, 0, &checked, -1);
 204         if (checked) {
 205           to_copy = TRUE;
 206         }
 207       }
 208       else
 209       {
 210         if (gtk_tree_selection_iter_is_selected(selection, &iter))
 211         {
 212           to_copy = TRUE;
 213         }
 214       }
 215       if (to_copy) {
 216         found_geoname *copied = copy_found_geoname(geoname_runner->data);
 217         selected_geonames = g_list_prepend(selected_geonames, copied);
 218       }
 219       geoname_runner = g_list_next(geoname_runner);
 220       gtk_tree_model_iter_next(GTK_TREE_MODEL(store), &iter);
 221     }
 222     if (selected_geonames)
 223     {
 224       gtk_widget_destroy ( dialog );
 225       return (selected_geonames);
 226     }
 227     a_dialog_error_msg(parent, _("Nothing was selected"));
 228   }
 229   gtk_widget_destroy ( dialog );
 230   return NULL;
 231 }
 232
 233 GList *get_entries_from_file(gchar *file_name)
 234 {
 235   gchar *text, *pat;
 236   GMappedFile *mf;
 237   gsize len;
 238   gboolean more = TRUE;
 239   gchar lat_buf[32], lon_buf[32];
 240   gchar *s;
 241   gint fragment_len;
 242   GList *found_places = NULL;
 243   found_geoname *geoname = NULL;
 244   gchar **found_entries;
 245   gchar *entry;
 246   int entry_runner;
 247   gchar *wikipedia_url = NULL;
 248   gchar *thumbnail_url = NULL;
 249
 250   lat_buf[0] = lon_buf[0] = '\0';
 251
 252   if ((mf = g_mapped_file_new(file_name, FALSE, NULL)) == NULL) {
 253     g_critical(_("couldn't map temp file"));
 254     exit(1);
 255   }
 256   len = g_mapped_file_get_length(mf);
 257   text = g_mapped_file_get_contents(mf);
 258
 259   if (g_strstr_len(text, len, GEONAMES_SEARCH_NOT_FOUND) != NULL) {
 260     more = FALSE;
 261   }
 262   found_entries = g_strsplit(text, "},", 0);
 263   entry_runner = 0;
 264   entry = found_entries[entry_runner];
 265   while (entry)
 266   {
 267     more = TRUE;
 268     geoname = new_found_geoname();
 269     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_COUNTRY_PATTERN))) {
 270       pat += strlen(GEONAMES_COUNTRY_PATTERN);
 271       fragment_len = 0;
 272       s = pat;
 273       while (*pat != '"') {
 274         fragment_len++;
 275         pat++;
 276       }
 277       geoname -> country = g_strndup(s, fragment_len);
 278     }
 279     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_LONGITUDE_PATTERN)) == NULL) {
 280       more = FALSE;
 281     }
 282     else {
 283       pat += strlen(GEONAMES_LONGITUDE_PATTERN);
 284       s = lon_buf;
 285       if (*pat == '-')
 286         *s++ = *pat++;
 287       while ((s < (lon_buf + sizeof(lon_buf))) && (pat < (text + len)) &&
 288               (g_ascii_isdigit(*pat) || (*pat == '.')))
 289         *s++ = *pat++;
 290       *s = '\0';
 291       if ((pat >= (text + len)) || (lon_buf[0] == '\0')) {
 292         more = FALSE;
 293       }
 294       geoname->ll.lon = g_ascii_strtod(lon_buf, NULL);
 295     }
 296     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_NAME_PATTERN))) {
 297       pat += strlen(GEONAMES_NAME_PATTERN);
 298       fragment_len = 0;
 299       s = pat;
 300       while (*pat != '"') {
 301         fragment_len++;
 302         pat++;
 303       }
 304       geoname -> name = g_strndup(s, fragment_len);
 305     }
 306     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_TITLE_PATTERN))) {
 307       pat += strlen(GEONAMES_TITLE_PATTERN);
 308       fragment_len = 0;
 309       s = pat;
 310       while (*pat != '"') {
 311         fragment_len++;
 312         pat++;
 313       }
 314       geoname -> name = g_strndup(s, fragment_len);
 315     }
 316     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_WIKIPEDIAURL_PATTERN))) {
 317       pat += strlen(GEONAMES_WIKIPEDIAURL_PATTERN);
 318       fragment_len = 0;
 319       s = pat;
 320       while (*pat != '"') {
 321         fragment_len++;
 322         pat++;
 323       }
 324       wikipedia_url = g_strndup(s, fragment_len);
 325     }
 326     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_THUMBNAILIMG_PATTERN))) {
 327       pat += strlen(GEONAMES_THUMBNAILIMG_PATTERN);
 328       fragment_len = 0;
 329       s = pat;
 330       while (*pat != '"') {
 331         fragment_len++;
 332         pat++;
 333       }
 334       thumbnail_url = g_strndup(s, fragment_len);
 335     }
 336     if ((pat = g_strstr_len(entry, strlen(entry), GEONAMES_LATITUDE_PATTERN)) == NULL) {
 337       more = FALSE;
 338     }
 339     else {
 340       pat += strlen(GEONAMES_LATITUDE_PATTERN);
 341       s = lat_buf;
 342       if (*pat == '-')
 343         *s++ = *pat++;
 344       while ((s < (lat_buf + sizeof(lat_buf))) && (pat < (text + len)) &&
 345               (g_ascii_isdigit(*pat) || (*pat == '.')))
 346         *s++ = *pat++;
 347       *s = '\0';
 348       if ((pat >= (text + len)) || (lat_buf[0] == '\0')) {
 349         more = FALSE;
 350       }
 351       geoname->ll.lat = g_ascii_strtod(lat_buf, NULL);
 352     }
 353     if (!more) {
 354       if (geoname) {
 355         g_free(geoname);
 356       }
 357     }
 358     else {
 359       if (wikipedia_url) {
 360         if (thumbnail_url) {
 361           geoname -> desc = g_strdup_printf("<a href=\"http://%s\" target=\"_blank\"><img src=\"%s\" border=\"0\"/></a>", wikipedia_url, thumbnail_url);
 362         }
 363         else {
 364           geoname -> desc = g_strdup_printf("<a href=\"http://%s\" target=\"_blank\">%s</a>", wikipedia_url, geoname->name);
 365         }
 366       }
 367       if (wikipedia_url) {
 368         g_free(wikipedia_url);
 369         wikipedia_url = NULL;
 370       }
 371       if (thumbnail_url) {
 372         g_free(thumbnail_url);
 373         thumbnail_url = NULL;
 374       }
 375       found_places = g_list_prepend(found_places, geoname);
 376     }
 377     entry_runner++;
 378     entry = found_entries[entry_runner];
 379   }
 380   g_strfreev(found_entries);
 381   found_places = g_list_reverse(found_places);
 382   g_mapped_file_free(mf);
 383   return(found_places);
 384 }
 385
 386
 387 gchar *download_url(gchar *uri)
 388 {
 389   FILE *tmp_file;
 390   int tmp_fd;
 391   gchar *tmpname;
 392
 393   if ((tmp_fd = g_file_open_tmp ("vikgsearch.XXXXXX", &tmpname, NULL)) == -1) {
 394     g_critical(_("couldn't open temp file"));
 395     exit(1);
 396   }
 397   tmp_file = fdopen(tmp_fd, "r+");
 398
 399   // TODO: curl may not be available
 400   if (curl_download_uri(uri, tmp_file, NULL, 0, NULL)) {  // error
 401     fclose(tmp_file);
 402     tmp_file = NULL;
 403     g_remove(tmpname);
 404     g_free(tmpname);
 405     return(NULL);
 406   }
 407   fclose(tmp_file);
 408   tmp_file = NULL;
 409   return(tmpname);
 410 }
 411
 412 void a_geonames_wikipedia_box(VikWindow *vw, VikTrwLayer *vtl, VikLayersPanel *vlp, struct LatLon maxmin[2])
 413 {
 414   gchar *uri;
 415   gchar *tmpname;
 416   GList *wiki_places;
 417   GList *selected;
 418   GList *wp_runner;
 419   VikWaypoint *wiki_wp;
 420   found_geoname *wiki_geoname;
 421
 422   /* encode doubles in a C locale */
 423   gchar *north = a_coords_dtostr(maxmin[0].lat);
 424   gchar *south = a_coords_dtostr(maxmin[1].lat);
 425   gchar *east = a_coords_dtostr(maxmin[0].lon);
 426   gchar *west = a_coords_dtostr(maxmin[1].lon);
 427   uri = g_strdup_printf(GEONAMES_WIKIPEDIA_URL_FMT, north, south, east, west);
 428   g_free(north); north = NULL;
 429   g_free(south); south = NULL;
 430   g_free(east);  east = NULL;
 431   g_free(west);  west = NULL;
 432   tmpname = download_url(uri);
 433   if (!tmpname) {
 434     none_found(vw);
 435     return;
 436   }
 437   wiki_places = get_entries_from_file(tmpname);
 438   if (g_list_length(wiki_places) == 0) {
 439     none_found(vw);
 440     return;
 441   }
 442   selected = a_select_geoname_from_list(VIK_GTK_WINDOW_FROM_WIDGET(vw), wiki_places, TRUE, "Select articles", "Select the articles you want to add.");
 443   wp_runner = selected;
 444   while (wp_runner) {
 445     wiki_geoname = (found_geoname *)wp_runner->data;
 446     wiki_wp = vik_waypoint_new();
 447     wiki_wp->visible = TRUE;
 448     vik_coord_load_from_latlon(&(wiki_wp->coord), vik_trw_layer_get_coord_mode ( vtl ), &(wiki_geoname->ll));
 449     vik_waypoint_set_comment(wiki_wp, wiki_geoname->desc);
 450     vik_trw_layer_filein_add_waypoint ( vtl, wiki_geoname->name, wiki_wp );
 451     wp_runner = g_list_next(wp_runner);
 452   }
 453   free_geoname_list(wiki_places);
 454   free_geoname_list(selected);
 455   g_free(uri);
 456   if (tmpname) {
 457     g_free(tmpname);
 458   }
 459   vik_layers_panel_emit_update(vlp);
 460 }