diff --git a/src/os/macosx/macos.mm b/src/os/macosx/macos.mm index f6b2d4e..7c50924 100644 --- a/src/os/macosx/macos.mm +++ b/src/os/macosx/macos.mm @@ -172,3 +172,20 @@ bool GetClipboardContents(char *buffer, size_t buff_len) return true; } #endif + +uint GetCPUCoreCount() +{ + uint count = 1; +#if (MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_5) + if (MacOSVersionIsAtLeast(10, 5, 0)) { + count = [ [ NSProcessInfo processInfo ] activeProcessorCount ]; + } else +#endif + { +#if (MAC_OS_X_VERSION_MIN_REQUIRED < MAC_OS_X_VERSION_10_5) + count = MPProcessorsScheduled(); +#endif + } + + return count; +} diff --git a/src/os/os2/os2.cpp b/src/os/os2/os2.cpp index bef44d4..21fd0c7 100644 --- a/src/os/os2/os2.cpp +++ b/src/os/os2/os2.cpp @@ -210,3 +210,8 @@ void CSleep(int milliseconds) const char *FS2OTTD(const char *name) {return name;} const char *OTTD2FS(const char *name) {return name;} + +uint GetCPUCoreCount() +{ + return 1; +} diff --git a/src/os/unix/unix.cpp b/src/os/unix/unix.cpp index 371beb2..700a905 100644 --- a/src/os/unix/unix.cpp +++ b/src/os/unix/unix.cpp @@ -28,10 +28,18 @@ #define HAS_STATVFS #endif +#if defined(OPENBSD) || defined(__NetBSD__) || defined(__FreeBSD__) + #define HAS_SYSCTL +#endif + #ifdef HAS_STATVFS #include #endif +#ifdef HAS_SYSCTL +#include +#endif + #ifdef __MORPHOS__ #include @@ -318,3 +326,26 @@ void CSleep(int milliseconds) usleep(milliseconds * 1000); #endif } + + +#ifndef __APPLE__ +uint GetCPUCoreCount() +{ + uint count = 1; +#ifdef HAS_SYSCTL + int ncpu = 0; + size_t len = sizeof(ncpu); + + if (sysctlbyname("hw.availcpu", &ncpu, &len, NULL, 0) < 0) { + sysctlbyname("hw.ncpu", &ncpu, &len, NULL, 0); + } + + if (ncpu > 0) count = ncpu; +#elif defined(_SC_NPROCESSORS_ONLN) + long res = sysconf(_SC_NPROCESSORS_ONLN); + if (res > 0) count = res; +#endif + + return count; +} +#endif diff --git a/src/os/windows/win32.cpp b/src/os/windows/win32.cpp index 39203cb..6c7a3ec 100644 --- a/src/os/windows/win32.cpp +++ b/src/os/windows/win32.cpp @@ -753,3 +753,11 @@ HRESULT OTTDSHGetFolderPath(HWND hwnd, int csidl, HANDLE hToken, DWORD dwFlags, static char retbuf[6] = {lang[0], lang[1], '_', country[0], country[1], 0}; return retbuf; } + +uint GetCPUCoreCount() +{ + SYSTEM_INFO info; + + GetSystemInfo(&info); + return info.dwNumberOfProcessors; +} diff --git a/src/thread/thread.h b/src/thread/thread.h index 83eeb73..26d209d 100644 --- a/src/thread/thread.h +++ b/src/thread/thread.h @@ -88,4 +88,10 @@ class ThreadMutex { virtual void SendSignal() = 0; }; +/** + * Get number of processor cores in the system, including HyperThreading or similar. + * @return Total number of processor cores. + */ +uint GetCPUCoreCount(); + #endif /* THREAD_H */