4 * Copyright 2007 Mikolaj Zalewski
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 #include "wine/debug.h"
39 #include "wine/unicode.h"
44 WINE_DEFAULT_DEBUG_CHANNEL(browseui
);
46 #define CANCEL_MSG_LINE 2
48 /* Note: to avoid a deadlock we don't want to send messages to the dialog
49 * with the critical section held. Instead we only mark what fields should be
50 * updated and the dialog proc does the update */
51 #define UPDATE_PROGRESS 0x1
52 #define UPDATE_TITLE 0x2
53 #define UPDATE_LINE1 0x4
54 #define UPDATE_LINE2 (UPDATE_LINE1<<1)
55 #define UPDATE_LINE3 (UPDATE_LINE2<<2)
58 #define WM_DLG_UPDATE (WM_APP+1) /* set to the dialog when it should update */
59 #define WM_DLG_DESTROY (WM_APP+2) /* DestroyWindow must be called from the owning thread */
61 typedef struct tagProgressDialog
{
62 const IProgressDialogVtbl
*vtbl
;
72 ULONGLONG ullCompleted
;
74 HWND hwndDisabledParent
; /* For modal dialog: the parent that need to be re-enabled when the dialog ends */
77 static void set_buffer(LPWSTR
*buffer
, LPCWSTR string
)
79 static const WCHAR empty_string
[] = {0};
84 string
= empty_string
;
85 CoGetMalloc(1, &malloc
);
87 cb
= (strlenW(string
) + 1)*sizeof(WCHAR
);
88 if (*buffer
== NULL
|| cb
> IMalloc_GetSize(malloc
, *buffer
))
89 *buffer
= IMalloc_Realloc(malloc
, *buffer
, cb
);
90 memcpy(*buffer
, string
, cb
);
100 static LPWSTR
load_string(HINSTANCE hInstance
, UINT uiResourceId
)
105 LoadStringW(hInstance
, uiResourceId
, string
, sizeof(string
)/sizeof(string
[0]));
106 ret
= HeapAlloc(GetProcessHeap(), 0, (strlenW(string
) + 1) * sizeof(WCHAR
));
107 strcpyW(ret
, string
);
111 static void set_progress_marquee(ProgressDialog
*This
)
113 HWND hProgress
= GetDlgItem(This
->hwnd
, IDC_PROGRESS_BAR
);
114 SetWindowLongW(hProgress
, GWL_STYLE
,
115 GetWindowLongW(hProgress
, GWL_STYLE
)|PBS_MARQUEE
);
118 void update_dialog(ProgressDialog
*This
, DWORD dwUpdate
)
122 if (dwUpdate
& UPDATE_TITLE
)
123 SetWindowTextW(This
->hwnd
, This
->title
);
125 if (dwUpdate
& UPDATE_LINE1
)
126 SetDlgItemTextW(This
->hwnd
, IDC_TEXT_LINE
, (This
->isCancelled
? empty
: This
->lines
[0]));
127 if (dwUpdate
& UPDATE_LINE2
)
128 SetDlgItemTextW(This
->hwnd
, IDC_TEXT_LINE
+1, (This
->isCancelled
? empty
: This
->lines
[1]));
129 if (dwUpdate
& UPDATE_LINE3
)
130 SetDlgItemTextW(This
->hwnd
, IDC_TEXT_LINE
+2, (This
->isCancelled
? This
->cancelMsg
: This
->lines
[2]));
132 if (dwUpdate
& UPDATE_PROGRESS
)
134 ULONGLONG ullTotal
= This
->ullTotal
;
135 ULONGLONG ullCompleted
= This
->ullCompleted
;
137 /* progress bar requires 32-bit coordinates */
138 while (ullTotal
>> 32)
144 SendDlgItemMessageW(This
->hwnd
, IDC_PROGRESS_BAR
, PBM_SETRANGE32
, 0, (DWORD
)ullTotal
);
145 SendDlgItemMessageW(This
->hwnd
, IDC_PROGRESS_BAR
, PBM_SETPOS
, (DWORD
)ullCompleted
, 0);
149 static void end_dialog(ProgressDialog
*This
)
151 SendMessageW(This
->hwnd
, WM_DLG_DESTROY
, 0, 0);
152 /* native doesn't reenable the window? */
153 if (This
->hwndDisabledParent
)
154 EnableWindow(This
->hwndDisabledParent
, TRUE
);
158 static INT_PTR CALLBACK
dialog_proc(HWND hwnd
, UINT msg
, WPARAM wParam
, LPARAM lParam
)
160 ProgressDialog
*This
= (ProgressDialog
*)GetWindowLongPtrW(hwnd
, DWLP_USER
);
166 struct create_params
*params
= (struct create_params
*)lParam
;
168 /* Note: until we set the hEvent, the object is protected by
169 * the critical section held by StartProgress */
170 SetWindowLongPtrW(hwnd
, DWLP_USER
, (LONG_PTR
)params
->This
);
174 if (This
->dwFlags
& PROGDLG_NOPROGRESSBAR
)
175 ShowWindow(GetDlgItem(hwnd
, IDC_PROGRESS_BAR
), SW_HIDE
);
176 if (This
->dwFlags
& PROGDLG_NOCANCEL
)
177 ShowWindow(GetDlgItem(hwnd
, IDCANCEL
), SW_HIDE
);
178 if (This
->dwFlags
& PROGDLG_MARQUEEPROGRESS
)
179 set_progress_marquee(This
);
180 if (This
->dwFlags
& PROGDLG_NOMINIMIZE
)
181 SetWindowLongW(hwnd
, GWL_STYLE
, GetWindowLongW(hwnd
, GWL_STYLE
) & (~WS_MINIMIZEBOX
));
183 update_dialog(This
, 0xffffffff);
185 This
->isCancelled
= FALSE
;
186 SetEvent(params
->hEvent
);
191 EnterCriticalSection(&This
->cs
);
192 update_dialog(This
, This
->dwUpdate
);
194 LeaveCriticalSection(&This
->cs
);
199 PostThreadMessageW(GetCurrentThreadId(), WM_NULL
, 0, 0); /* wake up the GetMessage */
204 if (msg
== WM_CLOSE
|| wParam
== IDCANCEL
)
206 EnterCriticalSection(&This
->cs
);
207 This
->isCancelled
= TRUE
;
209 if (!This
->cancelMsg
)
210 This
->cancelMsg
= load_string(BROWSEUI_hinstance
, IDS_CANCELLING
);
212 set_progress_marquee(This
);
213 EnableWindow(GetDlgItem(This
->hwnd
, IDCANCEL
), FALSE
);
214 update_dialog(This
, UPDATE_LINE1
|UPDATE_LINE2
|UPDATE_LINE3
);
215 LeaveCriticalSection(&This
->cs
);
222 static DWORD WINAPI
dialog_thread(LPVOID lpParameter
)
224 /* Note: until we set the hEvent in WM_INITDIALOG, the ProgressDialog object
225 * is protected by the critical section held by StartProgress */
226 struct create_params
*params
= (struct create_params
*)lpParameter
;
230 hwnd
= CreateDialogParamW(BROWSEUI_hinstance
, MAKEINTRESOURCEW(IDD_PROGRESS_DLG
),
231 params
->hwndParent
, dialog_proc
, (LPARAM
)params
);
233 while (GetMessageW(&msg
, NULL
, 0, 0) > 0)
237 if(!IsDialogMessageW(hwnd
, &msg
))
239 TranslateMessage(&msg
);
240 DispatchMessageW(&msg
);
247 static void ProgressDialog_Destructor(ProgressDialog
*This
)
249 TRACE("destroying %p\n", This
);
252 heap_free(This
->lines
[0]);
253 heap_free(This
->lines
[1]);
254 heap_free(This
->lines
[2]);
255 heap_free(This
->cancelMsg
);
256 heap_free(This
->title
);
261 static HRESULT WINAPI
ProgressDialog_QueryInterface(IProgressDialog
*iface
, REFIID iid
, LPVOID
*ppvOut
)
263 ProgressDialog
*This
= (ProgressDialog
*)iface
;
266 if (IsEqualIID(iid
, &IID_IUnknown
) || IsEqualIID(iid
, &IID_IProgressDialog
))
273 IUnknown_AddRef(iface
);
277 WARN("unsupported interface: %s\n", debugstr_guid(iid
));
278 return E_NOINTERFACE
;
281 static ULONG WINAPI
ProgressDialog_AddRef(IProgressDialog
*iface
)
283 ProgressDialog
*This
= (ProgressDialog
*)iface
;
284 return InterlockedIncrement(&This
->refCount
);
287 static ULONG WINAPI
ProgressDialog_Release(IProgressDialog
*iface
)
289 ProgressDialog
*This
= (ProgressDialog
*)iface
;
292 ret
= InterlockedDecrement(&This
->refCount
);
294 ProgressDialog_Destructor(This
);
298 static HRESULT WINAPI
ProgressDialog_StartProgressDialog(IProgressDialog
*iface
, HWND hwndParent
, IUnknown
*punkEnableModeless
, DWORD dwFlags
, LPCVOID reserved
)
300 ProgressDialog
*This
= (ProgressDialog
*)iface
;
301 struct create_params params
;
304 TRACE("(%p, %p, %x, %p)\n", iface
, punkEnableModeless
, dwFlags
, reserved
);
305 if (punkEnableModeless
|| reserved
)
306 FIXME("Reserved parameters not null (%p, %p)\n", punkEnableModeless
, reserved
);
307 if (dwFlags
& PROGDLG_AUTOTIME
)
308 FIXME("Flags PROGDLG_AUTOTIME not supported\n");
309 if (dwFlags
& PROGDLG_NOTIME
)
310 FIXME("Flags PROGDLG_NOTIME not supported\n");
312 EnterCriticalSection(&This
->cs
);
316 LeaveCriticalSection(&This
->cs
);
317 return S_OK
; /* as on XP */
319 This
->dwFlags
= dwFlags
;
321 params
.hwndParent
= hwndParent
;
322 params
.hEvent
= CreateEventW(NULL
, TRUE
, FALSE
, NULL
);
324 hThread
= CreateThread(NULL
, 0, dialog_thread
, ¶ms
, 0, NULL
);
325 WaitForSingleObject(params
.hEvent
, INFINITE
);
327 This
->hwndDisabledParent
= NULL
;
328 if (hwndParent
&& (dwFlags
& PROGDLG_MODAL
))
330 HWND hwndDisable
= GetAncestor(hwndParent
, GA_ROOT
);
331 if (EnableWindow(hwndDisable
, FALSE
))
332 This
->hwndDisabledParent
= hwndDisable
;
335 LeaveCriticalSection(&This
->cs
);
340 static HRESULT WINAPI
ProgressDialog_StopProgressDialog(IProgressDialog
*iface
)
342 ProgressDialog
*This
= (ProgressDialog
*)iface
;
344 EnterCriticalSection(&This
->cs
);
347 LeaveCriticalSection(&This
->cs
);
352 static HRESULT WINAPI
ProgressDialog_SetTitle(IProgressDialog
*iface
, LPCWSTR pwzTitle
)
354 ProgressDialog
*This
= (ProgressDialog
*)iface
;
357 TRACE("(%p, %s)\n", This
, wine_dbgstr_w(pwzTitle
));
359 EnterCriticalSection(&This
->cs
);
360 set_buffer(&This
->title
, pwzTitle
);
361 This
->dwUpdate
|= UPDATE_TITLE
;
363 LeaveCriticalSection(&This
->cs
);
366 SendMessageW(hwnd
, WM_DLG_UPDATE
, 0, 0);
371 static HRESULT WINAPI
ProgressDialog_SetAnimation(IProgressDialog
*iface
, HINSTANCE hInstance
, UINT uiResourceId
)
373 FIXME("(%p, %p, %d) - stub\n", iface
, hInstance
, uiResourceId
);
377 static BOOL WINAPI
ProgressDialog_HasUserCancelled(IProgressDialog
*iface
)
379 ProgressDialog
*This
= (ProgressDialog
*)iface
;
380 return This
->isCancelled
;
383 static HRESULT WINAPI
ProgressDialog_SetProgress64(IProgressDialog
*iface
, ULONGLONG ullCompleted
, ULONGLONG ullTotal
)
385 ProgressDialog
*This
= (ProgressDialog
*)iface
;
388 TRACE("(%p, 0x%s, 0x%s)\n", This
, wine_dbgstr_longlong(ullCompleted
), wine_dbgstr_longlong(ullTotal
));
390 EnterCriticalSection(&This
->cs
);
391 This
->ullTotal
= ullTotal
;
392 This
->ullCompleted
= ullCompleted
;
393 This
->dwUpdate
|= UPDATE_PROGRESS
;
395 LeaveCriticalSection(&This
->cs
);
398 SendMessageW(hwnd
, WM_DLG_UPDATE
, 0, 0);
400 return S_OK
; /* Windows sometimes returns S_FALSE */
403 static HRESULT WINAPI
ProgressDialog_SetProgress(IProgressDialog
*iface
, DWORD dwCompleted
, DWORD dwTotal
)
405 return IProgressDialog_SetProgress64(iface
, dwCompleted
, dwTotal
);
408 static HRESULT WINAPI
ProgressDialog_SetLine(IProgressDialog
*iface
, DWORD dwLineNum
, LPCWSTR pwzLine
, BOOL bPath
, LPCVOID reserved
)
410 ProgressDialog
*This
= (ProgressDialog
*)iface
;
413 TRACE("(%p, %d, %s, %d)\n", This
, dwLineNum
, wine_dbgstr_w(pwzLine
), bPath
);
416 FIXME("reserved pointer not null (%p)\n", reserved
);
419 if (dwLineNum
>= 3) /* Windows seems to do something like that */
422 EnterCriticalSection(&This
->cs
);
423 set_buffer(&This
->lines
[dwLineNum
], pwzLine
);
424 This
->dwUpdate
|= UPDATE_LINE1
<< dwLineNum
;
425 hwnd
= (This
->isCancelled
? NULL
: This
->hwnd
); /* no sense to send the message if window cancelled */
426 LeaveCriticalSection(&This
->cs
);
429 SendMessageW(hwnd
, WM_DLG_UPDATE
, 0, 0);
434 static HRESULT WINAPI
ProgressDialog_SetCancelMsg(IProgressDialog
*iface
, LPCWSTR pwzMsg
, LPCVOID reserved
)
436 ProgressDialog
*This
= (ProgressDialog
*)iface
;
439 TRACE("(%p, %s)\n", This
, wine_dbgstr_w(pwzMsg
));
442 FIXME("reserved pointer not null (%p)\n", reserved
);
444 EnterCriticalSection(&This
->cs
);
445 set_buffer(&This
->cancelMsg
, pwzMsg
);
446 This
->dwUpdate
|= UPDATE_LINE1
<< CANCEL_MSG_LINE
;
447 hwnd
= (This
->isCancelled
? This
->hwnd
: NULL
); /* no sense to send the message if window not cancelled */
448 LeaveCriticalSection(&This
->cs
);
451 SendMessageW(hwnd
, WM_DLG_UPDATE
, 0, 0);
456 static HRESULT WINAPI
ProgressDialog_Timer(IProgressDialog
*iface
, DWORD dwTimerAction
, LPCVOID reserved
)
458 ProgressDialog
*This
= (ProgressDialog
*)iface
;
460 FIXME("(%p, %d, %p) - stub\n", This
, dwTimerAction
, reserved
);
463 FIXME("Reserved field not NULL but %p\n", reserved
);
468 static const IProgressDialogVtbl ProgressDialogVtbl
=
470 ProgressDialog_QueryInterface
,
471 ProgressDialog_AddRef
,
472 ProgressDialog_Release
,
474 ProgressDialog_StartProgressDialog
,
475 ProgressDialog_StopProgressDialog
,
476 ProgressDialog_SetTitle
,
477 ProgressDialog_SetAnimation
,
478 ProgressDialog_HasUserCancelled
,
479 ProgressDialog_SetProgress
,
480 ProgressDialog_SetProgress64
,
481 ProgressDialog_SetLine
,
482 ProgressDialog_SetCancelMsg
,
486 HRESULT
ProgressDialog_Constructor(IUnknown
*pUnkOuter
, IUnknown
**ppOut
)
488 ProgressDialog
*This
;
490 return CLASS_E_NOAGGREGATION
;
492 This
= heap_alloc_zero(sizeof(ProgressDialog
));
494 return E_OUTOFMEMORY
;
496 This
->vtbl
= &ProgressDialogVtbl
;
498 InitializeCriticalSection(&This
->cs
);
500 TRACE("returning %p\n", This
);
501 *ppOut
= (IUnknown
*)This
;