]> git.mxchange.org Git - flightgear.git/blobdiff - src/AIModel/AIManager.cxx
Ground network distance tracking code. AIAircraft taxiing at airports
[flightgear.git] / src / AIModel / AIManager.cxx
index d5d6392920f6a03ce062aa2b5f8bbeca25963620..207913bb1ff5aae959e33dc829126c7c8c79ddb2 100644 (file)
 //
 // 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., 675 Mass Ave, Cambridge, MA 02139, USA.
+// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 
-#include <simgear/misc/sg_path.hxx>
-#include <Main/fg_props.hxx>
 #include <Main/globals.hxx>
 
 #include <Airports/simple.hxx>
-#include <Traffic/SchedFlight.hxx>
-#include <Traffic/Schedule.hxx>
 #include <Traffic/TrafficMgr.hxx>
 
-#include <list>
-
 #include "AIManager.hxx"
 #include "AIAircraft.hxx"
 #include "AIShip.hxx"
 #include "AIStorm.hxx"
 #include "AIThermal.hxx"
 #include "AICarrier.hxx"
-
-SG_USING_STD(list);
-
+#include "AIStatic.hxx"
+#include "AIMultiplayer.hxx"
 
 FGAIManager::FGAIManager() {
-  initDone = false;
-  for (int i=0; i < FGAIBase::MAX_OBJECTS; i++)
-     numObjects[i] = 0;
   _dt = 0.0;
-  dt_count = 9;
-  scenario_filename = "";
-  ai_list.clear();
+  mNumAiModels = 0;
+  for (unsigned i = 0; i < FGAIBase::MAX_OBJECTS; ++i)
+    mNumAiTypeModels[i] = 0;
 }
 
 FGAIManager::~FGAIManager() {
   ai_list_iterator ai_list_itr = ai_list.begin();
   while(ai_list_itr != ai_list.end()) {
       (*ai_list_itr)->unbind();
-      delete (*ai_list_itr);
       ++ai_list_itr;
-    }
-  ai_list.clear();
-  ModelVecIterator i = loadedModels.begin();
-  while (i != loadedModels.end())
-    {
-      i->getModelId()->deRef();
-    }
+  }
 }
 
 
