c++智能指针简单实现

#ifndef _smart_str
#define _smart_str
#include<iostream>
using namespace std;
class use {//用于测试->重载
public:
	void method(){ cout << "Method is called!" << endl; }
};
template <typename T>
class smart_ptr {
private:
	T* px;
	int* pn;//智能指针最重要的一点:计数
public:
	smart_ptr(T* p) ;//构造函数
	smart_ptr(const smart_ptr<T>& ori);//拷贝构造函数
	smart_ptr<T>& operator=(const smart_ptr<T>& rig);//重载=时注意 返回类型设置时要考虑连续赋值所以不要用void
	~smart_ptr();
	T& operator*()const;//对于指针*，->的实现必不可少
	T* operator->()const;
	void getcount() {
			return *pn;
		}
};
template <typename T>
smart_ptr<T>::smart_ptr(T* p) :px(p)
{
	try
	{
		pn = new int(1);
	}
	catch (const std::exception&)
	{
		delete px;
		px = nullptr;
		delete pn;
		pn = nullptr;
	}
	cout << "Constructor is called!" << endl;
}
template <typename T>
smart_ptr<T>::smart_ptr(const smart_ptr<T>& ori) {
	pn = ori.pn;
	px = ori.px;
	++(*ori.pn);//拷贝赋值计数器加一
	cout << "Copy Constructor is called!" << endl;
}
template <typename T>
smart_ptr<T>& smart_ptr<T>::operator=(const smart_ptr<T>& rig) {
	if (this->px == rig.px)//=赋值注意点1：自我复制处理
	{
		cout << "Self assignment is called!" << endl;
		return *this;
	}
	if (--(*pn) == 0) {//注意店2：要覆盖=左边的值前判断该指针的计数器是否为1，然后释放空间
		delete px;
		px=nullptr;
		delete pn;
		pn=nullptr;
	}
	++(*rig.pn);
	px = rig.px;
	pn = rig.pn;
	cout << "Assignment operator overloaded is called!" << endl;
	return *this;
	
}
template <typename T>
T& smart_ptr<T>::operator*()const {
	cout << "Assignment operator* overloaded is called!" << endl;
	return *px;
}
template <typename T>
T* smart_ptr<T>::operator->()const {
	cout << "Assignment operator-> overloaded is called!" << endl;
	return px;
}
template <typename T>
smart_ptr<T>::~smart_ptr() {
	if (--(*pn) == 0) {
		delete pn;
		delete px;
		px = nullptr;
		pn = nullptr;
		cout << "Destructor is called!" << endl;
	}
	
}
int main() {
	smart_ptr<int> p1(new int(0));
	p1 = p1;
	cout << *p1;
	smart_ptr<int> p2(p1);
	smart_ptr<int> p3(new int(1));
	smart_ptr<use> sp(new use);
	sp->method();
	p3 = p1;
	
	return 0;
}
#endif