抓图函数

//抓图函数

void CWallPaperDlg::PrintScreenToFile(char filename[])
{
 //获取当前整个屏幕DC
 HDC hDC = ::GetDC(NULL);

 //获得颜色模式
 int BitPerPixel = GetDeviceCaps(hDC, BITSPIXEL);
 int Width = GetDeviceCaps(hDC, HORZRES);
 int Height = GetDeviceCaps(hDC, VERTRES);
 
 //创建与获得的CDC兼容的内存设备描述
 HDC hMemDC = CreateCompatibleDC(hDC);
 
 //建立和屏幕兼容的bitmap
 HBITMAP hMemBitmap, hOldMemBitmap;
 
 //初始化memBitmap
 hMemBitmap = CreateCompatibleBitmap(hDC, Width, Height);
 
 //将memBitmap选入内存DC
 hOldMemBitmap = (HBITMAP)SelectObject(hMemDC, hMemBitmap);
 
 //复制屏幕图像到内存DC
 BitBlt(hMemDC,0, 0, Width, Height, hDC, 0, 0, SRCCOPY);
 
 //以下代码保存hMemDC中的位图到文件
 BITMAP bmp;
 //获得位图信息
 GetObject( hMemBitmap, (int)sizeof(BITMAP), &bmp );
 
 FILE *fp = fopen(filename, "w+b");
 
 //位图信息头
 BITMAPINFOHEADER bih = {0};
 //每个像素字节大小
 bih.biBitCount = bmp.bmBitsPixel;
 //无压缩
 bih.biCompression = BI_RGB;
 //高度
 bih.biHeight = bmp.bmHeight;
 bih.biPlanes = 1;
 bih.biSize = sizeof(BITMAPINFOHEADER);
 
 //图像数据大小
 bih.biSizeImage = bmp.bmWidthBytes * bmp.bmHeight;
 
 //宽度
 bih.biWidth = bmp.bmWidth;
 
 //位图文件头
 BITMAPFILEHEADER bfh = {0};
 
 //到位图数据的偏移量
 bfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
 
 //文件总的大小
 bfh.bfSize = bfh.bfOffBits + bmp.bmWidthBytes * bmp.bmHeight;
 
 //字符"BM"，表示位图文件
 bfh.bfType = (WORD)0x4d42;
 
 //写入位图文件头
 fwrite(&bfh, 1, sizeof(BITMAPFILEHEADER), fp);
 //写入位图信息头
 fwrite(&bih, 1, sizeof(BITMAPINFOHEADER), fp);
 //申请内存保存位图数据
 BYTE * p = new BYTE[bmp.bmWidthBytes * bmp.bmHeight];

 //获取位图数据
 GetDIBits(hMemDC, hMemBitmap, 0, Height, p, (LPBITMAPINFO) &bih, DIB_RGB_COLORS);
  
 //写入位图数据
 fwrite(p, 1, bmp.bmWidthBytes * bmp.bmHeight, fp);
 
 delete [] p;
 fclose(fp);
 SelectObject(hMemDC, hOldMemBitmap);
}

 

 

---------------------------------------------------------------------------------------------------

static HBITMAP CopyDCToBitmap(HDC hDC, LPRECT lpRect)
{
if(!hDC || !lpRect || IsRectEmpty(lpRect))
   return NULL;

HDC hMemDC; // 内存设备描述表
HBITMAP hBitmap, hOldBitmap; // 位图句柄
int nX, nY, nX2, nY2; // 选定区域坐标
int nWidth, nHeight;   // 位图宽度和高度

// 获得选定区域坐标
nX = lpRect->left;
nY = lpRect->top;
nX2 = lpRect->right;
nY2 = lpRect->bottom;
nWidth = nX2 - nX;
nHeight = nY2 - nY;

// 为指定设备描述表创建兼容的内存设备描述表
hMemDC = CreateCompatibleDC(hDC);

// 创建一个与指定设备描述表兼容的位图
hBitmap = CreateCompatibleBitmap(hDC, nWidth, nHeight);

// 把新位图选到内存设备描述表中
hOldBitmap = (HBITMAP)SelectObject(hMemDC, hBitmap);

// 把指定的设备描述表拷贝到内存设备描述表中
StretchBlt(hMemDC, 0, 0, nWidth, nHeight, hDC, nX, nY, nWidth, nHeight, SRCCOPY);

// 得到内存位图的句柄
hBitmap = (HBITMAP)SelectObject(hMemDC, hOldBitmap);

// 清除
DeleteDC(hMemDC);
DeleteObject(hOldBitmap);

// 返回位图句柄
return hBitmap;
}

