340 lines
9.9 KiB
C++
340 lines
9.9 KiB
C++
#include "./uiEraseMedium.h"
|
|
#include "api.h"
|
|
#include <api/service/waservicefactory.h>
|
|
#include "./resource.h"
|
|
#include <strsafe.h>
|
|
#include <commctrl.h>
|
|
|
|
|
|
#include "./uiCheckMedium.h"
|
|
|
|
#define ERASETIME_QUICKMODE 50 // in seconds
|
|
#define ERASETIME_FULLMODE 510 // in seconds
|
|
|
|
#define TIMER_CLOCK_ID 1979
|
|
#define TIMER_CLOCK_INTERVAL 1000
|
|
|
|
EraseMediumUI::EraseMediumUI(void)
|
|
{
|
|
eraseMedium = NULL;
|
|
|
|
}
|
|
|
|
EraseMediumUI::~EraseMediumUI(void)
|
|
{
|
|
if (eraseMedium)
|
|
{
|
|
delete(eraseMedium);
|
|
eraseMedium = NULL;
|
|
}
|
|
}
|
|
DWORD EraseMediumUI::SetEject(int ejectmode)
|
|
{
|
|
return (eraseMedium) ? eraseMedium->SetEject(ejectmode) : 0;
|
|
}
|
|
|
|
DWORD EraseMediumUI::Erase(DWORD drive, BOOL discCheck, HWND ownerWnd)
|
|
{
|
|
|
|
this->drive = drive;
|
|
this->discCheck = discCheck;
|
|
DWORD retCode;
|
|
LPCDLGTEMPLATE templ = NULL;
|
|
|
|
HRSRC hres = FindResourceExW(hResource, MAKEINTRESOURCEW(5), MAKEINTRESOURCEW(IDD_DLG_ERASEMEDIUMPREPARE), MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL));
|
|
templ = (hres) ? (LPCDLGTEMPLATE)LoadResource(hResource, hres) : NULL;
|
|
retCode = (DWORD)DialogBoxIndirectParamW(dllInstance, templ, ownerWnd, (DLGPROC)PrepareWndProc, (LPARAM)this);
|
|
if (ERASEMEDIUMUI_OK != retCode) return retCode;
|
|
|
|
hres = FindResourceExW(hResource, MAKEINTRESOURCEW(5), MAKEINTRESOURCEW(IDD_DLG_ERASEMEDIUMSTATUS), MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL));
|
|
templ = (hres) ? (LPCDLGTEMPLATE)LoadResource(hResource, hres) : NULL;
|
|
retCode = (DWORD)DialogBoxIndirectParamW(dllInstance, templ, ownerWnd, (DLGPROC)EraseWndProc, (LPARAM)this);
|
|
if (ERASEMEDIUM_ABORTED == retCode) return ERASEMEDIUMUI_CANCELED;
|
|
else if (retCode > ERASEMEDIUM_ERROR) return ERASEMEDIUMUI_ERROR;
|
|
return ERASEMEDIUMUI_OK;
|
|
}
|
|
|
|
void EraseMediumUI::OnPrepareInit(HWND hwndDlg)
|
|
{
|
|
prepareWnd = hwndDlg;
|
|
wchar_t format[96] = {0}, buffer[98] = {0};
|
|
|
|
LoadStringW(hResource, IDS_SELECTERASEMETHOD, format, 96);
|
|
StringCchPrintfW(buffer, 98, format, drive);
|
|
SetDlgItemTextW(prepareWnd, IDC_CAPTION, buffer);
|
|
HWND cmbWnd = GetDlgItem(prepareWnd, IDC_CMB_ERASEMETHOD);
|
|
|
|
LoadStringW(hResource, IDS_QUICKERASE, buffer, 98);
|
|
SendMessageW(cmbWnd, CB_ADDSTRING, 0, (LPARAM)buffer);
|
|
LoadStringW(hResource, IDS_COMPLETEERASE, buffer, 98);
|
|
SendMessageW(cmbWnd, CB_ADDSTRING, 0, (LPARAM)buffer);
|
|
SendMessageW(cmbWnd, CB_SETCURSEL, 0, 0);
|
|
PostMessageW(prepareWnd, WM_COMMAND, MAKEWPARAM(IDC_CMB_ERASEMETHOD, CBN_SELCHANGE), 0);
|
|
ShowWindow(prepareWnd, SW_SHOWNORMAL);
|
|
SetForegroundWindow(prepareWnd);
|
|
BringWindowToTop(prepareWnd);
|
|
}
|
|
|
|
void EraseMediumUI::OnPrepareOk()
|
|
{
|
|
switch(SendMessage(GetDlgItem(prepareWnd, IDC_CMB_ERASEMETHOD), CB_GETCURSEL, 0,0))
|
|
{
|
|
case 1: eraseMode = PRIMOSDK_ERASEFULL; break;
|
|
default: eraseMode = PRIMOSDK_ERASEQUICK; break;
|
|
|
|
}
|
|
if (discCheck)
|
|
{
|
|
// check that disc is erasable
|
|
obj_primo *primo=0;
|
|
waServiceFactory *sf = WASABI_API_SVC->service_getServiceByGuid(obj_primo::getServiceGuid());
|
|
if (sf) primo = reinterpret_cast<obj_primo *>(sf->getInterface());
|
|
if (primo)
|
|
{
|
|
WAMEDIUMINFO mi;
|
|
FillMemory(&mi, sizeof(WAMEDIUMINFO), 0xFF);
|
|
mi.erasable = TRUE;
|
|
CheckMediumUI cm;
|
|
wchar_t buffer[256] = {0};
|
|
LoadStringW(hResource, IDS_ERASEREQDISC, buffer, 256);
|
|
DWORD retCode = cm.Check(primo, &drive, &mi, buffer, TRUE, FALSE, prepareWnd);
|
|
|
|
switch(retCode)
|
|
{
|
|
|
|
case CHECKMEDIUMUI_MATCH: eraseCode = ERASEMEDIUMUI_OK; break;
|
|
case CHECKMEDIUMUI_CANCELED: eraseCode = ERASEMEDIUMUI_CANCELED; break;
|
|
default: eraseCode = ERASEMEDIUMUI_PRIMOSDKERROR; break;
|
|
}
|
|
|
|
sf->releaseInterface(primo);
|
|
}
|
|
else
|
|
eraseCode = ERASEMEDIUMUI_PRIMOSDKERROR;
|
|
}
|
|
else
|
|
{
|
|
eraseCode = ERASEMEDIUMUI_OK;
|
|
}
|
|
|
|
EndDialog(prepareWnd, eraseCode);
|
|
}
|
|
void EraseMediumUI::OnEraseTimerClock(void)
|
|
{
|
|
actualTime = (GetTickCount() - startTick) / 1000;
|
|
wchar_t time[32], *current;
|
|
current = time;
|
|
GetTimeString(current, 32, actualTime);
|
|
unsigned int len = lstrlenW(current);
|
|
current[len] = L'/';
|
|
len++;
|
|
current += len;
|
|
if (estimateTime < actualTime) estimateTime = actualTime;
|
|
GetTimeString(current, 32 - len, estimateTime);
|
|
SetWindowTextW(GetDlgItem(eraseWnd, IDC_LBL_TIME), time);
|
|
SendMessage(GetDlgItem(eraseWnd, IDC_PRG_PROGRESS), PBM_SETPOS, (int)(((float)actualTime/(float)estimateTime)*100), 0);
|
|
}
|
|
void EraseMediumUI::OnEraseInit(HWND hwndDlg)
|
|
{
|
|
eraseWnd = hwndDlg;
|
|
CheckDlgButton(eraseWnd, IDC_CHECK_EJECT, BST_CHECKED);
|
|
EnableWindow(GetDlgItem(eraseWnd, IDC_BTN_EJECT), FALSE);
|
|
|
|
eraseMedium = new(EraseMedium);
|
|
if (!eraseMedium) EndDialog(eraseWnd, ERASEMEDIUMUI_UNABLETOCREATEOBJECT);
|
|
|
|
|
|
startTick = GetTickCount();
|
|
OnEraseTimerClock();
|
|
|
|
ShowWindow(eraseWnd, SW_SHOWNORMAL);
|
|
SetForegroundWindow(eraseWnd);
|
|
BringWindowToTop(eraseWnd);
|
|
UpdateWindow(eraseWnd);
|
|
|
|
|
|
SetTimer(eraseWnd, TIMER_CLOCK_ID, TIMER_CLOCK_INTERVAL, NULL);
|
|
SendMessage(GetDlgItem(eraseWnd, IDC_PRG_PROGRESS), PBM_SETRANGE, 0, MAKELPARAM(0, 100));
|
|
eraseMedium->Start(drive, eraseMode, OnEraseNotify, this, FALSE); // will catch all errors in calback
|
|
EnableWindow(GetDlgItem(eraseWnd, IDC_BTN_EJECT), TRUE);
|
|
|
|
}
|
|
|
|
void EraseMediumUI::OnEraseClose(DWORD exitCode)
|
|
{
|
|
EnableWindow(GetDlgItem(eraseWnd, IDC_BTN_EJECT), FALSE);
|
|
if (eraseMedium)
|
|
{
|
|
HWND btnWnd = GetDlgItem(eraseWnd, IDCANCEL);
|
|
EnableWindow(btnWnd, FALSE);
|
|
eraseMedium->Stop();
|
|
wchar_t buffer[24] = {0};
|
|
LoadStringW(hResource, IDS_CLOSE, buffer, 24);
|
|
SetWindowTextW(btnWnd, buffer);
|
|
EnableWindow(btnWnd, TRUE);
|
|
primoCode = eraseMedium->GetErrorCode();
|
|
delete(eraseMedium);
|
|
eraseMedium = NULL;
|
|
}
|
|
eraseCode = exitCode;
|
|
KillTimer(eraseWnd, TIMER_CLOCK_ID);
|
|
estimateTime = 0; // will force it to be the same with actual
|
|
OnEraseTimerClock();
|
|
}
|
|
|
|
LRESULT EraseMediumUI::PrepareWndProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
static EraseMediumUI *object = NULL;
|
|
switch(uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
object = (EraseMediumUI*)lParam;
|
|
object->OnPrepareInit(hwndDlg);
|
|
break;
|
|
case WM_COMMAND:
|
|
switch(LOWORD(wParam))
|
|
{
|
|
case IDOK:
|
|
object->OnPrepareOk();
|
|
break;
|
|
case IDCANCEL:
|
|
{
|
|
wchar_t msg[256] = {0}, caption[64] = {0};
|
|
LoadStringW(hResource, IDS_MB_CANCELOPERATION, msg, 256);
|
|
LoadStringW(hResource, IDS_CONFIRMATION, caption, 64);
|
|
if (MessageBoxW(hwndDlg, msg, caption, MB_YESNO | MB_ICONQUESTION) == IDYES)
|
|
{
|
|
EndDialog(hwndDlg, LOWORD(wParam));
|
|
}
|
|
break;
|
|
}
|
|
case IDC_CMB_ERASEMETHOD:
|
|
switch(HIWORD(wParam))
|
|
{
|
|
case CBN_SELCHANGE:
|
|
switch(SendMessage(GetDlgItem(hwndDlg, IDC_CMB_ERASEMETHOD), CB_GETCURSEL, 0,0))
|
|
{
|
|
case 0:
|
|
object->estimateTime= ERASETIME_QUICKMODE;
|
|
break;
|
|
case 1:
|
|
object->estimateTime = ERASETIME_FULLMODE;
|
|
break;
|
|
default:
|
|
object->estimateTime =0;
|
|
break;
|
|
}
|
|
wchar_t time[16] = {0};
|
|
SetWindowTextW(GetDlgItem(hwndDlg, IDC_LBL_TIME), GetTimeString(time, 16, object->estimateTime));
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
LRESULT EraseMediumUI::EraseWndProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
static EraseMediumUI *object = NULL;
|
|
switch(uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
object = (EraseMediumUI*)lParam;
|
|
object->OnEraseInit(hwndDlg);
|
|
break;
|
|
case WM_CLOSE:
|
|
object->OnEraseClose((DWORD)wParam);
|
|
return 1;
|
|
case WM_COMMAND:
|
|
switch(LOWORD(wParam))
|
|
{
|
|
case IDCANCEL:
|
|
if (object->eraseMedium)
|
|
{
|
|
wchar_t msg[256] = {0}, caption[64] = {0};
|
|
LoadStringW(hResource, IDS_MB_CANCELOPERATION, msg, 256);
|
|
LoadStringW(hResource, IDS_CONFIRMATION, caption, 64);
|
|
if (MessageBoxW(hwndDlg, msg, caption, MB_YESNO | MB_ICONQUESTION) == IDYES) object->OnEraseClose(ERASEMEDIUM_ABORTED);
|
|
}
|
|
EndDialog(hwndDlg, object->eraseCode);
|
|
break;
|
|
case IDC_BTN_EJECT:
|
|
if (BN_CLICKED == HIWORD(wParam) && object->eraseMedium)
|
|
{
|
|
object->eraseMedium->SetEject(BST_CHECKED == IsDlgButtonChecked (hwndDlg,IDC_BTN_EJECT));
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
case WM_TIMER:
|
|
switch(wParam)
|
|
{
|
|
case TIMER_CLOCK_ID:
|
|
object->OnEraseTimerClock();
|
|
break;
|
|
|
|
}
|
|
break;
|
|
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
DWORD EraseMediumUI::OnEraseNotify(void *sender, void *param, DWORD eraseCode, DWORD primoCode)
|
|
{
|
|
EraseMediumUI *object = (EraseMediumUI*)param;
|
|
|
|
unsigned int strcode;
|
|
BOOL finished = FALSE;
|
|
|
|
if (ERASEMEDIUM_ERROR < eraseCode)
|
|
{// some error happened;
|
|
finished = TRUE;
|
|
|
|
switch(eraseCode)
|
|
{
|
|
case ERASEMEDIUM_ALREADYSTARTED: strcode = IDS_ALREADYERASING; break;
|
|
case ERASEMEDIUM_UNABLEINITPRIMO: strcode = IDS_PRIMOINITFAILED; break;
|
|
case ERASEMEDIUM_BEGINBURNFAILED: strcode = IDS_UNABLEINITERASE; break;
|
|
case ERASEMEDIUM_ENDBURNFAILED: strcode = IDS_UNABLEFINERASE; break;
|
|
case ERASEMEDIUM_ERASEMEDIUMFAILED: strcode = IDS_MEDIUMERASEFAILED; break;
|
|
case ERASEMEDIUM_DEVICENOTREADY: strcode = IDS_DRIVENOTREADY; break;
|
|
case ERASEMEDIUM_DISCINFOERROR: strcode = IDS_UNABLEGETDISCINFO; break;
|
|
case ERASEMEDIUM_DISCNOTERASABLE: strcode = IDS_DISCNONERASABLE; break;
|
|
default: strcode = IDS_UNKNOWNERROR; break;
|
|
}
|
|
MessageBeep(MB_ICONHAND);
|
|
}
|
|
else
|
|
{
|
|
switch(eraseCode)
|
|
{
|
|
case ERASEMEDIUM_READY: strcode = IDS_READY; break;
|
|
case ERASEMEDIUM_INITIALIZING: strcode = IDS_INITIALIZING; break;
|
|
case ERASEMEDIUM_ERASING: strcode = IDS_ERASING; break;
|
|
case ERASEMEDIUM_FINISHING: strcode = IDS_FINISHING; break;
|
|
case ERASEMEDIUM_CANCELING: strcode = IDS_CANCELING; break;
|
|
case ERASEMEDIUM_COMPLETED: strcode = IDS_COMPLETED; finished = TRUE; MessageBeep(MB_OK); break;
|
|
case ERASEMEDIUM_ABORTED: strcode = IDS_ABORTED; finished = TRUE; MessageBeep(MB_OK); break;
|
|
}
|
|
|
|
}
|
|
wchar_t buffer[224] = {0};
|
|
LoadStringW(hResource, strcode, buffer, 224);
|
|
|
|
if (ERASEMEDIUM_ERROR < eraseCode)
|
|
{
|
|
wchar_t txtStatus[256] = {0}, error[24] = {0};
|
|
LoadStringW(hResource, IDS_ERROR, error, 24);
|
|
StringCchPrintfW(txtStatus, 256, L"%s! %s", error, buffer);
|
|
}
|
|
else
|
|
{
|
|
SetWindowTextW(GetDlgItem(object->eraseWnd, IDC_LBL_STATUS), buffer);
|
|
}
|
|
|
|
if (finished) PostMessage(object->eraseWnd, WM_CLOSE, eraseCode, 0);
|
|
return ERASEMEDIUM_CONTINUE;
|
|
} |