#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->hwnd = NULL;
    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 64 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 449 of file CProgressDialog.cpp.

{
    return E_NOTIMPL;
}

◆ end_dialog()

void CProgressDialog::end_dialog ( )

Definition at line 141 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 441 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 103 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 345 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 414 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 391 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 386 of file CProgressDialog.cpp.

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

◆ SetProgress64()

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

override

Definition at line 362 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 329 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 272 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 319 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 433 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 110 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);
    }
}