在Windows开发有时需要实现截屏、桌面推送功能,网上搜的方案大多为ffmpeg实现。如果这个功能要放在sdk则引入的ffmpeg会导致模块体积太大。我参考ffmepg截屏部分的源码,在Windows上用的是gdi bitblt截屏,这完全可以自己用gdi实现一个截屏模块。最终实现如下,整个功能模块只依赖于<Windows.h>头文件,效果与ffmpeg基本一致,如果不是追求极致性能(mirro driver、duplication)的截屏,这种实现还是有价值的。下面是头文件的定义:
#pragma once
#include"IVideoCapture.h"
/************************************************************************
* @Project: GdiScreenCapture
* @Decription: Gdi屏幕采集工具
* @Verision: v1.0.0.1
* @Author: Xin
* @Create: 2020/3/12 16:26:00
* @LastUpdate: 2020/4/03 15:24:00
************************************************************************
* Copyright @ 2020. All rights reserved.
************************************************************************/
namespace Common{
/// <summary>
/// Gdi屏幕采集对象
/// </summary>
class GdiScreenCapture:IVideoCapture
{
public:
/// <summary>
/// 采集区域的模式
/// </summary>
enum RectMode {
/// <summary>
/// 按实际计算
/// </summary>
TrueValue,
/// <summary>
/// 按比例计算
/// </summary>
RatioValue
};
public:
/// <summary>
/// 构造函数
/// </summary>
/// <param name="callback">[in]回调接口对象</param>
/// <param name="hwnd">[in]窗口句柄,为NULL时采集屏幕</param>
/// <param name="format">[in]输出的像素格式</param>
GdiScreenCapture(void* hwnd, PixelFormat format);
/// <summary>
/// 析构函数
/// </summary>
~GdiScreenCapture();
// 通过 IVideoCapture 继承
virtual void SetCallback(Common::IVideoCaptureCallback *) override;
/// <summary>
/// 设置采集区域,可动态设置 -线程安全
/// </summary>
/// <param name="x">[in]x坐标</param>
/// <param name="y">[in]x坐标</param>
/// <param name="width">[in]宽</param>
/// <param name="height">[in]高</param>
/// <param name="rectMode">[in]采集区域计算方式。TrueValue为实际值。RatioValue为比例值范围0-1,全屏设为0,0,1,1。</param>
void SetRect(float x, float y, float width, float height, RectMode rectMode);
/// <summary>
/// 设置采集帧率,可动态设置 -线程安全
/// </summary>
/// <param name="framerate">采集帧率fps</param>
void SetFramerate(float framerate);
/// <summary>
/// 开启采集
/// </summary>
void Start();
/// <summary>
/// 停止采集
/// </summary>
void Stop();
/// <summary>
/// 快照图片缓存
/// </summary>
/// <param name="hwnd">[in]窗口句柄,为NULL时采集屏幕</param>
/// <param name="x">[in]x坐标</param>
/// <param name="y">[in]y坐标</param>
/// <param name="width">[in]宽</param>
/// <param name="height">[in]高</param>
/// <param name="format">[in]输出的像素格式</param>
/// <param name="buf">[in]图片缓存,由调用者初始化</param>
/// <param name="len">[in]图片缓存的长度,由调用者提供</param>
/// <returns>返回图像数据的长度</returns>
static int Snapshot(void* hwnd, int x, int y, int width, int height, PixelFormat format, unsigned char* buf, size_t len);
/// <summary>
/// 快照保存文件,bmp格式
/// </summary>
/// <param name="hwnd">[in]窗口句柄,为NULL时截取屏幕</param>
/// <param name="x">[in]x坐标</param>
/// <param name="y">[in]y坐标</param>
/// <param name="width">[in]宽</param>
/// <param name="height">[in]高</param>
/// <param name="filename">[in]保存的文件名</param>
static void Snapshot(void* hwnd, int x, int y, int width, int height, const char* filename);
private:
void*_implement;
};
}
采用pimp的设计头文件依赖较少,功能上支持:设置截屏的窗口句柄,设置截屏帧率,设置输出格式(bgr24、yu12、nv12等)。具体项目如下: