ASR::UnitSelecter Class Reference

#include <UnitSelecter.h>

Inheritance diagram for ASR::UnitSelecter:

[legend]Collaboration diagram for ASR::UnitSelecter:

[legend]List of all members.

Public Member Functions
	UnitSelecter (Team *team, Camera *camera)
	~UnitSelecter (void)
bool	updateBeforeFrame (const FrameEvent &e, const InputReader *inputDevice)
	Called each frame to perform the "update".
bool	updateAfterFrame (const FrameEvent &e, const InputReader *inputDevice)
Private Member Functions
void	_moveUnits ()
void	_selectUnits ()
void	_unselectDeadUnits ()
void	_createActiveGroup () const
void	_unselectCurrect ()
void	_selectUnit (Unit *unit)
void	_unselectUnit (Unit *unit)
Vector3	_getMousePosition ()
	Returns the world position of the current mouse position.
Private Attributes
Entity *	mCubes [5]
SceneNode *	mNodes [5]
Camera *	mCamera
RaySceneQuery *	mRayQuery
Team *	mTeam
float	mAbsMouse [2]
bool	mFirstPoint
Overlay *	mOverlay
Vector2	mBounds [2]
vector< Unit * >	mSelectedUnits

---

Detailed Description

Definition at line 36 of file UnitSelecter.h.

---

Constructor & Destructor Documentation

ASR::UnitSelecter::UnitSelecter	(	Team *	team,
		Camera *	camera
	)

Definition at line 40 of file UnitSelecter.cpp.

References ASR::Game::getSceneManager(), ASR::Game::getSingleton(), mAbsMouse, mBounds, mCubes, mFirstPoint, mNodes, mOverlay, and mRayQuery.

                : mTeam ( team ), mCamera ( camera )
        {
                SceneManager* sceneMgr = Game::getSingleton().getSceneManager ();

                // HACK
                //              Create four cubes for showing the bounding corners
                //              Create a cube for showing the right click location
                for ( int i = 0; i < 5; i++ )
                {
                        mCubes[i] = sceneMgr->createEntity ( StringConverter::toString(i), "cube.mesh" );
                        mNodes[i] = sceneMgr->getRootSceneNode ()->createChildSceneNode ( StringConverter::toString(i) );
                        mNodes[i]->attachObject( mCubes[i] );
                        mNodes[i]->scale ( 0.15, 0.15, 0.15  );

                        mNodes[i]->setVisible( false );
                }

                mOverlay = OverlayManager::getSingleton().getByName("ASR/SelectionOverlay");
                mOverlay->hide();

                mBounds[0] = Vector2::ZERO;
                mBounds[1] = Vector2::ZERO;

                mAbsMouse[0] = 0.0f;
                mAbsMouse[1] = 0.0f;

                mFirstPoint = false;

                mRayQuery = sceneMgr->createRayQuery( Ray() );
        }

ASR::UnitSelecter::~UnitSelecter

(

void

)

Definition at line 74 of file UnitSelecter.cpp.

References ASR::Game::getSceneManager(), ASR::Game::getSingletonPtr(), and mRayQuery.

        {
                SceneManager* sceneMgr = Game::getSingletonPtr ()->getSceneManager ();
                sceneMgr->destroyQuery ( mRayQuery );
        }

---

Member Function Documentation

void ASR::UnitSelecter::_createActiveGroup

(

)

const [private]

Definition at line 420 of file UnitSelecter.cpp.

References ASR::ControlGroup::addUnit(), ASR::Team::createControlGroup(), ASR::Team::destroyActiveControlGroup(), mSelectedUnits, mTeam, and ASR::Team::setActiveControlGroup().

Referenced by updateBeforeFrame().

        {
                ControlGroup* activeGroup;

                mTeam->destroyActiveControlGroup ();
                activeGroup = mTeam->createControlGroup ();

                vector<Unit*>::const_iterator iter = mSelectedUnits.begin();
                while ( iter != mSelectedUnits.end() )
                {
                        activeGroup->addUnit ( *iter );
                        iter++;
                }

                mTeam->setActiveControlGroup ( activeGroup );
        }

