Left: | ||
Right: |
OLD | NEW |
---|---|
(Empty) | |
1 // Parts of this code have been copied from boost/smart_ptr/intrusive_ptr.hpp. | |
2 // | |
3 // Copyright (c) 2001, 2002 Peter Dimov | |
4 // | |
5 // Distributed under the Boost Software License, Version 1.0. (See | |
6 // accompanying file LICENSE_1_0.txt or copy at | |
7 // http://www.boost.org/LICENSE_1_0.txt) | |
8 | |
9 #pragma once | |
sergei
2016/02/17 12:55:05
Yahoo! Finally we start it! I am looking forward t
| |
10 | |
11 #include <algorithm> | |
12 #include <type_traits> | |
13 | |
14 class ref_counted | |
15 { | |
16 public: | |
17 void AddRef() | |
18 { | |
19 mRefCount++; | |
20 } | |
21 | |
22 void ReleaseRef() | |
23 { | |
24 if (--mRefCount == 0) | |
25 delete this; | |
26 } | |
27 | |
28 protected: | |
29 ref_counted() | |
30 : mRefCount(0) | |
31 { | |
32 } | |
33 | |
34 // We need this virtual destructor, otherwise pointers to ref_counted and | |
35 // pointers to derived classes won't have the same value (e.g. converting from | |
36 // Filter* to ref_counted* decreases pointer value by 4). | |
37 virtual ~ref_counted() | |
38 { | |
39 } | |
40 | |
41 private: | |
42 int mRefCount; | |
43 }; | |
44 | |
45 template<typename T, | |
46 class = typename std::enable_if<std::is_base_of<ref_counted,T>::value>::type > | |
47 class intrusive_ptr | |
48 { | |
49 public: | |
50 intrusive_ptr() | |
51 : mPointer(nullptr) | |
52 { | |
53 } | |
54 | |
55 intrusive_ptr(T* pointer) | |
56 : mPointer(pointer) | |
57 { | |
58 if (mPointer) | |
59 mPointer->AddRef(); | |
60 } | |
61 | |
62 intrusive_ptr(const intrusive_ptr& other) | |
63 : mPointer(other.mPointer) | |
64 { | |
65 if (mPointer) | |
66 mPointer->AddRef(); | |
67 } | |
68 | |
69 intrusive_ptr(intrusive_ptr&& other) | |
70 : mPointer(other.mPointer) | |
71 { | |
72 other.mPointer = nullptr; | |
73 } | |
74 | |
75 ~intrusive_ptr() | |
76 { | |
77 if (mPointer) | |
78 mPointer->ReleaseRef(); | |
79 } | |
80 | |
81 intrusive_ptr& operator=(intrusive_ptr& other) | |
82 { | |
83 intrusive_ptr(other).swap(*this); | |
84 return *this; | |
85 } | |
86 | |
87 intrusive_ptr& operator=(intrusive_ptr&& other) | |
88 { | |
89 intrusive_ptr(other).swap(*this); | |
sergei
2016/02/17 12:55:07
JIC, `other` is just a reference here, I guess, to
Wladimir Palant
2016/02/18 16:07:14
I'll just trust you on this one, currently I have
| |
90 return *this; | |
91 } | |
92 | |
93 intrusive_ptr& operator=(T* other) | |
94 { | |
95 intrusive_ptr(other).swap(*this); | |
96 return *this; | |
97 } | |
98 | |
99 void reset() | |
100 { | |
101 intrusive_ptr().swap(*this); | |
102 } | |
103 | |
104 void reset(T* other) | |
105 { | |
106 intrusive_ptr(other).swap(*this); | |
107 } | |
108 | |
109 T* get() const | |
sergei
2016/02/17 12:55:06
Although it's a common practice, I'm personally no
Wladimir Palant
2016/02/18 16:07:13
Done, here and for the methods below. None of this
| |
110 { | |
111 return mPointer; | |
112 } | |
113 | |
114 T& operator*() const | |
115 { | |
116 return *mPointer; | |
117 } | |
118 | |
119 T* operator->() const | |
120 { | |
121 return mPointer; | |
122 } | |
123 | |
124 operator bool() const | |
125 { | |
126 return mPointer != nullptr; | |
127 } | |
128 | |
129 operator T*() const | |
130 { | |
131 return mPointer; | |
132 } | |
133 | |
134 bool operator!() const | |
135 { | |
136 return mPointer == nullptr; | |
137 } | |
138 | |
139 void swap(intrusive_ptr& other) | |
140 { | |
141 std::swap(mPointer, other.mPointer); | |
142 } | |
143 | |
144 private: | |
145 T* mPointer; | |
146 }; | |
147 | |
148 template<typename T, typename U> | |
149 inline bool operator==(const intrusive_ptr<T>& a, const intrusive_ptr<U>& b) | |
150 { | |
151 return a.get() == b.get(); | |
152 } | |
153 | |
154 template<typename T, typename U> | |
155 inline bool operator!=(const intrusive_ptr<T>& a, const intrusive_ptr<U>& b) | |
156 { | |
157 return a.get() != b.get(); | |
158 } | |
159 | |
160 template<typename T, typename U> | |
161 inline bool operator==(const intrusive_ptr<T>& a, const U* b) | |
162 { | |
163 return a.get() == b; | |
164 } | |
165 | |
166 template<typename T, typename U> | |
167 inline bool operator!=(const intrusive_ptr<T>& a, const U* b) | |
168 { | |
169 return a.get() != b; | |
170 } | |
171 | |
172 template<typename T, typename U> | |
173 inline bool operator==(const T* a, const intrusive_ptr<U>& b) | |
174 { | |
175 return a == b.get(); | |
176 } | |
177 | |
178 template<typename T, typename U> | |
179 inline bool operator!=(const T* a, intrusive_ptr<U> const& b) | |
180 { | |
181 return a != b.get(); | |
182 } | |
OLD | NEW |