]> git.mxchange.org Git - flightgear.git/blobdiff - src/AIModel/AIManager.cxx
Replace the data/Airports/basic.dat.gz and data/Airports/runways.dat.gz with
[flightgear.git] / src / AIModel / AIManager.cxx
index b53ac439b3686443dfcff2a3bd6ba7e9d62dde15..25f5425c7793cc5e0e9e9370084f7681e6c8bb6a 100644 (file)
@@ -1,5 +1,5 @@
 // AIManager.cxx  Based on David Luff's AIMgr:
-// - a global management class for AI objects
+// - a global management type for AI objects
 //
 // Written by David Culp, started October 2003.
 // - davidculp2@comcast.net
 #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 "AIBallistic.hxx"
 #include "AIStorm.hxx"
 #include "AIThermal.hxx"
+#include "AICarrier.hxx"
 
 SG_USING_STD(list);
 
 
 FGAIManager::FGAIManager() {
   initDone = false;
-  numObjects = 0;
+  for (int i=0; i < FGAIBase::MAX_OBJECTS; i++)
+     numObjects[i] = 0;
   _dt = 0.0;
   dt_count = 9;
   scenario_filename = "";
+  ai_list.clear();
 }
 
 FGAIManager::~FGAIManager() {
   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();
-  ids.clear();
+  ModelVecIterator i = loadedModels.begin();
+  while (i != loadedModels.end())
+    {
+      i->getModelId()->deRef();
+    }
 }
 
 
 void FGAIManager::init() {
-  int rval;
   root = fgGetNode("sim/ai", true);
 
   enabled = root->getNode("enabled", true)->getBoolValue();
   if (!enabled)
       return;
 
-
-  wind_from_down = fgGetNode("/environment/wind-from-down-fps", true);
-
-  for (int i = 0; i < root->nChildren(); i++) {
-    const SGPropertyNode * entry = root->getChild(i);
-
-    if (!strcmp(entry->getName(), "scenario")){
-      scenario_filename = entry->getStringValue();
-    }
-
-    if (!strcmp(entry->getName(), "entry")) {
-      if (!strcmp(entry->getStringValue("type", ""), "aircraft")) { 
-
-        rval = createAircraft( entry->getStringValue("class", ""),
-                               entry->getStringValue("path"),
-                               entry->getDoubleValue("latitude"),
-                               entry->getDoubleValue("longitude"),
-                               entry->getDoubleValue("altitude-ft"),
-                               entry->getDoubleValue("heading"),
-                               entry->getDoubleValue("speed-KTAS"),
-                               0.0, 
-                               entry->getDoubleValue("bank") );
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "ship")) {
-
-        rval = createShip( entry->getStringValue("path"),
-                           entry->getDoubleValue("latitude"),
-                           entry->getDoubleValue("longitude"),
-                           entry->getDoubleValue("altitude-ft"),
-                           entry->getDoubleValue("heading"),
-                           entry->getDoubleValue("speed-KTAS"),
-                           entry->getDoubleValue("rudder") );
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "ballistic")) {
-
-        rval = createBallistic( entry->getStringValue("path"),
-                                entry->getDoubleValue("latitude"),
-                                entry->getDoubleValue("longitude"),
-                                entry->getDoubleValue("altitude-ft"),
-                                entry->getDoubleValue("azimuth"),
-                                entry->getDoubleValue("elevation"),
-                                entry->getDoubleValue("speed") );
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "storm")) {
-
-        rval = createStorm( entry->getStringValue("path"),
-                            entry->getDoubleValue("latitude"),
-                            entry->getDoubleValue("longitude"),
-                            entry->getDoubleValue("altitude-ft"),
-                            entry->getDoubleValue("heading"),
-                            entry->getDoubleValue("speed-KTAS") );
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "thermal")) {
-
-        rval = createThermal( entry->getDoubleValue("latitude"),
-                              entry->getDoubleValue("longitude"),
-                              entry->getDoubleValue("strength-fps"),
-                              entry->getDoubleValue("diameter-ft") );
-
-      }       
-    }
-  }
-
+  wind_from_down_node = fgGetNode("/environment/wind-from-down-fps", true);
+  scenario_filename = root->getNode("scenario", true)->getStringValue();
   if (scenario_filename != "") processScenario( scenario_filename );
+
   initDone = true;
 }
 
 
 void FGAIManager::bind() {
    root = globals->get_props()->getNode("ai/models", true);
-   root->tie("count", SGRawValuePointer<int>(&numObjects));
+   root->tie("count", SGRawValuePointer<int>(&numObjects[0]));
 }
 
 
