D:/simple_rts/include/TeamUnits.h

Go to the documentation of this file.

#pragma once

#include <vector>
using std::vector;

#include "OgreVector3.h"
#include "OgreSceneNode.h"
#include "OgreString.h"
#include "OgreIteratorWrappers.h"

using Ogre::Vector3;
using Ogre::VectorIterator;
using Ogre::SceneNode;
using Ogre::String;
using Ogre::RaySceneQuery;


// ----------------------------------------------------------------------------
namespace ASR
{
        class Unit;
        class UnitList;
        class Team;

        // TODO
        //      Refactor into a TeamObjects templated class

        // ----------------------------------------------------------------------------
        class TeamUnits
        {
        // Data Types
        // ----------------------------------------------------------------------------
        public:
                typedef vector<Unit*>                           UnitList;
                typedef VectorIterator<UnitList>        UnitIterator;

        // Data Storage
        // ----------------------------------------------------------------------------
        private:
                SceneNode*                                      mTeamNode;
                RaySceneQuery*                          mRaySceneQuery;

                UnitList                                        mUnits;

                static int                                      mNumTeams;
                Team*                                           mTeam;


        // Spatial Unit Management Data
        // ----------------------------------------------------------------------------
        private:
                typedef vector <UnitList>                       SpatialColumn;
                typedef vector <SpatialColumn>          SpatialGrid;
                SpatialGrid                                                     mUnitGrid;


        // Construction
        // ----------------------------------------------------------------------------
        public:
                TeamUnits ( Team* parentTeam, unsigned int initialUnits = 0 );
                ~TeamUnits ();


        // Combat
        // ----------------------------------------------------------------------------
        public:
                void findTargets ();


        // List Maintenance
        // ----------------------------------------------------------------------------
        public:
                Unit* createUnit ( String meshName );
                void moveUnits ( float deltaTime );
                void animateUnits ( float deltaTime );

                UnitIterator getUnitIterator ();
                Unit* getUnit ( size_t index );
                size_t getNumUnits ();


        // Battle System
        // ----------------------------------------------------------------------------
        public:
                void performAttacks ();


        // Spatial Management
        // ----------------------------------------------------------------------------
        public:
                void setSpatialGridSize ( int numBucketsWide, int numBucketsHigh );

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

                void updateSpatialGrid ();

        private:
                void _clearSpatialGrid ();
        };
}

---