libs/pbd/cpus.cc

   1 /*
   2   Copyright (C) 2010 Paul Davis
   3   Author: Torben Hohn
   4
   5   This program is free software; you can redistribute it and/or modify
   6   it under the terms of the GNU General Public License as published by
   7   the Free Software Foundation; either version 2 of the License, or
   8   (at your option) any later version.
   9
  10   This program is distributed in the hope that it will be useful,
  11   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13   GNU General Public License for more details.
  14
  15   You should have received a copy of the GNU General Public License
  16   along with this program; if not, write to the Free Software
  17   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18
  19 */
  20
  21 #ifdef WAF_BUILD
  22 #include "libpbd-config.h"
  23 #endif
  24
  25 #include <stdlib.h>
  26
  27 #ifdef __linux__
  28 #include <unistd.h>
  29 #elif defined(__APPLE__) || defined(__FreeBSD__)
  30 #include <stddef.h>
  31 #include <sys/types.h>
  32 #include <sys/sysctl.h>
  33 #elif defined(PLATFORM_WINDOWS)
  34 #include <windows.h>
  35 #endif
  36
  37 #include "pbd/cpus.h"
  38
  39 #if defined(COMPILER_MSVC) && !defined(PTW32_VERSION)
  40 #include <ardourext/pthread.h>  // Gets us 'PTW32_VERSION'
  41 #endif
  42
  43 uint32_t
  44 hardware_concurrency()
  45 {
  46         if (getenv("CONCURRENCY")) {
  47                 int c = atoi (getenv("CONCURRENCY"));
  48                 if (c > 0) {
  49                         return c;
  50                 }
  51         }
  52 #if defined(PTW32_VERSION) || defined(__hpux)
  53         return pthread_num_processors_np();
  54 #elif defined(__APPLE__)
  55         int count;
  56         size_t size=sizeof(count);
  57         return sysctlbyname("hw.physicalcpu",&count,&size,NULL,0)?0:count;
  58 #elif defined(__FreeBSD__)
  59         int count;
  60         size_t size=sizeof(count);
  61         return sysctlbyname("hw.ncpu",&count,&size,NULL,0)?0:count;
  62 #elif defined(HAVE_UNISTD) && defined(_SC_NPROCESSORS_ONLN)
  63         int const count=sysconf(_SC_NPROCESSORS_ONLN);
  64         return (count>0)?count:0;
  65 #elif defined(PLATFORM_WINDOWS)
  66         SYSTEM_INFO sys_info;
  67         GetSystemInfo( &sys_info );
  68         return sys_info.dwNumberOfProcessors;
  69 #else
  70         return 0;
  71 #endif
  72 }