-
Notifications
You must be signed in to change notification settings - Fork 3
/
FunctionRunner.h
123 lines (104 loc) · 1.62 KB
/
FunctionRunner.h
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
* FunctionRunner.h
*
* Created on: Mar 16, 2021
* Author: tugrul
*/
#ifndef FUNCTIONRUNNER_H_
#define FUNCTIONRUNNER_H_
#include<mutex>
#include<queue>
#include<thread>
#include<condition_variable>
#include<memory>
template<typename T>
class Prefetcher
{
public:
Prefetcher(){ }
Prefetcher(Prefetcher & p) = default;
Prefetcher(T vaPrm)
{
va=vaPrm;
thr=std::thread([&](){
size_t currentIndex;
bool work = true;
while(work)
{
{
std::unique_lock<std::mutex> lck(mut);
if(!iQ.empty())
{
currentIndex = iQ.front();
iQ.pop();
}
else
{
cond.wait_for(lck,std::chrono::milliseconds(100));
continue;
}
}
if(currentIndex==-1)
{
work=false;
}
else
{
va.get(currentIndex);
}
}
{
std::unique_lock<std::mutex> lck(mut);
while(!iQ.empty())
{
iQ.pop();
}
iQ.push(-5);
}
});
}
void push(const size_t index)
{
{
std::unique_lock<std::mutex> lck(mut);
iQ.push(index);
}
cond.notify_one();
}
~Prefetcher()
{
push(-1);
bool work = true;
while(work)
{
{
std::unique_lock<std::mutex> lck(mut);
if(!iQ.empty())
{
size_t index = iQ.front();
iQ.pop();
if(index == -5)
{
// close signal received
work=false;
}
else if (index == -1)
{
iQ.push(-1);
}
}
}
cond.notify_one();
std::this_thread::yield();
}
if(thr.joinable())
thr.join();
}
private:
std::condition_variable cond;
std::thread thr;
std::mutex mut;
std::queue<size_t> iQ;
T va;
};
#endif /* FUNCTIONRUNNER_H_ */