简单封装的CMyWnd(封装sdk的窗口消息循环)

//CMyWnd.h

#pragma once
#include <Windows.h>
#include <vector>
using namespace std;

class CMyWnd
{
public:
	CMyWnd();
	~CMyWnd(void);

public:
	typedef LRESULT (CMyWnd::*PFNNORMALPROC)(HWND, UINT, WPARAM, LPARAM);
	typedef struct tagMsgInfo
	{
		UINT		m_uMsg;
		PFNNORMALPROC	m_pfunCall;
	}stMsgInfo, *LPMsgInfo;

	typedef LRESULT (CMyWnd::*PFNCOMMPROC)();

	typedef struct tagCommMsgInfo
	{
		UINT		m_uMsg;
		PFNCOMMPROC	m_pfunCall;
	}stCommMsgInfo, *LPCommMsgInfo;


public:
	//注册窗口类
	virtual BOOL RegisterClass(LPCTSTR lpszClassName, UINT style, int cbClsExtra, int cbWndExtra, HINSTANCE hInstance, 
		HICON hIcon, HCURSOR hCursor, HBRUSH hbrBackground, LPCTSTR lpszMenuName = NULL, HICON hIconSm = NULL);
	
	//创建窗口
	virtual BOOL Create(DWORD dwExStyle, LPCTSTR lpszWindowName, DWORD dwStyle, int x, int y, 
		int nWidth, int nHeight, HWND hWndParent, HMENU nIDorHMenu, HINSTANCE hInst, LPVOID lpParam = NULL);

	//Command消息循环
	virtual LRESULT CmdProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
	
	//消息循环
	static LRESULT WINAPI WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);

public:
	BOOL ShowWindow(int nCmdShow);

public:
	BOOL InitMsgFunList();
	BOOL AddCommandMsg(UINT uMsg, PFNNORMALPROC pfunCall);
	BOOL AddNoramlMsg(UINT uMsg, PFNNORMALPROC pfunCall);
	BOOL AddCommonMsg(UINT uMsg, PFNCOMMPROC pfunCall);

public:
	//窗口响应消息
	virtual LRESULT OnCreate();
	virtual LRESULT OnPaint();
	virtual LRESULT OnDestroy();

public:
	//辅助调用的函数
	virtual LRESULT OnDraw(HDC hdc, PAINTSTRUCT& ps);

public:
	HWND			m_hWnd;//保存窗口句柄

private:
	WNDCLASSEX		m_wndClass;//保存注册窗口类
	TCHAR			m_szClassName[MAX_PATH];//保存窗口类的名称

	//消息列表
	vector<stMsgInfo>		m_vtMsgInfo;//消息列表
	vector<stMsgInfo>		m_vtCmdInfo;//Command消息列表
	vector<stCommMsgInfo>	m_vtCommInfo;//常见的消息列表
};

 

 

//CMyWnd.cpp

#include "mywnd.h"
#include <tchar.h>
#include <stdio.h>

#pragma warning(disable: 4311)
#pragma warning(disable: 4312)
#pragma warning(disable: 4018)


CMyWnd::CMyWnd()
{
	m_hWnd = NULL;
	m_vtMsgInfo.clear();
	m_vtMsgInfo.clear();

	//添加常见消息映射
	InitMsgFunList();
}

CMyWnd::~CMyWnd(void)
{
}

BOOL CMyWnd::InitMsgFunList()
{
	AddCommonMsg(WM_CREATE, OnCreate);
	AddCommonMsg(WM_PAINT, OnPaint);
	AddCommonMsg(WM_DESTROY, OnDestroy);

	return TRUE;
}

BOOL CMyWnd::RegisterClass(LPCTSTR lpszClassName, UINT style, int cbClsExtra, int cbWndExtra, HINSTANCE hInstance, 
						   HICON hIcon, HCURSOR hCursor, HBRUSH hbrBackground, LPCTSTR lpszMenuName, HICON hIconSm)
{
	//初始化WNDCLASSEX
	::memset(&m_wndClass, 0, sizeof(m_wndClass));
	m_wndClass.cbSize = sizeof(m_wndClass);
	m_wndClass.style = style;
	m_wndClass.cbClsExtra = sizeof(this);
	m_wndClass.cbWndExtra = sizeof(this);
	m_wndClass.hInstance = hInstance;
	m_wndClass.hIcon = hIcon;
	m_wndClass.hCursor = hCursor;
	m_wndClass.hbrBackground = hbrBackground;

	if (NULL != lpszMenuName)
	{
		_tcsncpy(m_szClassName, lpszClassName, MAX_PATH);
	}
	else
	{
		_stprintf(m_szClassName, "mywnd-%d", rand());
	}

	m_wndClass.lpszClassName = m_szClassName;
	m_wndClass.lpszMenuName = lpszMenuName;
	m_wndClass.hIconSm = hIconSm;
	
	m_wndClass.lpfnWndProc = CMyWnd::WndProc;

	if (!::RegisterClassEx(&m_wndClass))
	{
		return FALSE;
	}

	return TRUE;
}