@@ -145,18 +98,21 @@ void FGAIManager::update(double dt) {
         // 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;      
+        _dt = dt;
 
         ai_list_itr = ai_list.begin();
         while(ai_list_itr != ai_list.end()) {
                 if ((*ai_list_itr)->getDie()) {      
-                   freeID((*ai_list_itr)->getID());
+                 tmgr->release((*ai_list_itr)->getID());
+                   --numObjects[(*ai_list_itr)->getType()];
+                   --numObjects[0];
+                   (*ai_list_itr)->unbind();
                    delete (*ai_list_itr);
-                   --numObjects;
                    if ( ai_list_itr == ai_list.begin() ) {
                        ai_list.erase(ai_list_itr);
                        ai_list_itr = ai_list.begin();
@@ -174,187 +130,182 @@ void FGAIManager::update(double dt) {
                 }
                 ++ai_list_itr;
         }
-        wind_from_down->setDoubleValue( strength );
-}
-
+        wind_from_down_node->setDoubleValue( strength );
 
-// This function returns the next available ID
-int FGAIManager::assignID() {
-  int maxint = 30000;
-  int x; 
-  bool used;
-  for (x=0; x<maxint; x++) {
-     used = false;
-     id_itr = ids.begin();
-     while( id_itr != ids.end() ) {
-       if ((*id_itr) == x) used = true;
-       ++id_itr;
-     }
-     if (!used) {
-       ids.push_back(x);
-       return x;
-     } 
-  }
-  return -1;  // no available ID's
 }
 
 
-// This function removes an ID from the ID array, making it
-// available for assignment to another AI object
-void FGAIManager::freeID( int ID ) {
-    id_itr = ids.begin();
-    while( id_itr != ids.end() ) {
-      if (*id_itr == ID) {
-        ids.erase( id_itr );
-        return;
-      }
-      ++id_itr;
-    }  
-}
-
-int FGAIManager::createAircraft( string model_class, string path,
-              double latitude, double longitude, double altitude,
-              double heading, double speed, double pitch, double roll ) {
+void*
+FGAIManager::createAircraft( FGAIModelEntity *entity,   FGAISchedule *ref) {
      
-        FGAIAircraft* ai_plane = new FGAIAircraft(this);
+        FGAIAircraft* ai_plane = new FGAIAircraft(this, ref);
         ai_list.push_back(ai_plane);
-        ai_plane->setID( assignID() );
-        ++numObjects;
-        if (model_class == "light") {
+        ++numObjects[0];
+        ++numObjects[FGAIBase::otAircraft];
+        if (entity->m_class == "light") {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::LIGHT]);
-        } else if (model_class == "ww2_fighter") {
+        } else if (entity->m_class == "ww2_fighter") {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::WW2_FIGHTER]);
-        } else if (model_class ==  "jet_transport") {
+        } else if (entity->m_class ==  "jet_transport") {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-        } else if (model_class == "jet_fighter") {
+        } else if (entity->m_class == "jet_fighter") {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_FIGHTER]);
-        } else {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-        }
-        ai_plane->setHeading(heading);
-        ai_plane->setSpeed(speed);
-        ai_plane->setPath(path.c_str());
-        ai_plane->setAltitude(altitude);
-        ai_plane->setLongitude(longitude);
-        ai_plane->setLatitude(latitude);
-        ai_plane->setBank(roll);
-        ai_plane->init();
-        ai_plane->bind();
-        return ai_plane->getID();
-}
-
-
-int FGAIManager::createAircraft( string model_class, string path,
-              FGAIFlightPlan* flightplan ) {
-     
-        FGAIAircraft* ai_plane = new FGAIAircraft(this);
-        ai_list.push_back(ai_plane);
-        ai_plane->setID( assignID() );
-        ++numObjects;
-        if (model_class == "light") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::LIGHT]);
-        } else if (model_class == "ww2_fighter") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::WW2_FIGHTER]);
-        } else if (model_class ==  "jet_transport") {
+        } else if (entity->m_class ==  "tanker") {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
-        } else if (model_class == "jet_fighter") {
-          ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_FIGHTER]);
+          ai_plane->SetTanker(true);
         } else {
           ai_plane->SetPerformance(&FGAIAircraft::settings[FGAIAircraft::JET_TRANSPORT]);
         }
-        ai_plane->setPath(path.c_str());
-        ai_plane->SetFlightPlan(flightplan);
+        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->getID();
+        return ai_plane;
 }
 
-
-int FGAIManager::createShip( string path, double latitude, double longitude,
-                             double altitude, double heading, double speed,
-                             double rudder ) {
+void*
+FGAIManager::createShip( FGAIModelEntity *entity ) {
+    
+     // cout << "creating ship" << endl;    
 
         FGAIShip* ai_ship = new FGAIShip(this);
         ai_list.push_back(ai_ship);
-        ai_ship->setID( assignID() );
-        ++numObjects;
-        ai_ship->setHeading(heading);
-        ai_ship->setSpeed(speed);
-        ai_ship->setPath(path.c_str());
-        ai_ship->setAltitude(altitude);
-        ai_ship->setLongitude(longitude);
-        ai_ship->setLatitude(latitude);
-        ai_ship->setBank(rudder);
+        ++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);
+
+        if ( entity->fp ) {
+           ai_ship->setFlightPlan(entity->fp);
+        }
+
         ai_ship->init();
         ai_ship->bind();
-        return ai_ship->getID();
+        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::otShip];
+        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->setRadius(entity->radius);
+        ai_carrier->setXoffset(entity->x_offset);
+        ai_carrier->setYoffset(entity->y_offset);
+        ai_carrier->setZoffset(entity->z_offset);
+
+        if ( entity->fp ) {
+           ai_carrier->setFlightPlan(entity->fp);
+        }
+
+        ai_carrier->init();
+        ai_carrier->bind();
+        return ai_carrier;
+}
 
