The SPtr is castable now. just say SPtrCast<TypeToCastTo>( ASPtr );

1 files changed, 15 insertions, 0 deletions

diff --git a/src/sptr.h b/src/sptr.h
index 23c21dc..0ea8bca 100644
--- a/src/sptr.h
+++ b/src/sptr.h
@@ -4,9 +4,14 @@
 #include <stdint.h>
 #include <stdio.h>
 
+template<typename T> class SPtr;
+template< typename Tb, typename Ta > SPtr<Tb> SPtrCast( SPtr<Ta> src );
+
 template<typename T>
 class SPtr
 {
+        template<typename Tb, typename Ta>
+        friend SPtr<Tb> SPtrCast( SPtr<Ta> pt );
 public:
         SPtr() :
                 pRefCnt( NULL ),
@@ -96,4 +101,14 @@ private:
         T *pData;
 };
 
+template< typename Tb, typename Ta > SPtr<Tb> SPtrCast( SPtr<Ta> src )
+{
+        SPtr<Tb> ret;
+        ret.pRefCnt = src.pRefCnt;
+        ret.pData = dynamic_cast<Tb *>(src.pData);
+        if( ret.pRefCnt )
+                (*(ret.pRefCnt)) += 1;
+        return ret;
+}
+
 #endif