Vector3 ASR::UnitSelecter::_getMousePosition

(

)

[private]

Returns the world position of the current mouse position.

Returns NEGATIVE_X_AXIS if the mouse doesn't intersect any terrain

Definition at line 367 of file UnitSelecter.cpp.

References mAbsMouse, mCamera, mNodes, and mRayQuery.

Referenced by _moveUnits().

        {
                int screenWidth = mCamera->getViewport()->getActualWidth() / 2;
                int screenHeight = mCamera->getViewport()->getActualHeight() / 2;

                // Cast a ray to find the intersection point
                float screenX = (mAbsMouse[0] + screenWidth ) / mCamera->getViewport()->getActualWidth();
                float screenY = (mAbsMouse[1] + screenHeight ) / mCamera->getViewport()->getActualHeight();

                Ray point = mCamera->getCameraToViewportRay( screenX, screenY );
                mRayQuery->setRay ( point );

                Vector3 position = Vector3::NEGATIVE_UNIT_X;

                RaySceneQueryResult& results = mRayQuery->execute();
                RaySceneQueryResult::iterator iter = results.begin();
                while ( iter != results.end () )
                {
                        if ( iter->worldFragment )
                        {
                                position = iter->worldFragment->singleIntersection;
                                break;
                        }
                        iter++;
                }

                if ( iter == results.end() )
                {
                        mNodes[0]->setVisible( false );
                }
                else
                {
                        mNodes[0]->setVisible( true );
                        mNodes[0]->setPosition ( position );
                }

                return position;
        }

void ASR::UnitSelecter::_moveUnits

(

)

[private]

Definition at line 82 of file UnitSelecter.cpp.

References _getMousePosition(), ASR::Team::getActiveControlGroup(), ASR::Level::getClosestNode(), ASR::ControlGroup::getLeaderUnit(), ASR::ControlGroup::getNumUnits(), ASR::Level::getSingleton(), mTeam, and ASR::ControlGroup::setDestination().

Referenced by updateBeforeFrame().

        {
                if ( mTeam->getActiveControlGroup ()->getNumUnits () == 0 )
                        return;

                Vector3 destPos = _getMousePosition();
                Unit* leader = mTeam->getActiveControlGroup ()->getLeaderUnit ();

                if ( leader->setDestination ( destPos ) )
                {
                        ControlGroup* activeGroup = mTeam->getActiveControlGroup ();

                        // TODO
                        //      Create a vector from the leaders last waypoint to the click position
                        //      and use this as an offset to put each and every other unit as well.

                        // Create paths for all other units to the final destination
                        activeGroup->setDestination ( destPos );
                        activeGroup->update ();


                        for ( unsigned int i = 1; i < activeGroup->getNumUnits (); i++ )
                        {
                                Level::LevelNode* endNode = Level::getSingleton ().getClosestNode ( activeGroup->getUnitDestination ( i ) );
                                activeGroup->getUnit ( i )->setDestination ( endNode->getWorldPosition () );
                        }
                }

        }

void ASR::UnitSelecter::_selectUnit

(

Unit *

unit

)

[private]

Definition at line 439 of file UnitSelecter.cpp.

Referenced by _selectUnits().

        {
                SceneNode* parentNode = unit->getEntity()->getParentSceneNode ();
                parentNode->showBoundingBox( true );
        }

void ASR::UnitSelecter::_selectUnits

(

)

[private]

Definition at line 116 of file UnitSelecter.cpp.

References _selectUnit(), ASR::Game::getSceneManager(), ASR::Game::getSingletonPtr(), ASR::Team::getUnitIterator(), mAbsMouse, mBounds, mCamera, mFirstPoint, mNodes, mOverlay, mRayQuery, mSelectedUnits, and mTeam.

