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 			, state(ThreadState::ALIVE)
59 			, workerIndex(0)
60 			, isExternalDescBuffer(true)
61 		{
62 			descBuffer = externalDescBuffer;
63 		}
64 
65 		ThreadContext::~ThreadContext()
66 		{
67 			if (isExternalDescBuffer == false)
68 			{
69 				Memory::Free(descBuffer);
70 			}
71 			descBuffer = nullptr;
72 		}
73 
74 		size_t ThreadContext::GetMemoryRequrementInBytesForDescBuffer()
75 		{
76 			return sizeof(internal::GroupedTask) * TASK_BUFFER_CAPACITY;
77 		}
78 
79 		void ThreadContext::SetThreadIndex(uint32 threadIndex)
80 		{
81 			workerIndex = threadIndex;
82 			random.SetSeed( GetPrimeNumber(threadIndex) );
83 		}
84 
85 #ifdef MT_INSTRUMENTED_BUILD
86 
87 		void ThreadContext::NotifyWaitStarted()
88 		{
89 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
90 			{
91 				eventListener->OnThreadWaitStarted();
92 			}
93 		}
94 
95 		void ThreadContext::NotifyWaitFinished()
96 		{
97 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
98 			{
99 				eventListener->OnThreadWaitFinished();
100 			}
101 		}
102 
103 		void ThreadContext::NotifyFiberAssignedToThread(uint32 fiberIndex, uint32 threadIndex)
104 		{
105 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
106 			{
107 				eventListener->OnFiberAssignedToThread(fiberIndex, threadIndex);
108 			}
109 		}
110 
111 		void ThreadContext::NotifyThreadAssignedToFiber()
112 		{
113 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
114 			{
115 				eventListener->OnThreadAssignedToFiber(workerIndex, 0xFFFFFFFF);
116 			}
117 		}
118 
119 		void ThreadContext::NotifyTaskExecuteStateChanged(MT::Color::Type debugColor, const mt_char* debugID, TaskExecuteState::Type type)
120 		{
121 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
122 			{
123 				eventListener->OnTaskExecuteStateChanged(debugColor, debugID, type);
124 			}
125 		}
126 
127 		void ThreadContext::NotifyThreadCreated(uint32 threadIndex)
128 		{
129 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
130 			{
131 				eventListener->OnThreadCreated(threadIndex);
132 			}
133 		}
134 
135 		void ThreadContext::NotifyThreadStarted(uint32 threadIndex)
136 		{
137 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
138 			{
139 				eventListener->OnThreadStarted(threadIndex);
140 			}
141 		}
142 
143 		void ThreadContext::NotifyThreadStoped(uint32 threadIndex)
144 		{
145 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
146 			{
147 				eventListener->OnThreadStoped(threadIndex);
148 			}
149 		}
150 
151 		void ThreadContext::NotifyThreadIdleStarted(uint32 threadIndex)
152 		{
153 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
154 			{
155 				eventListener->OnThreadIdleStarted(threadIndex);
156 			}
157 		}
158 
159 		void ThreadContext::NotifyThreadIdleFinished(uint32 threadIndex)
160 		{
161 			if (IProfilerEventListener* eventListener = taskScheduler->GetProfilerEventListener())
162 			{
163 				eventListener->OnThreadIdleFinished(threadIndex);
164 			}
165 		}
166 
167 #endif
168 
169 	}
170 
171 }
172