#include "config.h"
#include "MainThread.h"
#include "StdLibExtras.h"
#include "Threading.h"
#include "Vector.h"
namespace WTF {
struct FunctionWithContext {
MainThreadFunction* function;
void* context;
ThreadCondition* syncFlag;
FunctionWithContext(MainThreadFunction* function = 0, void* context = 0, ThreadCondition* syncFlag = 0)
: function(function)
, context(context)
, syncFlag(syncFlag)
{
}
};
typedef Vector<FunctionWithContext> FunctionQueue;
static bool callbacksPaused;
Mutex& mainThreadFunctionQueueMutex()
{
DEFINE_STATIC_LOCAL(Mutex, staticMutex, ());
return staticMutex;
}
static FunctionQueue& functionQueue()
{
DEFINE_STATIC_LOCAL(FunctionQueue, staticFunctionQueue, ());
return staticFunctionQueue;
}
#if !PLATFORM(WIN)
void initializeMainThread()
{
mainThreadFunctionQueueMutex();
}
#endif
void dispatchFunctionsFromMainThread()
{
ASSERT(isMainThread());
if (callbacksPaused)
return;
FunctionQueue queueCopy;
{
MutexLocker locker(mainThreadFunctionQueueMutex());
queueCopy.swap(functionQueue());
}
for (unsigned i = 0; i < queueCopy.size(); ++i) {
FunctionWithContext& invocation = queueCopy[i];
invocation.function(invocation.context);
if (invocation.syncFlag)
invocation.syncFlag->signal();
}
}
void callOnMainThread(MainThreadFunction* function, void* context)
{
ASSERT(function);
{
MutexLocker locker(mainThreadFunctionQueueMutex());
functionQueue().append(FunctionWithContext(function, context));
}
scheduleDispatchFunctionsOnMainThread();
}
void callOnMainThreadAndWait(MainThreadFunction* function, void* context)
{
ASSERT(function);
if (isMainThread()) {
function(context);
return;
}
ThreadCondition syncFlag;
Mutex conditionMutex;
{
MutexLocker locker(mainThreadFunctionQueueMutex());
functionQueue().append(FunctionWithContext(function, context, &syncFlag));
conditionMutex.lock();
}
scheduleDispatchFunctionsOnMainThread();
syncFlag.wait(conditionMutex);
}
void setMainThreadCallbacksPaused(bool paused)
{
ASSERT(isMainThread());
if (callbacksPaused == paused)
return;
callbacksPaused = paused;
if (!callbacksPaused)
scheduleDispatchFunctionsOnMainThread();
}
}