Referenced by updateBeforeFrame().

        {
                int screenWidth = mCamera->getViewport()->getActualWidth() / 2;
                int screenHeight = mCamera->getViewport()->getActualHeight() / 2;

                mOverlay->show ();

                // If the selection has already started, update the ending portion
                if ( !mFirstPoint )
                {
                        mBounds[0].x = mAbsMouse[0];
                        mBounds[0].y = mAbsMouse[1];

                        mBounds[1].x = mBounds[0].x;
                        mBounds[1].y = mBounds[0].y;

                        mFirstPoint = true;
                }
                else
                {
                        mBounds[1].x = mAbsMouse[0];
                        mBounds[1].y = mAbsMouse[1];
                }

                Vector2 topLeft;
                Vector2 bottomRight;

                topLeft.x = ( mBounds[0].x < mBounds[1].x ) ? mBounds[0].x : mBounds[1].x;
                topLeft.y = ( mBounds[0].y < mBounds[1].y ) ? mBounds[0].y : mBounds[1].y;

                bottomRight.x = ( mBounds[0].x > mBounds[1].x ) ? mBounds[0].x : mBounds[1].x;
                bottomRight.y = ( mBounds[0].y > mBounds[1].y ) ? mBounds[0].y : mBounds[1].y;

                int width = bottomRight.x - topLeft.x;
                int height = bottomRight.y - topLeft.y;

                OverlayContainer* cont = NULL;

                // Left Side
                cont = mOverlay->getChild( "ASR/Left" );
                cont->setLeft ( topLeft.x + screenWidth );
                cont->setTop ( topLeft.y + screenHeight );
                cont->setWidth ( 5 );
                cont->setHeight ( height );

                // Right Side
                cont = mOverlay->getChild( "ASR/Right" );
                cont->setLeft ( bottomRight.x + screenWidth );
                cont->setTop ( topLeft.y + screenHeight );
                cont->setWidth ( 5 );
                cont->setHeight ( height + 5 );

                // Top Side
                cont = mOverlay->getChild( "ASR/Top" );
                cont->setLeft ( topLeft.x + screenWidth );
                cont->setTop ( topLeft.y + screenHeight );
                cont->setWidth ( width );
                cont->setHeight ( 5 );

                // Bottom Side
                cont = mOverlay->getChild( "ASR/Bottom" );
                cont->setLeft ( topLeft.x + screenWidth );
                cont->setTop ( bottomRight.y + screenHeight );
                cont->setWidth ( width + 5 );
                cont->setHeight ( 5 );


                // Find the intersection with the terrain
                Ray points[2];
                float fLeft = (topLeft.x + screenWidth ) / mCamera->getViewport()->getActualWidth();
                float fTop = (topLeft.y + screenHeight ) / mCamera->getViewport()->getActualHeight();

                float fRight = (bottomRight.x + screenWidth ) / mCamera->getViewport()->getActualWidth();
                float fBottom = (bottomRight.y + screenHeight ) / mCamera->getViewport()->getActualHeight();

                points[0] = mCamera->getCameraToViewportRay( fLeft, fTop );
                points[1] = mCamera->getCameraToViewportRay( fRight, fBottom );

                Vector3 worldPos[2];

                mRayQuery->setRay ( points[0] );
                RaySceneQueryResult& results = mRayQuery->execute();
                RaySceneQueryResult::iterator iter = results.begin();
                if ( iter != results.end() && iter->worldFragment )
                {
                        worldPos[0] = iter->worldFragment->singleIntersection;
                }
                // Just put them at the top left if they missed the terrain
                else
                {
                        worldPos[0] = Vector3::ZERO;
                }

                mRayQuery->setRay ( points[1] );
                results = mRayQuery->execute();
                iter = results.begin();
                if ( iter != results.end() && iter->worldFragment )
                {
                        worldPos[1] = iter->worldFragment->singleIntersection;
                }
                else
                {
                        worldPos[1] = Vector3::ZERO;
                }

                mNodes[1]->setPosition( worldPos[0] );
                mNodes[1]->setVisible ( true );
                mNodes[2]->setPosition( worldPos[1] );
                mNodes[2]->setVisible ( true );



                // TODO
                //              Clean this approach up a lot
                // Grab all the entities within this rectangle
                SceneManager* sceneMgr = Game::getSingletonPtr ()->getSceneManager ();
                SceneNode* root = sceneMgr->getRootSceneNode();


                TeamUnits::UnitIterator unitIter = mTeam->getUnitIterator ();
                while ( unitIter.hasMoreElements () )
                {
                        Unit* curUnit = unitIter.getNext ();

                        if ( curUnit->isDead () )
                                continue;

                        Vector3 pos = curUnit->getPosition ();
                        if ( pos.x >= worldPos[0].x && pos.x <= worldPos[1].x )
                        {
                                if ( pos.z >= worldPos[0].z && pos.z <= worldPos[1].z )
                                {
                                        _selectUnit ( curUnit );
                                        mSelectedUnits.push_back( curUnit );
                                }
                        }
                }
        }

