shared_ptr.hpp

Go to the documentation of this file.

#ifndef VIENNACL_TOOLS_SHARED_PTR_HPP
#define VIENNACL_TOOLS_SHARED_PTR_HPP

/* =========================================================================
   Copyright (c) 2010-2016, Institute for Microelectronics,
                            Institute for Analysis and Scientific Computing,
                            TU Wien.
   Portions of this software are copyright by UChicago Argonne, LLC.

                            -----------------
                  ViennaCL - The Vienna Computing Library
                            -----------------

   Project Head:    Karl Rupp                   rupp@iue.tuwien.ac.at

   (A list of authors and contributors can be found in the manual)

   License:         MIT (X11), see file LICENSE in the base directory
============================================================================= */

#include <cstdlib>
#include <algorithm>

namespace viennacl
{
namespace tools
{

namespace detail
{

  class count
  {
  public:
    count(unsigned int val) : val_(val){ }
    void dec(){ --val_; }
    void inc(){ ++val_; }
    bool is_null(){ return val_ == 0; }
    unsigned int val(){ return val_; }
  private:
    unsigned int val_;
  };

  struct aux
  {
    detail::count count;

    aux() :count(1) {}
    virtual void destroy()=0;
    virtual ~aux() {}
  };

  template<class U, class Deleter>
  struct auximpl: public detail::aux
  {
    U* p;
    Deleter d;

    auximpl(U* pu, Deleter x) :p(pu), d(x) {}
    virtual void destroy() { d(p); }
  };

  template<class U>
  struct default_deleter
  {
    void operator()(U* p) const { delete p; }
  };

}

template<class T>
class shared_ptr
{
  template<class U>
  friend class shared_ptr;

  detail::aux* pa;
  T* pt;

public:

  shared_ptr() :pa(NULL), pt(NULL) {}

  template<class U, class Deleter>
  shared_ptr(U* pu, Deleter d) : pa(new detail::auximpl<U, Deleter>(pu, d)), pt(pu) {}

  template<class U>
  explicit shared_ptr(U* pu) : pa(new detail::auximpl<U, detail::default_deleter<U> >(pu, detail::default_deleter<U>())), pt(pu) {}

  T* get() const {  return pt; }

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

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

  shared_ptr(const shared_ptr& s) :pa(s.pa), pt(s.pt)
  {
    inc();
  }

  template<class U>
  shared_ptr(const shared_ptr<U>& s) :pa(s.pa), pt(s.pt)
  {
    inc();
  }

  ~shared_ptr()
  {
    dec();
  }

  void reset()
  {
    shared_ptr<T>().swap(*this);
  }

  void reset(T * ptr)
  {
    shared_ptr<T>(ptr).swap(*this);
  }

  void swap(shared_ptr<T> & other)
  {
    std::swap(pt,other.pt);
    std::swap(pa, other.pa);
  }


  shared_ptr& operator=(const shared_ptr& s)
  {
    if (this!=&s)
    {
      dec();
      pa = s.pa;
      pt = s.pt;
      inc();
    }
    return *this;
  }



  void inc()
  {
    if (pa) pa->count.inc();
  }

  void dec()
  {
    if (pa)
    {
      pa->count.dec();

      if (pa->count.is_null())
      {
        pa->destroy();
        delete pa;
        pa = NULL;
      }
    }
  }

};

}
}

#endif // VIENNACL_UTILS_SHARED_PTR_HPP