apps/elftosb/common/smart_ptr.h

/*
 *  smart_ptr.h
 *  elftosb
 *
 *  Created by Chris Reed on 4/18/06.
 *  Copyright 2006 __MyCompanyName__. All rights reserved.
 *
 */
#if !defined(_smart_ptr_h_)
#define _smart_ptr_h_

template <typename T>
class smart_ptr
{
public:
    typedef T data_type;
    typedef T *ptr_type;
    typedef const T *const_ptr_type;
    typedef T &ref_type;
    typedef const T &const_ref_type;

    smart_ptr()
        : _p(0)
    {
    }

    smart_ptr(ptr_type p)
        : _p(p)
    {
    }

    virtual ~smart_ptr() { safe_delete(); }
    ptr_type get() { return _p; }
    const_ptr_type get() const { return _p; }
    void set(ptr_type p)
    {
        if (_p && p != _p)
        {
            safe_delete();
        }
        _p = p;
    }

    void reset() { _p = 0; }
    void clear() { safe_delete(); }
    virtual void safe_delete()
    {
        if (_p)
        {
            delete _p;
            _p = 0;
        }
    }



    operator ptr_type() { return _p; }
    operator const_ptr_type() const { return _p; }
    operator ref_type() { return *_p; }
    operator const_ref_type() const { return *_p; }
    operator bool() const { return _p != 0; }
    smart_ptr<T> &operator=(const_ptr_type p)
    {
        set(const_cast<ptr_type>(p));
        return *this;
    }

    ptr_type operator->() { return _p; }
    const_ptr_type operator->() const { return _p; }
    //  //! Pointer dereferencing operator.
    //  ref_type operator * () const { return *_p; }
    //
    //  //! Const version of the pointer dereference operator.
    //  const_ref_type operator * () const { return *_p; }


protected:
    ptr_type _p; 
};

template <typename T>
class smart_array_ptr
{
public:
    typedef T data_type;
    typedef T *ptr_type;
    typedef const T *const_ptr_type;
    typedef T &ref_type;
    typedef const T &const_ref_type;

    smart_array_ptr()
        : _p(0)
    {
    }

    smart_array_ptr(ptr_type p)
        : _p(p)
    {
    }

    virtual ~smart_array_ptr() { safe_delete(); }
    ptr_type get() { return _p; }
    const_ptr_type get() const { return _p; }
    void set(ptr_type p)
    {
        if (_p && p != _p)
        {
            safe_delete();
        }
        _p = p;
    }

    void reset() { _p = 0; }
    void clear() { safe_delete(); }
    virtual void safe_delete()
    {
        if (_p)
        {
            delete[] _p;
            _p = 0;
        }
    }



    operator ptr_type() { return _p; }
    operator const_ptr_type() const { return _p; }
    operator ref_type() { return *_p; }
    operator const_ref_type() const { return *_p; }
    operator bool() const { return _p != 0; }
    smart_array_ptr<T> &operator=(const_ptr_type p)
    {
        set(const_cast<ptr_type>(p));
        return *this;
    }

    ptr_type operator->() { return _p; }
    const_ptr_type operator->() const { return _p; }
    ref_type operator[](unsigned index) { return _p[index]; }
    const_ref_type operator[](unsigned index) const { return _p[index]; }
    //  //! Pointer dereferencing operator.
    //  ref_type operator * () const { return *_p; }
    //
    //  //! Const version of the pointer dereference operator.
    //  const_ref_type operator * () const { return *_p; }


protected:
    ptr_type _p; 
};

#endif // _smart_ptr_h_