gsThreaded.h

Go to the documentation of this file.

#pragma once

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

namespace gismo
{

namespace util
{

// Usage:
// gsThreaded<C> a;
// a.mine();
template<class C, class Allocator = std::allocator<C> >
class gsThreaded
{
#ifdef _OPENMP
    std::vector<C,Allocator> m_array;
    #else
    C m_c;
#endif

public:

#ifdef _OPENMP
    gsThreaded() : m_array(omp_get_max_threads()) { }

    operator C&()             { return m_array[omp_get_thread_num()]; }
    operator const C&() const { return m_array[omp_get_thread_num()]; }

    C&       mine() { return m_array[omp_get_thread_num()]; }
    const C& mine() const { return m_array[omp_get_thread_num()]; }

    C& operator = (C other) { return m_array[omp_get_thread_num()] = give(other); }
#else
    operator C&()             { return m_c; }
    operator const C&() const { return m_c; }
    
    C&       mine() { return m_c; }
    const C& mine() const { return m_c; }
    
    C& operator = (C other) { return m_c = give(other); }
#endif
    
};//gsThreaded

}//util

}//gismo