001:
002:
003:
004:
005:
006:
007:
008:
009:
010:
011:
012:
013:
014:
015:
016:
017:
018:
019:
020:
021:
022:
023:
024:
025:
026:
027:
028:
029:
030:
031:
032:
033:
034:
035:
036:
037:
038:
039:
040:
041:
042:
043:
044:
045:
046:
047:
048:
049:
050:
051:
052:
053:
054:
055:
056:
057:
058:
059:
060:
061:
062:
063:
064:
065:
066:
067:
068:
069:
070:
071:
072:
073:
074:
075:
076:
077:
078:
079:
080:
081:
082:
083:
084:
085:
086:
087:
088:
089:
090:
091:
092:
093:
094:
095:
096:
097:
098:
099:
100:
101:
102:
103:
104:
105:
106:
107:
108:
109:
110:
111:
112:
113:
114:
115:
116:
117:
118:
119:
120:
121:
122:
123:
124:
125:
126:
127:
128:
129:
130:
131:
132:
133:
134:
135:
136:
137:
138:
139:
140:
141:
142:
143:
144:
145:
146:
147:
148:
149:
150:
151:
152:
153:
154:
155:
156:
157:
158:
159:
160:
161:
162:
163:
164:
165:
166:
167:
168:
169:
170:
171:
172:
173:
174:
175:
176:
177:
178:
179:
180:
181:
182:
183:
184:
185:
186:
187:
188:
189:
190:
191:
192:
193:
194:
195:
196:
197:
198:
199:
200:
201:
202:
203:
204:
205:
206:
207:
208:
209:
210:
211:
212:
213:
214:
215:
216:
217:
218:
219:
220:
221:
222:
223:
224:
225:
226:
227:
228:
229:
230:
231:
232:
233:
234:
235:
236:
237:
238:
239:
240:
241:
242:
243:
244:
245:
246:
247:
248:
249:
250:
251:
252:
253:
254:
255:
256:
257:
258:
259:
260:
261:
262:
263:
264:
265:
266:
267:
268:
269:
270:
271:
272:
273:
274:
275:
276:
277:
278:
279:
280:
281:
282:
283:
284:
285:
286:
287:
288:
289:
290:
291:
292:
293:
294:
295:
296:
297:
298:
299:
300:
301:
302:
303:
304:
305:
306:
307:
308:
309:
310:
311:
312:
313:
314:
315:
316:
317:
318:
319:
320:
321:
322:
323:
324:
325:
326:
327:
328:
329:
330:
331:
332:
333:
334:
335:
336:
337:
338:
339:
340:
341:
342:
343:
344:
345:
346:
347:
348:
349:
350:
351:
352:
353:
354:
355:
/*
* This file is part of the Omni C++ framework
*
* Copyright (c) 2016, Zeriph Enterprises, LLC
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* - Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* - Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* - Neither the name of Zeriph, Zeriph Enterprises, LLC, nor the names
* of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY ZERIPH AND CONTRIBUTORS "AS IS" AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL ZERIPH AND CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#ifndef OMNI_SMART_PTR_HPP
#define OMNI_SMART_PTR_HPP 1
#include <omni/defs/class_macros.hpp>
#include <omni/sync/lock.hpp>
#include <omni/sync/scoped_lock.hpp>
namespace omni {
template < typename T >
class smart_ptr
{
public:
/** Defines the underlying type of this instance */
typedef T pointer_t;
explicit smart_ptr(pointer_t *val = OMNI_NULL) :
OMNI_CTOR_FW(omni::smart_ptr<T>)
m_ptr(val),
m_cnt(new unsigned long(1))
{
OMNI_D5_FW("created by ptr");
}
smart_ptr(const smart_ptr<T> &cp) :
OMNI_CPCTOR_FW(cp)
m_ptr(cp.m_ptr),
m_cnt(cp.m_cnt)
{
this->_increment();
OMNI_DV5_FW("created by copy, ref count = ", *this->m_cnt);
}
template < typename T1 >
smart_ptr(smart_ptr<T1> &cp) :
OMNI_CPCTOR_FW(cp)
m_ptr(cp.m_ptr),
m_cnt(cp.m_cnt)
{
this->_increment();
OMNI_DV5_FW("created by copy, ref count = ", *this->m_cnt);
}
~smart_ptr()
{
OMNI_TRY_FW
OMNI_DTOR_FW
this->_decrement();
OMNI_CATCH_FW
OMNI_D5_FW("destroyed");
}
bool valid() const
{
return (this->m_ptr != OMNI_NULL);
}
operator bool() const
{
return valid();
}
bool operator!() const
{
return (!operator bool());
}
pointer_t& operator*() const
{
return *this->m_ptr;
}
pointer_t* operator->() const
{
return this->m_ptr;
}
bool operator==(const smart_ptr< T > &val) const
{
if (this == &val) { return true; }
return (this->m_ptr == val.m_ptr && this->m_cnt == val.m_cnt)
OMNI_EQUAL_FW(val);
}
bool operator!=(const smart_ptr< T > &val) const
{
return !(*this == val);
}
smart_ptr<T>& operator=(const smart_ptr<T> &val)
{
if (this != &val) {
this->_decrement();
OMNI_ASSIGN_FW(val)
this->m_ptr = val.m_ptr;
this->m_cnt = val.m_cnt;
this->_increment();
}
return *this;
}
template < typename T1 >
smart_ptr& operator=(const smart_ptr<T1> &val)
{
if (this != &val) {
this->_decrement();
OMNI_ASSIGN_FW(val)
this->m_ptr = val.m_ptr;
this->m_cnt = val.m_cnt;
this->_increment();
}
return *this;
}
smart_ptr<T>& operator=(pointer_t *val)
{
if (this->m_ptr != val) {
this->_decrement();
#if defined(OMNI_TYPE_INFO)
this->m_type = omni::type_id< omni::smart_ptr<T> >();
#endif
this->m_ptr = val;
this->_increment();
}
return *this;
}
OMNI_MEMBERS_FW(omni::smart_ptr<T>) // disposing,name,type(),hash()
private:
inline void _decrement()
{
if (--*this->m_cnt == 0) {
OMNI_D5_FW("ref count == 0, freeing pointers");
delete this->m_cnt;
delete this->m_ptr;
this->m_cnt = OMNI_NULL;
this->m_ptr = OMNI_NULL;
}
}
inline void _increment()
{
if (!this->m_cnt) { this->m_cnt = new unsigned long(0); }
++*this->m_cnt;
}
#if defined(OMNI_TYPE_INFO)
omni::type< omni::smart_ptr<T> > m_type;
#endif
T* m_ptr;
unsigned long* m_cnt;
// TODO: unsigned long long or std::size_t type?
};
template < typename T >
class smart_ptr_safe
{
public:
/** Defines the underlying type of this instance */
typedef T pointer_t;
explicit smart_ptr_safe(pointer_t *val = 0) :
OMNI_CTOR_FW(omni::smart_ptr_safe<T>)
m_ptr(val),
m_cnt(new unsigned long(1)),
m_mtx()
{
omni::sync::mutex_init(this->m_mtx);
OMNI_D5_FW("created by ptr");
}
smart_ptr_safe(const smart_ptr_safe<T> &cp) :
OMNI_CPCTOR_FW(cp)
m_ptr(cp.m_ptr),
m_cnt(cp.m_cnt),
m_mtx()
{
omni::sync::mutex_init(this->m_mtx);
this->_increment();
OMNI_DV5_FW("created by copy, ref count = ", *this->m_cnt);
}
template < typename T1 >
smart_ptr_safe(smart_ptr_safe<T1> &cp) :
OMNI_CPCTOR_FW(cp)
m_ptr(cp.m_ptr),
m_cnt(cp.m_cnt),
m_mtx()
{
omni::sync::mutex_init(this->m_mtx);
this->_increment();
OMNI_DV5_FW("created by copy, ref count = ", *this->m_cnt);
}
~smart_ptr_safe()
{
OMNI_TRY_FW
OMNI_DTOR_FW
this->_decrement();
omni::sync::mutex_destroy(this->m_mtx);
OMNI_CATCH_FW
OMNI_D5_FW("destroyed");
}
bool valid() const
{
omni::sync::scoped_lock<omni::sync::mutex_t> alock(&this->m_mtx);
return (this->m_ptr != 0);
}
operator bool() const
{
return this->valid();
}
bool operator!() const
{
return (!operator bool());
}
pointer_t& operator*() const
{
omni::sync::scoped_lock<omni::sync::mutex_t> alock(&this->m_mtx);
return *this->m_ptr;
}
pointer_t* operator->() const
{
omni::sync::scoped_lock<omni::sync::mutex_t> alock(&this->m_mtx);
return this->m_ptr;
}
bool operator==(const smart_ptr_safe< T > &val) const
{
if (this == &val) { return true; }
omni::sync::scoped_lock<omni::sync::mutex_t> a1(&this->m_mtx);
omni::sync::scoped_lock<omni::sync::mutex_t> a2(&val.m_mtx);
return (this->m_ptr == val.m_ptr && this->m_cnt == val.m_cnt)
OMNI_EQUAL_FW(val);
}
bool operator!=(const smart_ptr_safe< T > &val) const
{
return !(*this == val);
}
smart_ptr_safe<T>& operator=(const smart_ptr_safe<T> &val)
{
if (this != &val) {
this->_decrement();
omni::sync::mutex_lock(this->m_mtx);
omni::sync::mutex_lock(val.m_mtx);
OMNI_ASSIGN_FW(val)
this->m_ptr = val.m_ptr;
this->m_cnt = val.m_cnt;
omni::sync::mutex_unlock(this->m_mtx);
omni::sync::mutex_unlock(val.m_mtx);
this->_increment();
}
return *this;
}
template < typename T1 >
smart_ptr_safe& operator=(const smart_ptr_safe<T1> &val)
{
if (this != &val) {
this->_decrement();
omni::sync::mutex_lock(this->m_mtx);
omni::sync::mutex_lock(val.m_mtx);
OMNI_ASSIGN_FW(val)
this->m_ptr = val.m_ptr;
this->m_cnt = val.m_cnt;
omni::sync::mutex_unlock(this->m_mtx);
omni::sync::mutex_unlock(val.m_mtx);
this->_increment();
}
return *this;
}
smart_ptr_safe<T>& operator=(pointer_t *val)
{
if (this->m_ptr != val) {
this->_decrement();
omni::sync::mutex_lock(this->m_mtx);
#if defined(OMNI_TYPE_INFO)
this->m_type = omni::type_id< omni::smart_ptr_safe<T> >();
#endif
this->m_ptr = val;
omni::sync::mutex_unlock(this->m_mtx);
this->_increment();
}
return *this;
}
OMNI_MEMBERS_FW(omni::smart_ptr_safe<T>) // disposing,name,type(),hash()
private:
inline void _decrement()
{
omni::sync::scoped_lock<omni::sync::mutex_t> alock(&this->m_mtx);
if (--*this->m_cnt == 0) {
OMNI_D5_FW("ref count == 0, freeing pointers");
delete this->m_cnt;
delete this->m_ptr;
this->m_cnt = 0;
this->m_ptr = 0;
}
}
inline void _increment()
{
omni::sync::scoped_lock<omni::sync::mutex_t> alock(&this->m_mtx);
if (!this->m_cnt) { this->m_cnt = new unsigned long(0); }
++*this->m_cnt;
}
#if defined(OMNI_TYPE_INFO)
omni::type< omni::smart_ptr_safe<T> > m_type;
#endif
T* m_ptr;
unsigned long* m_cnt;
mutable omni::sync::mutex_t m_mtx;
};
}
#endif // OMNI_SMART_PTR_HPP