Added: Function to set thread priority

This commit is contained in:
kaetemi 2012-04-11 15:38:49 +02:00
parent 392b224799
commit 3aff28aab4
5 changed files with 47 additions and 7 deletions

View file

@ -48,6 +48,7 @@ public:
virtual void wait(); virtual void wait();
virtual bool setCPUMask(uint64 cpuMask); virtual bool setCPUMask(uint64 cpuMask);
virtual uint64 getCPUMask(); virtual uint64 getCPUMask();
virtual void setPriority(TThreadPriority priority);
virtual std::string getUserName(); virtual std::string getUserName();
virtual IRunnable *getRunnable() virtual IRunnable *getRunnable()

View file

@ -68,6 +68,16 @@ public:
} }
}; };
/// Thread priorities, numbering follows Win32 for now
enum TThreadPriority
{
ThreadPriorityLowest = -2,
ThreadPriorityLow = -1,
ThreadPriorityNormal = 0,
ThreadPriorityHigh = 1,
ThreadPriorityHighest = 2,
};
/** /**
* Thread base interface, must be implemented for all OS * Thread base interface, must be implemented for all OS
* \author Vianney Lecroart * \author Vianney Lecroart
@ -119,6 +129,9 @@ public:
*/ */
virtual uint64 getCPUMask()=0; virtual uint64 getCPUMask()=0;
/// Set the thread priority. Thread must have been started before.
virtual void setPriority(TThreadPriority priority) = 0;
/** /**
* Get the thread user name. * Get the thread user name.
* Under Linux return thread owner, under windows return the name of the logon user. * Under Linux return thread owner, under windows return the name of the logon user.

View file

@ -49,6 +49,7 @@ public:
virtual void wait(); virtual void wait();
virtual bool setCPUMask(uint64 cpuMask); virtual bool setCPUMask(uint64 cpuMask);
virtual uint64 getCPUMask(); virtual uint64 getCPUMask();
virtual void setPriority(TThreadPriority priority);
virtual std::string getUserName(); virtual std::string getUserName();
virtual IRunnable *getRunnable() virtual IRunnable *getRunnable()
@ -69,8 +70,7 @@ public:
void suspend(); void suspend();
// Resume the thread. No-op if already resumed // Resume the thread. No-op if already resumed
void resume(); void resume();
// set priority as defined by "SetThreadpriority" // Priority boost
void setPriority(int priority);
void enablePriorityBoost(bool enabled); void enablePriorityBoost(bool enabled);
/// private use /// private use

View file

@ -207,6 +207,32 @@ uint64 CPThread::getCPUMask()
return cpuMask; return cpuMask;
} }
void CPThread::setPriority(TThreadPriority priority)
{
// TODO: Verify and test this
switch (priority)
{
case ThreadPriorityHigh:
{
int minPrio = sched_get_priority_min(SCHED_FIFO);
int maxPrio = sched_get_priority_max(SCHED_FIFO);
int prio = ((maxPrio - minPrio) / 4) + minPrio;
pthread_setschedparam(_ThreadHandle, SCHED_FIFO, prio);
break;
}
case ThreadPriorityHighest:
{
int minPrio = sched_get_priority_min(SCHED_FIFO);
int maxPrio = sched_get_priority_max(SCHED_FIFO);
int prio = ((maxPrio - minPrio) / 2) + minPrio;
pthread_setschedparam(_ThreadHandle, SCHED_FIFO, prio);
break;
}
default:
pthread_setschedparam(_ThreadHandle, SCHED_OTHER, 0);
}
}
/* /*
* getUserName * getUserName
*/ */

View file

@ -84,6 +84,7 @@ public:
inline void enter() { EnterCriticalSection(&cs); } inline void enter() { EnterCriticalSection(&cs); }
inline void leave() { LeaveCriticalSection(&cs); } inline void leave() { LeaveCriticalSection(&cs); }
}; };
CWinCriticalSection s_CS;
}/* anonymous namespace */ }/* anonymous namespace */
CWinThread::CWinThread (void* threadHandle, uint32 threadId) CWinThread::CWinThread (void* threadHandle, uint32 threadId)
@ -112,12 +113,11 @@ CWinThread::CWinThread (void* threadHandle, uint32 threadId)
nlassert(0); // WARNING: following code has not tested! don't know if it work fo real ... nlassert(0); // WARNING: following code has not tested! don't know if it work fo real ...
// This is just a suggestion of a possible solution, should this situation one day occur ... // This is just a suggestion of a possible solution, should this situation one day occur ...
// Ensure that this thread don't get deleted, or we could suspend the main thread // Ensure that this thread don't get deleted, or we could suspend the main thread
static CWinCriticalSection cs; s_CS.enter();
cs.enter();
// the 2 following statement must be executed atomicaly among the threads of the current process ! // the 2 following statement must be executed atomicaly among the threads of the current process !
SuspendThread(threadHandle); SuspendThread(threadHandle);
_SuspendCount = ResumeThread(threadHandle); _SuspendCount = ResumeThread(threadHandle);
cs.leave(); s_CS.leave();
} }
} }
@ -159,10 +159,10 @@ void CWinThread::resume()
} }
} }
void CWinThread::setPriority(int priority) void CWinThread::setPriority(TThreadPriority priority)
{ {
nlassert(ThreadHandle); // 'start' was not called !! nlassert(ThreadHandle); // 'start' was not called !!
BOOL result = SetThreadPriority(ThreadHandle, priority); BOOL result = SetThreadPriority(ThreadHandle, (int)priority);
nlassert(result); nlassert(result);
} }