IReferenceCounted.h

// Copyright (C) 2002-2012 Nikolaus Gebhardt
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in irrlicht.h
 
#ifndef __I_IREFERENCE_COUNTED_H_INCLUDED__
#define __I_IREFERENCE_COUNTED_H_INCLUDED__
 
#include "irrTypes.h"
 
#ifdef _IRR_COMPILE_WITH_LEAK_HUNTER_
    #include "leakHunter.h"
#endif
 
namespace irr
{


    class IReferenceCounted
    {
    public:

        IReferenceCounted()
            : DebugName(0), ReferenceCounter(1)
        {
#ifdef _IRR_COMPILE_WITH_LEAK_HUNTER_
            LeakHunter::addObject(this);
#endif
        }

        virtual ~IReferenceCounted()
        {
            #ifdef _IRR_COMPILE_WITH_LEAK_HUNTER_
                LeakHunter::removeObject(this);
            #endif
        }


        void grab() const { ++ReferenceCounter; }


        bool drop() const
        {
            // someone is doing bad reference counting.
            _IRR_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 irr
 
#endif