1HEADmaster

1 files changed, 52 insertions, 0 deletions

diff --git a/game/client/tf2/hud_ammo.h b/game/client/tf2/hud_ammo.h
new file mode 100644
index 0000000..0c24d08
--- /dev/null
+++ b/game/client/tf2/hud_ammo.h
@@ -0,0 +1,52 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//=============================================================================//
+
+#ifndef HUD_AMMO_H
+#define HUD_AMMO_H
+
+#ifdef _WIN32
+#pragma once
+#endif
+
+// This is *like* a hud element, but isn't actually a hud element.
+// It's meant to be called from hud drawing methods in various places in the code
+class CHudAmmo
+{
+public:
+	// Accessor methods to set various state associated with the ammo display
+	void SetPrimaryAmmo( int nAmmoType, int nTotalAmmo, int nClipCount = -1, int nMaxClipCount = -1 );
+	void SetSecondaryAmmo( int nAmmoType, int nTotalAmmo, int nClipCount = -1, int nMaxClipCount = -1 );
+
+	bool	ShouldShowPrimaryClip() const;
+	bool	ShouldShowSecondary() const;
+
+	void	ShowHideHudControls();
+
+private:
+	int		m_nClip1;
+	int		m_nMaxClip1;
+	int		m_nTotalAmmo1;
+	int		m_nAmmoType1;
+
+	int		m_nClip2;
+	int		m_nMaxClip2;
+	int		m_nTotalAmmo2;
+ 	int		m_nAmmoType2;
+
+friend class CHudAmmoPrimary;
+friend class CHudAmmoPrimaryClip;
+friend class CHudAmmoSecondary;
+};
+
+//-----------------------------------------------------------------------------
+// Singleton...
+//-----------------------------------------------------------------------------
+CHudAmmo* GetHudAmmo();
+
+#endif // HUD_AMMO_H
+