]> git.mxchange.org Git - simgear.git/commitdiff
Copied original terrasync sources
authorThorstenB <brehmt@gmail.com>
Sat, 11 Jun 2011 20:55:57 +0000 (22:55 +0200)
committerThorstenB <brehmt@gmail.com>
Sat, 11 Jun 2011 20:55:57 +0000 (22:55 +0200)
simgear/scene/tsync/terrasync.cxx [new file with mode: 0644]

diff --git a/simgear/scene/tsync/terrasync.cxx b/simgear/scene/tsync/terrasync.cxx
new file mode 100644 (file)
index 0000000..d15078d
--- /dev/null
@@ -0,0 +1,654 @@
+// terrasync.cxx -- "JIT" scenery fetcher
+//
+// Written by Curtis Olson, started November 2002.
+//
+// Copyright (C) 2002  Curtis L. Olson  - http://www.flightgear.org/~curt
+// Copyright (C) 2008  Alexander R. Perry <alex.perry@ieee.org>
+//
+// 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 the Free Software Foundation; either version 2 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+//
+// $Id$
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#ifdef HAVE_WINDOWS_H
+#include <windows.h>
+#endif
+
+#ifdef __MINGW32__
+#include <time.h>
+#include <unistd.h>
+#elif defined(_MSC_VER)
+#   include <io.h>
+#   ifndef HAVE_SVN_CLIENT_H
+#       include <time.h>
+#       include <process.h>
+#   endif
+#endif
+
+#include <stdlib.h>             // atoi() atof() abs() system()
+#include <signal.h>             // signal()
+
+#include <simgear/compiler.h>
+
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <deque>
+#include <map>
+
+#include <simgear/io/raw_socket.hxx>
+#include <simgear/bucket/newbucket.hxx>
+#include <simgear/misc/sg_path.hxx>
+
+#ifdef HAVE_SVN_CLIENT_H
+#  ifdef HAVE_LIBSVN_CLIENT_1
+#    include <svn_auth.h>
+#    include <svn_client.h>
+#    include <svn_cmdline.h>
+#    include <svn_pools.h>
+#  else
+#    undef HAVE_SVN_CLIENT_H
+#  endif
+#endif
+
+using namespace std;
+
+const char* source_base = NULL;
+const char* svn_base =
+  "http://terrascenery.googlecode.com/svn/trunk/data/Scenery";
+const char* rsync_base = "scenery.flightgear.org::Scenery";
+const char* dest_base = "terrasyncdir";
+const char* rsync_cmd = 
+    "rsync --verbose --archive --delete --perms --owner --group";
+
+#ifdef HAVE_SVN_CLIENT_H
+bool use_svn = true;
+#else
+bool use_svn = false;
+const char* svn_cmd = "svn checkout";
+#endif
+
+// display usage
+static void usage( const string& prog ) {
+    cout << 
+"Usage:  terrasync [options]\n"
+"Options:  \n"
+" -d <dest>       destination directory [required]\n"
+" -R              transport using pipe to rsync\n"
+" -S              transport using built-in svn\n"
+" -p <port>       listen on UDP port [default: 5501]\n"
+" -s <source>     source base [default: '']\n"
+" -pid <pidfile>  write PID to file\n"
+" -v              be more verbose\n"
+;
+
+#ifdef HAVE_SVN_CLIENT_H
+    cout << "    (defaults to the built in subversion)" << endl;
+#else
+    cout << "    (defaults to rsync, using external commands)" << endl;
+#endif
+
+  cout << "\nExample:\n"
+"pid=$(cat $pidfile 2>/dev/null)\n"
+"if test -n \"$pid\" && kill -0 $pid ; then\n"
+"    echo \"terrasync already running: $pid\"\n"
+"else\n"
+"    nice /games/sport/fgs/utils/TerraSync/terrasync         \\\n"
+"      -v -pid $pidfile -S -p 5500 -d /games/orig/terrasync &\n"
+"fi" << endl;
+
+}
+
+deque<string> waitingTiles;
+typedef map<string,time_t> CompletedTiles;
+CompletedTiles completedTiles;
+simgear::Socket theSocket;
+
+#ifdef HAVE_SVN_CLIENT_H
+
+// Things we need for doing subversion checkout - often
+apr_pool_t *mysvn_pool = NULL;
+svn_client_ctx_t *mysvn_ctx = NULL;
+svn_opt_revision_t *mysvn_rev = NULL;
+svn_opt_revision_t *mysvn_rev_peg = NULL;
+
+static const svn_version_checklist_t mysvn_checklist[] = {
+    { "svn_subr",   svn_subr_version },
+    { "svn_client", svn_client_version },
+    { NULL, NULL }
+};
+
+// Configure our subversion session
+int mysvn_setup(void) {
+    // Are we already prepared?
+    if (mysvn_pool) return EXIT_SUCCESS;
+    // No, so initialize svn internals generally
+#ifdef _MSC_VER
+    // there is a segfault when providing an error stream.
+    //  Apparently, calling setvbuf with a nul buffer is
+    //  not supported under msvc 7.1 ( code inside svn_cmdline_init )
+    if (svn_cmdline_init("terrasync", 0) != EXIT_SUCCESS)
+        return EXIT_FAILURE;
+#else
+    if (svn_cmdline_init("terrasync", stderr) != EXIT_SUCCESS)
+        return EXIT_FAILURE;
+#endif
+    apr_pool_t *pool;
+    apr_pool_create(&pool, NULL);
+    svn_error_t *err = NULL;
+    SVN_VERSION_DEFINE(mysvn_version);
+    err = svn_ver_check_list(&mysvn_version, mysvn_checklist);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_ra_initialize(pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    char *config_dir = NULL;
+    err = svn_config_ensure(config_dir, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_client_create_context(&mysvn_ctx, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_config_get_config(&(mysvn_ctx->config),
+        config_dir, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    svn_config_t *cfg;
+    cfg = ( svn_config_t*) apr_hash_get(
+        mysvn_ctx->config,
+       SVN_CONFIG_CATEGORY_CONFIG,
+        APR_HASH_KEY_STRING);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    svn_auth_baton_t *ab;
+    err = svn_cmdline_setup_auth_baton(&ab,
+        TRUE, NULL, NULL, config_dir, TRUE, cfg,
+        mysvn_ctx->cancel_func, mysvn_ctx->cancel_baton, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    mysvn_ctx->auth_baton = ab;
+#if (SVN_VER_MINOR >= 5)
+    mysvn_ctx->conflict_func = NULL;
+    mysvn_ctx->conflict_baton = NULL;
+#endif
+    // Now our magic revisions
+    mysvn_rev = (svn_opt_revision_t*) apr_palloc(pool, 
+        sizeof(svn_opt_revision_t));
+    if (!mysvn_rev)
+        return EXIT_FAILURE;
+    mysvn_rev_peg = (svn_opt_revision_t*) apr_palloc(pool, 
+        sizeof(svn_opt_revision_t));
+    if (!mysvn_rev_peg)
+        return EXIT_FAILURE;
+    mysvn_rev->kind = svn_opt_revision_head;
+    mysvn_rev_peg->kind = svn_opt_revision_unspecified;
+    // Success if we got this far
+    mysvn_pool = pool;
+    return EXIT_SUCCESS;
+}
+
+#endif
+
+// sync one directory tree
+void sync_tree(const char* dir) {
+    int rc;
+    char command[512];
+    SGPath path( dest_base );
+
+    path.append( dir );
+    rc = path.create_dir( 0755 );
+    if (rc) {
+        cout << "Return code = " << rc << endl;
+        exit(1);
+    }
+
+    if (use_svn) {
+#ifdef HAVE_SVN_CLIENT_H
+        cout << dir << " ... ";
+       cout.flush();
+        char dest_base_dir[512];
+        snprintf( command, 512,
+            "%s/%s", source_base, dir);
+        snprintf( dest_base_dir, 512,
+            "%s/%s", dest_base, dir);
+       svn_error_t *err = NULL;
+       if (mysvn_setup() != EXIT_SUCCESS)
+           exit(1);
+       apr_pool_t *subpool = svn_pool_create(mysvn_pool);
+       
+#if (SVN_VER_MINOR >= 5)
+       err = svn_client_checkout3(NULL,
+           command,
+           dest_base_dir,
+           mysvn_rev_peg,
+           mysvn_rev,
+           svn_depth_infinity,
+           0, // ignore-externals = false
+           0, // allow unver obstructions = false
+           mysvn_ctx,
+           subpool);
+#else
+    // version 1.4 API
+    err = svn_client_checkout2(NULL,
+           command,
+           dest_base_dir,
+           mysvn_rev_peg,
+           mysvn_rev,
+           1, // recurse=true - same as svn_depth_infinity for checkout3 above
+           0, // ignore externals = false
+           mysvn_ctx,
+           subpool);
+#endif
+           
+       if (err) {
+           // Report errors from the checkout attempt
+           cout << "failed: " << endl
+                << err->message << endl;
+           svn_error_clear(err);
+           return;
+       } else {
+           cout << "done" << endl;
+       }
+       svn_pool_destroy(subpool);
+       return;
+#else
+
+        snprintf( command, 512,
+            "%s %s/%s %s/%s", svn_cmd,
+            source_base, dir,
+           dest_base, dir );
+#endif
+    } else {
+        snprintf( command, 512,
+            "%s %s/%s/ %s/%s/", rsync_cmd,
+            source_base, dir,
+           dest_base, dir );
+    }
+    cout << command << endl;
+    rc = system( command );
+    if (rc) {
+        cout << "Return code = " << rc << endl;
+        if (rc == 5120) exit(1);
+    }
+}
+
+#if defined(_MSC_VER) || defined(__MINGW32__)
+typedef void (__cdecl * sighandler_t)(int);
+#elif defined( __APPLE__ ) || defined (__FreeBSD__)
+typedef sig_t sighandler_t;
+#endif
+
+bool terminating = false;
+sighandler_t prior_signal_handlers[32];
+int termination_triggering_signals[] = {
+#if defined(_MSC_VER) || defined(__MINGW32__)
+    SIGINT, SIGILL, SIGFPE, SIGSEGV, SIGTERM, SIGBREAK, SIGABRT,
+#else
+    SIGHUP, SIGINT, SIGQUIT, SIGKILL, SIGTERM,
+#endif
+    0};  // zero terminated
+
+void terminate_request_handler(int param) {
+    char msg[] = "\nReceived signal XX, intend to exit soon.\n"
+         "repeat the signal to force immediate termination.\n";
+    msg[17] = '0' + param / 10;
+    msg[18] = '0' + param % 10;
+    write(1, msg, sizeof(msg) - 1);
+    terminating = true;
+    signal(param, prior_signal_handlers[param]);
+    theSocket.close();
+}
+
+
+const int nowhere = -9999;
+
+
+// parse message
+static void parse_message( const string &msg, int *lat, int *lon ) {
+    double dlat, dlon;
+    string text = msg;
+
+    // find GGA string and advance to start of lat
+    string::size_type pos = text.find( "$GPGGA" );
+    if ( pos == string::npos )
+    {
+       *lat = nowhere;
+       *lon = nowhere;
+       return;
+    }
+    string tmp = text.substr( pos + 7 );
+    pos = tmp.find( "," );
+    tmp = tmp.substr( pos + 1 );
+    // cout << "-> " << tmp << endl;
+
+    // find lat then advance to start of hemisphere
+    pos = tmp.find( "," );
+    string lats = tmp.substr( 0, pos );
+    dlat = atof( lats.c_str() ) / 100.0;
+    tmp = tmp.substr( pos + 1 );
+
+    // find N/S hemisphere and advance to start of lon
+    if ( tmp.substr( 0, 1 ) == "S" ) {
+        dlat = -dlat;
+    }
+    pos = tmp.find( "," );
+    tmp = tmp.substr( pos + 1 );
+
+    // find lon
+    pos = tmp.find( "," );
+    string lons = tmp.substr( 0, pos );
+    dlon = atof( lons.c_str() ) / 100.0;
+    tmp = tmp.substr( pos + 1 );
+
+    // find E/W hemisphere and advance to start of lon
+    if ( tmp.substr( 0, 1 ) == "W" ) {
+        dlon = -dlon;
+    }
+
+    if ( dlat < 0 ) {
+        *lat = (int)dlat - 1;
+    } else {
+        *lat = (int)dlat;
+    }
+
+    if ( dlon < 0 ) {
+        *lon = (int)dlon - 1;
+    } else {
+        *lon = (int)dlon;
+    }
+
+    if ((dlon == 0) && (dlat == 0)) {
+      *lon = nowhere;
+      *lat = nowhere;
+    }
+}
+
+
+// sync area
+static void sync_area( int lat, int lon ) {
+    if ( lat < -90 || lat > 90 || lon < -180 || lon > 180 )
+        return;
+    char NS, EW;
+    int baselat, baselon;
+
+    if ( lat < 0 ) {
+        int base = (int)(lat / 10);
+        if ( lat == base * 10 ) {
+            baselat = base * 10;
+        } else {
+            baselat = (base - 1) * 10;
+        }
+        NS = 's';
+    } else {
+        baselat = (int)(lat / 10) * 10;
+        NS = 'n';
+    }
+    if ( lon < 0 ) {
+        int base = (int)(lon / 10);
+        if ( lon == base * 10 ) {
+            baselon = base * 10;
+        } else {
+            baselon = (base - 1) * 10;
+        }
+        EW = 'w';
+    } else {
+        baselon = (int)(lon / 10) * 10;
+        EW = 'e';
+    }
+
+    const char* terrainobjects[3] = { "Terrain", "Objects", 0 };
+    
+    for (const char** tree = &terrainobjects[0]; *tree; tree++) {
+       char dir[512];
+       snprintf( dir, 512, "%s/%c%03d%c%02d/%c%03d%c%02d",
+               *tree,
+               EW, abs(baselon), NS, abs(baselat),
+               EW, abs(lon), NS, abs(lat) );
+       waitingTiles.push_back( dir );
+    }
+}
+
+
+// sync areas
+static void sync_areas( int lat, int lon, int lat_dir, int lon_dir ) {
+    // do current 1x1 degree area first
+    sync_area( lat, lon );
+
+    if ( lat_dir == 0 && lon_dir == 0 ) {
+        // now do surrounding 8 1x1 degree areas.
+        for ( int i = lat - 1; i <= lat + 1; ++i ) {
+            for ( int j = lon - 1; j <= lon + 1; ++j ) {
+                if ( i != lat || j != lon ) {
+                    sync_area( i, j );
+                }
+            }
+        }
+    } else {
+        if ( lat_dir != 0 ) {
+            sync_area( lat + lat_dir, lon );
+            sync_area( lat + lat_dir, lon - 1 );
+            sync_area( lat + lat_dir, lon + 1 );
+        }
+        if ( lon_dir != 0 ) {
+            sync_area( lat, lon + lon_dir );
+            sync_area( lat - 1, lon + lon_dir );
+            sync_area( lat + 1, lon + lon_dir );
+        }
+    }
+}
+
+void getWaitingTile() {
+    while ( !waitingTiles.empty() ) {
+       CompletedTiles::iterator ii =
+            completedTiles.find( waitingTiles.front() );
+       time_t now = time(0);
+       if ( ii == completedTiles.end() || ii->second + 600 < now ) {
+           sync_tree(waitingTiles.front().c_str());
+           completedTiles[ waitingTiles.front() ] = now;
+           waitingTiles.pop_front();
+           break;
+       }
+       waitingTiles.pop_front();
+    }
+}
+
+int main( int argc, char **argv ) {
+    int port = 5501;
+    char host[256] = "localhost";
+    bool testing = false;
+    bool do_checkout(true);
+    int verbose(0);
+    const char* pidfn = "";
+
+    // parse arguments
+    int i = 1;
+    while ( i < argc ) {
+        if ( (string)argv[i] == "-p" ) {
+            ++i;
+            port = atoi( argv[i] );
+        } else if ( string(argv[i]).find("-pid") == 0 ) {
+            ++i;
+            pidfn = argv[i];
+            cout << "pidfn: " << pidfn << endl;
+        } else if ( (string)argv[i] == "-s" ) {
+            ++i;
+            source_base = argv[i];
+        } else if ( (string)argv[i] == "-d" ) {
+            ++i;
+            dest_base = argv[i];
+        } else if ( (string)argv[i] == "-R" ) {
+           use_svn = false;
+        } else if ( (string)argv[i] == "-S" ) {
+           use_svn = true;
+        } else if ( (string)argv[i] == "-v" ) {
+           verbose++;
+        } else if ( (string)argv[i] == "-T" ) {
+           testing = true;
+        } else if ( (string)argv[i] == "-h" ) {
+            usage( argv[0] );
+            exit(0);
+        } else {
+            cerr << "Unrecognized verbiage '" << argv[i] << "'" << endl;
+            usage( argv[0] );
+            exit(-1);        
+        }
+        ++i;
+    }
+
+    if (*pidfn) {
+      ofstream pidstream;
+      pidstream.open(pidfn);
+      if (!pidstream.good()) {
+        cerr << "Cannot open pid file '" << pidfn << "': ";
+        perror(0);
+        exit(2);
+      }
+      pidstream << getpid() << endl;
+      pidstream.close();
+    }
+
+    // Use the appropriate default for the "-s" flag
+    if (source_base == NULL) {
+        if (use_svn)
+           source_base = svn_base;
+       else
+           source_base = rsync_base;
+    }
+    
+    // Must call this before any other net stuff
+    simgear::Socket::initSockets();
+
+    if ( ! theSocket.open( false ) ) {  // open a UDP socket
+        printf("error opening socket\n");
+        return -1;
+    }
+
+    if ( theSocket.bind( host, port ) == -1 ) {
+        printf("error binding to port %d\n", port);
+        return -1;
+    }
+
+    char msg[256];
+    int maxlen = 256;
+    int len;
+    int lat, lon;
+    int last_lat = nowhere;
+    int last_lon = nowhere;
+    bool recv_msg = false;
+
+    char synced_other;
+    if (do_checkout) {
+        for ( synced_other = 'K'; synced_other <= 'Z'; synced_other++ ) {
+            char dir[512];
+            snprintf( dir, 512, "Airports/%c", synced_other );
+            waitingTiles.push_back( dir );
+        }
+        for ( synced_other = 'A'; synced_other <= 'J'; synced_other++ ) {
+            char dir[512];
+            snprintf( dir, 512, "Airports/%c", synced_other );
+            waitingTiles.push_back( dir );
+        }
+        if ( use_svn ) {
+            waitingTiles.push_back( "Models" );
+        }
+    }
+
+
+    for (int* sigp=termination_triggering_signals; *sigp; sigp++) {
+        prior_signal_handlers[*sigp] =
+            signal(*sigp, *terminate_request_handler);
+        if (verbose) cout << "Intercepted signal " << *sigp << endl;
+    }
+
+    while ( !terminating ) {
+        // main loop
+        recv_msg = false;
+        if ( testing ) {
+            // No FGFS communications
+            lat = 37;
+            lon = -123;
+            recv_msg = (lat != last_lat) || (lon != last_lon);
+        } else {
+            if (verbose && waitingTiles.empty()) {
+                cout << "Idle; waiting for FlightGear position\n";
+            }
+            theSocket.setBlocking(waitingTiles.empty());
+            len = theSocket.recv(msg, maxlen, 0);
+            if (len >= 0) {
+                msg[len] = '\0';
+                recv_msg = true;
+                if (verbose) cout << "recv length: " << len << endl;
+                parse_message( msg, &lat, &lon );
+            }
+        }
+
+        if ( recv_msg ) {
+             // Ignore messages where the location does not change
+             if ( lat != last_lat || lon != last_lon ) {
+               cout << "pos in msg = " << lat << "," << lon << endl;
+               deque<string> oldRequests;
+               oldRequests.swap( waitingTiles );
+                int lat_dir, lon_dir, dist;
+                if ( last_lat == nowhere || last_lon == nowhere ) {
+                    lat_dir = lon_dir = 0;
+                } else {
+                    dist = lat - last_lat;
+                    if ( dist != 0 ) {
+                        lat_dir = dist / abs(dist);
+                    } else {
+                        lat_dir = 0;
+                    }
+                    dist = lon - last_lon;
+                    if ( dist != 0 ) {
+                        lon_dir = dist / abs(dist);
+                    } else {
+                        lon_dir = 0;
+                    }
+                }
+                cout << "lat = " << lat << " lon = " << lon << endl;
+                cout << "lat_dir = " << lat_dir << "  "
+                    << "lon_dir = " << lon_dir << endl;
+                sync_areas( lat, lon, lat_dir, lon_dir );
+               while ( !oldRequests.empty() ) {
+                   waitingTiles.push_back( oldRequests.front() );
+                   oldRequests.pop_front();
+               }
+                last_lat = lat;
+                last_lon = lon;
+            }
+       }
+
+        // No messages inbound, so process some pending work
+        else if ( !waitingTiles.empty() ) {
+           getWaitingTile();
+        }
+
+       else if ( testing ) {
+           terminating = true;
+       } else
+
+        #ifdef _WIN32
+                Sleep(1000);
+#else
+               sleep(1);
+#endif
+    } // while !terminating
+        
+    return 0;
+}