读C++ Primer 之智能指针

/*
* author: xizero00
* mail:[email protected]
* date:2011-08-07 20:19:11 
* smart pointer sample
*/


#include <iostream>
using namespace std;


//智能指针类
class smartPtr
{
	//友元类
	friend class HasPtr;
private:
	//实际的指针
	int *p;
	//引用计数
	size_t use;
	//智能指针类的构造函数
	smartPtr( int *ptr ): p( ptr ) , use (1) {}
	//析构函数
	~smartPtr() { delete p; }
	
};

//使用智能指针的类
//使用智能指针的类一般需要  复制构造函数、赋值操作符重载、析构函数
class HasPtr
{
private:
	smartPtr *ptr;
	int val;
public:
	//默认构造函数
	HasPtr(): ptr( NULL ) , val( 0 ) {}
	
	//带参数的构造函数
	HasPtr( int *p , int v ): ptr( new smartPtr( p ) ) , val( v ) {}
	
	//复制构造函数
	HasPtr( const HasPtr &h ): ptr( h.ptr ) , val( h.val ) { ++ptr->use; }
	
	//赋值重载，用于对赋值操作进行监控
	HasPtr& operator= ( const HasPtr& );
	
	//析构函数，监控删除操作
	~HasPtr() { if( 0 == --ptr->use ) delete ptr; } // 判断是否为最后一个引用
		
	
	//属性
	int get_int() const
	{
		return val;
	}
	
	void set_int( int v )
	{
		val = v;
	}
	
	int* get_ptr() const
	{
		return ptr->p;
	}
	
	
	void set_ptr( int *p )
	{
		ptr->p = p;
	}
	
	
};

//赋值操作符虫重载
HasPtr& HasPtr::operator= (const HasPtr &h)
{
	++ h.ptr->use;
	
	if( 0 == --ptr->use )
	{
		delete ptr;
	}
	
	ptr = h.ptr;
	val = h.val;
	return *this;
}




int main(int argc , char **argv)
{
	int *obj = new int[2]{1 , 2};
	HasPtr *h1 = new HasPtr( obj , 2 );
	//HasPtr *h2 = new HasPtr( obj , 1 );
	HasPtr h3 = *h1;//引用h1其中的指向数据的指针两次
	cout << "h1:" << *( h1->get_ptr() ) << endl;
	cout << "h3:" << *( h3.get_ptr() ) << endl;
	
	
	cout << "删除h1，此时剩下h3" << endl;
	//如果不用智能指针会导致obj所指向的数据被删除了，
	//而h3中的会成为悬垂指针
	//因为这里使用类智能指针
	//所以不会删除obj所指向的数据的空间，所以不会出错！
	delete h1;
	cout << "h3:" << *( h3.get_ptr() ) << endl;
	
	
	//cout << "h2:" << *h2.get_ptr() << endl;
	
	return 0;
}

编译的环境是g++ 因为数组的初始化用到了c++ 0x的特性，所以编译命令为

	g++ HasPtr.cc -std=c++0x