/
PTTasks.cs
182 lines (158 loc) · 4.82 KB
/
PTTasks.cs
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
/*
Copyright 2019, Daumantas Kavolis
This file is part of ParallelTasker.
ParallelTasker is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
ParallelTasker is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with ParallelTasker. If not, see <http: //www.gnu.org/licenses/>.
*/
using System;
namespace ParallelTasker
{
/// <summary>
/// A container for ParallelTasker tasks
/// </summary>
public class PTTask
{
private static readonly ObjectPool<PTTask> s_pool = new ObjectPool<PTTask>(() => new PTTask(), null, OnRelease);
public Func<object> initialize;
public Func<object, object> main;
public Action<object> finalize;
public uint period;
public PTTimePair EndTime
{
get;
internal set;
}
private uint m_counter;
public PTTask() : this(PTTimePair.DefaultUpdate, null, null, null)
{ }
public PTTask(PTTimePair endTime, Func<object> initialize, Func<object, object> main, Action<object> finalize, uint period = 1)
{
this.EndTime = endTime;
this.initialize = initialize;
this.main = main;
this.finalize = finalize;
this.period = period;
m_counter = period;
}
public bool ShouldExecuteNext()
{
return m_counter == period;
}
internal bool ShouldExecute()
{
if (m_counter == period)
{
m_counter = 1;
return true;
}
m_counter++;
return false;
}
protected static void OnRelease(PTTask task)
{
task.initialize = null;
task.main = null;
task.finalize = null;
task.period = 0;
}
public static PTTask Borrow(PTTimePair endTime, Func<object> initialize, Func<object, object> main, Action<object> finalize, uint period = 1)
{
var task = s_pool.Borrow();
task.initialize = initialize;
task.main = main;
task.finalize = finalize;
task.period = period;
task.EndTime = endTime;
task.m_counter = 1;
return task;
}
public static void Release(PTTask task)
{
task.Release();
}
public virtual void Release()
{
s_pool.Release(this);
}
}
/// <inheritdoc />
/// <summary>
/// A container for enqueued ParallelTasker tasks
/// </summary>
public class PTThreadTask : PTTask
{
private static readonly ObjectPool<PTThreadTask> s_pool = new ObjectPool<PTThreadTask>(() => new PTThreadTask(), null, OnRelease);
private object argument;
public PTThreadTask() : base()
{ }
private static void OnRelease(PTThreadTask task)
{
PTTask.OnRelease(task);
task.argument = null;
}
public static PTThreadTask Borrow(PTTimePair endTime, PTTask task)
{
var ttask = s_pool.Borrow();
ttask.initialize = task.initialize;
ttask.main = task.main;
ttask.finalize = task.finalize;
ttask.EndTime = endTime;
return ttask;
}
public override void Release()
{
s_pool.Release(this);
}
public static void Release(PTThreadTask task)
{
task.Release();
}
public PTThreadTask RunInitializer()
{
try
{
argument = initialize?.Invoke();
}
catch (Exception ex)
{
main = null;
finalize = null;
PTLogger.Exception(ex);
}
return this;
}
public PTThreadTask RunMainTask()
{
try
{
argument = main?.Invoke(argument);
}
catch (Exception ex)
{
finalize = null;
PTThreadSafeLogger.LogException(ex);
}
return this;
}
public void RunFinalizer()
{
try
{
finalize?.Invoke(argument);
}
catch (Exception ex)
{
PTLogger.Exception(ex);
}
Release();
}
}
}