]> git.mxchange.org Git - simgear.git/blob - simgear/scene/tsync/terrasync.cxx
Fix VS2010 lack of fminf
[simgear.git] / simgear / scene / tsync / terrasync.cxx
1 // terrasync.cxx -- scenery fetcher
2 //
3 // Started by Curtis Olson, November 2002.
4 //
5 // Copyright (C) 2002  Curtis L. Olson  - http://www.flightgear.org/~curt
6 // Copyright (C) 2008  Alexander R. Perry <alex.perry@ieee.org>
7 // Copyright (C) 2011  Thorsten Brehm <brehmt@gmail.com>
8 //
9 // This program is free software; you can redistribute it and/or
10 // modify it under the terms of the GNU General Public License as
11 // published by the Free Software Foundation; either version 2 of the
12 // License, or (at your option) any later version.
13 //
14 // This program is distributed in the hope that it will be useful, but
15 // WITHOUT ANY WARRANTY; without even the implied warranty of
16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17 // General Public License for more details.
18 //
19 // You should have received a copy of the GNU General Public License
20 // along with this program; if not, write to the Free Software
21 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
22 //
23 // $Id$
24
25 #ifdef HAVE_CONFIG_H
26 #  include <simgear_config.h>
27 #endif
28
29 #include <simgear/compiler.h>
30
31 #ifdef HAVE_WINDOWS_H
32 #include <windows.h>
33 #endif
34
35 #ifdef __MINGW32__
36 #  include <time.h>
37 #  include <unistd.h>
38 #elif defined(_MSC_VER)
39 #   include <io.h>
40 #   include <time.h>
41 #   include <process.h>
42 #endif
43
44 #include <stdlib.h>             // atoi() atof() abs() system()
45 #include <signal.h>             // signal()
46 #include <string.h>
47
48 #include <iostream>
49 #include <fstream>
50 #include <string>
51 #include <map>
52
53 #include <simgear/compiler.h>
54
55 #include "terrasync.hxx"
56
57 #include <simgear/bucket/newbucket.hxx>
58 #include <simgear/misc/sg_path.hxx>
59 #include <simgear/misc/strutils.hxx>
60 #include <simgear/threads/SGQueue.hxx>
61 #include <simgear/misc/sg_dir.hxx>
62 #include <simgear/debug/BufferedLogCallback.hxx>
63 #include <simgear/props/props_io.hxx>
64 #include <simgear/io/HTTPClient.hxx>
65 #include <simgear/io/SVNRepository.hxx>
66 #include <simgear/structure/exception.hxx>
67
68 static const bool svn_built_in_available = true;
69
70 using namespace simgear;
71 using namespace std;
72
73 const char* rsync_cmd = 
74         "rsync --verbose --archive --delete --perms --owner --group";
75
76 const char* svn_options =
77         "checkout -q";
78
79 namespace UpdateInterval
80 {
81     // interval in seconds to allow an update to repeat after a successful update (=daily)
82     static const double SuccessfulAttempt = 24*60*60;
83     // interval in seconds to allow another update after a failed attempt (10 minutes)
84     static const double FailedAttempt     = 10*60;
85 }
86
87 typedef map<string,time_t> CompletedTiles;
88
89 ///////////////////////////////////////////////////////////////////////////////
90 // helper functions ///////////////////////////////////////////////////////////
91 ///////////////////////////////////////////////////////////////////////////////
92 string stripPath(string path)
93 {
94     // svn doesn't like trailing white-spaces or path separators - strip them!
95     path = simgear::strutils::strip(path);
96     size_t slen = path.length();
97     while ((slen>0)&&
98             ((path[slen-1]=='/')||(path[slen-1]=='\\')))
99     {
100         slen--;
101     }
102     return path.substr(0,slen);
103 }
104
105 bool hasWhitespace(string path)
106 {
107     return path.find(' ')!=string::npos;
108 }
109
110 ///////////////////////////////////////////////////////////////////////////////
111 // SyncItem ////////////////////////////////////////////////////////////
112 ///////////////////////////////////////////////////////////////////////////////
113 class SyncItem
114 {
115 public:
116     enum Type
117     {
118         Stop = 0,   ///< special item indicating to stop the SVNThread
119         Tile,
120         AirportData,
121         SharedModels,
122         AIData
123     };
124     
125     enum Status
126     {
127         Invalid = 0,
128         Waiting,
129         Cached, ///< using already cached result
130         Updated,
131         NotFound, 
132         Failed
133     };
134     
135     SyncItem() :
136         _dir(),
137         _type(Stop),
138         _status(Invalid)
139     {
140     }
141     
142     SyncItem(string dir, Type ty) :
143         _dir(dir),
144         _type(ty),
145         _status(Waiting)
146     {}
147         
148     string _dir;
149     Type _type;
150     Status _status;
151 };
152
153 ///////////////////////////////////////////////////////////////////////////////
154
155 /**
156  * @brief SyncSlot encapsulates a queue of sync items we will fetch
157  * serially. Multiple slots exist to sync different types of item in
158  * parallel.
159  */
160 class SyncSlot
161 {
162 public:
163     SyncSlot() :
164         isNewDirectory(false),
165         busy(false)
166     {}
167     
168     SyncItem currentItem;
169     bool isNewDirectory;
170     std::queue<SyncItem> queue;
171     std::auto_ptr<SVNRepository> repository;
172     SGTimeStamp stamp;
173     bool busy; ///< is the slot working or idle
174     
175     unsigned int nextWarnTimeout;
176 };
177
178 static const int SYNC_SLOT_TILES = 0; ///< Terrain and Objects sync
179 static const int SYNC_SLOT_SHARED_DATA = 1; /// shared Models and Airport data
180 static const int SYNC_SLOT_AI_DATA = 2; /// AI traffic and models
181 static const unsigned int NUM_SYNC_SLOTS = 3;
182
183 /**
184  * @brief translate a sync item type into one of the available slots.
185  * This provides the scheduling / balancing / prioritising between slots.
186  */
187 static unsigned int syncSlotForType(SyncItem::Type ty)
188 {
189     switch (ty) {
190     case SyncItem::Tile: return SYNC_SLOT_TILES;
191     case SyncItem::SharedModels:
192     case SyncItem::AirportData:
193         return SYNC_SLOT_SHARED_DATA;
194     case SyncItem::AIData:
195         return SYNC_SLOT_AI_DATA;
196         
197     default:
198         return SYNC_SLOT_SHARED_DATA;
199     }
200 }
201
202
203 ///////////////////////////////////////////////////////////////////////////////
204 // SGTerraSync::SvnThread /////////////////////////////////////////////////////
205 ///////////////////////////////////////////////////////////////////////////////
206 class SGTerraSync::SvnThread : public SGThread
207 {
208 public:
209    SvnThread();
210    virtual ~SvnThread( ) { stop(); }
211
212    void stop();
213    bool start();
214
215     bool isIdle() {return !_busy; }
216    void request(const SyncItem& dir) {waitingTiles.push_front(dir);}
217    bool isDirty() { bool r = _is_dirty;_is_dirty = false;return r;}
218    bool hasNewTiles() { return !_freshTiles.empty();}
219    SyncItem getNewTile() { return _freshTiles.pop_front();}
220
221    void   setSvnServer(string server)       { _svn_server   = stripPath(server);}
222    void   setSvnDataServer(string server)   { _svn_data_server   = stripPath(server);}
223     
224    void   setExtSvnUtility(string svn_util) { _svn_command  = simgear::strutils::strip(svn_util);}
225    void   setRsyncServer(string server)     { _rsync_server = simgear::strutils::strip(server);}
226    void   setLocalDir(string dir)           { _local_dir    = stripPath(dir);}
227    string getLocalDir()                     { return _local_dir;}
228    void   setUseSvn(bool use_svn)           { _use_svn = use_svn;}
229    void   setAllowedErrorCount(int errors)  {_allowed_errors = errors;}
230
231    void   setCachePath(const SGPath& p)     {_persistentCachePath = p;}
232    void   setCacheHits(unsigned int hits)   {_cache_hits = hits;}
233    void   setUseBuiltin(bool built_in) { _use_built_in = built_in;}
234
235    volatile bool _active;
236    volatile bool _running;
237    volatile bool _busy;
238    volatile bool _stalled;
239    volatile int  _fail_count;
240    volatile int  _updated_tile_count;
241    volatile int  _success_count;
242    volatile int  _consecutive_errors;
243    volatile int  _allowed_errors;
244    volatile int  _cache_hits;
245    volatile int _transfer_rate;
246    // kbytes, not bytes, because bytes might overflow 2^31
247    volatile int _total_kb_downloaded;
248    
249 private:
250    virtual void run();
251     
252     // external model run and helpers
253     void runExternal();
254     void syncPathExternal(const SyncItem& next);
255     bool runExternalSyncCommand(const char* dir);
256
257     // internal mode run and helpers
258     void runInternal();
259     void updateSyncSlot(SyncSlot& slot);
260
261     // commond helpers between both internal and external models
262     
263     bool isPathCached(const SyncItem& next) const;
264     void initCompletedTilesPersistentCache();
265     void writeCompletedTilesPersistentCache() const;
266     void updated(SyncItem item, bool isNewDirectory);
267     void fail(SyncItem failedItem);
268     
269     bool _use_built_in;
270     HTTP::Client _http;
271     SyncSlot _syncSlots[NUM_SYNC_SLOTS];
272
273    volatile bool _is_dirty;
274    volatile bool _stop;
275    SGBlockingDeque <SyncItem> waitingTiles;
276    CompletedTiles _completedTiles;
277    SGBlockingDeque <SyncItem> _freshTiles;
278    bool _use_svn;
279    string _svn_server;
280    string _svn_data_server;
281    string _svn_command;
282    string _rsync_server;
283    string _local_dir;
284    SGPath _persistentCachePath;
285 };
286
287 SGTerraSync::SvnThread::SvnThread() :
288     _active(false),
289     _running(false),
290     _busy(false),
291     _stalled(false),
292     _fail_count(0),
293     _updated_tile_count(0),
294     _success_count(0),
295     _consecutive_errors(0),
296     _allowed_errors(6),
297     _cache_hits(0),
298     _transfer_rate(0),
299     _total_kb_downloaded(0),
300     _use_built_in(true),
301     _is_dirty(false),
302     _stop(false),
303     _use_svn(true)
304 {
305     _http.setUserAgent("terrascenery-" SG_STRINGIZE(SG_VERSION));
306 }
307
308 void SGTerraSync::SvnThread::stop()
309 {
310     // drop any pending requests
311     waitingTiles.clear();
312
313     if (!_running)
314         return;
315
316     // set stop flag and wake up the thread with an empty request
317     _stop = true;
318     SyncItem w(string(), SyncItem::Stop);
319     request(w);
320     join();
321     _running = false;
322 }
323
324 bool SGTerraSync::SvnThread::start()
325 {
326     if (_running)
327         return false;
328
329     if (_local_dir=="")
330     {
331         SG_LOG(SG_TERRAIN,SG_ALERT,
332                "Cannot start scenery download. Local cache directory is undefined.");
333         _fail_count++;
334         _stalled = true;
335         return false;
336     }
337
338     SGPath path(_local_dir);
339     if (!path.exists())
340     {
341         SG_LOG(SG_TERRAIN,SG_ALERT,
342                "Cannot start scenery download. Directory '" << _local_dir <<
343                "' does not exist. Set correct directory path or create directory folder.");
344         _fail_count++;
345         _stalled = true;
346         return false;
347     }
348
349     path.append("version");
350     if (path.exists())
351     {
352         SG_LOG(SG_TERRAIN,SG_ALERT,
353                "Cannot start scenery download. Directory '" << _local_dir <<
354                "' contains the base package. Use a separate directory.");
355         _fail_count++;
356         _stalled = true;
357         return false;
358     }
359
360     _use_svn |= _use_built_in;
361
362     if ((_use_svn)&&(_svn_server==""))
363     {
364         SG_LOG(SG_TERRAIN,SG_ALERT,
365                "Cannot start scenery download. Subversion scenery server is undefined.");
366         _fail_count++;
367         _stalled = true;
368         return false;
369     }
370     if ((!_use_svn)&&(_rsync_server==""))
371     {
372         SG_LOG(SG_TERRAIN,SG_ALERT,
373                "Cannot start scenery download. Rsync scenery server is undefined.");
374         _fail_count++;
375         _stalled = true;
376         return false;
377     }
378
379     _fail_count = 0;
380     _updated_tile_count = 0;
381     _success_count = 0;
382     _consecutive_errors = 0;
383     _stop = false;
384     _stalled = false;
385     _running = true;
386
387     string status;
388
389     if (_use_svn && _use_built_in)
390         status = "Using built-in SVN support. ";
391     else if (_use_svn)
392     {
393         status = "Using external SVN utility '";
394         status += _svn_command;
395         status += "'. ";
396     }
397     else
398     {
399         status = "Using RSYNC. ";
400     }
401
402     // not really an alert - but we want to (always) see this message, so user is
403     // aware we're downloading scenery (and using bandwidth).
404     SG_LOG(SG_TERRAIN,SG_ALERT,
405            "Starting automatic scenery download/synchronization. "
406            << status
407            << "Directory: '" << _local_dir << "'.");
408
409     SGThread::start();
410     return true;
411 }
412
413 bool SGTerraSync::SvnThread::runExternalSyncCommand(const char* dir)
414 {
415     ostringstream buf;
416     SGPath localPath( _local_dir );
417     localPath.append( dir );
418
419     if (_use_svn)
420     {
421         buf << "\"" << _svn_command << "\" "
422             << svn_options << " "
423             << "\"" << _svn_server << "/" << dir << "\" "
424             << "\"" << localPath.str_native() << "\"";
425     } else {
426         buf << rsync_cmd << " "
427             << "\"" << _rsync_server << "/" << dir << "/\" "
428             << "\"" << localPath.str_native() << "/\"";
429     }
430
431     string command;
432 #ifdef SG_WINDOWS
433         // windows command line parsing is just lovely...
434         // to allow white spaces, the system call needs this:
435         // ""C:\Program Files\something.exe" somearg "some other arg""
436         // Note: whitespace strings quoted by a pair of "" _and_ the 
437         //       entire string needs to be wrapped by "" too.
438         // The svn url needs forward slashes (/) as a path separator while
439         // the local path needs windows-native backslash as a path separator.
440     command = "\"" + buf.str() + "\"";
441 #else
442     command = buf.str();
443 #endif
444     SG_LOG(SG_TERRAIN,SG_DEBUG, "sync command '" << command << "'");
445
446 #ifdef SG_WINDOWS
447     // tbd: does Windows support "popen"?
448     int rc = system( command.c_str() );
449 #else
450     FILE* pipe = popen( command.c_str(), "r");
451     int rc=-1;
452     // wait for external process to finish
453     if (pipe)
454         rc = pclose(pipe);
455 #endif
456
457     if (rc)
458     {
459         SG_LOG(SG_TERRAIN,SG_ALERT,
460                "Failed to synchronize directory '" << dir << "', " <<
461                "error code= " << rc);
462         return false;
463     }
464     return true;
465 }
466
467 void SGTerraSync::SvnThread::run()
468 {
469     _active = true;
470     initCompletedTilesPersistentCache();
471     
472     if (_use_built_in) {
473         runInternal();
474     } else {
475         runExternal();
476     }
477     
478     _active = false;
479     _running = false;
480     _is_dirty = true;
481 }
482
483 void SGTerraSync::SvnThread::runExternal()
484 {
485     while (!_stop)
486     {
487         SyncItem next = waitingTiles.pop_front();
488         if (_stop)
489            break;
490
491         if (isPathCached(next)) {
492             _cache_hits++;
493             SG_LOG(SG_TERRAIN, SG_DEBUG,
494                    "Cache hit for: '" << next._dir << "'");
495             next._status = SyncItem::Cached;
496             _freshTiles.push_back(next);
497             _is_dirty = true;
498             continue;
499         }
500         
501         syncPathExternal(next);
502
503         if ((_allowed_errors >= 0)&&
504             (_consecutive_errors >= _allowed_errors))
505         {
506             _stalled = true;
507             _stop = true;
508         }
509     } // of thread running loop
510 }
511
512 void SGTerraSync::SvnThread::syncPathExternal(const SyncItem& next)
513 {
514     _busy = true;
515     SGPath path( _local_dir );
516     path.append( next._dir );
517     bool isNewDirectory = !path.exists();
518     
519     try {
520         if (isNewDirectory) {
521             int rc = path.create_dir( 0755 );
522             if (rc) {
523                 SG_LOG(SG_TERRAIN,SG_ALERT,
524                        "Cannot create directory '" << path << "', return code = " << rc );
525                 throw sg_exception("Cannot create directory for terrasync", path.str());
526             }
527         }
528         
529         if (!runExternalSyncCommand(next._dir.c_str())) {
530             throw sg_exception("Running external sync command failed");
531         }
532     } catch (sg_exception& e) {
533         fail(next);
534         _busy = false;
535         return;
536     }
537     
538     updated(next, isNewDirectory);
539     _busy = false;
540 }
541
542 void SGTerraSync::SvnThread::updateSyncSlot(SyncSlot &slot)
543 {
544     if (slot.repository.get()) {
545         if (slot.repository->isDoingSync()) {
546 #if 0
547             if (slot.stamp.elapsedMSec() > slot.nextWarnTimeout) {
548                 SG_LOG(SG_TERRAIN, SG_INFO, "sync taking a long time:" << slot.currentItem._dir << " taken " << slot.stamp.elapsedMSec());
549                 SG_LOG(SG_TERRAIN, SG_INFO, "HTTP status:" << _http.hasActiveRequests());
550                 slot.nextWarnTimeout += 10000;
551             }
552 #endif
553             return; // easy, still working
554         }
555         
556         // check result
557         SVNRepository::ResultCode res = slot.repository->failure();
558         if (res == SVNRepository::SVN_ERROR_NOT_FOUND) {
559             // this is fine, but maybe we should use a different return code
560             // in the future to higher layers can distinguish this case
561             slot.currentItem._status = SyncItem::NotFound;
562             _freshTiles.push_back(slot.currentItem);
563             _is_dirty = true;
564         } else if (res != SVNRepository::SVN_NO_ERROR) {
565             fail(slot.currentItem);
566         } else {
567             updated(slot.currentItem, slot.isNewDirectory);
568             SG_LOG(SG_TERRAIN, SG_DEBUG, "sync of " << slot.repository->baseUrl() << " finished ("
569                    << slot.stamp.elapsedMSec() << " msec");
570         }
571
572         // whatever happened, we're done with this repository instance
573         slot.busy = false;
574         slot.repository.reset();
575     }
576     
577     // init and start sync of the next repository
578     if (!slot.queue.empty()) {
579         slot.currentItem = slot.queue.front();
580         slot.queue.pop();
581         
582         SGPath path(_local_dir);
583         path.append(slot.currentItem._dir);
584         slot.isNewDirectory = !path.exists();
585         if (slot.isNewDirectory) {
586             int rc = path.create_dir( 0755 );
587             if (rc) {
588                 SG_LOG(SG_TERRAIN,SG_ALERT,
589                        "Cannot create directory '" << path << "', return code = " << rc );
590                 fail(slot.currentItem);
591                 return;
592             }
593         } // of creating directory step
594         
595         string serverUrl(_svn_server);
596         if (slot.currentItem._type == SyncItem::AIData) {
597             serverUrl = _svn_data_server;
598         }
599         
600         slot.repository.reset(new SVNRepository(path, &_http));
601         slot.repository->setBaseUrl(serverUrl + "/" + slot.currentItem._dir);
602         slot.repository->update();
603         
604         slot.nextWarnTimeout = 20000;
605         slot.stamp.stamp();
606         slot.busy = true;
607         SG_LOG(SG_TERRAIN, SG_DEBUG, "sync of " << slot.repository->baseUrl() << " started, queue size is " << slot.queue.size());
608     }
609 }
610
611 void SGTerraSync::SvnThread::runInternal()
612 {
613     while (!_stop) {
614         _http.update(100);
615         _transfer_rate = _http.transferRateBytesPerSec();
616         // convert from bytes to kbytes
617         _total_kb_downloaded = static_cast<int>(_http.totalBytesDownloaded() / 1024);
618         
619         if (_stop)
620             break;
621  
622         // drain the waiting tiles queue into the sync slot queues.
623         while (!waitingTiles.empty()) {
624             SyncItem next = waitingTiles.pop_front();
625             if (isPathCached(next)) {
626                 _cache_hits++;
627                 SG_LOG(SG_TERRAIN, SG_DEBUG, "TerraSync Cache hit for: '" << next._dir << "'");
628                 next._status = SyncItem::Cached;
629                 _freshTiles.push_back(next);
630                 _is_dirty = true;
631                 continue;
632             }
633
634             unsigned int slot = syncSlotForType(next._type);
635             _syncSlots[slot].queue.push(next);
636         }
637        
638         bool anySlotBusy = false;
639         // update each sync slot in turn
640         for (unsigned int slot=0; slot < NUM_SYNC_SLOTS; ++slot) {
641             updateSyncSlot(_syncSlots[slot]);
642             anySlotBusy |= _syncSlots[slot].busy;
643         }
644
645         _busy = anySlotBusy;
646         if (!anySlotBusy) {
647             // wait on the blocking deque here, otherwise we spin
648             // the loop very fast, since _http::update with no connections
649             // active returns immediately.
650             waitingTiles.waitOnNotEmpty();
651         }
652     } // of thread running loop
653 }
654
655 bool SGTerraSync::SvnThread::isPathCached(const SyncItem& next) const
656 {
657     CompletedTiles::const_iterator ii = _completedTiles.find( next._dir );
658     if (ii == _completedTiles.end()) {
659         return false;
660     }
661     
662     // check if the path still physically exists
663     SGPath p(_local_dir);
664     p.append(next._dir);
665     if (!p.exists()) {
666         return false;
667     }
668     
669     time_t now = time(0);
670     return (ii->second > now);
671 }
672
673 void SGTerraSync::SvnThread::fail(SyncItem failedItem)
674 {
675     time_t now = time(0);
676     _consecutive_errors++;
677     _fail_count++;
678     failedItem._status = SyncItem::Failed;
679     _freshTiles.push_back(failedItem);
680     _completedTiles[ failedItem._dir ] = now + UpdateInterval::FailedAttempt;
681     _is_dirty = true;
682 }
683
684 void SGTerraSync::SvnThread::updated(SyncItem item, bool isNewDirectory)
685 {
686     time_t now = time(0);
687     _consecutive_errors = 0;
688     _success_count++;
689     SG_LOG(SG_TERRAIN,SG_INFO,
690            "Successfully synchronized directory '" << item._dir << "'");
691     
692     item._status = SyncItem::Updated;
693     if (item._type == SyncItem::Tile) {
694         _updated_tile_count++;
695     }
696
697     _freshTiles.push_back(item);
698     _completedTiles[ item._dir ] = now + UpdateInterval::SuccessfulAttempt;
699     _is_dirty = true;
700     writeCompletedTilesPersistentCache();
701 }
702
703 void SGTerraSync::SvnThread::initCompletedTilesPersistentCache()
704 {
705     if (!_persistentCachePath.exists()) {
706         return;
707     }
708     
709     SGPropertyNode_ptr cacheRoot(new SGPropertyNode);
710     time_t now = time(0);
711     
712     try {
713         readProperties(_persistentCachePath.str(), cacheRoot);
714     } catch (sg_exception& e) {
715         SG_LOG(SG_TERRAIN, SG_INFO, "corrupted persistent cache, discarding");
716         return;
717     }
718     
719     for (int i=0; i<cacheRoot->nChildren(); ++i) {
720         SGPropertyNode* entry = cacheRoot->getChild(i);
721         string tileName = entry->getStringValue("path");
722         time_t stamp = entry->getIntValue("stamp");
723         if (stamp < now) {
724             continue;
725         }
726         
727         _completedTiles[tileName] = stamp;
728     }
729 }
730
731 void SGTerraSync::SvnThread::writeCompletedTilesPersistentCache() const
732 {
733     // cache is disabled
734     if (_persistentCachePath.isNull()) {
735         return;
736     }
737     
738     std::ofstream f(_persistentCachePath.c_str(), std::ios::trunc);
739     if (!f.is_open()) {
740         return;
741     }
742     
743     SGPropertyNode_ptr cacheRoot(new SGPropertyNode);
744     CompletedTiles::const_iterator it = _completedTiles.begin();
745     for (; it != _completedTiles.end(); ++it) {
746         SGPropertyNode* entry = cacheRoot->addChild("entry");
747         entry->setStringValue("path", it->first);
748         entry->setIntValue("stamp", it->second);
749     }
750     
751     writeProperties(f, cacheRoot, true /* write_all */);
752     f.close();
753 }
754
755 ///////////////////////////////////////////////////////////////////////////////
756 // SGTerraSync ////////////////////////////////////////////////////////////////
757 ///////////////////////////////////////////////////////////////////////////////
758 SGTerraSync::SGTerraSync() :
759     _svnThread(NULL),
760     _bound(false),
761     _inited(false)
762 {
763     _svnThread = new SvnThread();
764     _log = new BufferedLogCallback(SG_TERRAIN, SG_INFO);
765     _log->truncateAt(255);
766     
767     sglog().addCallback(_log);
768 }
769
770 SGTerraSync::~SGTerraSync()
771 {
772     delete _svnThread;
773     _svnThread = NULL;
774     sglog().removeCallback(_log);
775     delete _log;
776      _tiedProperties.Untie();
777 }
778
779 void SGTerraSync::setRoot(SGPropertyNode_ptr root)
780 {
781     _terraRoot = root->getNode("/sim/terrasync",true);
782 }
783
784 void SGTerraSync::init()
785 {
786     if (_inited) {
787         return;
788     }
789     
790     _inited = true;
791     
792     assert(_terraRoot);
793     _terraRoot->setBoolValue("built-in-svn-available",svn_built_in_available);
794         
795     reinit();
796 }
797
798 void SGTerraSync::shutdown()
799 {
800      _svnThread->stop();
801 }
802
803 void SGTerraSync::reinit()
804 {
805     // do not reinit when enabled and we're already up and running
806     if ((_terraRoot->getBoolValue("enabled",false))&&
807          (_svnThread->_active && _svnThread->_running))
808     {
809         return;
810     }
811     
812     _svnThread->stop();
813
814     if (_terraRoot->getBoolValue("enabled",false))
815     {
816         _svnThread->setSvnServer(_terraRoot->getStringValue("svn-server",""));
817         _svnThread->setSvnDataServer(_terraRoot->getStringValue("svn-data-server",""));
818         _svnThread->setRsyncServer(_terraRoot->getStringValue("rsync-server",""));
819         _svnThread->setLocalDir(_terraRoot->getStringValue("scenery-dir",""));
820         _svnThread->setAllowedErrorCount(_terraRoot->getIntValue("max-errors",5));
821         _svnThread->setUseBuiltin(_terraRoot->getBoolValue("use-built-in-svn",true));
822         _svnThread->setCachePath(SGPath(_terraRoot->getStringValue("cache-path","")));
823         _svnThread->setCacheHits(_terraRoot->getIntValue("cache-hit", 0));
824         _svnThread->setUseSvn(_terraRoot->getBoolValue("use-svn",true));
825         _svnThread->setExtSvnUtility(_terraRoot->getStringValue("ext-svn-utility","svn"));
826
827         if (_svnThread->start())
828         {
829             syncAirportsModels();
830         }
831     }
832
833     _stalledNode->setBoolValue(_svnThread->_stalled);
834 }
835
836 void SGTerraSync::bind()
837 {
838     if (_bound) {
839         return;
840     }
841     
842     _bound = true;
843     _tiedProperties.Tie( _terraRoot->getNode("busy", true), (bool*) &_svnThread->_busy );
844     _tiedProperties.Tie( _terraRoot->getNode("active", true), (bool*) &_svnThread->_active );
845     _tiedProperties.Tie( _terraRoot->getNode("update-count", true), (int*) &_svnThread->_success_count );
846     _tiedProperties.Tie( _terraRoot->getNode("error-count", true), (int*) &_svnThread->_fail_count );
847     _tiedProperties.Tie( _terraRoot->getNode("tile-count", true), (int*) &_svnThread->_updated_tile_count );
848     _tiedProperties.Tie( _terraRoot->getNode("cache-hits", true), (int*) &_svnThread->_cache_hits );
849     _tiedProperties.Tie( _terraRoot->getNode("transfer-rate-bytes-sec", true), (int*) &_svnThread->_transfer_rate );
850     
851     // use kbytes here because propety doesn't support 64-bit and we might conceivably
852     // download more than 2G in a single session
853     _tiedProperties.Tie( _terraRoot->getNode("downloaded-kbytes", true), (int*) &_svnThread->_total_kb_downloaded );
854     
855     _terraRoot->getNode("busy", true)->setAttribute(SGPropertyNode::WRITE,false);
856     _terraRoot->getNode("active", true)->setAttribute(SGPropertyNode::WRITE,false);
857     _terraRoot->getNode("update-count", true)->setAttribute(SGPropertyNode::WRITE,false);
858     _terraRoot->getNode("error-count", true)->setAttribute(SGPropertyNode::WRITE,false);
859     _terraRoot->getNode("tile-count", true)->setAttribute(SGPropertyNode::WRITE,false);
860     _terraRoot->getNode("use-built-in-svn", true)->setAttribute(SGPropertyNode::USERARCHIVE,false);
861     _terraRoot->getNode("use-svn", true)->setAttribute(SGPropertyNode::USERARCHIVE,false);
862     // stalled is used as a signal handler (to connect listeners triggering GUI pop-ups)
863     _stalledNode = _terraRoot->getNode("stalled", true);
864     _stalledNode->setBoolValue(_svnThread->_stalled);
865     _stalledNode->setAttribute(SGPropertyNode::PRESERVE,true);
866 }
867
868 void SGTerraSync::unbind()
869 {
870     _svnThread->stop();
871     _tiedProperties.Untie();
872     _bound = false;
873     _inited = false;
874     
875     _terraRoot.clear();
876     _stalledNode.clear();
877     _cacheHits.clear();
878 }
879
880 void SGTerraSync::update(double)
881 {
882     static SGBucket bucket;
883     if (_svnThread->isDirty())
884     {
885         if (!_svnThread->_active)
886         {
887             if (_svnThread->_stalled)
888             {
889                 SG_LOG(SG_TERRAIN,SG_ALERT,
890                        "Automatic scenery download/synchronization stalled. Too many errors.");
891             }
892             else
893             {
894                 // not really an alert - just always show this message
895                 SG_LOG(SG_TERRAIN,SG_ALERT,
896                         "Automatic scenery download/synchronization has stopped.");
897             }
898             _stalledNode->setBoolValue(_svnThread->_stalled);
899         }
900
901         while (_svnThread->hasNewTiles())
902         {
903             SyncItem next = _svnThread->getNewTile();
904             
905             if ((next._type == SyncItem::Tile) || (next._type == SyncItem::AIData)) {
906                 _activeTileDirs.erase(next._dir);
907             }
908         } // of freshly synced items
909     }
910 }
911
912 bool SGTerraSync::isIdle() {return _svnThread->isIdle();}
913
914 void SGTerraSync::syncAirportsModels()
915 {
916     static const char* bounds = "MZAJKL"; // airport sync order: K-L, A-J, M-Z
917     // note "request" method uses LIFO order, i.e. processes most recent request first
918     for( unsigned i = 0; i < strlen(bounds)/2; i++ )
919     {
920         for ( char synced_other = bounds[2*i]; synced_other <= bounds[2*i+1]; synced_other++ )
921         {
922             ostringstream dir;
923             dir << "Airports/" << synced_other;
924             SyncItem w(dir.str(), SyncItem::AirportData);
925             _svnThread->request( w );
926         }
927     }
928     
929     SyncItem w("Models", SyncItem::SharedModels);
930     _svnThread->request( w );
931 }
932
933 void SGTerraSync::syncAreaByPath(const std::string& aPath)
934 {
935     const char* terrainobjects[3] = { "Terrain/", "Objects/",  0 };
936     for (const char** tree = &terrainobjects[0]; *tree; tree++)
937     {
938         std::string dir = string(*tree) + aPath;
939         if (_activeTileDirs.find(dir) != _activeTileDirs.end()) {
940             continue;
941         }
942                 
943         _activeTileDirs.insert(dir);
944         SyncItem w(dir, SyncItem::Tile);
945         _svnThread->request( w );
946     }
947 }
948
949 bool SGTerraSync::scheduleTile(const SGBucket& bucket)
950 {
951     std::string basePath = bucket.gen_base_path();
952     syncAreaByPath(basePath);
953     return true;
954 }
955
956 bool SGTerraSync::isTileDirPending(const std::string& sceneryDir) const
957 {
958     if (!_svnThread->_running) {
959         return false;
960     }
961     
962     const char* terrainobjects[3] = { "Terrain/", "Objects/",  0 };
963     for (const char** tree = &terrainobjects[0]; *tree; tree++) {
964         string s = *tree + sceneryDir;
965         if (_activeTileDirs.find(s) != _activeTileDirs.end()) {
966             return true;
967         }
968     }
969
970     return false;
971 }
972
973 void SGTerraSync::scheduleDataDir(const std::string& dataDir)
974 {
975     if (_activeTileDirs.find(dataDir) != _activeTileDirs.end()) {
976         return;
977     }
978     
979     _activeTileDirs.insert(dataDir);
980     SyncItem w(dataDir, SyncItem::AIData);
981     _svnThread->request( w );
982
983 }
984
985 bool SGTerraSync::isDataDirPending(const std::string& dataDir) const
986 {
987     if (!_svnThread->_running) {
988         return false;
989     }
990     
991     return (_activeTileDirs.find(dataDir) != _activeTileDirs.end());
992 }
993
994 void SGTerraSync::reposition()
995 {
996     // stub, remove
997 }