将窗口截图存为Bmp的C++代码

 

备份一份代码，用于窗口截图~！

使用的时候只需要调用Catchfile函数就可以了。

当然，代码中的WIDTH、MARGIN、 HEIGHT等还得按需要进行设定。

 

Cpp代码

1. void CMyDialog::Catchfile()   
2. {  
3.     HBITMAP hBitmap = GetSrcBit();  
4.   
5.     if (hBitmap != NULL)   
6.     {  
7.         // szFilters is a text string that includes two file name filter:  
8.         // "*.bmp" for "bmp Files" and "*.*" for "All Files."  
9.         char szFilters[]=  
10.             "bmp Files (*.bmp)|*.bmp|All Files (*.*)|*.*||";  
11.   
12.         // Create a Save dialog; the default file name extension is ".bmp".  
13.         CFileDialog fileDlg (FALSE, "bmp", "*.bmp",  
14.             OFN_FILEMUSTEXIST| OFN_HIDEREADONLY, szFilters, this);  
15.   
16.         // Display the file dialog. When user clicks OK, fileDlg.DoModal()   
17.         // returns IDOK.  
18.         if( fileDlg.DoModal() == IDOK )  
19.         {  
20.             CString pathName = fileDlg.GetPathName();  
21.   
22.             // Implement opening and reading file in here.  
23.             //...  
24.             //Change the window's title to the opened file's title.  
25.             //CString fileName = fileDlg.GetFileTitle ();  
26.   
27.             //SetWindowText(fileName);  
28.             SaveBitmapToFile(hBitmap , pathName.GetBuffer()/*LPSTR lpFileName*/);  
29.         }  
30.     }  
31. }  
32.   
33.   
34. void CMyDialog::GetSrcBit(/*DWORD BitWidth, DWORD BitHeight*/void)  
35. {  
36.     HBITMAP hBitmap,hOldBitmap;  
37.     CPaintDC dc(this);  
38.     HDC hmyMemDC = CreateCompatibleDC(dc.m_hDC);  
39.     hBitmap = CreateCompatibleBitmap(dc.m_hDC, WIDTH-MARGIN, HEIGHT);  
40.     hOldBitmap = (HBITMAP)SelectObject(hmyMemDC, hBitmap);  
41.   
42.     BitBlt(hmyMemDC, 0, 0, WIDTH, HEIGHT,dc.m_hDC, 0, 0, SRCCOPY);  
43.   
44.   
45.     hBitmap = (HBITMAP)SelectObject(hmyMemDC, hOldBitmap);  
46.   
47.     return (hBitmap);  
48. }  
49.   
50. // 我们也可以把屏幕内容以位图格式存到磁盘文件上.  
51. // lpFileName 为位图文件脕E       
52. int CMyDialog::SaveBitmapToFile(HBITMAP hBitmap , LPSTR lpFileName)   
53. {  
54.       
55.     HDC hDC;      //设备描述句柄        
56.     int iBits;    //当前显示分辨率下每个像素所占字节数    
57.       
58.     WORD wBitCount;   //位图中每个像素所占字节数  
59.   
60.     //定义调色板大小， 位图中像素字节大小 ，  
61.     //位图文件大小  
62.     DWORD  dwPaletteSize=0,  
63.            dwBmBitsSize,  
64.            dwDIBSize, dwWritten;  
65.       
66.     BITMAP   Bitmap;  //位图属性结构        
67.     BITMAPFILEHEADER    bmfHdr;  //位图文件头结构  
68.   
69.     //位图信息头结构   
70.     BITMAPINFOHEADER    bi;  
71.   
72.     //指向位图信息头结构  
73.     LPBITMAPINFOHEADER lpbi;  
74.   
75.     //定义文件，分配内存句柄，调色板句眮E  
76.     HANDLE   fh, hDib;  
77.     HPALETTE hPal,hOldPal=NULL;  
78.   
79.     //计算位图文件每个像素所占字节数  
80.     hDC = CreateDC("DISPLAY",NULL,NULL,NULL);  
81.     iBits = GetDeviceCaps(hDC, BITSPIXEL) * GetDeviceCaps(hDC, PLANES);  
82.     DeleteDC(hDC);  
83.   
84.     if (iBits <= 1)  
85.         wBitCount = 1;  
86.     else if (iBits <= 4)  
87.         wBitCount = 4;  
88.     else if (iBits <= 8)  
89.         wBitCount = 8;  
90.     else if (iBits <= 24)  
91.         wBitCount = 24;  
92.     else wBitCount = 24;   
93.   
94.     //计算调色板大小  
95.     if (wBitCount <= 8)  
96.         dwPaletteSize = (1 << wBitCount) *sizeof(RGBQUAD);  
97.   
98.     //设置位图信息头结构  
99.     GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&Bitmap);  
100.     bi.biSize            = sizeof(BITMAPINFOHEADER);  
101.     bi.biWidth           = Bitmap.bmWidth;  
102.     bi.biHeight          = Bitmap.bmHeight;  
103.     bi.biPlanes          = 1;  
104.     bi.biBitCount         = wBitCount;  
105.     bi.biCompression      = BI_RGB;  
106.     bi.biSizeImage        = 0;  
107.     bi.biXPelsPerMeter     = 0;  
108.     bi.biYPelsPerMeter     = 0;  
109.     bi.biClrUsed         = 0;  
110.     bi.biClrImportant      = 0;  
111.   
112.     dwBmBitsSize = ((Bitmap.bmWidth *wBitCount+31)/32)* 4*Bitmap.bmHeight ;  
113.     //为位图内容分配内磥E  
114.     hDib  = GlobalAlloc(GHND,dwBmBitsSize+dwPaletteSize+sizeof(BITMAPINFOHEADER));  
115.     lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);  
116.     *lpbi = bi;  
117.   
118.     hPal = (HPALETTE)GetStockObject(DEFAULT_PALETTE);  
119.     if (hPal)  
120.     {  
121.         hDC  = ::GetDC(NULL);  
122.         hOldPal = SelectPalette(hDC, hPal, FALSE);  
123.         RealizePalette(hDC);  
124.     }  
125.   
126.     // 获取该调色板下新的像素值  
127.     GetDIBits(hDC, hBitmap, 0, (UINT)Bitmap.bmHeight,(LPSTR)lpbi + sizeof(BITMAPINFOHEADER)+dwPaletteSize,(LPBITMAPINFO)lpbi, DIB_RGB_COLORS);  
128.   
129.     //恢复调色皝E    
130.     if (hOldPal)  
131.     {  
132.         SelectPalette(hDC, hOldPal, TRUE);  
133.         RealizePalette(hDC);  
134.         ::ReleaseDC(NULL, hDC);  
135.     }  
136.   
137.     //创建位图文件      
138.     fh = CreateFile(lpFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,  
139.         FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);  
140.   
141.     if (fh == INVALID_HANDLE_VALUE)  
142.         return FALSE;  
143.   
144.     //设置位图文件头  
145.     bmfHdr.bfType = 0x4D42;  // "BM"  
146.     dwDIBSize    = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)  
147.         + dwPaletteSize + dwBmBitsSize;    
148.     bmfHdr.bfSize = dwDIBSize;  
149.     bmfHdr.bfReserved1 = 0;  
150.     bmfHdr.bfReserved2 = 0;  
151.     bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER)   
152.         + (DWORD)sizeof(BITMAPINFOHEADER)+ dwPaletteSize;  
153.                                                       
154.     WriteFile(fh, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);        
155.                                                                                                                                       
156.     WriteFile(fh, (LPSTR)lpbi, dwDIBSize, &dwWritten, NULL);                          
157.                                                                                       
158.     //清除                                                                              
159.     GlobalUnlock(hDib);                                                               
160.     GlobalFree(hDib);                                                                 
161.     CloseHandle(fh);                                                                  
162.                                                                                       
163.     return TRUE;                                                                      
164. }