三月 22, 2010

使用DirectX截屏

Written by

网上有很多关于DirectX截屏的文章,但大都是屏幕截图,很少有窗口截图,本文则两者都涉及到,先讲如何截取整个屏幕,再讲如何截取某个窗口,其实二者的区别不大,只是某个参数的设置不同而已,最后我们还将扩展到任意区域的截图。

首先看一下截屏用到的函数,最核心的当然是D3DXSaveSurfaceToFile,先看下函数原型

  1. 1 HRESULT D3DXSaveSurfaceToFile(
  2. 2  LPCTSTR pDestFile,
  3. 3  D3DXIMAGE_FILEFORMAT DestFormat,
  4. 4  LPDIRECT3DSURFACE9 pSrcSurface,
  5. 5  CONST PALETTEENTRY * pSrcPalette,
  6. 6  CONST RECT * pSrcRect
  7. 7 );

复制代码

第一个参数是指向设备的指针,不多说啦

第二个参数是截图文件的类型,支持的类型还不少,主要有下面这些

BMP,JPG,TGA,PNG,DDS,PPM,DIB,HDR,PFM

这里我们使用BMP-即位图格式

第三个参数是指向Surface的指针,也就是保存了截图数据的表面

第四个参数是Surface的调色板,这里不使用,设置为NULL

最后一个参数是Surface的矩形区域,也就是我们可以只截取Surface上某一矩形区域的数据,其实截取全屏和截取窗口的差别也就在这个参数的设置上

其他的函数在下面会逐一讲解

现在来定义我们的截屏函数,首先我们需要一个设备指针,因为在DX中,任何操作都与设备密切相关,所以设备指针几乎是每个DX函数都要用到的参数,我们这个函数也不例外,齐次需要一个窗口句柄,当我们截取窗口时,把窗口句柄传入,当我们截取整个屏幕时,直接传入NULL。最后我们需要一个字符串参数来指定截图对应的文件名,如下
1 BOOL ScreenShot(LPDIRECT3DDEVICE9 lpDevice, HWND hWnd, TCHAR* fileName)

详细步骤:

首先我们需要获取显示模式,注意这里获取的是显卡的显示模式,而不是设备的显示模式,因为设备的显示模式既有窗口模式,也有全屏模式,所以它的分辨率是不确定的,而显卡的显示模式返回的始终是最大分辨率,我们需要创建整个屏幕区域对应的Surface,当截取真个屏幕时,直接保存即可,当截取窗口时,我们将窗口所对应的区域保存即可

获取显卡显示模式的代码如下

  1. 1 HRESULT hr;
  2. 2
  3. 3 // Get adapter display mode
  4. 4 D3DDISPLAYMODE mode;
  5. 5 if (FAILED(hr = lpDevice->GetDisplayMode(0, &mode)))
  6. 6    return hr;
  7. 7

复制代码

下面开始创建表面,这个表面是对应整个屏幕的

  1. 1 // Create the surface to hold the screen image data
  2. 2 LPDIRECT3DSURFACE9 surf;
  3. 3 if (FAILED(hr = lpDevice->CreateOffscreenPlainSurface(mode.Width,
  4. 4    mode.Height, D3DFMT_A8R8G8B8, D3DPOOL_SYSTEMMEM, &surf, NULL))) //注意第四个参数不能是D3DPOOL_DEFAULT
  5. 5 {
  6. 6    return hr;
  7. 7 }
  8. 8

复制代码

接下来获取屏幕对应的数据,这个函数实际上是将显存中的数据拷贝到系统内存中

  1. 1 // Get the screen data
  2. 2 if (FAILED(hr = lpDevice->GetFrontBufferData(0, surf)))
  3. 3 {
  4. 4    surf->Release() ;
  5. 5    return hr ;
  6. 6 }
  7. 7

复制代码

接下来我们判断是截取窗口还是截取屏幕,很简单,只需判断hWnd是否为NULL即可,如果是截取窗口则设置窗口对应的矩形区域即可

  1. 1 // area to capture
  2. 2 RECT *rect = NULL ;
  3. 3
  4. 4 WINDOWINFO windowInfo ;
  5. 5 windowInfo.cbSize = sizeof(WINDOWINFO) ;
  6. 6
  7. 7 if(hWnd) // capture window
  8. 8 {
  9. 9    GetWindowInfo(hWnd, &windowInfo) ;
  10. 10    rect = &windowInfo.rcWindow ;
  11. 11 }
  12. 12

