抓屏

目录:
 
l          导言
l          GID 函数抓屏
l          DirectX 方式抓屏
l          Windows Media API 抓屏
 
导言
 
有时候我们需要编程抓取整个屏幕上的内容,下面我将介绍抓屏是如何实现的。典型地,我们可以用 GID DirectX 来完成,另外一个选择是 Windows Media API ,在这篇文章我会逐一加以分析。在每一种方法里,一旦我们把屏幕的内容保存到了程序定义的内存块或 bitmap 文件里,我们就可以进一步利用它们来生成动画和电影,这个过程你可以参考“ 从HBitmap创建电影 ”一文中,以获得更多的帮助。
 
GDI 函数抓屏
如果我们不太在意抓屏的效率,并且我们想要的只是一个屏幕快照的话,可以考虑使用 GDI 方式。这种抓屏机制是以“桌面也是一个窗口,桌面也有一个窗口句柄( HWND )”这个简单的常识为基础的,如果我们得到了桌面的设备上下文( DC ),就可以利用 blit (复制)它的内容到我们创建的 DC 中。我们可以用 GetDeskWindow ()得到桌面的窗口句柄,从句柄得到 DC 也是很容易的。具体的实现步骤为:
 
1.          通过 GetDesktopWindow ()函数得到桌面的窗口句柄
2.          GetDC ()取得桌面窗口的 DC
3.          创建和屏幕 DC 兼容的位图和 DC CreateCompatibleBitmap ()和 CreateCompatibleDC ()),并把这个位图选进该 DC SelectObject ())
4.          当你准备好抓屏时,就复制桌面窗口 DC 的内容到兼容 DC ,你就完成的抓屏过程,兼容位图中就是抓屏时刻的屏幕内容
5.          完成后别忘了释放你创建的对象,内存是宝贵的(对别的程序来说)
 
示例代码:
void CaptureScreen()
{
    int nScreenWidth = GetSystemMetrics(SM_CXSCREEN);
    int nScreenHeight = GetSystemMetrics(SM_CYSCREEN);
    HWND hDesktopWnd = GetDesktopWindow();
    HDC hDesktopDC = GetDC(hDesktopWnd);
    HDC hCaptureDC = CreateCompatibleDC(hDesktopDC);
    HBITMAP hCaptureBitmap =CreateCompatibleBitmap(hDesktopDC,
        nScreenWidth, nScreenHeight);
    SelectObject(hCaptureDC,hCaptureBitmap);
    BitBlt(hCaptureDC,0,0,nScreenWidth,nScreenHeight,hDesktopDC,0,0,SRCCOPY);
    SaveCapturedBitmap(hCaptureBitmap); //Place holder - Put your code
                                //here to save the captured image to disk
    ReleaseDC(hDesktopWnd,hDesktopDC);
    DeleteDC(hCaptureDC);
    DeleteObject(hCaptureBitmap);
}
 
上面代码段中, GetSystemMetrics ()返回屏幕的宽度( SM_CXSCREEN )和高度( SM_CYSCREEN )。关于如何保存抓到的位图到文件和如何置到剪贴板,请参看附带的源代码,很简单的。示例代码每隔一段时间就通过上述技术抓屏,并把图像序列保存到动画。
 
DirectX 方式
DreictX 进行抓屏也是很简单的, DirectX 提供了很优雅的实现。
每个 DirectX 程序都包含一个被我们称作缓冲的内存区域,其中保存了和该程序有关的显存内容,这在程序中被称作后台缓冲( Back Buffer ),有些程序有不止一个的后台缓冲。还有一个缓冲,在默认情况下每个程序都可以访问-前台缓冲。前台缓冲保存了和桌面相关的显存内容,实质上就是屏幕图像。
我们的程序通过访问前台缓冲就可以捕捉到当前屏幕的内容。由 DirectX 的底层优化机制做保证,我们的抓屏效率是很高的,至少比 GDI 方式高。
DirectX 程序中访问前台缓冲是很简单的, IDirect3DDevice8 接口提供了 GetFrontBuffer() 方法,它接收一个 IDirect3DSurface8 对象指针做参数,并复制前台缓冲的内容到该 Surface IDirect3DSurfce8 对象可以用 IDirect3DDevice8::CreateImageSurface() 得到。一旦屏幕内容被保存到了这个 surface ,我们就可以用 D3DXSaveSurfaceToFile() 方法直接把内容保存到磁盘 bmp 文件。示例代码如下:
extern IDirect3DDevice8* g_pd3dDevice;
Void CaptureScreen()
{
    IDirect3DSurface8 * pSurface;
    g_pd3dDeviceàCreateImageSurface(ScreenWidth,ScreenHeight,
        D3DFMT_A8R8G8B8,&pSurface);
    g_pd3dDevice->GetFrontBuffer(pSurface);
    D3DXSaveSurfaceToFile("Desktop.bmp",D3DXIFF_BMP,pSurface,
        NULL,NULL);
pSurface->Release();
}
上面, g_pd3dDevice 是一个初始化好的 IDirect3DDevice 对象,这个例子直接把捕捉到的图像保存到文件。然而,有时候我们想访问直接这个图像中的各个位,我们可以使用 IDirect3DSurface8::LockRect() ,它给我们一个执行 surface 内存的指针,也就是捕捉到的图像的数据。我们复制这些数据到程序定义的内存中就可以操作它了。看下面的代码:
extern void* pBits;
extern IDirect3DDevice8* g_pd3dDevice;
IDirect3DSurface8 * pSurface;
g_pd3dDeviceàCreateImageSurface(ScreenWidth,ScreenHeight,
                                D3DFMT_A8R8G8B8,&pSurface);
