模仿腾讯QQ截图功能:
// ScreenCapture.cpp : 定义应用程序的入口点。
//#include "stdafx.h"
#include "ScreenCapture.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);HDC g_srcMemDc;
HDC g_grayMemDc;
int screenW;
int screenH;RECT rect = {0}; //画图的矩形区域
bool isDrawing = false;
bool isSelect = false;void GetScreenCapture();
void CovertToGrayBitmap(HBITMAP hSourceBmp,HDC sourceDc);
void WriteDatatoClipBoard();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_SCREENCAPTURE, szWindowClass, MAX_LOADSTRING);MyRegisterClass(hInstance);// 执行应用程序初始化:if (!InitInstance (hInstance, nCmdShow)){return FALSE;}hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_SCREENCAPTURE));// 主消息循环: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 | CS_DBLCLKS;wcex.lpfnWndProc = WndProc;wcex.cbClsExtra = 0;wcex.cbWndExtra = 0;wcex.hInstance = hInstance;wcex.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_SCREENCAPTURE));wcex.hCursor = LoadCursor(NULL, IDC_ARROW);wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);wcex.lpszMenuName = 0;//MAKEINTRESOURCE(IDC_SCREENCAPTURE);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_POPUP,CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);if (!hWnd){return FALSE;}ShowWindow(hWnd, SW_MAXIMIZE);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;LOGBRUSH brush;brush.lbStyle=BS_NULL;HBRUSH hBrush=CreateBrushIndirect(&brush);LOGPEN pen;POINT penWidth;penWidth.x=2;penWidth.y=2;pen.lopnColor=0x0000FFFF;pen.lopnStyle=PS_SOLID;pen.lopnWidth=penWidth;HPEN hPen=CreatePenIndirect(&pen);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_EXIT:DestroyWindow(hWnd);break;default:return DefWindowProc(hWnd, message, wParam, lParam);}break;case WM_CREATE:GetScreenCapture();break;case WM_PAINT:{hdc = BeginPaint(hWnd, &ps);HDC memDc = CreateCompatibleDC(hdc);HBITMAP bmp = CreateCompatibleBitmap(hdc, screenW, screenH);SelectObject(memDc, bmp);BitBlt(memDc, 0, 0, screenW,screenH, g_grayMemDc, 0, 0, SRCCOPY);SelectObject(memDc, hBrush);SelectObject(memDc, hPen);if (isDrawing || isSelect){BitBlt(memDc, rect.left, rect.top, rect.right - rect.left, rect.bottom - rect.top, g_srcMemDc, rect.left, rect.top, SRCCOPY);Rectangle(memDc, rect.left, rect.top, rect.right, rect.bottom);}BitBlt(hdc, 0, 0, screenW, screenH, memDc, 0, 0, SRCCOPY);DeleteObject(bmp);DeleteObject(memDc);EndPaint(hWnd, &ps);}break;case WM_LBUTTONDOWN:{if (!isSelect){POINT pt;GetCursorPos(&pt);rect.left = pt.x;rect.top = pt.y;rect.right = pt.x;rect.bottom = pt.y;isDrawing = true;InvalidateRgn(hWnd, 0, false);}}break;case WM_LBUTTONUP:{if (isDrawing && !isSelect){isDrawing = false;POINT pt;GetCursorPos(&pt);rect.right = pt.x;rect.bottom = pt.y;isSelect = true;InvalidateRgn(hWnd, 0, false);}}break;case WM_MOUSEMOVE:{if (isDrawing&& !isSelect){POINT pt;GetCursorPos(&pt);rect.right = pt.x;rect.bottom = pt.y;InvalidateRgn(hWnd, 0, false);}}break;case WM_LBUTTONDBLCLK:{if (isSelect){WriteDatatoClipBoard();InvalidateRgn(hWnd, 0, false);ShowWindow(hWnd, SW_MINIMIZE);}isSelect = false;}break;case WM_DESTROY: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;
}void GetScreenCapture()
{HDC disDc = ::CreateDC(L"DISPLAY", 0, 0, 0); //创建屏幕相关的DCscreenW = GetDeviceCaps(disDc, HORZRES);//水平分辨率screenH = GetDeviceCaps(disDc, VERTRES);//垂直分辨率g_srcMemDc = CreateCompatibleDC(disDc); //创建于屏幕兼容的DC(内存DC)HBITMAP hbMap = CreateCompatibleBitmap(disDc, screenW, screenH); //模拟一张画布,其中是没有数据的SelectObject(g_srcMemDc, hbMap); //将画图选入内存DC,其中还是没有数据的BitBlt(g_srcMemDc, 0, 0, screenW, screenH, disDc, 0, 0, SRCCOPY); //将屏幕的dc中的画图,拷贝至内存DC中//获取屏幕的灰度图片g_grayMemDc = CreateCompatibleDC(disDc);HBITMAP grayMap = CreateCompatibleBitmap(disDc, screenW, screenH); //模拟一张画布,其中是没有数据的SelectObject(g_grayMemDc, grayMap); //将画图选入内存DC,其中还是没有数据的BitBlt(g_grayMemDc, 0, 0, screenW, screenH, disDc, 0, 0, SRCCOPY); //将屏幕的dc中的画图,拷贝至内存DC中CovertToGrayBitmap(grayMap, g_grayMemDc); //将彩色图片转换灰度图片DeleteObject(hbMap);DeleteObject(grayMap);DeleteObject(disDc);
}void CovertToGrayBitmap(HBITMAP hSourceBmp,HDC sourceDc)
{HBITMAP retBmp=hSourceBmp;BITMAPINFO bmpInfo;ZeroMemory(&bmpInfo,sizeof(BITMAPINFO));bmpInfo.bmiHeader.biSize=sizeof(BITMAPINFOHEADER);GetDIBits(sourceDc,retBmp,0,0,NULL,&bmpInfo,DIB_RGB_COLORS);BYTE* bits=new BYTE[bmpInfo.bmiHeader.biSizeImage];GetBitmapBits(retBmp,bmpInfo.bmiHeader.biSizeImage,bits);int bytePerPixel=4;//默认32位if(bmpInfo.bmiHeader.biBitCount==24){bytePerPixel=3;}for(DWORD i=0;i
{HDC hMemDc,hScrDc;HBITMAP hBmp,hOldBmp;int width,height;width=rect.right-rect.left;height=rect.bottom-rect.top;hScrDc=CreateDC(L"DISPLAY",NULL,NULL,NULL);hMemDc=CreateCompatibleDC(hScrDc);hBmp=CreateCompatibleBitmap(hScrDc,width,height);hOldBmp=(HBITMAP)SelectObject(hMemDc,hBmp);BitBlt(hMemDc,0,0,width,height,g_srcMemDc,rect.left,rect.top,SRCCOPY);hBmp=(HBITMAP)SelectObject(hMemDc,hOldBmp);DeleteDC(hMemDc);DeleteDC(hScrDc);//复制到剪贴板if(OpenClipboard(0)){EmptyClipboard();SetClipboardData(CF_BITMAP,hBmp);CloseClipboard();}DeleteObject(hBmp);DeleteObject(hMemDc);DeleteObject(hScrDc);}