多媒体编程——屏幕截图

多媒体编程——屏幕截图

 

为什么要先讲屏幕截图而不是硬件加速渲染图像呢，因为视频播放需要连续渲染图像，而最好的图像来源就是视频文件。但是视频文件解析是一个巨大的课题，哪怕是做一个测试例子，也至少要引入某些开源的解析库，非常麻烦。所以先讲屏幕截图，不停的抓图，得到一张张屏幕画面，进而后面章节可以讲解怎么渲染图像实现视频播放。

 

屏幕截图，依然会用到Direct，具体的说是DirectDraw。

DirectDraw是DirectX的一部分，在老的版本Direct中提供，新的版本虽然不提供开发接口，但是还是支持实现。意思就是说，老的版本的Direct SDK，比如9.0 提供了DirectDraw的头文件和lib库，新版本的DirectSDK 不再提供了，但是新版本的DirectSDK的运行库还是支持DirectDraw的。可以把老版本的头文件和lib库，放进高版本的SDK里面，依然可以正常使用。

 

DirectDraw进行截图的思路如下：

1首先需要创建DirectDraw的句柄。

2创建一个surface缓冲区，并且关联到整个屏幕（传入屏幕代表的HWND）。

3不停的从这个surface里读数据，得到就是一张RGB的位图，就是屏幕的截图。

反向操作是不是就是渲染图像呢？后面章节会讲到。

 

为了呈现截图的结果，默认将数据保存为位图，存入C盘。

 

需要头文件如下

 

#include "atlbase.h"
#include <ddraw.h>
#pragma comment(lib,"ddraw.lib")
#pragma comment(lib,"dxguid.lib")

完整代码如下：在OnInitDialog里面调用 CaptureScreenImage即可。解释都在注释里。

 

HRESULT CaptureScreenImage()
{
    ::CoInitialize(NULL); //初始化COM模块。
 
    LPDIRECTDRAW7 pDirectDraw= NULL ;
    HRESULT hr = DirectDrawCreateEx(NULL, (LPVOID*)&pDirectDraw, IID_IDirectDraw7,NULL);
    if (FAILED(hr))
       return hr;
 
    //设置显示模式，根据实际需要，强制改变当前屏幕显示的分辨率和颜色模式，如果不设置，则是默认模式。
    {
       //hr =pDirectDraw->SetDisplayMode(ScreenWidth,ScreenHeight,32,0,0);
       //if (FAILED(hr))
       //  returnhr;
    }
   
    DWORD dwScreenWidth= ::GetSystemMetrics(SM_CXSCREEN) ;
    DWORD dwScreenHeight= ::GetSystemMetrics(SM_CYSCREEN) ;
 
    // 设置协作层
    hr = pDirectDraw->SetCooperativeLevel(this->m_hWnd, DDSCL_NORMAL);
    if (FAILED(hr))
       return hr;
 
    /*
    什么是表面surface，surface就相当于是一个位图缓存
    默认当前屏幕被抽象成一个位图缓存，但是它是已经存在的。
    对于屏幕的surface，是关联，而不是创建。
    但是截图的时候我们需要创建一个后台surface。
    因为一般不直接从屏幕的缓存读数据，屏幕的更新速度较快，还没等你读完一张图，屏幕的画面已经发生了改变。
    所以过程是先一次将屏幕的数据拷贝到后台surface，做一个备份，然后慢慢从后台的surface读取。
    */
 
    // 创建主表面
    DDSURFACEDESC2    stPrimaryDdsd;
    ZeroMemory(&stPrimaryDdsd, sizeof(DDSURFACEDESC2));
    stPrimaryDdsd.dwSize = sizeof(DDSURFACEDESC2);
    stPrimaryDdsd.dwFlags = DDSD_CAPS;
    stPrimaryDdsd.ddsCaps.dwCaps= DDSCAPS_PRIMARYSURFACE; //指定关联到屏幕主surface。
 
    LPDIRECTDRAWSURFACE7  pDDSurfacePrimary= NULL ;
    hr = pDirectDraw->CreateSurface(&stPrimaryDdsd,&pDDSurfacePrimary, NULL);
    if (FAILED(hr))
       return hr;
 
    // 创建截屏缓存表面
    DDSURFACEDESC2    stBackgoundDdsd;
    ZeroMemory(&stBackgoundDdsd, sizeof(DDSURFACEDESC2));
    stBackgoundDdsd.dwSize = sizeof(DDSURFACEDESC2);
    stBackgoundDdsd.ddsCaps.dwCaps=  DDSCAPS_OFFSCREENPLAIN| DDSCAPS_VIDEOMEMORY ;
    stBackgoundDdsd.dwFlags = DDSD_WIDTH| DDSD_HEIGHT | DDSD_CAPS;
    stBackgoundDdsd.dwWidth = dwScreenWidth;
    stBackgoundDdsd.dwHeight = dwScreenHeight;
 
    LPDIRECTDRAWSURFACE7  pDDSurfaceBackground= NULL ;
    hr = pDirectDraw->CreateSurface(&stBackgoundDdsd,&pDDSurfaceBackground, NULL);
    if (FAILED(hr))
       return hr;
 
    for(int i=0; i<10; i++) //从屏幕连续抓图张
    {
       //第一步：以快速的方式从屏幕拷贝数据到后台表面
       HRESULT hr = pDDSurfaceBackground->BltFast(0,0,pDDSurfacePrimary,NULL,DDBLTFAST_NOCOLORKEY|DDBLTFAST_WAIT);
       if (FAILED(hr))
           return hr;
 
 
       //第二步：锁定缓存表面，读取里面的数据。它的像素模式依赖当前屏幕的显示模式
       hr = pDDSurfaceBackground->Lock(NULL,&stBackgoundDdsd,DDLOCK_WAIT|DDLOCK_READONLY,NULL);
       if (FAILED(hr))
           return hr;
       //第四步：将数据发送出去
       LPBYTE lpSurface = (LPBYTE)stBackgoundDdsd.lpSurface;
       if(lpSurface)
           GetPictureData(stBackgoundDdsd, i);
       //第五步：解锁表面。
       hr = pDDSurfaceBackground->Unlock(NULL);
       if (FAILED(hr))
           return hr;
    }
}
 
