脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → C 语言 → c++ 截图

c++ 封装一个截图服务

 更新时间：2021年01月07日 17:33:03   作者：xhubobo  
这篇文章主要介绍了c++ 封装一个截图服务的方法，帮助大家更好的理解和使用c++,感兴趣的朋友可以了解下

首先是抓图服务：

ICaptureHelper.h 

#pragma once

#include <windows.h>
#include <string>
using std::string;

class ICaptureHelper
{
public:
  virtual ~ICaptureHelper() {}
  virtual bool Init(const string& windowName) = 0;
  virtual bool Init(HWND hwnd) = 0;
  virtual void Cleanup() = 0;
  virtual bool RefreshWindow() = 0;
  virtual bool ChangeWindowHandle(const string& windowName) = 0;
  virtual bool ChangeWindowHandle(HWND hwnd) = 0;
  virtual bool Capture() = 0;

  virtual const RECT& GetWindowRect() const = 0;
  virtual const RECT& GetClientRect() const = 0;
  virtual int GetBitmapDataSize() const = 0;
  virtual HBITMAP GetBitmap() const = 0;
  virtual void* GetBitmapAddress() const = 0;
};

CaptureService.h

#pragma once

#include "ICaptureHelper.h"
#include <map>
using std::map;

class CaptureService
{
public:
  CaptureService() = default;
  static CaptureService& GetInstance();

  enum CaptureType
  {
    //使用CreateDIBSection抓图，速度快，但是无法抓取D3D等渲染的窗口
    CreateDibSection = 0,

    //使用PrintWindow抓图，速度慢(16ms左右)，但是可以抓取D3D等渲染的窗口
    PrintWindow
  };

  bool RegisterCapture(string name, string windowName, CaptureType type = CreateDibSection); //注册抓图服务
  bool RegisterCapture(string name, HWND hwnd, CaptureType type = CreateDibSection); //注册抓图服务
  void UnRegisterCapture(string name); //注销抓图服务
  bool IsRegister(string name); //获取是否已注册抓图服务

  bool RefreshWindow(string name); //刷新窗口
  bool ChangeWindowHandle(string name, string windowName); //修改窗口句柄
  bool ChangeWindowHandle(string name, HWND hwnd); //修改窗口句柄
  bool Capture(string name); //抓图

  bool GetWindowRect(string name, RECT& winRect); //获取窗口尺寸
  bool GetClientRect(string name, RECT& clientRect); //获取窗口客户区尺寸
  bool GetBitmapDataSize(string name, int& bmpDataSize); //获取抓图数据大小
  bool GetBitmap(string name, HBITMAP& bitmap); //获取窗口位图
  bool GetBitmapAddress(string name, void** bitsPtr); //获取窗口位图地址

  void Cleanup(); //清理所有抓图服务

private:
  ~CaptureService();

private:
  map<string, ICaptureHelper*> captureHelpers_;
};

其次是抓图代码封装：

AbsCaptureHelper.h

#pragma once

#include "ICaptureHelper.h"

class AbsCaptureHelper : public ICaptureHelper
{
public:
  AbsCaptureHelper();
  virtual ~AbsCaptureHelper();

  bool Init(const string& windowName) override;
  bool Init(HWND hwnd) override;
  void Cleanup() override;
  bool RefreshWindow() override;
  bool ChangeWindowHandle(const string& windowName) override;
  bool ChangeWindowHandle(HWND hwnd) override;
  bool Capture() override;

  const RECT& GetWindowRect() const override { return windowRect_; }
  const RECT& GetClientRect() const override { return clientRect_; }
  int GetBitmapDataSize() const override { return bmpDataSize_; }
  HBITMAP GetBitmap() const override { return bitmap_; }
  void* GetBitmapAddress() const override { return bitsPtr_; }

protected:
  virtual bool InitDC(const BITMAPINFO& bitmapInfo) = 0;
  virtual bool DoCapture() = 0;

protected:
  HWND hwnd_;
  HDC scrDc_;
  HDC memDc_;
  HBITMAP bitmap_;
  HBITMAP oldBitmap_;
  void* bitsPtr_;

  RECT windowRect_;
  RECT clientRect_;
  int bmpDataSize_;
};

AbsCaptureHelper.cpp

#include "stdafx.h"
#include "AbsCaptureHelper.h"


AbsCaptureHelper::AbsCaptureHelper()
  : hwnd_(nullptr)
  , scrDc_(nullptr)
  , memDc_(nullptr)
  , bitmap_(nullptr)
  , oldBitmap_(nullptr)
  , bitsPtr_(nullptr)
  , windowRect_{ 0, 0, 0, 0 }
  , clientRect_{ 0, 0, 0, 0 }
  , bmpDataSize_(0)
{
}

