(C) Zeriph Enterprises, LLC
omni/smart_ptr_safe.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:
/*
* 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_SAFE_HPP
#define OMNI_SMART_PTR_SAFE_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_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_DTOR_FW
                this->_decrement();
                omni::sync::mutex_destroy(this->m_mtx);
                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_SAFE_HPP