diff options
Diffstat (limited to 'game/client/tf/vgui/tf_teammenu.cpp')
| -rw-r--r-- | game/client/tf/vgui/tf_teammenu.cpp | 901 |
1 files changed, 901 insertions, 0 deletions
diff --git a/game/client/tf/vgui/tf_teammenu.cpp b/game/client/tf/vgui/tf_teammenu.cpp new file mode 100644 index 0000000..10be493 --- /dev/null +++ b/game/client/tf/vgui/tf_teammenu.cpp @@ -0,0 +1,901 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: +// +// $NoKeywords: $ +//============================================================================= + +#include "cbase.h" + +#include <vgui_controls/Label.h> +#include <vgui_controls/Button.h> +#include <vgui_controls/ImagePanel.h> +#include <vgui_controls/RichText.h> +#include <vgui_controls/Frame.h> +#include <vgui/IScheme.h> +#include <game/client/iviewport.h> +#include <vgui/IVGui.h> +#include <KeyValues.h> +#include <filesystem.h> +#include <vgui_controls/AnimationController.h> +#include "iclientmode.h" +#include "clientmode_shared.h" +#include "inputsystem/iinputsystem.h" + +#include "vguicenterprint.h" +#include "tf_controls.h" +#include "basemodelpanel.h" +#include "tf_teammenu.h" +#include <convar.h> +#include "IGameUIFuncs.h" // for key bindings +#include "hud.h" // for gEngfuncs +#include "c_tf_player.h" +#include "tf_gamerules.h" +#include "c_team.h" +#include "tf_hud_notification_panel.h" +#include "iinput.h" + +//----------------------------------------------------------------------------- +// Purpose: Constructor +//----------------------------------------------------------------------------- +CTFTeamButton::CTFTeamButton( vgui::Panel *parent, const char *panelName ) : CExButton( parent, panelName, "" ) +{ + m_szModelPanel[0] = '\0'; + m_iTeam = TEAM_UNASSIGNED; + m_flHoverTimeToWait = -1; + m_flHoverTime = -1; + m_bMouseEntered = false; + m_bTeamDisabled = false; + + vgui::ivgui()->AddTickSignal( GetVPanel(), 100 ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::ApplySettings( KeyValues *inResourceData ) +{ + BaseClass::ApplySettings( inResourceData ); + + Q_strncpy( m_szModelPanel, inResourceData->GetString( "associated_model", "" ), sizeof( m_szModelPanel ) ); + m_iTeam = inResourceData->GetInt( "team", TEAM_UNASSIGNED ); + m_flHoverTimeToWait = inResourceData->GetFloat( "hover", -1 ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::ApplySchemeSettings( vgui::IScheme *pScheme ) +{ + BaseClass::ApplySchemeSettings( pScheme ); + + SetDefaultColor( GetFgColor(), Color( 0, 0, 0, 0 ) ); + SetArmedColor( GetButtonFgColor(), Color( 0, 0, 0, 0 ) ); + SetDepressedColor( GetButtonFgColor(), Color( 0, 0, 0, 0 ) ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::SendAnimation( const char *pszAnimation ) +{ + Panel *pParent = GetParent(); + if ( pParent ) + { + CModelPanel *pModel = dynamic_cast< CModelPanel* >( pParent->FindChildByName( m_szModelPanel ) ); + if ( pModel ) + { + KeyValues *kvParms = new KeyValues( "SetAnimation" ); + if ( kvParms ) + { + kvParms->SetString( "animation", pszAnimation ); + PostMessage( pModel, kvParms ); + } + } + } +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::SetDefaultAnimation( const char *pszName ) +{ + Panel *pParent = GetParent(); + if ( pParent ) + { + CModelPanel *pModel = dynamic_cast< CModelPanel* >( pParent->FindChildByName( m_szModelPanel ) ); + if ( pModel ) + { + pModel->SetDefaultAnimation( pszName ); + } + } +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +bool CTFTeamButton::IsTeamFull() +{ + if ( TFGameRules() && TFGameRules()->IsInArenaMode() == true && tf_arena_use_queue.GetBool() == true ) + return false; + + bool bRetVal = false; + + if ( ( m_iTeam > TEAM_UNASSIGNED ) && GetParent() ) + { + CTFTeamMenu *pTeamMenu = dynamic_cast< CTFTeamMenu* >( GetParent() ); + if ( pTeamMenu ) + { + bRetVal = ( m_iTeam == TF_TEAM_BLUE ) ? pTeamMenu->IsBlueTeamDisabled() : pTeamMenu->IsRedTeamDisabled(); + } + } + + return bRetVal; +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::OnCursorEntered() +{ + BaseClass::OnCursorEntered(); + + SetMouseEnteredState( true ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::OnCursorExited() +{ + BaseClass::OnCursorExited(); + + SetMouseEnteredState( false ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::SetMouseEnteredState( bool state ) +{ + if ( state ) + { + m_bMouseEntered = true; + + if ( m_flHoverTimeToWait > 0 ) + { + m_flHoverTime = gpGlobals->curtime + m_flHoverTimeToWait; + } + else + { + m_flHoverTime = -1; + } + + if ( m_bTeamDisabled ) + { + SendAnimation( "enter_disabled" ); + } + else + { + SendAnimation( "enter_enabled" ); + } + } + else + { + m_bMouseEntered = false; + m_flHoverTime = -1; + + if ( m_bTeamDisabled ) + { + SendAnimation( "exit_disabled" ); + } + else + { + SendAnimation( "exit_enabled" ); + } + } +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamButton::OnTick() +{ + if ( GetParent() && !GetParent()->IsVisible() ) + return; + + // check to see if our state has changed + bool bDisabled = IsTeamFull(); + + if ( bDisabled != m_bTeamDisabled ) + { + m_bTeamDisabled = bDisabled; + + if ( m_bMouseEntered ) + { + // something has changed, so reset our state + SetMouseEnteredState( true ); + } + else + { + // the mouse isn't currently over the button, but we should update the status + if ( m_bTeamDisabled ) + { + SendAnimation( "idle_disabled" ); + } + else + { + SendAnimation( "idle_enabled" ); + } + } + } + + if ( ( m_flHoverTime > 0 ) && ( m_flHoverTime < gpGlobals->curtime ) ) + { + m_flHoverTime = -1; + + if ( m_bTeamDisabled ) + { + SendAnimation( "hover_disabled" ); + } + else + { + SendAnimation( "hover_enabled" ); + } + } +} + +//----------------------------------------------------------------------------- +// Purpose: Constructor +//----------------------------------------------------------------------------- +CTFTeamMenu::CTFTeamMenu( IViewPort *pViewPort ) : CTeamMenu( pViewPort ) +{ + SetMinimizeButtonVisible( false ); + SetMaximizeButtonVisible( false ); + SetCloseButtonVisible( false ); + SetVisible( false ); + SetKeyBoardInputEnabled( true ); + + m_iTeamMenuKey = BUTTON_CODE_INVALID; + + m_pBlueTeamButton = new CTFTeamButton( this, "teambutton0" ); + m_pRedTeamButton = new CTFTeamButton( this, "teambutton1" ); + m_pAutoTeamButton = new CTFTeamButton( this, "teambutton2" ); + m_pSpecTeamButton = new CTFTeamButton( this, "teambutton3" ); + m_pSpecLabel = new CExLabel( this, "TeamMenuSpectate", "" ); + +#ifdef _X360 + m_pFooter = new CTFFooter( this, "Footer" ); +#else + m_pCancelButton = new CExButton( this, "CancelButton", "#TF_Cancel" ); + + m_pHighlanderLabel = new CExLabel( this, "HighlanderLabel", "" ); + m_pHighlanderLabelShadow = new CExLabel( this, "HighlanderLabelShadow", "" ); + m_pTeamsFullLabel = new CExLabel( this, "TeamsFullLabel", "" ); + m_pTeamsFullLabelShadow = new CExLabel( this, "TeamsFullLabelShadow", "" ); + m_pTeamsFullArrow = new CTFImagePanel( this, "TeamsFullArrow" ); + +#endif + + vgui::ivgui()->AddTickSignal( GetVPanel(), 100 ); + + m_bRedDisabled = false; + m_bBlueDisabled = false; + + if ( g_pInputSystem && ::input->IsSteamControllerActive() ) + { + LoadControlSettings( "Resource/UI/Teammenu_SC.res" ); + SetMouseInputEnabled( false ); + } + else + { + LoadControlSettings( "Resource/UI/Teammenu.res" ); + SetMouseInputEnabled( true ); + } + + ListenForGameEvent( "server_spawn" ); +} + +//----------------------------------------------------------------------------- +// Purpose: Destructor +//----------------------------------------------------------------------------- +CTFTeamMenu::~CTFTeamMenu() +{ +} + +//----------------------------------------------------------------------------- +// Purpose: Destructor +//----------------------------------------------------------------------------- +void CTFTeamMenu::ApplySchemeSettings( IScheme *pScheme ) +{ + BaseClass::ApplySchemeSettings( pScheme ); + + if ( ::input->IsSteamControllerActive() ) + { + LoadControlSettings( "Resource/UI/Teammenu_SC.res" ); + m_pCancelHintIcon = dynamic_cast< CSCHintIcon* >( FindChildByName( "CancelHintIcon" ) ); + m_pJoinAutoHintIcon = dynamic_cast< CSCHintIcon* >( FindChildByName( "JoinAutoHintIcon" ) ); + m_pJoinBluHintIcon = dynamic_cast< CSCHintIcon* >( FindChildByName( "JoinBluHintIcon" ) ); + m_pJoinRedHintIcon = dynamic_cast< CSCHintIcon* >( FindChildByName( "JoinRedHintIcon" ) ); + m_pJoinSpectatorsHintIcon = dynamic_cast< CSCHintIcon* >( FindChildByName( "JoinSpectatorsHintIcon" ) ); + + SetMouseInputEnabled( false ); + } + else + { + LoadControlSettings( "Resource/UI/Teammenu.res" ); + + m_pCancelHintIcon = nullptr; + m_pJoinAutoHintIcon = m_pJoinRedHintIcon = m_pJoinBluHintIcon = m_pJoinSpectatorsHintIcon = nullptr; + + SetMouseInputEnabled( true ); + } + + Update(); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamMenu::ShowPanel( bool bShow ) +{ + if ( BaseClass::IsVisible() == bShow ) + return; + + if ( !gameuifuncs || !gViewPortInterface || !engine ) + return; + + if ( bShow ) + { + if ( !C_TFPlayer::GetLocalTFPlayer() ) + return; + + bool bDisallowChange = false; + if ( C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() >= FIRST_GAME_TEAM ) + { + const IMatchGroupDescription* pMatchDesc = GetMatchGroupDescription( TFGameRules()->GetCurrentMatchGroup() ); + if ( pMatchDesc && !pMatchDesc->m_params.m_bAllowTeamChange ) + { + bDisallowChange = true; + } + } + + if ( ( TFGameRules()->State_Get() == GR_STATE_TEAM_WIN + && C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() != TFGameRules()->GetWinningTeam() + && C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() != TEAM_SPECTATOR + && C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() != TEAM_UNASSIGNED ) + || TFGameRules()->State_Get() == GR_STATE_GAME_OVER + // [msmith] Don't allow the player to switch teams when in training. + || TFGameRules()->IsInTraining() + // or if they are coaching + || C_TFPlayer::GetLocalTFPlayer()->m_bIsCoaching + || bDisallowChange + ) + { + SetVisible( false ); + + CHudNotificationPanel *pNotifyPanel = GET_HUDELEMENT( CHudNotificationPanel ); + if ( pNotifyPanel ) + { + pNotifyPanel->SetupNotifyCustom( "#TF_CantChangeTeamNow", "ico_notify_flag_moving", C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() ); + } + + return; + } + + extern void Coaching_CheckIfEligibleForCoaching(); + Coaching_CheckIfEligibleForCoaching(); + + gViewPortInterface->ShowPanel( PANEL_CLASS_RED, false ); + gViewPortInterface->ShowPanel( PANEL_CLASS_BLUE, false ); + + engine->CheckPoint( "TeamMenu" ); + + // Force us to reload our scheme, in case Steam Controller stuff has changed. + InvalidateLayout( true, true ); + + Activate(); + + // get key bindings if shown + m_iTeamMenuKey = gameuifuncs->GetButtonCodeForBind( "changeteam" ); + m_iScoreBoardKey = gameuifuncs->GetButtonCodeForBind( "showscores" ); + + switch ( C_TFPlayer::GetLocalTFPlayer()->GetTeamNumber() ) + { + case TF_TEAM_BLUE: + if ( ::input->EnableJoystickMode() ) + { + m_pBlueTeamButton->OnCursorEntered(); + m_pBlueTeamButton->SetDefaultAnimation( "enter_enabled" ); + } + GetFocusNavGroup().SetCurrentFocus( m_pBlueTeamButton->GetVPanel(), m_pBlueTeamButton->GetVPanel() ); + break; + + case TF_TEAM_RED: + if ( ::input->EnableJoystickMode() ) + { + m_pRedTeamButton->OnCursorEntered(); + m_pRedTeamButton->SetDefaultAnimation( "enter_enabled" ); + } + GetFocusNavGroup().SetCurrentFocus( m_pRedTeamButton->GetVPanel(), m_pRedTeamButton->GetVPanel() ); + break; + + default: + if ( ::input->EnableJoystickMode() ) + { + m_pAutoTeamButton->OnCursorEntered(); + m_pAutoTeamButton->SetDefaultAnimation( "enter_enabled" ); + } + GetFocusNavGroup().SetCurrentFocus( m_pAutoTeamButton->GetVPanel(), m_pAutoTeamButton->GetVPanel() ); + break; + } + + ActivateSelectIconHint( GetFocusNavGroup().GetCurrentFocus() ? GetFocusNavGroup().GetCurrentFocus()->GetTabPosition() : -1 ); + } + else + { + SetVisible( false ); + + SetHighlanderTeamsFullPanels( false, true ); + + if ( ::input->EnableJoystickMode() ) + { + // Close the door behind us + CTFTeamButton *pButton = dynamic_cast< CTFTeamButton *> ( GetFocusNavGroup().GetCurrentFocus() ); + if ( pButton ) + { + pButton->OnCursorExited(); + } + } + } +} + +//----------------------------------------------------------------------------- +// Purpose: called to update the menu with new information +//----------------------------------------------------------------------------- +void CTFTeamMenu::Update( void ) +{ + BaseClass::Update(); + + C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( pLocalPlayer && ( pLocalPlayer->GetTeamNumber() != TEAM_UNASSIGNED ) ) + { +#ifdef _X360 + if ( m_pFooter ) + { + m_pFooter->ShowButtonLabel( "cancel", true ); + } +#else + if ( m_pCancelButton ) + { + m_pCancelButton->SetVisible( true ); + if ( m_pCancelHintIcon ) + { + m_pCancelHintIcon->SetVisible( true ); + } + } +#endif + } + else + { +#ifdef _X360 + if ( m_pFooter ) + { + m_pFooter->ShowButtonLabel( "cancel", false ); + } +#else + if ( m_pCancelButton && m_pCancelButton->IsVisible() ) + { + m_pCancelButton->SetVisible( false ); + if ( m_pCancelHintIcon ) + { + m_pCancelHintIcon->SetVisible( false ); + } + } +#endif + } +} + +#ifdef _X360 +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamMenu::Join_Team( const CCommand &args ) +{ + if ( args.ArgC() > 1 ) + { + char cmd[256]; + Q_snprintf( cmd, sizeof( cmd ), "jointeam_nomenus %s", args.Arg( 1 ) ); + OnCommand( cmd ); + } +} +#endif + +//----------------------------------------------------------------------------- +// Purpose: chooses and loads the text page to display that describes mapName map +//----------------------------------------------------------------------------- +void CTFTeamMenu::LoadMapPage( const char *mapName ) +{ + +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamMenu::OnKeyCodePressed( KeyCode code ) +{ + if ( ( m_iTeamMenuKey != BUTTON_CODE_INVALID && m_iTeamMenuKey == code ) || + code == KEY_XBUTTON_BACK || + code == KEY_XBUTTON_B || + code == STEAMCONTROLLER_B ) + { + C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( pLocalPlayer && ( pLocalPlayer->GetTeamNumber() != TEAM_UNASSIGNED ) ) + { + ShowPanel( false ); + } + } + else if( code == KEY_SPACE || code == STEAMCONTROLLER_Y ) + { + engine->ClientCmd( "jointeam auto" ); + + ShowPanel( false ); + OnClose(); + } + else if( code == KEY_XBUTTON_A || code == KEY_XBUTTON_RTRIGGER || code == STEAMCONTROLLER_A ) + { + // select the active focus + if ( GetFocusNavGroup().GetCurrentFocus() ) + { + ipanel()->SendMessage( GetFocusNavGroup().GetCurrentFocus()->GetVPanel(), new KeyValues( "PressButton" ), GetVPanel() ); + } + } + else if( code == KEY_XBUTTON_RIGHT || code == KEY_XSTICK1_RIGHT || code == STEAMCONTROLLER_DPAD_RIGHT ) + { + CTFTeamButton *pButton; + + pButton = dynamic_cast< CTFTeamButton *> ( GetFocusNavGroup().GetCurrentFocus() ); + if ( pButton ) + { + pButton->OnCursorExited(); + GetFocusNavGroup().RequestFocusNext( pButton->GetVPanel() ); + } + else + { + GetFocusNavGroup().RequestFocusNext( NULL ); + } + + pButton = dynamic_cast< CTFTeamButton * > ( GetFocusNavGroup().GetCurrentFocus() ); + if ( pButton ) + { + pButton->OnCursorEntered(); + } + + ActivateSelectIconHint( GetFocusNavGroup().GetCurrentFocus() ? GetFocusNavGroup().GetCurrentFocus()->GetTabPosition() : -1 ); + } + else if( code == KEY_XBUTTON_LEFT || code == KEY_XSTICK1_LEFT || code == STEAMCONTROLLER_DPAD_LEFT ) + { + CTFTeamButton *pButton; + + pButton = dynamic_cast< CTFTeamButton *> ( GetFocusNavGroup().GetCurrentFocus() ); + if ( pButton ) + { + pButton->OnCursorExited(); + GetFocusNavGroup().RequestFocusPrev( pButton->GetVPanel() ); + } + else + { + GetFocusNavGroup().RequestFocusPrev( NULL ); + } + + pButton = dynamic_cast< CTFTeamButton * > ( GetFocusNavGroup().GetCurrentFocus() ); + if ( pButton ) + { + pButton->OnCursorEntered(); + } + + ActivateSelectIconHint( GetFocusNavGroup().GetCurrentFocus() ? GetFocusNavGroup().GetCurrentFocus()->GetTabPosition() : -1 ); + } + else if ( m_iScoreBoardKey != BUTTON_CODE_INVALID && m_iScoreBoardKey == code ) + { + gViewPortInterface->ShowPanel( PANEL_SCOREBOARD, true ); + gViewPortInterface->PostMessageToPanel( PANEL_SCOREBOARD, new KeyValues( "PollHideCode", "code", code ) ); + } + else + { + BaseClass::OnKeyCodePressed( code ); + } +} + +//----------------------------------------------------------------------------- +// Purpose: Called when the user picks a team +//----------------------------------------------------------------------------- +void CTFTeamMenu::OnCommand( const char *command ) +{ + C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( Q_stricmp( command, "vguicancel" ) ) + { + // we're selecting a team, so make sure it's not the team we're already on before sending to the server + if ( pLocalPlayer && ( Q_strstr( command, "jointeam " ) ) ) + { + const char *pTeam = command + Q_strlen( "jointeam " ); + int iTeam = TEAM_INVALID; + + if ( Q_stricmp( pTeam, "spectate" ) == 0 ) + { + iTeam = TEAM_SPECTATOR; + } + else if ( Q_stricmp( pTeam, "red" ) == 0 ) + { + iTeam = TF_TEAM_RED; + } + else if ( Q_stricmp( pTeam, "blue" ) == 0 ) + { + iTeam = TF_TEAM_BLUE; + } + + if ( iTeam == TF_TEAM_RED && m_bRedDisabled ) + { + return; + } + + if ( iTeam == TF_TEAM_BLUE && m_bBlueDisabled ) + { + return; + } + + // are we selecting the team we're already on? + if ( pLocalPlayer->GetTeamNumber() != iTeam ) + { + engine->ClientCmd( command ); + } + } + else if ( pLocalPlayer && ( Q_strstr( command, "jointeam_nomenus " ) ) ) + { + engine->ClientCmd( command ); + } + } + + BaseClass::OnCommand( command ); + ShowPanel( false ); + OnClose(); +} + +void CTFTeamMenu::OnClose() +{ + C_BasePlayer *pLocalPlayer = C_BasePlayer::GetLocalPlayer(); + if ( pLocalPlayer ) + { + // Clear the HIDEHUD_HEALTH bit we hackily added. Turns out prediction + // was restoring these bits every frame. Unfortunately, prediction + // is off for karts which means the spell hud item would disappear if you + // brought up this menu and returned. + pLocalPlayer->m_Local.m_iHideHUD &= ~HIDEHUD_HEALTH; + } + + BaseClass::OnClose(); +} + +//----------------------------------------------------------------------------- +// Purpose: Activate the right selection hint icon, depending on the focus group number selected +//----------------------------------------------------------------------------- +void CTFTeamMenu::ActivateSelectIconHint( int focus_group_number ) +{ + if ( m_pJoinAutoHintIcon ) m_pJoinAutoHintIcon->SetVisible( false ); + if ( m_pJoinBluHintIcon ) m_pJoinBluHintIcon->SetVisible( false ); + if ( m_pJoinRedHintIcon ) m_pJoinRedHintIcon->SetVisible( false ); + if ( m_pJoinSpectatorsHintIcon ) m_pJoinSpectatorsHintIcon->SetVisible( false ); + + CSCHintIcon* icon = nullptr; + switch ( focus_group_number ) + { + case 1: icon = m_pJoinAutoHintIcon; break; + case 2: icon = m_pJoinSpectatorsHintIcon; break; + case 3: icon = m_pJoinBluHintIcon; break; + case 4: icon = m_pJoinRedHintIcon; break; + } + + if ( icon ) + { + icon->SetVisible( true ); + } +} + + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamMenu::SetHighlanderTeamsFullPanels( bool bTeamsFull, bool bForce /* = false */ ) +{ + if ( m_pTeamsFullLabel ) + { + if ( bForce || ( m_pTeamsFullLabel->IsVisible() != bTeamsFull ) ) + { + m_pTeamsFullLabel->SetVisible( bTeamsFull ); + } + } + + if ( m_pTeamsFullLabelShadow ) + { + if ( bForce || ( m_pTeamsFullLabelShadow->IsVisible() != bTeamsFull ) ) + { + m_pTeamsFullLabelShadow->SetVisible( bTeamsFull ); + } + } + + if ( !mp_allowspectators.GetBool() ) + { + // don't show the arrow if the server doesn't allow spectators + bTeamsFull = false; + } + + if ( m_pTeamsFullArrow ) + { + if ( bForce || ( m_pTeamsFullArrow->IsVisible() != bTeamsFull ) ) + { + m_pTeamsFullArrow->SetVisible( bTeamsFull ); + + if ( bTeamsFull ) + { + // turn on animation + g_pClientMode->GetViewportAnimationController()->StartAnimationSequence( "TeamsFullArrowAnimate" ); + } + else + { + // turn off animation + g_pClientMode->GetViewportAnimationController()->StartAnimationSequence( "TeamsFullArrowAnimateEnd" ); + } + } + } +} + +//----------------------------------------------------------------------------- +// Frame-based update +//----------------------------------------------------------------------------- +void CTFTeamMenu::OnTick() +{ + // update the number of players on each team + + // enable or disable buttons based on team limit + + if ( !IsVisible() ) + return; + + C_Team *pRed = GetGlobalTeam( TF_TEAM_RED ); + C_Team *pBlue = GetGlobalTeam( TF_TEAM_BLUE ); + + if ( !pRed || !pBlue ) + return; + + // set our team counts + SetDialogVariable( "bluecount", pBlue->Get_Number_Players() ); + SetDialogVariable( "redcount", pRed->Get_Number_Players() ); + + C_TFPlayer *pLocalPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( !pLocalPlayer ) + return; + + CTFGameRules *pRules = TFGameRules(); + + if ( !pRules ) + return; + + bool bHighlander = pRules->IsInHighlanderMode(); + + if ( m_pHighlanderLabel ) + { + if ( m_pHighlanderLabel->IsVisible() != bHighlander ) + { + m_pHighlanderLabel->SetVisible( bHighlander ); + } + } + + if ( m_pHighlanderLabelShadow ) + { + if ( m_pHighlanderLabelShadow->IsVisible() != bHighlander ) + { + m_pHighlanderLabelShadow->SetVisible( bHighlander ); + } + } + + // check if teams are unbalanced + m_bRedDisabled = m_bBlueDisabled = false; + + int iHeavyTeam, iLightTeam; + + bool bUnbalanced = pRules->AreTeamsUnbalanced( iHeavyTeam, iLightTeam ); + + int iCurrentTeam = pLocalPlayer->GetTeamNumber(); + + if ( ( bUnbalanced && iHeavyTeam == TF_TEAM_RED ) || + ( pRules->WouldChangeUnbalanceTeams( TF_TEAM_RED, iCurrentTeam ) ) || + ( bHighlander && GetGlobalTeam( TF_TEAM_RED )->GetNumPlayers() >= TF_LAST_NORMAL_CLASS - 1 ) || + ( pRules->IsMannVsMachineMode() && ( GetGlobalTeam( TF_TEAM_RED )->GetNumPlayers() >= kMVM_DefendersTeamSize ) ) ) + { + m_bRedDisabled = true; + } + + if ( ( bUnbalanced && iHeavyTeam == TF_TEAM_BLUE ) || + ( pRules->WouldChangeUnbalanceTeams( TF_TEAM_BLUE, iCurrentTeam ) ) || + ( bHighlander && GetGlobalTeam( TF_TEAM_BLUE )->GetNumPlayers() >= TF_LAST_NORMAL_CLASS - 1 ) || + ( pRules->IsMannVsMachineMode() ) ) + { + m_bBlueDisabled = true; + } + + bool bTeamsFull = m_bRedDisabled && m_bBlueDisabled; + SetHighlanderTeamsFullPanels( bHighlander && bTeamsFull ); + + if ( m_pSpecTeamButton && m_pSpecLabel && m_pAutoTeamButton ) + { + { + if ( mp_allowspectators.GetBool() ) + { + if ( !m_pSpecTeamButton->IsVisible() ) + { + m_pSpecTeamButton->SetVisible( true ); + m_pSpecLabel->SetVisible( true ); + } + + if ( !m_pAutoTeamButton->IsVisible() ) + { + m_pAutoTeamButton->SetVisible( true ); + } + } + else + { + if ( m_pSpecTeamButton->IsVisible() ) + { + m_pSpecTeamButton->SetVisible( false ); + m_pSpecLabel->SetVisible( false ); + } + + if ( bHighlander ) + { + if ( bTeamsFull ) + { + if ( m_pAutoTeamButton->IsVisible() ) + { + m_pAutoTeamButton->SetVisible( false ); + } + } + else + { + if ( !m_pAutoTeamButton->IsVisible() ) + { + m_pAutoTeamButton->SetVisible( true ); + } + } + } + } + } + } +} + +void CTFTeamMenu::OnThink() +{ + //Always hide the health... this needs to be done every frame because a message from the server keeps resetting this. + C_BasePlayer *pLocalPlayer = C_BasePlayer::GetLocalPlayer(); + if ( pLocalPlayer ) + { + pLocalPlayer->m_Local.m_iHideHUD |= HIDEHUD_HEALTH; + } + + BaseClass::OnThink(); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CTFTeamMenu::FireGameEvent( IGameEvent *event ) +{ + // when we are changing levels + if ( FStrEq( event->GetName(), "server_spawn" ) ) + { + SetHighlanderTeamsFullPanels( false, true ); + } +} |