RefCntDynObjPtr.hpp

#ifndef REFCNTDYNOBJPTR_HPP
#define REFCNTDYNOBJPTR_HPP

#include "DoBase.hpp" 
#include "DoError.h" 

template<class T>
struct RefCntDynObjPtr { 
    explicit RefCntDynObjPtr( T* pt=NULL, bool do_add_ref=true ) : m_pt(0) {
        if( !pt ) return;
        DoAssign( pt, do_add_ref );
    }
    
    RefCntDynObjPtr& operator = (T* pt){
        DoRelease();
        DoAssign( pt, true );
        return *this;
    }
    
    RefCntDynObjPtr& operator = (const RefCntDynObjPtr<T>& dop){
        return operator = ((T*)dop);
    }
    
    // Smart pointer access
    operator T* () { return m_pt; }
    operator T& () { return *m_pt; }
    T& operator * () { return *m_pt; }
    T* operator -> () { return m_pt; }
    T* Get(){ return m_pt; } 

    bool IsOk(){ return m_pt!=NULL; } 

    bool Release(){ T*pt=m_pt; DoRelease(); return pt && !m_pt; } 

protected:
    bool DoAssign( T* pt, bool do_add_ref ){
        RefCountI *prci = DoCasterSelect<T,RefCountI,DoIsTypeA<T,RefCountI>::v>::Cast(pt,NULL);
        if( !prci ) return false;
        m_pt = pt;
        if( do_add_ref )
            prci->IncRef();
        return true;
    }
    
    void DoRelease(){
        if( !m_pt ) return;
        RefCountI *prci = DoCasterSelect<T,RefCountI,DoIsTypeA<T,RefCountI>::v>::Cast(m_pt,NULL);
        DO_ASSERT_MSG( prci, "Should get RefCountI" );
        if( prci )
            prci->DecRef();
        m_pt = NULL;
    }
    
    T* m_pt;
};

#endif // REFCNTDYNOBJPTR_HPP

---