AutoPtr.h

// -*- c++ -*-

#ifndef _AUTO_PTR_H_
#define _AUTO_PTR_H_

// system
#include <stdlib.h> // size_t

// tools
#include "tools/NoCopy.h"


template<class T>class AutoPtr
{
public:
    AutoPtr() : ptr(0) {};

    AutoPtr(T *in) : ptr(in) {};

    AutoPtr(AutoPtr &rhs)
    {
        ptr = rhs.release();
    }

    ~AutoPtr()
    {
        assign(0);
    }

    AutoPtr & operator = (AutoPtr &rhs)
    {
        assign(rhs.release());
        return *this;
    }

    AutoPtr & operator = (T *p)
    {
        assign(p);
        return *this;
    }
    
    operator bool () const
    {
        return ptr != 0;
    }
    
    T &operator *() const
    {
        return *ptr;
    }

    T *operator ->() const
    {
        return ptr;
    }

    T *get() const
    {
        return ptr;
    }

    operator T * () const
    {
        return ptr;
    }

    void assign(T *new_ptr)
    {
        if (ptr)
            delete ptr;
        ptr = new_ptr;
    }

    T * release()
    {
        T *tmp = ptr;
        ptr = 0;
        return tmp;
    }
    
private:
    PROHIBIT_DEFAULT_COPY(AutoPtr);
    T *ptr;
};

template<class T>class AutoArrayPtr
{
public:
    AutoArrayPtr() : arr(0) {};

    AutoArrayPtr(T *in) : arr(in) {};

    AutoArrayPtr(AutoArrayPtr &rhs)
    {
        arr = rhs.release();
    }

    ~AutoArrayPtr()
    {
        assign(0);
    }

    AutoArrayPtr & operator = (AutoArrayPtr &rhs)
    {
        assign(rhs.release());
        return *this;
    }

    AutoArrayPtr & operator = (T *p)
    {
        assign(p);
        return *this;
    }
    
    operator bool () const
    {
        return arr != 0;
    }

    T * get() const
    {
        return arr;
    }
    
    T & operator [] (size_t i) const
    {
        return arr[i];
    }

    void assign(T *new_arr)
    {
        if (arr)
            delete [] arr;
        arr = new_arr;
    }

    T * release()
    {
        T *tmp = arr;
        arr = 0;
        return tmp;
    }
    
private:
    PROHIBIT_DEFAULT_COPY(AutoArrayPtr);
    T *arr;
};

#endif