SmartPtr.hpp

Go to the documentation of this file.

// This file is part of GranOO, a workbench for DEM simulation.
//
// Author(s)    : - Damien Andre       IRCER/UNILIM, Limoges France
//                  <damien.andre@unilim.fr>
//                - Jean-luc Charles   Arts et Metiers ParisTech, CNRS, I2M, Bordeaux France
//                  <jean-luc.charles@ensam.eu>
//                - Jeremie Girardot   Arts et Metiers ParisTech, CNRS, I2M, Bordeaux France
//                  <jeremie.girardot@ensam.eu>
//                - Cedric Hubert      LAMIH/UPHF, Valenciennes France
//                  <cedric.hubert@uphf.fr>
//                - Ivan Iordanoff     Arts et Metiers ParisTech, CNRS, I2M, Bordeaux France
//                  <ivan.iordanoff@ensam.eu>
//
// Copyright (C) 2008-2019 D. Andre, JL. Charles, J. Girardot, C. Hubert, I. Iordanoff
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 
#ifndef _libDEM_SmartPtr_hpp_
#define _libDEM_SmartPtr_hpp_
 
#include <set>
 
#include "GranOO3/Core/Macro.hpp"
 
namespace GranOO3
{
 
  namespace Core
  {
 
    template<typename type> class SmartPtr;
 
    template<typename type>
    class SmartPtrClass
    {
    public:
      SmartPtrClass();
      virtual ~SmartPtrClass();
 
      void add_ptr(SmartPtr<type>*);
      void erase_ptr(SmartPtr<type>*);
 
    private:
      std::set<SmartPtr<type>*> allPtr_;
    };
 
 
    template<typename type>
    class SmartPtr
    {
      friend class SmartPtrClass<type>;
 
    public:
      SmartPtr();
      SmartPtr(type*);
      SmartPtr(const SmartPtr<type>&);
 
      SmartPtr<type>& operator= ( const SmartPtr<type>&) = delete;
 
      ~SmartPtr();
      void attach_ptr(type*);
      void delete_ptr();
 
    private:
      type* ptr_;
    };
 
 
    template<typename type>
    SmartPtrClass<type>::SmartPtrClass()
      : allPtr_() {
    }
 
    template<typename type>
    SmartPtrClass<type>::~SmartPtrClass() {
      for (auto pt: allPtr_)
        pt->ptr_ = nullptr;
    }
 
    template<typename type>
    void
    SmartPtrClass<type>::add_ptr(SmartPtr<type>* pt) {
      SafeModeAssert(allPtr_.count(pt)==0, "");
      allPtr_.insert(pt);
    }
 
    template<typename type>
    void
    SmartPtrClass<type>::erase_ptr(SmartPtr<type>* pt) {
      SafeModeAssert(allPtr_.count(pt)==1, "");
      allPtr_.erase(pt);
    }
 
    template<typename type>
    SmartPtr<type>::SmartPtr()
      : ptr_(nullptr) {
    }
 
    template<typename type>
    SmartPtr<type>::SmartPtr(type* pt)
      : ptr_(nullptr) {
      attach_ptr(pt);
    }
 
    template<typename type>
    SmartPtr<type>::SmartPtr(const SmartPtr<type>& pt)
      : ptr_(nullptr) {
      if (pt.ptr_ != nullptr)
        attach_ptr(pt.ptr_);
    }
 
    template<typename type>
    SmartPtr<type>::~SmartPtr() {
      if (ptr_ != nullptr)
        static_cast<SmartPtrClass<type>*>(ptr_)->erase_ptr(this);
 
    }
 
    template<typename type>
    void
    SmartPtr<type>::attach_ptr(type* pt) {
      ptr_ = pt;
      static_cast<SmartPtrClass<type>*>(ptr_)->add_ptr(this);
    }
 
    template<typename type>
    void
    SmartPtr<type>::delete_ptr() {
      if (ptr_ != nullptr) {
        static_cast<SmartPtrClass<type>*>(ptr_)->erase_ptr(this);
        delete ptr_;
        ptr_ = nullptr;
      }
    }
 
  }
}
 
 
 
 
#endif