IReferenceCounted.hpp

// Copyright (C) 2002-2012 Nikolaus Gebhardt
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in nirtcpp/nirtcpp.hpp
 
#ifndef NIRT_I_IREFERENCE_COUNTED_HPP_INCLUDED
#define NIRT_I_IREFERENCE_COUNTED_HPP_INCLUDED
 
#include <nirtcpp/core/engine/nirt_types.hpp>
 
#ifdef _NIRT_COMPILE_WITH_LEAK_HUNTER_
    #include <nirtcpp/core/engine/leakHunter.hpp>
#endif
 
namespace nirt
{
 
 
    class IReferenceCounted
    {
    public:
 
        IReferenceCounted()
            : DebugName(0), ReferenceCounter(1)
        {
#ifdef _NIRT_COMPILE_WITH_LEAK_HUNTER_
            LeakHunter::addObject(this);
#endif
        }
 
        virtual ~IReferenceCounted()
        {
            #ifdef _NIRT_COMPILE_WITH_LEAK_HUNTER_
                LeakHunter::removeObject(this);
            #endif
        }
 
 
        void grab() const { ++ReferenceCounter; }
 
 
        bool drop() const
        {
            // someone is doing bad reference counting.
            NIRT_DEBUG_BREAK_IF(ReferenceCounter <= 0)
 
            --ReferenceCounter;
            if (!ReferenceCounter)
            {
                delete this;
                return true;
            }
 
            return false;
        }
 
 
        s32 getReferenceCount() const
        {
            return ReferenceCounter;
        }
 
 
        const c8* getDebugName() const
        {
            return DebugName;
        }
 
    protected:
 
 
        void setDebugName(const c8* newName)
        {
            DebugName = newName;
        }
 
    private:
 
        const c8* DebugName;
 
        mutable s32 ReferenceCounter;
    };
 
} // end namespace nirt
 
#endif