AbsCaptureHelper::~AbsCaptureHelper()
{
  AbsCaptureHelper::Cleanup();
}

bool AbsCaptureHelper::Init(const string& windowName)
{
  const auto handle = ::FindWindowA(nullptr, windowName.c_str());
  if (handle == nullptr)
  {
    return false;
  }

  return Init(handle);
}

bool AbsCaptureHelper::Init(HWND hwnd)
{
  hwnd_ = hwnd;

  //获取窗口大小
  if (!::GetWindowRect(hwnd_, &windowRect_) || !::GetClientRect(hwnd_, &clientRect_))
  {
    return false;
  }

  const auto clientRectWidth = clientRect_.right - clientRect_.left;
  const auto clientRectHeight = clientRect_.bottom - clientRect_.top;
  bmpDataSize_ = clientRectWidth * clientRectHeight * 4;

  //位图信息
  BITMAPINFO bitmapInfo;
  bitmapInfo.bmiHeader.biSize = sizeof(bitmapInfo);
  bitmapInfo.bmiHeader.biWidth = clientRectWidth;
  bitmapInfo.bmiHeader.biHeight = clientRectHeight;
  bitmapInfo.bmiHeader.biPlanes = 1;
  bitmapInfo.bmiHeader.biBitCount = 32;
  bitmapInfo.bmiHeader.biSizeImage = clientRectWidth * clientRectHeight;
  bitmapInfo.bmiHeader.biCompression = BI_RGB;

  return InitDC(bitmapInfo);
}

void AbsCaptureHelper::Cleanup()
{
  if (bitmap_ == nullptr)
  {
    return;
  }

  //删除用过的对象
  ::SelectObject(memDc_, oldBitmap_);
  ::DeleteObject(bitmap_);
  ::DeleteDC(memDc_);
  ::ReleaseDC(hwnd_, scrDc_);

  hwnd_ = nullptr;
  scrDc_ = nullptr;
  memDc_ = nullptr;
  bitmap_ = nullptr;
  oldBitmap_ = nullptr;
  bitsPtr_ = nullptr;
}

bool AbsCaptureHelper::RefreshWindow()
{
  const auto hwnd = hwnd_;
  Cleanup();
  return Init(hwnd);
}

bool AbsCaptureHelper::ChangeWindowHandle(const string& windowName)
{
  Cleanup();
  return Init(windowName);
}

bool AbsCaptureHelper::ChangeWindowHandle(HWND hwnd)
{
  Cleanup();
  return Init(hwnd);
}

bool AbsCaptureHelper::Capture()
{
  if (bitmap_ == nullptr || memDc_ == nullptr || scrDc_ == nullptr)
  {
    return false;
  }

  return DoCapture();
}

DibCaptureHelper.h

#pragma once

#include "AbsCaptureHelper.h"

class DibCaptureHelper : public AbsCaptureHelper
{
public:
  DibCaptureHelper();
  virtual ~DibCaptureHelper();

protected:
  bool InitDC(const BITMAPINFO& bitmapInfo) override;
  bool DoCapture() override;

private:
  bool saveBitmap_;
  int mockPageNumber;
  int bmpCount_;
};

DibCaptureHelper.cpp

#include "stdafx.h"
#include "DibCaptureHelper.h"
#include <sstream>

static int BmpCount = 0;
static int BmpMaxCount = 50;

DibCaptureHelper::DibCaptureHelper()
  : saveBitmap_(false)
  , mockPageNumber(++BmpCount)
  , bmpCount_(0)
{
}

DibCaptureHelper::~DibCaptureHelper()
{
}

bool DibCaptureHelper::InitDC(const BITMAPINFO& bitmapInfo)
{
  scrDc_ = ::GetWindowDC(hwnd_);
  memDc_ = ::CreateCompatibleDC(scrDc_);

  bitmap_ = ::CreateDIBSection(memDc_, &bitmapInfo, DIB_RGB_COLORS, &bitsPtr_, nullptr, 0);
  if (bitmap_ == nullptr)
  {
    ::DeleteDC(memDc_);
    ::ReleaseDC(hwnd_, scrDc_);
    return false;
  }

  oldBitmap_ = static_cast<HBITMAP>(::SelectObject(memDc_, bitmap_));
  return true;
}

