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

branches/ai/src/ai/ai_swarm.cc

@@ -1 @@
-
-
 /*
    orxonox - the future of 3D-vertical-scrollers
@@ -16 +14 @@
 */
 #define DEBUG_SPECIAL_MODULE DEBUG_MODULE_AI
-
 #include "ai_swarm.h"
-
+#include "debug.h"
 
 void AISwarm::process(float dt)
 {
-        Vector swarmCenter;
-
-        for(unsigned int i=0;i<swarmMembers.size();i++){
-                //swarmCenter=swarmCenter+swarmMembers.at(i)->getPosition();
-        }
-        swarmCenter=swarmCenter/swarmMembers.size();
-
-        for(unsigned int i=0;i<swarmMembers.size();i++)
-                swarmMembers.at(i)->process(dt);
+        std::set<AIModule*>::iterator it;
+        for (it= swarmMembers.begin(); it!= swarmMembers.end(); it++ )
+                (*it)->process(dt);
 }
 
-
-void AISwarm::addToSwarm(AIModule* newMember)
+void AISwarm::addAI(AIModule* aiModule)
 {
-        swarmMembers.push_back(newMember);
+        swarmMembers.insert(aiModule);
 }
 
-
-void AISwarm::removeFromSwarm(AIModule* member)
+void AISwarm::removeAI(AIModule* aiModule)
 {
-        std::vector<AIModule*>::iterator it = this->swarmMembers.begin();
-        for(; it < this->swarmMembers.end(); it++) {
-
-//      for(unsigned int i=0;i<swarmMembers.size();i++){
-                if(*it==member){
-                        swarmMembers.erase(it);
-                        return;
-                }
-        }
+        std::set<AIModule*>::iterator it=swarmMembers.find(aiModule);
+        if(it==swarmMembers.end())return;
+        delete (*it);
 }
 
-
-
+int AISwarm::getSwarmSize()
+{
+        return swarmMembers.size();
+}