g_pd3dDevice->GetFrontBuffer(pSurface);
D3DLOCKED_RECT lockedRect;
pSurfaceàLockRect(&lockedRect,NULL,
                  D3DLOCK_NO_DIRTY_UPDATE|D3DLOCK_NOSYSLOCK|
                  D3DLOCK_READONLY)));
for( int i=0 ; i < ScreenHeight ; i++)
{
    memcpy( (BYTE*) pBits + i * ScreenWidth * BITSPERPIXEL / 8 ,
        (BYTE*) lockedRect.pBits + i* lockedRect.Pitch ,
        ScreenWidth * BITSPERPIXEL / 8);
}
g_pSurface->UnlockRect();
pSurface->Release();
 
上面的 pBits 是一个 void* ,请保证为先为它分配组足够的内存空间。 BITSPERPIXEL 一般用 32 位色即可,它也取决于你的显示器当前配置。一个需要注意的是, surface 的宽度和被捕捉的屏幕宽度不一样。由于内存对齐的原因 ( WORD 对齐的内存通常在访问时效率较高 ) surface 在每行结尾处可能会有多余的 bits 以使它对齐到 word 边界上。 lockedRect.Pitch 给我们提供了两个连续行的开端之间的字节数。也就是说我们在读取一行时要向后移动指针 Pitch 字节而不是 Width 字节。你可以用下面的代码反序复制 surface
for( int i=0 ; i < ScreenHeight ; i++)
{
    memcpy((BYTE*) pBits +( ScreenHeight - i - 1) *
        ScreenWidth * BITSPERPIXEL/8 ,
        (BYTE*) lockedRect.pBits + i* lockedRect.Pitch ,
        ScreenWidth* BITSPERPIXEL/8);
}
这对于从 top-down 位图到 bottom-up 位图很有用。
 
我们还可以使用 IDirect3DSurface9 GetDC() 方法取得 DirectX surface GDI 兼容 DC ,然后复制它的内容到我们的兼容 DC 。如果你用的是 DirectX9 ,试试吧。
 
最后,需要注意的一点,文档提到: FrontBuffer 是一个比较慢的操作,设计就是如此,所以在效率很关键的程序中应避免使用。已经警告你了!本文附带的源代码用这种技术定时捕捉屏幕,并保存为动画。
 
[NextPage]
 
Windows Media API 抓屏
Windows Media 9.0 支持用 Windows Media Encoder 9 API 来抓屏。它有一个编码器叫 Windows Media Video 9 Screen codec ,特别为抓屏优化过。 Windows Media Encoder API 提供了一个 IWMEncoder2 接口可以用来高效地捕捉屏幕图像。
 
用这种技术进行抓屏也很简单,首先我们用 CoCreateInstance() 创建一个 IWMEncoder2 对象:
IWMEncoder2* g_pEncoder=NULL;
CoCreateInstance(CLSID_WMEncoder,NULL,CLSCTX_INPROC_SERVER,
        IID_IWMEncoder2,(void**)&g_pEncoder);
这个 Encoder 对象包含了捕捉屏幕的所需的全部操作,然而为了正确地工作,编码器对象的行为取决于被称作 profile 的设置。一个 profile 只是一个包含了所有控制编码操作设置的文件,我们可以根据被捕捉的数据的特性在运行时创建包含自定义设置的 profile 。为了在你的抓屏程序中使用 profile ,我们基于 Windows Media Video 9 Screen codec 来创建自定义的 profile 。自定义的 profile 对象从 IWMEncProfile2 开始就被支持了。我们可以用 CoCreateInstance 来创建自定义 profile
IWMEncProfile2* g_pProfile=NULL;
CoCreateInstance(CLSID_WMEncProfile2,NULL,CLSCTX_INPROC_SERVER,
        IID_IWMEncProfile2,(void**)&g_pProfile);
