// The MIT License (MIT) // // Copyright (c) 2015 Sergey Makeev, Vadim Slyusarev // // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // The above copyright notice and this permission notice shall be included in // all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN // THE SOFTWARE. #pragma once #ifndef __MT_THREAD__ #define __MT_THREAD__ #include namespace MT { class _Fiber; class Thread : public ThreadBase { MW_DWORD threadId; MW_HANDLE thread; static MW_DWORD __stdcall ThreadFuncInternal(void *pThread) { Thread* self = (Thread*)pThread; self->threadId = ::GetCurrentThreadId(); self->func(self->funcData); return 0; } public: Thread() : thread(nullptr) , threadId(0) { } ~Thread() { MT_ASSERT(thread == nullptr, "Thread is not stopped!"); } void Start(size_t stackSize, TThreadEntryPoint entryPoint, void *userData) { MT_ASSERT(thread == nullptr, "Thread already started"); func = entryPoint; funcData = userData; thread = ::CreateThread( nullptr, stackSize, ThreadFuncInternal, this, 0, nullptr ); MT_ASSERT(thread != nullptr, "Can't create thread"); } void Join() { if (thread == nullptr) { return; } ::WaitForSingleObject(thread, MW_INFINITE); MW_BOOL res = CloseHandle(thread); MT_USED_IN_ASSERT(res); MT_ASSERT(res != 0, "Can't close thread handle"); thread = nullptr; } bool IsCurrentThread() const { MW_DWORD id = ::GetCurrentThreadId(); return (threadId == id); } static void SetCurrentThreadName(const char* threadName) { MT_UNUSED(threadName); #ifdef MT_INSTRUMENTED_BUILD const int MW_EXCEPTION_EXECUTE_HANDLER = 1; const MW_DWORD MW_MSVC_EXCEPTION = 0x406D1388; #pragma pack(push,8) typedef struct tagTHREADNAME_INFO { MW_DWORD dwType; // Must be 0x1000. const char* szName; // Pointer to name (in user addr space). MW_DWORD dwThreadID; // Thread ID (-1=caller thread). MW_DWORD dwFlags; // Reserved for future use, must be zero. } THREADNAME_INFO; #pragma pack(pop) THREADNAME_INFO info; info.dwType = 0x1000; info.szName = threadName; info.dwThreadID = GetCurrentThreadId(); info.dwFlags = 0; __try { RaiseException(MW_MSVC_EXCEPTION, 0, sizeof(info) / sizeof(void*), (MW_ULONG_PTR*)&info); } __except (MW_EXCEPTION_EXECUTE_HANDLER) { } #endif } static int GetNumberOfHardwareThreads() { MW_SYSTEM_INFO sysinfo; ::GetSystemInfo( &sysinfo ); return sysinfo.dwNumberOfProcessors; } static void Sleep(uint32 milliseconds) { ::Sleep(milliseconds); } }; } #endif