1 // The MIT License (MIT)
2 //
3 // 	Copyright (c) 2015 Sergey Makeev, Vadim Slyusarev
4 //
5 // 	Permission is hereby granted, free of charge, to any person obtaining a copy
6 // 	of this software and associated documentation files (the "Software"), to deal
7 // 	in the Software without restriction, including without limitation the rights
8 // 	to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9 // 	copies of the Software, and to permit persons to whom the Software is
10 // 	furnished to do so, subject to the following conditions:
11 //
12 //  The above copyright notice and this permission notice shall be included in
13 // 	all copies or substantial portions of the Software.
14 //
15 // 	THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 // 	IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 // 	FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 // 	AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 // 	LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 // 	OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
21 // 	THE SOFTWARE.
22 #pragma once
23 
24 #ifndef __MT_THREAD__
25 #define __MT_THREAD__
26 
27 
28 #include <Platform/Common/MTThread.h>
29 
30 namespace MT
31 {
32 	class _Fiber;
33 
34 	class Thread : public ThreadBase
35 	{
36 		MW_DWORD threadId;
37 		MW_HANDLE thread;
38 
39 		static MW_DWORD __stdcall ThreadFuncInternal(void *pThread)
40 		{
41 			Thread* self = (Thread*)pThread;
42 			self->threadId = ::GetCurrentThreadId();
43 			self->func(self->funcData);
44 			return 0;
45 		}
46 	public:
47 
48 		Thread()
49 			: thread(nullptr)
50 			, threadId(0)
51 		{
52 		}
53 
54 		~Thread()
55 		{
56 			MT_ASSERT(thread == nullptr, "Thread is not stopped!");
57 		}
58 
59 		void Start(size_t stackSize, TThreadEntryPoint entryPoint, void *userData)
60 		{
61 			MT_ASSERT(thread == nullptr, "Thread already started");
62 
63 			func = entryPoint;
64 			funcData = userData;
65 			thread = ::CreateThread( nullptr, stackSize, ThreadFuncInternal, this, 0, nullptr );
66 			MT_ASSERT(thread != nullptr, "Can't create thread");
67 		}
68 
69 		void Join()
70 		{
71 			if (thread == nullptr)
72 			{
73 				return;
74 			}
75 
76 			::WaitForSingleObject(thread, MW_INFINITE);
77 			MW_BOOL res = CloseHandle(thread);
78 			MT_USED_IN_ASSERT(res);
79 			MT_ASSERT(res != 0, "Can't close thread handle");
80 			thread = nullptr;
81 		}
82 
83 		bool IsCurrentThread() const
84 		{
85 			MW_DWORD id = ::GetCurrentThreadId();
86 			return (threadId == id);
87 		}
88 
89 		static void SetCurrentThreadName(const char* threadName)
90 		{
91 			MT_UNUSED(threadName);
92 
93 #ifdef MT_INSTRUMENTED_BUILD
94 
95 			const int MW_EXCEPTION_EXECUTE_HANDLER = 1;
96 			const MW_DWORD MW_MSVC_EXCEPTION = 0x406D1388;
97 
98 #pragma pack(push,8)
99 			typedef struct tagTHREADNAME_INFO
100 			{
101 				MW_DWORD dwType; // Must be 0x1000.
102 				const char* szName; // Pointer to name (in user addr space).
103 				MW_DWORD dwThreadID; // Thread ID (-1=caller thread).
104 				MW_DWORD dwFlags; // Reserved for future use, must be zero.
105 			} THREADNAME_INFO;
106 #pragma pack(pop)
107 
108 			THREADNAME_INFO info;
109 			info.dwType = 0x1000;
110 			info.szName = threadName;
111 			info.dwThreadID = GetCurrentThreadId();
112 			info.dwFlags = 0;
113 
114 			__try
115 			{
116 				RaiseException(MW_MSVC_EXCEPTION, 0, sizeof(info) / sizeof(void*), (MW_ULONG_PTR*)&info);
117 			}
118 			__except (MW_EXCEPTION_EXECUTE_HANDLER)
119 			{
120 			}
121 #endif
122 		}
123 
124 
125 		static int GetNumberOfHardwareThreads()
126 		{
127 			MW_SYSTEM_INFO sysinfo;
128 			::GetSystemInfo( &sysinfo );
129 			return sysinfo.dwNumberOfProcessors;
130 		}
131 
132 		static void Sleep(uint32 milliseconds)
133 		{
134 		  ::Sleep(milliseconds);
135 		}
136 	};
137 
138 
139 }
140 
141 
142 #endif
143