我需要在 profile 里指定编码器的听众( audience )。每个 profile 可以包含多个听众配置,它们是 IWMEncAudienceObj 接口对象。这里我们为 profile 使用一个听众。我们可以通过 IWMEncProfile::AddAudience() 为我们的 profile 创建听众,这个函数返回一个 IWMEncAudienceObj 指针,可以用来配置视频编码器 ( IWMEncAudienceObj::put_VideoCodec() ) ,视频帧对象 ( IWMEncAudienceObj::put_VideoHeight() IWMEncAudienceObj::put_VideoWidth() ) 我们用下面的代码来配置视频编码器:
extern IWMEncAudienceObj* pAudience;
#define VIDEOCODEC MAKEFOURCC('M','S','S','2')
    //MSS2 is the fourcc for the screen codec
 
long lCodecIndex=-1;
g_pProfile->GetCodecIndexFromFourCC(WMENC_VIDEO,VIDEOCODEC,
    &lCodecIndex); //Get the Index of the Codec
pAudience->put_VideoCodec(0,lCodecIndex);
 
fourcc 是针对每个编码器的唯一的标识, Windows Media Video 9 Screen codec fourcc MSS2 IWMEncAudienceObj::put_VideoCodec() 接受 profile 索引来组织一个 profile ,索引可以用 IWMEncProfile::GetCodecIndexFromFourCC() 取得。
 
一旦我们配置完毕一个 profile 对象,我们就可以用 IWMEncSourceGroup :: put_Profile() 选择这个 profile 到我们的编码器。一个源组( SourceGruop )是一组视频流来源或音频流来源,或 html 来源。每个编码器可以使用许多源组,并从中取得输入数据。由于我们的程序仅仅使用视频流中是视频来源。这个视频来源需要用 IWMEncVideoSource2::SetInput(BSTR) Screen Device 来配置为输入来源:
 
extern IWMEncVideoSource2* pSrcVid;
pSrcVid->SetInput(CComBSTR("ScreenCap://ScreenCapture1");
 
目的输出可以用 IWMEncFile::put_LocalFileName() 配置为保存到视频文件( wmv 文件)。 IWMEncFile 对象可以用 IWMEncoder::get_File() 得到:
IWMEncFile* pOutFile=NULL;
g_pEncoder->get_File(&pOutFile);
pOutFile->put_LocalFileName(CComBSTR(szOutputFileName);
 
现在,一旦编码器对象的一切所需配置都完成后,我们就可以用 IWMEncoder::Start() 开始抓屏。 IWMEncoder::Stop() IWMEncoder::Pause 可以用来停止和暂停捕捉。
这些适用于全屏捕捉,我们也可以通过调整输入视频来源流的属性来选择一个区域进行捕捉。我们可以用 IWmEnVideoSource2 I PropertyBag 接口来实现:
#define WMSCRNCAP_WINDOWLEFT CComBSTR("Left")
#define WMSCRNCAP_WINDOWTOP CComBSTR("Top")
#define WMSCRNCAP_WINDOWRIGHT CComBSTR("Right")
#define WMSCRNCAP_WINDOWBOTTOM CComBSTR("Bottom")
#define WMSCRNCAP_FLASHRECT CComBSTR("FlashRect")
#define WMSCRNCAP_ENTIRESCREEN CComBSTR("Screen")
#define WMSCRNCAP_WINDOWTITLE CComBSTR("WindowTitle")
extern IWMEncVideoSource2* pSrcVid;
int nLeft, nRight, nTop, nBottom;
pSrcVid->QueryInterface(IID_IPropertyBag,(void**)&pPropertyBag);
CComVariant varValue = false;
pPropertyBag->Write(WMSCRNCAP_ENTIRESCREEN,&varValue);
varValue = nLeft;
pPropertyBag->Write( WMSCRNCAP_WINDOWLEFT, &varValue );
varValue = nRight;
pPropertyBag->Write( WMSCRNCAP_WINDOWRIGHT, &varValue );
varValue = nTop;
pPropertyBag->Write( WMSCRNCAP_WINDOWTOP, &varValue );
varValue = nBottom;
pPropertyBag->Write( WMSCRNCAP_WINDOWBOTTOM, &varValue );
 
本文附带的源码实现此中技术的抓屏。除去生成的动画质量很好外,一个有意思的地方是鼠标指针也被抓到了( GDI DirectX 默认是不抓取鼠标指针的)。
 
注意,为了适用 WindowMedia9.0 API ,你的电脑必须安装 Windows Media9.0 SDK ,你可以用下面地址下载:
最终用户必须安装 Windows Media Encoder 9 系列才能运行你的程序。在发布基于 Windows Media Encoder SDK 的程序时, Windows Media Encoder 软件也必须附带上去,要么在你的软件安装时自动安装 Windows Media Encoder 要么让用户自己下载安装。
 
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值