SmartPtrBase.h

Go to the documentation of this file.

/********************************************************************
**                Image Component Library (ICL)                    **
**                                                                 **
** Copyright (C) 2006-2013 CITEC, University of Bielefeld          **
**                         Neuroinformatics Group                  **
** Website: www.iclcv.org and                                      **
**          http://opensource.cit-ec.de/projects/icl               **
**                                                                 **
** File   : ICLUtils/src/ICLUtils/SmartPtrBase.h                   **
** Module : ICLUtils                                               **
** Authors: Christof Elbrechter, Robert Haschke                    **
**                                                                 **
**                                                                 **
** GNU LESSER GENERAL PUBLIC LICENSE                               **
** This file may be used under the terms of the GNU Lesser General **
** Public License version 3.0 as published by the                  **
**                                                                 **
** Free Software Foundation and appearing in the file LICENSE.LGPL **
** included in the packaging of this file.  Please review the      **
** following information to ensure the license requirements will   **
** be met: http://www.gnu.org/licenses/lgpl-3.0.txt                **
**                                                                 **
** The development of this software was supported by the           **
** Excellence Cluster EXC 277 Cognitive Interaction Technology.    **
** The Excellence Cluster EXC 277 is a grant of the Deutsche       **
** Forschungsgemeinschaft (DFG) in the context of the German       **
** Excellence Initiative.                                          **
**                                                                 **
********************************************************************/

#pragma once

#include <ICLUtils/Macros.h>
#include <cstdlib>

//#define ICL_USE_STD_SHARED_POINTER_IF_POSSIBLE
#if __cplusplus >= 201103L
#ifdef ICL_USE_STD_SHARED_POINTER_IF_POSSIBLE
#define ICL_USE_STD_SHARED_PTR
#endif
#endif

#ifdef ICL_USE_STD_SHARED_PTR
#include <memory>
#endif

namespace icl{
  namespace utils{

    class DelOpBase { public: virtual ~DelOpBase(){} };

    struct PointerDelOp : public DelOpBase{ template<class T> static void delete_func(T *t){ delete t; } };

    struct ArrayDelOp : public DelOpBase{ template<class T>  static void delete_func(T *t){ delete[] t; } };

    struct FreeDelOp : public DelOpBase{ static void delete_func(void *v){ free(v); } };


#ifdef ICL_USE_STD_SHARED_PTR
    //#warning "Using new std::shared_ptr - based SmartPtr class "
    template<class T, class delOp = ArrayDelOp >
    class SmartPtrBase{

      std::shared_ptr<T> impl;

      template<class DerivedT>
      inline SmartPtrBase<T,delOp> &assign(const SmartPtrBase<DerivedT,delOp>& r){
        impl = r.impl;
        return *this;
      }

      public:
      template<class A, class B> friend class SmartPtrBase;

      inline SmartPtrBase(){}

      template<class DerivedT>
      SmartPtrBase(DerivedT *ptData, bool bOwn=true) :
      impl(ptData, [bOwn](T *t){ if(bOwn) delOp::delete_func(t); }){
      }


      template<class DerivedT>
      SmartPtrBase(const SmartPtrBase<DerivedT,delOp> &r) :
        impl(std::static_pointer_cast<T,DerivedT>(r.impl)){
      }


      SmartPtrBase(const SmartPtrBase<T,delOp> &r): impl(r.impl){}


      template<class DerivedT>
      SmartPtrBase<T,delOp> &operator=(const SmartPtrBase<DerivedT,delOp>& r){
        return assign(r);
      }


      SmartPtrBase<T,delOp> &operator=(const SmartPtrBase<T,delOp>& r){
        return assign(r);
      }


      template<class DerivedT>
      SmartPtrBase<T,delOp> &operator=(DerivedT *p){
        return this->operator=(SmartPtrBase<T,delOp>(p));
      }

      SmartPtrBase<T,delOp> &operator=(T *p){
        return this->operator=(SmartPtrBase<T,delOp>(p));
      }

      virtual ~SmartPtrBase() {}


      T &operator* () { return *impl; }


      const T &operator* () const { return *impl; }



      T* get () { return impl.get(); }


      const T* get () const { return impl.get(); }



      T *operator-> () { return impl.get(); }


      const T *operator-> () const { return impl.get(); }


      operator bool() const { return (bool)impl; }

      int use_count() const { return impl.use_count(); }


      void setNull() {
        impl.reset();
      }
    };
#else



    template<class T, class delOp = ArrayDelOp >
    class SmartPtrBase{
      protected:
      T   *e; 
      int *c; 
      bool d; 
      void inc() {
        if(c) (*c)++;
      }

      void dec() {
        if(c && *c) {
          if ( --(*c) == 0) {
            //               if(d) delete[] e;
            if(d) delOp::delete_func(e);
            delete c;
            c = 0;
          }
        }
      }

      void set(T *e,int *c, bool d) {
        this->e=e;
        this->c=c;
        this->d=d;
      }

      template<class DerivedT>
      inline SmartPtrBase<T,delOp> &assign(const SmartPtrBase<DerivedT,delOp>& r){
        if(r.e == e) return *this;
        dec();
        set(r.e,r.c,r.d);
        inc();
        return *this;
      }

      public:

      template<class A, class B> friend class SmartPtrBase;

      SmartPtrBase(): e(0),c(0),d(0){
      }

      template<class DerivedT>
      SmartPtrBase(DerivedT *ptData, bool bOwn=true): e(ptData), c(new int(1)),d(bOwn){
      }


      template<class DerivedT>
      SmartPtrBase(const SmartPtrBase<DerivedT,delOp> &r): e(&dynamic_cast<T&>(*r.e)), c(r.c), d(r.d){
        //      SmartPtrBase(const SmartPtrBase<DerivedT,delOp> &r): e(r.e), c(r.c), d(r.d){
        inc();
      }


      SmartPtrBase(const SmartPtrBase<T,delOp> &r): e(r.e), c(r.c), d(r.d){
        inc();
      }


      template<class DerivedT>
      SmartPtrBase<T,delOp> &operator=(const SmartPtrBase<DerivedT,delOp>& r){
        return assign(r);

      }


      SmartPtrBase<T,delOp> &operator=(const SmartPtrBase<T,delOp>& r){
        return assign(r);
      }


      template<class DerivedT>
      SmartPtrBase<T,delOp> &operator=(DerivedT *p){
        return this->operator=(SmartPtrBase<T,delOp>(p));
      }

      SmartPtrBase<T,delOp> &operator=(T *p){
        return this->operator=(SmartPtrBase<T,delOp>(p));
      }

      virtual  ~SmartPtrBase() { dec(); }


      T &operator* () { ICLASSERT(e); return *e; }


      const T &operator* () const { ICLASSERT(e); return *e; }



      T* get () { return e; }


      const T* get () const { return e; }



      T *operator-> () { ICLASSERT(e); return e; }


      const T *operator-> () const { ICLASSERT(e); return e; }


      operator bool() const { return (e != 0); }

      int use_count() const { return c ? *c : 0; }


      void setNull() {
        dec();
        set(0,0,0);
      }
    };
#endif
  } // namespace utils
}