屏幕截取代码

阿杰

1. 
   
2. #include<windows.h>
   
3. HANDLE DDBtoDIB( HBITMAP bitmap, DWORD dwCompression, HPALETTE  hPal,DWORD * sizeimage) ;
   
4. BOOL CapScreen(LPTSTR FileName);//截屏函数
   
5. HANDLE DDBtoDIB( HBITMAP bitmap, DWORD dwCompression, HPALETTE  hPal,DWORD * sizeimage)
   
6. {
   
7.     BITMAP            bm;
   
8.     BITMAPINFOHEADER    bi;
   
9.      LPBITMAPINFOHEADER     lpbi;
   
10.     DWORD            dwLen;
    
11.     HANDLE            hDib;
    
12.     HANDLE            handle;
    
13.     HDC             hdc;
    
14.     //不支持BI_BITFIELDS类型
    
15.     if( dwCompression == BI_BITFIELDS )
    
16.         return NULL;
    
17.     //如果调色板为空，则用默认调色板
    
18.     if (hPal==NULL)
    
19.         hPal = (HPALETTE) GetStockObject(DEFAULT_PALETTE );
    
20.     //获取位图信息
    
21.     GetObject(bitmap,sizeof(bm),(LPSTR)&bm);
    
22.     //初始化位图信息头
    
23.     bi.biSize        = sizeof(BITMAPINFOHEADER);
    
24.     bi.biWidth        = bm.bmWidth;
    
25.     bi.biHeight         = bm.bmHeight;
    
26.     bi.biPlanes         = 1;
    
27.     bi.biBitCount        = bm.bmPlanes * bm.bmBitsPixel;
    
28.     bi.biCompression    = dwCompression;
    
29.     bi.biSizeImage        = 0;
    
30.     bi.biXPelsPerMeter    = 0;
    
31.     bi.biYPelsPerMeter    = 0;
    
32.     bi.biClrUsed        = 0;
    
33.     bi.biClrImportant    = 0;
    
34.     //计算信息头及颜色表大小
    
35.     int ncolors = (1 << bi.biBitCount); if( ncolors> 256 )
    
36.         ncolors = 0;
    
37.     dwLen  = bi.biSize + ncolors * sizeof(RGBQUAD);
    
38.     // we need a device context to get the dib from
    
39.     hdc = GetDC(NULL);
    
40.     hPal = SelectPalette(hdc,hPal,FALSE);
    
41.     RealizePalette(hdc);
    
42.     //为信息头及颜色表分配内存
    
43.     hDib = GlobalAlloc(GMEM_FIXED,dwLen);
    
44.     if (!hDib){
    
45.         SelectPalette(hdc,hPal,FALSE);
    
46.         ReleaseDC(NULL,hdc);
    
47.         return NULL;
    
48.     }
    
49.     lpbi = (LPBITMAPINFOHEADER)hDib;
    
50.     *lpbi = bi;
    
51.     //调用 GetDIBits 计算图像大小
    
52.     GetDIBits(hdc, bitmap, 0L, (DWORD)bi.biHeight,
    
53.             (LPBYTE)NULL, (LPBITMAPINFO)lpbi, (DWORD)DIB_RGB_COLORS );
    
54.     bi = *lpbi;
    
55.     //图像的每一行都对齐(32bit)边界
    
56.         if (bi.biSizeImage == 0){
    
57.         bi.biSizeImage = ((((bi.biWidth * bi.biBitCount) + 31) & ~31) / 8)
    
58.                         * bi.biHeight;
    
59.     if (dwCompression != BI_RGB)
    
60.             bi.biSizeImage = (bi.biSizeImage * 3) / 2;
    
61.     }
    
62.     //重新分配内存大小，以便放下所有数据
    
63.     dwLen += bi.biSizeImage;
    
64.     if (handle = GlobalReAlloc(hDib, dwLen, GMEM_MOVEABLE))
    
65.         hDib = handle;
    
66.     else{
    
67.         GlobalFree(hDib);
    
68.         //重选原始调色板
    
69.         SelectPalette(hdc,hPal,FALSE);
    
70.         ReleaseDC(NULL,hdc);
    
71.         return NULL;
    
72.     }
    
73.     //获取位图数据
    
74.     lpbi = (LPBITMAPINFOHEADER)hDib;
    
75.     //最终获得的DIB
    
76.     BOOL bgotbits = GetDIBits( hdc, bitmap,
    
77.                   0L,                      //扫描行起始处
    
78.                 (DWORD)bi.biHeight,      //扫描行数
    
79.                 (LPBYTE)lpbi             //位图数据地址
    
80.                 + (bi.biSize + ncolors * sizeof(RGBQUAD)),
    
81.                 (LPBITMAPINFO)lpbi,      //位图信息地址
    
82.                 (DWORD)DIB_RGB_COLORS);  //颜色板使用RGB
    
83.     if( !bgotbits )
    
84.     {
    
85.         GlobalFree(hDib);
    
86.         
    
87.         SelectPalette(hdc,hPal,FALSE);
    
88.         ReleaseDC(NULL,hdc);
    
89.                 return NULL;
    
90.     }
    
91.     SelectPalette(hdc,hPal,FALSE);
    
92.     ReleaseDC(NULL,hdc);
    
93.         *sizeimage=bi.biSizeImage;
    
94.     return hDib;
    
95. }
    
96. BOOL CapScreen(LPTSTR FileName)
    
97. {
    
98.         DWORD sizeimage;
    
99.         HDC hdc = CreateDC("DISPLAY", NULL, NULL, NULL);
    
100.         HDC CompatibleHDC = CreateCompatibleDC(hdc);
     
101.         HBITMAP BmpScreen = CreateCompatibleBitmap(hdc,GetDeviceCaps(hdc, HORZRES),GetDeviceCaps(hdc, VERTRES));
     
102.         SelectObject(CompatibleHDC, BmpScreen);
     
103.         BitBlt(CompatibleHDC,0,0,GetDeviceCaps(hdc, HORZRES), GetDeviceCaps(hdc, VERTRES),hdc,0,0,SRCCOPY);
     
104.         
     
105. HANDLE        pbitmapwithoutfileh=DDBtoDIB(BmpScreen, BI_RGB,0,&sizeimage);
     
106.         BITMAPFILEHEADER bfh;
     
107.    //设置位图信息头结构
     
108.         bfh.bfType = ((WORD)('M'<< 8)|'B');
     
109.         bfh.bfReserved1 = 0;
     
110.         bfh.bfReserved2 = 0;
     
111.         bfh.bfSize = 54+sizeimage;
     
112.         bfh.bfOffBits = 54;
     
113.         //创建位图文件   
     
114. HANDLE  hFile=CreateFile(FileName,GENERIC_WRITE,0,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);
     
115. DWORD dwWrite;
     
116. // 写入位图文件头
     
117. WriteFile(hFile,&bfh,sizeof(BITMAPFILEHEADER),&dwWrite,NULL);
     
118. // 写入位图文件其余内容
     
119. WriteFile(hFile,pbitmapwithoutfileh,bfh.bfSize,&dwWrite,NULL);  
     
120. DeleteDC(hdc);
     
121.    
     
122. CloseHandle(CompatibleHDC);
     
123. return true;
     
124. }
     
125. int main(void)
     
126. {
     
127. CapScreen("d:\\desktop.bmp");
     
128. return 0;
     
129. }
     

复制代码

8013

拿下，不说谢谢！~·····