reference_counting.h

Go to the documentation of this file.

/*******************************************************************\
 
Module: Reference Counting
 
Author: Daniel Kroening, kroening@kroening.com
 
\*******************************************************************/
 
 
#ifndef CPROVER_UTIL_REFERENCE_COUNTING_H
#define CPROVER_UTIL_REFERENCE_COUNTING_H
 
#include "invariant.h"
 
template <typename T, const T &empty = T::blank>
// NOLINTNEXTLINE(readability/identifiers)
class reference_counting
{
public:
  reference_counting():d(nullptr)
  {
  }
 
  explicit reference_counting(const T &other):d(NULL)
  {
    write()=other;
  }
 
  // copy constructor
  reference_counting(const reference_counting &other):d(other.d)
  {
    if(d!=nullptr)
    {
      PRECONDITION(d->ref_count != 0);
      d->ref_count++;
      #ifdef REFERENCE_COUNTING_DEBUG
      std::cout << "COPY " << d << " " << d->ref_count << '\n';
      #endif
    }
  }
 
  reference_counting &operator=(const reference_counting &other)
  {
    copy_from(other);
    return *this;
  }
 
  ~reference_counting()
  {
    remove_ref(d);
    d=nullptr;
  }
 
  void swap(reference_counting &other)
  {
    std::swap(other.d, d);
  }
 
  void clear()
  {
    remove_ref(d);
    d=nullptr;
  }
 
  const T &read() const
  {
    if(d==nullptr)
      return empty;
    return *d;
  }
 
  T &write()
  {
    detach();
    return *d;
  }
 
protected:
  class dt:public T
  {
  public:
    unsigned ref_count;
 
    dt():ref_count(1)
    {
    }
  };
 
  dt *d;
 
  void remove_ref(dt *old_d);
 
  void detach();
 
  void copy_from(const reference_counting &other)
  {
    PRECONDITION(&other != this); // check if we assign to ourselves
 
    #ifdef REFERENCE_COUNTING_DEBUG
    std::cout << "COPY " << (&other) << "\n";
    #endif
 
    remove_ref(d);
    d=other.d;
    if(d!=nullptr)
      d->ref_count++;
  }
 
public:
  dt *get_d() const
  {
    return d;
  }
};
 
template <class T, const T &empty>
void reference_counting<T, empty>::remove_ref(dt *old_d)
{
  if(old_d==nullptr)
    return;
 
  PRECONDITION(old_d->ref_count != 0);
 
  #ifdef REFERENCE_COUNTING_DEBUG
  std::cout << "R: " << old_d << " " << old_d->ref_count << '\n';
  #endif
 
  old_d->ref_count--;
  if(old_d->ref_count==0)
  {
    #ifdef REFERENCE_COUNTING_DEBUG
    std::cout << "DELETING " << old_d << '\n';
    old_d->clear();
    std::cout << "DEALLOCATING " << old_d << "\n";
    #endif
 
    delete old_d;
 
    #ifdef REFERENCE_COUNTING_DEBUG
    std::cout << "DONE\n";
    #endif
  }
}
 
template <class T, const T &empty>
void reference_counting<T, empty>::detach()
{
  #ifdef REFERENCE_COUNTING_DEBUG
  std::cout << "DETACH1: " << d << '\n';
  #endif
 
  if(d==nullptr)
  {
    d=new dt;
 
    #ifdef REFERENCE_COUNTING_DEBUG
    std::cout << "ALLOCATED " << d << '\n';
    #endif
  }
  else if(d->ref_count>1)
  {
    dt *old_d(d);
    d=new dt(*old_d);
 
    #ifdef REFERENCE_COUNTING_DEBUG
    std::cout << "ALLOCATED " << d << '\n';
    #endif
 
    d->ref_count=1;
    remove_ref(old_d);
  }
 
  POSTCONDITION(d->ref_count == 1);
 
  #ifdef REFERENCE_COUNTING_DEBUG
  std::cout << "DETACH2: " << d << '\n'
  #endif
}
 
template <class T, const T &empty>
bool operator==(
  const reference_counting<T, empty> &o1,
  const reference_counting<T, empty> &o2)
{
  if(o1.get_d()==o2.get_d())
    return true;
  return o1.read()==o2.read();
}
 
template <class T, const T &empty>
inline bool operator!=(
  const reference_counting<T, empty> &i1,
  const reference_counting<T, empty> &i2)
{ return !(i1==i2); }
 
#endif // CPROVER_UTIL_REFERENCE_COUNTING_H