//========= 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