BOOL CMyWnd::Create(DWORD dwExStyle, LPCTSTR lpszWindowName, DWORD dwStyle,int x, int y, 
					int nWidth, int nHeight, HWND hWndParent, HMENU nIDorHMenu, HINSTANCE hInst, LPVOID lpParam)
{
	m_hWnd = ::CreateWindowEx(dwExStyle, m_szClassName, lpszWindowName, dwStyle, x, y, nWidth, nHeight,
		hWndParent, nIDorHMenu, hInst, lpParam);

	if (NULL == m_hWnd)
	{
		return FALSE;
	}

	//创建完成之后调用OnCreate
    OnCreate();

	SetWindowLong(m_hWnd, GWL_USERDATA, (LONG)this);

	::ShowWindow(m_hWnd, SW_NORMAL);
	::UpdateWindow(m_hWnd);

	MSG msg;
	while (GetMessage(&msg, NULL, 0, 0))
	{
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}

	return TRUE;
}

BOOL CMyWnd::ShowWindow(int nCmdShow)
{
	if (!::IsWindow(m_hWnd))
	{
		return FALSE;
	}

	return ::ShowWindow(m_hWnd, nCmdShow);
}

LRESULT CMyWnd::WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	int		i		= 0;
	LONG	lVal	= 0;
	CMyWnd* pThis	= (CMyWnd*)GetWindowLong(hwnd, GWL_USERDATA);

	if (NULL == pThis)
	{
		return ::DefWindowProc(hwnd, uMsg, wParam, lParam);
	}

	//执行Command消息
	if (uMsg == WM_COMMAND)
	{
		return pThis->CmdProc(hwnd, uMsg, wParam, lParam);
	}

	//执行常见的消息
	for (i = 0; i < pThis->m_vtCommInfo.size(); i++)
	{
		if (uMsg == pThis->m_vtCommInfo[i].m_uMsg)
		{
			return (pThis->*(pThis->m_vtCommInfo[i].m_pfunCall))();
		}
	}

	//执行普通消息
	for (i = 0; i < pThis->m_vtMsgInfo.size(); i++)
	{
		if (uMsg == pThis->m_vtMsgInfo[i].m_uMsg)
		{
			return (pThis->*(pThis->m_vtMsgInfo[i].m_pfunCall))(hwnd, uMsg, wParam, lParam);
		}
	}

	return ::DefWindowProc(hwnd, uMsg, wParam, lParam);
}

LRESULT CMyWnd::CmdProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	int i = 0;

	//执行普通消息
	for (i = 0; i < m_vtCmdInfo.size(); i++)
	{
		if (uMsg == m_vtCmdInfo[i].m_uMsg)
		{
			return ( this->*(m_vtCmdInfo[i].m_pfunCall) )(hwnd, uMsg, wParam, lParam);
		}
	}

	return ::DefWindowProc(hwnd, uMsg, wParam, lParam);
}

BOOL CMyWnd::AddCommandMsg(UINT uMsg, PFNNORMALPROC pfunCall)
{
	stMsgInfo msgInfo = {uMsg, pfunCall};

	m_vtCmdInfo.push_back(msgInfo);

	return TRUE;
}

BOOL CMyWnd::AddNoramlMsg(UINT uMsg, PFNNORMALPROC pfunCall)
{
	stMsgInfo msgInfo = {uMsg, pfunCall};
	m_vtMsgInfo.push_back(msgInfo);

	return TRUE;
}

BOOL CMyWnd::AddCommonMsg(UINT uMsg, PFNCOMMPROC pfunCall)
{
	stCommMsgInfo msgInfo = {uMsg, pfunCall};
	m_vtCommInfo.push_back(msgInfo);

	return TRUE;
}

LRESULT CMyWnd::OnCreate()
{
	return DefWindowProc(m_hWnd, WM_CREATE, 0, 0);
}

LRESULT CMyWnd::OnPaint()
{
	HDC			hdc	= NULL;
	PAINTSTRUCT	ps	= {0};

	hdc = BeginPaint(m_hWnd, &ps);
	OnDraw(hdc, ps);
	EndPaint(m_hWnd, &ps);
	
	return S_OK;
}

LRESULT CMyWnd::OnDestroy()
{
	::PostQuitMessage(0);

	return S_OK;
}

LRESULT CMyWnd::OnDraw(HDC hdc, PAINTSTRUCT& ps)
{
	return S_OK;
}