1HEADmaster

1 files changed, 164 insertions, 0 deletions

diff --git a/game/client/tf2/c_obj_resourcepump.cpp b/game/client/tf2/c_obj_resourcepump.cpp
new file mode 100644
index 0000000..13858ce
--- /dev/null
+++ b/game/client/tf2/c_obj_resourcepump.cpp
@@ -0,0 +1,164 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================//
+#include "cbase.h"
+#include "hud.h"
+#include "c_obj_resourcepump.h"
+#include "commanderoverlay.h"
+#include "vgui_healthbar.h"
+#include "ObjectControlPanel.h"
+#include "tf_shareddefs.h"
+#include "vgui_bitmapbutton.h"
+#include "C_Func_Resource.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+BEGIN_RECV_TABLE_NOBASE( C_ObjectResourcePump, DT_ResourcePumpTeamOnlyVars )
+	RecvPropInt( RECVINFO(m_iPumpLevel) ),
+	RecvPropEHandle( RECVINFO(m_hResourceZone) ),
+END_RECV_TABLE()
+
+
+IMPLEMENT_CLIENTCLASS_DT(C_ObjectResourcePump, DT_ResourcePump, CObjectResourcePump)
+	RecvPropDataTable( "teamonly", 0, 0, &REFERENCE_RECV_TABLE( DT_ResourcePumpTeamOnlyVars ) )
+END_RECV_TABLE()
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+C_ObjectResourcePump::C_ObjectResourcePump()
+{
+	m_iPumpLevel = 1;
+	m_pResourceBar = NULL;
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void C_ObjectResourcePump::SetDormant( bool bDormant )
+{
+	BaseClass::SetDormant( bDormant );
+	ENTITY_PANEL_ACTIVATE( "resource_pump", !bDormant );
+}
+
+
+//-----------------------------------------------------------------------------
+// Control screen 
+//-----------------------------------------------------------------------------
+class CResourcePumpControlPanel : public CObjectControlPanel
+{
+	DECLARE_CLASS( CResourcePumpControlPanel, CObjectControlPanel );
+
+public:
+	CResourcePumpControlPanel( vgui::Panel *parent, const char *panelName );
+	virtual bool Init( KeyValues* pKeyValues, VGuiScreenInitData_t* pInitData );
+	virtual void OnTick();
+	virtual void OnCommand( const char *command );
+
+	void		 Upgrade( void );
+
+private:
+	vgui::Button *m_pUpgradeButton;
+	vgui::Label *m_pResourcesLabel;
+};
+
+
+DECLARE_VGUI_SCREEN_FACTORY( CResourcePumpControlPanel, "resourcepump_control_panel" );
+
+
+//-----------------------------------------------------------------------------
+// Constructor: 
+//-----------------------------------------------------------------------------
+CResourcePumpControlPanel::CResourcePumpControlPanel( vgui::Panel *parent, const char *panelName )
+	: BaseClass( parent, "CResourcePumpControlPanel" ) 
+{
+}
+
+//-----------------------------------------------------------------------------
+// Initialization 
+//-----------------------------------------------------------------------------
+bool CResourcePumpControlPanel::Init( KeyValues* pKeyValues, VGuiScreenInitData_t* pInitData )
+{
+	m_pUpgradeButton = new CBitmapButton( this, "UpgradeButton", "Upgrade" );
+	m_pResourcesLabel = new vgui::Label( this, "ResourcesLabel", "Resources: 0" );
+
+	if (!BaseClass::Init(pKeyValues, pInitData))
+		return false;
+
+	// ROBIN: Removed upgrading for now
+	m_pUpgradeButton->SetVisible( false );
+
+	return true;
+}
+
+
+//-----------------------------------------------------------------------------
+// Frame-based update
+//-----------------------------------------------------------------------------
+void CResourcePumpControlPanel::OnTick()
+{
+	BaseClass::OnTick();
+	
+	C_BaseObject *pObj = GetOwningObject();
+	if (!pObj)
+		return;
+
+	Assert( dynamic_cast<C_ObjectResourcePump*>(pObj) );
+	C_ObjectResourcePump *pPump = static_cast<C_ObjectResourcePump*>(pObj);
+
+	char buf[256];
+	int iPumpLevel = pPump->GetLevel();
+	int iCost = CalculateObjectUpgrade( OBJ_RESOURCEPUMP, iPumpLevel );
+	if ( iCost )
+	{
+		Q_snprintf( buf, sizeof( buf ), "Upgrade to Level %d\nCost: %d", iPumpLevel+1, iCost );
+	}
+	else
+	{
+		Q_snprintf( buf, sizeof( buf ), "Level %d", iPumpLevel );
+	}
+
+	m_pUpgradeButton->SetText( buf );
+
+	C_ResourceZone *pResourceZone = pPump->GetResourceZone();
+	if (pResourceZone)
+	{
+		Q_snprintf( buf, sizeof( buf ), "Resources: %d", pResourceZone->m_nResourcesLeft );
+		m_pResourcesLabel->SetText( buf );
+	}
+	else
+	{
+		m_pResourcesLabel->SetText( "Resources: 0" );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Dismantles the object 
+//-----------------------------------------------------------------------------
+void CResourcePumpControlPanel::Upgrade( void )
+{
+	C_BaseObject *pObj = GetOwningObject();
+	if (pObj)
+	{
+		pObj->SendClientCommand( "upgrade" );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Button click handlers
+//-----------------------------------------------------------------------------
+void CResourcePumpControlPanel::OnCommand( const char *command )
+{
+	if (!Q_strnicmp(command, "Upgrade", 7))
+	{
+		Upgrade();
+		return;
+	}
+
+	BaseClass::OnCommand(command);
+}
+