// 把HBITMAP保存成位图
static BOOL SaveBmp(HBITMAP hBitmap, const char *pchFileName)
{
if(!hBitmap || !pchFileName)
   return FALSE;

HDC hDC;
int iBits; // 当前分辨率下每象素所占字节数
WORD wBitCount; // 位图中每象素所占字节数

// 定义调色板大小， 位图中像素字节大小 ，位图文件大小 ， 写入文件字节数
DWORD dwPaletteSize=0, dwBmBitsSize=0, dwDIBSize=0, dwWritten=0;
BITMAP Bitmap;   // 位图属性结构
BITMAPFILEHEADER bmfHdr;   // 位图文件头结构
BITMAPINFOHEADER bi;   // 位图信息头结构
LPBITMAPINFOHEADER lpbi;   // 指向位图信息头结构
HANDLE fh, hDib, hPal, hOldPal = NULL; // 定义文件，分配内存句柄，调色板句柄

// 计算位图文件每个像素所占字节数
hDC = CreateDC("DISPLAY", NULL, NULL, NULL);
iBits = GetDeviceCaps(hDC, BITSPIXEL) * GetDeviceCaps(hDC, PLANES);
DeleteDC(hDC);
if (iBits <= 1)
   wBitCount = 1;
else if (iBits <= 4)
   wBitCount = 4;
else if (iBits <= 8)
   wBitCount = 8;
else
   wBitCount = 24;

GetObject(hBitmap, sizeof(Bitmap), (LPSTR)&Bitmap);
bi.biSize = sizeof(BITMAPINFOHEADER);
bi.biWidth = Bitmap.bmWidth;
bi.biHeight = Bitmap.bmHeight;
bi.biPlanes = 1;
bi.biBitCount = wBitCount;
bi.biCompression = BI_RGB;
bi.biSizeImage = 0;
bi.biXPelsPerMeter = 0;
bi.biYPelsPerMeter = 0;
bi.biClrImportant = 0;
bi.biClrUsed = 0;

dwBmBitsSize = ((Bitmap.bmWidth * wBitCount + 31) / 32) * 4 * Bitmap.bmHeight;

//为位图内容分配内存
hDib = GlobalAlloc(GHND,dwBmBitsSize + dwPaletteSize + sizeof(BITMAPINFOHEADER));
lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);
*lpbi = bi;

// 处理调色板  
hPal = GetStockObject(DEFAULT_PALETTE);
if (hPal) {
   hDC = ::GetDC(NULL);
   hOldPal = ::SelectPalette(hDC, (HPALETTE)hPal, FALSE);
   RealizePalette(hDC);
}

// 获取该调色板下新的像素值
GetDIBits(hDC, hBitmap, 0, (UINT) Bitmap.bmHeight,
   (LPSTR)lpbi + sizeof(BITMAPINFOHEADER) + dwPaletteSize,
   (BITMAPINFO *)lpbi, DIB_RGB_COLORS);

// 恢复调色板  
if(hOldPal) {
   ::SelectPalette(hDC, (HPALETTE)hOldPal, TRUE);
   RealizePalette(hDC);
   ::ReleaseDC(NULL, hDC);
}

// 创建位图文件
fh = CreateFile(pchFileName, GENERIC_WRITE,0, NULL, CREATE_ALWAYS,
   FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);

if (fh == INVALID_HANDLE_VALUE)
   return FALSE;

// 设置位图文件头
bmfHdr.bfType = 0x4D42; // "BM"
dwDIBSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)
   + dwPaletteSize + dwBmBitsSize;
bmfHdr.bfSize = dwDIBSize;
bmfHdr.bfReserved1 = 0;
bmfHdr.bfReserved2 = 0;
bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER)
   + (DWORD)sizeof(BITMAPINFOHEADER) + dwPaletteSize;

// 写入位图文件头
WriteFile(fh, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);

// 写入位图文件其余内容
WriteFile(fh, (LPSTR)lpbi, dwDIBSize, &dwWritten, NULL);

// 清除  
GlobalUnlock(hDib);
GlobalFree(hDib);
CloseHandle(fh);

return TRUE;
}

// 抓图

if (m_bRealPlay || m_bPlayBack) {
   CRect rect, barRect;
   HDC hDC = GetDC()->m_hDC;
   GetClientRect(&rect);
   if (m_bPlayBack) {
    m_bar.GetWindowRect(&barRect);
    rect.bottom = rect.bottom - barRect.Height();
   }
   HBITMAP bmp = CopyDCToBitmap(hDC, &rect);

   if (!SaveBmp(bmp, "D://a.bmp"))
    ::AfxMessageBox("save the file /"a.bmp/" failed.");
}