#include <CProgressDialog.h>

Inheritance diagram for CProgressDialog:

[legend]

Collaboration diagram for CProgressDialog:

[legend]

Classes
struct	progressMark

Public Member Functions
void	set_progress_marquee ()

void	update_dialog (DWORD dwUpdate)

void	end_dialog ()

	CProgressDialog ()

	~CProgressDialog ()

STDMETHOD()	StartProgressDialog (HWND hwndParent, IUnknown *punkEnableModeless, DWORD dwFlags, LPCVOID reserved) override

STDMETHOD()	StopProgressDialog () override

STDMETHOD()	SetTitle (LPCWSTR pwzTitle) override

STDMETHOD()	SetAnimation (HINSTANCE hInstance, UINT uiResourceId) override

	STDMETHOD_ (BOOL, HasUserCancelled)() override

STDMETHOD()	SetProgress64 (ULONGLONG ullCompleted, ULONGLONG ullTotal) override

STDMETHOD()	SetProgress (DWORD dwCompleted, DWORD dwTotal) override

STDMETHOD()	SetLine (DWORD dwLineNum, LPCWSTR pwzLine, BOOL bPath, LPCVOID reserved) override

STDMETHOD()	SetCancelMsg (LPCWSTR pwzMsg, LPCVOID reserved) override

STDMETHOD()	Timer (DWORD dwTimerAction, LPCVOID reserved) override

STDMETHOD()	GetWindow (HWND *phwnd) override

STDMETHOD()	ContextSensitiveHelp (BOOL fEnterMode) override

Public Member Functions inherited from ATL::CComObjectRootEx< CComMultiThreadModelNoCS >
	~CComObjectRootEx ()

ULONG	InternalAddRef ()

ULONG	InternalRelease ()

void	Lock ()

void	Unlock ()

HRESULT	_AtlInitialConstruct ()

Public Member Functions inherited from ATL::CComObjectRootBase
	CComObjectRootBase ()

	~CComObjectRootBase ()

void	SetVoid (void *)

HRESULT	_AtlFinalConstruct ()

HRESULT	FinalConstruct ()

void	InternalFinalConstructAddRef ()

void	InternalFinalConstructRelease ()

void	FinalRelease ()

HRESULT	GetWindow ([out] HWND *phwnd)

HRESULT	ContextSensitiveHelp ([in] BOOL fEnterMode)

Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)

ULONG	AddRef ()

ULONG	Release ()

Public Attributes
CRITICAL_SECTION	cs

HWND	hwnd

DWORD	dwFlags

DWORD	dwUpdate

LPWSTR	lines [3]

LPWSTR	cancelMsg

LPWSTR	title

BOOL	isCancelled

ULONGLONG	ullCompleted

ULONGLONG	ullTotal

HWND	hwndDisabledParent

UINT	clockHand

progressMark	progressClock [30]

DWORD	dwStartTime

Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef

Additional Inherited Members
Public Types inherited from IOleWindow
typedef IOleWindow *	LPOLEWINDOW

Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN

Static Public Member Functions inherited from ATL::CComObjectRootBase
static void WINAPI	ObjectMain (bool)

static const struct _ATL_CATMAP_ENTRY *	GetCategoryMap ()

static HRESULT WINAPI	InternalQueryInterface (void pThis, const _ATL_INTMAP_ENTRY pEntries, REFIID iid, void **ppvObject)

Detailed Description

Definition at line 24 of file CProgressDialog.h.

Constructor & Destructor Documentation

◆ CProgressDialog()

CProgressDialog::CProgressDialog ( )

Definition at line 45 of file CProgressDialog.cpp.

{
    this->lines[0]  = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
    this->lines[1]  = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
    this->lines[2]  = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
    this->cancelMsg = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
    this->title     = (LPWSTR) HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
 
    this->lines[0][0] = this->lines[1][0] = this->lines[2][0] = UNICODE_NULL;
    this->cancelMsg[0] = this->title[0] = UNICODE_NULL;
 
    this->clockHand = -1;
    this->progressClock[29].ullMark = 0ull;
    this->dwStartTime = GetTickCount();
 
    InitializeCriticalSection(&this->cs);
}

◆ ~CProgressDialog()

CProgressDialog::~CProgressDialog ( )

Definition at line 63 of file CProgressDialog.cpp.

{
    if (this->hwnd)
        this->end_dialog();
    HeapFree(GetProcessHeap(), 0, this->lines[0]);
    HeapFree(GetProcessHeap(), 0, this->lines[1]);
    HeapFree(GetProcessHeap(), 0, this->lines[2]);
    HeapFree(GetProcessHeap(), 0, this->cancelMsg);
    HeapFree(GetProcessHeap(), 0, this->title);
    DeleteCriticalSection(&this->cs);
}

Member Function Documentation

◆ ContextSensitiveHelp()

HRESULT WINAPI CProgressDialog::ContextSensitiveHelp ( BOOL fEnterMode )

override

Implements IOleWindow.

Definition at line 448 of file CProgressDialog.cpp.

