众人拾柴火焰高,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 screen
full_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 screen
save_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;
}