自写的智能指针c++

来源:互联网 发布:淘宝客服是职责是什么 编辑:程序博客网 时间:2024/05/22 11:44

/*!
     @class  引用计数类
 */
    class RefCounted
    {
   
 public:
       
  //! @brief 增加计数
  virtual void G_CALL ref() = 0;

        //! @brief 减少计数
  virtual void G_CALL unref() = 0;
 
 protected:
 
        ~RefCounted()
  {
  }
    };

template< typename T >
   class RefPtr
   {
   public:
      RefPtr(T* ptr = 0)
      {
         mPtr = 0;
         *this = ptr;
      }

      RefPtr(const RefPtr<T>& ptr)
      {
         mPtr = 0;
         *this = ptr;
      }

      ~RefPtr()
      {
         if (mPtr)
         {
            mPtr->unref();
            mPtr = 0;
         }
      }

      RefPtr<T>& operator=(T* ptr)
      {
         if (ptr != mPtr)
         {
            if (mPtr)
            {
               mPtr->unref();
            }
            mPtr = ptr;
            if (mPtr)
            {
               mPtr->ref();
            }
         }
         return *this;
      }

      RefPtr<T>& operator=(const RefPtr<T>& ptr)
      {
         *this = ptr.mPtr;
         return *this;
      }

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

      T& operator*() const
      {
         return *mPtr;
      }

      operator bool() const
      {
         return (mPtr != 0);
      }

      T* get() const
      {
         return mPtr;
      }

   private:
      T* mPtr;
   };

原创粉丝点击