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
23namespace MT
24{
25
26	template<class TTask>
27	void FiberContext::RunSubtasksAndYield(TaskGroup taskGroup, const TTask* taskArray, size_t taskCount)
28	{
29		MT_ASSERT(taskCount < (internal::TASK_BUFFER_CAPACITY - 1), "Too many tasks per one Run.");
30		MT_ASSERT(threadContext, "ThreadContext is nullptr");
31
32		TaskScheduler& scheduler = *(threadContext->taskScheduler);
33
34		ArrayView<internal::GroupedTask> buffer(threadContext->descBuffer, taskCount);
35
36		size_t bucketCount = MT::Min((size_t)scheduler.GetWorkersCount(), taskCount);
37		ArrayView<internal::TaskBucket> buckets(MT_ALLOCATE_ON_STACK(sizeof(internal::TaskBucket) * bucketCount), bucketCount);
38
39		internal::DistibuteDescriptions(taskGroup, taskArray, buffer, buckets);
40		RunSubtasksAndYieldImpl(buckets);
41	}
42
43	template<class TTask>
44	void FiberContext::RunAsync(TaskGroup taskGroup, const TTask* taskArray, size_t taskCount)
45	{
46		MT_ASSERT(taskCount < (internal::TASK_BUFFER_CAPACITY - 1), "Too many tasks per one Run.");
47
48		MT_ASSERT(threadContext, "ThreadContext is nullptr");
49		MT_ASSERT(threadContext->taskScheduler->IsWorkerThread(), "Can't use RunAsync outside Task. Use TaskScheduler.RunAsync() instead.");
50
51		TaskScheduler& scheduler = *(threadContext->taskScheduler);
52
53		ArrayView<internal::GroupedTask> buffer(threadContext->descBuffer, taskCount);
54
55		size_t bucketCount = MT::Min((size_t)scheduler.GetWorkersCount(), taskCount);
56		ArrayView<internal::TaskBucket>	buckets(MT_ALLOCATE_ON_STACK(sizeof(internal::TaskBucket) * bucketCount), bucketCount);
57
58		internal::DistibuteDescriptions(taskGroup, taskArray, buffer, buckets);
59		scheduler.RunTasksImpl(buckets, nullptr, false);
60	}
61
62
63
64
65}
66