auto_restore.h

/*****************************************************/
/* lean Smart                   (c) Tobias Zirr 2011 */
/*****************************************************/

#ifndef LEAN_SMART_AUTO_RESTORE
#define LEAN_SMART_AUTO_RESTORE

#include "../lean.h"
#include "../tags/noncopyable.h"

namespace lean
{
namespace smart
{

template <class Value>
class auto_restore : public noncopyable
{
private:
    Value &m_value;
    Value m_initialValue;

public:
    typedef Value value_type;

    explicit auto_restore(value_type &value)
        : m_value(value),
        m_initialValue(value) { }
    auto_restore(value_type &value, const value_type &newValue)
        : m_value(value),
        m_initialValue(value)
    {
        m_value = newValue;
    }

    ~auto_restore()
    {
        m_value = m_initialValue;
    }
    
    LEAN_INLINE value_type& get(void) { return m_value; };
    LEAN_INLINE const value_type& get(void) const { return m_value; };

    LEAN_INLINE value_type& operator *() { return get(); };
    LEAN_INLINE const value_type& operator *() const { return get(); };
    LEAN_INLINE value_type* operator ->() { return &get(); };
    LEAN_INLINE const value_type* operator ->() const { return &get(); };

    LEAN_INLINE operator value_type&() { return get(); };
    LEAN_INLINE operator const value_type&() const { return get(); };
};

} // namespace

using smart::auto_restore;

} // namespace

#endif