#include "mxdirectdraw.h"

#include "decomp.h"

DECOMP_SIZE_ASSERT(MxDirectDraw, 0x880);
DECOMP_SIZE_ASSERT(MxDirectDraw::DeviceModesInfo, 0x17c);

#define RELEASE(x)                                                                                                     \
	if (x != NULL) {                                                                                                   \
		x->Release();                                                                                                  \
		x = NULL;                                                                                                      \
	}

#ifndef DDSCAPS_3DDEVICE
#define DDSCAPS_3DDEVICE 0x00002000l
#endif

// GLOBAL: LEGO1 0x10100c68
BOOL g_isPaletteIndexed8 = 0;

// FUNCTION: LEGO1 0x1009d490
MxDirectDraw::MxDirectDraw()
{
	m_pFrontBuffer = NULL;
	m_pBackBuffer = NULL;
	m_pZBuffer = NULL;
	m_pClipper = NULL;
	m_pPalette = NULL;
	m_pDirectDraw = NULL;
	m_pText1Surface = NULL;
	m_pText2Surface = NULL;
	m_hWndMain = NULL;
	m_bIgnoreWMSIZE = FALSE;
	m_bPrimaryPalettized = FALSE;
	m_bOnlySystemMemory = FALSE;
	m_bFullScreen = FALSE;
	m_bOnlySoftRender = FALSE;
	m_pauseCount = 0;
	m_pErrorHandler = NULL;
	m_pFatalErrorHandler = NULL;
	m_pErrorHandlerArg = NULL;
	m_pFatalErrorHandlerArg = NULL;
	m_pCurrentDeviceModesList = NULL;
	m_bIsOnPrimaryDevice = TRUE;
	m_hFont = NULL;
}

// FUNCTION: LEGO1 0x1009d530
MxDirectDraw::~MxDirectDraw()
{
	Destroy();

	if (m_pCurrentDeviceModesList != NULL) {
		delete m_pCurrentDeviceModesList;
		m_pCurrentDeviceModesList = NULL;
	}
}

// FUNCTION: LEGO1 0x1009d570
int MxDirectDraw::GetPrimaryBitDepth()
{
	DWORD dwRGBBitCount;
	LPDIRECTDRAW pDDraw;
	DDSURFACEDESC ddsd;

	HRESULT result = DirectDrawCreate(NULL, &pDDraw, NULL);
	dwRGBBitCount = 8;
	if (!result) {
		memset(&ddsd, 0, sizeof(ddsd));
		ddsd.dwSize = sizeof(ddsd);

		pDDraw->GetDisplayMode(&ddsd);
		dwRGBBitCount = ddsd.ddpfPixelFormat.dwRGBBitCount;
		g_isPaletteIndexed8 = (ddsd.ddpfPixelFormat.dwFlags & DDPF_PALETTEINDEXED8) != 0;
		pDDraw->Release();
	}

	return dwRGBBitCount;
}

