BGI/SOAPsnp/ThreadManager.cpp

   1 #include "ThreadManager.h"
   2
   3 pthread_mutex_t         mutex_ctrl = PTHREAD_MUTEX_INITIALIZER;
   4
   5 void *__processwork(void* p)
   6 {       THRPARAM*       param = (THRPARAM*)p;
   7
   8         while(param->pMgr->m_nindex < param->pMgr->m_nmaxthread)
   9         {
  10                 int             nindex = 0;
  11                 pthread_mutex_lock(&mutex_ctrl);
  12                 if(param->pMgr->m_nindex < param->pMgr->m_nmaxthread)
  13                 {
  14                         nindex = param->pMgr->m_nindex++;
  15                 }
  16                 else
  17                 {
  18                         pthread_mutex_unlock(&mutex_ctrl);
  19                         break;
  20                 }
  21                 pthread_mutex_unlock(&mutex_ctrl);
  22                 param->pMgr->m_vproc[nindex](param->pMgr->m_vparam[nindex]);
  23         }
  24         return NULL;
  25 }
  26
  27 ThreadManager::ThreadManager()
  28 {
  29         m_nindex = 0;
  30         m_nmaxthread = 0;
  31         m_ftimer = 0.1;
  32         m_cpu = 1;
  33 }
  34
  35 ThreadManager::ThreadManager( int cpu ):m_cpu(cpu)
  36 {
  37         m_nindex = 0;
  38         m_nmaxthread = 0;
  39         m_ftimer = 0.5;
  40
  41         if(m_cpu <= 0)
  42                 m_cpu = 1;
  43 }
  44
  45 ThreadManager::~ThreadManager()
  46 {
  47         // UPDATE at 2010-10-21
  48         for (int i = 0; i < m_vparam.size(); ++i)
  49         {
  50                 if (m_vparam[i] != NULL)
  51                         delete m_vparam[i];
  52                 m_vparam[i] = NULL;
  53         }
  54 }
  55
  56 void ThreadManager::AddThread( pthreadproc pProc, void* param )
  57 {
  58         m_vproc.push_back(pProc);
  59         m_vparam.push_back(param);
  60         m_nmaxthread++;
  61 }
  62
  63 void ThreadManager::Run()
  64 {
  65         if(m_cpu >= m_nmaxthread)
  66                 m_cpu = m_nmaxthread;
  67         vector<pthread_t>       vid(m_cpu);
  68         vector<THRPARAM>        vparam(m_cpu);
  69         for (int i=0; i<m_cpu; i++)
  70         {
  71                 vparam[i].pMgr = this;
  72                 vparam[i].flag = 0;
  73                 pthread_create(&vid[i], NULL, __processwork, (void*)&vparam[i]);
  74                 usleep(int(m_ftimer*float(1000000)));
  75         }
  76
  77         for (int i=0; i<m_cpu; i++)
  78         {
  79                 pthread_join(vid[i], NULL);
  80         }
  81 }
  82
  83 void ThreadManager::SetTimer( float t /*= 0.5*/ )
  84 {
  85         m_ftimer = t;
  86 }
  87
  88 void ThreadManager::Reset()
  89 {
  90         m_ftimer = 1.0;
  91         m_cpu = 1;
  92         m_nmaxthread = 0;
  93         m_nindex = 0;
  94
  95         // UPDATE at 2010-10-12
  96         for (int i = 0; i < m_vparam.size(); ++i)
  97         {
  98                 delete m_vparam[i];
  99                 m_vparam[i] = NULL;
 100         }
 101         m_vparam.clear();
 102         m_vproc.clear();
 103 }
 104
 105 void ThreadManager::SetCPU( int count )
 106 {
 107         m_cpu = count;
 108         if (m_cpu >= m_nmaxthread) m_cpu = m_nmaxthread;
 109 }