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 
23 #include <MTScheduler.h>
24 
25 namespace MT
26 {
27 	namespace internal
28 	{
29 		// Prime numbers for linear congruential generator seed
30 		static const uint32 primeNumbers[] = {
31 			128473, 135349, 159499, 173839, 209213, 241603, 292709, 314723,
32 			343943, 389299, 419473, 465169, 518327, 649921, 748271, 851087,
33 			862171, 974551, 1002973, 1034639, 1096289, 1153123, 1251037, 1299269,
34 			1272941, 1252151, 1231091, 1206761, 1185469, 1169933, 1141351, 1011583 };
35 
36 		uint32 GetPrimeNumber(uint32 index)
37 		{
38 			return primeNumbers[index % MT_ARRAY_SIZE(primeNumbers)];
39 		}
40 
41 
42 
43 		ThreadContext::ThreadContext()
44 			: lastActiveFiberContext(nullptr)
45 			, taskScheduler(nullptr)
46 			, hasNewTasksEvent(EventReset::AUTOMATIC, true)
47 			, state(ThreadState::ALIVE)
48 			, workerIndex(0)
49 			, isExternalDescBuffer(false)
50 		{
51 			 descBuffer = Memory::Alloc( GetMemoryRequrementInBytesForDescBuffer() );
52 		}
53 
54 		ThreadContext::ThreadContext(void* externalDescBuffer)
55 			: lastActiveFiberContext(nullptr)
56 			, taskScheduler(nullptr)
57 			, queue(DummyQueueFlag::IS_DUMMY_QUEUE)
58 			, hasNewTasksEvent(EventReset::AUTOMATIC, true)
59 			, state(ThreadState::ALIVE)
60 			, workerIndex(0)
61 			, isExternalDescBuffer(true)
62 		{
63 			descBuffer = externalDescBuffer;
64 		}
65 
66 		ThreadContext::~ThreadContext()
67 		{
68 			if (isExternalDescBuffer == false)
69 			{
70 				Memory::Free(descBuffer);
71 			}
72 			descBuffer = nullptr;
73 		}
74 
75 		size_t ThreadContext::GetMemoryRequrementInBytesForDescBuffer()
76 		{
77 			return sizeof(internal::GroupedTask) * TASK_BUFFER_CAPACITY;
78 		}
79 
80 		void ThreadContext::SetThreadIndex(uint32 threadIndex)
81 		{
82 			workerIndex = threadIndex;
83 			random.SetSeed( GetPrimeNumber(threadIndex) );
84 		}
85 
86 #ifdef MT_INSTRUMENTED_BUILD
87 
88 		void ThreadContext::NotifyWaitStarted()
89 		{
90 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
91 			{
92 				eventListener->OnThreadWaitStarted();
93 			}
94 		}
95 
96 		void ThreadContext::NotifyWaitFinished()
97 		{
98 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
99 			{
100 				eventListener->OnThreadWaitFinished();
101 			}
102 		}
103 
104 
105 		void ThreadContext::NotifyTaskExecuteStateChanged(MT::Color::Type debugColor, const mt_char* debugID, TaskExecuteState::Type type)
106 		{
107 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
108 			{
109 				eventListener->NotifyTaskExecuteStateChanged(debugColor, debugID, type);
110 			}
111 		}
112 
113 		void ThreadContext::NotifyThreadCreated(uint32 threadIndex)
114 		{
115 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
116 			{
117 				eventListener->OnThreadCreated(threadIndex);
118 			}
119 		}
120 
121 		void ThreadContext::NotifyThreadStarted(uint32 threadIndex)
122 		{
123 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
124 			{
125 				eventListener->OnThreadStarted(threadIndex);
126 			}
127 		}
128 
129 		void ThreadContext::NotifyThreadStoped(uint32 threadIndex)
130 		{
131 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
132 			{
133 				eventListener->OnThreadStoped(threadIndex);
134 			}
135 		}
136 
137 		void ThreadContext::NotifyThreadIdleStarted(uint32 threadIndex)
138 		{
139 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
140 			{
141 				eventListener->OnThreadIdleStarted(threadIndex);
142 			}
143 		}
144 
145 		void ThreadContext::NotifyThreadIdleFinished(uint32 threadIndex)
146 		{
147 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
148 			{
149 				eventListener->OnThreadIdleFinished(threadIndex);
150 			}
151 		}
152 
153 #endif
154 
155 	}
156 
157 }
158