SimpleRemoter/server/2015Remote/CGridDialog.cpp

#include "stdafx.h"
#include "afxdialogex.h"
#include "CGridDialog.h"
#include "Resource.h"

BEGIN_MESSAGE_MAP(CGridDialog, CDialog)
	ON_WM_SIZE()
	ON_WM_LBUTTONDBLCLK()
	ON_MESSAGE(WM_CHILD_CLOSED, &CGridDialog::OnChildClosed)
END_MESSAGE_MAP()

CGridDialog::CGridDialog() : CDialog()
{
}

BOOL CGridDialog::OnInitDialog()
{
	m_hIcon = ::LoadIconA(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDR_MAINFRAME));
	SetIcon(m_hIcon, FALSE);

	CDialog::OnInitDialog();
	return TRUE;
}

void CGridDialog::SetGrid(int rows, int cols)
{
	m_rows = rows;
	m_cols = cols;
	m_max = rows * cols;
	LayoutChildren();
}

BOOL CGridDialog::AddChild(CDialog* pDlg)
{
	if (!pDlg || !::IsWindow(pDlg->GetSafeHwnd()) || m_children.size() >= m_max)
		return FALSE;

	pDlg->SetParent(this);
	pDlg->ShowWindow(SW_SHOW);

	// 去掉标题栏和调整大小
	LONG style = ::GetWindowLong(pDlg->GetSafeHwnd(), GWL_STYLE);
	style &= ~(WS_CAPTION | WS_THICKFRAME | WS_SIZEBOX | WS_BORDER);
	::SetWindowLong(pDlg->GetSafeHwnd(), GWL_STYLE, style);
	::SetWindowPos(pDlg->GetSafeHwnd(), nullptr, 0, 0, 0, 0, SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);

	m_children.push_back(pDlg);
	LayoutChildren();
	return TRUE;
}

void CGridDialog::RemoveChild(CDialog* pDlg)
{
	auto it = std::find(m_children.begin(), m_children.end(), pDlg);
	if (it != m_children.end())
	{
		(*it)->ShowWindow(SW_HIDE);
		(*it)->SetParent(nullptr);
		m_children.erase(it);

		// 删除 m_origState 中对应条目
		auto itState = m_origState.find(pDlg);
		if (itState != m_origState.end())
			m_origState.erase(itState);

		// 如果关闭的子窗口是当前最大化窗口，重置 m_pMaxChild
		if (m_pMaxChild == pDlg)
			m_pMaxChild = nullptr;

		LayoutChildren();
	}
}

LRESULT CGridDialog::OnChildClosed(WPARAM wParam, LPARAM lParam)
{
	CDialog* pDlg = (CDialog*)wParam;
	RemoveChild(pDlg);
	return 0;
}

void CGridDialog::LayoutChildren()
{
	if (m_children.size() == 0) {
		// 恢复父窗口标题栏
		if (m_parentStyle != 0)
		{
			::SetWindowLong(m_hWnd, GWL_STYLE, m_parentStyle);
			::SetWindowPos(m_hWnd, nullptr, 0, 0, 0, 0,
				SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);
			m_parentStyle = 0;
		}
		ShowWindow(SW_HIDE);
		return;
	}

	if (m_rows <= 0 || m_cols <= 0 || m_children.empty() || m_pMaxChild != nullptr)
		return;

	CRect rcClient;
	GetClientRect(&rcClient);

	int w = rcClient.Width() / m_cols;
	int h = rcClient.Height() / m_rows;

	for (size_t i = 0; i < m_children.size(); ++i)
	{
		int r = (int)i / m_cols;
		int c = (int)i % m_cols;

		if (r >= m_rows)
			break; // 超过网格范围

		int x = c * w;
		int y = r * h;

		m_children[i]->MoveWindow(x, y, w, h, TRUE);
		m_children[i]->ShowWindow(SW_SHOW);
	}
}

