Fix crashes on using wrong tools
[viking/gosmore.git] / src / vikdemlayer.c
blobe755d674cfd49b1a37cf51d68c3bfdca07cb896f
1 /*
2 * viking -- GPS Data and Topo Analyzer, Explorer, and Manager
4 * Copyright (C) 2003-2005, Evan Battaglia <gtoevan@gmx.net>
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.
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.
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
21 #include <math.h>
22 #include <sys/types.h>
23 #include <sys/stat.h>
24 #include <string.h>
25 #include <stdlib.h>
27 #include "config.h"
28 #include "globals.h"
29 #include "coords.h"
30 #include "vikcoord.h"
31 #include "download.h"
32 #include "background.h"
33 #include "vikwaypoint.h"
34 #include "viktrack.h"
35 #include "vikviewport.h"
36 #include "viktreeview.h"
37 #include "viklayer.h"
38 #include "vikaggregatelayer.h"
39 #include "viklayerspanel.h"
40 #include "vikdemlayer.h"
41 #include "vikdemlayer_pixmap.h"
42 #include "vikmapslayer.h"
43 #include "dialog.h"
45 #include "dem.h"
46 #include "dems.h"
48 #define MAPS_CACHE_DIR maps_layer_default_dir()
50 #define SRTM_CACHE_TEMPLATE "%ssrtm3-%s%c%02d%c%03d.hgt.zip"
51 #define SRTM_FTP_SITE "e0srp01u.ecs.nasa.gov"
52 #define SRTM_FTP_URI "/srtm/version2/SRTM3/"
54 #ifdef VIK_CONFIG_DEM24K
55 #define DEM24K_DOWNLOAD_SCRIPT "dem24k.pl"
56 #endif
59 static VikDEMLayer *dem_layer_copy ( VikDEMLayer *vdl, gpointer vp );
60 static void dem_layer_marshall( VikDEMLayer *vdl, guint8 **data, gint *len );
61 static VikDEMLayer *dem_layer_unmarshall( guint8 *data, gint len, VikViewport *vvp );
62 static gboolean dem_layer_set_param ( VikDEMLayer *vdl, guint16 id, VikLayerParamData data, VikViewport *vp );
63 static VikLayerParamData dem_layer_get_param ( VikDEMLayer *vdl, guint16 id );
64 static void dem_layer_update_gc ( VikDEMLayer *vdl, VikViewport *vp, const gchar *color );
65 static void dem_layer_post_read ( VikLayer *vl, VikViewport *vp, gboolean from_file );
66 static void srtm_draw_existence ( VikViewport *vp );
68 #ifdef VIK_CONFIG_DEM24K
69 static void dem24k_draw_existence ( VikViewport *vp );
70 #endif
72 static VikLayerParamScale param_scales[] = {
73 { 1, 10000, 10, 1 },
74 { 1, 10, 1, 0 },
77 static gchar *params_source[] = {
78 "SRTM Global 90m (3 arcsec)",
79 #ifdef VIK_CONFIG_DEM24K
80 "USA 10m (USGS 24k)",
81 #endif
82 "None",
85 enum { DEM_SOURCE_SRTM,
86 #ifdef VIK_CONFIG_DEM24K
87 DEM_SOURCE_DEM24K,
88 #endif
89 DEM_SOURCE_NONE,
92 static VikLayerParam dem_layer_params[] = {
93 { "files", VIK_LAYER_PARAM_STRING_LIST, VIK_LAYER_GROUP_NONE, "DEM Files:", VIK_LAYER_WIDGET_FILELIST },
94 { "source", VIK_LAYER_PARAM_UINT, VIK_LAYER_GROUP_NONE, "Download Source:", VIK_LAYER_WIDGET_RADIOGROUP_STATIC, params_source, NULL },
95 { "color", VIK_LAYER_PARAM_STRING, VIK_LAYER_GROUP_NONE, "Color:", VIK_LAYER_WIDGET_ENTRY },
96 { "max_elev", VIK_LAYER_PARAM_DOUBLE, VIK_LAYER_GROUP_NONE, "Max Elev:", VIK_LAYER_WIDGET_SPINBUTTON, param_scales + 0 },
97 { "line_thickness", VIK_LAYER_PARAM_UINT, VIK_LAYER_GROUP_NONE, "Line Thickness:", VIK_LAYER_WIDGET_SPINBUTTON, param_scales + 1 },
101 enum { PARAM_FILES=0, PARAM_SOURCE, PARAM_COLOR, PARAM_MAX_ELEV, PARAM_LINE_THICKNESS, NUM_PARAMS };
103 static gpointer dem_layer_download_create ( VikWindow *vw, VikViewport *vvp);
104 static gboolean dem_layer_download_release ( VikDEMLayer *vdl, GdkEventButton *event, VikViewport *vvp );
105 static gboolean dem_layer_download_click ( VikDEMLayer *vdl, GdkEventButton *event, VikViewport *vvp );
107 static VikToolInterface dem_tools[] = {
108 { "DEM Download/Import", (VikToolConstructorFunc) dem_layer_download_create, NULL, NULL, NULL,
109 (VikToolMouseFunc) dem_layer_download_click, NULL, (VikToolMouseFunc) dem_layer_download_release },
116 static gchar *dem_colors[] = {
117 "#0000FF",
118 "#9b793c",
119 "#9c7d40",
120 "#9d8144",
121 "#9e8549",
122 "#9f894d",
123 "#a08d51",
124 "#a29156",
125 "#a3955a",
126 "#a4995e",
127 "#a69d63",
128 "#a89f65",
129 "#aaa267",
130 "#ada569",
131 "#afa76b",
132 "#b1aa6d",
133 "#b4ad6f",
134 "#b6b071",
135 "#b9b373",
136 "#bcb676",
137 "#beb978",
138 "#c0bc7a",
139 "#c2c07d",
140 "#c4c37f",
141 "#c6c681",
142 "#c8ca84",
143 "#cacd86",
144 "#ccd188",
145 "#cfd58b",
146 "#c2ce84",
147 "#b5c87e",
148 "#a9c278",
149 "#9cbb71",
150 "#8fb56b",
151 "#83af65",
152 "#76a95e",
153 "#6aa358",
154 "#5e9d52",
155 "#63a055",
156 "#69a458",
157 "#6fa85c",
158 "#74ac5f",
159 "#7ab063",
160 "#80b467",
161 "#86b86a",
162 "#8cbc6e",
163 "#92c072",
164 "#94c175",
165 "#97c278",
166 "#9ac47c",
167 "#9cc57f",
168 "#9fc682",
169 "#a2c886",
170 "#a4c989",
171 "#a7cb8d",
172 "#aacd91",
173 "#afce99",
174 "#b5d0a1",
175 "#bbd2aa",
176 "#c0d3b2",
177 "#c6d5ba",
178 "#ccd7c3",
179 "#d1d9cb",
180 "#d7dbd4",
181 "#DDDDDD",
182 "#e0e0e0",
183 "#e4e4e4",
184 "#e8e8e8",
185 "#ebebeb",
186 "#efefef",
187 "#f3f3f3",
188 "#f7f7f7",
189 "#fbfbfb",
190 "#ffffff"
194 "#9b793c",
195 "#9e8549",
196 "#a29156",
197 "#a69d63",
198 "#ada569",
199 "#b4ad6f",
200 "#bcb676",
201 "#c2c07d",
202 "#c8ca84",
203 "#cfd58b",
204 "#a9c278",
205 "#83af65",
206 "#5e9d52",
207 "#6fa85c",
208 "#80b467",
209 "#92c072",
210 "#9ac47c",
211 "#a2c886",
212 "#aacd91",
213 "#bbd2aa",
214 "#ccd7c3",
215 "#DDDDDD",
216 "#e8e8e8",
217 "#f3f3f3",
218 "#FFFFFF"
222 static const guint DEM_N_COLORS = sizeof(dem_colors)/sizeof(dem_colors[0]);
225 VikLayerInterface vik_dem_layer_interface = {
226 "DEM",
227 &demlayer_pixbuf,
229 dem_tools,
230 sizeof(dem_tools) / sizeof(dem_tools[0]),
232 dem_layer_params,
233 NUM_PARAMS,
234 NULL,
237 VIK_MENU_ITEM_ALL,
239 (VikLayerFuncCreate) vik_dem_layer_create,
240 (VikLayerFuncRealize) NULL,
241 dem_layer_post_read,
242 (VikLayerFuncFree) vik_dem_layer_free,
244 (VikLayerFuncProperties) NULL,
245 (VikLayerFuncDraw) vik_dem_layer_draw,
246 (VikLayerFuncChangeCoordMode) NULL,
248 (VikLayerFuncSetMenuItemsSelection) NULL,
249 (VikLayerFuncGetMenuItemsSelection) NULL,
251 (VikLayerFuncAddMenuItems) NULL,
252 (VikLayerFuncSublayerAddMenuItems) NULL,
254 (VikLayerFuncSublayerRenameRequest) NULL,
255 (VikLayerFuncSublayerToggleVisible) NULL,
257 (VikLayerFuncCopy) dem_layer_copy,
258 (VikLayerFuncMarshall) dem_layer_marshall,
259 (VikLayerFuncUnmarshall) dem_layer_unmarshall,
261 (VikLayerFuncSetParam) dem_layer_set_param,
262 (VikLayerFuncGetParam) dem_layer_get_param,
264 (VikLayerFuncReadFileData) NULL,
265 (VikLayerFuncWriteFileData) NULL,
267 (VikLayerFuncDeleteItem) NULL,
268 (VikLayerFuncCopyItem) NULL,
269 (VikLayerFuncPasteItem) NULL,
270 (VikLayerFuncFreeCopiedItem) NULL,
271 (VikLayerFuncDragDropRequest) NULL,
274 struct _VikDEMLayer {
275 VikLayer vl;
276 GdkGC *gc;
277 GdkGC **gcs;
278 GList *files;
279 gdouble max_elev;
280 guint8 line_thickness;
281 gchar *color;
282 guint source;
285 GType vik_dem_layer_get_type ()
287 static GType vdl_type = 0;
289 if (!vdl_type)
291 static const GTypeInfo vdl_info =
293 sizeof (VikDEMLayerClass),
294 NULL, /* base_init */
295 NULL, /* base_finalize */
296 NULL, /* class init */
297 NULL, /* class_finalize */
298 NULL, /* class_data */
299 sizeof (VikDEMLayer),
301 NULL /* instance init */
303 vdl_type = g_type_register_static ( VIK_LAYER_TYPE, "VikDEMLayer", &vdl_info, 0 );
306 return vdl_type;
309 static VikDEMLayer *dem_layer_copy ( VikDEMLayer *vdl, gpointer vp )
311 VikDEMLayer *rv = vik_dem_layer_new ( );
312 gint i;
314 rv->files = a_dems_list_copy ( vdl->files );
316 for ( i = 0; i < DEM_N_COLORS; i++ ) {
317 rv->gcs[i] = vdl->gcs[i];
318 g_object_ref ( rv->gcs[i] );
321 rv->source = vdl->source;
322 rv->max_elev = vdl->max_elev;
323 rv->line_thickness = vdl->line_thickness;
325 rv->color = g_strdup ( vdl->color );
326 rv->gc = vdl->gc;
327 g_object_ref ( rv->gc );
328 return rv;
331 static void dem_layer_marshall( VikDEMLayer *vdl, guint8 **data, gint *len )
333 vik_layer_marshall_params ( VIK_LAYER(vdl), data, len );
336 static VikDEMLayer *dem_layer_unmarshall( guint8 *data, gint len, VikViewport *vvp )
338 VikDEMLayer *rv = vik_dem_layer_new ();
339 gint i;
341 /* TODO: share GCS between layers */
342 for ( i = 0; i < DEM_N_COLORS; i++ )
343 rv->gcs[i] = vik_viewport_new_gc ( vvp, dem_colors[i], rv->line_thickness );
345 vik_layer_unmarshall_params ( VIK_LAYER(rv), data, len, vvp );
346 return rv;
349 gboolean dem_layer_set_param ( VikDEMLayer *vdl, guint16 id, VikLayerParamData data, VikViewport *vp )
351 switch ( id )
353 case PARAM_COLOR: if ( vdl->color ) g_free ( vdl->color ); vdl->color = g_strdup ( data.s ); break;
354 case PARAM_SOURCE: vdl->source = data.u; break;
355 case PARAM_MAX_ELEV: vdl->max_elev = data.d; break;
356 case PARAM_LINE_THICKNESS: if ( data.u >= 1 && data.u <= 15 ) vdl->line_thickness = data.u; break;
357 case PARAM_FILES: a_dems_load_list ( &(data.sl) ); a_dems_list_free ( vdl->files ); vdl->files = data.sl; break;
359 return TRUE;
362 static VikLayerParamData dem_layer_get_param ( VikDEMLayer *vdl, guint16 id )
364 VikLayerParamData rv;
365 switch ( id )
367 case PARAM_FILES: rv.sl = vdl->files; break;
368 case PARAM_SOURCE: rv.u = vdl->source; break;
369 case PARAM_COLOR: rv.s = vdl->color ? vdl->color : ""; break;
370 case PARAM_MAX_ELEV: rv.d = vdl->max_elev; break;
371 case PARAM_LINE_THICKNESS: rv.i = vdl->line_thickness; break;
373 return rv;
376 static void dem_layer_post_read ( VikLayer *vl, VikViewport *vp, gboolean from_file )
378 VikDEMLayer *vdl = VIK_DEM_LAYER(vl);
379 if ( vdl->gc )
380 g_object_unref ( G_OBJECT(vdl->gc) );
382 vdl->gc = vik_viewport_new_gc ( vp, vdl->color, vdl->line_thickness );
385 VikDEMLayer *vik_dem_layer_new ( )
387 VikDEMLayer *vdl = VIK_DEM_LAYER ( g_object_new ( VIK_DEM_LAYER_TYPE, NULL ) );
389 vik_layer_init ( VIK_LAYER(vdl), VIK_LAYER_DEM );
391 vdl->files = NULL;
394 vdl->gc = NULL;
396 vdl->gcs = g_malloc(sizeof(GdkGC *)*DEM_N_COLORS);
397 /* make new gcs only if we need it (copy layer -> use old) */
399 vdl->max_elev = 1000.0;
400 vdl->source = DEM_SOURCE_SRTM;
401 vdl->line_thickness = 3;
402 vdl->color = NULL;
403 return vdl;
408 static void vik_dem_layer_draw_dem ( VikDEMLayer *vdl, VikViewport *vp, VikDEM *dem )
410 VikDEMColumn *column;
412 struct LatLon dem_northeast, dem_southwest;
413 gdouble max_lat, max_lon, min_lat, min_lon;
415 /**** Check if viewport and DEM data overlap ****/
417 /* get min, max lat/lon of viewport */
418 vik_viewport_get_min_max_lat_lon ( vp, &min_lat, &max_lat, &min_lon, &max_lon );
420 /* get min, max lat/lon of DEM data */
421 if ( dem->horiz_units == VIK_DEM_HORIZ_LL_ARCSECONDS ) {
422 dem_northeast.lat = dem->max_north / 3600.0;
423 dem_northeast.lon = dem->max_east / 3600.0;
424 dem_southwest.lat = dem->min_north / 3600.0;
425 dem_southwest.lon = dem->min_east / 3600.0;
426 } else if ( dem->horiz_units == VIK_DEM_HORIZ_UTM_METERS ) {
427 struct UTM dem_northeast_utm, dem_southwest_utm;
428 dem_northeast_utm.northing = dem->max_north;
429 dem_northeast_utm.easting = dem->max_east;
430 dem_southwest_utm.northing = dem->min_north;
431 dem_southwest_utm.easting = dem->min_east;
432 dem_northeast_utm.zone = dem_southwest_utm.zone = dem->utm_zone;
433 dem_northeast_utm.letter = dem_southwest_utm.letter = dem->utm_letter;
435 a_coords_utm_to_latlon(&dem_northeast_utm, &dem_northeast);
436 a_coords_utm_to_latlon(&dem_southwest_utm, &dem_southwest);
439 if ( (max_lat > dem_northeast.lat && min_lat > dem_northeast.lat) ||
440 (max_lat < dem_southwest.lat && min_lat < dem_southwest.lat) )
441 return;
442 else if ( (max_lon > dem_northeast.lon && min_lon > dem_northeast.lon) ||
443 (max_lon < dem_southwest.lon && min_lon < dem_southwest.lon) )
444 return;
445 /* else they overlap */
447 /**** End Overlap Check ****/
448 /* boxes to show where we have DEM instead of actually drawing the DEM.
449 * useful if we want to see what areas we have coverage for (if we want
450 * to get elevation data for a track) but don't want to cover the map.
453 #if 0
454 /* draw a box if a DEM is loaded. in future I'd like to add an option for this
455 * this is useful if we want to see what areas we have dem for but don't want to
456 * cover the map (or maybe we just need translucent DEM?) */
458 VikCoord demne, demsw;
459 gint x1, y1, x2, y2;
460 vik_coord_load_from_latlon(&demne, vik_viewport_get_coord_mode(vp), &dem_northeast);
461 vik_coord_load_from_latlon(&demsw, vik_viewport_get_coord_mode(vp), &dem_southwest);
463 vik_viewport_coord_to_screen ( vp, &demne, &x1, &y1 );
464 vik_viewport_coord_to_screen ( vp, &demsw, &x2, &y2 );
466 if ( x1 > vik_viewport_get_width(vp) ) x1=vik_viewport_get_width(vp);
467 if ( y2 > vik_viewport_get_height(vp) ) y2=vik_viewport_get_height(vp);
468 if ( x2 < 0 ) x2 = 0;
469 if ( y1 < 0 ) y1 = 0;
470 vik_viewport_draw_rectangle ( vp, GTK_WIDGET(vp)->style->black_gc,
471 FALSE, x2, y1, x1-x2, y2-y1 );
472 return;
474 #endif
476 if ( dem->horiz_units == VIK_DEM_HORIZ_LL_ARCSECONDS ) {
477 VikCoord tmp; /* TODO: don't use coord_load_from_latlon, especially if in latlon drawing mode */
479 gdouble max_lat_as, max_lon_as, min_lat_as, min_lon_as;
480 gdouble start_lat_as, end_lat_as, start_lon_as, end_lon_as;
482 gdouble start_lat, end_lat, start_lon, end_lon;
484 struct LatLon counter;
486 guint x, y, start_x, start_y;
488 gint16 elev;
490 guint skip_factor = ceil ( vik_viewport_get_xmpp(vp) / 40 ); /* todo: smarter calculation. */
492 gdouble nscale_deg = dem->north_scale / ((gdouble) 3600);
493 gdouble escale_deg = dem->east_scale / ((gdouble) 3600);
495 max_lat_as = max_lat * 3600;
496 min_lat_as = min_lat * 3600;
497 max_lon_as = max_lon * 3600;
498 min_lon_as = min_lon * 3600;
500 start_lat_as = MAX(min_lat_as, dem->min_north);
501 end_lat_as = MIN(max_lat_as, dem->max_north);
502 start_lon_as = MAX(min_lon_as, dem->min_east);
503 end_lon_as = MIN(max_lon_as, dem->max_east);
505 start_lat = floor(start_lat_as / dem->north_scale) * nscale_deg;
506 end_lat = ceil (end_lat_as / dem->north_scale) * nscale_deg;
507 start_lon = floor(start_lon_as / dem->east_scale) * escale_deg;
508 end_lon = ceil (end_lon_as / dem->east_scale) * escale_deg;
510 vik_dem_east_north_to_xy ( dem, start_lon_as, start_lat_as, &start_x, &start_y );
512 for ( x=start_x, counter.lon = start_lon; counter.lon <= end_lon; counter.lon += escale_deg * skip_factor, x += skip_factor ) {
513 if ( x > 0 && x < dem->n_columns ) {
514 column = g_ptr_array_index ( dem->columns, x );
515 for ( y=start_y, counter.lat = start_lat; counter.lat <= end_lat; counter.lat += nscale_deg * skip_factor, y += skip_factor ) {
516 if ( y > column->n_points )
517 break;
518 elev = column->points[y];
519 if ( elev > vdl->max_elev ) elev=vdl->max_elev;
521 gint a, b;
523 vik_coord_load_from_latlon(&tmp, vik_viewport_get_coord_mode(vp), &counter);
524 vik_viewport_coord_to_screen(vp, &tmp, &a, &b);
525 if ( elev == VIK_DEM_INVALID_ELEVATION )
526 ; /* don't draw it */
527 else if ( elev <= 0 )
528 vik_viewport_draw_rectangle(vp, vdl->gcs[0], TRUE, a-2, b-2, 4, 4 );
529 else
530 vik_viewport_draw_rectangle(vp, vdl->gcs[(gint)floor(elev/vdl->max_elev*(DEM_N_COLORS-2))+1], TRUE, a-2, b-2, 4, 4 );
532 } /* for y= */
534 } /* for x= */
535 } else if ( dem->horiz_units == VIK_DEM_HORIZ_UTM_METERS ) {
536 gdouble max_nor, max_eas, min_nor, min_eas;
537 gdouble start_nor, start_eas, end_nor, end_eas;
539 gint16 elev;
541 guint x, y, start_x, start_y;
543 VikCoord tmp; /* TODO: don't use coord_load_from_latlon, especially if in latlon drawing mode */
544 struct UTM counter;
546 guint skip_factor = ceil ( vik_viewport_get_xmpp(vp) / 10 ); /* todo: smarter calculation. */
548 VikCoord tleft, tright, bleft, bright;
550 vik_viewport_screen_to_coord ( vp, 0, 0, &tleft );
551 vik_viewport_screen_to_coord ( vp, vik_viewport_get_width(vp), 0, &tright );
552 vik_viewport_screen_to_coord ( vp, 0, vik_viewport_get_height(vp), &bleft );
553 vik_viewport_screen_to_coord ( vp, vik_viewport_get_width(vp), vik_viewport_get_height(vp), &bright );
556 vik_coord_convert(&tleft, VIK_COORD_UTM);
557 vik_coord_convert(&tright, VIK_COORD_UTM);
558 vik_coord_convert(&bleft, VIK_COORD_UTM);
559 vik_coord_convert(&bright, VIK_COORD_UTM);
561 max_nor = MAX(tleft.north_south, tright.north_south);
562 min_nor = MIN(bleft.north_south, bright.north_south);
563 max_eas = MAX(bright.east_west, tright.east_west);
564 min_eas = MIN(bleft.east_west, tleft.east_west);
566 start_nor = MAX(min_nor, dem->min_north);
567 end_nor = MIN(max_nor, dem->max_north);
568 if ( tleft.utm_zone == dem->utm_zone && bleft.utm_zone == dem->utm_zone
569 && (tleft.utm_letter >= 'N') == (dem->utm_letter >= 'N')
570 && (bleft.utm_letter >= 'N') == (dem->utm_letter >= 'N') ) /* if the utm zones/hemispheres are different, min_eas will be bogus */
571 start_eas = MAX(min_eas, dem->min_east);
572 else
573 start_eas = dem->min_east;
574 if ( tright.utm_zone == dem->utm_zone && bright.utm_zone == dem->utm_zone
575 && (tright.utm_letter >= 'N') == (dem->utm_letter >= 'N')
576 && (bright.utm_letter >= 'N') == (dem->utm_letter >= 'N') ) /* if the utm zones/hemispheres are different, min_eas will be bogus */
577 end_eas = MIN(max_eas, dem->max_east);
578 else
579 end_eas = dem->max_east;
581 start_nor = floor(start_nor / dem->north_scale) * dem->north_scale;
582 end_nor = ceil (end_nor / dem->north_scale) * dem->north_scale;
583 start_eas = floor(start_eas / dem->east_scale) * dem->east_scale;
584 end_eas = ceil (end_eas / dem->east_scale) * dem->east_scale;
586 vik_dem_east_north_to_xy ( dem, start_eas, start_nor, &start_x, &start_y );
588 /* TODO: why start_x and start_y are -1 -- rounding error from above? */
590 counter.zone = dem->utm_zone;
591 counter.letter = dem->utm_letter;
593 for ( x=start_x, counter.easting = start_eas; counter.easting <= end_eas; counter.easting += dem->east_scale * skip_factor, x += skip_factor ) {
594 if ( x > 0 && x < dem->n_columns ) {
595 column = g_ptr_array_index ( dem->columns, x );
596 for ( y=start_y, counter.northing = start_nor; counter.northing <= end_nor; counter.northing += dem->north_scale * skip_factor, y += skip_factor ) {
597 if ( y > column->n_points )
598 continue;
599 elev = column->points[y];
600 if ( elev > vdl->max_elev ) elev=vdl->max_elev;
602 gint a, b;
603 vik_coord_load_from_utm(&tmp, vik_viewport_get_coord_mode(vp), &counter);
604 vik_viewport_coord_to_screen(vp, &tmp, &a, &b);
605 if ( elev == VIK_DEM_INVALID_ELEVATION )
606 ; /* don't draw it */
607 else if ( elev <= 0 )
608 vik_viewport_draw_rectangle(vp, vdl->gcs[0], TRUE, a-2, b-2, 4, 4 );
609 else
610 vik_viewport_draw_rectangle(vp, vdl->gcs[(gint)floor(elev/vdl->max_elev*(DEM_N_COLORS-2))+1], TRUE, a-2, b-2, 4, 4 );
612 } /* for y= */
614 } /* for x= */
618 /* return the continent for the specified lat, lon */
619 /* TODO */
620 const gchar *srtm_continent_dir ( gint lat, gint lon )
622 return "North_America/";
625 void vik_dem_layer_draw ( VikDEMLayer *vdl, gpointer data )
627 VikViewport *vp = (VikViewport *) data;
628 GList *dems_iter = vdl->files;
629 VikDEM *dem;
632 /* search for SRTM3 90m */
634 if ( vdl->source == DEM_SOURCE_SRTM )
635 srtm_draw_existence ( vp );
636 #ifdef VIK_CONFIG_DEM24K
637 else if ( vdl->source == DEM_SOURCE_DEM24K )
638 dem24k_draw_existence ( vp );
639 #endif
641 while ( dems_iter ) {
642 dem = a_dems_get ( (const char *) (dems_iter->data) );
643 if ( dem )
644 vik_dem_layer_draw_dem ( vdl, vp, dem );
645 dems_iter = dems_iter->next;
649 void vik_dem_layer_free ( VikDEMLayer *vdl )
651 gint i;
652 if ( vdl->gc != NULL )
653 g_object_unref ( G_OBJECT(vdl->gc) );
655 if ( vdl->color != NULL )
656 g_free ( vdl->color );
658 if ( vdl->gcs )
659 for ( i = 0; i < DEM_N_COLORS; i++ )
660 g_object_unref ( vdl->gcs[i] );
661 g_free ( vdl->gcs );
663 a_dems_list_free ( vdl->files );
666 static void dem_layer_update_gc ( VikDEMLayer *vdl, VikViewport *vp, const gchar *color )
668 if ( vdl->color )
669 g_free ( vdl->color );
671 vdl->color = g_strdup ( color );
673 if ( vdl->gc )
674 g_object_unref ( G_OBJECT(vdl->gc) );
676 vdl->gc = vik_viewport_new_gc ( vp, vdl->color, vdl->line_thickness );
679 VikDEMLayer *vik_dem_layer_create ( VikViewport *vp )
681 VikDEMLayer *vdl = vik_dem_layer_new ();
682 gint i;
684 /* TODO: share GCS between layers */
685 for ( i = 0; i < DEM_N_COLORS; i++ )
686 vdl->gcs[i] = vik_viewport_new_gc ( vp, dem_colors[i], vdl->line_thickness );
688 dem_layer_update_gc ( vdl, vp, "red" );
689 return vdl;
691 /**************************************************************
692 **** SOURCES & DOWNLOADING
693 **************************************************************/
694 typedef struct {
695 gchar *dest;
696 gdouble lat, lon;
698 GMutex *mutex;
699 VikDEMLayer *vdl; /* NULL if not alive */
701 guint source;
702 } DEMDownloadParams;
705 /**************************************************
706 * SOURCE: SRTM *
707 **************************************************/
709 static void srtm_dem_download_thread ( DEMDownloadParams *p, gpointer threaddata )
711 gint intlat, intlon;
713 intlat = (int)floor(p->lat);
714 intlon = (int)floor(p->lon);
715 gchar *src_fn = g_strdup_printf("%s%s%c%02d%c%03d.hgt.zip",
716 SRTM_FTP_URI,
717 srtm_continent_dir(intlat, intlon),
718 (intlat >= 0) ? 'N' : 'S',
719 ABS(intlat),
720 (intlon >= 0) ? 'E' : 'W',
721 ABS(intlon) );
723 DownloadOptions options = { NULL, 0 };
724 a_ftp_download_get_url ( SRTM_FTP_SITE, src_fn, p->dest, &options );
725 g_free ( src_fn );
728 static gchar *srtm_lat_lon_to_dest_fn ( gdouble lat, gdouble lon )
730 gint intlat, intlon;
731 intlat = (int)floor(lat);
732 intlon = (int)floor(lon);
733 return g_strdup_printf("srtm3-%s%c%02d%c%03d.hgt.zip",
734 srtm_continent_dir(intlat, intlon),
735 (intlat >= 0) ? 'N' : 'S',
736 ABS(intlat),
737 (intlon >= 0) ? 'E' : 'W',
738 ABS(intlon) );
742 /* TODO: generalize */
743 static void srtm_draw_existence ( VikViewport *vp )
745 gdouble max_lat, max_lon, min_lat, min_lon;
746 gchar buf[strlen(MAPS_CACHE_DIR)+strlen(SRTM_CACHE_TEMPLATE)+30];
747 gint i, j;
749 vik_viewport_get_min_max_lat_lon ( vp, &min_lat, &max_lat, &min_lon, &max_lon );
751 for (i = floor(min_lat); i <= floor(max_lat); i++) {
752 for (j = floor(min_lon); j <= floor(max_lon); j++) {
753 g_snprintf(buf, sizeof(buf), SRTM_CACHE_TEMPLATE,
754 MAPS_CACHE_DIR,
755 srtm_continent_dir(i, j),
756 (i >= 0) ? 'N' : 'S',
757 ABS(i),
758 (j >= 0) ? 'E' : 'W',
759 ABS(j) );
760 if ( access(buf, F_OK ) == 0 ) {
761 VikCoord ne, sw;
762 gint x1, y1, x2, y2;
763 sw.north_south = i;
764 sw.east_west = j;
765 sw.mode = VIK_COORD_LATLON;
766 ne.north_south = i+1;
767 ne.east_west = j+1;
768 ne.mode = VIK_COORD_LATLON;
769 vik_viewport_coord_to_screen ( vp, &sw, &x1, &y1 );
770 vik_viewport_coord_to_screen ( vp, &ne, &x2, &y2 );
771 if ( x1 < 0 ) x1 = 0;
772 if ( y2 < 0 ) y2 = 0;
773 vik_viewport_draw_rectangle ( vp, GTK_WIDGET(vp)->style->black_gc,
774 FALSE, x1, y2, x2-x1, y1-y2 );
781 /**************************************************
782 * SOURCE: USGS 24K *
783 **************************************************/
785 #ifdef VIK_CONFIG_DEM24K
787 static void dem24k_dem_download_thread ( DEMDownloadParams *p, gpointer threaddata )
789 /* TODO: dest dir */
790 gchar *cmdline = g_strdup_printf("%s %.03f %.03f",
791 DEM24K_DOWNLOAD_SCRIPT,
792 floor(p->lat*8)/8,
793 ceil(p->lon*8)/8 );
794 /* FIX: don't use system, use execv or something. check for existence */
795 system(cmdline);
798 static gchar *dem24k_lat_lon_to_dest_fn ( gdouble lat, gdouble lon )
800 return g_strdup_printf("dem24k/%d/%d/%.03f,%.03f.dem",
801 (gint) lat,
802 (gint) lon,
803 floor(lat*8)/8,
804 ceil(lon*8)/8);
807 /* TODO: generalize */
808 static void dem24k_draw_existence ( VikViewport *vp )
810 gdouble max_lat, max_lon, min_lat, min_lon;
811 gchar buf[strlen(MAPS_CACHE_DIR)+40];
812 gdouble i, j;
814 vik_viewport_get_min_max_lat_lon ( vp, &min_lat, &max_lat, &min_lon, &max_lon );
816 for (i = floor(min_lat*8)/8; i <= floor(max_lat*8)/8; i+=0.125) {
817 /* check lat dir first -- faster */
818 g_snprintf(buf, sizeof(buf), "%sdem24k/%d/",
819 MAPS_CACHE_DIR,
820 (gint) i );
821 if ( access(buf, F_OK) != 0 )
822 continue;
823 for (j = floor(min_lon*8)/8; j <= floor(max_lon*8)/8; j+=0.125) {
824 /* check lon dir first -- faster */
825 g_snprintf(buf, sizeof(buf), "%sdem24k/%d/%d/",
826 MAPS_CACHE_DIR,
827 (gint) i,
828 (gint) j );
829 if ( access(buf, F_OK) != 0 )
830 continue;
831 g_snprintf(buf, sizeof(buf), "%sdem24k/%d/%d/%.03f,%.03f.dem",
832 MAPS_CACHE_DIR,
833 (gint) i,
834 (gint) j,
835 floor(i*8)/8,
836 floor(j*8)/8 );
837 if ( access(buf, F_OK ) == 0 ) {
838 VikCoord ne, sw;
839 gint x1, y1, x2, y2;
840 sw.north_south = i;
841 sw.east_west = j-0.125;
842 sw.mode = VIK_COORD_LATLON;
843 ne.north_south = i+0.125;
844 ne.east_west = j;
845 ne.mode = VIK_COORD_LATLON;
846 vik_viewport_coord_to_screen ( vp, &sw, &x1, &y1 );
847 vik_viewport_coord_to_screen ( vp, &ne, &x2, &y2 );
848 if ( x1 < 0 ) x1 = 0;
849 if ( y2 < 0 ) y2 = 0;
850 vik_viewport_draw_rectangle ( vp, GTK_WIDGET(vp)->style->black_gc,
851 FALSE, x1, y2, x2-x1, y1-y2 );
856 #endif
858 /**************************************************
859 * SOURCES -- DOWNLOADING & IMPORTING TOOL *
860 **************************************************
863 static void weak_ref_cb ( gpointer ptr, GObject * dead_vdl )
865 DEMDownloadParams *p = ptr;
866 g_mutex_lock ( p->mutex );
867 p->vdl = NULL;
868 g_mutex_unlock ( p->mutex );
871 /* Try to add file full_path.
872 * full_path will be copied.
873 * returns FALSE if file does not exists, TRUE otherwise.
875 static gboolean dem_layer_add_file ( VikDEMLayer *vdl, const gchar *full_path )
877 if ( access(full_path, F_OK ) == 0 ) {
878 /* only load if file size is not 0 (not in progress */
879 struct stat sb;
880 stat (full_path, &sb);
881 if ( sb.st_size ) {
882 gchar *duped_path = g_strdup(full_path);
883 vdl->files = g_list_prepend ( vdl->files, duped_path );
884 a_dems_load ( duped_path );
885 g_warning(duped_path);
886 vik_layer_emit_update ( VIK_LAYER(vdl) );
888 return TRUE;
889 } else
890 return FALSE;
893 static void dem_download_thread ( DEMDownloadParams *p, gpointer threaddata )
895 if ( p->source == DEM_SOURCE_SRTM )
896 srtm_dem_download_thread ( p, threaddata );
897 #ifdef VIK_CONFIG_DEM24K
898 else if ( p->source == DEM_SOURCE_DEM24K )
899 dem24k_dem_download_thread ( p, threaddata );
900 #endif
902 gdk_threads_enter();
903 g_mutex_lock ( p->mutex );
904 if ( p->vdl ) {
905 g_object_weak_unref ( G_OBJECT(p->vdl), weak_ref_cb, p );
907 if ( dem_layer_add_file ( p->vdl, p->dest ) )
908 vik_layer_emit_update ( VIK_LAYER(p->vdl) );
910 g_mutex_unlock ( p->mutex );
911 gdk_threads_leave();
915 static void free_dem_download_params ( DEMDownloadParams *p )
917 g_mutex_free ( p->mutex );
918 g_free ( p->dest );
919 g_free ( p );
922 static gpointer dem_layer_download_create ( VikWindow *vw, VikViewport *vvp)
924 return vvp;
928 static gboolean dem_layer_download_release ( VikDEMLayer *vdl, GdkEventButton *event, VikViewport *vvp )
930 VikCoord coord;
931 struct LatLon ll;
933 gchar *full_path;
934 gchar *dem_file = NULL;
936 if ( vdl->source == DEM_SOURCE_NONE )
937 a_dialog_error_msg ( VIK_GTK_WINDOW_FROM_LAYER(vdl), "No download source selected. Edit layer properties." );
939 vik_viewport_screen_to_coord ( vvp, event->x, event->y, &coord );
940 vik_coord_to_latlon ( &coord, &ll );
943 if ( vdl->source == DEM_SOURCE_SRTM )
944 dem_file = srtm_lat_lon_to_dest_fn ( ll.lat, ll.lon );
945 #ifdef VIK_CONFIG_DEM24K
946 else if ( vdl->source == DEM_SOURCE_DEM24K )
947 dem_file = dem24k_lat_lon_to_dest_fn ( ll.lat, ll.lon );
948 #endif
950 if ( ! dem_file )
951 return TRUE;
953 full_path = g_strdup_printf("%s%s", MAPS_CACHE_DIR, dem_file );
955 g_warning(full_path);
957 // TODO: check if already in filelist
959 if ( ! dem_layer_add_file(vdl, full_path) ) {
960 gchar *tmp = g_strdup_printf ( "Downloading DEM %s ", dem_file );
961 DEMDownloadParams *p = g_malloc(sizeof(DEMDownloadParams));
962 p->dest = g_strdup(full_path);
963 p->lat = ll.lat;
964 p->lon = ll.lon;
965 p->vdl = vdl;
966 p->mutex = g_mutex_new();
967 p->source = vdl->source;
968 g_object_weak_ref(G_OBJECT(p->vdl), weak_ref_cb, p );
970 a_background_thread ( VIK_GTK_WINDOW_FROM_LAYER(vdl), tmp,
971 (vik_thr_func) dem_download_thread, p,
972 (vik_thr_free_func) free_dem_download_params, NULL, 1 );
975 g_free ( dem_file );
976 g_free ( full_path );
978 return TRUE;
981 static gboolean dem_layer_download_click ( VikDEMLayer *vdl, GdkEventButton *event, VikViewport *vvp )
983 /* choose & keep track of cache dir
984 * download in background thread
985 * download over area */
986 return TRUE;