RefCounted.h

Go to the documentation of this file.

#ifndef MiscLib__REFCOUNTED_HEADER__
#define MiscLib__REFCOUNTED_HEADER__
#ifdef DOPARALLEL
#include <omp.h>
#endif
 
namespace MiscLib
{
    template <class T>
    class RefCounted : public T
    {
    public:
        RefCounted() : m_refCount(1)
        {
        }
 
        RefCounted(const RefCounted<T>& r) : T(r), m_refCount(1)
        {
            // do not copy the ref count!
        }
 
        unsigned int
        AddRef() const
        {
            //#pragma omp atomic
            ++m_refCount;
            return m_refCount;
        }
 
        unsigned int
        Release() const
        {
            if (m_refCount == 1)
            {
                //#pragma omp critical
                {
                    if (m_refCount)
                    {
                        m_refCount = 0;
                        delete this;
                    }
                }
                return 0;
            }
            //#pragma omp atomic
            --m_refCount;
            return m_refCount;
        }
 
        RefCounted&
        operator=(const RefCounted& r)
        {
            *((T*)this) = r;
            return *this; // do not copy the ref count!
        }
 
    protected:
        virtual ~RefCounted()
        {
        }
 
    private:
        mutable unsigned int m_refCount;
    };
}; // namespace MiscLib
 
#endif