void ASR::UnitSelecter::_unselectCurrect

(

)

[private]

Definition at line 408 of file UnitSelecter.cpp.

References ASR::Team::getActiveControlGroup(), ASR::ControlGroup::getNumUnits(), ASR::ControlGroup::getUnit(), and mTeam.

Referenced by updateBeforeFrame().

        {
                ControlGroup* activeGroup = mTeam->getActiveControlGroup ();

                for ( unsigned int i = 0; i < activeGroup->getNumUnits (); i++ )
                {
                        activeGroup->getUnit( i )->getEntity ()->getParentSceneNode ()->showBoundingBox ( false );
                }
        }

void ASR::UnitSelecter::_unselectDeadUnits

(

)

[private]

Definition at line 332 of file UnitSelecter.cpp.

References _unselectUnit(), ASR::Team::createControlGroup(), ASR::Team::destroyActiveControlGroup(), ASR::Team::getActiveControlGroup(), ASR::ControlGroup::getNumUnits(), ASR::ControlGroup::getUnit(), mTeam, ASR::ControlGroup::removeUnit(), and ASR::Team::setActiveControlGroup().

Referenced by updateBeforeFrame().

        {
                ControlGroup* activeGroup = mTeam->getActiveControlGroup ();
                if ( activeGroup->getNumUnits () == 0 )
                        return;

                vector<Unit*> deadUnits;
                Unit* curUnit;
                for ( unsigned int i = 0; i < activeGroup->getNumUnits(); i++ )
                {
                        curUnit = activeGroup->getUnit(i);
                        if ( curUnit->isDead() )
                        {
                                deadUnits.push_back ( curUnit );
                        }
                }

                // Simply replace the entire active group with a blank one
                if ( deadUnits.size() == activeGroup->getNumUnits () )
                {
                        mTeam->destroyActiveControlGroup ();
                        mTeam->setActiveControlGroup ( mTeam->createControlGroup () );
                }
                else
                {
                        for ( unsigned int i = 0; i < deadUnits.size(); i++ )
                        {
                                activeGroup->removeUnit ( deadUnits[i] );
                                _unselectUnit ( deadUnits[i] );
                        }
                }
        }

void ASR::UnitSelecter::_unselectUnit

(

Unit *

unit

)

[private]

Definition at line 447 of file UnitSelecter.cpp.

Referenced by _unselectDeadUnits().

        {
                SceneNode* parentNode = unit->getEntity()->getParentSceneNode ();
                parentNode->showBoundingBox( false );
        }

bool ASR::UnitSelecter::updateAfterFrame	(	const FrameEvent &	e,
		const InputReader *	inputDevice
	)			[virtual]

Implements ASR::Updater.

Definition at line 325 of file UnitSelecter.cpp.

        {
                return true;
        }

bool ASR::UnitSelecter::updateBeforeFrame	(	const FrameEvent &	e,
		const InputReader *	inputDevice
	)			[virtual]

Called each frame to perform the "update".

Return true if rendering should continue, false otherwise.

Implements ASR::Updater.

Definition at line 257 of file UnitSelecter.cpp.

