D:/simple_rts/include/Team.h

Go to the documentation of this file.

#ifndef __TEAM_H__
#define __TEAM_H__

#pragma once

#include "UnitType.h"
#include "TeamUnits.h"

#include "OgreString.h"
#include "OgreVector3.h"
#include "OgreColourValue.h"
#include <vector>

using Ogre::String;
using Ogre::Vector3;
using Ogre::ColourValue;
using std::vector;

namespace ASR
{
        // Forward Declarations
        class Unit;
        class Building;
        class TeamBuildings;
        class ControlGroup;
        class TeamUpdater;
        class FogOfWar;

        // ----------------------------------------------------------------------------
        class Team
        {
                friend class TeamUpdater;

        // Data Storage
        // ----------------------------------------------------------------------------
        private:
                TeamUnits*                      mUnits;
                TeamBuildings*          mBuildings;
                FogOfWar*                       mFogOfWar;

                TeamUpdater*            mUpdater;

                typedef vector<ControlGroup*>   ControlGroups;
                ControlGroups           mControlGroups;
                ControlGroup*           mActiveGroup;

                String                          mName;

                // HACKS
                String                          mBaseMaterialName;
                String                          mFeathersMaterialName;
                //String                                mShieldMaterialName;
                String                          mShoulderMaterialName;


        // Construction
        // ----------------------------------------------------------------------------
        public:
                Team ( const String& name, const ColourValue& teamColor );

                ~Team ();


        // ----------------------------------------------------------------------------
        public:
                const String& getName () const;


        // Creation Methods
        // ----------------------------------------------------------------------------
        public:
                Unit* createUnit ( const UnitType& unitType );


                Building* createBuilding ( const String& meshName, const Vector3& position );

        private:
                void _applyTeamColors ( Unit* unit );


                void _applyTeamColors ( Building* building );


        // Control Groups
        // ----------------------------------------------------------------------------
        public:
                ControlGroup* createControlGroup ();
                void destroyControlGroup ( ControlGroup* group );

                void setActiveControlGroup ( ControlGroup* group );
                void destroyActiveControlGroup ();
                ControlGroup* getActiveControlGroup () const;

        private:
                void _destroyAllControlGroups ();


        // Unit Management
        // ----------------------------------------------------------------------------
        public:
                TeamUnits::UnitIterator getUnitIterator () const;

                TeamUnits::UnitList getAdjacentUnits ( const Vector3& worldPos, float radius );

        
        // Misc Methods
        // ----------------------------------------------------------------------------
        public:
                FogOfWar* getFogOfWar () const;


        // Update Methods
        // ----------------------------------------------------------------------------
        public:
                void findTargets ();
        };
}

#endif

---