thread_local_storage.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2011 - 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------
 
#ifndef dealii_thread_local_storage_h
#  define dealii_thread_local_storage_h
 
 
#  include <deal.II/base/config.h>
 
DEAL_II_DISABLE_EXTRA_DIAGNOSTICS
#  ifdef DEAL_II_WITH_THREADS
#    include <tbb/enumerable_thread_specific.h>
#  endif
DEAL_II_ENABLE_EXTRA_DIAGNOSTICS
 
 
 
DEAL_II_NAMESPACE_OPEN
 
 
 
namespace Threads
{
  template <typename T>
  class ThreadLocalStorage
  {
  public:
    ThreadLocalStorage() = default;
 
    explicit ThreadLocalStorage(const T &t);
 
    ThreadLocalStorage(const ThreadLocalStorage<T> &t);
 
    T &
    get();
 
    T &
    get(bool &exists);
 
    operator T &();
 
    ThreadLocalStorage<T> &
    operator=(const T &t);
 
    void
    clear();
 
#  ifdef DEAL_II_WITH_THREADS
    tbb::enumerable_thread_specific<T> &
#  else
    T &
#  endif
    get_implementation();
 
  private:
#  ifdef DEAL_II_WITH_THREADS
 
    tbb::enumerable_thread_specific<T> data;
#  else
    T data;
#  endif
  };
 
  // ----------------- inline and template functions --------------------------
 
  template <typename T>
  inline ThreadLocalStorage<T>::ThreadLocalStorage(const T &t)
    : data(t)
  {}
 
 
  template <typename T>
  inline ThreadLocalStorage<T>::ThreadLocalStorage(
    const ThreadLocalStorage<T> &t)
    : data(t)
  {}
 
 
  template <typename T>
  inline T &
  ThreadLocalStorage<T>::get()
  {
#  ifdef DEAL_II_WITH_THREADS
    return data.local();
#  else
    return data;
#  endif
  }
 
 
  template <typename T>
  inline T &
  ThreadLocalStorage<T>::get(bool &exists)
  {
#  ifdef DEAL_II_WITH_THREADS
    return data.local(exists);
#  else
    exists = true;
    return data;
#  endif
  }
 
 
  template <typename T>
  inline ThreadLocalStorage<T>::operator T &()
  {
    return get();
  }
 
 
  template <typename T>
  inline ThreadLocalStorage<T> &
  ThreadLocalStorage<T>::operator=(const T &t)
  {
    get() = t;
    return *this;
  }
 
 
  template <typename T>
  inline
#  ifdef DEAL_II_WITH_THREADS
    tbb::enumerable_thread_specific<T> &
#  else
    T &
#  endif
    ThreadLocalStorage<T>::get_implementation()
  {
    return data;
  }
 
 
 
  template <typename T>
  inline void
  ThreadLocalStorage<T>::clear()
  {
#  ifdef DEAL_II_WITH_THREADS
    data.clear();
#  else
    data = T{};
#  endif
  }
} // namespace Threads
 
//---------------------------------------------------------------------------
DEAL_II_NAMESPACE_CLOSE
// end of #ifndef dealii_thread_local_storage_h
#endif
//---------------------------------------------------------------------------