Magic_Game/core/modules/Window.cpp

#include "..\e2dmodule.h"
#include "..\e2dmanager.h"
#include "..\e2dobject.h"
#include <imm.h>
#pragma comment (lib ,"imm32.lib")

#define WINDOW_STYLE	WS_OVERLAPPEDWINDOW & ~WS_MAXIMIZEBOX & ~WS_THICKFRAME
#define REGISTER_CLASS	L"Easy2DApp"


e2d::Window * e2d::Window::instance_ = nullptr;

e2d::Window * e2d::Window::GetInstance()
{
	if (!instance_)
		instance_ = new (std::nothrow) Window;
	return instance_;
}

void e2d::Window::DestroyInstance()
{
	if (instance_)
	{
		delete instance_;
		instance_ = nullptr;
	}
}

e2d::Size e2d::Window::GetScreenSize()
{
	int screen_width = ::GetSystemMetrics(SM_CXSCREEN);
	int screen_height = ::GetSystemMetrics(SM_CYSCREEN);
	Size screen_size(
		static_cast<float>(screen_width),
		static_cast<float>(screen_height)
	);
	return std::move(screen_size);
}

e2d::Window::Window()
	: hwnd_(nullptr)
	, width_(640)
	, height_(480)
	, title_(L"Easy2D Game")
	, icon_id_(0)
	, dpi_(0.f)
{
	::CoInitialize(nullptr);

	// <20><>ȡϵͳ DPI
	Renderer::GetFactory()->GetDesktopDpi(&dpi_, &dpi_);
}

e2d::Window::~Window()
{
	if (::GetConsoleWindow())
		::FreeConsole();

	if (hwnd_)
		::DestroyWindow(hwnd_);

	::CoUninitialize();
}

bool e2d::Window::CheckUnique()
{
	HANDLE mutex = ::CreateMutex(nullptr, TRUE, LPCWSTR(L"Easy2DApp-" + title_));

	if (mutex == nullptr)
	{
		WARN("CreateMutex Failed!");
		return false;
	}
	else if (::GetLastError() == ERROR_ALREADY_EXISTS)
	{
		::CloseHandle(mutex);
		return false;
	}
	return true;
}

e2d::Rect e2d::Window::Locate(int width, int height)
{
	Rect result;
	RECT rect		= { 0, 0, LONG(ceil(width * dpi_ / 96.f)), LONG(ceil(height * dpi_ / 96.f)) };
	int max_width	= ::GetSystemMetrics(SM_CXSCREEN);
	int max_height	= ::GetSystemMetrics(SM_CYSCREEN);

	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʵĴ<CAB5><C4B4>ڴ<EFBFBD>С
	::AdjustWindowRectEx(&rect, WINDOW_STYLE, FALSE, NULL);
	width = static_cast<int>(rect.right - rect.left);
	height = static_cast<int>(rect.bottom - rect.top);

	// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD><C4B4>ڴ<EFBFBD>С<EFBFBD>ȷֱ<C8B7><D6B1>ʴ<EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	WARN_IF(max_width < width || max_height < height, "The window Is larger than screen!");
	width = std::min(width, max_width);
	height = std::min(height, max_height);

	float x = float((max_width - width) / 2), y = float((max_height - height) / 2);
	return std::move(Rect(x, y, float(width), float(height)));
}

void e2d::Window::Poll()
{
	while (::PeekMessage(&msg_, nullptr, 0, 0, PM_REMOVE))
	{
		::TranslateMessage(&msg_);
		::DispatchMessage(&msg_);
	}
}

int e2d::Window::GetWidth() const
{
	return width_;
}

int e2d::Window::GetHeight() const
{
	return height_;
}

e2d::Size e2d::Window::GetSize() const
{
	e2d::Size size(
		static_cast<float>(width_),
		static_cast<float>(height_)
	);
	return std::move(size);
}

float e2d::Window::GetDpi() const
{
	return dpi_;
}

const e2d::String& e2d::Window::GetTitle() const
{
	return title_;
}

