ThreadVec.H

Go to the documentation of this file.

 
#ifndef THREADVEC_H_
#define THREADVEC_H_
 
#include <vector>
#ifdef _OPENMP
#   include <omp.h>
#endif
#include "Types.H"
 
namespace criptic {
 
  template <class T> class ThreadVec {
 
  public:
 
    ThreadVec() {
#ifdef _OPENMP
#pragma omp parallel
      {
    const int nthreads = omp_get_num_threads();
#pragma omp single
    obj.resize(nthreads);
      }
#else
      obj.resize(1);
#endif
    }
 
    IdxType size() const { return obj.size(); }
    
    T& operator()() {
#ifdef _OPENMP
      const int ithread = omp_get_thread_num();
#else
      const int ithread = 0;
#endif
      return obj[ithread];
    }
 
    const T& operator()() const {
#ifdef _OPENMP
      const int ithread = omp_get_thread_num();
#else
      const int ithread = 0;
#endif
      return obj[ithread];
    }
 
    T& operator[] (const int i) {
      return obj[i];
    }
 
    const T& operator[] (const int i) const {
      return obj[i];
    }
    
  private:
 
    std::vector<T> obj;  
  };
 
}
 
#endif
// _THREADVEC_H_