omp.cpp

00001 
00002 /* Copyright (c) 2007, Stefan Eilemann <eile@equalizergraphics.com> 
00003  *
00004  * This library is free software; you can redistribute it and/or modify it under
00005  * the terms of the GNU Lesser General Public License version 2.1 as published
00006  * by the Free Software Foundation.
00007  *  
00008  * This library is distributed in the hope that it will be useful, but WITHOUT
00009  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
00010  * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
00011  * details.
00012  * 
00013  * You should have received a copy of the GNU Lesser General Public License
00014  * along with this library; if not, write to the Free Software Foundation, Inc.,
00015  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
00016  */
00017 
00018 #include "omp.h"
00019 #include "debug.h"
00020 
00021 namespace eq
00022 {
00023 namespace base
00024 {
00025 static int _setupNThreads();
00026 
00027 int OMP::_nThreads = _setupNThreads();
00028 
00029 static int _setupNThreads()
00030 {
00031 #ifdef EQ_USE_OPENMP
00032     const char* nThreadsEnv = getenv( "OMP_NUM_THREADS" );
00033     int         nThreads    = 0;
00034 
00035     if( nThreadsEnv )
00036         nThreads = atoi( nThreadsEnv );
00037 
00038     if( nThreads < 1 )
00039         nThreads = omp_get_num_procs();
00040 
00041     EQASSERT( nThreads > 0 );
00042     omp_set_num_threads( nThreads );
00043 
00044     return nThreads;
00045 #else
00046     return 1;
00047 #endif
00048 }
00049 
00050 }
00051 }