@@ -74,6 +57,9 @@ void FGAIManager::init() {
       return;
 
   wind_from_down_node = fgGetNode("/environment/wind-from-down-fps", true);
+  wind_from_east_node  = fgGetNode("/environment/wind-from-east-fps",true);
+  wind_from_north_node = fgGetNode("/environment/wind-from-north-fps",true);
+
   user_latitude_node  = fgGetNode("/position/latitude-deg", true);
   user_longitude_node = fgGetNode("/position/longitude-deg", true);
   user_altitude_node  = fgGetNode("/position/altitude-ft", true);
@@ -81,17 +67,29 @@ void FGAIManager::init() {
   user_pitch_node     = fgGetNode("/orientation/pitch-deg", true);
   user_yaw_node       = fgGetNode("/orientation/side-slip-deg", true);
   user_speed_node     = fgGetNode("/velocities/uBody-fps", true);
-  scenario_filename = root->getNode("scenario", true)->getStringValue();
-  if (scenario_filename != "") processScenario( scenario_filename );
 
-  initDone = true;
+
+  /// Move that into the constructor
+  for(int i = 0 ; i < root->nChildren() ; i++) {
+    SGPropertyNode *aiEntry = root->getChild( i );
+    if( !strcmp( aiEntry->getName(), "scenario" ) ) {
+      scenario_filename = aiEntry->getStringValue();
+      if (!scenario_filename.empty())
+        processScenario( scenario_filename );
+    }
+  }
+}
+
+
+void FGAIManager::reinit() {
+   update(0.0);
 }
 
 
 void FGAIManager::bind() {
    root = globals->get_props()->getNode("ai/models", true);
-   root->tie("count", SGRawValuePointer<int>(&numObjects[0]));
+   root->tie("count", SGRawValueMethods<FGAIManager, int>(*this,
+             &FGAIManager::getNumAiObjects));
 }
 
 
@@ -101,232 +99,72 @@ void FGAIManager::unbind() {
 
 
 void FGAIManager::update(double dt) {
+  // initialize these for finding nearest thermals
+  range_nearest = 10000.0;
+  strength = 0.0;
+  if (!enabled)
+    return;
 
-        // initialize these for finding nearest thermals
-        range_nearest = 10000.0;
-        strength = 0.0;
-       FGTrafficManager *tmgr = (FGTrafficManager*) globals->get_subsystem("Traffic Manager");
-
-        if (!enabled)
-            return;
-
-        _dt = dt;
-
-        ai_list_iterator ai_list_itr = ai_list.begin();
-        while(ai_list_itr != ai_list.end()) {
-                if ((*ai_list_itr)->getDie()) {      
-                 tmgr->release((*ai_list_itr)->getID());
-                   --numObjects[(*ai_list_itr)->getType()];
-                   --numObjects[0];
-                   (*ai_list_itr)->unbind();
-                   delete (*ai_list_itr);
-                   if ( ai_list_itr == ai_list.begin() ) {
-                       ai_list.erase(ai_list_itr);
-                       ai_list_itr = ai_list.begin();
-                       continue;
-                   } else {
-                       ai_list.erase(ai_list_itr--);
-                   }
-                } else {
-                   fetchUserState();
-                   if ((*ai_list_itr)->isa(FGAIBase::otThermal)) {
-                       processThermal((FGAIThermal*)*ai_list_itr); 
-                   } else { 
-                      (*ai_list_itr)->update(_dt);
-                   }
-                }
-                ++ai_list_itr;
-        }
-        wind_from_down_node->setDoubleValue( strength ); // for thermals
-
-}
-
-
-void*
-FGAIManager::createAircraft( FGAIModelEntity *entity,   FGAISchedule *ref) {
-     
-        FGAIAircraft* ai_plane = new FGAIAircraft(this, ref);
-        ai_list.push_back(ai_plane);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otAircraft];
-        if (entity->m_class == "light") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::LIGHT]);
-        } else if (entity->m_class == "ww2_fighter") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::WW2_FIGHTER]);
-        } else if (entity->m_class ==  "jet_transport") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-        } else if (entity->m_class == "jet_fighter") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_FIGHTER]);
-        } else if (entity->m_class ==  "tanker") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-          ai_plane->SetTanker(true);
-        } else {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-        }
-       ai_plane->setAcType(entity->acType);
-       ai_plane->setCompany(entity->company);
-        ai_plane->setHeading(entity->heading);
-        ai_plane->setSpeed(entity->speed);
-        ai_plane->setPath(entity->path.c_str());
-        ai_plane->setAltitude(entity->altitude);
-        ai_plane->setLongitude(entity->longitude);
-        ai_plane->setLatitude(entity->latitude);
-        ai_plane->setBank(entity->roll);
-
-        if ( entity->fp ) {
-          ai_plane->SetFlightPlan(entity->fp);
-        }
-
-        ai_plane->init();
-        ai_plane->bind();
-        return ai_plane;
-}
-
-void*
-FGAIManager::createShip( FGAIModelEntity *entity ) {
-    
-     // cout << "creating ship" << endl;    
-
-        FGAIShip* ai_ship = new FGAIShip(this);
-        ai_list.push_back(ai_ship);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otShip];
-        ai_ship->setHeading(entity->heading);
-        ai_ship->setSpeed(entity->speed);
-        ai_ship->setPath(entity->path.c_str());
-        ai_ship->setAltitude(entity->altitude);
-        ai_ship->setLongitude(entity->longitude);
-        ai_ship->setLatitude(entity->latitude);
-        ai_ship->setBank(entity->rudder);
-        ai_ship->setName(entity->name);
-
-        if ( entity->fp ) {
-           ai_ship->setFlightPlan(entity->fp);
-        }
-
-        ai_ship->init();
-        ai_ship->bind();
-        return ai_ship;
-}
-
-void*
-FGAIManager::createCarrier( FGAIModelEntity *entity ) {
-    
-    // cout << "creating carrier" << endl;
-
-        FGAICarrier* ai_carrier = new FGAICarrier(this);
-        ai_list.push_back(ai_carrier);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otCarrier];
-        ai_carrier->setHeading(entity->heading);
-        ai_carrier->setSpeed(entity->speed);
-        ai_carrier->setPath(entity->path.c_str());
-        ai_carrier->setAltitude(entity->altitude);
-        ai_carrier->setLongitude(entity->longitude);
-        ai_carrier->setLatitude(entity->latitude);
-        ai_carrier->setBank(entity->rudder);
-        ai_carrier->setSolidObjects(entity->solid_objects);
-        ai_carrier->setWireObjects(entity->wire_objects);
-        ai_carrier->setCatapultObjects(entity->catapult_objects);
-        ai_carrier->setParkingPositions(entity->ppositions);
-        ai_carrier->setRadius(entity->radius);
-        ai_carrier->setSign(entity->pennant_number);
-        ai_carrier->setName(entity->name);
-        ai_carrier->setFlolsOffset(entity->flols_offset);
-
-        if ( entity->fp ) {
-           ai_carrier->setFlightPlan(entity->fp);
-        }
+  FGTrafficManager *tmgr = (FGTrafficManager*) globals->get_subsystem("Traffic Manager");
+  _dt = dt;
 
