threadsafevalue.hh

#ifndef DUNE_FEM_THREADSAFEVALUES_HH
#define DUNE_FEM_THREADSAFEVALUES_HH
 
#include <vector>
 
#include <dune/fem/misc/mpimanager.hh>
 
namespace Dune {
 
  namespace Fem {
 
 
    template <class T>
    class ThreadSafeValue
    {
#ifdef USE_SMP_PARALLEL
      std::vector< T > value_;
#else
      T value_;
#endif
    public:
      typedef T ValueType ;
 
      template< class ...Args >
      ThreadSafeValue( Args&& ...args )
#ifdef USE_SMP_PARALLEL
        : value_( MPIManager::maxThreads(), ValueType( std::forward< Args >( args )... ) )
#else
        : value_( std::forward< Args >( args )... )
#endif
      {}
 
      ThreadSafeValue()
        : value_(
#ifdef USE_SMP_PARALLEL
            MPIManager::maxThreads()
#endif
             )
      {}
 
      size_t size() const { return MPIManager::maxThreads(); }
 
      ValueType& operator * () { return this->operator[]( MPIManager::thread() ); }
      const ValueType& operator * () const { return this->operator[]( MPIManager::thread() ); }
 
      operator const ValueType& () const { return this->operator[]( MPIManager::thread() ); }
      operator ValueType& () { return this->operator[]( MPIManager::thread() ); }
 
      ValueType& operator [] ( const unsigned int thread ) {
        assert( thread < size() );
#ifdef USE_SMP_PARALLEL
        assert( thread < value_.size() );
#endif
        return value_
#ifdef USE_SMP_PARALLEL
          [ thread ]
#endif
          ;
      }
 
      const ValueType& operator [] ( const unsigned int thread ) const {
        assert( thread < size() );
#ifdef USE_SMP_PARALLEL
        assert( thread < value_.size() );
#endif
        return value_
#ifdef USE_SMP_PARALLEL
          [ thread ]
#endif
          ;
      }
    };
 
  } // end namespace Fem
 
} // end namespace Dune
 
 
#endif