1 files changed, 314 insertions, 0 deletions
diff --git a/game/server/colorcorrection.cpp b/game/server/colorcorrection.cpp
new file mode 100644
index 0000000..16d0122
--- /dev/null
+++ b/game/server/colorcorrection.cpp
@@ -0,0 +1,314 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: Color correction entity.
+//
+// $NoKeywords: $
+//===========================================================================//
+
+#include <string.h>
+
+#include "cbase.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+#define COLOR_CORRECTION_ENT_THINK_RATE TICK_INTERVAL
+
+static const char *s_pFadeInContextThink = "ColorCorrectionFadeInThink";
+static const char *s_pFadeOutContextThink = "ColorCorrectionFadeOutThink";
+ 
+//------------------------------------------------------------------------------
+// FIXME: This really should inherit from something	more lightweight
+//------------------------------------------------------------------------------
+
+
+//------------------------------------------------------------------------------
+// Purpose : Shadow control entity
+//------------------------------------------------------------------------------
+class CColorCorrection : public CBaseEntity
+{
+	DECLARE_CLASS( CColorCorrection, CBaseEntity );
+public:
+	DECLARE_SERVERCLASS();
+	DECLARE_DATADESC();
+
+	CColorCorrection();
+
+	void Spawn( void );
+	int  UpdateTransmitState();
+	void Activate( void );
+
+	virtual int	ObjectCaps( void ) { return BaseClass::ObjectCaps() & ~FCAP_ACROSS_TRANSITION; }
+
+	// Inputs
+	void	InputEnable( inputdata_t &inputdata );
+	void	InputDisable( inputdata_t &inputdata );
+	void	InputSetFadeInDuration ( inputdata_t &inputdata );
+	void	InputSetFadeOutDuration ( inputdata_t &inputdata );
+
+private:
+	void	FadeIn ( void );
+	void	FadeOut ( void );
+
+	void FadeInThink( void );	// Fades lookup weight from Cur->MaxWeight 
+	void FadeOutThink( void );	// Fades lookup weight from CurWeight->0.0
+
+	
+	
+	float	m_flFadeInDuration;		// Duration for a full 0->MaxWeight transition
+	float	m_flFadeOutDuration;	// Duration for a full Max->0 transition
+	float	m_flStartFadeInWeight;
+	float	m_flStartFadeOutWeight;
+	float	m_flTimeStartFadeIn;
+	float	m_flTimeStartFadeOut;
+	
+	float	m_flMaxWeight;
+
+	bool	m_bStartDisabled;
+	CNetworkVar( bool, m_bEnabled );
+
+	CNetworkVar( float, m_MinFalloff );
+	CNetworkVar( float, m_MaxFalloff );
+	CNetworkVar( float, m_flCurWeight );
+	CNetworkString( m_netlookupFilename, MAX_PATH );
+
+	string_t	m_lookupFilename;
+};
+
+LINK_ENTITY_TO_CLASS(color_correction, CColorCorrection);
+
+BEGIN_DATADESC( CColorCorrection )
+
+	DEFINE_THINKFUNC( FadeInThink ),
+	DEFINE_THINKFUNC( FadeOutThink ),
+
+	DEFINE_FIELD( m_flCurWeight,	      FIELD_FLOAT ),
+	DEFINE_FIELD( m_flTimeStartFadeIn,	  FIELD_FLOAT ),
+	DEFINE_FIELD( m_flTimeStartFadeOut,	  FIELD_FLOAT ),
+	DEFINE_FIELD( m_flStartFadeInWeight,  FIELD_FLOAT ),
+	DEFINE_FIELD( m_flStartFadeOutWeight, FIELD_FLOAT ),
+
+	DEFINE_KEYFIELD( m_MinFalloff,		  FIELD_FLOAT,   "minfalloff" ),
+	DEFINE_KEYFIELD( m_MaxFalloff,		  FIELD_FLOAT,   "maxfalloff" ),
+	DEFINE_KEYFIELD( m_flMaxWeight,		  FIELD_FLOAT,	 "maxweight" ),
+	DEFINE_KEYFIELD( m_flFadeInDuration,  FIELD_FLOAT,	 "fadeInDuration" ),
+	DEFINE_KEYFIELD( m_flFadeOutDuration,  FIELD_FLOAT,	 "fadeOutDuration" ),
+	DEFINE_KEYFIELD( m_lookupFilename,	  FIELD_STRING,  "filename" ),
+
+	DEFINE_KEYFIELD( m_bEnabled,		  FIELD_BOOLEAN, "enabled" ),
+	DEFINE_KEYFIELD( m_bStartDisabled,    FIELD_BOOLEAN, "StartDisabled" ),
+//	DEFINE_ARRAY( m_netlookupFilename, FIELD_CHARACTER, MAX_PATH ), 
+
+	DEFINE_INPUTFUNC( FIELD_VOID, "Enable", InputEnable ),
+	DEFINE_INPUTFUNC( FIELD_VOID, "Disable", InputDisable ),
+	DEFINE_INPUTFUNC( FIELD_FLOAT, "SetFadeInDuration", InputSetFadeInDuration ),
+	DEFINE_INPUTFUNC( FIELD_FLOAT, "SetFadeOutDuration", InputSetFadeOutDuration ),
+
+END_DATADESC()
+
+extern void SendProxy_Origin( const SendProp *pProp, const void *pStruct, const void *pData, DVariant *pOut, int iElement, int objectID );
+IMPLEMENT_SERVERCLASS_ST_NOBASE(CColorCorrection, DT_ColorCorrection)
+	SendPropVector( SENDINFO(m_vecOrigin), -1,  SPROP_NOSCALE, 0.0f, HIGH_DEFAULT, SendProxy_Origin ),
+	SendPropFloat(  SENDINFO(m_MinFalloff) ),
+	SendPropFloat(  SENDINFO(m_MaxFalloff) ),
+	SendPropFloat(  SENDINFO(m_flCurWeight) ),
+	SendPropString( SENDINFO(m_netlookupFilename) ),
+	SendPropBool( SENDINFO(m_bEnabled) ),
+END_SEND_TABLE()
+
+
+CColorCorrection::CColorCorrection() : BaseClass()
+{
+	m_bEnabled = true;
+	m_MinFalloff = 0.0f;
+	m_MaxFalloff = 1000.0f;
+	m_flMaxWeight = 1.0f;
+	m_flCurWeight.Set( 0.0f );
+	m_flFadeInDuration = 0.0f;
+	m_flFadeOutDuration = 0.0f;
+	m_flStartFadeInWeight = 0.0f;
+	m_flStartFadeOutWeight = 0.0f;
+	m_flTimeStartFadeIn = 0.0f;
+	m_flTimeStartFadeOut = 0.0f;
+	m_netlookupFilename.GetForModify()[0] = 0;
+	m_lookupFilename = NULL_STRING;
+}
+
+
+//------------------------------------------------------------------------------
+// Purpose : Send even though we don't have a model
+//------------------------------------------------------------------------------
+int CColorCorrection::UpdateTransmitState()
+{
+	// ALWAYS transmit to all clients.
+	return SetTransmitState( FL_EDICT_ALWAYS );
+}
+
+//------------------------------------------------------------------------------
+// Purpose :
+//------------------------------------------------------------------------------
+void CColorCorrection::Spawn( void )
+{
+	AddEFlags( EFL_FORCE_CHECK_TRANSMIT | EFL_DIRTY_ABSTRANSFORM );
+	Precache();
+	SetSolid( SOLID_NONE );
+
+	// To fade in/out the weight.
+	SetContextThink( &CColorCorrection::FadeInThink, TICK_NEVER_THINK, s_pFadeInContextThink );
+	SetContextThink( &CColorCorrection::FadeOutThink, TICK_NEVER_THINK, s_pFadeOutContextThink );
+	
+	if( m_bStartDisabled )
+	{
+		m_bEnabled = false;
+		m_flCurWeight.Set ( 0.0f );
+	}
+	else
+	{
+		m_bEnabled = true;
+		m_flCurWeight.Set ( 1.0f );
+	}
+
+	BaseClass::Spawn();
+}
+
+void CColorCorrection::Activate( void )
+{
+	BaseClass::Activate();
+
+	Q_strncpy( m_netlookupFilename.GetForModify(), STRING( m_lookupFilename ), MAX_PATH );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Sets up internal vars needed for fade in lerping
+//-----------------------------------------------------------------------------
+void CColorCorrection::FadeIn ( void )
+{
+	m_bEnabled = true;
+	m_flTimeStartFadeIn = gpGlobals->curtime;
+	m_flStartFadeInWeight = m_flCurWeight;
+	SetNextThink ( gpGlobals->curtime + COLOR_CORRECTION_ENT_THINK_RATE, s_pFadeInContextThink );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Sets up internal vars needed for fade out lerping
+//-----------------------------------------------------------------------------
+void CColorCorrection::FadeOut ( void )
+{
+	m_bEnabled = false;
+	m_flTimeStartFadeOut = gpGlobals->curtime;
+	m_flStartFadeOutWeight = m_flCurWeight;
+	SetNextThink ( gpGlobals->curtime + COLOR_CORRECTION_ENT_THINK_RATE, s_pFadeOutContextThink );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Fades lookup weight from CurWeight->MaxWeight
+//-----------------------------------------------------------------------------
+void CColorCorrection::FadeInThink( void )
+{
+	// Check for conditions where we shouldnt fade in
+	if (		m_flFadeInDuration <= 0 ||  // not set to fade in
+		 m_flCurWeight >= m_flMaxWeight ||  // already past max weight
+							!m_bEnabled ||  // fade in/out mutex
+				  m_flMaxWeight == 0.0f ||  // min==max
+  m_flStartFadeInWeight >= m_flMaxWeight )  // already at max weight
+	{
+		SetNextThink ( TICK_NEVER_THINK, s_pFadeInContextThink );
+		return;
+	}
+	
+	// If we started fading in without fully fading out, use a truncated duration
+    float flTimeToFade = m_flFadeInDuration;
+	if ( m_flStartFadeInWeight > 0.0f )
+	{	
+		float flWeightRatio		= m_flStartFadeInWeight / m_flMaxWeight;
+		flWeightRatio = clamp ( flWeightRatio, 0.0f, 0.99f );
+		flTimeToFade			= m_flFadeInDuration * (1.0 - flWeightRatio);
+	}	
+	
+	Assert ( flTimeToFade > 0.0f );
+	float flFadeRatio = (gpGlobals->curtime - m_flTimeStartFadeIn) / flTimeToFade;
+	flFadeRatio = clamp ( flFadeRatio, 0.0f, 1.0f );
+	m_flStartFadeInWeight = clamp ( m_flStartFadeInWeight, 0.0f, 1.0f );
+
+	m_flCurWeight = Lerp( flFadeRatio, m_flStartFadeInWeight, m_flMaxWeight );
+
+	SetNextThink( gpGlobals->curtime + COLOR_CORRECTION_ENT_THINK_RATE, s_pFadeInContextThink );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Fades lookup weight from CurWeight->0.0 
+//-----------------------------------------------------------------------------
+void CColorCorrection::FadeOutThink( void )
+{
+	// Check for conditions where we shouldn't fade out
+	if ( m_flFadeOutDuration <= 0 || // not set to fade out
+			m_flCurWeight <= 0.0f || // already faded out
+					   m_bEnabled || // fade in/out mutex
+		   m_flMaxWeight == 0.0f  || // min==max
+	 m_flStartFadeOutWeight <= 0.0f )// already at min weight
+	{
+		SetNextThink ( TICK_NEVER_THINK, s_pFadeOutContextThink );
+		return;
+	}
+
+	// If we started fading out without fully fading in, use a truncated duration
+    float flTimeToFade = m_flFadeOutDuration;
+	if ( m_flStartFadeOutWeight < m_flMaxWeight )
+	{	
+		float flWeightRatio		= m_flStartFadeOutWeight / m_flMaxWeight;
+		flWeightRatio = clamp ( flWeightRatio, 0.01f, 1.0f );
+		flTimeToFade			= m_flFadeOutDuration * flWeightRatio;
+	}	
+	
+	Assert ( flTimeToFade > 0.0f );
+	float flFadeRatio = (gpGlobals->curtime - m_flTimeStartFadeOut) / flTimeToFade;
+	flFadeRatio = clamp ( flFadeRatio, 0.0f, 1.0f );
+	m_flStartFadeOutWeight = clamp ( m_flStartFadeOutWeight, 0.0f, 1.0f );
+
+	m_flCurWeight = Lerp( 1.0f - flFadeRatio, 0.0f, m_flStartFadeOutWeight );
+
+	SetNextThink( gpGlobals->curtime + COLOR_CORRECTION_ENT_THINK_RATE, s_pFadeOutContextThink );
+}
+
+//------------------------------------------------------------------------------
+// Purpose : Input handlers
+//------------------------------------------------------------------------------
+void CColorCorrection::InputEnable( inputdata_t &inputdata )
+{
+	m_bEnabled = true;
+
+	if ( m_flFadeInDuration > 0.0f )
+	{
+		FadeIn();
+	}
+	else
+	{
+		m_flCurWeight = m_flMaxWeight;
+	}
+	
+}
+
+void CColorCorrection::InputDisable( inputdata_t &inputdata )
+{
+	m_bEnabled = false;
+
+	if ( m_flFadeOutDuration > 0.0f )
+	{
+		FadeOut();
+	}
+	else
+	{
+		m_flCurWeight = 0.0f;
+	}
+	
+}
+
+void CColorCorrection::InputSetFadeInDuration( inputdata_t& inputdata )
+{
+	m_flFadeInDuration = inputdata.value.Float();
+}
+
+void CColorCorrection::InputSetFadeOutDuration( inputdata_t& inputdata )
+{
+	m_flFadeOutDuration = inputdata.value.Float();
+}