mirror of
https://github.com/ddnet/ddnet.git
synced 2024-11-11 02:28:18 +00:00
93 lines
2 KiB
C++
93 lines
2 KiB
C++
/* (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_NumThreads = 0;
|
|
m_Shutdown = false;
|
|
m_Lock = lock_create();
|
|
sphore_init(&m_Semaphore);
|
|
m_pFirstJob = 0;
|
|
m_pLastJob = 0;
|
|
}
|
|
|
|
CJobPool::~CJobPool()
|
|
{
|
|
m_Shutdown = true;
|
|
for(int i = 0; i < m_NumThreads; i++)
|
|
sphore_signal(&m_Semaphore);
|
|
for(int i = 0; i < m_NumThreads; i++)
|
|
thread_wait(m_apThreads[i]);
|
|
lock_destroy(m_Lock);
|
|
sphore_destroy(&m_Semaphore);
|
|
}
|
|
|
|
void CJobPool::WorkerThread(void *pUser)
|
|
{
|
|
CJobPool *pPool = (CJobPool *)pUser;
|
|
|
|
while(!pPool->m_Shutdown)
|
|
{
|
|
CJob *pJob = 0;
|
|
|
|
// fetch job from queue
|
|
sphore_wait(&pPool->m_Semaphore);
|
|
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_unlock(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;
|
|
if(pJob->m_pfnCallback)
|
|
pJob->m_pfnCallback(pJob, pJob->m_pFuncData);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
int CJobPool::Init(int NumThreads)
|
|
{
|
|
// start threads
|
|
m_NumThreads = NumThreads > MAX_THREADS ? MAX_THREADS : NumThreads;
|
|
for(int i = 0; i < NumThreads; i++)
|
|
m_apThreads[i] = thread_init(WorkerThread, this);
|
|
return 0;
|
|
}
|
|
|
|
int CJobPool::Add(CJob *pJob, JOBFUNC pfnFunc, void *pData, CBFUNC pfnCallback)
|
|
{
|
|
mem_zero(pJob, sizeof(CJob));
|
|
pJob->m_pfnFunc = pfnFunc;
|
|
pJob->m_pFuncData = pData;
|
|
pJob->m_pfnCallback = pfnCallback;
|
|
|
|
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_unlock(m_Lock);
|
|
sphore_signal(&m_Semaphore);
|
|
return 0;
|
|
}
|