Changeset 10137 in orxonox.OLD for branches/ai/src/ai/ai_team.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_team.cc (modified) (2 diffs)

branches/ai/src/ai/ai_team.cc

@@ -1 @@
-
-
 /*
    orxonox - the future of 3D-vertical-scrollers
@@ -16 +14 @@
 */
 #define DEBUG_SPECIAL_MODULE DEBUG_MODULE_AI
-
 #include "ai_team.h"
-
-AITeam::AITeam(){
-        for(int i=0; i < maxSwarms; i++ )swarms[i]=NULL;
-}
-
-
-AISwarm* AITeam::getSwarm(int swarmNumber)
-{
-        if(swarmNumber>maxSwarms || swarmNumber<0)return NULL;
-        return swarms[swarmNumber];
-}
-
-
-AISwarm* AITeam::getCreateSwarm(int swarmNumber)
-{
-        if(swarmNumber>maxSwarms || swarmNumber<0)return NULL;
-        if(swarms[swarmNumber]==NULL)swarms[swarmNumber]=new AISwarm();
-        return swarms[swarmNumber];
-}
-
-
-void AITeam::addSwarm(int swarmNumber)
-{
-        if(swarmNumber>maxSwarms || swarmNumber<0)return;
-        swarms[swarmNumber]=new AISwarm();
-}
-
-
-void AITeam::removeSwarm(int swarmNumber)
-{
-        if(swarmNumber>maxSwarms || swarmNumber<0)return;
-        swarms[swarmNumber]=NULL;
-}
-
+#include "debug.h"
 
 void AITeam::process(float dt)
 {
-        for(int i=0; i < maxSwarms; i++ ){
-                if(swarms[i]!=NULL)swarms[i]->process(dt);
+        std::map<int,AISwarm*>::iterator it;
+        for (it= swarms.begin(); it!= swarms.end(); it++ )
+                it->second->process(dt);
+}
+
+void AITeam::addAI(int swarmNumber, AIModule* aiModule)
+{
+        std::pair<std::map<int,AISwarm*>::iterator,bool> p;
+        AISwarm* newSwarm=new AISwarm();
+        p=swarms.insert(std::make_pair(swarmNumber,newSwarm));
+        if(!p.second)delete newSwarm;
+        p.first->second->addAI(aiModule);
+}
+
+void AITeam::removeAI(int swarmNumber, AIModule* aiModule)
+{
+        std::map<int,AISwarm*>::iterator it = swarms.find(swarmNumber);
+        if(it==swarms.end())return;
+        it->second->removeAI(aiModule);
+        if(it->second->getSwarmSize()==0){
+                delete it->second;
+                swarms.erase(it);
         }
 }
 
-
+int AITeam::getTeamSize()
+{
+        return swarms.size();
+}