]> git.mxchange.org Git - flightgear.git/blobdiff - src/AIModel/AIManager.cxx
Make the vertical acceleration rate scale with vertical performance. The
[flightgear.git] / src / AIModel / AIManager.cxx
index 2d68fc3d98aff8508a2e8e04bd007b1bb1527d40..207913bb1ff5aae959e33dc829126c7c8c79ddb2 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
 //
 // 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 <list>
+#include <Airports/simple.hxx>
+#include <Traffic/TrafficMgr.hxx>
 
 #include "AIManager.hxx"
 #include "AIAircraft.hxx"
 #include "AIShip.hxx"
 #include "AIBallistic.hxx"
-
-SG_USING_STD(list);
-
+#include "AIStorm.hxx"
+#include "AIThermal.hxx"
+#include "AICarrier.hxx"
+#include "AIStatic.hxx"
+#include "AIMultiplayer.hxx"
 
 FGAIManager::FGAIManager() {
-  initDone = false;
+  _dt = 0.0;
+  mNumAiModels = 0;
+  for (unsigned i = 0; i < FGAIBase::MAX_OBJECTS; ++i)
+    mNumAiTypeModels[i] = 0;
 }
 
 FGAIManager::~FGAIManager() {
-  ai_list.clear(); 
+  ai_list_iterator ai_list_itr = ai_list.begin();
+  while(ai_list_itr != ai_list.end()) {
+      (*ai_list_itr)->unbind();
+      ++ai_list_itr;
+  }
 }
 
+
 void FGAIManager::init() {
-  SGPropertyNode * node = fgGetNode("sim/ai", true);
-  for (int i = 0; i < node->nChildren(); i++) {
-    const SGPropertyNode * entry = node->getChild(i);
-    if (!strcmp(entry->getName(), "entry")) {
-      if (!strcmp(entry->getStringValue("type", ""), "aircraft")) { 
-        FGAIAircraft* ai_plane = new FGAIAircraft;
-        ai_list.push_back(ai_plane);
-        if (!strcmp(entry->getStringValue("class", ""), "light")) {
-          PERF_STRUCT ps = {2.0, 2.0, 450.0, 1000.0, 70.0, 80.0, 100.0, 80.0, 60.0};
-          ai_plane->SetPerformance(ps);
-        } else if (!strcmp(entry->getStringValue("class", ""), "ww2_fighter")) {
-          PERF_STRUCT ps = {4.0, 2.0, 3000.0, 1500.0, 110.0, 180.0, 250.0, 200.0, 100.0};
-          ai_plane->SetPerformance(ps);
-        } else if (!strcmp(entry->getStringValue("class", ""), "jet_transport")) {
-          PERF_STRUCT ps = {5.0, 2.0, 3000.0, 1500.0, 140.0, 300.0, 430.0, 300.0, 130.0};
-          ai_plane->SetPerformance(ps);
-        } else if (!strcmp(entry->getStringValue("class", ""), "jet_fighter")) {
-          PERF_STRUCT ps = {7.0, 3.0, 4000.0, 2000.0, 150.0, 350.0, 500.0, 350.0, 150.0};
-          ai_plane->SetPerformance(ps);
-        }
-        ai_plane->setHeading(entry->getDoubleValue("heading"));
-        ai_plane->setSpeed(entry->getDoubleValue("speed-KTAS"));
-        ai_plane->setPath(entry->getStringValue("path"));
-        ai_plane->setAltitude(entry->getDoubleValue("altitude-ft"));
-        ai_plane->setLongitude(entry->getDoubleValue("longitude"));
-        ai_plane->setLatitude(entry->getDoubleValue("latitude"));
-        ai_plane->init();
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "ship")) {
-        FGAIShip* ai_ship = new FGAIShip;
-        ai_list.push_back(ai_ship);
-        ai_ship->setHeading(entry->getDoubleValue("heading"));
-        ai_ship->setSpeed(entry->getDoubleValue("speed-KTAS"));
-        ai_ship->setPath(entry->getStringValue("path"));
-        ai_ship->setAltitude(entry->getDoubleValue("altitude-ft"));
-        ai_ship->setLongitude(entry->getDoubleValue("longitude"));
-        ai_ship->setLatitude(entry->getDoubleValue("latitude"));
-        ai_ship->init();
-
-      } else if (!strcmp(entry->getStringValue("type", ""), "ballistic")) {
-        FGAIBallistic* ai_ballistic = new FGAIBallistic;
-        ai_list.push_back(ai_ballistic);
-        ai_ballistic->setAzimuth(entry->getDoubleValue("azimuth"));
-        ai_ballistic->setElevation(entry->getDoubleValue("elevation"));
-        ai_ballistic->setSpeed(entry->getDoubleValue("speed-fps"));
-        ai_ballistic->setPath(entry->getStringValue("path"));
-        ai_ballistic->setAltitude(entry->getDoubleValue("altitude-ft"));
-        ai_ballistic->setLongitude(entry->getDoubleValue("longitude"));
-        ai_ballistic->setLatitude(entry->getDoubleValue("latitude"));
-        ai_ballistic->init();
-      } 
-     }
+  root = fgGetNode("sim/ai", true);
+
+  enabled = root->getNode("enabled", true)->getBoolValue();
+  if (!enabled)
+      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);
+  user_heading_node   = fgGetNode("/orientation/heading-deg", true);
+  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);
+
+
+  /// 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 );
+    }
   }
+}
+
 
-  initDone = true;
+void FGAIManager::reinit() {
+   update(0.0);
 }
 
 
 void FGAIManager::bind() {
+   root = globals->get_props()->getNode("ai/models", true);
+   root->tie("count", SGRawValueMethods<FGAIManager, int>(*this,
+             &FGAIManager::getNumAiObjects));
 }
 
 
 void FGAIManager::unbind() {
+    root->untie("count");
 }
 
 
 void FGAIManager::update(double dt) {
-#if 0
-       if(!initDone) {
-          init();
-          SG_LOG(SG_ATC, SG_WARN, "Warning - AIManager::update(...) called before AIManager::init()");
-       }
-#endif
-       
-       ai_list_itr = ai_list.begin();
-       while(ai_list_itr != ai_list.end()) {
-                if ((*ai_list_itr)->getDie()) {
-                   ai_list.erase(ai_list_itr, ai_list_itr);
-                } else {
-                  (*ai_list_itr)->update(dt);
-               }
-                ++ai_list_itr;
-       }
+  // initialize these for finding nearest thermals
+  range_nearest = 10000.0;
+  strength = 0.0;
+  if (!enabled)
+    return;
+
+  FGTrafficManager *tmgr = (FGTrafficManager*) globals->get_subsystem("Traffic Manager");
+  _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());
+      --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::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::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;
+  }
+}
+
+int
+FGAIManager::getNumAiObjects(void) const
+{
+  return mNumAiModels;
+}
+
+void FGAIManager::fetchUserState( void ) {
+     user_latitude  = user_latitude_node->getDoubleValue();
+     user_longitude = user_longitude_node->getDoubleValue();
+     user_altitude  = user_altitude_node->getDoubleValue();
+     user_heading   = user_heading_node->getDoubleValue();
+     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();
 }
+
+
+// only keep the results from the nearest thermal
+void FGAIManager::processThermal( FGAIThermal* thermal ) {
+  thermal->update(_dt);
+  if ( thermal->_getRange() < range_nearest ) {
+     range_nearest = thermal->_getRange();
+     strength = thermal->getStrength();
+  }
+}
+
+
+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);
+      
+    }
+  }
+}
+
+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) const
+{
+  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));
+}
+
+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