FFMPEG录屏（22）--- Linux 下基于X11枚举所有显示屏，并获取大小和截图等信息

众人拾柴火焰高，github给个star行不行？
open-traa/traa
traa is a versatile project aimed at recording anything, anywhere. The primary focus is to provide robust solutions for various recording scenarios, making it a highly adaptable tool for multiple use cases.

在 Linux X11 窗口系统下枚举显示屏

在 Linux 系统中，X11 是一个广泛使用的窗口系统，它提供了丰富的 API 用于管理和操作显示屏。在这篇博客中，我们将介绍如何使用 X11 枚举系统中的显示屏，并获取显示屏的 ID、坐标和截图等信息。

前置条件

在开始之前，请确保你已经安装了 X11 相关的开发库，例如 libX11 和 libXrandr。你可以使用以下命令进行安装：

sudo apt-get install libx11-dev libxrandr-dev

额外的说明

现在的Ubuntu系统，默认是Wayland的，所以一些X11功能可能会报错，尤其是 获取屏幕截图, 作为LInux新手…我搞了好久，切换X11环境后，一次就成了，可以在Login界面选择
在这里插入图片描述

枚举显示屏的基本步骤

我们将通过以下几个步骤来实现枚举显示屏的功能：

打开 X11 显示连接。
检查 Xrandr 扩展是否可用。
获取根窗口。
获取显示屏信息。
获取显示屏的截图。

代码实现

以下是一个完整的代码示例，展示了如何实现上述功能：

#include <X11/Xlib.h>
#include <X11/extensions/Xrandr.h>
#include <iostream>
#include <vector>
#include <string>
#include <memory>
#include <cstring>// 获取显示屏的截图
bool get_screen_image_data(Display *display, ::Window root, const XRRMonitorInfo &monitor, const int thumbnail_width, const int thumbnail_height, std::vector<uint8_t> &data) {XImage *image = XGetImage(display, root, monitor.x, monitor.y, monitor.width, monitor.height, AllPlanes, ZPixmap);if (!image) {std::cerr << "Failed to get image for screen" << std::endl;return false;}int scaled_width = thumbnail_width;int scaled_height = thumbnail_height;data.resize(scaled_width * scaled_height * 4);// 使用 libyuv 进行图像缩放libyuv::ARGBScale(reinterpret_cast<uint8_t *>(image->data), image->bytes_per_line, image->width, image->height, data.data(), scaled_width * 4, scaled_width, scaled_height, libyuv::kFilterBox);XDestroyImage(image);return true;
}// 枚举显示屏
void enum_screens(int thumbnail_width, int thumbnail_height) {Display *display = XOpenDisplay(NULL);if (!display) {std::cerr << "Failed to open display" << std::endl;return;}// 检查 Xrandr 扩展是否可用int event_base, error_base;if (!XRRQueryExtension(display, &event_base, &error_base)) {std::cerr << "Xrandr extension is not available" << std::endl;XCloseDisplay(display);return;}// 获取根窗口::Window root = DefaultRootWindow(display);// 获取显示屏信息int monitor_count = 0;XRRMonitorInfo *monitors = XRRGetMonitors(display, root, True, &monitor_count);if (!monitors) {std::cerr << "Failed to get monitors" << std::endl;XCloseDisplay(display);return;}for (int i = 0; i < monitor_count; ++i) {std::cout << "Monitor ID: " << i << std::endl;std::cout << "Position: (" << monitors[i].x << ", " << monitors[i].y << ")" << std::endl;std::cout << "Size: " << monitors[i].width << "x" << monitors[i].height << std::endl;// 获取显示屏的截图std::vector<uint8_t> image_data;if (get_screen_image_data(display, root, monitors[i], thumbnail_width, thumbnail_height, image_data)) {std::cout << "Captured screen image" << std::endl;}}XRRFreeMonitors(monitors);XCloseDisplay(display);
}int main() {int thumbnail_width = 200;  // 缩略图宽度int thumbnail_height = 150; // 缩略图高度enum_screens(thumbnail_width, thumbnail_height);return 0;
}

代码说明

获取显示屏的截图：使用

XGetImage

函数获取显示屏的截图数据，并使用 libyuv 进行图像缩放。
2. 枚举显示屏：使用

XRRGetMonitors

函数获取显示屏信息，并遍历每个显示屏获取其详细信息。

总结

通过上述步骤，我们可以在 Linux X11 窗口系统下枚举系统中的显示屏，并获取显示屏的 ID、坐标和截图等信息。这些功能可以用于开发桌面管理工具、屏幕录制软件等应用。希望这篇博客对你有所帮助！

更多细节请前往 TRAA

额外的实现

真实的获取图片的代码是用了一个额外的辅助类，参考了webrtc的代码，本系列会大量借鉴webrtc代码，比较大树底下好乘凉？XServerPixelBuffer

这个东西，有一个坑点，搞了许久，也不知道为什么，望大佬赐教，我们是利用Xrandr插件来获取每个显示屏的坐标的，一旦你先调用了XRRGetMonitors枚举函数，XServerPixelBuffer::Init中的XGetWindowAttributes必然失败，错误是什么BadMatch。。。你懂的，X11的错误我至今不知道应该如何调查，所以一定要在XRRGetMonitors之前把pixelBuffer初始化掉
如下

  std::unique_ptr<basic_desktop_frame> full_screen_frame;if (thumbnail_size.width > 0 && thumbnail_size.height > 0) {// must init the pixel buffer before calling XRRGetMonitors, ohterwise init will raise an// BadMatch error.I don't know why, just do it.x_server_pixel_buffer pixel_buffer;if (!pixel_buffer.init(&atom_cache, root)) {LOG_ERROR("failed to init pixel buffer for window {}", root);return false;}// capture the whole screenfull_screen_frame = std::make_unique<basic_desktop_frame>(pixel_buffer.window_size());pixel_buffer.synchronize();if (!pixel_buffer.capture_rect(desktop_rect::make_size(full_screen_frame->size()),full_screen_frame.get())) {LOG_ERROR("failed to capture rect for screen {}", root);return false;}#if 0// create a file to save the full screensave_pixel_to_ppm("full_screen.ppm", full_screen_frame->data(), full_screen_frame->size().width(),full_screen_frame->size().height());
#endif}int monitor_count = 0;XRRMonitorInfo *monitors = XRRGetMonitors(display, root, True, &monitor_count);if (!monitors) {LOG_ERROR("failed to get monitors");XCloseDisplay(display);return traa_error::TRAA_ERROR_UNKNOWN;}