-int FGAIManager::createBallistic( string path, double latitude, double longitude,
-                                  double altitude, double azimuth, double elevation,
-                                  double speed ) {
+void*
+FGAIManager::createBallistic( FGAIModelEntity *entity ) {
 
         FGAIBallistic* ai_ballistic = new FGAIBallistic(this);
         ai_list.push_back(ai_ballistic);
-        ai_ballistic->setID( assignID() );    
-        ++numObjects;
-        ai_ballistic->setAzimuth(azimuth);
-        ai_ballistic->setElevation(elevation);
-        ai_ballistic->setSpeed(speed);
-        ai_ballistic->setPath(path.c_str());
-        ai_ballistic->setAltitude(altitude);
-        ai_ballistic->setLongitude(longitude);
-        ai_ballistic->setLatitude(latitude);
+        ++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->getID();
+        return ai_ballistic;
 }
 
-int FGAIManager::createStorm( string path, double latitude, double longitude,
-                             double altitude, double heading, double speed ) {
+void*
+FGAIManager::createStorm( FGAIModelEntity *entity ) {
 
         FGAIStorm* ai_storm = new FGAIStorm(this);
-        ai_list.push_back(ai_storm);
-        ai_storm->setID( assignID() );
-        ++numObjects;
-        ai_storm->setHeading(heading);
-        ai_storm->setSpeed(speed);
-        ai_storm->setPath(path.c_str());
-        ai_storm->setAltitude(altitude);
-        ai_storm->setLongitude(longitude);
-        ai_storm->setLatitude(latitude);
+        ++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->setLongitude(entity->longitude);
+        ai_storm->setLatitude(entity->latitude);
         ai_storm->init();
         ai_storm->bind();
-        return ai_storm->getID();
+        ai_list.push_back(ai_storm);
+        return ai_storm;
 }
 
-int FGAIManager::createThermal( double latitude, double longitude,
-                                double strength, double diameter ) {
+void*
+FGAIManager::createThermal( FGAIModelEntity *entity ) {
 
         FGAIThermal* ai_thermal = new FGAIThermal(this);
-        ai_list.push_back(ai_thermal);
-        ai_thermal->setID( assignID() );
-        ++numObjects;
-        ai_thermal->setLongitude(longitude);
-        ai_thermal->setLatitude(latitude);
-        ai_thermal->setMaxStrength(strength);
-        ai_thermal->setDiameter(diameter / 6076.11549);
+        ++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->init();
         ai_thermal->bind();
-        return ai_thermal->getID();
+        ai_list.push_back(ai_thermal);
+        return ai_thermal;
 }
 
-void FGAIManager::destroyObject( int ID ) {
+void FGAIManager::destroyObject( void* ID ) {
         ai_list_itr = ai_list.begin();
         while(ai_list_itr != ai_list.end()) {
             if ((*ai_list_itr)->getID() == ID) {
-              freeID( ID );
+              --numObjects[0];
+              --numObjects[(*ai_list_itr)->getType()];
+              (*ai_list_itr)->unbind();
               delete (*ai_list_itr);
               ai_list.erase(ai_list_itr);
-              --ai_list_itr;
-              --numObjects;
-              return;
+
+              break;
             }
             ++ai_list_itr;
         }
@@ -386,30 +337,55 @@ void FGAIManager::processThermal( FGAIThermal* thermal ) {
 }
 
 
-void FGAIManager::processScenario( string filename ) {
-  //cout << "AIManager: creating a scenario." << endl;
+void FGAIManager::processScenario( string &filename ) {
   FGAIScenario* s = new FGAIScenario( filename );
   for (int i=0;i<s->nEntries();i++) {
-    FGAIScenario::entry* en = s->getNextEntry();
+    FGAIModelEntity* en = s->getNextEntry();
+
     if (en) {
-      FGAIFlightPlan* f = new FGAIFlightPlan( en->flightplan );
-      if (en->aitype == "aircraft"){
-        createAircraft( en->aircraft_class, en->model_path, f);
-      }
+      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 );
+      }      
     }
   }
+
   delete s;
 }
 
-int FGAIManager::getNum( FGAIBase::object_type ot ) {
-  ai_list_iterator itr = ai_list.begin();
-  int count = 0;
-  while(itr != ai_list.end()) {
-      if ((*itr)->getType() == ot) {
-         ++count;
-      }
-      ++itr;
-  }  
-  return count;
+// 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)
+{
+  ModelVecIterator i = loadedModels.begin();
+  while (i != loadedModels.end())
+    {
+      if (i->getPath() == path)
+       return i->getModelId();
+      i++;
+    }
+  return 0;
+}
+
+void FGAIManager::setModel(const string& path, ssgBranch *model)
+{
+  loadedModels.push_back(FGModelID(path,model));
 }
 
+
+//end AIManager.cxx