BMP截屏

1. void CSaveBmpDlg::OnButton1()    
2. {   
3.     // TODO: Add your control notification handler code here   
4.     HWND hWnd = ::GetDesktopWindow();//获得屏幕的HWND.   
5.     HDC hScreenDC = ::GetDC(hWnd);   //获得屏幕的HDC.   
6.     int width = ::GetSystemMetrics(SM_CXSCREEN);   
7.     int height = ::GetSystemMetrics(SM_CYSCREEN);   
8.     HBITMAP hBitmap = GetSrcBit(hScreenDC, width, height);   
9.     ASSERT(hBitmap);   
10.     SaveBmp(hBitmap, L"d://screen.bmp");   
11.        
12. }   
13. HBITMAP CSaveBmpDlg::GetSrcBit(HDC hDC,DWORD BitWidth, DWORD BitHeight)   
14. {   
15.     HDC hBufDC;   
16.     HBITMAP hBitmap, hBitTemp;   
17.     //创建设备上下文(HDC)   
18.     hBufDC = CreateCompatibleDC(hDC);   
19.     //创建HBITMAP   
20.     hBitmap = CreateCompatibleBitmap(hDC, BitWidth, BitHeight);   
21.     hBitTemp = (HBITMAP) SelectObject(hBufDC, hBitmap);   
22.     //得到位图缓冲区   
23.     StretchBlt(hBufDC, 0, 0, BitWidth, BitHeight,   
24.         hDC, 0, 0, BitWidth, BitHeight, SRCCOPY);   
25.     //得到最终的位图信息   
26.     hBitmap = (HBITMAP) SelectObject(hBufDC, hBitTemp);   
27.     //释放内存   
28.     DeleteObject(hBitTemp);   
29.     ::DeleteDC(hBufDC);   
30.     return hBitmap;   
31. }   
32. BOOL CSaveBmpDlg::SaveBmp(HBITMAP hBitmap, CString FileName)   
33. {   
34.     //设备描述表   
35.     HDC hDC;   
36.     //当前分辨率下每象素所占字节数   
37.     int iBits;   
38.     //位图中每象素所占字节数   
39.     WORD wBitCount;   
40.     //定义调色板大小， 位图中像素字节大小 ，位图文件大小 ， 写入文件字节数    
41.     DWORD dwPaletteSize=0, dwBmBitsSize=0, dwDIBSize=0, dwWritten=0;    
42.     //位图属性结构    
43.     BITMAP Bitmap;     
44.     //位图文件头结构   
45.     BITMAPFILEHEADER bmfHdr;     
46.     //位图信息头结构    
47.     BITMAPINFOHEADER bi;     
48.     //指向位图信息头结构     
49.     LPBITMAPINFOHEADER lpbi;     
50.     //定义文件，分配内存句柄，调色板句柄    
51.     HANDLE fh, hDib, hPal,hOldPal=NULL;    
52.     //计算位图文件每个像素所占字节数    
53.     hDC = CreateDC("DISPLAY", NULL, NULL, NULL);   
54.     iBits = GetDeviceCaps(hDC, BITSPIXEL) * GetDeviceCaps(hDC, PLANES);    
55.     DeleteDC(hDC);    
56.     if (iBits <= 1) wBitCount = 1;    
57.     else if (iBits <= 4)  wBitCount = 4;    
58.     else if (iBits <= 8)  wBitCount = 8;    
59.     else   wBitCount = 24;    
60.     GetObject(hBitmap, sizeof(Bitmap), (LPSTR)&Bitmap);   
61.     bi.biSize = sizeof(BITMAPINFOHEADER);   
62.     bi.biWidth = Bitmap.bmWidth;   
63.     bi.biHeight = Bitmap.bmHeight;   
64.     bi.biPlanes = 1;   
65.     bi.biBitCount = wBitCount;   
66.     bi.biCompression = BI_RGB;   
67.     bi.biSizeImage = 0;   
68.     bi.biXPelsPerMeter = 0;   
69.     bi.biYPelsPerMeter = 0;   
70.     bi.biClrImportant = 0;   
71.     bi.biClrUsed = 0;   
72.     dwBmBitsSize = ((Bitmap.bmWidth * wBitCount + 31) / 32) * 4 * Bitmap.bmHeight;   
73.     //为位图内容分配内存    
74.     hDib = GlobalAlloc(GHND,dwBmBitsSize + dwPaletteSize + sizeof(BITMAPINFOHEADER));    
75.     lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);    
76.     *lpbi = bi;    
77.        
78.     // 处理调色板     
79.     hPal = GetStockObject(DEFAULT_PALETTE);    
80.     if (hPal)    
81.     {    
82.         hDC = ::GetDC(NULL);    
83.         hOldPal = ::SelectPalette(hDC, (HPALETTE)hPal, FALSE);    
84.         RealizePalette(hDC);    
85.     }   
86.     // 获取该调色板下新的像素值    
87.     GetDIBits(hDC, hBitmap, 0, (UINT) Bitmap.bmHeight, (LPSTR)lpbi + sizeof(BITMAPINFOHEADER) +dwPaletteSize, (BITMAPINFO *)lpbi, DIB_RGB_COLORS);    
88.     //恢复调色板     
89.     if (hOldPal)    
90.     {    
91.         ::SelectPalette(hDC, (HPALETTE)hOldPal, TRUE);    
92.         RealizePalette(hDC);    
93.         ::ReleaseDC(NULL, hDC);    
94.     }    
95.     //创建位图文件     
96.     fh = CreateFile(FileName, GENERIC_WRITE,0, NULL, CREATE_ALWAYS,    
97.         FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);    
98.     if (fh == INVALID_HANDLE_VALUE)  return FALSE;    
99.     // 设置位图文件头    
100.     bmfHdr.bfType = 0x4D42; // "BM"    
101.     dwDIBSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + dwPaletteSize + dwBmBitsSize;     
102.     bmfHdr.bfSize = dwDIBSize;    
103.     bmfHdr.bfReserved1 = 0;    
104.     bmfHdr.bfReserved2 = 0;    
105.     bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + (DWORD)sizeof(BITMAPINFOHEADER) + dwPaletteSize;    
106.     // 写入位图文件头    
107.     WriteFile(fh, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);    
108.     // 写入位图文件其余内容    
109.     WriteFile(fh, (LPSTR)lpbi, dwDIBSize, &dwWritten, NULL);    
110.     //清除     
111.     GlobalUnlock(hDib);    
112.     GlobalFree(hDib);    
113.     CloseHandle(fh);    
114.        
115.     return TRUE;   
116. }