{
    return E_NOTIMPL;
}

◆ end_dialog()

void CProgressDialog::end_dialog ( )

Definition at line 140 of file CProgressDialog.cpp.

{
    SendMessageW(this->hwnd, WM_DLG_DESTROY, 0, 0);
    /* native doesn't re-enable the window? */
    if (this->hwndDisabledParent)
        EnableWindow(this->hwndDisabledParent, TRUE);
    this->hwnd = NULL;
}

Referenced by StopProgressDialog(), and ~CProgressDialog().

◆ GetWindow()

HRESULT WINAPI CProgressDialog::GetWindow ( HWND * phwnd )

override

Implements IOleWindow.

Definition at line 440 of file CProgressDialog.cpp.

{
    EnterCriticalSection(&this->cs);
    *phwnd = this->hwnd;
    LeaveCriticalSection(&this->cs);
    return S_OK;
}

◆ set_progress_marquee()

void CProgressDialog::set_progress_marquee ( )

Definition at line 102 of file CProgressDialog.cpp.

{
    HWND hProgress = GetDlgItem(this->hwnd, IDC_PROGRESS_BAR);
    SetWindowLongW(hProgress, GWL_STYLE,
        GetWindowLongW(hProgress, GWL_STYLE)|PBS_MARQUEE);
}

◆ SetAnimation()

HRESULT WINAPI CProgressDialog::SetAnimation	(	HINSTANCE	hInstance,
		UINT	uiResourceId
	)

override

Definition at line 344 of file CProgressDialog.cpp.

{
    HWND hAnimation = GetDlgItem(this->hwnd, IDD_PROGRESS_DLG);
    SetWindowLongW(hAnimation, GWL_STYLE,
        GetWindowLongW(hAnimation, GWL_STYLE)|ACS_TRANSPARENT|ACS_CENTER|ACS_AUTOPLAY);
 
    if(!Animate_OpenEx(hAnimation,hInstance,MAKEINTRESOURCEW(uiResourceId)))
        return S_FALSE;
 
    return S_OK;
}

◆ SetCancelMsg()

HRESULT WINAPI CProgressDialog::SetCancelMsg	(	LPCWSTR	pwzMsg,
		LPCVOID	reserved
	)

override

Definition at line 413 of file CProgressDialog.cpp.

{
    HWND hwnd;
 
    if (reserved)
        FIXME("reserved pointer not null (%p)\n", reserved);
 
    EnterCriticalSection(&this->cs);
    set_buffer(&this->cancelMsg, pwzMsg);
    this->dwUpdate |= UPDATE_LINE1 << CANCEL_MSG_LINE;
    hwnd = (this->isCancelled ? this->hwnd : NULL); /* no sense to send the message if window not cancelled */
    LeaveCriticalSection(&this->cs);
 
    if (hwnd)
        SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 
    return S_OK;
}

◆ SetLine()

HRESULT WINAPI CProgressDialog::SetLine	(	DWORD	dwLineNum,
		LPCWSTR	pwzLine,
		BOOL	bPath,
		LPCVOID	reserved
	)

override

Definition at line 390 of file CProgressDialog.cpp.

{
    HWND hwnd;
 
    if (reserved)
        FIXME("reserved pointer not null (%p)\n", reserved);
 
    dwLineNum--;
    if (dwLineNum >= 3)  /* Windows seems to do something like that */
        dwLineNum = 0;
 
    EnterCriticalSection(&this->cs);
    set_buffer(&this->lines[dwLineNum], pwzLine);
    this->dwUpdate |= UPDATE_LINE1 << dwLineNum;
    hwnd = (this->isCancelled ? NULL : this->hwnd); /* no sense to send the message if window cancelled */
    LeaveCriticalSection(&this->cs);
 
    if (hwnd)
        SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 
    return S_OK;
}

◆ SetProgress()

HRESULT WINAPI CProgressDialog::SetProgress	(	DWORD	dwCompleted,
		DWORD	dwTotal
	)

override

Definition at line 385 of file CProgressDialog.cpp.

{
    return this->SetProgress64(dwCompleted, dwTotal);
}

◆ SetProgress64()

HRESULT WINAPI CProgressDialog::SetProgress64	(	ULONGLONG	ullCompleted,
		ULONGLONG	ullTotal
	)

override

Definition at line 361 of file CProgressDialog.cpp.

{
    HWND hwnd;
 
    EnterCriticalSection(&this->cs);
    this->ullTotal = ullTotal;
    this->ullCompleted = ullCompleted;
 
    if (GetTickCount() - this->progressClock[(this->clockHand + 29) % 30].dwTime > 20) {
        this->clockHand = (this->clockHand + 1) % 30;
        this->progressClock[this->clockHand].ullMark = ullCompleted;
        this->progressClock[this->clockHand].dwTime = GetTickCount();
    }
 
    this->dwUpdate |= UPDATE_PROGRESS;
    hwnd = this->hwnd;
    LeaveCriticalSection(&this->cs);
 
    if (hwnd)
        SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 
    return S_OK;  /* Windows sometimes returns S_FALSE */
}

