referenced.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.1 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_REFERENCED_H
#define COBASE_REFERENCED_H

#include <co/base/api.h>      // for COBASE_API
#include <co/base/atomic.h>   // member
#include <co/base/debug.h>    // for EQERROR
#include <co/base/refPtr.h>   // EQ_REFERENCED_ARGS

#ifdef EQ_REFERENCED_DEBUG
#  include <co/base/hash.h>
#  include <co/base/lock.h>
#  include <co/base/lockable.h>
#  include <co/base/scopedMutex.h>
#endif

#include <typeinfo>

namespace co
{
namespace base
{
    class Referenced 
    {
    public:
        void ref( EQ_REFERENCED_ARGS ) const
        {
#ifndef NDEBUG
            EQASSERTINFO( !_hasBeenDeleted, className( this ));
#endif
            ++_refCount;

#ifdef EQ_REFERENCED_DEBUG
            std::stringstream cs;
            cs << backtrace;
            ScopedMutex<> referencedMutex( _holders );
            HolderHash::iterator i = _holders->find( holder );
            EQASSERTINFO( i == _holders->end(), i->second );
            _holders.data[ holder ] = cs.str();
#endif
        }

        void unref( EQ_REFERENCED_ARGS ) const
            { 
#ifndef NDEBUG
                EQASSERT( !_hasBeenDeleted );
#endif
                EQASSERT( _refCount > 0 ); 
                const bool deleteMe = (--_refCount==0);
                if( deleteMe )
                    deleteReferenced( this );
#ifdef EQ_REFERENCED_DEBUG
                else
                {
                    ScopedMutex<> referencedMutex( _holders );
                    HolderHash::iterator i = _holders->find( holder );
                    EQASSERT( i != _holders->end( ));
                    _holders->erase( i );
                }
#endif
            }

        int  getRefCount() const { return _refCount; }

        void printHolders( std::ostream& os ) const
            {
#ifdef EQ_REFERENCED_DEBUG
                os << disableFlush << disableHeader;
                ScopedMutex<> referencedMutex( _holders );
                for( HolderHash::const_iterator i = _holders->begin();
                     i != _holders->end(); ++i )
                {
                    os << "Holder " << i->first << ": " << i->second 
                       << std::endl;
                }
                os << enableHeader << enableFlush;
#endif
            }

    protected:
        Referenced()
                : _refCount( 0 )
                , _hasBeenDeleted( false )
            {}

        Referenced( const Referenced& ) 
                : _refCount( 0 )
                , _hasBeenDeleted( false )
            {}

        virtual ~Referenced() 
            {
#ifndef NDEBUG
                EQASSERT( !_hasBeenDeleted );
                _hasBeenDeleted = true;
#endif
                EQASSERTINFO( _refCount == 0,
                              "Deleting object with ref count " << _refCount );
            }

        COBASE_API void deleteReferenced( const Referenced* object ) const;

    private:
        mutable a_int32_t _refCount;
        bool _hasBeenDeleted;

#ifdef EQ_REFERENCED_DEBUG
        typedef PtrHash< const void*, std::string > HolderHash;
        mutable Lockable< HolderHash, Lock > _holders;
#endif
    };
}

}
#endif //COBASE_REFERENCED_H