source: orxonox.OLD/branches/ai/src/ai/ai_team.cc @ 10137

/*
   orxonox - the future of 3D-vertical-scrollers

   Copyright (C) 2004 orx

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2, or (at your option)
   any later version.

   ### File Specific:
   main-programmer: Thomas Fahrni
   co-programmer:
*/
#define DEBUG_SPECIAL_MODULE DEBUG_MODULE_AI
#include "ai_team.h"
#include "debug.h"

void AITeam::process(float 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();
}