ScreenRender.cpp

#include "stdafx.h"
#include "ScreenRender.h"
#include "ProjControl.h"
#include "resource.h"

#define SetDlgBtnChecked(dlg, id, val) CheckDlgButton(dlg, id, (val ? BST_CHECKED : BST_UNCHECKED))
#define GetDlgBtnChecked(dlg, id) (IsDlgButtonChecked(dlg, id) ? TRUE : FALSE)

HINSTANCE hInst;
SCRCFG cfg;

// Rendering resources
HDC hBufferDC;

void UpdateBuffer()
{
	HBITMAP hBitOld;
	HBITMAP hBitNew;
	// Create a Bitmap which is compatible with the desktop
	HDC hDesktop = GetDC(NULL);
	hBitNew = CreateCompatibleBitmap(hDesktop, cfg.w, cfg.h);
	ReleaseDC(NULL, hDesktop);

	// Select the new Bitmap, free the old one
	hBitOld = (HBITMAP) SelectObject(hBufferDC, hBitNew);
	DeleteObject(hBitOld);
}

BOOL CALLBACK EnumScreens(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData)
{
	HWND hCombo = (HWND) dwData;
	MONITORINFOEX mInfo;
	mInfo.cbSize = sizeof(MONITORINFOEX);
	if (GetMonitorInfo(hMonitor, &mInfo)) {
		int id = ComboBox_AddString(hCombo, mInfo.szDevice);
		if (id >= 0) { // errors are negative
			MONITORINFOEX* pInfo = (MONITORINFOEX*) malloc(sizeof(MONITORINFOEX));
			if (pInfo != NULL) {
				memcpy(pInfo, &mInfo, sizeof(MONITORINFOEX));
				ComboBox_SetItemData(hCombo, id, (LPARAM) pInfo);
			} else {
				ComboBox_SetItemData(hCombo, id, NULL);
			}
		}
	}
	return TRUE;
}

INT_PTR CALLBACK ScrSelProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	HWND hCombo = GetDlgItem(hDlg, IDC_SCREENS);
	MONITORINFOEX* pInfo;
	int count, id, interval;
	BOOL bSuccess;
	UNREFERENCED_PARAMETER(lParam);
	switch (message)
	{
	case WM_INITDIALOG:
		// Set up the combo box
		ComboBox_ResetContent(hCombo); // Just to be sure
		// Create the default value
		ComboBox_AddString(hCombo, TEXT("(none)"));
		ComboBox_SetItemData(hCombo, 0, NULL);
		// Enumerate Monitors
		EnumDisplayMonitors(NULL, NULL, EnumScreens, (LPARAM) hCombo);
		if (ComboBox_SelectString(hCombo, -1, cfg.szScreenName) == CB_ERR) 
			ComboBox_SetCurSel(hCombo, 0);

		// Init the other stuff
		SetDlgItemInt(hDlg, IDC_TIME, cfg.iUpdateInterval, FALSE);
		SetDlgBtnChecked(hDlg, IDC_MOUSE, cfg.bShowCursor);
		SetDlgBtnChecked(hDlg, IDC_DETAIL, cfg.bHighQuality);
		return (INT_PTR)TRUE;
		break;

	case WM_COMMAND:
		switch(LOWORD(wParam))
		{
		case IDOK:
			// Save data
			cfg.bShowCursor = false;
			id = ComboBox_GetCurSel(hCombo);
			if (id >= 0) { // Negative is CB_ERR
				pInfo = (MONITORINFOEX*) ComboBox_GetItemData(hCombo, id);
				if (pInfo != NULL) {
					StringCbCopy(cfg.szScreenName, sizeof(cfg.szScreenName), pInfo->szDevice);
					cfg.x = pInfo->rcMonitor.left;
					cfg.y = pInfo->rcMonitor.top;
					cfg.w = pInfo->rcMonitor.right - cfg.x;
					cfg.h = pInfo->rcMonitor.bottom - cfg.y;
					cfg.bActive = true;
					UpdateBuffer();
				} else {
					cfg.bActive = false;
					cfg.szScreenName[0] = (TCHAR) 0;
				}
				ForceResize();
				ForceRedraw();
			}
			interval = GetDlgItemInt(hDlg, IDC_TIME, &bSuccess, FALSE);
			if (bSuccess) cfg.iUpdateInterval = interval;
			cfg.bShowCursor = GetDlgBtnChecked(hDlg, IDC_MOUSE);
			cfg.bHighQuality = GetDlgBtnChecked(hDlg, IDC_DETAIL);
			if (cfg.bActive) {
				StartTimedRedraw(cfg.iUpdateInterval);
			} else {
				StopTimedRedraw();
			}
			// NO BREAK
		case IDCANCEL:
			// Clean up the MonitorInfo structures
			count = ComboBox_GetCount(hCombo);
			for (int i=0; i<count; i++) {
				pInfo = (MONITORINFOEX*) ComboBox_GetItemData(hCombo, i);
				if (pInfo == NULL) continue;
				free(pInfo);
			}

			EndDialog(hDlg, LOWORD(wParam));
			return (INT_PTR)TRUE;
		}
		break;
	}
	return (INT_PTR)FALSE;
}

void Screen_Init(HINSTANCE hInstance)
{
	HDC hDesktop;
	hInst = hInstance;

	cfg.bActive = false;
	cfg.bHighQuality = false;
	cfg.bShowCursor = false;
	cfg.iUpdateInterval = 100;
	cfg.szScreenName[0] = (TCHAR) 0; // Default to empty string

	// Initialize the buffer DC
	hDesktop = GetDC(NULL);
	hBufferDC = CreateCompatibleDC(hDesktop);
	ReleaseDC(NULL, hDesktop);
}

void Screen_OpenSettingsBox(HWND parent)
{
	DialogBox(hInst, MAKEINTRESOURCE(IDD_SELECT_SCREEN), parent, ScrSelProc);
}

double Screen_GetRatio()
{
	if (cfg.bActive) {
		return (double)cfg.w / (double)cfg.h;
	} else {
		return 0.0;
	}
}

void Screen_Draw(HDC hdc)
{
//	TCHAR str[100];
//	size_t len;
	if (cfg.bActive)
	{
		RECT rcWnd = GetClientRect();
		HDC hDesktop = GetDC(NULL);
		BitBlt(hBufferDC, 0, 0, cfg.w, cfg.h, hDesktop, cfg.x, cfg.y, SRCCOPY);
		ReleaseDC(NULL, hDesktop);
/*		if (cfg.bActive) {
			StringCbPrintf(str, sizeof(str), TEXT("On %s, x = %i, y = %i, w = %i, h = %i\nClient w = %i, h = %i"), cfg.szScreenName, cfg.x, cfg.y, cfg.w, cfg.h, rcWnd.right, rcWnd.bottom);
		} else {
			StringCbPrintf(str, sizeof(str), TEXT("No screen selected\nClient w = %i, h = %i"), rcWnd.right, rcWnd.bottom);
		}
		StringCchLength(str, 100, &len);
		TextOut(hdc, 0, 0, str, (int)len); */
		SetStretchBltMode(hdc, cfg.bHighQuality ? HALFTONE : COLORONCOLOR);
		StretchBlt(hdc, 0, 0, rcWnd.right, rcWnd.bottom,
			hBufferDC, 0, 0, cfg.w, cfg.h, SRCCOPY);
	}
}