-        ai_carrier->init();
-        ai_carrier->bind();
-        return ai_carrier;
+  ai_list_iterator ai_list_itr = ai_list.begin();
+  while(ai_list_itr != ai_list.end()) {
+    if ((*ai_list_itr)->getDie()) {      
+      tmgr->release((*ai_list_itr)->getID());
+      --mNumAiModels;
+      --(mNumAiTypeModels[(*ai_list_itr)->getType()]);
+      (*ai_list_itr)->unbind();
+      ai_list_itr = ai_list.erase(ai_list_itr);
+    } else {
+      fetchUserState();
+      if ((*ai_list_itr)->isa(FGAIBase::otThermal)) {
+        FGAIBase *base = *ai_list_itr;
+        processThermal((FGAIThermal*)base); 
+      } else { 
+        (*ai_list_itr)->update(_dt);
+      }
+      ++ai_list_itr;
+    }
+  }
+  wind_from_down_node->setDoubleValue( strength ); // for thermals
 }
 
-void*
-FGAIManager::createBallistic( FGAIModelEntity *entity ) {
-
-        FGAIBallistic* ai_ballistic = new FGAIBallistic(this);
-        ai_list.push_back(ai_ballistic);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otBallistic];
-        ai_ballistic->setAzimuth(entity->azimuth);
-        ai_ballistic->setElevation(entity->elevation);
-        ai_ballistic->setSpeed(entity->speed);
-        ai_ballistic->setPath(entity->path.c_str());
-        ai_ballistic->setAltitude(entity->altitude);
-        ai_ballistic->setLongitude(entity->longitude);
-        ai_ballistic->setLatitude(entity->latitude);
-        ai_ballistic->setDragArea(entity->eda);
-        ai_ballistic->setLife(entity->life);
-        ai_ballistic->setBuoyancy(entity->buoyancy);
-        ai_ballistic->setWind_from_east(entity->wind_from_east);
-        ai_ballistic->setWind_from_north(entity->wind_from_north);
-        ai_ballistic->setWind(entity->wind);
-        ai_ballistic->setRoll(entity->roll);
-        ai_ballistic->setCd(entity->cd);
-        ai_ballistic->setMass(entity->mass);
-        ai_ballistic->setStabilisation(entity->aero_stabilised);
-        ai_ballistic->init();
-        ai_ballistic->bind();
-        return ai_ballistic;
+void
+FGAIManager::attach(SGSharedPtr<FGAIBase> model)
+{
+  unsigned idx = mNumAiTypeModels[model->getType()];
+  const char* typeString = model->getTypeString();
+  SGPropertyNode* root = globals->get_props()->getNode("ai/models", true);
+  SGPropertyNode* p = root->getNode(typeString, idx, true);
+  model->setManager(this, p);
+  ai_list.push_back(model);
+  ++mNumAiModels;
+  ++(mNumAiTypeModels[model->getType()]);
+  model->init();
+  model->bind();
 }
 
