(C) Zeriph Enterprises, LLC
omni/smart_ptr.hpp
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:









































































































































































































































































































































/*
* Copyright (c), Zeriph Enterprises
* All rights reserved.
*
* Contributor(s):
* Zechariah Perez, omni (at) zeriph (dot) com
*
* 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:
            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 uint64_t(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();
                    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 uint64_t(0); }
                ++*this->m_cnt;
            }
            
            T* m_ptr;
            uint64_t* m_cnt;
    };
    
    template < typename T >
    class smart_ptr_safe
    {
        public:
            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 uint64_t(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);
                    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 uint64_t(0); }
                ++*this->m_cnt;
            }
            
            T* m_ptr;
            uint64_t* m_cnt;
            mutable omni::sync::mutex_t m_mtx;
    };
}

#endif // OMNI_SMART_PTR_HPP