#ifndef BOOST_ENABLE_SHARED_FROM_THIS_HPP_INCLUDED
#define BOOST_ENABLE_SHARED_FROM_THIS_HPP_INCLUDED

//
//  enable_shared_from_this.hpp
//
//  Copyright (c) 2002 Peter Dimov
//
// Distributed under the Boost Software License, Version 1.0. (See
// accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//
//  http://www.boost.org/libs/smart_ptr/enable_shared_from_this.html
//

#include <boost/weak_ptr.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/assert.hpp>
#include <boost/config.hpp>

namespace boost
{

class enable_shared_from_this;
template< class Y > void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe );
template< class Y > void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe, void * /*pd*/ );
template< class T > shared_ptr<T> shared_from_this( T * p );

class enable_shared_from_this
{
protected:

    enable_shared_from_this()
    {
    }

    enable_shared_from_this(enable_shared_from_this const &)
    {
    }

    enable_shared_from_this & operator=(enable_shared_from_this const &)
    {
        return *this;
    }

    ~enable_shared_from_this()
    {
    }

private:

    mutable weak_ptr<void> _weak_ptr;

    template< class Y > friend void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe );
    template< class Y > friend void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe, void * /*pd*/ );
    template< class T > friend shared_ptr<T> shared_from_this( T * p );
};

template< class Y > inline void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe )
{
    if( pe != 0 && pe->_weak_ptr.use_count() == 0 )
    {
        pe->_weak_ptr = ptr;
    }
}

template< class Y > inline void sp_accept_owner( shared_ptr<Y> * ptr, enable_shared_from_this const * pe, void * /*pd*/ )
{
    if( pe != 0 && pe->_weak_ptr.use_count() == 0 )
    {
        pe->_weak_ptr = ptr;
    }
}

template< class T > inline shared_ptr<T> shared_from_this( T * p )
{
    return shared_ptr<T>( shared_ptr<void>(_weak_ptr), p );
}

} // namespace boost

#endif  // #ifndef BOOST_ENABLE_SHARED_FROM_THIS_HPP_INCLUDED
