FFMPEG录屏（17）--- 使用 DwmRegisterThumbnail 捕获指定窗口图像数据

使用 DwmRegisterThumbnail 捕获指定窗口图像数据

在 Windows 平台上，捕获指定窗口的图像数据可以通过多种方法实现，其中一种高效的方法是使用 [DwmRegisterThumbnail] 本文将介绍如何使用 [DwmRegisterThumbnail] 捕获窗口图像数据，并提供一个完整的示例代码。

前提条件

在开始之前，请确保您的开发环境满足以下条件：

Windows Vista 或更高版本（因为 DWM API 在 Windows Vista 中引入）。
安装了 Visual Studio 或其他支持 Windows API 开发的编译器。

步骤

1. 包含必要的头文件

首先，包含必要的头文件：

#include <dwmapi.h>
#include <windows.h>
#include <iostream>

2. 检查 DWM 是否支持

在使用 DWM API 之前，检查当前系统是否支持 DWM：

bool is_dwm_supported() {HINSTANCE dwmapi = ::LoadLibraryW(L"dwmapi.dll");if (dwmapi != nullptr) {::FreeLibrary(dwmapi);return true;}return false;
}

3. 注册缩略图

使用 DwmRegisterThumbnail 注册窗口的缩略图：

HTHUMBNAIL register_thumbnail(HWND dest_window, HWND src_window) {HTHUMBNAIL thumbnail_id = nullptr;if (FAILED(::DwmRegisterThumbnail(dest_window, src_window, &thumbnail_id))) {std::cerr << "Register thumbnail failed: " << GetLastError() << std::endl;return nullptr;}return thumbnail_id;
}

4. 更新缩略图属性

设置缩略图的属性，例如可见性、透明度等：

bool update_thumbnail_properties(HTHUMBNAIL thumbnail_id, int width, int height) {DWM_THUMBNAIL_PROPERTIES properties = {};properties.fVisible = TRUE;properties.fSourceClientAreaOnly = FALSE;properties.opacity = 180; // 255 * 0.7properties.dwFlags = DWM_TNP_VISIBLE | DWM_TNP_RECTDESTINATION | DWM_TNP_SOURCECLIENTAREAONLY;properties.rcDestination = {0, 0, width, height};if (FAILED(::DwmUpdateThumbnailProperties(thumbnail_id, &properties))) {std::cerr << "Update thumbnail properties failed: " << GetLastError() << std::endl;return false;}return true;
}

5. 捕获窗口图像数据

使用 GDI 或其他方法捕获窗口图像数据：

bool capture_window_image(HWND window, int width, int height, uint8_t** data) {// 使用 GDI 或其他方法捕获图像数据// 这里省略具体实现return true;
}

6. 完整示例

以下是一个完整的示例代码，展示了如何使用 DwmRegisterThumbnail 捕获指定窗口的图像数据：

#include <dwmapi.h>
#include <windows.h>
#include <iostream>bool is_dwm_supported() {HINSTANCE dwmapi = ::LoadLibraryW(L"dwmapi.dll");if (dwmapi != nullptr) {::FreeLibrary(dwmapi);return true;}return false;
}HTHUMBNAIL register_thumbnail(HWND dest_window, HWND src_window) {HTHUMBNAIL thumbnail_id = nullptr;if (FAILED(::DwmRegisterThumbnail(dest_window, src_window, &thumbnail_id))) {std::cerr << "Register thumbnail failed: " << GetLastError() << std::endl;return nullptr;}return thumbnail_id;
}bool update_thumbnail_properties(HTHUMBNAIL thumbnail_id, int width, int height) {DWM_THUMBNAIL_PROPERTIES properties = {};properties.fVisible = TRUE;properties.fSourceClientAreaOnly = FALSE;properties.opacity = 180; // 255 * 0.7properties.dwFlags = DWM_TNP_VISIBLE | DWM_TNP_RECTDESTINATION | DWM_TNP_SOURCECLIENTAREAONLY;properties.rcDestination = {0, 0, width, height};if (FAILED(::DwmUpdateThumbnailProperties(thumbnail_id, &properties))) {std::cerr << "Update thumbnail properties failed: " << GetLastError() << std::endl;return false;}return true;
}bool capture_window_image(HWND window, int width, int height, uint8_t** data) {// 使用 GDI 或其他方法捕获图像数据// 这里省略具体实现return true;
}int main() {if (!is_dwm_supported()) {std::cerr << "DWM is not supported on this system." << std::endl;return -1;}HWND src_window = ::FindWindow(nullptr, L"Source Window Title");HWND dest_window = ::CreateWindowEx(WS_EX_LAYERED, L"STATIC", L"Destination Window",WS_POPUP | WS_VISIBLE, 0, 0, 800, 600, nullptr, nullptr,nullptr, nullptr);if (!src_window || !dest_window) {std::cerr << "Failed to find or create window." << std::endl;return -1;}HTHUMBNAIL thumbnail_id = register_thumbnail(dest_window, src_window);if (!thumbnail_id) {return -1;}if (!update_thumbnail_properties(thumbnail_id, 800, 600)) {::DwmUnregisterThumbnail(thumbnail_id);return -1;}uint8_t* data = nullptr;if (!capture_window_image(dest_window, 800, 600, &data)) {::DwmUnregisterThumbnail(thumbnail_id);return -1;}// 处理捕获的图像数据// ...::DwmUnregisterThumbnail(thumbnail_id);return 0;
}