atsobjptr.h

来自「在你的程序中实现脚本功能,该脚本引擎名为AtScript , 是类似于VBScr」· C头文件 代码 · 共 97 行

#ifndef __AMObjPtr
#define __AMObjPtr

/////////////////////////////////////////////
// 自动删除指针
/////////////////////////////////////////////
template <class T>
class CAtsObjPtr
{
public:
	typedef T _PtrClass;
	CAtsObjPtr()
	{
		p = NULL;
	}
	CAtsObjPtr(T* lp)
	{
		p = lp;
	}
	CAtsObjPtr(const CAtsObjPtr<T>& lp)
	{
		if ((p = lp.p) != NULL)
			p->AddRef();
	}
	~CAtsObjPtr()
	{
		Release();
	}
	long Release()
	{
		if (p)
			delete p;

		return 0;
	}
	operator T*() const
	{
		return (T*)p;
	}
	T& operator*() const
	{
		ATLASSERT(p!=NULL);
		return *p;
	}
	T* operator->() const
	{
		return p;
	}
	T* operator=(T* lp)
	{
		p = lp;
		return lp;
	}
	T* operator=(const CAtsObjPtr<T>& lp)
	{
		if (lp != NULL)
			lp.p->AddRef();
		if (p)
			p->Release();
		p = lp;
		return lp;
	}
	bool operator!() const
	{
		return (p == NULL);
	}
	bool operator<(T* pT) const
	{
		return p < pT;
	}
	bool operator==(T* pT) const
	{
		return p == pT;
	}
	bool IsEqualObject(T* pOther)
	{
		if ( p == pOther )
			return true;

		return false;
	}
	void Attach(T* p2)
	{
		if (p)
			p->Release();
		p = p2;
	}
	T* Detach()
	{
		T* pt = p;
		p = NULL;
		return pt;
	}
	T* p;
};

#endif // __AMObjPtr