bool DibCaptureHelper::DoCapture()
{
  const auto clientRectWidth = clientRect_.right - clientRect_.left;
  const auto clientRectHeight = clientRect_.bottom - clientRect_.top;

  const auto ret = ::BitBlt(
    memDc_, 0, 0, clientRectWidth, clientRectHeight,
    scrDc_, 0, 0, SRCCOPY);

  return ret != 0;
}

PrintCaptureHelper.h

#pragma once

#include "AbsCaptureHelper.h"

class PrintCaptureHelper : public AbsCaptureHelper
{
public:
  PrintCaptureHelper();
  virtual ~PrintCaptureHelper();

protected:
  bool InitDC(const BITMAPINFO& bitmapInfo) override;
  bool DoCapture() override;
};

PrintCaptureHelper.cpp

#include "stdafx.h"
#include "PrintCaptureHelper.h"


PrintCaptureHelper::PrintCaptureHelper()
{
}

PrintCaptureHelper::~PrintCaptureHelper()
{
}

bool PrintCaptureHelper::InitDC(const BITMAPINFO& bitmapInfo)
{
  scrDc_ = ::GetWindowDC(hwnd_);
  memDc_ = ::CreateCompatibleDC(scrDc_);

  bitmap_ = ::CreateDIBSection(scrDc_, &bitmapInfo, DIB_RGB_COLORS, &bitsPtr_, nullptr, 0);
  if (bitmap_ == nullptr)
  {
    ::DeleteDC(memDc_);
    ::ReleaseDC(hwnd_, scrDc_);
    return false;
  }
  
  oldBitmap_ = static_cast<HBITMAP>(::SelectObject(memDc_, bitmap_));
  return true;
}

bool PrintCaptureHelper::DoCapture()
{
  const auto ret = ::PrintWindow(hwnd_, memDc_, PW_CLIENTONLY | PW_RENDERFULLCONTENT);
  return ret != 0;
}

以上就是c++ 封装一个截图服务的详细内容，更多关于c++ 截图的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • C语言实现黎曼和求定积分

    C语言实现黎曼和求定积分

    这篇文章主要为大家详细介绍了用C语言程序实现黎曼和求定积分，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2020-02-02
  • c++中cin/cout与scanf/printf的区别比较

    c++中cin/cout与scanf/printf的区别比较

    这篇文章主要介绍了c++中cin/cout与scanf/printf的区别比较,需要的朋友可以参考下
    2017-06-06
  • QT网络编程UDP下C/S架构广播通信(实例讲解)

    QT网络编程UDP下C/S架构广播通信(实例讲解)

    下面小编就为大家带来一篇QT网络编程UDP下C/S架构广播通信(实例讲解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-07-07
  • C++ 基类指针和子类指针相互赋值的实现方法

    C++ 基类指针和子类指针相互赋值的实现方法

    下面小编就为大家带来一篇C++ 基类指针和子类指针相互赋值的实现方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2016-12-12
  • C++实现屏幕截图（全屏截图）

    C++实现屏幕截图（全屏截图）

    屏幕截图已经成为了所有IM即时通讯软件的必备模块，也是日常办公中使用最频繁的功能之一。今天我们从C++开发的角度，来看看屏幕截图的主要功能点是如何实现的，感兴趣的可以了解一下
    2021-11-11
  • C++/C 回文字符串的实例详解

    C++/C 回文字符串的实例详解

    这篇文章主要介绍了C++ 回文字符串的实例详解的相关资料,需要的朋友可以参考下
    2017-07-07
  • C++实现数组的排序/插入重新排序/以及逆置操作详解

    C++实现数组的排序/插入重新排序/以及逆置操作详解

    将新的数字与已经排序好的数组中的数字一一比较，直到找到插入点，然后将插入点以后的数字都向后移动一个单位（a[i+1]=a[i]），然后将数据插入即可
    2013-10-10
  • C++使用waveIn实现声音采集

    C++使用waveIn实现声音采集

    在Windows上实现录音比较简单的方法是使用winmm,其中的waveIn模块就可以打开录音设备,这篇文章主要为大家介绍了C++如何使用waveIn实现声音采集,需要的可以了解下
    2023-10-10
  • c语言 字符串的拼接和分割实例

    c语言 字符串的拼接和分割实例

    今天小编就为大家分享一篇c语言 字符串的拼接和分割实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-12-12
  • VS2019项目打包生成.exe文件与Setup的步骤实现

    VS2019项目打包生成.exe文件与Setup的步骤实现

    这篇文章主要介绍了VS2019项目打包生成.exe文件与Setup的步骤实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-03-03

最新评论