45fan.com - 路饭网

搜索: 您的位置主页 > 网络频道 > 阅读资讯:如何抓取屏幕中选中区域并存取为bmp位图的类?

如何抓取屏幕中选中区域并存取为bmp位图的类?

2016-09-01 19:43:02 来源:www.45fan.com 【

如何抓取屏幕中选中区域并存取为bmp位图的类?

/******************************************************************************

* FILE NAME: CopyScreenToBitmap.h *

* *

* DESCRIPTION: *

* 截屏位图文件 *

* *

* COPYRIGHT: *

* (C) Copyright LiuYong *

* All Rights Reserved *

* *

* AUTHOR: *

* liu yong *

* DATE : 2006-11-29 *

* *

*******************************************************************************/

class CCopyScreenToBitmap

{

public:

CCopyScreenToBitmap();

~CCopyScreenToBitmap();

//截取指定区域的屏幕到位图文件,文件目录默认让用户自己选择

BOOL SaveRectToBitmap( LPRECT lpRect, LPSTR lpFileName = NULL );

//截取全屏幕到位图文件,文件目录默认让用户自己选择

BOOL SaveScreenToBitmap( LPSTR lpFileName = NULL );

private:

HBITMAP SaveBitmapCopyScreenToBitmap( LPRECT lpRect );

BOOL SaveBitmapToFile( HBITMAP hBitmap , LPSTR lpFileName ) ;

void SaveBitmapToClipboard( HBITMAP hBitmap, HWND hWnd );

};

///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

// CopyScreenToBitmap.cpp: implementation of the CCopyScreenToBitmap class.

//

//////////////////////////////////////////////////////////////////////

#include "stdafx.h"

#include "CopyScreenToBitmap.h"

#include <afxdlgs.h>

#ifdef _DEBUG

#undef THIS_FILE

static char THIS_FILE[]=__FILE__;

#define new DEBUG_NEW

#endif

//////////////////////////////////////////////////////////////////////

// Construction/Destruction

//////////////////////////////////////////////////////////////////////

CCopyScreenToBitmap::CCopyScreenToBitmap()

{

}

CCopyScreenToBitmap::~CCopyScreenToBitmap()

{

}

BOOL CCopyScreenToBitmap::SaveRectToBitmap(LPRECT lpRect, LPSTR lpFileName)

{

if(!lpRect)

{

::MessageBox( NULL,"请指定要截图的区域!","提示",MB_ICONINFORMATION );

return FALSE;

}

HBITMAP hBitmap = SaveBitmapCopyScreenToBitmap(lpRect);

if( !lpFileName )

{

CString Path;

CFileDialog dlg(FALSE,"*.*","新图形",NULL,

"位图文件(*.bmp)|*.bmp|所有文件(*.*)|*.*||",NULL);

//not change the default dir

dlg.m_ofn.Flags |= OFN_EXPLORER|OFN_OVERWRITEPROMPT|OFN_HIDEREADONLY|OFN_NOCHANGEDIR;

dlg.m_ofn.lpstrTitle = "保存图形";

if (dlg.DoModal() == IDCANCEL)

{

::DeleteObject(hBitmap);

return FALSE;

}

Path = dlg.GetPathName();

//save the bitmap

return SaveBitmapToFile(hBitmap,Path.GetBuffer( Path.GetLength() ) );

}

else

{

return SaveBitmapToFile( hBitmap, lpFileName );

}

}

BOOL CCopyScreenToBitmap::SaveScreenToBitmap( LPSTR lpFileName )

{

int cxSrc = GetSystemMetrics (SM_CXSCREEN) ; //源设备宽--屏幕宽

int cySrc = GetSystemMetrics (SM_CYSCREEN) ; //源设备高--屏幕高

CRect rect(0,0,cxSrc,cySrc);

HBITMAP hBitmap = SaveBitmapCopyScreenToBitmap( &rect );

if( !lpFileName )

{

CString Path;

CFileDialog dlg(FALSE,"*.*","新图形",NULL,

"位图文件(*.bmp)|*.bmp|所有文件(*.*)|*.*||",NULL);

//not change the default dir

dlg.m_ofn.Flags |= OFN_EXPLORER|OFN_OVERWRITEPROMPT|OFN_HIDEREADONLY|OFN_NOCHANGEDIR;

dlg.m_ofn.lpstrTitle = "保存图形";

if (dlg.DoModal() == IDCANCEL)

{

::DeleteObject(hBitmap);

return FALSE;

}

Path = dlg.GetPathName();

//save the bitmap

return SaveBitmapToFile(hBitmap,Path.GetBuffer( Path.GetLength() ) );

}

else

{

return SaveBitmapToFile( hBitmap, lpFileName );

}

}

//lpRect 代表选定区域

HBITMAP CCopyScreenToBitmap::SaveBitmapCopyScreenToBitmap(LPRECT lpRect)

