#include "config.h"
#include "RunLoop.h"
#include "WindowsExtras.h"
#include <wtf/CurrentTime.h>
using namespace std;
namespace WebCore {
static const UINT PerformWorkMessage = WM_USER + 1;
static const LPWSTR kRunLoopMessageWindowClassName = L"RunLoopMessageWindow";
LRESULT CALLBACK RunLoop::RunLoopWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
if (RunLoop* runLoop = static_cast<RunLoop*>(getWindowPointer(hWnd, 0)))
return runLoop->wndProc(hWnd, message, wParam, lParam);
if (message == WM_CREATE) {
LPCREATESTRUCT createStruct = reinterpret_cast<LPCREATESTRUCT>(lParam);
setWindowPointer(hWnd, 0, createStruct->lpCreateParams);
return 0;
}
return ::DefWindowProc(hWnd, message, wParam, lParam);
}
LRESULT RunLoop::wndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
switch (message) {
case PerformWorkMessage:
performWork();
return 0;
case WM_TIMER:
RunLoop::TimerBase::timerFired(this, wParam);
return 0;
}
return ::DefWindowProc(hWnd, message, wParam, lParam);
}
void RunLoop::run()
{
MSG message;
while (BOOL result = ::GetMessage(&message, 0, 0, 0)) {
if (result == -1)
break;
::TranslateMessage(&message);
::DispatchMessage(&message);
}
}
void RunLoop::stop()
{
::PostQuitMessage(0);
}
bool RunLoop::registerRunLoopMessageWindowClass()
{
WNDCLASS windowClass = { 0 };
windowClass.lpfnWndProc = RunLoop::RunLoopWndProc;
windowClass.cbWndExtra = sizeof(RunLoop*);
windowClass.lpszClassName = kRunLoopMessageWindowClassName;
return !!::RegisterClass(&windowClass);
}
RunLoop::RunLoop()
{
registerRunLoopMessageWindowClass();
m_runLoopMessageWindow = ::CreateWindow(kRunLoopMessageWindowClassName, 0, 0,
CW_USEDEFAULT, 0, CW_USEDEFAULT, 0,
HWND_MESSAGE, 0, 0, this);
ASSERT(::IsWindow(m_runLoopMessageWindow));
}
RunLoop::~RunLoop()
{
}
void RunLoop::wakeUp()
{
::PostMessage(m_runLoopMessageWindow, PerformWorkMessage, reinterpret_cast<WPARAM>(this), 0);
}
void RunLoop::TimerBase::timerFired(RunLoop* runLoop, uint64_t ID)
{
TimerMap::iterator it = runLoop->m_activeTimers.find(ID);
if (it == runLoop->m_activeTimers.end()) {
return;
}
TimerBase* timer = it->second;
if (!timer->m_isRepeating) {
runLoop->m_activeTimers.remove(it);
::KillTimer(runLoop->m_runLoopMessageWindow, ID);
}
timer->fired();
}
static uint64_t generateTimerID()
{
static uint64_t uniqueTimerID = 1;
return uniqueTimerID++;
}
RunLoop::TimerBase::TimerBase(RunLoop* runLoop)
: m_runLoop(runLoop)
, m_ID(generateTimerID())
, m_isRepeating(false)
{
}
RunLoop::TimerBase::~TimerBase()
{
stop();
}
void RunLoop::TimerBase::start(double nextFireInterval, bool repeat)
{
m_isRepeating = repeat;
m_runLoop->m_activeTimers.set(m_ID, this);
::SetTimer(m_runLoop->m_runLoopMessageWindow, m_ID, nextFireInterval * 1000, 0);
}
void RunLoop::TimerBase::stop()
{
TimerMap::iterator it = m_runLoop->m_activeTimers.find(m_ID);
if (it == m_runLoop->m_activeTimers.end())
return;
m_runLoop->m_activeTimers.remove(it);
::KillTimer(m_runLoop->m_runLoopMessageWindow, m_ID);
}
bool RunLoop::TimerBase::isActive() const
{
return m_runLoop->m_activeTimers.contains(m_ID);
}
}