upload "kind" alien swarm

game/server/ai_addon.h

@@ -0,0 +1,344 @@
+// ai_addon.h
+
+#ifndef AI_ADDON_H
+#define AI_ADDON_H
+#ifdef _WIN32
+#pragma once
+#endif
+
+#include "tier1/utlvector.h"
+
+#include "baseanimating.h"
+#include "ai_agent.h"
+#include "ai_behavior.h"
+#include "player_pickup.h"
+#include "IEffects.h"
+
+class CAI_BaseNPC;
+
+#define INVALID_ADDON_ATTACHMENT_ID	-1
+
+int GetIDForAttachmentName( char *pszAttachmentName );
+int GetNumAddOnAttachmentPoints();
+bool IsAddOnAttachmentAvailable( CAI_BaseNPC *pHost, char *pszAttachmentName );
+int CountAddOns( CAI_BaseNPC *pHost );
+
+//=========================================================
+//=========================================================
+class CAI_AddOn : public CBaseAnimating, public CAI_Agent, public CDefaultPlayerPickupVPhysics
+{
+public:
+	DECLARE_CLASS( CAI_AddOn, CBaseAnimating );
+
+	//---------------------------------
+	// CTor/DTor
+	//---------------------------------
+	CAI_AddOn() { m_iAttachmentID = INVALID_ADDON_ATTACHMENT_ID; }
+
+	//---------------------------------
+	// Precache/Spawn/etc
+	//---------------------------------
+	virtual void Precache();
+	virtual void Spawn();
+	virtual void UpdateOnRemove();
+
+	const char *GetDebugName()	{ return CBaseEntity::GetDebugName(); }
+	int entindex()				{ return CBaseEntity::entindex(); }
+
+	virtual int			DrawDebugTextOverlays(void);
+
+	//---------------------------------
+	//---------------------------------
+	bool SUB_AllowedToFade( void )
+	{
+		return true;
+	}
+
+	int Save( ISave &save )
+	{
+		int result = CBaseAnimating::Save( save );
+		if ( result )
+		{
+			result = CAI_Agent::Save( save );
+		}
+		return result;
+	}
+
+	//-------------------------------------
+
+	int Restore( IRestore &restore )
+	{
+		int result = CBaseAnimating::Restore( restore );
+		if ( result )
+		{
+			result = CAI_Agent::Restore( restore );
+		}
+		return result;
+	}
+
+	//---------------------------------
+	// Agent stuff
+	//---------------------------------
+	virtual void		GatherConditions();
+	virtual int			SelectSchedule();
+	virtual void		StartTask( const Task_t *pTask );
+	virtual void		RunTask( const Task_t *pTask );
+
+	//---------------------------------
+	// Install/Remove AddOns
+	//---------------------------------
+	virtual bool Install( CAI_BaseNPC *pHost, bool bRemoveOnFail = true );
+	void SetPhysReplacement( CBaseEntity *pEntity );
+	bool Attach( CAI_BaseNPC *pHost );
+	void Dettach( void );
+	virtual void Remove() { Unbind(); Dettach(); }
+	virtual void Bind() {}
+	virtual void Unbind() {}
+
+	//---------------------------------
+	// Hosts/Outer accessors
+	//---------------------------------
+	CAI_BaseNPC *GetNPCHost();
+	CBaseEntity *GetHostEnemy();
+
+	//---------------------------------
+	// Thinking
+	//---------------------------------
+	void DispatchAddOnThink();
+
+	virtual float GetThinkInterval() { return 0.1f; }
+
+	//---------------------------------
+	// Appearance & Position
+	//---------------------------------
+	virtual void PickAttachment( CAI_BaseNPC *pHost, char *pchAttachment ) = 0;
+	virtual char *GetAddOnModelName() = 0;
+	virtual Vector GetAttachOffset( QAngle &attachmentAngles )			{ return vec3_origin; }
+	virtual QAngle GetAttachOrientation( QAngle &attachmentAngles )		{ return attachmentAngles; }
+	virtual QAngle GetLocalOrientation( void );
+	int GetAttachmentID() { return m_iAttachmentID; }
+	virtual void EjectFromHost();
+
+	//---------------------------------
+	// Entity I/O
+	//---------------------------------
+	void InputInstall( inputdata_t &data );
+	void InputRemove( inputdata_t &data );
+
+	//---------------------------------
+	// Schedule/Task/Conditions
+	//---------------------------------
+	enum
+	{
+		TASK_ADDON_WAIT = NEXT_TASK,
+		TASK_ADDON_WAIT_RANDOM,
+		NEXT_TASK,
+	};
+
+	enum
+	{
+		COND_ADDON_LOST_HOST = NEXT_CONDITION,
+		NEXT_CONDITION,
+	};
+
+	enum
+	{
+		SCHED_ADDON_NO_OWNER = NEXT_SCHEDULE,
+		NEXT_SCHEDULE,
+	};
+
+	DECLARE_DATADESC();
+
+protected:
+	CHandle<CAI_BaseNPC>	m_hNPCHost;
+	CHandle<CBaseEntity>	m_hPhysReplacement;
+	int						m_iPhysReplacementSolidFlags;
+	int						m_iPhysReplacementMoveType;
+	QAngle					m_angPhysReplacementLocalOrientation;
+	Vector					m_vecPhysReplacementDetatchForce;
+
+	bool	m_bWasAttached;
+
+private:
+	//---------------------------------
+	// Fields used by AI
+	//---------------------------------
+	float	m_flWaitFinished;	// Same as for AI_BaseNPC
+	int		m_iAttachmentID;	// Which attachment point am I connected to on my host's model?
+	float 	m_flNextAttachTime;
+
+
+public:
+	//---------------------------------
+	// Hammer keyfields
+	//---------------------------------
+
+	DEFINE_AGENT();
+};
+
+//=========================================================
+//=========================================================
+
+class CAI_AddOnBehaviorBase : public CAI_SimpleBehavior
+{
+public:
+	virtual bool ShouldNPCSave() 
+	{ 
+		return false; 
+	}
+
+	virtual CAI_AddOn **GetAddOnsBase() { return NULL; }
+	const CAI_AddOn **GetAddOnsBase() const { return (const CAI_AddOn **)const_cast<CAI_AddOnBehaviorBase *>(this)->GetAddOnsBase(); }
+	virtual int NumAddOns() const { return 0; }
+};
+
+
+template <class ADDON>
+class CAI_AddOnBehavior : public CAI_AddOnBehaviorBase
+{
+public:
+	virtual int BindAddOn( ADDON *pAddOn )	
+	{ 
+		Assert( m_AddOns.Find( pAddOn ) == m_AddOns.InvalidIndex() );
+		Assert( static_cast<CAI_AddOn *>(pAddOn) ); // Satic cast is used to trap improper inheritance
+		m_AddOns.AddToTail( pAddOn );
+		return m_AddOns.Count(); 
+	}
+
+	virtual int UnbindAddOn( ADDON *pAddOn ) 
+	{ 
+		Assert( m_AddOns.Find( pAddOn ) != m_AddOns.InvalidIndex() );
+		m_AddOns.FindAndRemove( pAddOn );
+		return m_AddOns.Count(); 
+	}
+
+	virtual CAI_AddOn **GetAddOnsBase() 
+	{ 
+		return (CAI_AddOn **)m_AddOns.Base(); 
+	}
+
+	virtual ADDON **GetAddOns() 
+	{ 
+		return m_AddOns.Base(); 
+	}
+
+	virtual int NumAddOns() const
+	{ 
+		return m_AddOns.Count(); 
+	}
+
+protected:
+	CUtlVector<ADDON *> m_AddOns;
+};
+
+
+//=========================================================
+//=========================================================
+template <class ADDON, class BEHAVIOR>
+class CAI_AddOnBehaviorConnector : public ADDON
+{
+	DECLARE_CLASS( CAI_AddOnBehaviorConnector, ADDON );
+
+	virtual int ObjectCaps( void )
+	{
+		return BaseClass::ObjectCaps() | FCAP_IMPULSE_USE;
+	}
+
+	virtual void Use( CBaseEntity *pActivator, CBaseEntity *pCaller, USE_TYPE useType, float value )
+	{
+		CBasePlayer *pPlayer = ToBasePlayer( pActivator );
+
+		if ( pPlayer == NULL )
+			return;
+
+		if ( this->m_bWasAttached )
+		{
+			this->Remove();
+
+			g_pEffects->Sparks( this->GetAbsOrigin(), 2, 2 );
+		}
+
+		pPlayer->PickupObject( this );
+	}
+
+	virtual void Bind()
+	{
+		if ( this->m_hNPCHost )
+		{
+			BEHAVIOR *pBehavior;
+			if ( !this->m_hNPCHost->GetBehavior( &pBehavior ) )
+			{
+				pBehavior =  new BEHAVIOR;
+				this->m_hNPCHost->AddBehavior( pBehavior );
+			}
+			pBehavior->BindAddOn( this );
+		}
+	}
+
+	virtual void Unbind()
+	{
+		if ( this->m_hNPCHost )
+		{
+			BEHAVIOR *pBehavior;
+			if ( this->m_hNPCHost->GetBehavior( &pBehavior ) )
+			{
+				if ( pBehavior->UnbindAddOn( this ) == 0 )
+				{
+					this->m_hNPCHost->RemoveAndDestroyBehavior( pBehavior );
+				}
+			}
+		}
+	}
+
+	//-------------------------------------
+
+	int Save( ISave &save )
+	{
+		int result = BaseClass::Save( save );
+		if ( result )
+		{
+			bool bSaved = false;
+			BEHAVIOR *pBehavior;
+			if ( this->m_hNPCHost && this->m_hNPCHost->GetBehavior( &pBehavior ) )
+			{
+				bSaved = true;
+				CAI_BehaviorBase::SaveBehaviors( save, pBehavior, (CAI_BehaviorBase **)&pBehavior, 1, false );
+			}
+
+			if ( !bSaved )
+			{
+				CAI_BehaviorBase::SaveBehaviors( save, NULL, NULL, 0 );
+			}
+		}
+
+		return result;
+	}
+
+	//-------------------------------------
+
+	int Restore( IRestore &restore )
+	{
+		int result = BaseClass::Restore( restore );
+		if ( result )
+		{
+			Bind();
+			BEHAVIOR *pBehavior;
+			if ( this->m_hNPCHost && this->m_hNPCHost->GetBehavior( &pBehavior ) )
+			{
+				CAI_BehaviorBase::RestoreBehaviors( restore, (CAI_BehaviorBase **)&pBehavior, 1, false );
+			}
+			else
+			{
+				CAI_BehaviorBase::RestoreBehaviors( restore, NULL, 0,false );
+			}
+		}
+		return result;
+	}
+
+};
+
+#define LINK_ENTITY_TO_ADDON_AND_BEHAVIOR( mapClassName, DLLClassName, BehaviorName ) \
+	typedef CAI_AddOnBehaviorConnector<DLLClassName, BehaviorName> DLLClassName##_##BehaviorName##_Binder; \
+	LINK_ENTITY_TO_CLASS( mapClassName, DLLClassName##_##BehaviorName##_Binder )
+
+#endif // AI_ADDON_H