FastThreadable.h

#ifndef _RAVELIN_FAST_THREADABLE_H_
#define _RAVELIN_FAST_THREADABLE_H_

#ifdef _OPENMP
#include <omp.h>
#include <vector>
#endif

namespace Ravelin {

template <class T>
class FastThreadable
{
  private:
    #ifdef _OPENMP
    std::vector<T> _x;
    #else
    T _x;
    #endif

  public:
    FastThreadable()
    {
      #ifdef _OPENMP
      _x.resize(omp_get_max_threads());
      #endif
    }

    T& operator()()
    {
      #ifdef _OPENMP
      return _x[omp_get_thread_num()];
      #else
      return _x;
      #endif
    }
};

} // end namespace

#endif