src/file.c

   1 /*
   2  * viking -- GPS Data and Topo Analyzer, Explorer, and Manager
   3  *
   4  * Copyright (C) 2003-2005, Evan Battaglia <gtoevan@gmx.net>
   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
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25 #include "viking.h"
  26
  27 #include "gpx.h"
  28
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <stdio.h>
  32 #ifdef HAVE_UNISTD_H
  33 #include <unistd.h>
  34 #endif
  35 #include <glib.h>
  36 #include <glib/gstdio.h>
  37 #include <glib/gi18n.h>
  38
  39 /* Relax some dependencies */
  40 #if ! GLIB_CHECK_VERSION(2,12,0)
  41 static gboolean return_true (gpointer a, gpointer b, gpointer c) { return TRUE; }
  42 static g_hash_table_remove_all (GHashTable *ght) { g_hash_table_foreach_remove ( ght, (GHRFunc) return_true, FALSE ); }
  43 #endif
  44
  45 #define TEST_BOOLEAN(str) (! ((str)[0] == '\0' || (str)[0] == '0' || (str)[0] == 'n' || (str)[0] == 'N' || (str)[0] == 'f' || (str)[0] == 'F') )
  46 #define VIK_MAGIC "#VIK"
  47 #define GPX_MAGIC "<?xm"
  48 #define VIK_MAGIC_LEN 4
  49
  50 #ifdef WINDOWS
  51 #define FILE_SEP '\\'
  52 #else
  53 #define FILE_SEP '/'
  54 #endif
  55
  56 typedef struct _Stack Stack;
  57
  58 struct _Stack {
  59   Stack *under;
  60   gpointer *data;
  61 };
  62
  63 static void pop(Stack **stack) {
  64   Stack *tmp = (*stack)->under;
  65   g_free ( *stack );
  66   *stack = tmp;
  67 }
  68
  69 static void push(Stack **stack)
  70 {
  71   Stack *tmp = g_malloc ( sizeof ( Stack ) );
  72   tmp->under = *stack;
  73   *stack = tmp;
  74 }
  75
  76 static gboolean check_magic ( FILE *f, const gchar *magic_number )
  77 {
  78   gchar magic[VIK_MAGIC_LEN];
  79   gboolean rv = FALSE;
  80   gint8 i;
  81   if ( fread(magic, 1, sizeof(magic), f) == sizeof(magic) &&
  82       strncmp(magic, magic_number, sizeof(magic)) == 0 )
  83     rv = TRUE;
  84   for ( i = sizeof(magic)-1; i >= 0; i-- ) /* the ol' pushback */
  85     ungetc(magic[i],f);
  86   return rv;
  87 }
  88
  89
  90 static gboolean str_starts_with ( const gchar *haystack, const gchar *needle, guint16 len_needle, gboolean must_be_longer )
  91 {
  92   if ( strlen(haystack) > len_needle - (!must_be_longer) && strncasecmp ( haystack, needle, len_needle ) == 0 )
  93     return TRUE;
  94   return FALSE;
  95 }
  96
  97 static guint16 layer_type_from_string ( const gchar *str )
  98 {
  99   guint8 i;
 100   for ( i = 0; i < VIK_LAYER_NUM_TYPES; i++ )
 101     if ( strcasecmp ( str, vik_layer_get_interface(i)->name ) == 0 )
 102       return i;
 103   return -1;
 104 }
 105
 106 void file_write_layer_param ( FILE *f, const gchar *name, guint8 type, VikLayerParamData data ) {
 107       /* string lists are handled differently. We get a GList (that shouldn't
 108        * be freed) back for get_param and if it is null we shouldn't write
 109        * anything at all (otherwise we'd read in a list with an empty string,
 110        * not an empty string list.
 111        */
 112       if ( type == VIK_LAYER_PARAM_STRING_LIST ) {
 113         if ( data.sl ) {
 114           GList *iter = (GList *)data.sl;
 115           while ( iter ) {
 116             fprintf ( f, "%s=", name );
 117             fprintf ( f, "%s\n", (gchar *)(iter->data) );
 118             iter = iter->next;
 119           }
 120         }
 121       } else {
 122         fprintf ( f, "%s=", name );
 123         switch ( type )
 124         {
 125           case VIK_LAYER_PARAM_DOUBLE: {
 126   //          char buf[15]; /* locale independent */
 127   //          fprintf ( f, "%s\n", (char *) g_dtostr (data.d, buf, sizeof (buf)) ); break;
 128               fprintf ( f, "%f\n", data.d );
 129               break;
 130          }
 131           case VIK_LAYER_PARAM_UINT: fprintf ( f, "%d\n", data.u ); break;
 132           case VIK_LAYER_PARAM_INT: fprintf ( f, "%d\n", data.i ); break;
 133           case VIK_LAYER_PARAM_BOOLEAN: fprintf ( f, "%c\n", data.b ? 't' : 'f' ); break;
 134           case VIK_LAYER_PARAM_STRING: fprintf ( f, "%s\n", data.s ); break;
 135           case VIK_LAYER_PARAM_COLOR: fprintf ( f, "#%.2x%.2x%.2x\n", (int)(data.c.red/256),(int)(data.c.green/256),(int)(data.c.blue/256)); break;
 136         }
 137       }
 138 }
 139
 140 static void write_layer_params_and_data ( VikLayer *l, FILE *f )
 141 {
 142   VikLayerParam *params = vik_layer_get_interface(l->type)->params;
 143   VikLayerFuncGetParam get_param = vik_layer_get_interface(l->type)->get_param;
 144
 145   fprintf ( f, "name=%s\n", l->name ? l->name : "" );
 146   if ( !l->visible )
 147     fprintf ( f, "visible=f\n" );
 148
 149   if ( params && get_param )
 150   {
 151     VikLayerParamData data;
 152     guint16 i, params_count = vik_layer_get_interface(l->type)->params_count;
 153     for ( i = 0; i < params_count; i++ )
 154     {
 155       data = get_param(l,i);
 156       file_write_layer_param(f, params[i].name, params[i].type, data);
 157     }
 158   }
 159   if ( vik_layer_get_interface(l->type)->write_file_data )
 160   {
 161     fprintf ( f, "\n\n~LayerData\n" );
 162     vik_layer_get_interface(l->type)->write_file_data ( l, f );
 163     fprintf ( f, "~EndLayerData\n" );
 164   }
 165   /* foreach param:
 166      write param, and get_value, etc.
 167      then run layer data, and that's it.
 168   */
 169 }
 170
 171 static void file_write ( VikAggregateLayer *top, FILE *f, gpointer vp )
 172 {
 173   Stack *stack = NULL;
 174   VikLayer *current_layer;
 175   struct LatLon ll;
 176   VikViewportDrawMode mode;
 177   gchar *modestring;
 178
 179   push(&stack);
 180   stack->data = (gpointer) vik_aggregate_layer_get_children(VIK_AGGREGATE_LAYER(top));
 181   stack->under = NULL;
 182
 183   /* crazhy CRAZHY */
 184   vik_coord_to_latlon ( vik_viewport_get_center ( VIK_VIEWPORT(vp) ), &ll );
 185
 186   mode = vik_viewport_get_drawmode ( VIK_VIEWPORT(vp) );
 187   switch ( mode ) {
 188     case VIK_VIEWPORT_DRAWMODE_UTM: modestring = "utm"; break;
 189     case VIK_VIEWPORT_DRAWMODE_EXPEDIA: modestring = "expedia"; break;
 190     case VIK_VIEWPORT_DRAWMODE_MERCATOR: modestring = "mercator"; break;
 191     default:
 192       g_critical("Houston, we've had a problem. mode=%d", mode);
 193   }
 194
 195   fprintf ( f, "#VIKING GPS Data file " VIKING_URL "\n\nxmpp=%f\nympp=%f\nlat=%f\nlon=%f\nmode=%s\ncolor=%s\ndrawscale=%s\ndrawcentermark=%s",
 196       vik_viewport_get_xmpp ( VIK_VIEWPORT(vp) ), vik_viewport_get_ympp ( VIK_VIEWPORT(vp) ), ll.lat, ll.lon,
 197       modestring, vik_viewport_get_background_color(VIK_VIEWPORT(vp)),
 198       vik_viewport_get_draw_scale(VIK_VIEWPORT(vp)) ? "t" : "f",
 199       vik_viewport_get_draw_centermark(VIK_VIEWPORT(vp)) ? "t" : "f" );
 200
 201   if ( ! VIK_LAYER(top)->visible )
 202     fprintf ( f, "visible=f\n" );
 203
 204   while (stack && stack->data)
 205   {
 206     current_layer = VIK_LAYER(((GList *)stack->data)->data);
 207     fprintf ( f, "\n~Layer %s\n", vik_layer_get_interface(current_layer->type)->name );
 208     write_layer_params_and_data ( current_layer, f );
 209     if ( current_layer->type == VIK_LAYER_AGGREGATE && !vik_aggregate_layer_is_empty(VIK_AGGREGATE_LAYER(current_layer)) )
 210     {
 211       push(&stack);
 212       stack->data = (gpointer) vik_aggregate_layer_get_children(VIK_AGGREGATE_LAYER(current_layer));
 213     }
 214     else if ( current_layer->type == VIK_LAYER_GPS && !vik_gps_layer_is_empty(VIK_GPS_LAYER(current_layer)) )
 215     {
 216       push(&stack);
 217       stack->data = (gpointer) vik_gps_layer_get_children(VIK_GPS_LAYER(current_layer));
 218     }
 219     else
 220     {
 221       stack->data = (gpointer) ((GList *)stack->data)->next;
 222       fprintf ( f, "~EndLayer\n\n" );
 223       while ( stack && (!stack->data) )
 224       {
 225         pop(&stack);
 226         if ( stack )
 227         {
 228           stack->data = (gpointer) ((GList *)stack->data)->next;
 229           fprintf ( f, "~EndLayer\n\n" );
 230         }
 231       }
 232     }
 233   }
 234 /*
 235   get vikaggregatelayer's children (?)
 236   foreach write ALL params,
 237   then layer data (IF function exists)
 238   then endlayer
 239
 240   impl:
 241   stack of layers (LIST) we are working on
 242   when layer->next == NULL ...
 243   we move on.
 244 */
 245 }
 246
 247 static void string_list_delete ( gpointer key, gpointer l, gpointer user_data )
 248 {
 249   /* 20071021 bugfix */
 250   GList *iter = (GList *) l;
 251   while ( iter ) {
 252     g_free ( iter->data );
 253     iter = iter->next;
 254   }
 255   g_list_free ( (GList *) l );
 256 }
 257
 258 static void string_list_set_param (gint i, GList *list, gpointer *layer_and_vp)
 259 {
 260   VikLayerParamData x;
 261   x.sl = list;
 262   vik_layer_set_param ( VIK_LAYER(layer_and_vp[0]), i, x, layer_and_vp[1] );
 263 }
 264
 265 static void file_read ( VikAggregateLayer *top, FILE *f, VikViewport *vp )
 266 {
 267   Stack *stack;
 268   struct LatLon ll = { 0.0, 0.0 };
 269   gchar buffer[4096];
 270   gchar *line;
 271   guint16 len;
 272   long line_num = 0;
 273
 274   VikLayerParam *params = NULL; /* for current layer, so we don't have to keep on looking up interface */
 275   guint8 params_count = 0;
 276
 277   GHashTable *string_lists = g_hash_table_new(g_direct_hash,g_direct_equal);
 278
 279   push(&stack);
 280   stack->under = NULL;
 281   stack->data = (gpointer) top;
 282
 283   while ( fgets ( buffer, 4096, f ) )
 284   {
 285     line_num++;
 286
 287     line = buffer;
 288     while ( *line == ' ' || *line =='\t' )
 289       line++;
 290
 291     if ( line[0] == '#' )
 292       continue;
 293
 294
 295     len = strlen(line);
 296     if ( len > 0 && line[len-1] == '\n' )
 297       line[--len] = '\0';
 298     if ( len > 0 && line[len-1] == '\r' )
 299       line[--len] = '\0';
 300
 301     if ( len == 0 )
 302       continue;
 303
 304
 305     if ( line[0] == '~' )
 306     {
 307       line++; len--;
 308       if ( *line == '\0' )
 309         continue;
 310       else if ( str_starts_with ( line, "Layer ", 6, TRUE ) )
 311       {
 312         int parent_type = VIK_LAYER(stack->data)->type;
 313         if ( ( ! stack->data ) || ((parent_type != VIK_LAYER_AGGREGATE) && (parent_type != VIK_LAYER_GPS)) )
 314         {
 315           g_warning ( "Line %ld: Layer command inside non-Aggregate Layer (type %d)", line_num, parent_type );
 316           push(&stack); /* inside INVALID layer */
 317           stack->data = NULL;
 318           continue;
 319         }
 320         else
 321         {
 322           gint16 type = layer_type_from_string ( line+6 );
 323           push(&stack);
 324           if ( type == -1 )
 325           {
 326             g_warning ( "Line %ld: Unknown type %s", line_num, line+6 );
 327             stack->data = NULL;
 328           }
 329           else if (parent_type == VIK_LAYER_GPS)
 330           {
 331             stack->data = (gpointer) vik_gps_layer_get_a_child(VIK_GPS_LAYER(stack->under->data));
 332             params = vik_layer_get_interface(type)->params;
 333             params_count = vik_layer_get_interface(type)->params_count;
 334           }
 335           else
 336           {
 337             stack->data = (gpointer) vik_layer_create ( type, vp, NULL, FALSE );
 338             params = vik_layer_get_interface(type)->params;
 339             params_count = vik_layer_get_interface(type)->params_count;
 340           }
 341         }
 342       }
 343       else if ( str_starts_with ( line, "EndLayer", 8, FALSE ) )
 344       {
 345         if ( stack->under == NULL )
 346           g_warning ( "Line %ld: Mismatched ~EndLayer command", line_num );
 347         else
 348         {
 349           /* add any string lists we've accumulated */
 350           gpointer layer_and_vp[2];
 351           layer_and_vp[0] = stack->data;
 352           layer_and_vp[1] = vp;
 353           g_hash_table_foreach ( string_lists, (GHFunc) string_list_set_param, layer_and_vp );
 354           g_hash_table_remove_all ( string_lists );
 355
 356           if ( stack->data && stack->under->data )
 357           {
 358             if (VIK_LAYER(stack->under->data)->type == VIK_LAYER_AGGREGATE) {
 359               vik_aggregate_layer_add_layer ( VIK_AGGREGATE_LAYER(stack->under->data), VIK_LAYER(stack->data) );
 360               vik_layer_post_read ( VIK_LAYER(stack->data), vp, TRUE );
 361             }
 362             else if (VIK_LAYER(stack->under->data)->type == VIK_LAYER_GPS) {
 363               /* TODO: anything else needs to be done here ? */
 364             }
 365             else
 366               g_warning ( "Line %ld: EndLayer command inside non-Aggregate Layer (type %d)", line_num, VIK_LAYER(stack->data)->type );
 367           }
 368           pop(&stack);
 369         }
 370       }
 371       else if ( str_starts_with ( line, "LayerData", 9, FALSE ) )
 372       {
 373         if ( stack->data && vik_layer_get_interface(VIK_LAYER(stack->data)->type)->read_file_data )
 374           vik_layer_get_interface(VIK_LAYER(stack->data)->type)->read_file_data ( VIK_LAYER(stack->data), f );
 375           /* must read until hits ~EndLayerData */
 376
 377         else
 378         { /* simply skip layer data over */
 379           while ( fgets ( buffer, 4096, f ) )
 380           {
 381             line_num++;
 382
 383             line = buffer;
 384
 385             len = strlen(line);
 386             if ( len > 0 && line[len-1] == '\n' )
 387               line[--len] = '\0';
 388             if ( len > 0 && line[len-1] == '\r' )
 389               line[--len] = '\0';
 390             if ( strcasecmp ( line, "~EndLayerData" ) == 0 )
 391               break;
 392           }
 393           continue;
 394         }
 395       }
 396       else
 397       {
 398         g_warning ( "Line %ld: Unknown tilde command", line_num );
 399       }
 400     }
 401     else
 402     {
 403       gint32 eq_pos = -1;
 404       guint16 i;
 405       if ( ! stack->data )
 406         continue;
 407
 408       for ( i = 0; i < len; i++ )
 409         if ( line[i] == '=' )
 410           eq_pos = i;
 411
 412       if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "xmpp", eq_pos ) == 0) /* "hard coded" params: global & for all layer-types */
 413         vik_viewport_set_xmpp ( VIK_VIEWPORT(vp), strtod ( line+5, NULL ) );
 414       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "ympp", eq_pos ) == 0)
 415         vik_viewport_set_ympp ( VIK_VIEWPORT(vp), strtod ( line+5, NULL ) );
 416       else if ( stack->under == NULL && eq_pos == 3 && strncasecmp ( line, "lat", eq_pos ) == 0 )
 417         ll.lat = strtod ( line+4, NULL );
 418       else if ( stack->under == NULL && eq_pos == 3 && strncasecmp ( line, "lon", eq_pos ) == 0 )
 419         ll.lon = strtod ( line+4, NULL );
 420       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "mode", eq_pos ) == 0 && strcasecmp ( line+5, "utm" ) == 0)
 421         vik_viewport_set_drawmode ( VIK_VIEWPORT(vp), VIK_VIEWPORT_DRAWMODE_UTM);
 422       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "mode", eq_pos ) == 0 && strcasecmp ( line+5, "expedia" ) == 0)
 423         vik_viewport_set_drawmode ( VIK_VIEWPORT(vp), VIK_VIEWPORT_DRAWMODE_EXPEDIA );
 424       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "mode", eq_pos ) == 0 && strcasecmp ( line+5, "google" ) == 0)
 425       {
 426         g_warning ( _("Draw mode '%s' no more supported"), "google" );
 427       }
 428       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "mode", eq_pos ) == 0 && strcasecmp ( line+5, "kh" ) == 0)
 429       {
 430         g_warning ( _("Draw mode '%s' no more supported"), "kh" );
 431       }
 432       else if ( stack->under == NULL && eq_pos == 4 && strncasecmp ( line, "mode", eq_pos ) == 0 && strcasecmp ( line+5, "mercator" ) == 0)
 433         vik_viewport_set_drawmode ( VIK_VIEWPORT(vp), VIK_VIEWPORT_DRAWMODE_MERCATOR );
 434       else if ( stack->under == NULL && eq_pos == 5 && strncasecmp ( line, "color", eq_pos ) == 0 )
 435         vik_viewport_set_background_color ( VIK_VIEWPORT(vp), line+6 );
 436       else if ( stack->under == NULL && eq_pos == 9 && strncasecmp ( line, "drawscale", eq_pos ) == 0 )
 437         vik_viewport_set_draw_scale ( VIK_VIEWPORT(vp), TEST_BOOLEAN(line+10) );
 438       else if ( stack->under == NULL && eq_pos == 14 && strncasecmp ( line, "drawcentermark", eq_pos ) == 0 )
 439         vik_viewport_set_draw_centermark ( VIK_VIEWPORT(vp), TEST_BOOLEAN(line+15) );
 440       else if ( stack->under && eq_pos == 4 && strncasecmp ( line, "name", eq_pos ) == 0 )
 441         vik_layer_rename ( VIK_LAYER(stack->data), line+5 );
 442       else if ( eq_pos == 7 && strncasecmp ( line, "visible", eq_pos ) == 0 )
 443         VIK_LAYER(stack->data)->visible = TEST_BOOLEAN(line+8);
 444       else if ( eq_pos != -1 && stack->under )
 445       {
 446         gboolean found_match = FALSE;
 447
 448         /* go thru layer params. if len == eq_pos && starts_with jazz, set it. */
 449         /* also got to check for name and visible. */
 450
 451         if ( ! params )
 452         {
 453           g_warning ( "Line %ld: No options for this kind of layer", line_num );
 454           continue;
 455         }
 456
 457         for ( i = 0; i < params_count; i++ )
 458           if ( strlen(params[i].name) == eq_pos && strncasecmp ( line, params[i].name, eq_pos ) == 0 )
 459           {
 460             VikLayerParamData x;
 461             line += eq_pos+1;
 462             if ( params[i].type == VIK_LAYER_PARAM_STRING_LIST ) {
 463               GList *l = g_list_append ( g_hash_table_lookup ( string_lists, GINT_TO_POINTER ((gint) i) ),
 464                                          g_strdup(line) );
 465               g_hash_table_replace ( string_lists, GINT_TO_POINTER ((gint)i), l );
 466               /* add the value to a list, possibly making a new list.
 467                * this will be passed to the layer when we read an ~EndLayer */
 468             } else {
 469               switch ( params[i].type )
 470               {
 471                 case VIK_LAYER_PARAM_DOUBLE: x.d = strtod(line, NULL); break;
 472                 case VIK_LAYER_PARAM_UINT: x.u = strtoul(line, NULL, 10); break;
 473                 case VIK_LAYER_PARAM_INT: x.i = strtol(line, NULL, 10); break;
 474                 case VIK_LAYER_PARAM_BOOLEAN: x.b = TEST_BOOLEAN(line); break;
 475                 case VIK_LAYER_PARAM_COLOR: memset(&(x.c), 0, sizeof(x.c)); /* default: black */
 476                                           gdk_color_parse ( line, &(x.c) ); break;
 477                 /* STRING or STRING_LIST -- if STRING_LIST, just set param to add a STRING */
 478                 default: x.s = line;
 479               }
 480               vik_layer_set_param ( VIK_LAYER(stack->data), i, x, vp );
 481             }
 482             found_match = TRUE;
 483             break;
 484           }
 485         if ( ! found_match )
 486           g_warning ( "Line %ld: Unknown parameter. Line:\n%s", line_num, line );
 487       }
 488       else
 489         g_warning ( "Line %ld: Invalid parameter or parameter outside of layer.", line_num );
 490     }
 491 /* could be:
 492 [Layer Type=Bla]
 493 [EndLayer]
 494 [LayerData]
 495 name=this
 496 #comment
 497 */
 498   }
 499
 500   while ( stack )
 501   {
 502     if ( stack->under && stack->under->data && stack->data )
 503     {
 504       vik_aggregate_layer_add_layer ( VIK_AGGREGATE_LAYER(stack->under->data), VIK_LAYER(stack->data) );
 505       vik_layer_post_read ( VIK_LAYER(stack->data), vp, TRUE );
 506     }
 507     pop(&stack);
 508   }
 509
 510   if ( ll.lat != 0.0 || ll.lon != 0.0 )
 511     vik_viewport_set_center_latlon ( VIK_VIEWPORT(vp), &ll );
 512
 513   if ( ( ! VIK_LAYER(top)->visible ) && VIK_LAYER(top)->realized )
 514     vik_treeview_item_set_visible ( VIK_LAYER(top)->vt, &(VIK_LAYER(top)->iter), FALSE );
 515
 516   /* delete anything we've forgotten about -- should only happen when file ends before an EndLayer */
 517   g_hash_table_foreach ( string_lists, string_list_delete, NULL );
 518   g_hash_table_destroy ( string_lists );
 519 }
 520
 521 /*
 522 read thru file
 523 if "[Layer Type="
 524   push(&stack)
 525   new default layer of type (str_to_type) (check interface->name)
 526 if "[EndLayer]"
 527   VikLayer *vl = stack->data;
 528   pop(&stack);
 529   vik_aggregate_layer_add_layer(stack->data, vl);
 530 if "[LayerData]"
 531   vik_layer_data ( VIK_LAYER_DATA(stack->data), f, vp );
 532
 533 */
 534
 535 /* ---------------------------------------------------- */
 536
 537 static FILE *xfopen ( const char *fn, const char *mode )
 538 {
 539   if ( strcmp(fn,"-") == 0 )
 540     return stdin;
 541   else
 542     return g_fopen(fn, "r");
 543 }
 544
 545 static void xfclose ( FILE *f )
 546 {
 547   if ( f != stdin && f != stdout ) {
 548     fclose ( f );
 549     f = NULL;
 550   }
 551 }
 552
 553 /* 0 on failure, 1 on success (vik file) 2 on success (other file) */
 554 gshort a_file_load ( VikAggregateLayer *top, VikViewport *vp, const gchar *filename_or_uri )
 555 {
 556   char *filename = filename_or_uri;
 557   if (strncmp(filename, "file://", 7) == 0)
 558     filename = filename + 7;
 559
 560   gboolean is_gpx_file = check_file_ext ( filename, ".gpx" );
 561   FILE *f = xfopen ( filename, "r" );
 562
 563   g_assert ( vp );
 564
 565   if ( ! f )
 566     return 0;
 567
 568   if ( !is_gpx_file && check_magic ( f, VIK_MAGIC ) )
 569   {
 570     file_read ( top, f, vp );
 571     if ( f != stdin )
 572       xfclose(f);
 573     return 1;
 574   }
 575   else
 576   {
 577     VikCoord new_center;
 578     VikLayer *vtl = vik_layer_create ( VIK_LAYER_TRW, vp, NULL, FALSE );
 579     vik_layer_rename ( vtl, a_file_basename ( filename ) );
 580
 581     if ( is_gpx_file || check_magic ( f, GPX_MAGIC ) )
 582       a_gpx_read_file ( VIK_TRW_LAYER(vtl), f );
 583     else
 584      a_gpspoint_read_file ( VIK_TRW_LAYER(vtl), f );
 585
 586     vik_layer_post_read ( vtl, vp, TRUE );
 587
 588     vik_aggregate_layer_add_layer ( top, vtl );
 589
 590     if ( vik_trw_layer_find_center ( VIK_TRW_LAYER(vtl), &new_center ) )
 591       vik_viewport_set_center_coord ( VIK_VIEWPORT(vp), &new_center );
 592     xfclose(f);
 593     return 2;
 594   }
 595 }
 596
 597 gboolean a_file_save ( VikAggregateLayer *top, gpointer vp, const gchar *filename )
 598 {
 599   FILE *f = g_fopen(filename, "w");
 600
 601   if ( ! f )
 602     return FALSE;
 603
 604   file_write ( top, f, vp );
 605
 606   fclose(f);
 607   f = NULL;
 608
 609   return TRUE;
 610 }
 611
 612
 613 const gchar *a_file_basename ( const gchar *filename )
 614 {
 615   const gchar *t = filename + strlen(filename) - 1;
 616   while ( --t > filename )
 617     if ( *(t-1) == FILE_SEP )
 618       break;
 619   if ( t >= filename )
 620     return t;
 621   return filename;
 622 }
 623
 624 /* example:
 625      gboolean is_gpx = check_file_ext ( "a/b/c.gpx", ".gpx" );
 626 */
 627 gboolean check_file_ext ( const gchar *filename, const gchar *fileext )
 628 {
 629   const gchar *basename = a_file_basename(filename);
 630   g_assert( filename );
 631   g_assert( fileext && fileext[0]=='.' );
 632   if (!basename)
 633     return FALSE;
 634
 635   const char * dot = strrchr(basename, '.');
 636   if (dot && !strcmp(dot, fileext))
 637     return TRUE;
 638
 639   return FALSE;
 640 }
 641
 642 gboolean a_file_export ( VikTrwLayer *vtl, const gchar *filename, gshort file_type )
 643 {
 644   FILE *f = g_fopen ( filename, "w" );
 645   if ( f )
 646   {
 647     if ( file_type == FILE_TYPE_GPSMAPPER )
 648       a_gpsmapper_write_file ( vtl, f );
 649     else if ( file_type == FILE_TYPE_GPX )
 650       a_gpx_write_file ( vtl, f );
 651     else
 652       a_gpspoint_write_file ( vtl, f );
 653     fclose ( f );
 654     f = NULL;
 655     return TRUE;
 656   }
 657   return FALSE;
 658 }
 659
 660 const gchar *a_get_viking_dir()
 661 {
 662   static gchar *viking_dir = NULL;
 663
 664   // TODO: use g_get_user_config_dir ?
 665
 666   if (!viking_dir) {
 667     const gchar *home = g_getenv("HOME");
 668     if (!home || g_access(home, W_OK))
 669       home = g_get_home_dir ();
 670 #ifdef HAVE_MKDTEMP
 671     if (!home || g_access(home, W_OK))
 672     {
 673       static gchar temp[] = {"/tmp/vikXXXXXX"};
 674       home = mkdtemp(temp);
 675     }
 676 #endif
 677     if (!home || g_access(home, W_OK))
 678       /* Fatal error */
 679       g_critical("Unable to find a base directory");
 680
 681     /* Build the name of the directory */
 682     viking_dir = g_build_filename(home, ".viking", NULL);
 683     if (g_file_test(viking_dir, G_FILE_TEST_EXISTS) == FALSE)
 684       g_mkdir(viking_dir, 0755);
 685   }
 686
 687   return viking_dir;
 688 }