Added a simple smart-pointer. It deletes what's in it automatically and does

ref-counting.  You just have to be careful to not hand it an array...

1 files changed, 85 insertions, 0 deletions

diff --git a/src/sptr.h b/src/sptr.h
new file mode 100644
index 0000000..c711a89
--- /dev/null
+++ b/src/sptr.h
@@ -0,0 +1,85 @@
+#ifndef SPTR_H
+#define SPTR_H
+
+#include <stdint.h>
+#include <stdio.h>
+
+template<typename T>
+class SPtr
+{
+public:
+        SPtr() :
+                pRefCnt( NULL ),
+                pData( NULL )
+        {
+        }
+
+        ~SPtr()
+        {
+                decCount();
+        }
+
+        SPtr( SPtr<T> &src ) :
+                pRefCnt( src.pRefCnt ),
+                pData( src.pData )
+        {
+                (*pRefCnt) += 1;
+        }
+
+        SPtr( T *pSrc ) :
+                pRefCnt( NULL ),
+                pData( pSrc )
+        {
+                pRefCnt = new uint32_t;
+                (*pRefCnt) = 1;
+        }
+
+        uint32_t count()
+        {
+                return *pRefCnt;
+        }
+
+        T *operator->() const
+        {
+                return pData;
+        }
+
+        T *operator*() const
+        {
+                return pData;
+        }
+
+        SPtr<T> operator=( const SPtr<T> &src )
+        {
+                decCount();
+                pRefCnt = src.pRefCnt;
+                pData = src.pData;
+                (*pRefCnt) += 1;
+        }
+
+        bool operator==( const SPtr<T> &src )
+        {
+                return pData == src.pData;
+        }
+
+private:
+        void decCount()
+        {
+                if( pRefCnt )
+                {
+                        (*pRefCnt) -= 1;
+                        if( (*pRefCnt) == 0 )
+                        {
+                                delete pRefCnt;
+                                delete pData;
+                                pRefCnt = NULL;
+                                pData = NULL;
+                        }
+                }
+        }
+
+        uint32_t *pRefCnt;
+        T *pData;
+};
+
+#endif