blob: 4bc09325e67942f7f520ea12e68a9deb1cd42354 (
plain)
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
|
/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
/* If you are missing that file, acquire a complete release at teeworlds.com. */
#include <base/system.h>
#include "jobs.h"
CJobPool::CJobPool()
{
// empty the pool
m_Lock = lock_create();
m_pFirstJob = 0;
m_pLastJob = 0;
}
void CJobPool::WorkerThread(void *pUser)
{
CJobPool *pPool = (CJobPool *)pUser;
while(1)
{
CJob *pJob = 0;
// fetch job from queue
lock_wait(pPool->m_Lock);
if(pPool->m_pFirstJob)
{
pJob = pPool->m_pFirstJob;
pPool->m_pFirstJob = pPool->m_pFirstJob->m_pNext;
if(pPool->m_pFirstJob)
pPool->m_pFirstJob->m_pPrev = 0;
else
pPool->m_pLastJob = 0;
}
lock_release(pPool->m_Lock);
// do the job if we have one
if(pJob)
{
pJob->m_Status = CJob::STATE_RUNNING;
pJob->m_Result = pJob->m_pfnFunc(pJob->m_pFuncData);
pJob->m_Status = CJob::STATE_DONE;
}
else
thread_sleep(10);
}
}
int CJobPool::Init(int NumThreads)
{
// start threads
for(int i = 0; i < NumThreads; i++)
thread_create(WorkerThread, this);
return 0;
}
int CJobPool::Add(CJob *pJob, JOBFUNC pfnFunc, void *pData)
{
mem_zero(pJob, sizeof(CJob));
pJob->m_pfnFunc = pfnFunc;
pJob->m_pFuncData = pData;
lock_wait(m_Lock);
// add job to queue
pJob->m_pPrev = m_pLastJob;
if(m_pLastJob)
m_pLastJob->m_pNext = pJob;
m_pLastJob = pJob;
if(!m_pFirstJob)
m_pFirstJob = pJob;
lock_release(m_Lock);
return 0;
}
|