Referenced by SetProgress().

◆ SetTitle()

HRESULT WINAPI CProgressDialog::SetTitle ( LPCWSTR pwzTitle )

override

Definition at line 328 of file CProgressDialog.cpp.

{
    HWND hwnd;
 
    EnterCriticalSection(&this->cs);
    set_buffer(&this->title, pwzTitle);
    this->dwUpdate |= UPDATE_TITLE;
    hwnd = this->hwnd;
    LeaveCriticalSection(&this->cs);
 
    if (hwnd)
        SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 
    return S_OK;
}

◆ StartProgressDialog()

HRESULT WINAPI CProgressDialog::StartProgressDialog	(	HWND	hwndParent,
		IUnknown *	punkEnableModeless,
		DWORD	dwFlags,
		LPCVOID	reserved
	)

override

Definition at line 271 of file CProgressDialog.cpp.

{
    static const INITCOMMONCONTROLSEX init = { sizeof(init), ICC_ANIMATE_CLASS };
 
    struct create_params params;
    HANDLE hThread;
 
    // TRACE("(%p, %p, %x, %p)\n", this, punkEnableModeless, dwFlags, reserved);
    if (punkEnableModeless || reserved)
        FIXME("Reserved parameters not null (%p, %p)\n", punkEnableModeless, reserved);
    if (dwFlags & PROGDLG_AUTOTIME)
        FIXME("Flags PROGDLG_AUTOTIME not supported\n");
    if (dwFlags & PROGDLG_NOTIME)
        FIXME("Flags PROGDLG_NOTIME not supported\n");
 
    InitCommonControlsEx( &init );
 
    EnterCriticalSection(&this->cs);
 
    if (this->hwnd)
    {
        LeaveCriticalSection(&this->cs);
        return S_OK;  /* as on XP */
    }
    this->dwFlags = dwFlags;
    params.This = this;
    params.hwndParent = hwndParent;
    params.hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
 
    hThread = CreateThread(NULL, 0, dialog_thread, &params, 0, NULL);
    WaitForSingleObject(params.hEvent, INFINITE);
    CloseHandle(params.hEvent);
    CloseHandle(hThread);
 
    this->hwndDisabledParent = NULL;
    if (hwndParent && (dwFlags & PROGDLG_MODAL))
    {
        HWND hwndDisable = GetAncestor(hwndParent, GA_ROOT);
        if (EnableWindow(hwndDisable, FALSE))
            this->hwndDisabledParent = hwndDisable;
    }
 
    LeaveCriticalSection(&this->cs);
 
    return S_OK;
}

◆ STDMETHOD_()

CProgressDialog::STDMETHOD_	(	BOOL	,
		HasUserCancelled
	)

override

◆ StopProgressDialog()

HRESULT WINAPI CProgressDialog::StopProgressDialog ( )

override

Definition at line 318 of file CProgressDialog.cpp.

{
    EnterCriticalSection(&this->cs);
    if (this->hwnd)
        this->end_dialog();
    LeaveCriticalSection(&this->cs);
 
    return S_OK;
}

◆ Timer()

HRESULT WINAPI CProgressDialog::Timer	(	DWORD	dwTimerAction,
		LPCVOID	reserved
	)

override

Definition at line 432 of file CProgressDialog.cpp.

{
     if (reserved)
         FIXME("Reserved field not NULL but %p\n", reserved);
 
    return S_OK;
}

◆ update_dialog()

void CProgressDialog::update_dialog ( DWORD dwUpdate )

Definition at line 109 of file CProgressDialog.cpp.

{
    WCHAR empty[] = {0};
 
    if (dwUpdate & UPDATE_TITLE)
        SetWindowTextW(this->hwnd, this->title);
 
    if (dwUpdate & UPDATE_LINE1)
        SetDlgItemTextW(this->hwnd, IDC_TEXT_LINE, (this->isCancelled ? empty : this->lines[0]));
    if (dwUpdate & UPDATE_LINE2)
        SetDlgItemTextW(this->hwnd, IDC_TEXT_LINE+1, (this->isCancelled ? empty : this->lines[1]));
    if (dwUpdate & UPDATE_LINE3)
        SetDlgItemTextW(this->hwnd, IDC_TEXT_LINE+2, (this->isCancelled ? this->cancelMsg : this->lines[2]));
 
    if (dwUpdate & UPDATE_PROGRESS)
    {
        ULONGLONG ullTotal = this->ullTotal;
        ULONGLONG ullCompleted = this->ullCompleted;
 
        /* progress bar requires 32-bit coordinates */
        while (ullTotal >> 32)
        {
            ullTotal >>= 1;
            ullCompleted >>= 1;
        }
 
        SendDlgItemMessageW(this->hwnd, IDC_PROGRESS_BAR, PBM_SETRANGE32, 0, (DWORD)ullTotal);
        SendDlgItemMessageW(this->hwnd, IDC_PROGRESS_BAR, PBM_SETPOS, (DWORD)ullCompleted, 0);
    }
}