void savebmp(LPBYTE pdata, LPCTSTR bmp_file,int bits,int width,int height);
 
void GetPictureData(DDSURFACEDESC2& stOverlayDdsd,int idx)
{
    DWORD dwWidth= stOverlayDdsd.dwWidth;
    DWORD dwHeight= stOverlayDdsd.dwHeight;
    LPBYTE lpSurface = (LPBYTE)stOverlayDdsd.lpSurface;
 
    if(stOverlayDdsd.ddpfPixelFormat.dwFlags & DDPF_RGB)
    {//源数据是RGB数据
 
       DWORD dwRGBBitCount = stOverlayDdsd.ddpfPixelFormat.dwRGBBitCount;//RGB每一像素所占的位数
       DWORD dwRGBLineSize = dwWidth*dwRGBBitCount/8 ;
       LPBYTE pRGBData = newBYTE[dwWidth* dwHeight * dwRGBBitCount/8];
       CString fileName ;
       switch(dwRGBBitCount)
       {
       case 32: //32位RGB色
       case 24: //24位RGB色
           for(DWORD h=0; h<dwHeight; h++) //截图得到的lpSurface里面是图像数据，但它并不是整齐排列的。
               memcpy(pRGBData+h*dwRGBLineSize,lpSurface+h*stOverlayDdsd.lPitch,dwRGBLineSize);
          
           fileName.Format(_T("C:/screenshot_%d.bmp"),idx);
           savebmp(pRGBData, fileName,dwRGBBitCount, dwWidth,dwHeight);
           break ;
       case 16: //16色时，RGB分别占用位数或者，在GetScreenPicture中可以通过掩码判断结果
           break ;
       case 8: //8色时，使用当前显示驱动的调色板
           break ;
       }
    }
    else if(stOverlayDdsd.ddpfPixelFormat.dwFlags& DDPF_YUV)
    {
 
       //AfxMessageBox(_T("目前还不支持YUV格式解析、请扩展！"));
    }
}
 
void savebmp(LPBYTE pdata, LPCTSTR bmp_file,int bits,int width,int height) 
{      //分别为rgb数据，要保存的bmp文件名，图片长宽
    int size = width*height*bits/8; // 每个像素点个字节
    // 位图第一部分，文件信息
    BITMAPFILEHEADER bfh; 
    bfh.bfType= (WORD)0x4d42;  //bm 
    bfh.bfSize= size // data size 
       + sizeof( BITMAPFILEHEADER ) //first section size 
       + sizeof( BITMAPINFOHEADER ) //second section size 
       ; 
    bfh.bfReserved1= 0; // reserved 
    bfh.bfReserved2= 0; // reserved 
    bfh.bfOffBits= sizeof( BITMAPFILEHEADER)+ sizeof( BITMAPINFOHEADER);//真正的数据的位置
 
    // 位图第二部分，数据信息
    BITMAPINFOHEADER bih; 
    bih.biSize= sizeof(BITMAPINFOHEADER); 
    bih.biWidth= width; 
    bih.biHeight= -height;//BMP图片从最后一个点开始扫描，显示时图片是倒着的，所以用-height，这样图片就正了
    bih.biPlanes= 1;//为，不用改
    bih.biBitCount= bits; 
    bih.biCompression= 0;//不压缩
    bih.biSizeImage= size; 
    bih.biXPelsPerMeter= 0 ;//2835 ;//像素每米
    bih.biYPelsPerMeter= 0 ;//2835 ; 
    bih.biClrUsed= 0;//已用过的颜色，位的为
    bih.biClrImportant= 0;//每个像素都重要
    FILE * fp = _tfopen( bmp_file,_T("wb")); 
    if( !fp )
       return; 
 
    fwrite(&bfh, 8, 1,  fp );//由于linux上字节对齐，而信息头大小为字节，第一部分字节，第二部分字节，所以会将第一部分补齐为自己，直接用sizeof，打开图片时就会遇到prematureend-of-file encountered错误
    fwrite(&bfh.bfReserved2,sizeof(bfh.bfReserved2), 1, fp); 
    fwrite(&bfh.bfOffBits, sizeof(bfh.bfOffBits), 1, fp); 
    fwrite(&bih, sizeof(BITMAPINFOHEADER),1,fp); 
    fwrite(pdata,size,1,fp); 
    fclose(fp); 
}

保存位图代码，参考文档 http://blog.csdn.net/chyxwzn/article/details/8443883