diff options
Diffstat (limited to 'game/client/tf/tf_hud_sapper_charge.cpp')
| -rw-r--r-- | game/client/tf/tf_hud_sapper_charge.cpp | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/game/client/tf/tf_hud_sapper_charge.cpp b/game/client/tf/tf_hud_sapper_charge.cpp new file mode 100644 index 0000000..ce88fe9 --- /dev/null +++ b/game/client/tf/tf_hud_sapper_charge.cpp @@ -0,0 +1,149 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: +// +// $NoKeywords: $ +//============================================================================= + +#include "cbase.h" +#include "hud.h" +#include "hudelement.h" +#include "c_tf_player.h" +#include "iclientmode.h" +#include "ienginevgui.h" +#include <vgui/ILocalize.h> +#include <vgui/ISurface.h> +#include <vgui/IVGui.h> +#include <vgui_controls/EditablePanel.h> +#include <vgui_controls/ProgressBar.h> +#include "tf_weaponbase.h" +//#include "c_tf_projectile_arrow.h" + +// memdbgon must be the last include file in a .cpp file!!! +#include "tier0/memdbgon.h" + +using namespace vgui; + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +class CHudSapperChargeMeter : public CHudElement, public EditablePanel +{ + DECLARE_CLASS_SIMPLE( CHudSapperChargeMeter, EditablePanel ); + +public: + CHudSapperChargeMeter( const char *pElementName ); + + virtual void ApplySchemeSettings( IScheme *scheme ); + virtual bool ShouldDraw( void ); + virtual void OnTick( void ); + +private: + vgui::ContinuousProgressBar *m_pChargeMeter; +}; + +#ifdef STAGING_ONLY +DECLARE_HUDELEMENT( CHudSapperChargeMeter ); +#endif + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +CHudSapperChargeMeter::CHudSapperChargeMeter( const char *pElementName ) : CHudElement( pElementName ), BaseClass( NULL, "HudSapperCharge" ) +{ + Panel *pParent = g_pClientMode->GetViewport(); + SetParent( pParent ); + + m_pChargeMeter = new ContinuousProgressBar( this, "ChargeMeter" ); + + SetHiddenBits( HIDEHUD_MISCSTATUS ); + + vgui::ivgui()->AddTickSignal( GetVPanel() ); + + RegisterForRenderGroup( "inspect_panel" ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CHudSapperChargeMeter::ApplySchemeSettings( IScheme *pScheme ) +{ + // load control settings... + LoadControlSettings( "resource/UI/HudSapperCharge.res" ); + + BaseClass::ApplySchemeSettings( pScheme ); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +bool CHudSapperChargeMeter::ShouldDraw( void ) +{ + C_TFPlayer *pPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( !pPlayer || !pPlayer->IsPlayerClass( TF_CLASS_SPY ) || !pPlayer->IsAlive() ) + return false; + + CTFWeaponBase *pWpn = pPlayer->GetActiveTFWeapon(); + if ( !pWpn ) + return false; + + int iWeaponID = pWpn->GetWeaponID(); + if ( iWeaponID != TF_WEAPON_BUILDER ) + return false; + + int iCustomHUD = 0; + CALL_ATTRIB_HOOK_INT_ON_OTHER( pWpn, iCustomHUD, custom_charge_meter ); + if ( !iCustomHUD ) + return false; + + ITFChargeUpWeapon *pChargeupWeapon = dynamic_cast<ITFChargeUpWeapon *>( pWpn ); + if ( !pChargeupWeapon || !pChargeupWeapon->CanCharge() ) + return false; + + // Only draw if we have any charge + if ( pChargeupWeapon->GetChargeBeginTime() <= 0 ) + return false; + + return CHudElement::ShouldDraw(); +} + +//----------------------------------------------------------------------------- +// Purpose: +//----------------------------------------------------------------------------- +void CHudSapperChargeMeter::OnTick( void ) +{ + C_TFPlayer *pPlayer = C_TFPlayer::GetLocalTFPlayer(); + + if ( !pPlayer ) + return; + + CTFWeaponBase *pWpn = pPlayer->GetActiveTFWeapon(); + ITFChargeUpWeapon *pChargeupWeapon = dynamic_cast< ITFChargeUpWeapon *>( pWpn ); + + if ( !pWpn || !pChargeupWeapon ) + return; + + if ( m_pChargeMeter ) + { + float flChargeMaxTime = pChargeupWeapon->GetChargeMaxTime(); + + if ( flChargeMaxTime != 0 ) + { + float flChargeBeginTime = pChargeupWeapon->GetChargeBeginTime(); + + if ( flChargeBeginTime > 0 ) + { + float flTimeCharged = MAX( 0, gpGlobals->curtime - flChargeBeginTime ); + flTimeCharged = MIN( flTimeCharged, flChargeMaxTime ); + float flPercentCharged = MIN( 1.0, flTimeCharged / flChargeMaxTime ); + + m_pChargeMeter->SetProgress( flPercentCharged ); + } + else + { + m_pChargeMeter->SetProgress( 0.0f ); + } + } + } +}
\ No newline at end of file |