Win32项目实现基本绘图

// PEN.cpp : 定义应用程序的入口点。
//


#include "stdafx.h"
#include "PEN.h"
#define MAX_LOADSTRING 100


// 全局变量: 
HINSTANCE hInst; // 当前实例
TCHAR szTitle[MAX_LOADSTRING]; // 标题栏文本
TCHAR szWindowClass[MAX_LOADSTRING]; // 主窗口类名


// 此代码模块中包含的函数的前向声明: 
ATOM MyRegisterClass(HINSTANCE hInstance);
BOOL InitInstance(HINSTANCE, int);
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK About(HWND, UINT, WPARAM, LPARAM);


int APIENTRY _tWinMain(_In_ HINSTANCE hInstance,
                     _In_opt_ HINSTANCE hPrevInstance,
                     _In_ LPTSTR    lpCmdLine,
                     _In_ int       nCmdShow)
{
UNREFERENCED_PARAMETER(hPrevInstance);
UNREFERENCED_PARAMETER(lpCmdLine);


  // TODO:  在此放置代码。
MSG msg;
HACCEL hAccelTable;


// 初始化全局字符串
LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
LoadString(hInstance, IDC_PEN, szWindowClass, MAX_LOADSTRING);
MyRegisterClass(hInstance);


// 执行应用程序初始化: 
if (!InitInstance (hInstance, nCmdShow))
{
return FALSE;
}


hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_PEN));


// 主消息循环: 
while (GetMessage(&msg, NULL, 0, 0))
{
if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}


return (int) msg.wParam;
}






//
//  函数:  MyRegisterClass()
//
//  目的:  注册窗口类。
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
WNDCLASSEX wcex;


wcex.cbSize = sizeof(WNDCLASSEX);


wcex.style = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc = WndProc;
wcex.cbClsExtra = 0;
wcex.cbWndExtra = 0;
wcex.hInstance = hInstance;
wcex.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_PEN));
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
wcex.lpszMenuName = MAKEINTRESOURCE(IDC_PEN);
wcex.lpszClassName = szWindowClass;
wcex.hIconSm = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));


return RegisterClassEx(&wcex);
}


//
//   函数:  InitInstance(HINSTANCE, int)
//
//   目的:  保存实例句柄并创建主窗口
//
//   注释: 
//
//        在此函数中，我们在全局变量中保存实例句柄并
//        创建和显示主程序窗口。
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   HWND hWnd;


   hInst = hInstance; // 将实例句柄存储在全局变量中


   hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);


   if (!hWnd)
   {
      return FALSE;
   }


   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);


   return TRUE;
}


