gmOmp.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND,
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR
** A PARTICULAR PURPOSE.
**
************************************************************************/
 
#ifndef _GEMA_OPENMP_H_
#define _GEMA_OPENMP_H_
 
#ifdef _OPENMP
#include <omp.h>
#endif
 
#include <assert.h>
 
inline void GmOmpInit(int maxThreads)
{
  assert(maxThreads > 0);
#ifdef _OPENMP
  omp_set_num_threads(maxThreads);
  omp_set_dynamic(0);
#endif
}
 
inline int GmOmpAdjustNumThreads(int nt) 
{ 
#ifdef _OPENMP
  int mt = omp_get_max_threads();
  assert(mt > 0);
  return (nt > 0 && nt <= mt) ? nt : mt;
#else
  return 1;
#endif
}
 
inline int GmOmpThreadNum()
{
#ifdef _OPENMP
  return omp_get_thread_num();
#else
  return 0;
#endif
}
 
 
class GmOmpTempUpdateNumThreads
{
public:
  GmOmpTempUpdateNumThreads(int nthreads) { 
#ifdef _OPENMP
          _oldNumThreads = omp_get_max_threads(); omp_set_num_threads(nthreads);
#endif
  }
  ~GmOmpTempUpdateNumThreads() { 
#ifdef _OPENMP
          omp_set_num_threads(_oldNumThreads);
#endif
  }
private:
  int _oldNumThreads;
};
 
 
#ifdef _OPENMP
#define GM_OMP_ASSERT_NO_DYNAMIC assert(omp_get_dynamic() == 0)
#else
#define GM_OMP_ASSERT_NO_DYNAMIC ((void)0)
#endif
 
#endif