首页 > > 程序设计 > C/C++ >

C++抓图服务

2020-03-31 16:02:46来源：博客园 阅读 ()

新老客户大回馈,云服务器低至5折

C++抓图服务

基于前两篇抓图文章，本文将抓图提取为一个服务，实现不同场景下抓图需求的封装。

C++使用BitBlt进行窗口抓图

C++使用PrintWindow进行窗口抓图

首先是抓图服务：

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_;
};

CaptureService.cpp

#include "stdafx.h"
#include "CaptureService.h"
#include "DibCaptureHelper.h"
#include "PrintCaptureHelper.h"


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

CaptureService& CaptureService::GetInstance()
{
    static CaptureService instance;
    return instance;
}

bool CaptureService::RegisterCapture(string name, string windowName, CaptureType type /* = CreateDibSection */)
{
    const auto hwnd = ::FindWindowA(nullptr, windowName.c_str());
    return RegisterCapture(name, hwnd, type);
}

bool CaptureService::RegisterCapture(string name, HWND hwnd, CaptureType type /* = CreateDibSection */)
{
    if (name.empty() || captureHelpers_.find(name) != captureHelpers_.end())
    {
        return false;
    }

    ICaptureHelper* helper;
    switch (type)
    {
    case CreateDibSection:
        helper = new DibCaptureHelper();
        break;
    case PrintWindow:
        helper = new PrintCaptureHelper();
        break;
    default:
        return false;
    }

    if (helper == nullptr)
    {
        return false;
    }

    if (!helper->Init(hwnd))
    {
        delete helper;
        return false;
    }

    captureHelpers_[name] = helper;
    return true;
}

void CaptureService::UnRegisterCapture(string name)
{
    if (name.empty() || captureHelpers_.find(name) == captureHelpers_.end())
    {
        return;
    }

    auto* captureHelper = captureHelpers_[name];
    if (captureHelper != nullptr)
    {
        captureHelper->Cleanup();
        delete captureHelper;
    }

    captureHelpers_.erase(name);
}

bool CaptureService::IsRegister(string name)
{
    return !name.empty() && captureHelpers_.find(name) != captureHelpers_.end();
}

bool CaptureService::RefreshWindow(string name)
{
    if (!IsRegister(name))
    {
        return false;
    }
    return captureHelpers_[name]->RefreshWindow();
}

bool CaptureService::ChangeWindowHandle(string name, string windowName)
{
    if (!IsRegister(name))
    {
        return false;
    }
    return captureHelpers_[name]->ChangeWindowHandle(windowName);
}

bool CaptureService::ChangeWindowHandle(string name, HWND hwnd)
{
    if (!IsRegister(name))
    {
        return false;
    }
    return captureHelpers_[name]->ChangeWindowHandle(hwnd);
}

bool CaptureService::Capture(string name)
{
    if (!IsRegister(name))
    {
        return false;
    }
    return captureHelpers_[name]->Capture();
}

bool CaptureService::GetWindowRect(string name, RECT& winRect)
{
    if (!IsRegister(name))
    {
        return false;
    }
    winRect = captureHelpers_[name]->GetWindowRect();
    return true;
}

bool CaptureService::GetClientRect(string name, RECT& clientRect)
{
    if (!IsRegister(name))
    {
        return false;
    }
    clientRect = captureHelpers_[name]->GetClientRect();
    return true;
}

bool CaptureService::GetBitmapDataSize(string name, int& bmpDataSize)
{
    if (!IsRegister(name))
    {
        return false;
    }
    bmpDataSize = captureHelpers_[name]->GetBitmapDataSize();
    return true;
}

bool CaptureService::GetBitmap(string name, HBITMAP& bitmap)
{
    if (!IsRegister(name))
    {
        return false;
    }
    bitmap = captureHelpers_[name]->GetBitmap();
    return true;
}

bool CaptureService::GetBitmapAddress(string name, void** bitsPtr)
{
    if (!IsRegister(name))
    {
        return false;
    }
    *bitsPtr = captureHelpers_[name]->GetBitmapAddress();
    return true;
}

void CaptureService::Cleanup()
{
    for (auto iter = captureHelpers_.cbegin(); iter != captureHelpers_.cend(); ++iter)
    {
        auto* captureHelper = iter->second;
        if (captureHelper != nullptr)
        {
            captureHelper->Cleanup();
            delete captureHelper;
        }
    }
    captureHelpers_.clear();
}

其次是抓图代码封装：

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;
}

 


原文链接:https://www.cnblogs.com/xhubobo/p/12604983.html
如有疑问请与原作者联系

标签：

版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点，本站所提供的摄影照片，插画，设计作品，如需使用，请与原作者联系，版权归原作者所有

上一篇：【题解】Building Strings Gym - 102152E

下一篇：Window中的shellcode编写框架（入门篇）

相关文章

IDC资讯： 主机资讯 注册资讯 托管资讯 vps资讯 网站建设

网站运营： 建站经验 策划盈利 搜索优化 网站推广 免费资源

网站联盟： 联盟新闻 联盟介绍 联盟点评 网赚技巧

行业资讯： 搜索引擎 网络游戏 电子商务 广告传媒

网络编程： Asp.Net编程 Asp编程 Php编程 Xml编程 Access Mssql Mysql 其它

服务器技术： Web服务器 Ftp服务器 Mail服务器 Dns服务器 安全防护

软件技巧： 其它软件 Word Excel Powerpoint Ghost Vista QQ空间 QQ FlashGet 迅雷

网页制作： FrontPages Dreamweaver Javascript css photoshop fireworks Flash

程序设计： Java技术 C/C++ VB delphi

网络知识： 网络协议 网络安全 网络管理 组网方案 Cisco技术

操作系统： Win2000 WinXP Win2003 Mac OS Linux FreeBSD

热门词条
最新资讯
热门关注
热门标签