source: orxonox.OLD/branches/ai/src/ai/swarm_module.h @ 10244

#ifndef _SWARM_MODULE_H
#define _SWARM_MODULE_H
#include "world_entity.h"
#include "ai_module.h"

class SwarmModule {
 public:
        inline SwarmModule(){taskComplete=false;}
        virtual ~SwarmModule(){}
        virtual void process(float dt){}
        
        void addAI(WorldEntity*);
        void removeAI(WorldEntity*);
        void getAttributesFrom(SwarmModule* oldModule);
        virtual void initialize(){}
        inline bool taskDone(){return taskComplete;}
        
        inline void setEnemyList(std::vector<WorldEntity*>* enemys){this->enemys=enemys;}       
        
        inline void orderRelPos(Vector taskRelPos){this->taskRelPos=taskRelPos;}
        inline void orderAbsPos(Vector taskAbsPos){this->taskAbsPos=taskAbsPos;}
        inline void orderView(Vector taskView){this->taskView=taskView;}
        inline void orderSpeed(float taskSpeed){this->taskSpeed=taskSpeed;}
        inline void orderRelObject(WorldEntity* taskRelObject){this->taskRelObject=taskRelObject;}
        inline void orderMaxTime(float taskMaxTime){this->taskMaxTime=taskMaxTime;}

        Vector getPosition();
        inline float getSpeed(){return speed;}
        inline Vector getView(){return view;}
        inline int getSwarmSize(){return members.size();}
        inline std::vector<WorldEntity*>* getEnemyList(){return enemys;}
        inline std::map<WorldEntity*,AIModule*> getMembers(){return members;}
        
 protected:
        float getRadius(WorldEntity* object);
        void changeAIModule(std::map<WorldEntity*,AIModule*>::iterator it,AIModule* newAI);
        Vector taskRelPos;
        Vector taskAbsPos;
        Vector taskView;
        float taskSpeed;
        WorldEntity* taskRelObject;
        bool taskComplete;
        float taskMaxTime;
        
        Vector view;
        float speed;
        
        std::map<WorldEntity*,AIModule*> members;       
        std::vector<WorldEntity*>* enemys;
};


#endif /* _SWARM_MODULE_H */