HWND e2d::Window::GetHWnd()
{
	if (!hwnd_)
	{
		WNDCLASSEX wcex = { 0 };
		wcex.cbSize = sizeof(WNDCLASSEX);
		wcex.lpszClassName = REGISTER_CLASS;
		wcex.style = CS_HREDRAW | CS_VREDRAW | CS_DBLCLKS;
		wcex.lpfnWndProc = Window::WndProc;
		wcex.hIcon = nullptr;
		wcex.cbClsExtra = 0;
		wcex.cbWndExtra = sizeof(LONG_PTR);
		wcex.hInstance = HINST_THISCOMPONENT;
		wcex.hbrBackground = nullptr;
		wcex.lpszMenuName = nullptr;
		wcex.hCursor = ::LoadCursor(nullptr, IDC_ARROW);

		if (icon_id_ != 0)
		{
			wcex.hIcon = (HICON)::LoadImage(
				HINST_THISCOMPONENT,
				MAKEINTRESOURCE(icon_id_),
				IMAGE_ICON,
				0,
				0,
				LR_DEFAULTCOLOR | LR_CREATEDIBSECTION | LR_DEFAULTSIZE
			);
		}

		// ע<>ᴰ<EFBFBD><E1B4B0><EFBFBD><EFBFBD>
		RegisterClassEx(&wcex);

		// <20><><EFBFBD>㴰<EFBFBD>ڴ<EFBFBD>С
		Rect clientRect = Locate(width_, height_);

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
		hwnd_ = ::CreateWindowEx(
			NULL,
			REGISTER_CLASS,
			(LPCTSTR)title_,
			WINDOW_STYLE,
			int(clientRect.origin.x),
			int(clientRect.origin.y),
			int(clientRect.size.width),
			int(clientRect.size.height),
			nullptr,
			nullptr,
			HINST_THISCOMPONENT,
			this
		);

		if (hwnd_)
		{
			// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>뷨
			SetTypewritingEnabled(false);
			// <20><><EFBFBD>ÿ<EFBFBD><C3BF><EFBFBD>̨<EFBFBD>رհ<D8B1>ť
			HWND console_hwnd = ::GetConsoleWindow();
			if (console_hwnd)
			{
				HMENU hmenu = ::GetSystemMenu(console_hwnd, FALSE);
				::RemoveMenu(hmenu, SC_CLOSE, MF_BYCOMMAND);
			}
		}
		else
		{
			::UnregisterClass(REGISTER_CLASS, HINST_THISCOMPONENT);
			throw RuntimeException("Create window failed");
		}
	}
	return hwnd_;
}

void e2d::Window::SetSize(int width, int height)
{
	if (width_ == width && height_ == height)
		return;

	width_ = width;
	height_ = height;

	if (hwnd_)
	{
		Rect rect = Locate(width, height);
		::MoveWindow(
			hwnd_,
			int(rect.origin.x),
			int(rect.origin.y),
			int(rect.size.width),
			int(rect.size.height),
			TRUE
		);
	}
}

void e2d::Window::SetTitle(const String& title)
{
	title_ = title;
	if (hwnd_)
	{
		::SetWindowText(hwnd_, (LPCWSTR)title);
	}
}

void e2d::Window::SetIcon(int resource_id)
{
	this->icon_id_ = resource_id;
	if (hwnd_)
	{
		HICON icon = (HICON)::LoadImage(
			HINST_THISCOMPONENT,
			MAKEINTRESOURCE(resource_id),
			IMAGE_ICON,
			0,
			0,
			LR_DEFAULTCOLOR | LR_CREATEDIBSECTION | LR_DEFAULTSIZE
		);
		// <20><><EFBFBD>ô<EFBFBD><C3B4>ڵ<EFBFBD>ͼ<EFBFBD><CDBC>
		::SendMessage(hwnd_, WM_SETICON, ICON_BIG, (LPARAM)icon);
		::SendMessage(hwnd_, WM_SETICON, ICON_SMALL, (LPARAM)icon);
	}
}

