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...

3 files changed, 141 insertions, 0 deletions

diff --git a/src/sptr.cpp b/src/sptr.cpp
new file mode 100644
index 0000000..7f5e894
--- /dev/null
+++ b/src/sptr.cpp
@@ -0,0 +1 @@
+#include "sptr.h"
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
diff --git a/src/tests/sptr.cpp b/src/tests/sptr.cpp
new file mode 100644
index 0000000..38d3675
--- /dev/null
+++ b/src/tests/sptr.cpp
@@ -0,0 +1,55 @@
+#include <stdio.h>
+#include "sptr.h"
+
+class Annoy
+{
+public:
+        Annoy() : nCnt( 0 )
+        {
+                printf("Created.\n");
+        }
+
+        ~Annoy()
+        {
+                printf("Destroyed.\n");
+        }
+
+        void go()
+        {
+                printf("%d: I'm annoying.\n", ++nCnt);
+        }
+
+        int nCnt;
+};
+
+void beAnnoying( SPtr<Annoy> bob )
+{
+        printf("bob-Count: %d\n", bob.count() );
+        bob->go();
+}
+
+int main()
+{
+        SPtr<Annoy> pt( new Annoy );
+        printf("Count: %d\n", pt.count() );
+        pt->go();
+
+        {
+                SPtr<Annoy> pt2 = pt;
+                printf("Count: %d\n", pt2.count() );
+
+                pt2->go();
+
+                {
+                        SPtr<Annoy> pt3( pt2 );
+                        printf("Count: %d\n", pt3.count() );
+
+                        pt3->go();
+
+                        beAnnoying( pt3 );
+                }
+                printf("Count: %d\n", pt.count() );
+        }
+        printf("Count: %d\n", pt.count() );
+}
+