#include <string>
#include <vector>
#include <algorithm>
+#include <boost/foreach.hpp>
#include <simgear/compiler.h>
#include <simgear/misc/sg_path.hxx>
/******************************************************************************
* TrafficManager
*****************************************************************************/
-FGTrafficManager::FGTrafficManager()
+FGTrafficManager::FGTrafficManager() :
+ inited(false),
+ doingInit(false),
+ enabled("/sim/traffic-manager/enabled"),
+ aiEnabled("/sim/ai/enabled"),
+ realWxEnabled("/environment/realwx/enabled"),
+ metarValid("/environment/metar/valid")
{
//score = 0;
//runCount = 0;
//cerr << "Saving AI traffic heuristics" << endl;
saveData = true;
cachefile.open(cacheData.str().c_str());
+ cachefile << "[TrafficManagerCachedata:ref:2011:09:04]" << endl;
}
}
for (ScheduleVectorIterator sched = scheduledAircraft.begin();
if (saveData) {
cachefile << (*sched)->getRegistration() << " "
<< (*sched)->getRunCount() << " "
- << (*sched)->getHits() << endl;
+ << (*sched)->getHits() << " "
+ << (*sched)->getLastUsed() << endl;
}
delete(*sched);
}
void FGTrafficManager::init()
{
- heuristicsVector heuristics;
- HeuristicMap heurMap;
+ if (!enabled || !aiEnabled) {
+ return;
+ }
+ assert(!doingInit);
+ doingInit = true;
if (string(fgGetString("/sim/traffic-manager/datafile")) == string("")) {
simgear::Dir trafficDir(SGPath(globals->get_fg_root(), "AI/Traffic"));
simgear::PathList d = trafficDir.children(simgear::Dir::TYPE_DIR | simgear::Dir::NO_DOT_OR_DOTDOT);
- for (unsigned int i=0; i<d.size(); ++i) {
- simgear::Dir d2(d[i]);
+ BOOST_FOREACH(SGPath p, d) {
+ simgear::Dir d2(p);
simgear::PathList trafficFiles = d2.children(simgear::Dir::TYPE_FILE, ".xml");
- for (unsigned int j=0; j<trafficFiles.size(); ++j) {
- SGPath curFile = trafficFiles[j];
- SG_LOG(SG_GENERAL, SG_DEBUG,
- "Scanning " << curFile.str() << " for traffic");
- readXML(curFile.str(), *this);
- }
+ schedulesToRead.insert(schedulesToRead.end(), trafficFiles.begin(), trafficFiles.end());
}
} else {
fgSetBool("/sim/traffic-manager/heuristics", false);
readTimeTableFromFile(path);
}
} else {
- SG_LOG(SG_GENERAL, SG_ALERT,
+ SG_LOG(SG_AI, SG_ALERT,
"Unknown data format " << path.str()
<< " for traffic");
}
//exit(1);
}
- if (fgGetBool("/sim/traffic-manager/heuristics")) {
- //cerr << "Processing Heuristics" << endl;
- // Load the heuristics data
- SGPath cacheData(fgGetString("/sim/fg-home"));
- cacheData.append("ai");
- string airport = fgGetString("/sim/presets/airport-id");
- if ((airport) != "") {
- char buffer[128];
- ::snprintf(buffer, 128, "%c/%c/%c/",
- airport[0], airport[1], airport[2]);
- cacheData.append(buffer);
- cacheData.append(airport + "-cache.txt");
- if (cacheData.exists()) {
- ifstream data(cacheData.c_str());
- while (1) {
- Heuristic h; // = new Heuristic;
- data >> h.registration >> h.runCount >> h.hits;
- if (data.eof())
- break;
- HeuristicMapIterator itr = heurMap.find(h.registration);
- if (itr != heurMap.end()) {
- SG_LOG(SG_GENERAL, SG_WARN,"Traffic Manager Warning: found duplicate tailnumber " <<
- h.registration << " for AI aircraft");
- }
- heurMap[h.registration] = h;
- heuristics.push_back(h);
- }
- }
- }
- for (currAircraft = scheduledAircraft.begin();
- currAircraft != scheduledAircraft.end(); currAircraft++) {
- string registration = (*currAircraft)->getRegistration();
- HeuristicMapIterator itr = heurMap.find(registration);
- //cerr << "Processing heuristics for" << (*currAircraft)->getRegistration() << endl;
- if (itr == heurMap.end()) {
- //cerr << "No heuristics found for " << registration << endl;
- } else {
- (*currAircraft)->setrunCount(itr->second.runCount);
- (*currAircraft)->setHits(itr->second.hits);
- //cerr <<"Runcount " << itr->second->runCount << ".Hits " << itr->second->hits << endl;
- }
- }
- //cerr << "Done" << endl;
- //for (heuristicsVectorIterator hvi = heuristics.begin();
- // hvi != heuristics.end(); hvi++) {
- // delete(*hvi);
- //}
+}
+
+void FGTrafficManager::initStep()
+{
+ assert(doingInit);
+ if (schedulesToRead.empty()) {
+ finishInit();
+ return;
}
+
+ SGPath path = schedulesToRead.front();
+ schedulesToRead.erase(schedulesToRead.begin());
+ SG_LOG(SG_AI, SG_DEBUG, path << " for traffic");
+ readXML(path.str(), *this);
+}
+
+void FGTrafficManager::finishInit()
+{
+ assert(doingInit);
+ SG_LOG(SG_AI, SG_INFO, "finishing AI-Traffic init");
+ loadHeuristics();
+
// Do sorting and scoring separately, to take advantage of the "homeport| variable
for (currAircraft = scheduledAircraft.begin();
currAircraft != scheduledAircraft.end(); currAircraft++) {
(*currAircraft)->setScore();
}
+
sort(scheduledAircraft.begin(), scheduledAircraft.end(),
compareSchedules);
currAircraft = scheduledAircraft.begin();
currAircraftClosest = scheduledAircraft.begin();
+
+ doingInit = false;
+ inited = true;
+}
+
+void FGTrafficManager::loadHeuristics()
+{
+ if (!fgGetBool("/sim/traffic-manager/heuristics")) {
+ return;
+ }
+
+ HeuristicMap heurMap;
+ //cerr << "Processing Heuristics" << endl;
+ // Load the heuristics data
+ SGPath cacheData(fgGetString("/sim/fg-home"));
+ cacheData.append("ai");
+ string airport = fgGetString("/sim/presets/airport-id");
+ if ((airport) != "") {
+ char buffer[128];
+ ::snprintf(buffer, 128, "%c/%c/%c/",
+ airport[0], airport[1], airport[2]);
+ cacheData.append(buffer);
+ cacheData.append(airport + "-cache.txt");
+ string revisionStr;
+ if (cacheData.exists()) {
+ ifstream data(cacheData.c_str());
+ data >> revisionStr;
+ if (revisionStr != "[TrafficManagerCachedata:ref:2011:09:04]") {
+ SG_LOG(SG_GENERAL, SG_ALERT,"Traffic Manager Warning: discarding outdated cachefile " <<
+ cacheData.c_str() << " for Airport " << airport);
+ } else {
+ while (1) {
+ Heuristic h; // = new Heuristic;
+ data >> h.registration >> h.runCount >> h.hits >> h.lastRun;
+ if (data.eof())
+ break;
+ HeuristicMapIterator itr = heurMap.find(h.registration);
+ if (itr != heurMap.end()) {
+ SG_LOG(SG_GENERAL, SG_WARN,"Traffic Manager Warning: found duplicate tailnumber " <<
+ h.registration << " for AI aircraft");
+ } else {
+ heurMap[h.registration] = h;
+ }
+ }
+ }
+ }
+ }
+
+ for(currAircraft = scheduledAircraft.begin(); currAircraft != scheduledAircraft.end(); ++currAircraft) {
+ string registration = (*currAircraft)->getRegistration();
+ HeuristicMapIterator itr = heurMap.find(registration);
+ if (itr != heurMap.end()) {
+ (*currAircraft)->setrunCount(itr->second.runCount);
+ (*currAircraft)->setHits(itr->second.hits);
+ (*currAircraft)->setLastUsed(itr->second.lastRun);
+ }
+ }
}
void FGTrafficManager::update(double /*dt */ )
{
- if (fgGetBool("/environment/metar/valid") == false) {
+ if (!enabled || !aiEnabled || (realWxEnabled && !metarValid)) {
return;
}
+
+ if (!inited) {
+ if (!doingInit) {
+ init();
+ }
+
+ initStep();
+ if (!inited) {
+ return; // still more to do on next update() call
+ }
+ }
+
time_t now = time(NULL) + fgGetLong("/sim/time/warp");
if (scheduledAircraft.size() == 0) {
return;
}
- SGVec3d userCart =
- SGVec3d::fromGeod(SGGeod::
- fromDeg(fgGetDouble("/position/longitude-deg"),
- fgGetDouble("/position/latitude-deg")));
+ SGVec3d userCart = globals->get_aircraft_positon_cart();
if (currAircraft == scheduledAircraft.end()) {
currAircraft = scheduledAircraft.begin();
}
//cerr << "Processing << " << (*currAircraft)->getRegistration() << " with score " << (*currAircraft)->getScore() << endl;
if (!((*currAircraft)->update(now, userCart))) {
- // NOTE: With traffic manager II, this statement below is no longer true
- // after proper initialization, we shouldnt get here.
- // But let's make sure
- //SG_LOG( SG_GENERAL, SG_ALERT, "Failed to update aircraft schedule in traffic manager");
+ (*currAircraft)->taint();
}
currAircraft++;
}