1 files changed, 368 insertions, 0 deletions
diff --git a/game/shared/tf/tf_wheel_of_doom.h b/game/shared/tf/tf_wheel_of_doom.h
new file mode 100644
index 0000000..e05dabf
--- /dev/null
+++ b/game/shared/tf/tf_wheel_of_doom.h
@@ -0,0 +1,368 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose:
+//
+// $NoKeywords: $
+//=============================================================================//
+
+#ifndef TF_WHEEL_OF_DOOM_H
+#define TF_WHEEL_OF_DOOM_H
+#pragma once
+
+#include "GameEventListener.h"
+
+// Client specific.
+#ifdef CLIENT_DLL
+#define CWheelOfDoom C_WheelOfDoom
+#endif
+
+enum eWheelOfDoomEffectSkinIndex
+{
+	EFFECT_WHAMMY = 1,
+	EFFECT_JUMP_HEIGHT,
+	EFFECT_SMALL_HEAD,
+	EFFECT_SPEED,
+	EFFECT_LOW_GRAVITY,
+	EFFECT_BIG_HEAD,
+	EFFECT_UBER,
+	EFFECT_CRITS,
+	EFFECT_DANCE,
+	EFFECT_BASEBALLS,
+	EFFECT_DECAPITATE,
+	EFFECT_BARS,
+	EFFECT_BELLS,
+	EFFECT_SEVENS,
+	EFFECT_CHERRIES,
+
+	EFFECT_COUNT
+};
+
+
+#ifdef GAME_DLL
+
+class CWheelOfDoomSpiral;
+
+//=============================================================================
+//
+// Wheel of doom class
+//
+class CWheelOfDoom : public CBaseAnimating, public CGameEventListener
+{
+public:
+	DECLARE_CLASS( CWheelOfDoom, CBaseAnimating );
+	DECLARE_DATADESC();
+
+
+	CWheelOfDoom( void );
+	~CWheelOfDoom( void );
+
+	// Initialization
+	virtual void Spawn( void );
+	virtual void Precache( void );
+
+	// Thinks
+	void IdleThink( void );
+	void SpinThink( void );
+
+	virtual void FireGameEvent( IGameEvent *gameEvent );
+
+	void StartSpin( void );
+	void Spin( inputdata_t& inputdata );
+	void ClearAllEffects( inputdata_t& inputdata );
+	bool IsDoneBoardcastingEffectSound() const;
+
+	float GetDuration() const { return m_flDuration; }
+
+	void DBG_ApplyEffectByName( const char* pszEffectName );
+
+private:
+
+	enum eEffectProperty
+	{
+		PROPERTY_WHAMMY,
+		PROPERTY_DOES_NOT_REAPPLY_ON_SPAWN
+	};
+
+	struct EffectData_t
+	{
+		CUtlVector<CTFPlayer*> m_vecPlayers;
+		CWheelOfDoom* m_pWheel;
+	};
+
+	struct EffectManager;
+	class WOD_BaseEffect
+	{
+	public:
+		WOD_BaseEffect();
+
+		virtual void InitEffect( float flDefaultDuration );
+
+		virtual void ActivateEffect( EffectData_t& data )	{};
+		virtual void UpdateEffect( EffectData_t& data  )	{};
+		virtual void DeactivateEffect( EffectData_t& data )	{};
+
+		void SetListFlags( int iFlags );
+		int GetListFlags() const			{ return m_iListFlags; }
+		const char* GetName()				{ return m_pszName; }
+		eWheelOfDoomEffectSkinIndex GetSkinIndex() const { return m_nSkin; }
+	protected:
+
+		eWheelOfDoomEffectSkinIndex m_nSkin;
+		float m_flExpireTime;
+		const char* m_pszEffectAnnouncementSound;
+		int m_iListFlags;
+
+		const char* m_pszName;
+
+		friend struct EffectManager;
+	};
+
+	struct EffectManager
+	{
+	public:
+		explicit EffectManager( CWheelOfDoom* pWheel ) { m_pWheel = pWheel; }
+		~EffectManager();
+
+		int AddEffect( WOD_BaseEffect* pEffect, float flDefaultDuration );
+		void ApplyAllEffectsToPlayer( CTFPlayer* pPlayer );
+		void ClearEffects();
+		bool UpdateAndClearExpiredEffects();
+		void Precache();
+
+	private:
+		CWheelOfDoom*	m_pWheel;
+		CUtlVector<WOD_BaseEffect*> m_vecActiveEffects;
+	} m_EffectManager;
+
+	WOD_BaseEffect* GetRandomEffectWithFlags();
+	void EndSpin();
+
+	CUtlVector<WOD_BaseEffect*> m_vecEffects;
+
+	void RegisterEffect( WOD_BaseEffect* pEffect, int nFlags = 0 );
+
+	static void ApplyAttributeToAllPlayers( const char* pszAttribName, float flValue );
+	static void ApplyAttributeToPlayer( CTFPlayer* pPlayer, const char* pszAttribName, float flValue );
+	static void RemoveAttributeFromAllPlayers( const char* pszAttribName );
+	static void RemoveAttributeFromPlayer( CTFPlayer* pPlayer, const char* pszAttribName );
+
+	static void SpeakMagicConceptToAllPlayers( const char* pszEffect );
+
+
+	WOD_BaseEffect* m_pChosenEffect;
+	CWheelOfDoomSpiral* m_pSpiral;
+
+	static const char* GetScreenModelName();
+	float CalcNextTickTime() const;
+	float CalcSpinCompletion() const;
+	void SetSkin( int nSkin );
+	void SetScale( float flScale );
+	void PlaySound( const char* pszSound );
+
+	bool	m_bAnnounced;
+	bool	m_bHasSpiral;
+	float	m_flDuration;
+	float	m_flNextTickTime;
+	float	m_flNextAnnounceTime;
+	float	m_flStopSpinTime;
+	CUtlVector<CWheelOfDoom*> m_vecOtherWODs;
+
+	float	m_flFinishBroadcastingEffectTime;
+
+	COutputEvent m_EffectApplied;
+	COutputEvent m_EffectExpired;
+
+
+	//-----------------------------------------------------------------------
+	// Effects
+
+	class WOD_CritsEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_CritsEffect()
+		{
+			m_pszName = "Crits";
+			m_nSkin = EFFECT_CRITS;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelCrits";
+		};
+
+		void ActivateEffect( EffectData_t& data );
+	};
+
+	class WOD_UberEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_UberEffect()
+		{
+			m_pszName = "Uber";
+			m_nSkin = EFFECT_UBER;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelUber";
+		};
+
+		void InitEffect( float flDefaultExpireTime );
+		void ActivateEffect( EffectData_t& data );
+	};
+
+	class WOD_SuperSpeedEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_SuperSpeedEffect()
+		{
+			m_pszName = "Super Speed";
+			m_nSkin = EFFECT_SPEED;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelSuperSpeed";
+		};
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_SuperJumpEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_SuperJumpEffect()
+		{
+			m_pszName = "Super Jump";
+			m_nSkin = EFFECT_JUMP_HEIGHT;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelSuperJump";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_BigHeadEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_BigHeadEffect()
+		{
+			m_pszName = "Big Head";
+			m_nSkin = EFFECT_BIG_HEAD;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelBigHead";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_SmallHeadEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_SmallHeadEffect()
+		{
+			m_pszName = "Small Head";
+			m_nSkin = EFFECT_SMALL_HEAD;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelShrunkHead";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_LowGravityEffect : public WOD_BaseEffect
+	{
+	public:
+		WOD_LowGravityEffect()
+		{
+			m_pszName = "Low Gravity";
+			m_nSkin = EFFECT_LOW_GRAVITY;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelGravity";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_Pee : public WOD_BaseEffect
+	{
+	public:
+		WOD_Pee()
+		{
+			m_pszName = "Pee";
+			m_nSkin = EFFECT_WHAMMY;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelJarate";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void UpdateEffect( EffectData_t& data );
+
+	private:
+
+		float m_flNextPeeTime;
+		CUtlVector<CBaseEntity*> m_vecClouds;
+	};
+
+	class WOD_Burn : public WOD_BaseEffect
+	{
+	public:
+		WOD_Burn()
+		{
+			m_pszName = "Burn";
+			m_nSkin = EFFECT_WHAMMY;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelBurn";
+		}
+
+		void InitEffect( float flDefaultDuration );
+		void ActivateEffect( EffectData_t& data );
+	};
+
+	class WOD_Ghosts : public WOD_BaseEffect
+	{
+	public:
+		WOD_Ghosts()
+		{
+			m_pszName = "Ghosts";
+			m_nSkin = EFFECT_WHAMMY;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelGhosts";
+		}
+
+		void ActivateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+	};
+
+	class WOD_Dance : public WOD_BaseEffect
+	{
+	public:
+		WOD_Dance()
+		{
+			m_pszName = "Dance";
+			m_nSkin = EFFECT_DANCE;
+			m_pszEffectAnnouncementSound = "Halloween.MerasmusWheelDance";
+			m_flNextDanceTime = 0.f;
+		}
+
+		void InitEffect( float flExpireTime );
+
+		void UpdateEffect( EffectData_t& data );
+		void DeactivateEffect( EffectData_t& data );
+
+	private:
+		int GetNumOFTeamDancing( int nTeam ) const;
+		void SlamPosAndAngles( CTFPlayer* pPlayer, const Vector& vPos, const QAngle& vAng );
+		
+		struct Dancer_t
+		{
+			Vector m_vecPos;
+			QAngle m_vecAngles;
+			CHandle<CTFPlayer> m_hPlayer;
+		};
+
+		CUtlVector<Dancer_t*> m_vecDancers;
+
+		struct MerasmusCreateInfo_t
+		{
+			MerasmusCreateInfo_t( const Vector& p, const QAngle& ang ) : m_vecPos( p ), m_vecAngles( ang ) {}
+			Vector m_vecPos;
+			QAngle m_vecAngles;
+		};
+		CUtlVector< MerasmusCreateInfo_t > m_vecMerasmusDancerCreateInfos;
+		int m_iCurrentMerasmusCreateInfo;
+		CHandle<CMerasmusDancer> m_hMerasmusDancer;
+
+		float m_flNextDanceTime;
+	};
+};
+
+#endif // GAME_DLL
+
+#endif // TF_WHEEL_OF_DOOM_H