2014-12-17 00:38:14 -05:00
|
|
|
// Copyright 2013 Dolphin Emulator Project / 2014 Citra Emulator Project
|
|
|
|
// Licensed under GPLv2 or any later version
|
2013-09-04 20:17:46 -04:00
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
2014-04-08 20:15:08 -04:00
|
|
|
#include "common/thread.h"
|
2013-09-04 20:17:46 -04:00
|
|
|
#ifdef __APPLE__
|
2016-09-17 20:38:01 -04:00
|
|
|
#include <mach/mach.h>
|
2014-09-03 01:05:45 -04:00
|
|
|
#elif defined(_WIN32)
|
2016-12-03 15:08:02 -05:00
|
|
|
#include <windows.h>
|
2015-06-20 17:45:15 -04:00
|
|
|
#else
|
2016-10-09 22:25:19 -04:00
|
|
|
#if defined(__Bitrig__) || defined(__DragonFly__) || defined(__FreeBSD__) || defined(__OpenBSD__)
|
2016-09-17 20:38:01 -04:00
|
|
|
#include <pthread_np.h>
|
|
|
|
#else
|
|
|
|
#include <pthread.h>
|
|
|
|
#endif
|
|
|
|
#include <sched.h>
|
2015-06-20 17:45:15 -04:00
|
|
|
#endif
|
|
|
|
#ifndef _WIN32
|
2016-09-17 20:38:01 -04:00
|
|
|
#include <unistd.h>
|
2013-09-04 20:17:46 -04:00
|
|
|
#endif
|
|
|
|
|
2016-10-09 22:33:41 -04:00
|
|
|
#ifdef __FreeBSD__
|
|
|
|
#define cpu_set_t cpuset_t
|
|
|
|
#endif
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
namespace Common {
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2014-12-29 22:59:14 -05:00
|
|
|
#ifdef _MSC_VER
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetThreadAffinity(std::thread::native_handle_type thread, u32 mask) {
|
2014-04-01 18:20:08 -04:00
|
|
|
SetThreadAffinityMask(thread, mask);
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetCurrentThreadAffinity(u32 mask) {
|
2014-04-01 18:20:08 -04:00
|
|
|
SetThreadAffinityMask(GetCurrentThread(), mask);
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
void SwitchCurrentThread() {
|
2014-04-01 18:20:08 -04:00
|
|
|
SwitchToThread();
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
// Sets the debugger-visible name of the current thread.
|
|
|
|
// Uses undocumented (actually, it is now documented) trick.
|
|
|
|
// http://msdn.microsoft.com/library/default.asp?url=/library/en-us/vsdebug/html/vxtsksettingthreadname.asp
|
2014-11-19 03:49:13 -05:00
|
|
|
|
2013-09-04 20:17:46 -04:00
|
|
|
// This is implemented much nicer in upcoming msvc++, see:
|
|
|
|
// http://msdn.microsoft.com/en-us/library/xcb2z8hs(VS.100).aspx
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetCurrentThreadName(const char* szThreadName) {
|
2014-04-01 18:20:08 -04:00
|
|
|
static const DWORD MS_VC_EXCEPTION = 0x406D1388;
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
#pragma pack(push, 8)
|
|
|
|
struct THREADNAME_INFO {
|
|
|
|
DWORD dwType; // must be 0x1000
|
|
|
|
LPCSTR szName; // pointer to name (in user addr space)
|
2014-04-01 18:20:08 -04:00
|
|
|
DWORD dwThreadID; // thread ID (-1=caller thread)
|
2016-09-17 20:38:01 -04:00
|
|
|
DWORD dwFlags; // reserved for future use, must be zero
|
2014-04-01 18:20:08 -04:00
|
|
|
} info;
|
2016-09-17 20:38:01 -04:00
|
|
|
#pragma pack(pop)
|
2014-04-01 18:20:08 -04:00
|
|
|
|
|
|
|
info.dwType = 0x1000;
|
|
|
|
info.szName = szThreadName;
|
2016-09-17 20:38:01 -04:00
|
|
|
info.dwThreadID = -1; // dwThreadID;
|
2014-04-01 18:20:08 -04:00
|
|
|
info.dwFlags = 0;
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
__try {
|
|
|
|
RaiseException(MS_VC_EXCEPTION, 0, sizeof(info) / sizeof(ULONG_PTR), (ULONG_PTR*)&info);
|
|
|
|
} __except (EXCEPTION_CONTINUE_EXECUTION) {
|
2014-04-01 18:20:08 -04:00
|
|
|
}
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
2014-11-19 03:49:13 -05:00
|
|
|
|
2014-11-29 00:38:20 -05:00
|
|
|
#else // !MSVC_VER, so must be POSIX threads
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetThreadAffinity(std::thread::native_handle_type thread, u32 mask) {
|
2013-09-04 20:17:46 -04:00
|
|
|
#ifdef __APPLE__
|
2016-09-17 20:38:01 -04:00
|
|
|
thread_policy_set(pthread_mach_thread_np(thread), THREAD_AFFINITY_POLICY, (integer_t*)&mask, 1);
|
2016-10-09 22:33:41 -04:00
|
|
|
#elif (defined __linux__ || defined __FreeBSD__) && !(defined ANDROID)
|
2014-04-01 18:20:08 -04:00
|
|
|
cpu_set_t cpu_set;
|
|
|
|
CPU_ZERO(&cpu_set);
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2014-04-01 18:20:08 -04:00
|
|
|
for (int i = 0; i != sizeof(mask) * 8; ++i)
|
|
|
|
if ((mask >> i) & 1)
|
|
|
|
CPU_SET(i, &cpu_set);
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2014-04-01 18:20:08 -04:00
|
|
|
pthread_setaffinity_np(thread, sizeof(cpu_set), &cpu_set);
|
2013-09-04 20:17:46 -04:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetCurrentThreadAffinity(u32 mask) {
|
2014-04-01 18:20:08 -04:00
|
|
|
SetThreadAffinity(pthread_self(), mask);
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
|
|
|
|
2014-11-29 00:38:20 -05:00
|
|
|
#ifndef _WIN32
|
2016-09-17 20:38:01 -04:00
|
|
|
void SwitchCurrentThread() {
|
2014-04-01 18:20:08 -04:00
|
|
|
usleep(1000 * 1);
|
2013-09-04 20:17:46 -04:00
|
|
|
}
|
2014-11-29 00:38:20 -05:00
|
|
|
#endif
|
2013-09-04 20:17:46 -04:00
|
|
|
|
2014-11-29 00:38:20 -05:00
|
|
|
// MinGW with the POSIX threading model does not support pthread_setname_np
|
2014-12-29 22:59:14 -05:00
|
|
|
#if !defined(_WIN32) || defined(_MSC_VER)
|
2016-09-17 20:38:01 -04:00
|
|
|
void SetCurrentThreadName(const char* szThreadName) {
|
2013-09-04 20:17:46 -04:00
|
|
|
#ifdef __APPLE__
|
2014-04-01 18:20:08 -04:00
|
|
|
pthread_setname_np(szThreadName);
|
2016-10-09 22:25:19 -04:00
|
|
|
#elif defined(__Bitrig__) || defined(__DragonFly__) || defined(__FreeBSD__) || defined(__OpenBSD__)
|
2014-08-08 00:56:58 -04:00
|
|
|
pthread_set_name_np(pthread_self(), szThreadName);
|
2016-10-09 22:25:19 -04:00
|
|
|
#elif defined(__NetBSD__)
|
|
|
|
pthread_setname_np(pthread_self(), "%s", (void*)szThreadName);
|
2013-09-04 20:17:46 -04:00
|
|
|
#else
|
2014-04-01 18:20:08 -04:00
|
|
|
pthread_setname_np(pthread_self(), szThreadName);
|
2013-09-04 20:17:46 -04:00
|
|
|
#endif
|
|
|
|
}
|
2014-11-29 00:38:20 -05:00
|
|
|
#endif
|
2013-09-04 20:17:46 -04:00
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
} // namespace Common
|