source-engine/game/client/tf2/resourcezoneoverlay.cpp

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
// $Revision:     $
// $NoKeywords: $
//=============================================================================//
#include "cbase.h"
#include <VGUI_EntityPanel.h>
#include <KeyValues.h>
#include "commanderoverlay.h"
#include "clientmode_tfnormal.h"
#include "tf_shareddefs.h"
#include "shareddefs.h"
#include "c_func_resource.h"
#include "techtree.h"
#include "c_basetfplayer.h"
#include "vgui_HealthBar.h"
#include "vgui_bitmapimage.h"

// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"

class CResourceZoneOverlay : public CEntityPanel
{
	DECLARE_CLASS( CResourceZoneOverlay, CEntityPanel );

public:
	CResourceZoneOverlay( vgui::Panel *parent, const char *panelName );
	virtual ~CResourceZoneOverlay( void );

	bool Init( KeyValues* pKeyValues, C_BaseEntity* pEntity );
	bool InitResourceBitmaps( KeyValues* pKeyValues );

	void SetColor( int r, int g, int b, int a );
	void SetImage( BitmapImage *pImage );

	virtual void OnTick();

	virtual void Paint( void );
	virtual void PaintBackground( void ) {}

private:
	class CResourceBitmaps
	{
	public:
		CResourceBitmaps() : m_pImage(0) {}

		BitmapImage		*m_pImage;
		Color		m_Color;
	};

	struct Rect_t
	{
		int x, y, w, h;
	};

	bool ParseSingleResourceBitmap( KeyValues *pKeyValues, 
			const char *pResourceName, CResourceBitmaps *pResourceBitmap );
	bool ParseTeamResourceBitmaps( CResourceBitmaps *pT, KeyValues *pTeam );

	int					m_r, m_g, m_b, m_a;
	CHealthBarPanel		m_UsageBar;
	Rect_t				m_Icon;
	C_ResourceZone		*m_pResourceZone;
	CResourceBitmaps	m_pResourceBitmap;
	BitmapImage			*m_pImage;

};


//-----------------------------------------------------------------------------
// Class factory
//-----------------------------------------------------------------------------

DECLARE_OVERLAY_FACTORY( CResourceZoneOverlay, "resourcezone" );


//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------

CResourceZoneOverlay::CResourceZoneOverlay( vgui::Panel *parent, const char *panelName )
: BaseClass( parent, "CResourceZoneOverlay" )
{
	m_pImage = 0;
	SetPaintBackgroundEnabled( false );
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------

CResourceZoneOverlay::~CResourceZoneOverlay( void )
{
	if ( m_pResourceBitmap.m_pImage )
	{
		delete m_pResourceBitmap.m_pImage;
	}
}


//-----------------------------------------------------------------------------
// Parse class icons
//-----------------------------------------------------------------------------

bool CResourceZoneOverlay::ParseSingleResourceBitmap( KeyValues *pKeyValues, 
				const char *pResourceName, CResourceBitmaps *pResourceBitmap )
{
	const char *image;
	KeyValues *pResource;
	pResource = pKeyValues->FindKey( pResourceName );
	if ( !pResource )
		return false;

	image = pResource->GetString( "material" );
	if ( image && image[ 0 ] )
	{
		pResourceBitmap->m_pImage = new BitmapImage( GetVPanel(), image );
	}
	else
	{
		return( false );
	}

	return ParseRGBA( pResource, "color", pResourceBitmap->m_Color );
}

bool CResourceZoneOverlay::ParseTeamResourceBitmaps( CResourceBitmaps *pT, KeyValues *pTeam )
{
	if ( !ParseSingleResourceBitmap( pTeam, "Alpha", pT ) )
		return false;
	return true;
}

//-----------------------------------------------------------------------------
// Initialization
//-----------------------------------------------------------------------------

bool CResourceZoneOverlay::InitResourceBitmaps( KeyValues* pKeyValues )
{
//	char teamkey[ 128 ];
//	for (int i = 0; i < 3; ++i)
//	{
//		Q_snprintf( teamkey, sizeof( teamkey ), "Team%i", i );
//		KeyValues *pTeam = pKeyValues->getSection( teamkey );
//		if (pTeam)
		{
			if (!ParseTeamResourceBitmaps( &m_pResourceBitmap, pKeyValues ))
				return false;
		}
//	}

	return true;
}

//-----------------------------------------------------------------------------
// Initialization
//-----------------------------------------------------------------------------

bool CResourceZoneOverlay::Init( KeyValues* pKeyValues, C_BaseEntity* pEntity )
{
	if (!BaseClass::Init( pKeyValues, pEntity))
		return false;

	if (!pKeyValues)
		return false;

	// We gotta be attached to a resource zone
	m_pResourceZone = dynamic_cast<C_ResourceZone*>(GetEntity());
	if (!m_pResourceZone)
		return false;

	KeyValues* pUsage = pKeyValues->FindKey("Usage");
	if (!m_UsageBar.Init( pUsage ))
		return false;
	m_UsageBar.SetParent( this );

	// get the icon info...
	if (!ParseRect( pKeyValues, "iconposition", m_Icon.x, m_Icon.y, m_Icon.w, m_Icon.h ))
		return false;

	if (!InitResourceBitmaps( pKeyValues ))
		return false;

	SetImage( 0 );
	SetColor( m_pResourceBitmap.m_Color[0], m_pResourceBitmap.m_Color[1], m_pResourceBitmap.m_Color[2], m_pResourceBitmap.m_Color[3] );

	// we need updating
	return true;
}


//-----------------------------------------------------------------------------
// called when we're ticked...
//-----------------------------------------------------------------------------

void CResourceZoneOverlay::OnTick()
{
	// Update position
	CEntityPanel::OnTick();

	SetImage( m_pResourceBitmap.m_pImage );
	int r, g, b, a;
	m_pResourceBitmap.m_Color.GetColor( r, g, b, a );
	SetColor( r, g, b, a );

	m_UsageBar.SetHealth( m_pResourceZone->m_flClientResources );
}

//-----------------------------------------------------------------------------
// Purpose: 
// Input  : pImage - Class specific image
//-----------------------------------------------------------------------------

void CResourceZoneOverlay::SetImage( BitmapImage *pImage )
{
	m_pImage = pImage;
}

//-----------------------------------------------------------------------------
// Sets the draw color 
//-----------------------------------------------------------------------------

void CResourceZoneOverlay::SetColor( int r, int g, int b, int a )
{
	m_r = r;
	m_g = g;
	m_b = b;
	m_a = a;
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------

void CResourceZoneOverlay::Paint( void )
{
	if ( !m_pImage )
		return;

	ComputeAndSetSize();

	Color color;
	color.SetColor( m_r, m_g, m_b, m_a );
	m_pImage->SetPos( m_Icon.x, m_Icon.y );
	m_pImage->SetColor( color );
	m_pImage->DoPaint( GetVPanel() );
}