D:/simple_rts/include/ControlGroup.h

Go to the documentation of this file.

#ifndef __UNIT_GROUP_H__
#define __UNIT_GROUP_H__

#pragma once

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

using Ogre::Vector3;
using std::vector;

namespace ASR
{
        class Formation;
        class Unit;

        class ControlGroup
        {
        // Data Members
        // ----------------------------------------------------------------------------
        private:
                bool                    mDirty;
                Formation*              mCurFormation;
                
                typedef vector<Unit*> UnitList;
                UnitList                mUnits;

                Vector3                 mFormationCenter;
                Vector3                 mDestination;

                Vector3                 mForward;
                Vector3                 mRight;


        // Construction
        // ----------------------------------------------------------------------------
        public:
                ControlGroup ();
                ~ControlGroup ();

                int addUnit ( Unit* unit );
                void removeUnit ( size_t index );
                void removeUnit ( Unit* unit );
                int getUnitIndex ( Unit* unit );
                size_t getNumUnits ();
                
                Unit* getUnit ( int index );
                Unit* getLeaderUnit ();

        // Positioning
        // ----------------------------------------------------------------------------
                Vector3 getFormationCenter ();

                Vector3 getUnitOffset ( size_t index );
                Vector3 getUnitOffset ( Unit* unit );

                Vector3 getUnitDestination ( int index );
                Vector3 getUnitDestination ( Unit* unit );

                void setDestination ( const Vector3& dest );



        // ----------------------------------------------------------------------------
        public:
                void update ();


        // Helpers
        // ----------------------------------------------------------------------------
        protected:
                void _updateFormation ();

                void _calcCenter ();

                void _calcOrientation ();
        };
}

#endif

---