Initial community commit

1 files changed, 340 insertions, 0 deletions

diff --git a/Src/burnlib/uiEraseMedium.cpp b/Src/burnlib/uiEraseMedium.cpp
new file mode 100644
index 00000000..69ffec14
--- /dev/null
+++ b/Src/burnlib/uiEraseMedium.cpp
@@ -0,0 +1,340 @@
+#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;
+}
\ No newline at end of file