void CGridDialog::OnSize(UINT nType, int cx, int cy)
{
	CDialog::OnSize(nType, cx, cy);

	if (m_pMaxChild == nullptr) {
		LayoutChildren();
	}
	else {
		// 最大化状态下，保持铺满父对话框
		CRect rcClient;
		GetClientRect(&rcClient);
		m_pMaxChild->MoveWindow(rcClient, TRUE);
	}
}

void CGridDialog::OnLButtonDblClk(UINT nFlags, CPoint point)
{
	// 如果当前有最大化的子窗口，双击任何地方都恢复
	if (m_pMaxChild != nullptr)
	{
		// 恢复子窗口样式和位置
		for (auto& kv : m_origState)
		{
			CDialog* dlg = kv.first;
			const ChildState& state = kv.second;

			::SetWindowLong(dlg->GetSafeHwnd(), GWL_STYLE, state.style);
			::SetWindowPos(dlg->GetSafeHwnd(), nullptr, 0, 0, 0, 0,
				SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);

			dlg->MoveWindow(state.rect, TRUE);
			dlg->ShowWindow(SW_SHOW);
		}

		// 恢复父窗口标题栏
		if (m_parentStyle != 0)
		{
			::SetWindowLong(m_hWnd, GWL_STYLE, m_parentStyle);
			::SetWindowPos(m_hWnd, nullptr, 0, 0, 0, 0,
				SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);
			m_parentStyle = 0;
		}

		// 刷新父窗口
		m_pMaxChild = nullptr;
		m_origState.clear();
		LayoutChildren();
		return; // 已处理，返回
	}

	// 没有最大化子窗口，则按原逻辑找到点击的子窗口进行最大化
	for (auto dlg : m_children)
	{
		CRect rc;
		dlg->GetWindowRect(&rc);
		ScreenToClient(&rc);

		if (rc.PtInRect(point))
		{
			// 保存所有子窗口原始状态
			m_origState.clear();
			for (auto d : m_children)
			{
				ChildState state;
				d->GetWindowRect(&state.rect);
				ScreenToClient(&state.rect);
				state.style = ::GetWindowLong(d->GetSafeHwnd(), GWL_STYLE);
				m_origState[d] = state;
			}

			// 最大化点击的子窗口
			LONG style = m_origState[dlg].style;
			style |= (WS_CAPTION | WS_THICKFRAME | WS_SYSMENU | WS_MINIMIZEBOX | WS_MAXIMIZEBOX);
			::SetWindowLong(dlg->GetSafeHwnd(), GWL_STYLE, style);
			::SetWindowPos(dlg->GetSafeHwnd(), nullptr, 0, 0, 0, 0,
				SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);

			// 隐藏父窗口标题栏
			if (m_parentStyle == 0)
				m_parentStyle = ::GetWindowLong(m_hWnd, GWL_STYLE);
			LONG parentStyle = m_parentStyle & ~(WS_CAPTION | WS_THICKFRAME);
			::SetWindowLong(m_hWnd, GWL_STYLE, parentStyle);
			::SetWindowPos(m_hWnd, nullptr, 0, 0, 0, 0,
				SWP_NOZORDER | SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED);

			// 全屏显示子窗口
			CRect rcClient;
			GetClientRect(&rcClient);
			dlg->MoveWindow(rcClient, TRUE);
			m_pMaxChild = dlg;

			// 隐藏其他子窗口
			for (auto d : m_children)
				if (d != dlg) d->ShowWindow(SW_HIDE);

			break;
		}
	}

	CDialog::OnLButtonDblClk(nFlags, point);
}

BOOL CGridDialog::PreTranslateMessage(MSG* pMsg)
{
	if (pMsg->wParam == VK_RETURN || pMsg->wParam == VK_ESCAPE) {
		return TRUE;// 屏蔽Enter和ESC关闭对话
	}
	return CDialog::PreTranslateMessage(pMsg);
}