345 lines
8.4 KiB
C
345 lines
8.4 KiB
C
|
// 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
|