{

// 屏幕和内存设备描述表

HDC hScrDC, hMemDC;

// 位图句柄

HBITMAP hBitmap, hOldBitmap;

// 选定区域坐标

int nX, nY, nX2, nY2;

// 位图宽度和高度

int nWidth, nHeight;

// 屏幕分辨率

int xScrn, yScrn;

// 确保选定区域不为空矩形

if (IsRectEmpty(lpRect))

return NULL;

//为屏幕创建设备描述表

hScrDC = CreateDC("DISPLAY", NULL, NULL, NULL);

//为屏幕设备描述表创建兼容的内存设备描述表

hMemDC = CreateCompatibleDC(hScrDC);

// 获得选定区域坐标

nX = lpRect->left;

nY = lpRect->top;

nX2 = lpRect->right;

nY2 = lpRect->bottom;

// 获得屏幕分辨率

xScrn = GetDeviceCaps(hScrDC, HORZRES);

yScrn = GetDeviceCaps(hScrDC, VERTRES);

//确保选定区域是可见的

if (nX < 0)

nX = 0;

if (nY < 0)

nY = 0;

if (nX2 > xScrn)

nX2 = xScrn;

if (nY2 > yScrn)

nY2 = yScrn;

nWidth = nX2 - nX;

nHeight = nY2 - nY;

// 创建一个与屏幕设备描述表兼容的位图

hBitmap = CreateCompatibleBitmap(hScrDC, nWidth, nHeight);

// 把新位图选到内存设备描述表中

hOldBitmap = (HBITMAP)SelectObject(hMemDC, hBitmap);

// 把屏幕设备描述表拷贝到内存设备描述表中

BitBlt(hMemDC, 0, 0, nWidth, nHeight,hScrDC, nX, nY, SRCCOPY);

//得到屏幕位图的句柄

hBitmap = (HBITMAP)SelectObject(hMemDC, hOldBitmap);

//清除

DeleteDC(hScrDC);

DeleteDC(hMemDC);

// 返回位图句柄

return hBitmap;

}

/////得到屏幕位图句柄以后,我们

/////可以把屏幕内容粘贴到剪贴板上.

void CCopyScreenToBitmap::SaveBitmapToClipboard(HBITMAP hBitmap,HWND hWnd)

{

//hWnd为程序窗口句柄

if (OpenClipboard(hWnd))

{

//清空剪贴板

EmptyClipboard();

SetClipboardData(CF_BITMAP, hBitmap);

//关闭剪贴板

CloseClipboard();

}

}

//我们也可以把屏幕内容以位图格式存到磁盘文件上.

//lpFileName 为位图文件名

BOOL CCopyScreenToBitmap::SaveBitmapToFile(HBITMAP hBitmap , LPSTR lpFileName)

{

//设备描述表

HDC hDC;

//当前显示分辨率下每个像素所占字节数

int iBits;

//位图中每个像素所占字节数

WORD wBitCount=32;

//定义调色板大小, 位图中像素字节大小 ,

//位图文件大小 , 写入文件字节数

DWORD dwPaletteSize=0,

dwBmBitsSize,

dwDIBSize, dwWritten;

//位图属性结构

BITMAP Bitmap;

//位图文件头结构

BITMAPFILEHEADER bmfHdr;

//位图信息头结构

BITMAPINFOHEADER bi;

//指向位图信息头结构

LPBITMAPINFOHEADER lpbi;

//定义文件,分配内存句柄,调色板句柄

HANDLE fh, hDib;

HPALETTE 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 if (iBits <= 24)

wBitCount = 24;

//计算调色板大小

if (wBitCount <= 8)

dwPaletteSize = (1 << wBitCount) *sizeof(RGBQUAD);

//设置位图信息头结构

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.biClrUsed = 0;

bi.biClrImportant = 0;

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

//为位图内容分配内存

hDib = GlobalAlloc(GHND,dwBmBitsSize+dwPaletteSize+sizeof(BITMAPINFOHEADER));

lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);

*lpbi = bi;

// 处理调色板

hPal = (HPALETTE)GetStockObject(DEFAULT_PALETTE);

if (hPal)

{

hDC = GetDC(NULL);

hOldPal = SelectPalette(hDC, hPal, FALSE);

RealizePalette(hDC);

}

// 获取该调色板下新的像素值

GetDIBits(hDC, hBitmap, 0, (UINT)Bitmap.bmHeight,(LPSTR)lpbi + sizeof(BITMAPINFOHEADER)+dwPaletteSize,(LPBITMAPINFO)lpbi, DIB_RGB_COLORS);

//恢复调色板

if (hOldPal)

{

SelectPalette(hDC, hOldPal, TRUE);

RealizePalette(hDC);

ReleaseDC(NULL, hDC);

}

//创建位图文件

fh = CreateFile(lpFileName, 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;

}

 

本文地址:http://www.45fan.com/a/question/70922.html
Tags: 屏幕 抓取 选中
编辑:路饭网
关于我们 | 联系我们 | 友情链接 | 网站地图 | Sitemap | App | 返回顶部