-void*
-FGAIManager::createStorm( FGAIModelEntity *entity ) {
-
-        FGAIStorm* ai_storm = new FGAIStorm(this);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otStorm];
-        ai_storm->setHeading(entity->heading);
-        ai_storm->setSpeed(entity->speed);
-        ai_storm->setPath(entity->path.c_str());
-        ai_storm->setAltitude(entity->altitude); 
-        ai_storm->setDiameter(entity->diameter / 6076.11549); 
-        ai_storm->setHeight(entity->height_msl); 
-        ai_storm->setStrengthNorm(entity->strength); 
-        ai_storm->setLongitude(entity->longitude);
-        ai_storm->setLatitude(entity->latitude);
-        ai_storm->init();
-        ai_storm->bind();
-        ai_list.push_back(ai_storm);
-        return ai_storm;
-}
 
-void*
-FGAIManager::createThermal( FGAIModelEntity *entity ) {
-
-        FGAIThermal* ai_thermal = new FGAIThermal(this);
-        ++numObjects[0];
-        ++numObjects[FGAIBase::otThermal];
-        ai_thermal->setLongitude(entity->longitude);
-        ai_thermal->setLatitude(entity->latitude);
-        ai_thermal->setMaxStrength(entity->strength);
-        ai_thermal->setDiameter(entity->diameter / 6076.11549);
-        ai_thermal->setHeight(entity->height_msl);
-        ai_thermal->init();
-        ai_thermal->bind();
-        ai_list.push_back(ai_thermal);
-        return ai_thermal;
+void FGAIManager::destroyObject( int ID ) {
+  ai_list_iterator ai_list_itr = ai_list.begin();
+  while(ai_list_itr != ai_list.end()) {
+    if ((*ai_list_itr)->getID() == ID) {
+      --mNumAiModels;
+      --(mNumAiTypeModels[(*ai_list_itr)->getType()]);
+      (*ai_list_itr)->unbind();
+      ai_list_itr = ai_list.erase(ai_list_itr);
+    } else
+      ++ai_list_itr;
+  }
 }
 
-void FGAIManager::destroyObject( void* ID ) {
-        ai_list_iterator ai_list_itr = ai_list.begin();
-        while(ai_list_itr != ai_list.end()) {
-            if ((*ai_list_itr)->getID() == ID) {
-              --numObjects[0];
-              --numObjects[(*ai_list_itr)->getType()];
-              (*ai_list_itr)->unbind();
-              delete (*ai_list_itr);
-              ai_list.erase(ai_list_itr);
-
-              break;
-            }
-            ++ai_list_itr;
-        }
+int
+FGAIManager::getNumAiObjects(void) const
+{
+  return mNumAiModels;
 }
 
-
 void FGAIManager::fetchUserState( void ) {
      user_latitude  = user_latitude_node->getDoubleValue();
      user_longitude = user_longitude_node->getDoubleValue();
@@ -335,6 +173,8 @@ void FGAIManager::fetchUserState( void ) {
      user_pitch     = user_pitch_node->getDoubleValue();
      user_yaw       = user_yaw_node->getDoubleValue();
      user_speed     = user_speed_node->getDoubleValue() * 0.592484;
+     wind_from_east = wind_from_east_node->getDoubleValue();
+     wind_from_north = wind_from_north_node->getDoubleValue();
 }
 
 
@@ -348,40 +188,78 @@ void FGAIManager::processThermal( FGAIThermal* thermal ) {
 }
 
 
-void FGAIManager::processScenario( string &filename ) {
-  FGAIScenario* s = new FGAIScenario( filename );
-  for (int i=0;i<s->nEntries();i++) {
-    FGAIModelEntity* en = s->getNextEntry();
-
-    if (en) {
-      if ( en->m_type == "aircraft") {
-        createAircraft( en );
-
-      } else if ( en->m_type == "ship") {
-        createShip( en );
-
-      } else if ( en->m_type == "carrier") {
-        createCarrier( en );
-
-      } else if ( en->m_type == "thunderstorm") {
-        createStorm( en );
-
-      } else if ( en->m_type == "thermal") {
-        createThermal( en );
-
-      } else if ( en->m_type == "ballistic") {
-        createBallistic( en );
-      }      
+void FGAIManager::processScenario( const string &filename ) {
+
+  SGPropertyNode_ptr scenarioTop = loadScenarioFile(filename);
+  if (!scenarioTop)
+    return;
+  SGPropertyNode* scenarios = scenarioTop->getChild("scenario");
+  if (!scenarios)
+    return;
+
+  for (int i = 0; i < scenarios->nChildren(); i++) { 
+    SGPropertyNode* scEntry = scenarios->getChild(i);
+    std::string type = scEntry->getStringValue("type", "aircraft");
+
+    if (type == "aircraft") {
+      FGAIAircraft* aircraft = new FGAIAircraft;
+      aircraft->readFromScenario(scEntry);
+      attach(aircraft);
+
+    } else if (type == "ship") {
+      FGAIShip* ship = new FGAIShip;
+      ship->readFromScenario(scEntry);
+      attach(ship);
+      
+    } else if (type == "carrier") {
+      FGAICarrier* carrier = new FGAICarrier;
+      carrier->readFromScenario(scEntry);
+      attach(carrier);
+      
+    } else if (type == "thunderstorm") {
+      FGAIStorm* storm = new FGAIStorm;
+      storm->readFromScenario(scEntry);
+      attach(storm);
+      
+    } else if (type == "thermal") {
+      FGAIThermal* thermal = new FGAIThermal;
+      thermal->readFromScenario(scEntry);
+      attach(thermal);
+      
+    } else if (type == "ballistic") {
+      FGAIBallistic* ballistic = new FGAIBallistic;
+      ballistic->readFromScenario(scEntry);
+      attach(ballistic);
+      
+    } else if (type == "static") {
+      FGAIStatic* aistatic = new FGAIStatic;
+      aistatic->readFromScenario(scEntry);
+      attach(aistatic);
+      
     }
   }
+}
 
-  delete s;
+SGPropertyNode_ptr
+FGAIManager::loadScenarioFile(const std::string& filename)
+{
+  SGPath path(globals->get_fg_root());
+  path.append("AI/" + filename + ".xml");
+  try {
+    SGPropertyNode_ptr root = new SGPropertyNode;
+    readProperties(path.str(), root);
+    return root;
+  } catch (const sg_exception &e) {
+    SG_LOG(SG_GENERAL, SG_ALERT, "Incorrect path specified for AI "
+           "scenario: \"" << path.str() << "\"");
+    return 0;
+  }
 }
 
 // This code keeps track of models that have already been loaded
 // Eventually we'd prbably need to find a way to keep track of models
 // that are unloaded again
-ssgBranch * FGAIManager::getModel(const string& path)
+ssgBranch * FGAIManager::getModel(const string& path) const
 {
   ModelVecIterator i = loadedModels.begin();
   while (i != loadedModels.end())
@@ -398,5 +276,42 @@ void FGAIManager::setModel(const string& path, ssgBranch *model)
   loadedModels.push_back(FGModelID(path,model));
 }
 
+bool FGAIManager::getStartPosition(const string& id, const string& pid,
+                                   SGGeod& geodPos, double& hdng, SGVec3d& uvw)
+{
+  bool found = false;
+  SGPropertyNode* root = fgGetNode("sim/ai", true);
+  if (!root->getNode("enabled", true)->getBoolValue())
+      return found;
+
+  for(int i = 0 ; (!found) && i < root->nChildren() ; i++) {
+    SGPropertyNode *aiEntry = root->getChild( i );
+    if( !strcmp( aiEntry->getName(), "scenario" ) ) {
+      string filename = aiEntry->getStringValue();
+      SGPropertyNode_ptr scenarioTop = loadScenarioFile(filename);
+      if (scenarioTop) {
+        SGPropertyNode* scenarios = scenarioTop->getChild("scenario");
+        if (scenarios) {
+          for (int i = 0; i < scenarios->nChildren(); i++) { 
+            SGPropertyNode* scEntry = scenarios->getChild(i);
+            std::string type = scEntry->getStringValue("type");
+            std::string pnumber = scEntry->getStringValue("pennant-number");
+            std::string name = scEntry->getStringValue("name");
+            if (type == "carrier" && (pnumber == id || name == id)) {
+              SGSharedPtr<FGAICarrier> carrier = new FGAICarrier;
+              carrier->readFromScenario(scEntry);
+              
+              if (carrier->getParkPosition(pid, geodPos, hdng, uvw)) {
+                found = true;
+                break;
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+  return found;
+}
 
 //end AIManager.cxx