MFC实现屏幕截图

BOOL CMyDlg::SaveBitmapToFile(HDC hDc, CBitmap & bitmap, LPCTSTR lpszFileName)
{
BOOL ret = TRUE;
BITMAP btm;
bitmap.GetBitmap(&btm);
//本函数用于查看CBitmap对象的信息。返回的信息存放在pBitMap指向的BITMAP结构中
DWORD size = btm.bmWidthBytes * btm.bmHeight;
//分配内存
HGLOBAL hMem = GlobalAlloc(GMEM_FIXED|GMEM_ZEROINIT, size);
if(hMem == NULL)
{
return FALSE;
}

//加锁 保护内存
LPSTR lpData = (LPSTR)GlobalLock(hMem);
//定义位图头结构并初始化
BITMAPINFOHEADER bih;
bih.biSize = sizeof(BITMAPINFOHEADER);
bih.biWidth = btm.bmWidth;
bih.biHeight = btm.bmHeight;
bih.biPlanes = 1;
bih.biBitCount = btm.bmBitsPixel;
bih.biCompression = 0;
bih.biSizeImage = size;
bih.biXPelsPerMeter = 0;
bih.biYPelsPerMeter = 0;
bih.biClrUsed = 0;
bih.biClrImportant = 0;

//将位图中的数据以bits形式放入lpData指向的位图数组中
if(GetDIBits(hDc, bitmap, 0, bih.biHeight, lpData,(BITMAPINFO *)&bih, DIB_RGB_COLORS) == 0)
{
GlobalFree(hMem);
return FALSE;
}

//位图文件结构初始化
BITMAPFILEHEADER bfh;
bfh.bfType = ((WORD)('M'<< 8)|'B');
bfh.bfReserved1 = 0;
bfh.bfReserved2 = 0;
bfh.bfSize = sizeof(bfh) + size;
bfh.bfOffBits = sizeof(bfh);

//创建位图文件并写入数据
CFile bf;
if(bf.Open(lpszFileName, CFile::modeCreate | CFile::modeWrite))
{
bf.WriteHuge(&bfh, sizeof(BITMAPFILEHEADER));
bf.WriteHuge(&bih, sizeof(BITMAPINFOHEADER));
bf.WriteHuge(lpData, size);
bf.Close();
}
else
{
ret = FALSE;
}

//释放内存
GlobalFree(hMem);
return ret;
}