1 files changed, 355 insertions, 0 deletions
diff --git a/game/client/tf/tf_hud_arena_class_layout.cpp b/game/client/tf/tf_hud_arena_class_layout.cpp
new file mode 100644
index 0000000..e739147
--- /dev/null
+++ b/game/client/tf/tf_hud_arena_class_layout.cpp
@@ -0,0 +1,355 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================
+
+#include "cbase.h"
+#include "hud.h"
+#include "hud_macros.h"
+#include "c_tf_player.h"
+#include "c_tf_team.h"
+#include "c_tf_playerresource.h"
+#include "iclientmode.h"
+#include "ienginevgui.h"
+#include "tf_gamerules.h"
+#include <vgui/ILocalize.h>
+#include <vgui/ISurface.h>
+#include <vgui/IVGui.h>
+#include <vgui_controls/EditablePanel.h>
+#include "tf_hud_arena_class_layout.h"
+#include "tf_hud_menu_spy_disguise.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+using namespace vgui;
+
+const char *g_sImagesBlue[] = {
+	"",
+	"class_sel_sm_scout_blu",
+	"class_sel_sm_sniper_blu",
+	"class_sel_sm_soldier_blu",
+	"class_sel_sm_demo_blu",
+	"class_sel_sm_medic_blu",
+	"class_sel_sm_heavy_blu",
+	"class_sel_sm_pyro_blu",
+	"class_sel_sm_spy_blu",
+	"class_sel_sm_engineer_blu",
+	"",
+};
+
+const char *g_sImagesRed[] = {
+	"",
+	"class_sel_sm_scout_red",
+	"class_sel_sm_sniper_red",
+	"class_sel_sm_soldier_red",
+	"class_sel_sm_demo_red",
+	"class_sel_sm_medic_red",
+	"class_sel_sm_heavy_red",
+	"class_sel_sm_pyro_red",
+	"class_sel_sm_spy_red",
+	"class_sel_sm_engineer_red",
+	"",
+};
+
+
+DECLARE_HUDELEMENT( CHudArenaClassLayout );
+
+bool ArenaClassLayoutKeyInput( int down, ButtonCode_t keynum, const char *pszCurrentBinding )
+{
+	CHudArenaClassLayout *pArenaClassLayoutPanel = ( CHudArenaClassLayout * )GET_HUDELEMENT( CHudArenaClassLayout );
+
+	if ( pArenaClassLayoutPanel && pArenaClassLayoutPanel->ShouldDraw() )
+	{
+		return pArenaClassLayoutPanel->HandleKeyCodePressed( keynum );
+	}
+
+	return false;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+CHudArenaClassLayout::CHudArenaClassLayout( const char *pElementName ) : CHudElement( pElementName ), BaseClass( NULL, "HudArenaClassLayout" )
+{
+	Panel *pParent = g_pClientMode->GetViewport();
+	SetParent( pParent );
+
+	vgui::SETUP_PANEL( this );
+
+	SetKeyBoardInputEnabled( true );
+
+	vgui::ivgui()->AddTickSignal( GetVPanel(), 100 );
+
+	m_pBackground = new CTFImagePanel( this, "background" );
+	m_pLocalPlayerBG = new CTFImagePanel( this, "localPlayerBG" );
+	m_pTitle = new CExLabel( this, "title", "" );
+	m_pChangeLabel = new CExLabel( this, "changeLabel", "" );
+	m_pChangeLabelShadow = new CExLabel( this, "changeLabelShadow", "" );
+
+	char tempName[MAX_PATH];
+	for ( int i = 0 ; i < MAX_CLASS_IMAGES ; ++i )
+	{
+		Q_snprintf( tempName, sizeof( tempName ), "classImage%d", i );
+		m_ClassImages[i] = new CTFImagePanel( this, tempName );
+	}
+
+	SetVisible( false );
+
+	RegisterForRenderGroup( "mid" );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CHudArenaClassLayout::Init( void )
+{
+	CHudElement::Init();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CHudArenaClassLayout::ApplySchemeSettings( IScheme *pScheme )
+{
+	// load control settings...
+	LoadControlSettings( "resource/UI/HudArenaClassLayout.res" );
+
+	BaseClass::ApplySchemeSettings( pScheme );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CHudArenaClassLayout::PerformLayout( void )
+{
+	if ( !g_TF_PR )
+		return;
+
+	int i = 0;
+	CUtlVector<int> teamPlayers;
+	int nLocalPlayerTeam = GetLocalPlayerTeam();
+	int iImageIndex = 0;
+	CTFImagePanel *pImage = NULL;
+	int nImageXPos = 0, nImageYPos = 0;
+
+	if ( GetLocalPlayerTeam() <= LAST_SHARED_TEAM )
+		return;
+
+	int nClass = g_TF_PR->GetPlayerClass( GetLocalPlayerIndex() );
+	if ( nClass == TF_CLASS_UNDEFINED )
+		return;
+
+	// count the number of players on our team that have chosen a playerclass
+	for ( i = 1 ; i <= MAX_PLAYERS ; i++ )
+	{
+		if ( g_TF_PR->GetTeam( i ) == GetLocalPlayerTeam() )
+		{
+			if ( g_TF_PR->GetPlayerClass( i ) > TF_CLASS_UNDEFINED ) 
+			{
+				teamPlayers.AddToTail( i );
+			}
+		}
+
+		if ( teamPlayers.Count() >= MAX_CLASS_IMAGES )
+		{
+			break;
+		}
+	}
+
+	if ( teamPlayers.Count() > 0 )
+	{
+		int nBackgroundXPos, nBackgroundYPos, nBackgroundWide, nBackgroundTall;
+		int nImageWide = m_ClassImages[iImageIndex]->GetWide();
+		int nTotalWidth = ( teamPlayers.Count() * nImageWide ) + ( 2 * XRES( 10 ) ); // the XRES(10) is for the background to be scaled to cover the images on both ends
+		int nXPos = ( GetWide() - nTotalWidth ) * 0.5;
+
+		m_pBackground->GetBounds( nBackgroundXPos, nBackgroundYPos, nBackgroundWide, nBackgroundTall );
+		m_pBackground->SetBounds( nXPos, nBackgroundYPos, nTotalWidth, nBackgroundTall );
+
+		// this is where our first image will start
+		nXPos += XRES( 10 );
+
+		// the first image on the left is always the local player (and we'll have a special background behind it)
+		pImage = m_ClassImages[iImageIndex];
+		iImageIndex++;
+
+		if ( pImage )
+		{
+			pImage->SetVisible( true );
+			pImage->GetPos( nImageXPos, nImageYPos ); // only really care about the YPos here
+			pImage->SetPos( nXPos, nImageYPos );
+			pImage->SetImage( nLocalPlayerTeam == TF_TEAM_BLUE ? g_sImagesBlue[nClass] : g_sImagesRed[nClass] );
+
+			if ( teamPlayers.Count() > 1 )
+			{
+				// local player background
+				int nBGXPos, nBGYPos;
+				m_pLocalPlayerBG->SetVisible( true );
+				m_pLocalPlayerBG->GetPos( nBGXPos, nBGYPos );
+				m_pLocalPlayerBG->SetPos( nXPos, nBGYPos );
+			}
+			else
+			{
+				m_pLocalPlayerBG->SetVisible( false );
+			}
+
+			nXPos += nImageWide;
+		}
+
+		for ( i = 0 ; i < teamPlayers.Count() ; i++ )
+		{
+			int iPlayerIndex = teamPlayers[i];
+
+			if ( iPlayerIndex == GetLocalPlayerIndex() )
+				continue;
+
+			if ( iImageIndex >= MAX_CLASS_IMAGES )
+				continue;
+
+			if ( g_TF_PR->GetTeam( iPlayerIndex ) == GetLocalPlayerTeam() )
+			{
+				nClass = g_TF_PR->GetPlayerClass( iPlayerIndex );
+
+				if ( nClass == TF_CLASS_UNDEFINED )
+					continue;
+
+				pImage = m_ClassImages[iImageIndex];
+
+				if ( pImage )
+				{
+					pImage->SetVisible( true );
+					pImage->SetPos( nXPos, nImageYPos );
+					pImage->SetImage( nLocalPlayerTeam == TF_TEAM_BLUE ? g_sImagesBlue[nClass] : g_sImagesRed[nClass] );
+					nXPos += nImageWide;
+				}
+
+				iImageIndex++;
+			}
+		}
+	}
+
+	// turn off any unused images
+	while ( iImageIndex < MAX_CLASS_IMAGES )
+	{
+		pImage = m_ClassImages[iImageIndex];
+		if ( pImage )
+		{
+			pImage->SetVisible( false );
+		}
+
+		iImageIndex++;
+	}
+
+	C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer();
+	if ( pLocalPlayer && m_pChangeLabel && m_pChangeLabelShadow )
+	{
+		bool bShow = true;
+
+		if ( ( pLocalPlayer->m_Shared.GetArenaNumChanges() >= tf_arena_change_limit.GetInt() ) ||
+			 ( tf_arena_force_class.GetBool() == false ) )
+		{
+			bShow = false;
+		}
+
+		if ( m_pChangeLabel->IsVisible() != bShow )
+		{
+			m_pChangeLabel->SetVisible( bShow );
+			m_pChangeLabelShadow->SetVisible( bShow );
+		}
+	}
+
+	BaseClass::PerformLayout();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+bool CHudArenaClassLayout::ShouldDraw( void )
+{
+	if ( TFGameRules() == NULL )
+		return false;
+
+	if ( CHudElement::ShouldDraw() == false )
+		return false;
+
+	if ( TFGameRules()->State_Get() != GR_STATE_PREROUND )
+		return false;
+
+	if ( TFGameRules()->IsInArenaMode() == false )
+		return false;
+
+	if ( GetLocalPlayerTeam() > LAST_SHARED_TEAM )
+	{
+		if ( ( tf_arena_force_class.GetBool() == true ) || ( GetGlobalTeam( GetLocalPlayerTeam() )->Get_Number_Players() > 1 ) )
+		{
+			C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer();
+			if ( pLocalPlayer && pLocalPlayer->IsAlive() )
+			{
+				C_TFPlayerClass *pClass = pLocalPlayer->GetPlayerClass();
+				if ( pClass && pClass->GetClassIndex() != TF_CLASS_UNDEFINED )
+				{
+					return true;
+				}					
+			}
+		}
+	}
+
+	return false;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CHudArenaClassLayout::SetVisible( bool state )
+{
+	IGameEvent *event = gameeventmanager->CreateEvent( "show_class_layout" );
+	if ( event )
+	{
+		event->SetBool( "show", state );
+		gameeventmanager->FireEventClientSide( event );
+	}
+
+	BaseClass::SetVisible( state );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CHudArenaClassLayout::OnTick( void )
+{
+	bool bVisible = ShouldDraw();
+
+	if ( bVisible != IsVisible() )
+	{
+		SetVisible( bVisible );
+	}
+
+	if ( !bVisible )
+		return;
+	
+	InvalidateLayout( true );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+bool CHudArenaClassLayout::HandleKeyCodePressed( vgui::KeyCode code )
+{
+	if ( code == KEY_F4 )
+	{
+		if ( ShouldDraw() && ( tf_arena_force_class.GetBool() == true ) )
+		{
+			C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer();
+			if ( pLocalPlayer && ( pLocalPlayer->m_Shared.GetArenaNumChanges() < tf_arena_change_limit.GetInt() ) )
+			{
+				engine->ClientCmd( "arena_changeclass" );
+			}
+		}
+	}
+
+	return false;
+}
+