From 78659a9fc97aedbea0c59a477516f41e90932c70 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=82=B9=E6=99=93=E8=88=AA?= <1210603696@qq.com>
Date: Thu, 12 Mar 2015 11:47:21 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85unique=5Fptr?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TinySTL/Memory.h | 37 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

diff --git a/TinySTL/Memory.h b/TinySTL/Memory.h
index dfa804f..2e05e67 100644
--- a/TinySTL/Memory.h
+++ b/TinySTL/Memory.h
@@ -18,6 +18,7 @@ namespace TinySTL{
 	public:
 		typedef T element_type;
 		typedef D deleter_type;
+		typedef element_type *pointer;
 	public:
 		explicit unique_ptr(T *data = nullptr) :data_(data){}
 		unique_ptr(T *data, deleter_type del) :data_(data), deleter(del){}
@@ -29,7 +30,7 @@ namespace TinySTL{
 		unique_ptr& operator = (unique_ptr&& up){
 			if (&up != this){
 				clean();
-				swap(data_, up.data_);
+				swap(*this, up);
 			}
 		}
 
@@ -37,6 +38,28 @@ namespace TinySTL{
 		unique_ptr& operator = (const unique_ptr&) = delete;
 
 		~unique_ptr(){ clean(); }
+
+		pointer get()const{ return data_; }
+		deleter_type& get_deleter(){ return deleter; }
+		const deleter_type& get_deleter()const{ return deleter; }
+
+		operator bool()const{ return get() != nullptr; }
+
+		pointer release(){ 
+			auto p = nullptr; 
+			swap(p, data_);
+			return p;
+		}
+		void reset(pointer p = pointer()){
+			clean();
+			data_ = p;
+		}
+		void swap(unique_ptr& up){
+			swap(data_, up.data_);
+		}
+
+		element_type operator *()const{ return *data_; }
+		pointer operator ->()const{ return data_; }
 	private:
 		inline void clean(){
 			deleter(data_);
@@ -46,6 +69,18 @@ namespace TinySTL{
 		element_type *data_;
 		deleter_type deleter;
 	};
+	template <class T, class D>
+	void swap(unique_ptr<T, D>& x, unique_ptr<T, D>& y){
+		x.swap(y);
+	}
+	template <class T1, class D1, class T2, class D2>
+	bool operator == (const unique_ptr<T1, D1>& lhs, const unique_ptr<T2, D2>& rhs){
+		return lhs.get() == rhs.get();
+	}
+	template <class T1, class D1, class T2, class D2>
+	bool operator != (const unique_ptr<T1, D1>& lhs, const unique_ptr<T2, D2>& rhs){
+		return !(lhs == rhs);
+	}
 
 	template <class T, class... Args>
 	unique_ptr<T> make_unique(Args&&... args){
-- 
GitLab