1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
/*
* CThreadHelper.h, part of VCMI engine
*
* Authors: listed in file AUTHORS in main folder
*
* License: GNU General Public License v2.0 or later
* Full text of license available in license.txt file, in main folder
*
*/
#pragma once
VCMI_LIB_NAMESPACE_BEGIN
///DEPRECATED
/// Can assign CPU work to other threads/cores
class DLL_LINKAGE CThreadHelper
{
public:
typedef std::function<void()> Task;
CThreadHelper(std::vector<std::function<void()> > *Tasks, int Threads);
void run();
private:
boost::mutex rtinm;
int currentTask, amount, threads;
std::vector<Task> *tasks;
void processTasks();
};
template<typename Payload>
class ThreadPool
{
public:
using Task = std::function<void(std::shared_ptr<Payload>)>;
using Tasks = std::vector<Task>;
ThreadPool(Tasks * tasks_, std::vector<std::shared_ptr<Payload>> context_)
: currentTask(0),
amount(tasks_->size()),
threads(context_.size()),
tasks(tasks_),
context(context_)
{}
void run()
{
boost::thread_group grupa;
for(size_t i=0; i<threads; i++)
{
std::shared_ptr<Payload> payload = context.at(i);
grupa.create_thread(std::bind(&ThreadPool::processTasks, this, payload));
}
grupa.join_all();
//thread group deletes threads, do not free manually
}
private:
boost::mutex rtinm;
size_t currentTask, amount, threads;
Tasks * tasks;
std::vector<std::shared_ptr<Payload>> context;
void processTasks(std::shared_ptr<Payload> payload)
{
while(true)
{
size_t pom;
{
boost::unique_lock<boost::mutex> lock(rtinm);
if((pom = currentTask) >= amount)
break;
else
++currentTask;
}
(*tasks)[pom](payload);
}
}
};
void DLL_LINKAGE setThreadName(const std::string &name);
VCMI_LIB_NAMESPACE_END
|
