ibrcommon/ibrcommon/refcnt_ptr.h

Go to the documentation of this file.

#ifndef IBRCOMMON_refcnt_ptr_h
#define IBRCOMMON_refcnt_ptr_h 1

#include "ibrcommon/thread/Mutex.h"
#include "ibrcommon/thread/MutexLock.h"

template <class T> class refcnt_ptr
{
        protected:
                // a helper class that holds the pointer to the managed object
                // and its reference count
                class Holder
                {
                public:
                        Holder( T* ptr) : ptr_(ptr), count_(1) {};
                        virtual ~Holder() { delete ptr_;};

                        T* ptr_;
                        unsigned count_;
                        ibrcommon::Mutex lock;
                };

                Holder* h_;

        private:
                void down()
                {
                        try {
                                ibrcommon::MutexLock l(h_->lock);
                                if (--h_->count_ == 0) throw true;
                        } catch (const bool&) {
                                delete h_;
                        }
                }

        public:
                // ctor of refcnt_ptr (p must not be NULL)
                explicit refcnt_ptr(T* p) : h_(new Holder(p))
                {
                }

                // dtor of refcnt_ptr
                ~refcnt_ptr()
                {
                        down();
                }

                // copy and assignment of refcnt_ptr
                refcnt_ptr (const refcnt_ptr<T>& right) : h_(right.h_)
                {
                        ibrcommon::MutexLock l(h_->lock);
                        ++h_->count_;
                }

                refcnt_ptr<T>& operator= (const refcnt_ptr<T>& right)
                {
                        ibrcommon::MutexLock l(right.h_->lock);
                        ++right.h_->count_;

                        down();

                        h_ = right.h_;

                        return *this;
                }

                refcnt_ptr<T>& operator= (refcnt_ptr<T>& right)
                {
                        ibrcommon::MutexLock l(right.h_->lock);
                        ++right.h_->count_;

                        down();

                        h_ = right.h_;

                        return *this;
                }

                // access to the managed object
                T* operator-> () { return h_->ptr_; }
                T& operator* () { return *h_->ptr_; }
                T* getPointer() { return h_->ptr_; }

                const T* operator-> () const { return h_->ptr_; }
                const T& operator* () const { return *h_->ptr_; }
                const T* getPointer() const { return h_->ptr_; }
};
#endif