Initial community commit

1 files changed, 300 insertions, 0 deletions

diff --git a/Src/burnlib/uiCheckMedium.cpp b/Src/burnlib/uiCheckMedium.cpp
new file mode 100644
index 00000000..de2b1791
--- /dev/null
+++ b/Src/burnlib/uiCheckMedium.cpp
@@ -0,0 +1,300 @@
+#include "./uiCheckMedium.h"
+#include "./resource.h"
+#include <strsafe.h>
+
+#include "./uiEraseMedium.h"
+#include "./uiUnitReady.h"
+
+#define AINSTATE_DISABLE	PRIMOSDK_LOCK
+#define AINSTATE_ENABLE		PRIMOSDK_UNLOCK
+
+
+CheckMediumUI::CheckMediumUI(void)
+{
+	hwnd = NULL;
+	drive = NULL;
+	primoSDK = NULL;
+	errPrimo = 0;
+	errReady = 0;
+}
+CheckMediumUI::~CheckMediumUI(void)
+{
+	if (hwnd) DestroyWindow(hwnd);
+}
+
+DWORD CheckMediumUI::Check(obj_primo *primoSDK, DWORD *drive, WAMEDIUMINFO *medium, const wchar_t *description, BOOL disableAIN, BOOL showErase, HWND ownerWnd)
+{
+
+	if (!medium) return CHECKMEDIUMUI_DISCNOTSET;
+	if (!drive) return CHECKMEDIUMUI_DRIVENOTSET;
+	if (!primoSDK) return CHECKMEDIUMUI_PRIMOSDKNOTSET;
+	this->desiredMedium	= medium;
+	this->drive = drive;
+	this->primoSDK = primoSDK;
+	this->disableAIN = (disableAIN) ? AINSTATE_DISABLE : 0 ;
+	this->showErase = showErase;
+	this->description = description;
+	this->ownerWnd = ownerWnd;
+	
+	hwnd = NULL;
+	errPrimo = PRIMOSDK_OK;
+	errReady = CHECKMEDIUMUI_NOMATCH;
+	
+	deadLoop = 0;
+
+	LPCDLGTEMPLATE templ = NULL;
+	HRSRC hres = FindResourceExW(hResource,  MAKEINTRESOURCEW(5), MAKEINTRESOURCEW(IDD_DLG_WRONGMEDIUM), MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL)); 
+	if (hres) templ = (LPCDLGTEMPLATE)LoadResource(hResource, hres);
+	HWND dlgWnd = CreateDialogIndirectParamW(dllInstance, templ, ownerWnd, (DLGPROC)WndProc, (LPARAM)this);
+	
+	if (!dlgWnd) return CHECKMEDIUMUI_UNABLETOCREATEDIALOG;
+	
+	MSG msg;
+	BOOL ret;
+	while( 0 != (ret = GetMessageW(&msg, hwnd, 0, 0)))
+	{	 
+		if (ret == -1) 
+		{
+			errReady = CHECKMEDIUMUI_MESSAGEPUMPERROR;
+			break;
+		}
+		if (IsDialogMessage(hwnd, &msg)) continue;
+		TranslateMessage(&msg);
+		DispatchMessageW(&msg);
+	}
+	return errReady;
+}
+
+DWORD CheckMediumUI::Rescan(void)
+{
+	wchar_t buffer[256] = {0};
+	if (hwnd)
+	{
+		EnableWindow(GetDlgItem(hwnd, IDCANCEL), FALSE);
+		
+		LoadStringW(hResource, IDS_UPDATINGDATA, buffer, 256);
+		SetWindowLong(GetDlgItem(hwnd, IDC_LBL_DETECTEDMEDIUM), GWL_STYLE, WS_CHILD | WS_VISIBLE | SS_CENTER | SS_CENTERIMAGE);
+		SetDlgItemTextW(hwnd, IDC_LBL_DETECTEDMEDIUM, buffer);
+	}
+	if (!drive) return CHECKMEDIUMUI_DRIVENOTSET;
+	if (!primoSDK) return CHECKMEDIUMUI_PRIMOSDKNOTSET;
+	
+	
+	UpdatingDataUI updateDlg;
+	LoadStringW(hResource, IDS_READINGDISCINFO, buffer, 256);
+	updateDlg.Show(800, buffer, TRUE, hwnd);
+
+	WAMEDIUMINFO detectedMedium;
+	ZeroMemory(&detectedMedium, sizeof(WAMEDIUMINFO));
+	
+	errPrimo = GetMediumInfo(primoSDK, drive, &detectedMedium);
+    updateDlg.Hide();
+
+
+	if (PRIMOSDK_OK != errPrimo)
+	{
+
+		SetWindowLong(GetDlgItem(hwnd, IDC_LBL_DETECTEDMEDIUM), GWL_STYLE,  WS_CHILD | WS_VISIBLE | SS_CENTER | SS_CENTERIMAGE);
+		LoadStringW(hResource, IDS_UNABLEGETDISCINFO, buffer, 256);
+		SetDlgItemTextW(hwnd, IDC_LBL_DETECTEDMEDIUM, buffer);
+		EnableWindow(GetDlgItem(hwnd, IDCANCEL), FALSE);
+		EnableWindow(GetDlgItem(hwnd, IDC_BTN_ERASE), FALSE);
+		EnableWindow(hwnd, FALSE);
+		if (AINSTATE_DISABLE == disableAIN && primoSDK) 
+		{
+			primoSDK->UnitAIN(drive, disableAIN);
+			disableAIN = AINSTATE_ENABLE;
+		}
+		UnitReadyUI ready;
+		DWORD test = ready.Check(primoSDK, drive, FALSE, hwnd);
+		if (UNITREADYUI_DRIVEREADY == test) deadLoop++;
+		if (deadLoop == 5) 
+		{ /// dead loop detected - PRIMOSDK broken - recomended restart of the computer
+			errReady = CHECKMEDIUMUI_DEADLOOP; 
+			if(hwnd) PostMessage(hwnd, WM_DESTROY, 0, 0);
+			return errReady;
+		}
+		EnableWindow(hwnd, TRUE);
+		UpdateWindow(hwnd);
+		switch(test)
+		{
+			case UNITREADYUI_DRIVEREADY:
+				break;	
+			case UNITREADYUI_PRIMOSDKERROR:
+				errReady = CHECKMEDIUMUI_PRIMOSDKERROR; 
+				break;
+			case UNITREADYUI_CANCELED:
+				errReady = CHECKMEDIUMUI_CANCELED; 
+				break;
+			default:
+				errReady = CHECKMEDIUMUI_PRIMOSDKERROR; 
+				break;
+
+		}
+		if (UNITREADYUI_DRIVEREADY != test)
+		{
+			if(hwnd) PostMessage(hwnd, WM_DESTROY, 0, 0);
+			return errReady;
+		}
+		
+		PostMessage(hwnd, WM_COMMAND, MAKEWPARAM(IDOK, BN_CLICKED), 0); // start other rescan
+		errReady = CHECKMEDIUMUI_DRIVENOTREADY; 
+		return errReady;
+
+	}
+
+	if (hwnd)
+	{
+		EnableWindow(GetDlgItem(hwnd, IDCANCEL), TRUE);
+		SetDlgItemTextW(hwnd, IDC_LBL_DETECTEDMEDIUM, L"");
+	}
+	
+	errReady = CHECKMEDIUMUI_MATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->medium && desiredMedium->medium != detectedMedium.medium)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->mediumType && desiredMedium->mediumType != detectedMedium.mediumType)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->mediumFormat && desiredMedium->mediumFormat != detectedMedium.mediumFormat)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->tracks && desiredMedium->tracks != detectedMedium.tracks)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->erasable && desiredMedium->erasable != detectedMedium.erasable)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->freeSectors && desiredMedium->freeSectors > detectedMedium.freeSectors)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->usedSectors && desiredMedium->usedSectors < detectedMedium.usedSectors)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->recordable && desiredMedium->recordable != detectedMedium.recordable)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->isCD && desiredMedium->isCD != detectedMedium.isCD)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->isDCD && desiredMedium->isDCD != detectedMedium.isDCD)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->isDVD && desiredMedium->isDVD != detectedMedium.isDVD)  errReady = CHECKMEDIUMUI_NOMATCH;
+	if (CHECKMEDIUMUI_MATCH == errReady && MAXDWORD != desiredMedium->isDLDVD && desiredMedium->isDLDVD != detectedMedium.isDLDVD)  errReady = CHECKMEDIUMUI_NOMATCH;
+	
+
+	if (CHECKMEDIUMUI_MATCH == errReady)
+	{
+		if(hwnd) PostMessage(hwnd, WM_DESTROY, 0, 0);
+	}
+	else if (hwnd)
+	{
+		wchar_t buffer[4096] = {0};
+		SetWindowLong(GetDlgItem(hwnd, IDC_LBL_DETECTEDMEDIUM), GWL_STYLE, WS_CHILD | WS_VISIBLE | SS_LEFT);
+		SetDlgItemTextW(hwnd, IDC_LBL_DETECTEDMEDIUM, GetMediumInfoText(buffer, 4096, &detectedMedium));
+
+		if (showErase) EnableWindow(GetDlgItem(hwnd, IDC_BTN_ERASE), (detectedMedium.erasable && detectedMedium.usedSectors > 0));
+		ShowWindow(hwnd, SW_SHOWNORMAL);
+		SetForegroundWindow(hwnd);
+		BringWindowToTop(hwnd);
+		MessageBeep(MB_ICONEXCLAMATION);
+		UpdateWindow(hwnd);
+	}
+	return errReady;
+}
+
+wchar_t* CheckMediumUI::GetMediumInfoText(wchar_t *buffer, unsigned int cchBuffer, WAMEDIUMINFO *info)
+{
+	buffer[0] = 0x0000;
+
+	wchar_t format[64] = {0}, freeSize[256] = {0}, usedSize[256] = {0}, medium[32] = {0}, mtype[128] = {0}, mformat[128] = {0}, erasable[32] = {0};
+	LoadStringW(hResource, IDS_DISCINFOFORMAT, format, 64);
+	LoadStringW(hResource, (info->erasable) ? IDS_DISCERASABLE : IDS_DISCNONERASABLE, erasable, 32);
+	StringCchPrintfW(buffer, cchBuffer, format, GetMediumText(medium, 32, info->medium),
+																erasable,
+																GetSizeText(freeSize, 256, info->freeSectors),
+																GetSizeText(usedSize, 256, info->usedSectors),
+																GetMediumFormatText(mformat, 128,info->mediumFormat), 
+																GetMediumTypeText(mtype, 128, info->mediumType));
+	return buffer;
+}
+
+wchar_t* CheckMediumUI::GetSizeText(wchar_t *buffer, unsigned int cchBuffer, unsigned int sectors)
+{
+	wchar_t format[32] = {0},tmp[256] = {0};
+	DWORD minutes = sectors/(75*60);
+	DWORD kb = MulDiv(sectors, 2048,1024);
+	DWORD mb = kb / 1024;
+	LoadStringW(hResource, IDS_DISCSIZEFORMAT, format, 32);
+	StringCchPrintfW(tmp, 256, format, minutes, mb, sectors);
+	StringCchCatW(buffer, cchBuffer, tmp);
+	return buffer;
+}
+void CheckMediumUI::OnInitDialog(HWND hwndDlg)
+{
+	hwnd = hwndDlg;
+
+	HANDLE hImage = LoadBitmapW(hResource,  MAKEINTRESOURCEW(IDB_DISC1));
+	if(hImage==NULL){
+		hImage = LoadBitmapW(dllInstance,  MAKEINTRESOURCEW(IDB_DISC1));
+	}
+	SendDlgItemMessage(hwnd, IDC_PIC, STM_SETIMAGE, IMAGE_BITMAP, (LPARAM)hImage);
+
+	wchar_t format[64] = {0}, buffer[66] = {0};
+	
+	LoadStringW(hResource, IDS_WRONGMEDIUM, format, 64);
+	StringCchPrintfW(buffer, 64, format, (char)*drive);
+	SetWindowTextW(GetDlgItem(hwnd, IDC_CAPTION), buffer);
+	SetWindowTextW(GetDlgItem(hwnd, IDC_LBL_REQUESTEDMEDIUM), description);
+	HWND btnWnd = GetDlgItem(hwnd, IDC_BTN_ERASE);
+	EnableWindow(btnWnd, showErase);
+	ShowWindow(btnWnd, showErase);
+	Rescan();
+}
+
+void CheckMediumUI::OnCancel(void)
+{
+	wchar_t msg[256] = {0}, caption[64] = {0};
+	LoadStringW(hResource, IDS_MB_CANCELOPERATION, msg, 256);
+	LoadStringW(hResource, IDS_CONFIRMATION, caption, 64);
+	if (MessageBoxW(hwnd, msg, caption, MB_YESNO | MB_ICONQUESTION) == IDYES)
+	{
+		errReady = CHECKMEDIUMUI_CANCELED;
+		if(hwnd) DestroyWindow(hwnd);
+		hwnd = NULL;
+	}
+}
+
+void CheckMediumUI::OnDestroy(void)
+{
+	if (AINSTATE_ENABLE == disableAIN && primoSDK) 
+	{
+		primoSDK->UnitAIN(drive, disableAIN);
+		disableAIN = AINSTATE_DISABLE;
+	}
+	ShowWindow(hwnd, SW_HIDE);
+	hwnd = NULL;
+	drive = NULL;
+	primoSDK = NULL;
+	deadLoop = 0;
+}
+
+void CheckMediumUI::OnEraseClicked(void)
+{
+	EraseMediumUI eraseMedium;
+	eraseMedium.Erase(*drive, FALSE, hwnd); // do not perform disc check
+	UpdateWindow(hwnd);
+	Rescan();
+}
+
+LRESULT CheckMediumUI::WndProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+	static CheckMediumUI *object = NULL;
+	switch(uMsg)
+	{
+		case WM_INITDIALOG:
+			object = (CheckMediumUI*)lParam;
+			object->OnInitDialog(hwndDlg);
+			break;
+		case WM_DESTROY:
+			object->OnDestroy();
+			 PostQuitMessage(object->errReady);
+			break;
+		case WM_COMMAND:
+			switch(LOWORD(wParam))
+			{
+				case IDOK:
+					object->Rescan();
+					break;
+				case IDCANCEL:
+					object->OnCancel();
+					break;
+				case IDC_BTN_ERASE:
+					if (BN_CLICKED == HIWORD(wParam)) object->OnEraseClicked();
+					break;
+			}
+			break;
+	}
+	return 0;
+}
\ No newline at end of file