void e2d::Window::SetCursor(Cursor cursor)
{
	LPCWSTR cursor_name = nullptr;
	switch (cursor)
	{
	case Cursor::Normal:
		cursor_name = IDC_ARROW;
		break;

	case Cursor::Hand:
		cursor_name = IDC_HAND;
		break;

	case Cursor::No:
		cursor_name = IDC_NO;
		break;

	case Cursor::Wait:
		cursor_name = IDC_WAIT;
		break;

	case Cursor::ArrowWait:
		cursor_name = IDC_APPSTARTING;
		break;

	default:
		break;
	}

	HCURSOR hcursor = ::LoadCursor(nullptr, cursor_name);
	if (hcursor)
	{
		::SetCursor(hcursor);
	}
}

void e2d::Window::SetConsoleEnabled(bool enabled)
{
	// <20><><EFBFBD><EFBFBD><EFBFBD>Ѵ<EFBFBD><D1B4>ڵĿ<DAB5><C4BF><EFBFBD>̨<EFBFBD><CCA8><EFBFBD><EFBFBD>
	HWND hwnd = ::GetConsoleWindow();
	// <20>رտ<D8B1><D5BF><EFBFBD>̨
	if (enabled)
	{
		if (hwnd)
		{
			::ShowWindow(hwnd, SW_SHOWNORMAL);
		}
		else
		{
			// <20><>ʾһ<CABE><D2BB><EFBFBD>¿<EFBFBD><C2BF><EFBFBD>̨
			if (::AllocConsole())
			{
				hwnd = ::GetConsoleWindow();
				// <20>ض<EFBFBD><D8B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
				FILE * stdoutStream, * stdinStream, * stderrStream;
				freopen_s(&stdoutStream, "conout$", "w+t", stdout);
				freopen_s(&stdinStream, "conin$", "r+t", stdin);
				freopen_s(&stderrStream, "conout$", "w+t", stderr);
				// <20><><EFBFBD>ÿ<EFBFBD><C3BF><EFBFBD>̨<EFBFBD>رհ<D8B1>ť
				HMENU hmenu = ::GetSystemMenu(hwnd, FALSE);
				::RemoveMenu(hmenu, SC_CLOSE, MF_BYCOMMAND);
			}
		}
	}
	else 
	{
		if (hwnd)
		{
			::ShowWindow(hwnd, SW_HIDE);
		}
	}
}

void e2d::Window::SetTypewritingEnabled(bool enabled)
{
	static HIMC himc = nullptr;

	if (enabled)
	{
		if (himc != nullptr)
		{
			::ImmAssociateContext(GetHWnd(), himc);
			himc = nullptr;
		}
	}
	else
	{
		if (himc == nullptr)
		{
			himc = ::ImmAssociateContext(GetHWnd(), nullptr);
		}
	}
}

bool e2d::Window::Popup(const String & text, const String & title, PopupStyle style, bool has_cancel)
{
	UINT type = 0;
	switch (style)
	{
	case PopupStyle::Info:
		type = MB_ICONINFORMATION;
		break;
	case PopupStyle::Warning:
		type = MB_ICONWARNING;
		break;
	case PopupStyle::Error:
		type = MB_ICONERROR;
		break;
	default:
		break;
	}

	if (has_cancel)
	{
		type |= MB_OKCANCEL;
	}

	Game::GetInstance()->Pause();
	int ret = ::MessageBox(hwnd_, (LPCWSTR)text, (LPCWSTR)title, type);
	Game::GetInstance()->Resume();
	return ret == IDOK;
}


