ptr.h

Go to the documentation of this file.

/*
 * /phoenix/include/triton/ptr.h
 *
 * This file is a part of Phoenix operating system.
 * Copyright (c) 2011-2012, Artyom Lebedev <artyom.lebedev@gmail.com>
 * All rights reserved.
 * See COPYING file for copyright details.
 */

#ifndef PTR_H_
#define PTR_H_

namespace triton {

template <class T>
class Ptr {
private:
    T *_obj;
    bool _canDelete;

    inline void
    _ReleaseObj()
    {
        if (_obj && !_obj->Release() && _canDelete) {
            delete _obj;
        }
        _obj = nullptr;
    }
public:

    inline
    Ptr() {
        _obj = nullptr;
    }

    inline
    Ptr(T *objPtr) {
        if (objPtr) {
            objPtr->AddRef();
        }
        _obj = objPtr;
        _canDelete = true;
    }

    inline
    Ptr(T &obj) {
        obj.AddRef();
        _obj = &obj;
        _canDelete = false;
    }

    inline
    Ptr(Ptr<T> &ptr) {
        if (ptr._obj) {
            ptr._obj->AddRef();
            _obj = ptr._obj;
            _canDelete = ptr._canDelete;
        } else {
            _obj = nullptr;
        }
    }

    inline
    Ptr(Ptr<T> &&ptr) {
        _obj = ptr._obj;
        _canDelete = ptr._canDelete;
        ptr._obj = nullptr;
    }

    inline
    ~Ptr() {
        _ReleaseObj();
    }

    inline Ptr<T> &
    operator =(Ptr<T> &ptr)
    {
        _ReleaseObj();
        if (ptr._obj) {
            ptr._obj->AddRef();
            _obj = ptr._obj;
            _canDelete = ptr._canDelete;
        } else {
            _obj = nullptr;
        }
        return *this;
    }

    inline Ptr<T> &
    operator =(Ptr<T> &&ptr)
    {
        _ReleaseObj();
        _obj = ptr._obj;
        _canDelete = ptr._canDelete;
        ptr._obj = nullptr;
        return *this;
    }

    inline Ptr<T> &
    operator =(T *objPtr)
    {
        _ReleaseObj();
        if (objPtr) {
            objPtr->AddRef();
        }
        _obj = objPtr;
        _canDelete = true;
        return *this;
    }

    inline Ptr<T> &
    operator =(T &obj)
    {
        _ReleaseObj();
        obj.AddRef();
        _obj = &obj;
        _canDelete = false;
        return *this;
    }

    inline bool
    operator ==(const Ptr<T> &ptr) const
    {
        return _obj == ptr._obj;
    }

    inline bool
    operator !=(const Ptr<T> &ptr) const
    {
        return _obj != ptr._obj;
    }

    inline
    operator bool() const
    {
        return _obj != nullptr;
    }

    inline T &
    operator *() const
    {
        if (UNLIKELY(!_obj)) {
            throw NullPtrError();
        }
        return *_obj;
    }

    inline T *
    operator ->() const
    {
        if (UNLIKELY(!_obj)) {
            throw NullPtrError();
        }
        return _obj;
    }
};

template <class T>
inline size_t
len(Ptr<T> &objPtr)
{
    return objPtr->__len__();
}

template <class T>
inline size_t
len(Ptr<T> &&objPtr)
{
    return objPtr->__len__();
}

template <class T>
inline Object::hash_t
hash(Ptr<T> &objPtr)
{
    return objPtr->__hash__();
}

template <class T>
inline Object::hash_t
hash(Ptr<T> &&objPtr)
{
    return objPtr->__hash__();
}

} /* namespace triton */

#endif /* PTR_H_ */