Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

243 lines
5.8 KiB

5 years ago
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================//
#include "resource.h"
#include "ProgressDialog.h"
#include "mxtk/mx.h"
#include "mdlviewer.h"
#include "tier1/utlstring.h"
#include "tier1/strtools.h"
#include "tier1/fmtstr.h"
#include <CommCtrl.h>
class CProgressDialog : public IProgressDialog
{
public:
CProgressDialog();
void Start( char const *pchTitle, char const *pchText, bool bShowCancel );
void Update( float flZeroToOneFraction );
void UpdateText( char const *pchFmt, ... );
bool IsCancelled();
void Finish();
static BOOL CALLBACK ProgressDialogProc( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam );
private:
BOOL ProgressDialogProcImpl( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam );
CUtlString m_sTitle;
CUtlString m_sStatus;
float m_flFraction;
bool m_bShowCancel;
bool m_bWantsCancel;
HWND m_hwndDlg;
double m_flStartTime;
};
static CProgressDialog g_ProgressDialog;
IProgressDialog *g_pProgressDialog = &g_ProgressDialog;
CProgressDialog::CProgressDialog() :
m_flFraction( 0.0f ), m_hwndDlg( 0 ), m_bShowCancel( false ), m_bWantsCancel( false ), m_flStartTime( 0.0f )
{
}
bool CProgressDialog::IsCancelled()
{
return m_bShowCancel && m_bWantsCancel;
}
//-----------------------------------------------------------------------------
// Purpose:
// Input : hwndDlg -
// uMsg -
// wParam -
// lParam -
// Output : static BOOL CALLBACK
//-----------------------------------------------------------------------------
BOOL CALLBACK CProgressDialog::ProgressDialogProc( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
{
return g_ProgressDialog.ProgressDialogProcImpl( hwndDlg, uMsg, wParam, lParam );
}
//-----------------------------------------------------------------------------
// Purpose:
// Input : *view -
// *actor -
// Output : int
//-----------------------------------------------------------------------------
void CProgressDialog::Start( char const *pchTitle, char const *pchText, bool bShowCancel )
{
if ( m_hwndDlg )
{
Finish();
}
Assert( NULL == m_hwndDlg );
m_sTitle = pchTitle;
m_sStatus = pchText;
m_flFraction = 0.0f;
m_bShowCancel = bShowCancel;
m_bWantsCancel = false;
m_flStartTime = Plat_FloatTime();
m_hwndDlg = CreateDialog( (HINSTANCE)GetModuleHandle( 0 ),
MAKEINTRESOURCE( IDD_PROGRESS ),
(HWND)g_MDLViewer->getHandle(),
(DLGPROC)ProgressDialogProc );
}
void CProgressDialog::Update( float flZeroToOneFraction )
{
m_flFraction = clamp( flZeroToOneFraction, 0.0f, 1.0f );
// Update the progress bar
HWND pb = GetDlgItem( m_hwndDlg, IDC_FP_PROGRESS );
int pos = clamp( (int)( 1000.0f * flZeroToOneFraction ), 0, 1000 );
SendMessage( pb, (UINT) PBM_SETPOS, pos, 0 );
HWND btn = GetDlgItem( m_hwndDlg, IDCANCEL );
LRESULT lr = SendMessage( btn, BM_GETSTATE, 0, 0 );
if ( lr & BST_PUSHED )
{
m_bWantsCancel = true;
}
if ( GetAsyncKeyState( VK_ESCAPE ) )
{
m_bWantsCancel = true;
}
mx::check();
}
void CProgressDialog::UpdateText( char const *pchFmt, ... )
{
char buf[ 2048 ];
va_list argptr;
va_start( argptr, pchFmt );
Q_vsnprintf( buf, sizeof( buf ), pchFmt, argptr );
va_end( argptr );
m_sStatus = buf;
SetDlgItemText( m_hwndDlg, IDC_FP_PROGRESS_TEXT, CFmtStr( "%s", m_sStatus.String() ) );
SetDlgItemText( m_hwndDlg, IDC_FP_PROGRESS_PERCENT, CFmtStr( "%.2f %%", m_flFraction * 100.0f ) );
double elapsed = Plat_FloatTime() - m_flStartTime;
double flPercentagePerSecond = 0.0f;
if ( m_flFraction > 0.0f )
{
flPercentagePerSecond = elapsed / m_flFraction;
}
double flSecondsRemaining = flPercentagePerSecond * ( 1.0f - m_flFraction );
int seconds = (int)flSecondsRemaining;
CFmtStr string;
int hours = 0;
int minutes = seconds / 60;
if ( minutes > 0 )
{
seconds -= (minutes * 60);
hours = minutes / 60;
if ( hours > 0 )
{
minutes -= (hours * 60);
}
}
if ( hours > 0 )
{
string.sprintf( "Time Remaining: %2i:%02i:%02i", hours, minutes, seconds );
}
else
{
string.sprintf( "Time Remaining: %02i:%02i", minutes, seconds );
}
SetDlgItemText( m_hwndDlg, IDC_FP_PROGRESS_ETA, string.Access() );
}
void CProgressDialog::Finish()
{
if ( !m_hwndDlg )
return;
DestroyWindow( m_hwndDlg );
m_hwndDlg = NULL;
}
BOOL CProgressDialog::ProgressDialogProcImpl( HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
{
switch(uMsg)
{
case WM_INITDIALOG:
// Insert code here to put the string (to find and replace with)
// into the edit controls.
// ...
{
RECT rcDlg;
GetWindowRect( hwndDlg, &rcDlg );
// Get relative to primary monitor instead of actual window parent
RECT rcParent;
rcParent.left = 0;
rcParent.right = rcParent.left + GetSystemMetrics( SM_CXFULLSCREEN );
rcParent.top = 0;
rcParent.bottom = rcParent.top + GetSystemMetrics( SM_CYFULLSCREEN );
int dialogw, dialogh;
int parentw, parenth;
parentw = rcParent.right - rcParent.left;
parenth = rcParent.bottom - rcParent.top;
dialogw = rcDlg.right - rcDlg.left;
dialogh = rcDlg.bottom - rcDlg.top;
int dlgleft, dlgtop;
dlgleft = ( parentw - dialogw ) / 2;
dlgtop = ( parenth - dialogh ) / 2;
MoveWindow( hwndDlg,
dlgleft,
dlgtop,
dialogw,
dialogh,
TRUE
);
SetDlgItemText( hwndDlg, IDC_FP_PROGRESS_TITLE, m_sTitle.String() );
SetDlgItemText( hwndDlg, IDC_FP_PROGRESS_TEXT, m_sStatus.String() );
HWND pb = GetDlgItem( hwndDlg, IDC_FP_PROGRESS );
SendMessage( pb, (UINT) PBM_SETRANGE, 0, MAKELPARAM( 0, 1000 ) );
ShowWindow( GetDlgItem( hwndDlg, IDCANCEL ), m_bShowCancel ? SW_SHOW : SW_HIDE );
Update( 0.0f );
}
return FALSE;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDCANCEL:
m_bWantsCancel = true;
break;
}
return TRUE;
}
return FALSE;
}