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 		DWORD threadId;
37 		HANDLE thread;
38 
39 		static DWORD __stdcall ThreadFuncInternal(void *pThread)
40 		{
41 			Thread* self = (Thread*)pThread;
42 
43 			self->func(self->funcData);
44 
45 			return 0;
46 		}
47 	public:
48 
49 		Thread()
50 			: thread(nullptr)
51 			, threadId(0)
52 		{
53 		}
54 
55 		~Thread()
56 		{
57 			MT_ASSERT(thread == nullptr, "Thread is not stopped!");
58 		}
59 
60 		void Start(size_t stackSize, TThreadEntryPoint entryPoint, void *userData)
61 		{
62 			MT_ASSERT(thread == nullptr, "Thread already started");
63 
64 			func = entryPoint;
65 			funcData = userData;
66 			thread = ::CreateThread( nullptr, stackSize, ThreadFuncInternal, this, 0, &threadId );
67 			MT_ASSERT(thread != nullptr, "Can't create thread");
68 		}
69 
70 		void Stop()
71 		{
72 			if (thread == nullptr)
73 			{
74 				return;
75 			}
76 
77 			::WaitForSingleObject(thread, INFINITE);
78 			BOOL res = CloseHandle(thread);
79 			MT_USED_IN_ASSERT(res);
80 			MT_ASSERT(res != 0, "Can't close thread handle");
81 			thread = nullptr;
82 		}
83 
84 		bool IsCurrentThread() const
85 		{
86 			return threadId == ::GetCurrentThreadId();
87 		}
88 
89 		static int GetNumberOfHardwareThreads()
90 		{
91 			SYSTEM_INFO sysinfo;
92 			::GetSystemInfo( &sysinfo );
93 			return sysinfo.dwNumberOfProcessors;
94 		}
95 
96 		static void Sleep(uint32 milliseconds)
97 		{
98 		  ::Sleep(milliseconds);
99 		}
100 	};
101 
102 
103 }
104 
105 
106 #endif