//
//  函数:  WndProc(HWND, UINT, WPARAM, LPARAM)
//
//  目的:    处理主窗口的消息。
//
//  WM_COMMAND - 处理应用程序菜单
//  WM_PAINT - 绘制主窗口
//  WM_DESTROY - 发送退出消息并返回
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int wmId, wmEvent;
PAINTSTRUCT ps;
HDC hdc;
//TCHAR szHello[MAX_LOADSTRING];
//LoadString(hInst, IDS_HELLO, szHello, MAX_LOADSTRING);
static POINT pStart;
static POINT pEnd;
static POINT pPrev;
static BOOL bMouseMove = false;
static HPEN hpen;
static HPEN holdpen;
static HBRUSH hBr;
static HBRUSH hOldBr;  //画一个图形，其内部将变为灰色
static COLORREF color;
static HMENU hMenu;
static HMENU hMenu2;
static HMENU hMenu3;
static INT iWidth;
static INT iShape;//1:Line;2:Circle;3:Rectangle;4:Pen;
static BOOL bFill = TRUE;
switch (message)
{
case WM_COMMAND:
wmId = LOWORD(wParam);
wmEvent = HIWORD(wParam);
// 分析菜单选择: 
switch (wmId)
{
case IDM_ABOUT:
DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
break;
case IDM_DRAW_LINE:
iShape = 1;
//CheckMenuItem()
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_DRAW_ELLIPSE:
iShape = 2;
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_DRAW_RECT:
iShape = 3;
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_DRAW_CURVE:
iShape = 4;
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_CHECKED | MF_BYCOMMAND);
break;
case IDM_COLORRED:
color = RGB(255, 0, 0);
DeleteObject(hpen);
DeleteObject(hBr);
iWidth = 2;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
CheckMenuItem(hMenu, IDM_COLORRED,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu,IDM_COLORBLACK,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_COLORGREEN:
color = RGB(0, 255, 0);
DeleteObject(hpen);
DeleteObject(hBr);
iWidth = 2;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
CheckMenuItem(hMenu, IDM_COLORRED,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLACK,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_COLORBLUE:
color = RGB(0, 0, 255);
DeleteObject(hpen);
DeleteObject(hBr);
iWidth = 2;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
CheckMenuItem(hMenu, IDM_COLORRED,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLACK,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_COLORBLACK:
color = RGB(0, 0, 0);
DeleteObject(hpen);
DeleteObject(hBr);
iWidth = 2;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
CheckMenuItem(hMenu, IDM_COLORRED,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLACK,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_UNCHECKED | MF_BYCOMMAND);
break;
case IDM_COLORWHITE:
color = RGB(255, 255, 255);
DeleteObject(hpen);
DeleteObject(hBr);
iWidth = 5;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
iShape = 4;
CheckMenuItem(hMenu, IDM_COLORRED,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLACK,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_CHECKED | MF_BYCOMMAND);
break;
case IDM_DRAW_FILL:
bFill = !bFill;
//CheckMenuItem(hMenu, IDM_DRAW_FILL,
// MF_BYCOMMAND | (bFill ? MF_CHECKED : MF_UNCHECKED));
if (bFill)
{
CheckMenuItem(hMenu3, IDM_DRAW_FILL,
MF_CHECKED | MF_BYCOMMAND);
}
else
{
CheckMenuItem(hMenu3, IDM_DRAW_FILL,
MF_UNCHECKED | MF_BYCOMMAND);
}
break;
case IDM_EXIT:
DestroyWindow(hWnd);
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
break;
//case WM_SIZE:


// break;
case WM_CREATE:
color = RGB(255, 0, 0);
iWidth = 2;
hpen = CreatePen(PS_SOLID, iWidth, color);
hBr = CreateSolidBrush(color);
hMenu = GetMenu(hWnd);
hMenu2 = GetMenu(hWnd);
hMenu3 = GetMenu(hWnd);
CheckMenuItem(hMenu, IDM_COLORRED,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORGREEN,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLUE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORBLACK,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu, IDM_COLORWHITE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_LINE,
MF_CHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_ELLIPSE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_RECT,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu2, IDM_DRAW_CURVE,
MF_UNCHECKED | MF_BYCOMMAND);
CheckMenuItem(hMenu3, IDM_DRAW_FILL,
MF_CHECKED | MF_BYCOMMAND);
iShape = 1;
break;
case WM_PAINT:
hdc = BeginPaint(hWnd, &ps);
// TODO:  在此添加任意绘图代码...
//GetClientRect(hWnd, &rect);
/* Ellipse(hdc, 0, 0,rect.right,rect.bottom);*/
// TODO: Add any drawing code here...
EndPaint(hWnd, &ps);
break;
case WM_LBUTTONDOWN:
hdc = GetDC(hWnd);
pStart.x = GET_X_LPARAM(lParam);
pStart.y = GET_Y_LPARAM(lParam);
pEnd = pStart;
pPrev = pStart;
SetROP2(hdc, R2_NOTXORPEN);  //R2_NOTXORPEN的意思是由画笔的颜色和屏幕的颜色异或，然后取反，由于屏幕的颜色是白色，而画笔的颜色是其他色，由于白色的颜色值是1，所以异或之后是白色，然后取反是画笔的颜色；
SelectObject(hdc,hpen);
if (bFill)
{
SelectObject(hdc, hBr);
}
else
{
SelectObject(hdc, GetStockObject(NULL_BRUSH));
}
switch (iShape)
{
case 1:
MoveToEx(hdc, pStart.x, pStart.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
break;
case 2:
Ellipse(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 3:
Rectangle(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 4:
break;
}
ReleaseDC(hWnd, hdc);
bMouseMove = true;
SetCapture(hWnd);
break;
case WM_LBUTTONUP:
hdc = GetDC(hWnd);
SetROP2(hdc, R2_NOTXORPEN);//取非
SelectObject(hdc, hpen);
if (bFill)
{
SelectObject(hdc, hBr);
}
else
{
SelectObject(hdc, GetStockObject(NULL_BRUSH));
}
switch (iShape)
{
case 1:
MoveToEx(hdc, pStart.x, pStart.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
break;
case 2:
Ellipse(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 3:
Rectangle(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
}
pEnd.x = GET_X_LPARAM(lParam);
pEnd.y = GET_Y_LPARAM(lParam);
SetROP2(hdc, R2_COPYPEN);
if (bFill)
{
SelectObject(hdc, hBr);
}
else
{
SelectObject(hdc, GetStockObject(NULL_BRUSH));
}
switch (iShape)
{
case 1:
MoveToEx(hdc, pStart.x, pStart.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
break;
case 2:
Ellipse(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 3:
Rectangle(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 4:
SetROP2(hdc, R2_COPYPEN);
MoveToEx(hdc, pPrev.x, pPrev.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
pPrev = pEnd;
break;
}
ReleaseDC(hWnd, hdc);
bMouseMove = false;
ReleaseCapture();
break;
case WM_MOUSEMOVE:
if (bMouseMove)
{
hdc = GetDC(hWnd);
SetROP2(hdc, R2_NOTXORPEN); //由于现在画笔的颜色还是其他色，不是白色，所以最后得到的还是画笔的颜色；
SelectObject(hdc, hpen);
if (bFill)
{
SelectObject(hdc, hBr);
}
else
{
SelectObject(hdc, GetStockObject(NULL_BRUSH));
}
switch (iShape)
{
case 1:
MoveToEx(hdc, pStart.x, pStart.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
break;
case 2:
Ellipse(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 3:
Rectangle(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 4:
break;
}
pEnd.x = GET_X_LPARAM(lParam);
pEnd.y = GET_Y_LPARAM(lParam);
switch (iShape)
{
case 1:
MoveToEx(hdc, pStart.x, pStart.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
break;
case 2:
Ellipse(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 3:
Rectangle(hdc, pStart.x, pStart.y,
pEnd.x, pEnd.y);
break;
case 4:
SetROP2(hdc, R2_COPYPEN);
MoveToEx(hdc, pPrev.x, pPrev.y, NULL);
LineTo(hdc, pEnd.x, pEnd.y);
pPrev = pEnd;
}
ReleaseDC(hWnd, hdc);
}
break;
case WM_DESTROY:
DeleteObject(hpen);
DeleteObject(hBr);
PostQuitMessage(0);
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
return 0;
}


// “关于”框的消息处理程序。
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
UNREFERENCED_PARAMETER(lParam);
switch (message)
{
case WM_INITDIALOG:
return (INT_PTR)TRUE;


case WM_COMMAND:
if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
{
EndDialog(hDlg, LOWORD(wParam));
return (INT_PTR)TRUE;
}
break;
}
return (INT_PTR)FALSE;
}