Replace WTF::move with WTFMove
[WebKit-https.git] / Source / WTF / wtf / win / RunLoopWin.cpp
1 /*
2  * Copyright (C) 2010 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "RunLoop.h"
28
29 #include <wtf/CurrentTime.h>
30 #include <wtf/WindowsExtras.h>
31
32 namespace WTF {
33
34 static const UINT PerformWorkMessage = WM_USER + 1;
35 static const LPWSTR kRunLoopMessageWindowClassName = L"RunLoopMessageWindow";
36
37 LRESULT CALLBACK RunLoop::RunLoopWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
38 {
39     if (RunLoop* runLoop = static_cast<RunLoop*>(getWindowPointer(hWnd, 0)))
40         return runLoop->wndProc(hWnd, message, wParam, lParam);
41
42     if (message == WM_CREATE) {
43         LPCREATESTRUCT createStruct = reinterpret_cast<LPCREATESTRUCT>(lParam);
44
45         // Associate the RunLoop with the window.
46         setWindowPointer(hWnd, 0, createStruct->lpCreateParams);
47         return 0;
48     }
49
50     return ::DefWindowProc(hWnd, message, wParam, lParam);
51 }
52
53 LRESULT RunLoop::wndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
54 {
55     switch (message) {
56     case PerformWorkMessage:
57         performWork();
58         return 0;
59     case WM_TIMER:
60         RunLoop::TimerBase::timerFired(this, wParam);
61         return 0;
62     }
63
64     return ::DefWindowProc(hWnd, message, wParam, lParam);
65 }
66
67 void RunLoop::run()
68 {
69     MSG message;
70     while (BOOL result = ::GetMessage(&message, 0, 0, 0)) {
71         if (result == -1)
72             break;
73         ::TranslateMessage(&message);
74         ::DispatchMessage(&message);
75     }
76 }
77
78 void RunLoop::stop()
79 {
80     ::PostQuitMessage(0);
81 }
82
83 bool RunLoop::registerRunLoopMessageWindowClass()
84 {
85     // FIXME: This really only needs to be called once.
86
87     WNDCLASS windowClass = { 0 };
88     windowClass.lpfnWndProc     = RunLoop::RunLoopWndProc;
89     windowClass.cbWndExtra      = sizeof(RunLoop*);
90     windowClass.lpszClassName   = kRunLoopMessageWindowClassName;
91
92     return !!::RegisterClass(&windowClass);
93 }
94
95 RunLoop::RunLoop()
96 {
97     registerRunLoopMessageWindowClass();
98
99     m_runLoopMessageWindow = ::CreateWindow(kRunLoopMessageWindowClassName, 0, 0,
100         CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, HWND_MESSAGE, 0, 0, this);
101     ASSERT(::IsWindow(m_runLoopMessageWindow));
102 }
103
104 RunLoop::~RunLoop()
105 {
106     // FIXME: Tear down the work item queue here.
107 }
108
109 void RunLoop::wakeUp()
110 {
111     // FIXME: No need to wake up the run loop if we've already called dispatch
112     // before the run loop has had the time to respond.
113     ::PostMessage(m_runLoopMessageWindow, PerformWorkMessage, reinterpret_cast<WPARAM>(this), 0);
114 }
115
116 // RunLoop::Timer
117
118 void RunLoop::TimerBase::timerFired(RunLoop* runLoop, uint64_t ID)
119 {
120     TimerMap::iterator it = runLoop->m_activeTimers.find(ID);
121     if (it == runLoop->m_activeTimers.end()) {
122         // The timer must have been stopped after the WM_TIMER message was posted to the message queue.
123         return;
124     }
125
126     TimerBase* timer = it->value;
127
128     if (!timer->m_isRepeating) {
129         runLoop->m_activeTimers.remove(it);
130         ::KillTimer(runLoop->m_runLoopMessageWindow, ID);
131     }
132
133     timer->fired();
134 }
135
136 static uint64_t generateTimerID()
137 {
138     static uint64_t uniqueTimerID = 1;
139     return uniqueTimerID++;
140 }
141
142 RunLoop::TimerBase::TimerBase(RunLoop& runLoop)
143     : m_runLoop(runLoop)
144     , m_ID(generateTimerID())
145     , m_isRepeating(false)
146 {
147 }
148
149 RunLoop::TimerBase::~TimerBase()
150 {
151     stop();
152 }
153
154 void RunLoop::TimerBase::start(double nextFireInterval, bool repeat)
155 {
156     m_isRepeating = repeat;
157     m_runLoop.m_activeTimers.set(m_ID, this);
158     ::SetTimer(m_runLoop.m_runLoopMessageWindow, m_ID, nextFireInterval * 1000, 0);
159 }
160
161 void RunLoop::TimerBase::stop()
162 {
163     TimerMap::iterator it = m_runLoop.m_activeTimers.find(m_ID);
164     if (it == m_runLoop.m_activeTimers.end())
165         return;
166
167     m_runLoop.m_activeTimers.remove(it);
168     ::KillTimer(m_runLoop.m_runLoopMessageWindow, m_ID);
169 }
170
171 bool RunLoop::TimerBase::isActive() const
172 {
173     return m_runLoop.m_activeTimers.contains(m_ID);
174 }
175
176 } // namespace WTF