References _createActiveGroup(), _moveUnits(), _selectUnits(), _unselectCurrect(), _unselectDeadUnits(), mAbsMouse, mBounds, mCamera, mFirstPoint, mNodes, MOUSE_SPEED, mOverlay, and mSelectedUnits.

        {
                int screenWidth = mCamera->getViewport()->getActualWidth() / 2;
                int screenHeight = mCamera->getViewport()->getActualHeight() / 2;

                // TODO
                //              Abstract this into a Cursor class
                const int MOUSE_SPEED = 2;
                mAbsMouse[0] += inputDevice->getMouseRelativeX() * MOUSE_SPEED;
                mAbsMouse[1] += inputDevice->getMouseRelativeY() * MOUSE_SPEED;

                if ( mAbsMouse[0] < -screenWidth )
                        mAbsMouse[0] = -screenWidth;
                if ( mAbsMouse[0] > screenWidth )
                        mAbsMouse[0] = screenWidth;

                if ( mAbsMouse[1] < -screenHeight )
                        mAbsMouse[1] = -screenHeight;
                if ( mAbsMouse[1] > screenHeight )
                        mAbsMouse[1] = screenHeight;

                static bool rightUp = true;

                // The right button was pushed
                if ( inputDevice->getMouseButton( 1 ) && rightUp )
                {
                        rightUp = false;
                        _moveUnits ();
                }

                // The left button is currently down
                else if ( inputDevice->getMouseButton( 0 ) )
                {
                        mSelectedUnits.clear ();

                        mNodes[1]->setVisible( true );
                        mNodes[2]->setVisible( true );

                        _unselectCurrect ();
                        _selectUnits ();
                        _createActiveGroup ();

                        mSelectedUnits.clear ();
                }
                // Otherwise, they aren't trying to select anything
                else
                {
                        mFirstPoint = false;

                        mOverlay->hide ();
                        mNodes[1]->setVisible( false );
                        mNodes[2]->setVisible( false );

                        // Reset our bounds
                        mBounds[0].x = -1;
                        mBounds[1].x = -1;
                }

                if ( !inputDevice->getMouseButton(1) )
                        rightUp = true;

                _unselectDeadUnits ();

                return true;
        }

---

Member Data Documentation

float ASR::UnitSelecter::mAbsMouse[2] [private]

Definition at line 53 of file UnitSelecter.h.

Referenced by _getMousePosition(), _selectUnits(), UnitSelecter(), and updateBeforeFrame().

Vector2 ASR::UnitSelecter::mBounds[2] [private]

Definition at line 60 of file UnitSelecter.h.

Referenced by _selectUnits(), UnitSelecter(), and updateBeforeFrame().

Camera* ASR::UnitSelecter::mCamera [private]

Definition at line 47 of file UnitSelecter.h.

Referenced by _getMousePosition(), _selectUnits(), and updateBeforeFrame().

Entity* ASR::UnitSelecter::mCubes[5] [private]

Definition at line 41 of file UnitSelecter.h.

Referenced by UnitSelecter().

bool ASR::UnitSelecter::mFirstPoint [private]

Definition at line 54 of file UnitSelecter.h.

Referenced by _selectUnits(), UnitSelecter(), and updateBeforeFrame().

SceneNode* ASR::UnitSelecter::mNodes[5] [private]

Definition at line 42 of file UnitSelecter.h.

Referenced by _getMousePosition(), _selectUnits(), UnitSelecter(), and updateBeforeFrame().

Overlay* ASR::UnitSelecter::mOverlay [private]

Definition at line 57 of file UnitSelecter.h.

Referenced by _selectUnits(), UnitSelecter(), and updateBeforeFrame().

RaySceneQuery* ASR::UnitSelecter::mRayQuery [private]

Definition at line 49 of file UnitSelecter.h.

Referenced by _getMousePosition(), _selectUnits(), UnitSelecter(), and ~UnitSelecter().

vector<Unit*> ASR::UnitSelecter::mSelectedUnits [private]

Definition at line 62 of file UnitSelecter.h.

Referenced by _createActiveGroup(), _selectUnits(), and updateBeforeFrame().

Team* ASR::UnitSelecter::mTeam [private]

Definition at line 50 of file UnitSelecter.h.

Referenced by _createActiveGroup(), _moveUnits(), _selectUnits(), _unselectCurrect(), and _unselectDeadUnits().

---

The documentation for this class was generated from the following files:

• D:/simple_rts/include/UnitSelecter.h
• D:/simple_rts/src/UnitSelecter.cpp

---