source: orxonox.OLD/branches/blink/src/ai/ai_engine.cc

/*
   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_engine.h"
#include "debug.h"
#include "npcs/npc_test.h"
#include "player.h"
#include "playable.h"
#include "state.h"

AIEngine* AIEngine::singletonRef = NULL;

void AIEngine::tick(float dt)
{
        std::map<int,AITeam*>::iterator it;
        std::vector<WorldEntity*>* enemyList=new std::vector<WorldEntity*>;

        for (it=teams.begin(); it!= teams.end(); it++ )
        {
                //add player to enemys (player belongs to team 0)
                if(it->first!=0){
                        Player* pl = State::getPlayer();
                        if( pl!=NULL && pl->getPlayable()!=NULL )
                                enemyList->push_back(pl->getPlayable());
                }

                //find other enemys for this team
                for(ObjectList<NPC2>::const_iterator npc = NPC2::objectList().begin(); npc != NPC2::objectList().end(); ++npc)
                        if((*npc)->getTeam() != it->first)enemyList->push_back(*npc);

                //process the team
                it->second->setEnemyList(enemyList);
                it->second->process(dt);
                enemyList->clear();
        }

        delete enemyList;
}

void AIEngine::addAI(int teamNumber, int swarmNumber, WorldEntity* npc, float maxSpeed, float attackDistance)
{
        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, npc, maxSpeed, attackDistance);
}

void AIEngine::removeAI(int teamNumber, int swarmNumber, WorldEntity* npc)
{
        std::map<int,AITeam*>::iterator it = teams.find(teamNumber);
        if(it==teams.end())return;
        it->second->removeAI(swarmNumber,npc);
        if(it->second->getTeamSize()==0){
                delete it->second;
                teams.erase(it);
        }
}