1 files changed, 143 insertions, 0 deletions
diff --git a/game/shared/tf/tf_weapon_sword.h b/game/shared/tf/tf_weapon_sword.h
new file mode 100644
index 0000000..1ea60dd
--- /dev/null
+++ b/game/shared/tf/tf_weapon_sword.h
@@ -0,0 +1,143 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+//=============================================================================
+
+#ifndef TF_WEAPON_SWORD_H
+#define TF_WEAPON_SWORD_H
+#ifdef _WIN32
+#pragma once
+#endif
+
+#include "tf_weaponbase_melee.h"
+#ifdef GAME_DLL
+#include "GameEventListener.h"
+#else
+#include "tf_viewmodel.h"
+#include "bone_setup.h"
+#include "tf_wearable_item_demoshield.h"
+#endif
+
+#ifdef CLIENT_DLL
+#define CTFSword C_TFSword
+#define CTFKatana C_TFKatana
+#endif
+
+//=============================================================================
+//
+// Decapitation melee weapon base class.
+//
+
+#ifdef GAME_DLL
+class CTFDecapitationMeleeWeaponBase : public CTFWeaponBaseMelee, public CGameEventListener
+#else
+class CTFDecapitationMeleeWeaponBase : public CTFWeaponBaseMelee
+#endif
+{
+public:
+
+	DECLARE_CLASS( CTFDecapitationMeleeWeaponBase, CTFWeaponBaseMelee );
+	
+	CTFDecapitationMeleeWeaponBase();
+
+	virtual void		Precache( void );
+	virtual int			GetWeaponID( void ) const			{ return TF_WEAPON_SWORD; }
+	virtual int			GetCustomDamageType() const			{ return TF_DMG_CUSTOM_DECAPITATION; }
+	virtual float		GetMeleeDamage( CBaseEntity *pTarget, int* piDamageType, int* piCustomDamage );
+	virtual Activity	TranslateViewmodelHandActivityInternal( Activity actBase );
+
+	virtual bool		CanDecapitate( void );
+	virtual void		SetupGameEventListeners( void );
+	virtual void		OnDecapitation( CTFPlayer *pDeadPlayer ) {}
+
+#ifdef GAME_DLL
+	virtual bool		Holster( CBaseCombatWeapon *pSwitchingTo = NULL );
+	virtual void		FireGameEvent( IGameEvent *event );
+#endif
+
+#ifdef CLIENT_DLL
+	virtual void		UpdateAttachmentModels( void );
+	virtual int			DrawOverriddenViewmodel( C_BaseViewModel *pViewmodel, int flags );
+#endif // CLIENT_DLL
+
+protected:
+	bool				m_bHolstering;
+
+private:
+#ifdef CLIENT_DLL
+	EHANDLE				m_hShield;
+#endif // CLIENT_DLL
+};
+
+//=============================================================================
+//
+// Sword class. (Consumes heads for speed/health bonus.)
+//
+class CTFSword : public CTFDecapitationMeleeWeaponBase
+{
+public:
+
+	DECLARE_CLASS( CTFSword, CTFDecapitationMeleeWeaponBase );
+	DECLARE_NETWORKCLASS(); 
+	DECLARE_PREDICTABLE();
+
+	CTFSword() {}
+	virtual ~CTFSword() {}
+
+	virtual void		WeaponReset( void );
+	virtual float		GetSwordSpeedMod( void );
+	virtual int			GetSwordHealthMod();
+	virtual int			GetSwingRange( void );
+	virtual void		OnDecapitation( CTFPlayer *pDeadPlayer );
+	
+	virtual bool		Deploy( void );
+
+	float				GetProgress( void ) { return 0.f; }
+	int					GetCount( void );
+	const char*			GetEffectLabelText( void ) { return "#TF_BERZERK"; }
+
+#ifdef CLIENT_DLL
+	virtual void		WeaponIdle( void );
+#endif
+
+private:
+
+	CTFSword( const CTFSword& ) {}
+
+#ifdef CLIENT_DLL
+	float				m_flNextIdleWavRoll;
+	int					m_iPrevWavDecap;
+#endif
+};
+
+//=============================================================================
+//
+// Sword class. (On head-take causes short-term soldier buff effects.)
+//
+class CTFKatana : public CTFDecapitationMeleeWeaponBase
+{
+public:
+
+	DECLARE_CLASS( CTFKatana, CTFDecapitationMeleeWeaponBase );
+	DECLARE_NETWORKCLASS(); 
+	DECLARE_PREDICTABLE();
+
+	CTFKatana();
+	virtual ~CTFKatana() {}
+
+	virtual bool	Deploy( void );
+	virtual float	GetMeleeDamage( CBaseEntity *pTarget, int* piDamageType, int* piCustomDamage );
+	virtual void	OnDecapitation( CTFPlayer *pDeadPlayer );
+
+	virtual int		GetActivityWeaponRole() const OVERRIDE;
+
+protected:
+	virtual int		GetSkinOverride() const;
+	
+private:
+	CNetworkVar( bool, m_bIsBloody );
+};
+
+
+#endif // TF_WEAPON_SWORD_H