]> git.street.me.uk Git - andy/viking.git/blame - src/babel.c
Add Terraserver
[andy/viking.git] / src / babel.c
CommitLineData
1d1bc3c1
EB
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
28c82d8b
EB
22/* babel.c: running external programs and redirecting to TRWLayers.
23 * GPSBabel may not be necessary for everything -- for instance,
24 * use a_babel_convert_from_shellcommand with input_file_type == NULL
25 * for an external program that outputs GPX.
26 */
27
45acf79e
MA
28#ifdef HAVE_CONFIG_H
29#include "config.h"
30#endif
31
1d1bc3c1
EB
32#include "viking.h"
33#include "gpx.h"
34#include "babel.h"
8c060406 35#include <stdio.h>
e81e41fb 36#ifdef HAVE_SYS_WAIT_H
29a5a545 37#include <sys/wait.h>
e81e41fb 38#endif
45acf79e
MA
39#ifdef HAVE_UNISTD_H
40#include <unistd.h>
41#endif
42#include <glib.h>
43#include <glib/gstdio.h>
1d1bc3c1 44
8cf048bd
EB
45/* in the future we could have support for other shells (change command strings), or not use a shell at all */
46#define BASH_LOCATION "/bin/bash"
47
7b3479e3 48gboolean a_babel_convert( VikTrwLayer *vt, const char *babelargs, BabelStatusFunc cb, gpointer user_data )
1d1bc3c1
EB
49{
50 int fd_src;
51 FILE *f;
52 gchar *name_src;
53 gboolean ret = FALSE;
3adc68b0 54 gchar *bargs = g_strconcat(babelargs, " -i gpx", NULL);
1d1bc3c1 55
2ecf5998 56 if ((fd_src = g_file_open_tmp("tmp-viking.XXXXXX", &name_src, NULL)) >= 0) {
1d1bc3c1
EB
57 f = fdopen(fd_src, "w");
58 a_gpx_write_file(vt, f);
59 fclose(f);
8c060406 60 f = NULL;
7b3479e3 61 ret = a_babel_convert_from ( vt, bargs, cb, name_src, user_data );
1d1bc3c1
EB
62 }
63
64 g_free(bargs);
8c060406 65 g_remove(name_src);
1d1bc3c1
EB
66 g_free(name_src);
67 return ret;
68}
69
28c82d8b
EB
70/* Runs args[0] with the arguments and uses the GPX module
71 * to import the GPX data into layer vt. Assumes that upon
72 * running the command, the data will appear in the (usually
73 * temporary) file name_dst.
74 *
75 * cb: callback that is run upon new data from STDOUT (?)
76 * (TODO: STDERR would be nice since we usually redirect STDOUT)
77 * user_data: passed along to cb
78 *
79 * returns TRUE on success
80 */
70548902
MA
81#ifdef WINDOWS
82gboolean babel_general_convert_from( VikTrwLayer *vt, BabelStatusFunc cb, gchar **args, const gchar *name_dst, gpointer user_data )
83{
84 gboolean ret;
85 FILE *f;
86 gchar *cmd;
87 gchar **args2;
88
89 STARTUPINFO si;
90 PROCESS_INFORMATION pi;
91
92
93 ZeroMemory( &si, sizeof(si) );
94 ZeroMemory( &pi, sizeof(pi) );
95 si.cb = sizeof(si);
96 si.dwFlags = STARTF_USESHOWWINDOW;
97 si.wShowWindow = SW_HIDE;
98
99 cmd = g_strjoinv( " ", args);
100 args2 = g_strsplit(cmd, "\\", 0);
101 g_free(cmd);
102 cmd = g_strjoinv( "\\\\", args2);
103 g_free(args2);
104 args2 = g_strsplit(cmd, "/", 0);
105 g_free(cmd);
106 cmd = g_strjoinv( "\\\\", args2);
107
108 if( !CreateProcess(
109 NULL, // No module name (use command line).
110 (LPTSTR)cmd, // Command line.
111 NULL, // Process handle not inheritable.
112 NULL, // Thread handle not inheritable.
113 FALSE, // Set handle inheritance to FALSE.
114 0, // No creation flags.
115 NULL, // Use parent's environment block.
116 NULL, // Use parent's starting directory.
117 &si, // Pointer to STARTUPINFO structure.
118 &pi ) // Pointer to PROCESS_INFORMATION structure.
119 ){
120 g_warning( "CreateProcess failed");
121 ret = FALSE;
122 }
123 else {
124 WaitForSingleObject(pi.hProcess, INFINITE);
125 WaitForSingleObject(pi.hThread, INFINITE);
126
127 CloseHandle(pi.hThread);
128 CloseHandle(pi.hProcess);
129
130 if ( cb )
131 cb(BABEL_DONE, NULL, user_data);
132
133 f = g_fopen(name_dst, "r");
134 a_gpx_read_file( vt, f );
135 fclose(f);
136 ret = TRUE;
137 }
138
139 g_strfreev( args2 );
140 g_free( cmd );
141
142 return ret;
143}
144/* Windows */
145#else
146/* Posix */
7b3479e3 147gboolean babel_general_convert_from( VikTrwLayer *vt, BabelStatusFunc cb, gchar **args, const gchar *name_dst, gpointer user_data )
8cf048bd 148{
2ecf5998 149 gboolean ret = FALSE;
8cf048bd 150 GPid pid;
6c641b1a
MA
151 GError *error = NULL;
152 gint babel_stdout;
8cf048bd
EB
153 FILE *f;
154
155
6c641b1a
MA
156 if (!g_spawn_async_with_pipes (NULL, args, NULL, 0, NULL, NULL, &pid, NULL, &babel_stdout, NULL, &error)) {
157 g_warning("Error : %s", error->message);
158 g_error_free(error);
8cf048bd
EB
159 ret = FALSE;
160 } else {
161 gchar line[512];
162 FILE *diag;
163 diag = fdopen(babel_stdout, "r");
164 setvbuf(diag, NULL, _IONBF, 0);
165
166 while (fgets(line, sizeof(line), diag)) {
7b3479e3
EB
167 if ( cb )
168 cb(BABEL_DIAG_OUTPUT, line, user_data);
8cf048bd 169 }
7b3479e3
EB
170 if ( cb )
171 cb(BABEL_DONE, NULL, user_data);
8cf048bd 172 fclose(diag);
8c060406 173 diag = NULL;
8cf048bd
EB
174 waitpid(pid, NULL, 0);
175 g_spawn_close_pid(pid);
176
8c060406
MA
177 f = g_fopen(name_dst, "r");
178 if (f) {
179 a_gpx_read_file ( vt, f );
180 fclose(f);
181 f = NULL;
182 ret = TRUE;
2ecf5998 183 }
8cf048bd
EB
184 }
185
186 return ret;
187}
70548902 188#endif /* Posix */
8cf048bd 189
7b3479e3 190gboolean a_babel_convert_from( VikTrwLayer *vt, const char *babelargs, BabelStatusFunc cb, const char *from, gpointer user_data )
1d1bc3c1
EB
191{
192 int fd_dst;
1d1bc3c1 193 gchar *name_dst;
92255687 194 gchar *cmd;
1d1bc3c1
EB
195 gboolean ret = FALSE;
196 gchar **args;
c5b782e3 197 gint nb_args;
1d1bc3c1 198
2ecf5998 199 if ((fd_dst = g_file_open_tmp("tmp-viking.XXXXXX", &name_dst, NULL)) >= 0) {
92255687 200 gchar *gpsbabel_loc;
1d1bc3c1
EB
201 close(fd_dst);
202
92255687
EB
203 gpsbabel_loc = g_find_program_in_path("gpsbabel");
204
205 if (gpsbabel_loc ) {
206 gchar *unbuffer_loc = g_find_program_in_path("unbuffer");
207 cmd = g_strdup_printf ( "%s%s%s %s -o gpx %s %s",
5f304fd7 208 unbuffer_loc ? unbuffer_loc : "",
92255687
EB
209 unbuffer_loc ? " " : "",
210 gpsbabel_loc,
211 babelargs,
212 from,
213 name_dst );
214
215 if ( unbuffer_loc )
216 g_free ( unbuffer_loc );
217
c5b782e3
GB
218 if ( g_shell_parse_argv(cmd, &nb_args, &args, NULL) ) {
219 ret = babel_general_convert_from ( vt, cb, args, name_dst, user_data );
220 g_strfreev(args);
221 }
92255687
EB
222 g_free ( cmd );
223 }
1d1bc3c1
EB
224 }
225
8c060406 226 g_remove(name_dst);
1d1bc3c1
EB
227 g_free(name_dst);
228 return ret;
229}
8cf048bd 230
28c82d8b
EB
231/* Runs the input command in a shell (bash) and optionally uses GPSBabel to convert from input_file_type.
232 * If input_file_type is NULL, doesn't use GPSBabel. Input must be GPX (or Geocaching *.loc)
233 *
234 * Uses babel_general_convert_from to actually run the command. This function
235 * prepares the command and temporary file, and sets up the arguments for bash.
236 */
237gboolean a_babel_convert_from_shellcommand ( VikTrwLayer *vt, const char *input_cmd, const char *input_file_type, BabelStatusFunc cb, gpointer user_data )
8cf048bd
EB
238{
239 int fd_dst;
240 gchar *name_dst;
241 gboolean ret = FALSE;
242 gchar **args;
243
2ecf5998 244 if ((fd_dst = g_file_open_tmp("tmp-viking.XXXXXX", &name_dst, NULL)) >= 0) {
28c82d8b
EB
245 gchar *shell_command;
246 if ( input_file_type )
247 shell_command = g_strdup_printf("%s | gpsbabel -i %s -f - -o gpx -F %s", input_cmd, input_file_type, name_dst);
248 else
249 shell_command = g_strdup_printf("%s > %s", input_cmd, name_dst);
250
29a5a545 251 g_debug("%s: %s", __FUNCTION__, shell_command);
8cf048bd
EB
252 close(fd_dst);
253
254 args = g_malloc(sizeof(gchar *)*4);
255 args[0] = BASH_LOCATION;
256 args[1] = "-c";
257 args[2] = shell_command;
258 args[3] = NULL;
259
7b3479e3 260 ret = babel_general_convert_from ( vt, cb, args, name_dst, user_data );
8cf048bd
EB
261 g_free ( args );
262 g_free ( shell_command );
263 }
264
8c060406 265 g_remove(name_dst);
8cf048bd
EB
266 g_free(name_dst);
267 return ret;
268}
269
533bbf34
MA
270gboolean a_babel_convert_from_url ( VikTrwLayer *vt, const char *url, const char *input_type, BabelStatusFunc cb, gpointer user_data )
271{
1ac37c09 272 static DownloadOptions options = {NULL, 0, a_check_html_file};
533bbf34 273 gint fd_src;
8bd9fe17
GB
274 int fetch_ret;
275 gboolean ret = FALSE;
533bbf34
MA
276 gchar *name_src;
277 gchar *babelargs;
278
279 g_debug("%s: input_type=%s url=%s", __FUNCTION__, input_type, url);
280
8bd9fe17 281 if ((fd_src = g_file_open_tmp("tmp-viking.XXXXXX", &name_src, NULL)) >= 0) {
533bbf34
MA
282 close(fd_src);
283 g_remove(name_src);
284
285 babelargs = g_strdup_printf(" -i %s", input_type);
286
1ac37c09 287 fetch_ret = a_http_download_get_url(url, "", name_src, &options);
8bd9fe17
GB
288 if (fetch_ret == 0)
289 ret = a_babel_convert_from( vt, babelargs, NULL, name_src, NULL);
533bbf34
MA
290
291 g_remove(name_src);
292 g_free(babelargs);
293 g_free(name_src);
294 }
295
296 return ret;
297}
298
70548902
MA
299#ifdef WINDOWS
300gboolean babel_general_convert_to( VikTrwLayer *vt, BabelStatusFunc cb, gchar **args, const gchar *name_src, gpointer user_data )
301{
302 gboolean ret;
303 gchar *cmd;
304 gchar **args2;
305
306 if (!a_file_export(vt, name_src, FILE_TYPE_GPX)) {
307 g_warning("%s(): error exporting to %s", __FUNCTION__, name_src);
308 return(FALSE);
309 }
310
311 STARTUPINFO si;
312 PROCESS_INFORMATION pi;
313
314 ZeroMemory( &si, sizeof(si) );
315 ZeroMemory( &pi, sizeof(pi) );
316 si.cb = sizeof(si);
317 si.dwFlags = STARTF_USESHOWWINDOW;
318 si.wShowWindow = SW_HIDE;
319
320
321 cmd = g_strjoinv( " ", args);
322 args2 = g_strsplit(cmd, "\\", 0);
323 cmd = g_strjoinv( "\\\\", args2);
324 g_free(args2);
325 args2 = g_strsplit(cmd, "/", 0);
326 g_free(cmd);
327 cmd = g_strjoinv( "\\\\", args2);
328
329 if( !CreateProcess(
330 NULL, // No module name (use command line).
331 (LPTSTR)cmd, // Command line.
332 NULL, // Process handle not inheritable.
333 NULL, // Thread handle not inheritable.
334 FALSE, // Set handle inheritance to FALSE.
335 0, // No creation flags.
336 NULL, // Use parent's environment block.
337 NULL, // Use parent's starting directory.
338 &si, // Pointer to STARTUPINFO structure.
339 &pi ) // Pointer to PROCESS_INFORMATION structure.
340 ){
341 g_warning( "CreateProcess failed" );
342 ret = FALSE;
343 }
344 else {
345
346 WaitForSingleObject(pi.hProcess, INFINITE);
347 WaitForSingleObject(pi.hThread, INFINITE);
348
349 CloseHandle(pi.hThread);
350 CloseHandle(pi.hProcess);
351
352 if ( cb )
353 cb(BABEL_DONE, NULL, user_data);
354
355 ret = TRUE;
356 }
357
358 g_strfreev(args2);
359 g_free( cmd );
360
361 return ret;
362}
363/* Windows */
364#else
365/* Posix */
b364d6bc
QT
366gboolean babel_general_convert_to( VikTrwLayer *vt, BabelStatusFunc cb, gchar **args, const gchar *name_src, gpointer user_data )
367{
2ecf5998 368 gboolean ret = FALSE;
b364d6bc 369 GPid pid;
6c641b1a
MA
370 GError *error = NULL;
371 gint babel_stdout;
b364d6bc
QT
372
373 if (!a_file_export(vt, name_src, FILE_TYPE_GPX)) {
29a5a545 374 g_warning("%s(): error exporting to %s", __FUNCTION__, name_src);
b364d6bc
QT
375 return(FALSE);
376 }
377
6c641b1a
MA
378 if (!g_spawn_async_with_pipes (NULL, args, NULL, 0, NULL, NULL, &pid, NULL, &babel_stdout, NULL, &error)) {
379 g_warning("Error : %s", error->message);
380 g_error_free(error);
b364d6bc
QT
381 ret = FALSE;
382 } else {
383 gchar line[512];
384 FILE *diag;
385 diag = fdopen(babel_stdout, "r");
386 setvbuf(diag, NULL, _IONBF, 0);
387
388 while (fgets(line, sizeof(line), diag)) {
389 if ( cb )
390 cb(BABEL_DIAG_OUTPUT, line, user_data);
391 }
392 if ( cb )
393 cb(BABEL_DONE, NULL, user_data);
394 fclose(diag);
8c060406 395 diag = NULL;
b364d6bc
QT
396 waitpid(pid, NULL, 0);
397 g_spawn_close_pid(pid);
398
399 ret = TRUE;
400 }
401
402 return ret;
403}
70548902 404#endif /* Posix */
b364d6bc
QT
405
406gboolean a_babel_convert_to( VikTrwLayer *vt, const char *babelargs, BabelStatusFunc cb, const char *to, gpointer user_data )
407{
408 int fd_src;
409 gchar *name_src;
410 gchar *cmd;
411 gboolean ret = FALSE;
412 gchar **args;
c5b782e3 413 gint nb_args;
b364d6bc 414
2ecf5998 415 if ((fd_src = g_file_open_tmp("tmp-viking.XXXXXX", &name_src, NULL)) >= 0) {
b364d6bc
QT
416 gchar *gpsbabel_loc;
417 close(fd_src);
418
419 gpsbabel_loc = g_find_program_in_path("gpsbabel");
420
421 if (gpsbabel_loc ) {
422 gchar *unbuffer_loc = g_find_program_in_path("unbuffer");
423 cmd = g_strdup_printf ( "%s%s%s %s -i gpx %s %s",
424 unbuffer_loc ? unbuffer_loc : "",
425 unbuffer_loc ? " " : "",
426 gpsbabel_loc,
427 babelargs,
428 name_src,
429 to);
430
431 if ( unbuffer_loc )
432 g_free ( unbuffer_loc );
29a5a545 433 g_debug ( "%s: %s", __FUNCTION__, cmd );
c5b782e3
GB
434 if ( g_shell_parse_argv(cmd, &nb_args, &args, NULL) ) {
435 ret = babel_general_convert_to ( vt, cb, args, name_src, user_data );
436 g_strfreev(args);
437 }
b364d6bc
QT
438 g_free ( cmd );
439 }
440 }
441
8c060406 442 g_remove(name_src);
b364d6bc
QT
443 g_free(name_src);
444 return ret;
445}