SharedPtr.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
 
AST_NAMESPACE_BEGIN
 
template<typename _Object>
class SharedPtr
{
public:
    SharedPtr()
        :object_{ nullptr }
    {}
    SharedPtr(std::nullptr_t)
        :object_{ nullptr }
    {
    }
    SharedPtr(_Object* obj)
        :object_{ obj }
    {
        _incRef();
    }
    SharedPtr(const SharedPtr& ptr)
        :SharedPtr{ptr.get()}
    {
 
    }
    ~SharedPtr()
    {
        _decRef();
    }
    SharedPtr& operator=(const SharedPtr& ptr)
    {
        return this->operator=(ptr.get());
    }
 
    SharedPtr& operator=(_Object* obj)
    {
        if(obj != object_){ 
            if(obj)
                obj->incRef();
            _decRef();
            object_ = obj;
        }
        return *this;
    }
    SharedPtr& operator=(std::nullptr_t)
    {
        reset();
        return *this;
    }
    // 这里还是设置为支持隐式转换，因为在很多情况下，我们需要将SharedPtr转换为_Object*
    // explicit 
    operator _Object*() const
    {
        return object_;
    }
    explicit operator bool() const
    {
        return object_ != nullptr;
    }
    _Object* operator->() const
    {
        return object_;
    }
    _Object* get() const
    {
        return object_;
    }
    _Object* take()
    {
        _Object* obj = object_;
        if(obj){
            obj->decRefNoDelete();
            object_ = nullptr;
        }
        return obj;
    }
    void reset()
    {
        _decRef();
        object_ = nullptr;
    }
protected:
    void _incRef()
    {
        if(object_)
            object_->incRef();
    }
    void _decRef()
    {
        if(object_)
            object_->decRef();
    }
protected:
    _Object* object_{nullptr};
};
 
 
 
AST_NAMESPACE_END