CSDN博客

img zyyuce

捕捉屏幕并保存位图文件

发表于2004/10/26 19:04:00  940人阅读

 
    一百年前的示例代码,贴出来算权当收藏Emotion
////////////////////////////////////////////////////////////////////////////////////////////////////
// 预处理
#include

// 每行数据对齐起到四个字节,下面的定义仅仅用于 24 位色 DIB
#define _WidthPitch(w) ((w + w + w + 3) & 0xFFFFFFFC)
////////////////////////////////////////////////////////////////////////////////////////////////////



////////////////////////////////////////////////////////////////////////////////////////////////////
// 函数声明
HBITMAP WINAPI CaptureScreen(LPCRECT prtRect);
BOOL WINAPI SaveBitmap(HBITMAP hBitmap, LPCSTR ptzFileName);
////////////////////////////////////////////////////////////////////////////////////////////////////



////////////////////////////////////////////////////////////////////////////////////////////////////
// 程序入口,使用自定义入口将生成非常小的 PE 文件,仅有 3K
#ifndef _DEBUG
#pragma comment(linker, "/ENTRY:EntryPoint")
VOID WINAPI EntryPoint()
{
ExitProcess(WinMain(GetModuleHandle(NULL), NULL, GetCommandLine(), SW_SHOWNORMAL));
}
#endif // _DEBUG
////////////////////////////////////////////////////////////////////////////////////////////////////



////////////////////////////////////////////////////////////////////////////////////////////////////
// 主函数
INT WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR pszCmdLine, INT iCmdShow)
{
HBITMAP hBitmap;

hBitmap = CaptureScreen(NULL);
if (hBitmap)
{
if (SaveBitmap(hBitmap, "C://CaptureScreen.bmp"))
{
DeleteObject(hBitmap);

MessageBox(NULL, "屏幕图像保存在 C://CaptureScreen.bmp", "完成", MB_ICONINFORMATION);

return 0;
}

DeleteObject(hBitmap);
}

MessageBox(NULL, "不知道出了什么错误:)……", "错误", MB_ICONSTOP);

return -1;
}
////////////////////////////////////////////////////////////////////////////////////////////////////



////////////////////////////////////////////////////////////////////////////////////////////////////
// 捕捉窗口窗口矩形区域。prtRect 为要捕捉的矩形区域,如果为 NULL,则捕捉整个窗口
HBITMAP WINAPI CaptureScreen(LPCRECT prtRect)
{
INT x;
INT y;
HDC hdcSrc;
HDC hdcMem;
INT iWidth;
INT iHeight;
HBITMAP hbmpTemp;
HBITMAP hbmpReturn;

// 创建屏幕设备场景
hdcSrc = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL);

// 设置捕捉矩形
if (prtRect == NULL)
{
x = 0;
y = 0;
iWidth = GetDeviceCaps(hdcSrc, HORZRES);
iHeight = GetDeviceCaps(hdcSrc, VERTRES);
}
else
{
x = prtRect->left;
y = prtRect->top;
iWidth = prtRect->right - prtRect->left;
iHeight = prtRect->bottom - prtRect->top;
}

// 创建内存兼容设备场景
hdcMem = CreateCompatibleDC(hdcSrc);
hbmpReturn = CreateCompatibleBitmap(hdcSrc, iWidth, iHeight);

// 传送位图
hbmpTemp = (HBITMAP) SelectObject(hdcMem, hbmpReturn);
BitBlt(hdcMem, 0, 0,  iWidth, iHeight, hdcSrc, x, y, SRCCOPY);
SelectObject(hdcMem, hbmpTemp);

// 删除设备场景
DeleteDC(hdcMem);
DeleteDC(hdcSrc);

return hbmpReturn;
}
////////////////////////////////////////////////////////////////////////////////////////////////////



////////////////////////////////////////////////////////////////////////////////////////////////////
// 保存屏幕兼容位图为 24 位色位图文件
BOOL WINAPI SaveBitmap(HBITMAP hBitmap, LPCSTR ptzFileName)
{
HDC hDC;
DWORD dwSize;
BOOL bReturn;
LPBYTE pBits;
HANDLE hFile;
BITMAP bmBitmap;
BITMAPFILEHEADER bfhFile;
BITMAPINFOHEADER bihInfo;

// 返回值初始化为 FALSE
bReturn = FALSE;

// 取得屏幕兼容位图的对象信息
if (GetObject(hBitmap, sizeof(BITMAP), &bmBitmap))
{
// 设置 BITMAPINFOHEADER 结构
bihInfo.biSize = sizeof(BITMAPINFOHEADER);
bihInfo.biWidth = bmBitmap.bmWidth;
bihInfo.biHeight = bmBitmap.bmHeight;
bihInfo.biPlanes = 1;
bihInfo.biBitCount = 24;
bihInfo.biCompression = BI_RGB;
bihInfo.biSizeImage =  _WidthPitch(bmBitmap.bmWidth) * bmBitmap.bmHeight;
bihInfo.biXPelsPerMeter = 0;
bihInfo.biYPelsPerMeter = 0;
bihInfo.biClrUsed = 0;
bihInfo.biClrImportant = 0;

// 设置 BITMAPFILEHEADER 结构
bfhFile.bfType = 0x4D42; // "BM"
bfhFile.bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + bihInfo.biSizeImage;
bfhFile.bfReserved1 = 0;
bfhFile.bfReserved2 = 0;
bfhFile.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);

// 分配内存
pBits = (LPBYTE) HeapAlloc(GetProcessHeap(), 0, bihInfo.biSizeImage);
if (pBits)
{
// 获取屏幕设备场景
hDC = CreateIC("DISPLAY", NULL, NULL, NULL);

// 获取 DIB 数据
if (GetDIBits(hDC, hBitmap, 0, bmBitmap.bmHeight,
pBits, (LPBITMAPINFO) &bihInfo, DIB_RGB_COLORS))
{
// 创建文件
hFile = CreateFile(ptzFileName, GENERIC_WRITE, 0, NULL,
CREATE_ALWAYS, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if (hFile != INVALID_HANDLE_VALUE)
{
// 写入 BITMAPFILEHEADER  信息
bReturn = WriteFile(hFile, &bfhFile, sizeof(BITMAPFILEHEADER), &dwSize, 0);
if (bReturn)
{
// 写入 BITMAPINFOHEADER 信息
bReturn = WriteFile(hFile, &bihInfo, sizeof(BITMAPINFOHEADER), &dwSize, 0);
if (bReturn)
{
// 写入 DIB 数据
bReturn = WriteFile(hFile, pBits, bihInfo.biSizeImage, &dwSize, 0);
}
}

// 关闭文件句柄
CloseHandle(hFile);
}
}

// 删除设备场景,释放内存
DeleteDC(hDC);
HeapFree(GetProcessHeap(), 0, (LPVOID) pBits);
}
}

return bReturn;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
阅读全文
0 0

相关文章推荐

img
取 消
img