Changeset 10137 in orxonox.OLD for branches/ai/src/ai/ai_engine.cc

Timestamp:

Dec 21, 2006, 3:49:51 PM (17 years ago)

Author:

tfahrni

Message:

discoverd std::map and std::set

File:

• branches/ai/src/ai/ai_engine.cc (modified) (3 diffs)

branches/ai/src/ai/ai_engine.cc

@@ -1 @@
-
-
 /*
    orxonox - the future of 3D-vertical-scrollers
@@ -12 +10 @@
 
    ### File Specific:
-   main-programmer: Patrick Boenzli
+   main-programmer: Thomas Fahrni
    co-programmer:
 */
 #define DEBUG_SPECIAL_MODULE DEBUG_MODULE_AI
-
 #include "ai_engine.h"
 #include "debug.h"
@@ -22 +19 @@
 AIEngine* AIEngine::singletonRef = NULL;
 
-AIEngine::AIEngine(){
-        for(int i=0; i < maxTeams; i++ )teams[i]=NULL;
+void AIEngine::tick(float dt)
+{
+        std::map<int,AITeam*>::iterator it;
+        for (it= teams.begin(); it!= teams.end(); it++ )
+                it->second->process(dt);
 }
 
-AITeam* AIEngine::getTeam(int teamNumber)
+void AIEngine::addAI(int teamNumber, int swarmNumber, AIModule* aiModule)
 {
-        if(teamNumber>maxTeams || teamNumber<0)return NULL;
-        return teams[teamNumber];
+        std::pair<std::map<int,AITeam*>::iterator,bool> p;
+        AITeam* newTeam=new AITeam();
+        p=teams.insert(std::make_pair(teamNumber,newTeam));
+        if(!p.second)delete newTeam;
+        p.first->second->addAI(swarmNumber, aiModule);
 }
 
-AITeam* AIEngine::getCreateTeam(int teamNumber)
+void AIEngine::removeAI(int teamNumber, int swarmNumber, AIModule* aiModule)
 {
-        if(teamNumber>maxTeams || teamNumber<0)return NULL;
-        if(teams[teamNumber]==NULL)teams[teamNumber]=new AITeam();
-        return teams[teamNumber];
+        std::map<int,AITeam*>::iterator it = teams.find(swarmNumber);
+        if(it==teams.end())return;
+        it->second->removeAI(swarmNumber,aiModule);
+        if(it->second->getTeamSize()==0){
+                delete it->second;
+                teams.erase(it);
+        }
 }
 
-
-void AIEngine::addTeam(int teamNumber)
-{
-        if(teamNumber>maxTeams || teamNumber<0)return;
-        teams[teamNumber]=new AITeam();
-}
-
-
-void AIEngine::removeTeam(int teamNumber)
-{
-        if(teamNumber>maxTeams || teamNumber<0)return;
-        teams[teamNumber]=NULL;
-}
-
-
-void AIEngine::tick(float dt)
-{
-        for(int i=0; i < maxTeams; i++ )
-                if(teams[i]!=NULL)teams[i]->process(dt);
-}
-
-
-void AIEngine::rebuildAIVector(){
-        //AIVector.clear();
-
-        //for(int i=0; i < maxTeams; i++ )
-                //if(teams[i]!=NULL)AIVector.push_back(teams[i]);
-}