* viking -- GPS Data and Topo Analyzer, Explorer, and Manager
*
* Copyright (C) 2003-2008, Evan Battaglia <gtoevan@gmx.net>
+ * Copyright (C) 2007, Quy Tonthat <qtonthat@gmail.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
*
*/
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
#include <stdio.h>
+#ifdef HAVE_STRING_H
#include <string.h>
+#endif
#include <glib.h>
+#ifdef HAVE_MATH_H
#include <math.h>
+#endif
+#ifdef HAVE_STDLIB_H
#include <stdlib.h>
-#include <zlib.h>
+#endif
#include <sys/types.h>
#include <sys/stat.h>
+#ifdef HAVE_UNISTD_H
#include <unistd.h>
-#include <sys/mman.h>
-
+#endif
+#include <glib/gstdio.h>
#include <glib/gi18n.h>
+#include "compression.h"
#include "dem.h"
-#include "file.h"
+#include "coords.h"
+#include "fileutils.h"
+
+/* Compatibility */
+#if ! GLIB_CHECK_VERSION(2,22,0)
+#define g_mapped_file_unref g_mapped_file_free
+#endif
#define DEM_BLOCK_SIZE 1024
#define GET_COLUMN(dem,n) ((VikDEMColumn *)g_ptr_array_index( (dem)->columns, (n) ))
}
}
-/* return size of unzip data or 0 if failed */
-/* can be made generic to uncompress zip, gzip, bzip2 data */
-static guint uncompress_data(void *uncompressed_buffer, guint uncompressed_size, void *compressed_data, guint compressed_size)
-{
- z_stream stream;
- int err;
-
- stream.next_in = compressed_data;
- stream.avail_in = compressed_size;
- stream.next_out = uncompressed_buffer;
- stream.avail_out = uncompressed_size;
- stream.zalloc = (alloc_func)0;
- stream.zfree = (free_func)0;
- stream.opaque = (voidpf)0;
-
- /* negative windowBits to inflateInit2 means "no header" */
- if ((err = inflateInit2(&stream, -MAX_WBITS)) != Z_OK) {
- g_warning("%s(): inflateInit2 failed", __PRETTY_FUNCTION__);
- return 0;
- }
-
- err = inflate(&stream, Z_FINISH);
- if ((err != Z_OK) && (err != Z_STREAM_END) && stream.msg) {
- g_warning("%s() inflate failed err=%d \"%s\"", __PRETTY_FUNCTION__, err, stream.msg == NULL ? "unknown" : stream.msg);
- inflateEnd(&stream);
- return 0;
- }
-
- inflateEnd(&stream);
- return(stream.total_out);
-}
-
-static void *unzip_hgt_file(gchar *zip_file, gulong *unzip_size)
-{
- void *unzip_data = NULL;
- gchar *zip_data;
- struct _lfh {
- guint32 sig;
- guint16 extract_version;
- guint16 flags;
- guint16 comp_method;
- guint16 time;
- guint16 date;
- guint32 crc_32;
- guint32 compressed_size;
- guint32 uncompressed_size;
- guint16 filename_len;
- guint16 extra_field_len;
- } __attribute__ ((__packed__)) *local_file_header = NULL;
-
-
- local_file_header = (struct _lfh *) zip_file;
- if (local_file_header->sig != 0x04034b50) {
- g_warning("%s(): wrong format", __PRETTY_FUNCTION__);
- g_free(unzip_data);
- goto end;
- }
-
- zip_data = zip_file + sizeof(struct _lfh) + local_file_header->filename_len + local_file_header->extra_field_len;
- unzip_data = g_malloc(local_file_header->uncompressed_size);
- gulong uncompressed_size = local_file_header->uncompressed_size;
-
- if (!(*unzip_size = uncompress_data(unzip_data, uncompressed_size, zip_data, local_file_header->compressed_size))) {
- g_free(unzip_data);
- unzip_data = NULL;
- goto end;
- }
-
-end:
- return(unzip_data);
-}
-
-static VikDEM *vik_dem_read_srtm_hgt(FILE *f, const gchar *basename, gboolean zip)
+static VikDEM *vik_dem_read_srtm_hgt(const gchar *file_name, const gchar *basename, gboolean zip)
{
gint i, j;
VikDEM *dem;
- struct stat stat;
off_t file_size;
gint16 *dem_mem = NULL;
- gint16 *dem_file = NULL;
+ gchar *dem_file = NULL;
const gint num_rows_3sec = 1201;
const gint num_rows_1sec = 3601;
gint num_rows;
- int fd = fileno(f);
+ GMappedFile *mf;
gint arcsec;
+ GError *error = NULL;
dem = g_malloc(sizeof(VikDEM));
dem->columns = g_ptr_array_new();
dem->n_columns = 0;
- if (fstat(fd, &stat) == -1)
- g_error("%s(): fstat failed on %s\n", __PRETTY_FUNCTION__, basename);
- if ((dem_file = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0)) == (void *) -1)
- g_error("%s(): mmap failed on %s\n", __PRETTY_FUNCTION__, basename);
-
- file_size = stat.st_size;
- dem_mem = dem_file;
+ if ((mf = g_mapped_file_new(file_name, FALSE, &error)) == NULL) {
+ g_critical(_("Couldn't map file %s: %s"), file_name, error->message);
+ g_error_free(error);
+ g_free(dem);
+ return NULL;
+ }
+ file_size = g_mapped_file_get_length(mf);
+ dem_file = g_mapped_file_get_contents(mf);
+
if (zip) {
void *unzip_mem = NULL;
gulong ucsize;
- if ((unzip_mem = unzip_hgt_file((gchar *)dem_file, &ucsize)) == NULL) {
- munmap(dem_file, file_size);
+ if ((unzip_mem = unzip_file(dem_file, &ucsize)) == NULL) {
+ g_mapped_file_unref(mf);
+ g_ptr_array_foreach ( dem->columns, (GFunc)g_free, NULL );
g_ptr_array_free(dem->columns, TRUE);
g_free(dem);
return NULL;
dem_mem = unzip_mem;
file_size = ucsize;
}
+ else
+ dem_mem = (gint16 *)dem_file;
if (file_size == (num_rows_3sec * num_rows_3sec * sizeof(gint16)))
arcsec = 3;
arcsec = 1;
else {
g_warning("%s(): file %s does not have right size", __PRETTY_FUNCTION__, basename);
- munmap(dem_file, file_size);
- g_ptr_array_free(dem->columns, TRUE);
+ g_mapped_file_unref(mf);
g_free(dem);
return NULL;
}
if (zip)
g_free(dem_mem);
- munmap(dem_file, stat.st_size);
+ g_mapped_file_unref(mf);
return dem;
}
VikDEM *vik_dem_new_from_file(const gchar *file)
{
- FILE *f;
+ FILE *f=NULL;
VikDEM *rv;
gchar buffer[DEM_BLOCK_SIZE+1];
gint cur_row = -1;
const gchar *basename = a_file_basename(file);
- /* FILE IO */
- f = fopen(file, "r");
- if ( !f )
+ if ( g_access ( file, R_OK ) != 0 )
return NULL;
if ( (strlen(basename)==11 || ((strlen(basename) == 15) && (basename[11] == '.' && basename[12] == 'z' && basename[13] == 'i' && basename[14] == 'p'))) &&
basename[7]=='.' && basename[8]=='h' && basename[9]=='g' && basename[10]=='t' &&
(basename[0] == 'N' || basename[0] == 'S') && (basename[3] == 'E' || basename[3] =='W')) {
gboolean is_zip_file = (strlen(basename) == 15);
- rv = vik_dem_read_srtm_hgt(f, basename, is_zip_file);
- fclose(f);
+ rv = vik_dem_read_srtm_hgt(file, basename, is_zip_file);
return(rv);
}
rv = g_malloc(sizeof(VikDEM));
/* Header */
+ f = g_fopen(file, "r");
+ if ( !f ) {
+ g_free ( rv );
+ return NULL;
+ }
buffer[fread(buffer, 1, DEM_BLOCK_SIZE, f)] = '\0';
if ( ! dem_parse_header ( buffer, rv ) ) {
g_free ( rv );
+ fclose(f);
return NULL;
}
/* TODO: actually use header -- i.e. GET # OF COLUMNS EXPECTED */
/* TODO - class C records (right now says 'Invalid' and dies) */
fclose(f);
+ f = NULL;
/* 24k scale */
if ( rv->horiz_units == VIK_DEM_HORIZ_UTM_METERS && rv->n_columns >= 2 )
guint i;
for ( i = 0; i < dem->n_columns; i++)
g_free ( GET_COLUMN(dem, i)->points );
+ g_ptr_array_foreach ( dem->columns, (GFunc)g_free, NULL );
g_ptr_array_free ( dem->columns, TRUE );
g_free ( dem );
}
projects / andy / viking.git / blobdiff