refPtr.h

/* Copyright (c) 2005-2011, Stefan Eilemann <eile@equalizergraphics.com> 
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 2.1q as published
 * by the Free Software Foundation.
 *  
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef COBASE_REFPTR_H
#define COBASE_REFPTR_H

#include <co/base/debug.h>

#include <iostream>
#include <typeinfo>
#include <stdlib.h>

//#define EQ_REFERENCED_DEBUG
#ifdef EQ_REFERENCED_DEBUG
#  define EQ_REFERENCED_ARGS const void* holder
#  define EQ_REFERENCED_PARAM (const void*)(this)
#else
#  define EQ_REFERENCED_ARGS
#  define EQ_REFERENCED_PARAM
#endif

namespace co
{
namespace base
{
    template<class T> class RefPtr 
    {
    public:
        RefPtr()                     : _ptr( 0 )         {}

        RefPtr( T* const ptr )       : _ptr( ptr )       { _ref(); }

        RefPtr( const RefPtr& from ) : _ptr( from._ptr ) { _ref(); }

        template< class O > RefPtr( RefPtr< O > from )
                : _ptr( from.get( )) { _ref(); }

        ~RefPtr() { _unref(); _ptr = 0; }

        RefPtr& operator = ( const RefPtr& rhs )
            {
                if( _ptr == rhs._ptr )
                    return *this;

                T* tmp = _ptr;
                _ptr = rhs._ptr;
                _ref();
                if( tmp ) tmp->unref( EQ_REFERENCED_PARAM );
                return *this;
            }

        RefPtr& operator = ( T* ptr )
            {
                if( _ptr == ptr )
                    return *this;

                T* tmp = _ptr;
                _ptr = ptr;
                _ref();
                if( tmp ) tmp->unref( EQ_REFERENCED_PARAM );
                return *this;
            }

        bool operator == ( const RefPtr& rhs ) const 
            { return ( _ptr == rhs._ptr ); }

        bool operator != ( const RefPtr& rhs ) const
            { return ( _ptr != rhs._ptr ); }

        bool operator < ( const RefPtr& rhs ) const
            { return ( _ptr < rhs._ptr ); }

        bool operator > ( const RefPtr& rhs ) const
            { return ( _ptr > rhs._ptr ); }

        bool operator ! () const               { return ( _ptr==0 ); }

        bool operator == ( const T* ptr ) const { return ( _ptr == ptr ); }

        bool operator != ( const T* ptr ) const { return ( _ptr != ptr ); }

        T*       operator->()       
            { EQASSERTINFO( _ptr, typeid(*this).name( )); return _ptr; }
        const T* operator->() const
            { EQASSERTINFO( _ptr, typeid(*this).name( )); return _ptr; }
        T&       operator*()        
            { EQASSERTINFO( _ptr, typeid(*this).name( )); return *_ptr; }
        const T& operator*() const  
            { EQASSERTINFO( _ptr, typeid(*this).name( )); return *_ptr; }

        T*       get()                { return _ptr; }
        const T* get() const          { return _ptr; }

        bool isValid() const { return ( _ptr != 0 ); }
        
    private:
        T* _ptr;

        void _ref()   { if(_ptr) _ptr->ref( EQ_REFERENCED_PARAM ); }

        void _unref() 
        {
            if(_ptr)
            {
#ifndef NDEBUG
                const bool abondon = (_ptr->getRefCount() == 1);
                _ptr->unref( EQ_REFERENCED_PARAM );
                if( abondon ) 
                    _ptr = 0;
#else
                _ptr->unref( EQ_REFERENCED_PARAM );
#endif
            }
        }
    };

    template< class T >
    inline std::ostream& operator << ( std::ostream& os, const RefPtr<T>& rp )
    {
        const T* p = rp.get();
        if( p )
        {
            os << disableFlush << "RP<" << *p << ">";
#ifdef EQ_REFERENCED_DEBUG
            os << std::endl;
            p->printHolders( os );
#endif
            os << enableFlush;
        }
        else
            os << "RP< NULL >";

        return os;
    }

    template< class T > inline std::string className( const RefPtr<T>& rp )
    { return className( rp.get( )); }
}

}
#endif //COBASE_REFPTR_H