LRESULT e2d::Window::WndProc(HWND hwnd, UINT msg, WPARAM w_param, LPARAM l_param)
{
	LRESULT result = 0;

	if (msg == WM_CREATE)
	{
		LPCREATESTRUCT pcs = (LPCREATESTRUCT)l_param;
		Window *window = (Window *)pcs->lpCreateParams;

		::SetWindowLongPtrW(
			hwnd,
			GWLP_USERDATA,
			PtrToUlong(window)
		);

		result = 1;
	}
	else
	{
		bool has_handled = false;
		Window *window = reinterpret_cast<Window *>(
			static_cast<LONG_PTR>(
				::GetWindowLongPtrW(hwnd, GWLP_USERDATA)
			)
		);

		switch (msg)
		{

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ
		case WM_LBUTTONUP:
		case WM_LBUTTONDOWN:
		case WM_LBUTTONDBLCLK:
		case WM_MBUTTONUP:
		case WM_MBUTTONDOWN:
		case WM_MBUTTONDBLCLK:
		case WM_RBUTTONUP:
		case WM_RBUTTONDOWN:
		case WM_RBUTTONDBLCLK:
		case WM_MOUSEMOVE:
		case WM_MOUSEWHEEL:
		{
			auto game = Game::GetInstance();
			if (game->IsTransitioning())
				break;

			auto curr_scene = game->GetCurrentScene();
			if (curr_scene)
			{
				curr_scene->Dispatch(MouseEvent(msg, w_param, l_param));
			}
		}
		result = 0;
		has_handled = true;
		break;

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ
		case WM_KEYDOWN:
		case WM_KEYUP:
		{
			auto game = Game::GetInstance();
			if (game->IsTransitioning())
				break;

			auto curr_scene = game->GetCurrentScene();
			if (curr_scene)
			{
				curr_scene->Dispatch(KeyEvent(msg, w_param, l_param));
			}
		}
		result = 0;
		has_handled = true;
		break;

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڴ<EFBFBD>С<EFBFBD>仯<EFBFBD><E4BBAF>Ϣ
		case WM_SIZE:
		{
			UINT width = LOWORD(l_param);
			UINT height = HIWORD(l_param);

			if (w_param == SIZE_RESTORED)
			{
				window->width_ = static_cast<int>(width * 96.f / window->dpi_);
				window->height_ = static_cast<int>(height * 96.f / window->dpi_);
			}

			// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>յ<EFBFBD>һ<EFBFBD><D2BB> WM_SIZE <20><>Ϣ<EFBFBD><CFA2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ⱦ
			// Ŀ<><C4BF><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ܻ<EFBFBD><DCBB><EFBFBD><EFBFBD><EFBFBD>ʧ<EFBFBD>ܣ<EFBFBD><DCA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ժ<EFBFBD><D4BA><EFBFBD><EFBFBD>п<EFBFBD><D0BF>ܵ<EFBFBD>
			// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD>ε<EFBFBD><CEB5><EFBFBD> EndDraw ʱ<><CAB1><EFBFBD><EFBFBD>
			auto render_target = Renderer::GetInstance()->GetRenderTarget();
			if (render_target)
			{
				render_target->Resize(D2D1::SizeU(width, height));
			}
		}
		break;

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڱ<EFBFBD><DAB1><EFBFBD><EFBFBD>仯<EFBFBD><E4BBAF>Ϣ
		case WM_SETTEXT:
		{
			window->title_ = (const wchar_t*)l_param;
		}
		break;

		// <20><><EFBFBD><EFBFBD><EFBFBD>ֱ<EFBFBD><D6B1>ʱ仯<CAB1><E4BBAF>Ϣ
		case WM_DISPLAYCHANGE:
		{
			// <20>ػ<EFBFBD><D8BB>ͻ<EFBFBD><CDBB><EFBFBD>
			::InvalidateRect(hwnd, nullptr, FALSE);
		}
		result = 0;
		has_handled = true;
		break;

		// <20>ػ洰<D8BB><E6B4B0>
		case WM_PAINT:
		{
			Game::GetInstance()->DrawScene();
			::ValidateRect(hwnd, nullptr);
		}
		result = 0;
		has_handled = true;
		break;

		// <20><><EFBFBD>ڹر<DAB9><D8B1><EFBFBD>Ϣ
		case WM_CLOSE:
		{
			auto game = Game::GetInstance();
			auto currScene = game->GetCurrentScene();
			if (!currScene || currScene->OnCloseWindow())
			{
				game->Quit();
			}
		}
		result = 0;
		has_handled = true;
		break;

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ
		case WM_DESTROY:
		{
			::PostQuitMessage(0);
		}
		result = 1;
		has_handled = true;
		break;

		}

		if (!has_handled)
		{
			result = ::DefWindowProc(hwnd, msg, w_param, l_param);
		}
	}
	return result;
}