// FUNCTION: LEGO1 0x1009d5e0
BOOL MxDirectDraw::Create(
	HWND hWnd,
	BOOL fullscreen,
	BOOL surface_fullscreen,
	BOOL onlySystemMemory,
	int width,
	int height,
	int bpp,
	const PALETTEENTRY* pPaletteEntries,
	int paletteEntryCount
)
{
	m_hWndMain = hWnd;

	CacheOriginalPaletteEntries();

	if (!RecreateDirectDraw(&m_pCurrentDeviceModesList->m_guid)) {
		return FALSE;
	}

	m_bFlipSurfaces = surface_fullscreen;
	m_bOnlySystemMemory = onlySystemMemory;
	m_bIsOnPrimaryDevice = !m_pCurrentDeviceModesList->m_guid;

	if (!m_bIsOnPrimaryDevice) {
		fullscreen = TRUE;
	}

	if (!SetPaletteEntries(pPaletteEntries, paletteEntryCount, fullscreen)) {
		return FALSE;
	}

	if (!DDInit(fullscreen)) {
		return FALSE;
	}

	if (!DDSetMode(width, height, bpp)) {
		return FALSE;
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009d690
BOOL MxDirectDraw::RecreateDirectDraw(GUID** ppGUID)
{
	RELEASE(m_pDirectDraw);
	return (DirectDrawCreate(*ppGUID, &m_pDirectDraw, 0) == DD_OK);
}

// FUNCTION: LEGO1 0x1009d6c0
BOOL MxDirectDraw::CacheOriginalPaletteEntries()
{
	HDC hdc;

	if (g_isPaletteIndexed8) {
		hdc = GetDC(NULL);
		GetSystemPaletteEntries(hdc, 0, (1 << 8), &m_originalPaletteEntries[0]);
		ReleaseDC(NULL, hdc);
	}
	return TRUE;
}

// FUNCTION: LEGO1 0x1009d700
BOOL MxDirectDraw::SetPaletteEntries(const PALETTEENTRY* pPaletteEntries, int paletteEntryCount, BOOL fullscreen)
{
	int reservedLowEntryCount = 10;
	int reservedHighEntryCount = 10;
	int arraySize = sizeOfArray(m_paletteEntries);
	HDC hdc;
	int i;

	if (g_isPaletteIndexed8) {
		hdc = GetDC(NULL);
		GetSystemPaletteEntries(hdc, 0, (1 << 8), m_paletteEntries);
		ReleaseDC(NULL, hdc);
	}

	for (i = 0; i < reservedLowEntryCount; i++) {
		m_paletteEntries[i].peFlags = 0x80;
	}

	for (i = reservedLowEntryCount; i < 142; i++) {
		m_paletteEntries[i].peFlags = 0x44;
	}

	for (i = 142; i < arraySize - reservedHighEntryCount; i++) {
		m_paletteEntries[i].peFlags = 0x84;
	}

	for (i = 256 - reservedHighEntryCount; i < 256; i++) {
		m_paletteEntries[i].peFlags = 0x80;
	}

	if (paletteEntryCount != 0) {
		for (i = reservedLowEntryCount; (i < paletteEntryCount) && (i < 256 - reservedHighEntryCount); i++) {
			m_paletteEntries[i].peRed = pPaletteEntries[i].peRed;
			m_paletteEntries[i].peGreen = pPaletteEntries[i].peGreen;
			m_paletteEntries[i].peBlue = pPaletteEntries[i].peBlue;
		}
	}

	if (m_pPalette) {
		HRESULT result;

		result = m_pPalette->SetEntries(0, 0, sizeOfArray(m_paletteEntries), m_paletteEntries);
		if (result != DD_OK) {
			Error("SetEntries failed", result);
			return FALSE;
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009d800
void MxDirectDraw::Destroy()
{
	DestroyButNotDirectDraw();

	FUN_1009d920();

	RELEASE(m_pDirectDraw);

	m_bIsOnPrimaryDevice = TRUE;

	if (m_pCurrentDeviceModesList != NULL) {
		delete m_pCurrentDeviceModesList;
		m_pCurrentDeviceModesList = NULL;
	}
}

// FUNCTION: LEGO1 0x1009d860
void MxDirectDraw::DestroyButNotDirectDraw()
{
	RestoreOriginalPaletteEntries();
	if (m_bFullScreen) {
		if (m_pDirectDraw) {
			m_bIgnoreWMSIZE = TRUE;
			m_pDirectDraw->RestoreDisplayMode();
			m_bIgnoreWMSIZE = FALSE;
		}
	}

	RELEASE(m_pPalette);
	RELEASE(m_pClipper);
	RELEASE(m_pText1Surface);
	RELEASE(m_pText2Surface);
	RELEASE(m_pZBuffer);
	RELEASE(m_pBackBuffer);
	RELEASE(m_pFrontBuffer);
}

// FUNCTION: LEGO1 0x1009d920
void MxDirectDraw::FUN_1009d920()
{
	RestoreOriginalPaletteEntries();
	if (m_pDirectDraw != NULL) {
		m_bIgnoreWMSIZE = TRUE;
		m_pDirectDraw->RestoreDisplayMode();
		m_pDirectDraw->SetCooperativeLevel(NULL, DDSCL_NORMAL);
		m_bIgnoreWMSIZE = FALSE;
	}
}

// FUNCTION: LEGO1 0x1009d960
BOOL MxDirectDraw::DDInit(BOOL fullscreen)
{
	HRESULT result;

	if (fullscreen) {
		m_bIgnoreWMSIZE = TRUE;
		result = m_pDirectDraw->SetCooperativeLevel(m_hWndMain, DDSCL_EXCLUSIVE | DDSCL_FULLSCREEN);
		m_bIgnoreWMSIZE = FALSE;
	}
	else {
		result = m_pDirectDraw->SetCooperativeLevel(m_hWndMain, DDSCL_NORMAL);
	}

	if (result != DD_OK) {
		Error("SetCooperativeLevel failed", result);
		return FALSE;
	}

	m_bFullScreen = fullscreen;

	return TRUE;
}

// FUNCTION: LEGO1 0x1009d9d0
BOOL MxDirectDraw::IsSupportedMode(int width, int height, int bpp)
{
	Mode mode = {width, height, bpp};

	for (int i = 0; i < m_pCurrentDeviceModesList->m_count; i++) {
		if (m_pCurrentDeviceModesList->m_modeArray[i] == mode) {
			return TRUE;
		}
	}

	return FALSE;
}

// FUNCTION: LEGO1 0x1009da20
void EnableResizing(HWND p_hwnd, BOOL p_flag)
{
	static DWORD g_dwStyle;

	if (!p_flag) {
		g_dwStyle = GetWindowLong(p_hwnd, GWL_STYLE);
		if (g_dwStyle & WS_THICKFRAME) {
			SetWindowLong(p_hwnd, GWL_STYLE, GetWindowLong(p_hwnd, GWL_STYLE) ^ WS_THICKFRAME);
		}
	}
	else {
		SetWindowLong(p_hwnd, GWL_STYLE, g_dwStyle);
	}
}

// FUNCTION: LEGO1 0x1009da80
BOOL MxDirectDraw::DDSetMode(int width, int height, int bpp)
{
	HRESULT result;

	if (m_bFullScreen) {
		LPDIRECTDRAW lpDD;

		EnableResizing(m_hWndMain, FALSE);

		if (!m_bIsOnPrimaryDevice) {
			lpDD = NULL;
			result = DirectDrawCreate(0, &lpDD, 0);
			if (result == DD_OK) {
				result = lpDD->SetCooperativeLevel(m_hWndMain, DDSCL_FULLSCREEN | DDSCL_EXCLUSIVE | DDSCL_ALLOWREBOOT);
				if (result == DD_OK) {
					lpDD->SetDisplayMode(width, height, 8);
				}
			}
		}

		if (!IsSupportedMode(width, height, bpp)) {
			width = m_pCurrentDeviceModesList->m_modeArray[0].width;
			height = m_pCurrentDeviceModesList->m_modeArray[0].height;
			bpp = m_pCurrentDeviceModesList->m_modeArray[0].bitsPerPixel;
		}

		m_bIgnoreWMSIZE = TRUE;
		result = m_pDirectDraw->SetDisplayMode(width, height, bpp);
		m_bIgnoreWMSIZE = FALSE;
		if (result != DD_OK) {
			Error("SetDisplayMode failed", result);
			return FALSE;
		}
	}
	else {
		RECT rc;
		DWORD dwStyle;

		if (!m_bIsOnPrimaryDevice) {
			Error(
				"Attempt made enter a windowed mode on a DirectDraw device that is not the primary display",
				DDERR_GENERIC
			);
			return FALSE;
		}

		m_bIgnoreWMSIZE = TRUE;
		dwStyle = GetWindowLong(m_hWndMain, GWL_STYLE);
		dwStyle &= ~WS_POPUP;
		dwStyle |= WS_OVERLAPPED | WS_CAPTION | WS_THICKFRAME;
		SetWindowLong(m_hWndMain, GWL_STYLE, dwStyle);
		SetRect(&rc, 0, 0, width - 1, height - 1);
		AdjustWindowRectEx(
			&rc,
			GetWindowLong(m_hWndMain, GWL_STYLE),
			GetMenu(m_hWndMain) != NULL,
			GetWindowLong(m_hWndMain, GWL_EXSTYLE)
		);
		SetWindowPos(
			m_hWndMain,
			NULL,
			0,
			0,
			rc.right - rc.left + 1,
			rc.bottom - rc.top + 1,
			SWP_NOMOVE | SWP_NOZORDER | SWP_NOACTIVATE
		);
		SetWindowPos(m_hWndMain, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE | SWP_NOACTIVATE);
		m_bIgnoreWMSIZE = FALSE;
	}

	m_currentMode.width = width;
	m_currentMode.height = height;
	m_currentMode.bitsPerPixel = bpp;

	if (!DDCreateSurfaces()) {
		return FALSE;
	}

	DDSURFACEDESC ddsd;

	FUN_1009e020();

	if (!GetDDSurfaceDesc(&ddsd, m_pBackBuffer)) {
		return FALSE;
	}

	if (ddsd.ddpfPixelFormat.dwFlags & DDPF_PALETTEINDEXED8) {
		m_bPrimaryPalettized = TRUE;
	}
	else {
		m_bPrimaryPalettized = FALSE;
	}

	if (m_bPrimaryPalettized) {
		result = m_pDirectDraw->CreatePalette(
			DDPCAPS_8BIT | DDPCAPS_ALLOW256 | DDPCAPS_INITIALIZE, // 0x4c
			m_paletteEntries,
			&m_pPalette,
			NULL
		);
		if (result != DD_OK) {
			Error("CreatePalette failed", result);
			return 0;
		}
		result = m_pBackBuffer->SetPalette(m_pPalette); // TODO: add FIX_BUGS define and fix this
		result = m_pFrontBuffer->SetPalette(m_pPalette);
		if (result != DD_OK) {
			Error("SetPalette failed", result);
			return FALSE;
		}
	}

	// create debug text only in windowed mode?
	return m_bFullScreen || CreateTextSurfaces();
}

// FUNCTION: LEGO1 0x1009dd80
HRESULT MxDirectDraw::CreateDDSurface(
	LPDDSURFACEDESC p_lpDDSurfDesc,
	LPDIRECTDRAWSURFACE FAR* p_lpDDSurface,
	IUnknown FAR* p_pUnkOuter
)
{
	return m_pDirectDraw->CreateSurface(p_lpDDSurfDesc, p_lpDDSurface, p_pUnkOuter);
}

// FUNCTION: LEGO1 0x1009dda0
BOOL MxDirectDraw::GetDDSurfaceDesc(LPDDSURFACEDESC lpDDSurfDesc, LPDIRECTDRAWSURFACE lpDDSurf)
{
	HRESULT result;

	memset(lpDDSurfDesc, 0, sizeof(DDSURFACEDESC));
	lpDDSurfDesc->dwSize = sizeof(DDSURFACEDESC);
	result = lpDDSurf->GetSurfaceDesc(lpDDSurfDesc);
	if (result != DD_OK) {
		Error("Error getting a surface description", result);
	}

	return (result == DD_OK);
}

// FUNCTION: LEGO1 0x1009ddf0
BOOL MxDirectDraw::DDCreateSurfaces()
{
	HRESULT result;
	DDSURFACEDESC ddsd;
	DDSCAPS ddscaps;

	if (m_bFlipSurfaces) {
		memset(&ddsd, 0, sizeof(DDSURFACEDESC));
		ddsd.dwSize = sizeof(ddsd);
		ddsd.dwFlags = DDSD_CAPS | DDSD_BACKBUFFERCOUNT;
		ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE | DDSCAPS_FLIP | DDSCAPS_3DDEVICE | DDSCAPS_COMPLEX;
		if (m_bOnlySystemMemory) {
			ddsd.ddsCaps.dwCaps |= DDSCAPS_SYSTEMMEMORY;
		}
		ddsd.dwBackBufferCount = 1;
		result = CreateDDSurface(&ddsd, &m_pFrontBuffer, NULL);
		if (result != DD_OK) {
			Error("CreateSurface for front/back fullScreen buffer failed", result);
			return FALSE;
		}
		ddscaps.dwCaps = DDSCAPS_BACKBUFFER;
		result = m_pFrontBuffer->GetAttachedSurface(&ddscaps, &m_pBackBuffer);
		if (result != DD_OK) {
			Error("GetAttachedSurface failed to get back buffer", result);
			return FALSE;
		}
		if (!GetDDSurfaceDesc(&ddsd, m_pBackBuffer)) {
			return FALSE;
		}
	}
	else {
		memset(&ddsd, 0, sizeof(DDSURFACEDESC));
		ddsd.dwSize = sizeof(DDSURFACEDESC);
		ddsd.dwFlags = DDSD_CAPS;
		ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE;
		result = CreateDDSurface(&ddsd, &m_pFrontBuffer, NULL);
		if (result != DD_OK) {
			Error("CreateSurface for window front buffer failed", result);
			return FALSE;
		}
		ddsd.dwFlags = DDSD_WIDTH | DDSD_HEIGHT | DDSD_CAPS;
		ddsd.dwHeight = m_currentMode.height;
		ddsd.dwWidth = m_currentMode.width;
		ddsd.ddsCaps.dwCaps = DDSCAPS_OFFSCREENPLAIN | DDSCAPS_3DDEVICE;
		if (m_bOnlySystemMemory) {
			ddsd.ddsCaps.dwCaps |= DDSCAPS_SYSTEMMEMORY;
		}
		result = CreateDDSurface(&ddsd, &m_pBackBuffer, NULL);
		if (result != DD_OK) {
			Error("CreateSurface for window back buffer failed", result);
			return FALSE;
		}
		if (!GetDDSurfaceDesc(&ddsd, m_pBackBuffer)) {
			return FALSE;
		}

		result = m_pDirectDraw->CreateClipper(0, &m_pClipper, NULL);
		if (result != DD_OK) {
			Error("CreateClipper failed", result);
			return FALSE;
		}
		result = m_pClipper->SetHWnd(0, m_hWndMain);
		if (result != DD_OK) {
			Error("Clipper SetHWnd failed", result);
			return FALSE;
		}
		result = m_pFrontBuffer->SetClipper(m_pClipper);
		if (result != DD_OK) {
			Error("SetClipper failed", result);
			return FALSE;
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e020
void MxDirectDraw::FUN_1009e020()
{
	HRESULT result;
	byte* line;
	DDSURFACEDESC ddsd;
	int j;
	int count = m_bFlipSurfaces ? 2 : 1;

	for (int i = 0; i < count; i++) {
		memset(&ddsd, 0, sizeof(ddsd));
		ddsd.dwSize = sizeof(ddsd);

		result = m_pBackBuffer->Lock(NULL, &ddsd, 1, NULL);
		if (result == DDERR_SURFACELOST) {
			m_pBackBuffer->Restore();
			result = m_pBackBuffer->Lock(NULL, &ddsd, 1, NULL);
		}

		if (result != DD_OK) {
			// lock failed
			return;
		}

		// clear backBuffer
		line = (byte*) ddsd.lpSurface;
		for (j = ddsd.dwHeight; j--;) {
			memset(line, 0, ddsd.dwWidth);
			line += ddsd.lPitch;
		}

		m_pBackBuffer->Unlock(ddsd.lpSurface);

		if (m_bFlipSurfaces) {
			m_pFrontBuffer->Flip(NULL, DDFLIP_WAIT);
		}
	}
}

// FUNCTION: LEGO1 0x1009e110
BOOL MxDirectDraw::TextToTextSurface(const char* text, IDirectDrawSurface* pSurface, SIZE& textSizeOnSurface)
{
	HRESULT result;
	HDC hdc;
	RECT rc;
	size_t textLength;

	if (!pSurface) {
		return FALSE;
	}

	result = pSurface->GetDC(&hdc);
	if (result != DD_OK) {
		Error("GetDC for text surface failed", result);
		return FALSE;
	}

	textLength = strlen(text);

	SelectObject(hdc, m_hFont);
	SetTextColor(hdc, RGB(255, 255, 0));
	SetBkColor(hdc, RGB(0, 0, 0));
	SetBkMode(hdc, OPAQUE);
	GetTextExtentPoint32(hdc, text, textLength, &textSizeOnSurface);
	SetRect(&rc, 0, 0, textSizeOnSurface.cx, textSizeOnSurface.cy);
	ExtTextOut(hdc, 0, 0, ETO_OPAQUE, &rc, text, textLength, NULL);
	pSurface->ReleaseDC(hdc);

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e210
BOOL MxDirectDraw::TextToTextSurface1(const char* text)
{
	return TextToTextSurface(text, m_pText1Surface, m_text1SizeOnSurface);
}

// FUNCTION: LEGO1 0x1009e230
BOOL MxDirectDraw::TextToTextSurface2(const char* text)
{
	return TextToTextSurface(text, m_pText2Surface, m_text2SizeOnSurface);
}

// FUNCTION: LEGO1 0x1009e250
BOOL MxDirectDraw::CreateTextSurfaces()
{
	HRESULT result;
	DDCOLORKEY ddck;
	DDSURFACEDESC ddsd;
	HDC hdc;
	char dummyinfo[] = "000x000x00 (RAMP) 0000";
	char dummyfps[] = "000.00 fps (000.00 fps (000.00 fps) 00000 tps)";

	if (m_hFont != NULL) {
		DeleteObject(m_hFont);
	}
	m_hFont = CreateFont(
		m_currentMode.width <= 600 ? 12 : 24,
		0,
		0,
		0,
		FW_NORMAL,
		FALSE,
		FALSE,
		FALSE,
		ANSI_CHARSET,
		OUT_DEFAULT_PRECIS,
		CLIP_DEFAULT_PRECIS,
		DEFAULT_QUALITY,
		VARIABLE_PITCH,
		"Arial"
	);

	hdc = GetDC(NULL);
	SelectObject(hdc, m_hFont);
	GetTextExtentPoint(hdc, dummyfps, strlen(dummyfps), &m_text1SizeOnSurface);
	GetTextExtentPoint(hdc, dummyinfo, strlen(dummyinfo), &m_text2SizeOnSurface);
	ReleaseDC(NULL, hdc);

	memset(&ddsd, 0, sizeof(ddsd));
	ddsd.dwSize = sizeof(ddsd);
	ddsd.dwFlags = DDSD_CAPS | DDSD_HEIGHT | DDSD_WIDTH;
	ddsd.ddsCaps.dwCaps = DDSCAPS_OFFSCREENPLAIN;
	if (m_bOnlySystemMemory) {
		ddsd.ddsCaps.dwCaps |= DDSCAPS_SYSTEMMEMORY;
	}
	ddsd.dwHeight = m_text1SizeOnSurface.cy;
	ddsd.dwWidth = m_text1SizeOnSurface.cx;
	result = CreateDDSurface(&ddsd, &m_pText1Surface, NULL);
	if (result != DD_OK) {
		Error("CreateSurface for text surface 1 failed", result);
		return FALSE;
	}
	memset(&ddck, 0, sizeof(ddck));
	m_pText1Surface->SetColorKey(DDCKEY_SRCBLT, &ddck);
	if (!TextToTextSurface1(dummyfps)) {
		return FALSE;
	}

	memset(&ddsd, 0, sizeof(ddsd));
	ddsd.dwSize = sizeof(ddsd);
	ddsd.dwFlags = DDSD_CAPS | DDSD_HEIGHT | DDSD_WIDTH;
	ddsd.ddsCaps.dwCaps = DDSCAPS_OFFSCREENPLAIN;
	if (m_bOnlySystemMemory) {
		ddsd.ddsCaps.dwCaps |= DDSCAPS_SYSTEMMEMORY;
	}
	ddsd.dwHeight = m_text2SizeOnSurface.cy;
	ddsd.dwWidth = m_text2SizeOnSurface.cx;
	result = CreateDDSurface(&ddsd, &m_pText2Surface, NULL);
	if (result != DD_OK) {
		Error("CreateSurface for text surface 2 failed", result);
		return FALSE;
	}
	memset(&ddck, 0, sizeof(ddck));
	m_pText2Surface->SetColorKey(DDCKEY_SRCBLT, &ddck);
	if (!TextToTextSurface2(dummyinfo)) {
		return FALSE;
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e4d0
BOOL MxDirectDraw::RestoreSurfaces()
{
	HRESULT result;

	if (m_pFrontBuffer != NULL) {
		if (m_pFrontBuffer->IsLost() == DDERR_SURFACELOST) {
			result = m_pFrontBuffer->Restore();
			if (result != DD_OK) {
				Error("Restore of front buffer failed", result);
				return FALSE;
			}
		}
	}

	if (m_pBackBuffer != NULL) {
		if (m_pBackBuffer->IsLost() == DDERR_SURFACELOST) {
			result = m_pBackBuffer->Restore();
			if (result != DD_OK) {
				Error("Restore of back buffer failed", result);
				return FALSE;
			}
		}
	}

	if (m_pZBuffer != NULL) {
		if (m_pZBuffer->IsLost() == DDERR_SURFACELOST) {
			result = m_pZBuffer->Restore();
			if (result != DD_OK) {
				Error("Restore of Z-buffer failed", result);
				return FALSE;
			}
		}
	}

	if (m_pText1Surface != NULL) {
		if (m_pText1Surface->IsLost() == DDERR_SURFACELOST) {
			result = m_pText1Surface->Restore();
			if (result != DD_OK) {
				Error("Restore of text surface 1 failed", result);
				return FALSE;
			}
		}
	}

	if (m_pText2Surface != NULL) {
		if (m_pText2Surface->IsLost() == DDERR_SURFACELOST) {
			result = m_pText2Surface->Restore();
			if (result != DD_OK) {
				Error("Restore of text surface 2 failed", result);
				return FALSE;
			}
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e5e0
BOOL MxDirectDraw::CreateZBuffer(DWORD memorytype, DWORD depth)
{
	HRESULT result;                // eax
	LPDIRECTDRAWSURFACE lpZBuffer; // [esp+8h] [ebp-70h] BYREF
	DDSURFACEDESC ddsd;

	memset(&ddsd, 0, sizeof(ddsd));
	ddsd.dwSize = sizeof(ddsd);
	ddsd.dwFlags = DDSD_WIDTH | DDSD_HEIGHT | DDSD_CAPS | DDSD_ZBUFFERBITDEPTH;
	ddsd.dwHeight = m_currentMode.height;
	ddsd.dwWidth = m_currentMode.width;
	ddsd.dwZBufferBitDepth = depth;
	ddsd.ddsCaps.dwCaps = DDSCAPS_ZBUFFER | memorytype;
	result = CreateDDSurface(&ddsd, &lpZBuffer, 0);

	if (result != DD_OK) {
		Error("CreateSurface for fullScreen Z-buffer failed", result);
		return FALSE;
	}

	result = m_pBackBuffer->AddAttachedSurface(lpZBuffer);
	if (result != DD_OK) {
		Error("AddAttachedBuffer failed for Z-Buffer", result);
		return FALSE;
	}

	m_pZBuffer = lpZBuffer;
	return TRUE;
}

// FUNCTION: LEGO1 0x1009e6a0
int MxDirectDraw::Pause(BOOL p_pause)
{
	if (p_pause) {
		++m_pauseCount;

		if (m_pauseCount > 1) {
			return TRUE;
		}

		if (!RestoreOriginalPaletteEntries()) {
			return FALSE;
		}

		if (m_bFullScreen) {
			if (!FlipToGDISurface()) {
				return FALSE;
			}

			DrawMenuBar(m_hWndMain);
			RedrawWindow(m_hWndMain, NULL, NULL, RDW_FRAME);
		}
	}
	else {
		--m_pauseCount;

		if (m_pauseCount > 0) {
			return TRUE;
		}

		if (m_pauseCount < 0) {
			m_pauseCount = 0;
		}

		if (!RestorePaletteEntries()) {
			return FALSE;
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e750
BOOL MxDirectDraw::RestorePaletteEntries()
{

	if (m_bFullScreen && m_bPrimaryPalettized) {
		if (m_pPalette) {
			HRESULT result;

			result =
				m_pPalette->SetEntries(0, 0, sizeof(m_paletteEntries) / sizeof(m_paletteEntries[0]), m_paletteEntries);
			if (result != DD_OK) {
				Error("SetEntries failed", result);
				return FALSE;
			}
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e7a0
BOOL MxDirectDraw::RestoreOriginalPaletteEntries()
{
	if (m_bPrimaryPalettized) {
		if (m_pPalette) {
			HRESULT result;

			result = m_pPalette->SetEntries(
				0,
				0,
				sizeof(m_originalPaletteEntries) / sizeof(m_originalPaletteEntries[0]),
				m_originalPaletteEntries
			);
			if (result != DD_OK) {
				Error("SetEntries failed", result);
				return FALSE;
			}
		}
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e7f0
int MxDirectDraw::FlipToGDISurface()
{

	if (m_pDirectDraw) {
		HRESULT result;

		result = m_pDirectDraw->FlipToGDISurface();
		if (result != DD_OK) {
			Error("FlipToGDISurface failed", result);
		}
		return (result == DD_OK);
	}

	return TRUE;
}

// FUNCTION: LEGO1 0x1009e830
void MxDirectDraw::Error(const char* p_message, int p_error)
{
	// GLOBAL: LEGO1 0x10100c70
	static BOOL g_isInsideError = FALSE;
	if (!g_isInsideError) {
		g_isInsideError = TRUE;
		Destroy();
		if (m_pErrorHandler) {
			m_pErrorHandler(p_message, p_error, m_pErrorHandlerArg);
		}
		g_isInsideError = FALSE;
	}
}

// FUNCTION: LEGO1 0x1009e880
const char* MxDirectDraw::ErrorToString(HRESULT p_error)
{
	switch (p_error) {
	case DD_OK:
		return "No error.";
	case DDERR_ALREADYINITIALIZED:
		return "This object is already initialized.";
	case DDERR_BLTFASTCANTCLIP:
		return "Return if a clipper object is attached to the source surface passed into a BltFast call.";
	case DDERR_CANNOTATTACHSURFACE:
		return "This surface can not be attached to the requested surface.";
	case DDERR_CANNOTDETACHSURFACE:
		return "This surface can not be detached from the requested surface.";
	case DDERR_CANTCREATEDC:
		return "Windows can not create any more DCs.";
	case DDERR_CANTDUPLICATE:
		return "Can't duplicate primary & 3D surfaces, or surfaces that are implicitly created.";
	case DDERR_CLIPPERISUSINGHWND:
		return "An attempt was made to set a cliplist for a clipper object that is already monitoring an hwnd.";
	case DDERR_COLORKEYNOTSET:
		return "No src color key specified for this operation.";
	case DDERR_CURRENTLYNOTAVAIL:
		return "Support is currently not available.";
	case DDERR_DIRECTDRAWALREADYCREATED:
		return "A DirectDraw object representing this driver has already been created for this process.";
	case DDERR_EXCEPTION:
		return "An exception was encountered while performing the requested operation.";
	case DDERR_EXCLUSIVEMODEALREADYSET:
		return "An attempt was made to set the cooperative level when it was already set to exclusive.";
	case DDERR_GENERIC:
		return "Generic failure.";
	case DDERR_HEIGHTALIGN:
		return "Height of rectangle provided is not a multiple of reqd alignment.";
	case DDERR_HWNDALREADYSET:
		return "The CooperativeLevel HWND has already been set. It can not be reset while the process has surfaces or "
			   "palettes created.";
	case DDERR_HWNDSUBCLASSED:
		return "HWND used by DirectDraw CooperativeLevel has been subclassed, this prevents DirectDraw from restoring "
			   "state.";
	case DDERR_IMPLICITLYCREATED:
		return "This surface can not be restored because it is an implicitly created surface.";
	case DDERR_INCOMPATIBLEPRIMARY:
		return "Unable to match primary surface creation request with existing primary surface.";
	case DDERR_INVALIDCAPS:
		return "One or more of the caps bits passed to the callback are incorrect.";
	case DDERR_INVALIDCLIPLIST:
		return "DirectDraw does not support the provided cliplist.";
	case DDERR_INVALIDDIRECTDRAWGUID:
		return "The GUID passed to DirectDrawCreate is not a valid DirectDraw driver identifier.";
	case DDERR_INVALIDMODE:
		return "DirectDraw does not support the requested mode.";
	case DDERR_INVALIDOBJECT:
		return "DirectDraw received a pointer that was an invalid DIRECTDRAW object.";
	case DDERR_INVALIDPARAMS:
		return "One or more of the parameters passed to the function are incorrect.";
	case DDERR_INVALIDPIXELFORMAT:
		return "The pixel format was invalid as specified.";
	case DDERR_INVALIDPOSITION:
		return "Returned when the position of the overlay on the destination is no longer legal for that "
			   "destination.";
	case DDERR_INVALIDRECT:
		return "Rectangle provided was invalid.";
	case DDERR_LOCKEDSURFACES:
		return "Operation could not be carried out because one or more surfaces are locked.";
	case DDERR_NO3D:
		return "There is no 3D present.";
	case DDERR_NOALPHAHW:
		return "Operation could not be carried out because there is no alpha accleration hardware present or "
			   "available.";
	case DDERR_NOBLTHW:
		return "No blitter hardware present.";
	case DDERR_NOCLIPLIST:
		return "No cliplist available.";
	case DDERR_NOCLIPPERATTACHED:
		return "No clipper object attached to surface object.";
	case DDERR_NOCOLORCONVHW:
		return "Operation could not be carried out because there is no color conversion hardware present or "
			   "available.";
	case DDERR_NOCOLORKEY:
		return "Surface doesn't currently have a color key";
	case DDERR_NOCOLORKEYHW:
		return "Operation could not be carried out because there is no hardware support of the destination color "
			   "key.";
	case DDERR_NOCOOPERATIVELEVELSET:
		return "Create function called without DirectDraw object method SetCooperativeLevel being called.";
	case DDERR_NODC:
		return "No DC was ever created for this surface.";
	case DDERR_NODDROPSHW:
		return "No DirectDraw ROP hardware.";
	case DDERR_NODIRECTDRAWHW:
		return "A hardware-only DirectDraw object creation was attempted but the driver did not support any "
			   "hardware.";
	case DDERR_NOEMULATION:
		return "Software emulation not available.";
	case DDERR_NOEXCLUSIVEMODE:
		return "Operation requires the application to have exclusive mode but the application does not have exclusive "
			   "mode.";
	case DDERR_NOFLIPHW:
		return "Flipping visible surfaces is not supported.";
	case DDERR_NOGDI:
		return "There is no GDI present.";
	case DDERR_NOHWND:
		return "Clipper notification requires an HWND or no HWND has previously been set as the CooperativeLevel "
			   "HWND.";
	case DDERR_NOMIRRORHW:
		return "Operation could not be carried out because there is no hardware present or available.";
	case DDERR_NOOVERLAYDEST:
		return "Returned when GetOverlayPosition is called on an overlay that UpdateOverlay has never been called on "
			   "to establish a destination.";
	case DDERR_NOOVERLAYHW:
		return "Operation could not be carried out because there is no overlay hardware present or available.";
	case DDERR_NOPALETTEATTACHED:
		return "No palette object attached to this surface.";
	case DDERR_NOPALETTEHW:
		return "No hardware support for 16 or 256 color palettes.";
	case DDERR_NORASTEROPHW:
		return "Operation could not be carried out because there is no appropriate raster op hardware present or "
			   "available.";
	case DDERR_NOROTATIONHW:
		return "Operation could not be carried out because there is no rotation hardware present or available.";
	case DDERR_NOSTRETCHHW:
		return "Operation could not be carried out because there is no hardware support for stretching.";
	case DDERR_NOT4BITCOLOR:
		return "DirectDrawSurface is not in 4 bit color palette and the requested operation requires 4 bit color "
			   "palette.";
	case DDERR_NOT4BITCOLORINDEX:
		return "DirectDrawSurface is not in 4 bit color index palette and the requested operation requires 4 bit color "
			   "index palette.";
	case DDERR_NOT8BITCOLOR:
		return "DirectDrawSurface is not in 8 bit color mode and the requested operation requires 8 bit color.";
	case DDERR_NOTAOVERLAYSURFACE:
		return "Returned when an overlay member is called for a non-overlay surface.";
	case DDERR_NOTEXTUREHW:
		return "Operation could not be carried out because there is no texture mapping hardware present or "
			   "available.";
	case DDERR_NOTFLIPPABLE:
		return "An attempt has been made to flip a surface that is not flippable.";
	case DDERR_NOTFOUND:
		return "Requested item was not found.";
	case DDERR_NOTLOCKED:
		return "Surface was not locked.  An attempt to unlock a surface that was not locked at all, or by this "
			   "process, has been attempted.";
	case DDERR_NOTPALETTIZED:
		return "The surface being used is not a palette-based surface.";
	case DDERR_NOVSYNCHW:
		return "Operation could not be carried out because there is no hardware support for vertical blank "
			   "synchronized operations.";
	case DDERR_NOZBUFFERHW:
		return "Operation could not be carried out because there is no hardware support for zbuffer blitting.";
	case DDERR_NOZOVERLAYHW:
		return "Overlay surfaces could not be z layered based on their BltOrder because the hardware does not support "
			   "z layering of overlays.";
	case DDERR_OUTOFCAPS:
		return "The hardware needed for the requested operation has already been allocated.";
	case DDERR_OUTOFMEMORY:
		return "DirectDraw does not have enough memory to perform the operation.";
	case DDERR_OUTOFVIDEOMEMORY:
		return "DirectDraw does not have enough memory to perform the operation.";
	case DDERR_OVERLAYCANTCLIP:
		return "The hardware does not support clipped overlays.";
	case DDERR_OVERLAYCOLORKEYONLYONEACTIVE:
		return "Can only have ony color key active at one time for overlays.";
	case DDERR_OVERLAYNOTVISIBLE:
		return "Returned when GetOverlayPosition is called on a hidden overlay.";
	case DDERR_PALETTEBUSY:
		return "Access to this palette is being refused because the palette is already locked by another thread.";
	case DDERR_PRIMARYSURFACEALREADYEXISTS:
		return "This process already has created a primary surface.";
	case DDERR_REGIONTOOSMALL:
		return "Region passed to Clipper::GetClipList is too small.";
	case DDERR_SURFACEALREADYATTACHED:
		return "This surface is already attached to the surface it is being attached to.";
	case DDERR_SURFACEALREADYDEPENDENT:
		return "This surface is already a dependency of the surface it is being made a dependency of.";
	case DDERR_SURFACEBUSY:
		return "Access to this surface is being refused because the surface is already locked by another thread.";
	case DDERR_SURFACEISOBSCURED:
		return "Access to surface refused because the surface is obscured.";
	case DDERR_SURFACELOST:
		return "Access to this surface is being refused because the surface memory is gone. The DirectDrawSurface "
			   "object representing this surface should have Restore called on it.";
	case DDERR_SURFACENOTATTACHED:
		return "The requested surface is not attached.";
	case DDERR_TOOBIGHEIGHT:
		return "Height requested by DirectDraw is too large.";
	case DDERR_TOOBIGSIZE:
		return "Size requested by DirectDraw is too large, but the individual height and width are OK.";
	case DDERR_TOOBIGWIDTH:
		return "Width requested by DirectDraw is too large.";
	case DDERR_UNSUPPORTED:
		return "Action not supported.";
	case DDERR_UNSUPPORTEDFORMAT:
		return "FOURCC format requested is unsupported by DirectDraw.";
	case DDERR_UNSUPPORTEDMASK:
		return "Bitmask in the pixel format requested is unsupported by DirectDraw.";
	case DDERR_VERTICALBLANKINPROGRESS:
		return "Vertical blank is in progress.";
	case DDERR_WASSTILLDRAWING:
		return "Informs DirectDraw that the previous Blt which is transfering information to or from this Surface is "
			   "incomplete.";
	case DDERR_WRONGMODE:
		return "This surface can not be restored because it was created in a different mode.";
	case DDERR_XALIGN:
		return "Rectangle provided was not horizontally aligned on required boundary.";
	default:
		return "Unrecognized error value.";
	}
}

// FUNCTION: LEGO1 0x1009efb0
MxDirectDraw::DeviceModesInfo::DeviceModesInfo()
{
	memset(this, 0, sizeof(*this));
}

// FUNCTION: LEGO1 0x1009efd0
MxDirectDraw::DeviceModesInfo::~DeviceModesInfo()
{
	if (m_guid != NULL) {
		delete m_guid;
	}

	if (m_modeArray != NULL) {
		delete[] m_modeArray;
	}
}