tf2/tf2_src/game/server/damagemodifier.h

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
// $NoKeywords: $
//=============================================================================//

#ifndef DAMAGEMODIFIER_H
#define DAMAGEMODIFIER_H
#ifdef _WIN32
#pragma once
#endif


class CBaseEntity;


#include "ehandle.h"

//-----------------------------------------------------------------------------
// Purpose: Class handling generic damage modification to & from a player
//-----------------------------------------------------------------------------
class CDamageModifier
{
public:
				CDamageModifier();

	void		AddModifierToEntity( CBaseEntity *pChar );
	void		RemoveModifier();

	void		SetModifier( float flDamageScale );
	float		GetModifier() const;

	void		SetDoneToMe( bool bDoneToMe );
	bool		IsDamageDoneToMe() const;

	CBaseEntity	*GetCharacter() const;

private:
	float							m_flModifier;
	CHandle<CBaseEntity>			m_hEnt;
	bool							m_bDoneToMe;	// True = modifies damage done to the entity, false = damage done by the entity
};


#endif // DAMAGEMODIFIER_H