复制代码

最后一部,保存截图!

  1. 1 // Save the screen date to file
  2. 2 hr = D3DXSaveSurfaceToFile(fileName, D3DXIFF_BMP, surf, NULL, rect);
  3. 3
  4. 4 surf->Release() ;
  5. 5
  6. 6 return hr ;
  7. 7

复制代码

大功告成!

完整代码

  1. 1 BOOL ScreenShot(LPDIRECT3DDEVICE9 lpDevice, HWND hWnd, TCHAR* fileName)
  2. 2 {
  3. 3    HRESULT hr;
  4. 4   
  5. 5    // Get adapter display mode
  6. 6    D3DDISPLAYMODE mode;
  7. 7    if (FAILED(hr = lpDevice->GetDisplayMode(0, &mode)))
  8. 8        return hr;
  9. 9
  10. 10    // Create the surface to hold the screen image data
  11. 11    LPDIRECT3DSURFACE9 surf;
  12. 12    if (FAILED(hr = lpDevice->CreateOffscreenPlainSurface(mode.Width,
  13. 13        mode.Height, D3DFMT_A8R8G8B8, D3DPOOL_SYSTEMMEM, &surf, NULL))) //注意第四个参数不能是D3DPOOL_DEFAULT
  14. 14    {
  15. 15        return hr;
  16. 16    }
  17. 17
  18. 18    // Get the screen data
  19. 19    if (FAILED(hr = lpDevice->GetFrontBufferData(0, surf)))
  20. 20    {
  21. 21        surf->Release() ;
  22. 22        return hr ;
  23. 23    }
  24. 24
  25. 25    // area to capture
  26. 26    RECT *rect = NULL ;
  27. 27
  28. 28    WINDOWINFO windowInfo ;
  29. 29    windowInfo.cbSize = sizeof(WINDOWINFO) ;
  30. 30
  31. 31    if(hWnd) // capture window
  32. 32    {
  33. 33        GetWindowInfo(hWnd, &windowInfo) ;
  34. 34        rect = &windowInfo.rcWindow ;
  35. 35    }
  36. 36
  37. 37    // Save the screen date to file
  38. 38    hr = D3DXSaveSurfaceToFile(fileName, D3DXIFF_BMP, surf, NULL, rect);
  39. 39
  40. 40    surf->Release() ;
  41. 41
  42. 42    return hr ;
  43. 43 }

复制代码

那么如何实现任意区域截屏呢,我想大家已经想到了,假设使用鼠标拖拽的方法截图,记下鼠标按下和抬起时的坐标,构造一个RECT,然后传递给 D3DXSaveSurfaceToFile函数就可以了,需要注意到是,由于鼠标拖拽到方向是任意的,所以在构造RECT的时候要注意right < left或者bottom < top 的情况,用下面的方法可以处理

  1. 1 int left = 0 ;
  2. 2 int right = 0 ;
  3. 3 int top = 0 ;
  4. 4 int bottom = 0 ;
  5. 5 RECT rect ;
  6. 6
  7. 7 case WM_LBUTTONDOWN:
  8. 8    left = ( short )LOWORD( lParam );
  9. 9    top = ( short )HIWORD( lParam );
  10. 10    break ;
  11. 11
  12. 12 case WM_LBUTTONUP:
  13. 13    right = ( short )LOWORD( lParam );
  14. 14    bottom = ( short )HIWORD( lParam );
  15. 15
  16. 16    rect.left = min(left, right) ;
  17. 17    rect.right = max(left, right) ;
  18. 18    rect.top = min(top, bottom) ;
  19. 19    rect.bottom = max(top, bottom) ;
  20. 20    // 调用截图函数
  21. 21

复制代码

(文/AutumnWinter

来自:http://www.pin5i.com/showtopic-26129.html

 

Category : VC++Window$

Tags :

发表评论

电子邮件地址不会被公开。

Proudly powered by WordPress and Sweet Tech Theme