关于ubuntu 22.0采用pipewire + dbus的屏幕录制

最新推荐文章于 2024-06-24 10:20:13 发布

砖农L

最新推荐文章于 2024-06-24 10:20:13 发布

阅读量759

点赞数

分类专栏：音视频文章标签： ubuntu linux c语言

本文链接：https://blog.csdn.net/liusarzn/article/details/128849603

版权

本文介绍了在Ubuntu 22.0系统上，结合pipewire 0.3和dbus进行屏幕录制的实现过程。首先，通过apt安装libpipewire-0.3-dev和libglib2.0-dev，然后编译pipewire源代码。接着，使用glib封装的dbus和libsdl2-dev进行画面渲染。确保系统中存在dbus总线和运行的pipewire进程，最后展示了一个简单的demo代码片段和运行效果。

摘要由CSDN通过智能技术生成

最近因为项目需要，研究了一下在wayland上的屏幕录制功能，开发机如图
在这里插入图片描述

所需环境：
pipewire0.3，ubuntu可采用apt命令安装：sudo apt install libpipewire-0.3-dev
或者直接编译pipewire源代码，代码git地址：pipewire gitlab
dbus采用glib封装版本，glib版本2.0，安装命令：sudo apt install libglib2.0-dev
demo中画面渲染采用sdl2，安装命令：sudo apt install libsdl2-dev

此外需要确认系统中有相关的dbus数据总线，查看方法可以通过d-feet工具，ubuntu software中可以找到，也可以通过命令安装：sudo apt install d-feet
在d-feet的Session Bus中可以找到如图所示的dbus总线
在这里插入图片描述
还要顺便检查下在系统中是否存在pipewire进程，只有在pipewire进程正常运行的情况下，demo才能正常工作，可通过ps -e | grep pipewire命令查看

至此准备工作结束，一下是demo代码：

sdl.h


#include <SDL2/SDL.h>

#include <spa/utils/type.h>
#include <spa/pod/builder.h>
#include <spa/param/video/raw.h>
#include <spa/param/video/format.h>

static struct {
   
	Uint32 format;
	uint32_t id;
} sdl_video_formats[] = {
   
#if SDL_BYTEORDER == SDL_BIG_ENDIAN
	{
    SDL_PIXELFORMAT_UNKNOWN, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_UNKNOWN, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1MSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX4LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX4MSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX8, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB332, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGR555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ARGB4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ABGR4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGRA4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ARGB1555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA5551, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ABGR1555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGRA5551, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB565, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGR565, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB24, SPA_VIDEO_FORMAT_RGB,},
	{
    SDL_PIXELFORMAT_RGB888, SPA_VIDEO_FORMAT_RGB,},
	{
    SDL_PIXELFORMAT_RGBX8888, SPA_VIDEO_FORMAT_RGBx,},
	{
    SDL_PIXELFORMAT_BGR24, SPA_VIDEO_FORMAT_BGR,},
	{
    SDL_PIXELFORMAT_BGR888, SPA_VIDEO_FORMAT_BGR,},
	{
    SDL_PIXELFORMAT_BGRX8888, SPA_VIDEO_FORMAT_BGRx,},
	{
    SDL_PIXELFORMAT_ARGB2101010, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA8888, SPA_VIDEO_FORMAT_RGBA,},
	{
    SDL_PIXELFORMAT_ARGB8888, SPA_VIDEO_FORMAT_ARGB,},
	{
    SDL_PIXELFORMAT_BGRA8888, SPA_VIDEO_FORMAT_BGRA,},
	{
    SDL_PIXELFORMAT_ABGR8888, SPA_VIDEO_FORMAT_ABGR,},
	{
    SDL_PIXELFORMAT_YV12, SPA_VIDEO_FORMAT_YV12,},
	{
    SDL_PIXELFORMAT_IYUV, SPA_VIDEO_FORMAT_I420,},
	{
    SDL_PIXELFORMAT_YUY2, SPA_VIDEO_FORMAT_YUY2,},
	{
    SDL_PIXELFORMAT_UYVY, SPA_VIDEO_FORMAT_UYVY,},
	{
    SDL_PIXELFORMAT_YVYU, SPA_VIDEO_FORMAT_YVYU,},
#if SDL_VERSION_ATLEAST(2,0,4)
	{
    SDL_PIXELFORMAT_NV12, SPA_VIDEO_FORMAT_NV12,},
	{
    SDL_PIXELFORMAT_NV21, SPA_VIDEO_FORMAT_NV21,},
#endif
#else
	{
    SDL_PIXELFORMAT_UNKNOWN, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_UNKNOWN, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX1MSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX4LSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX4MSB, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_INDEX8, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB332, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGR555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ARGB4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ABGR4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGRA4444, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ARGB1555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA5551, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_ABGR1555, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGRA5551, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB565, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_BGR565, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGB24, SPA_VIDEO_FORMAT_BGR,},
	{
    SDL_PIXELFORMAT_RGB888, SPA_VIDEO_FORMAT_BGR,},
	{
    SDL_PIXELFORMAT_RGBX8888, SPA_VIDEO_FORMAT_xBGR,},
	{
    SDL_PIXELFORMAT_BGR24, SPA_VIDEO_FORMAT_RGB,},
	{
    SDL_PIXELFORMAT_BGR888, SPA_VIDEO_FORMAT_RGB,},
	{
    SDL_PIXELFORMAT_BGRX8888, SPA_VIDEO_FORMAT_xRGB,},
	{
    SDL_PIXELFORMAT_XRGB8888, SPA_VIDEO_FORMAT_BGRx,},
	{
    SDL_PIXELFORMAT_ARGB2101010, SPA_VIDEO_FORMAT_UNKNOWN,},
	{
    SDL_PIXELFORMAT_RGBA8888, SPA_VIDEO_FORMAT_ABGR,},
	{
    SDL_PIXELFORMAT_ARGB8888, SPA_VIDEO_FORMAT_BGRA,},
	{
    SDL_PIXELFORMAT_BGRA8888, SPA_VIDEO_FORMAT_ARGB,},
	{
    SDL_PIXELFORMAT_ABGR8888, SPA_VIDEO_FORMAT_RGBA,},
	{
    SDL_PIXELFORMAT_YV12, SPA_VIDEO_FORMAT_YV12,},
	{
    SDL_PIXELFORMAT_IYUV, SPA_VIDEO_FORMAT_I420,},
	{
    SDL_PIXELFORMAT_YUY2, SPA_VIDEO_FORMAT_YUY2,},
	{
    SDL_PIXELFORMAT_UYVY, SPA_VIDEO_FORMAT_UYVY,},
	{
    SDL_PIXELFORMAT_YVYU, SPA_VIDEO_FORMAT_YVYU,},
#if SDL_VERSION_ATLEAST(2,0,4)
	{
    SDL_PIXELFORMAT_NV12, SPA_VIDEO_FORMAT_NV12,},
	{
    SDL_PIXELFORMAT_NV21, SPA_VIDEO_FORMAT_NV21,},
#endif
#endif
};

#define SPA_FOR_EACH_ELEMENT_VAR(arr, var) \
	for(__typeof__((arr)[0])* (var) = arr; (void*)(var) < SPA_PTROFF(arr, sizeof(arr), void); (var)++)

static inline uint32_t sdl_format_to_id(Uint32 format)
{
   
	SPA_FOR_EACH_ELEMENT_VAR(sdl_video_formats, f) {
   
		if (f->format == format)
			return f->id;
	}
	return SPA_VIDEO_FORMAT_UNKNOWN;
}

static inline Uint32 id_to_sdl_format(uint32_t id)
{
   
	SPA_FOR_EACH_ELEMENT_VAR(sdl_video_formats, f) {
   
		if (f->id == id)
			return f->format;
	}
	return SDL_PIXELFORMAT_UNKNOWN;
}

static inline struct spa_pod *sdl_build_formats(SDL_RendererInfo *info, struct spa_pod_builder *b)
{
   
	uint32_t i, c;
	struct spa_pod_frame f[2];

	/* make an object of type SPA_TYPE_OBJECT_Format and id SPA_PARAM_EnumFormat.
	 * The object type is important because it defines the properties that are
	 * acceptable. The id gives more context about what the object is meant to
	 * contain. In this case we enumerate supported formats. */
	spa_pod_builder_push_object(b, &f[0], SPA_TYPE_OBJECT_Format, SPA_PARAM_EnumFormat);
	/* add media type and media subtype properties */
	spa_pod_builder_prop(b, SPA_FORMAT_mediaType, 0);
	spa_pod_builder_id(b, SPA_MEDIA_TYPE_video);
	spa_pod_builder_prop(b, SPA_FORMAT_mediaSubtype, 0);
	spa_pod_builder_id(b, SPA_MEDIA_SUBTYPE_raw);

	/* build an enumeration of formats */
	spa_pod_builder_prop(b, SPA_FORMAT_VIDEO_format, 0);
	spa_pod_builder_push_choice(b, &f[1], SPA_CHOICE_Enum, 0);
	/* first the formats supported by the textures */
	for (i = 0, c = 0; i < info->num_texture_formats; i++) {
   
		uint32_t id = sdl_format_to_id(info->texture_formats[i]);
		if (id == 0)
			continue;
		if (c++ == 0)
			spa_pod_builder_id(b, id);
		spa_pod_builder_id(b, id);
	}
	/* then all the other ones SDL can convert from/to */
	SPA_FOR_EACH_ELEMENT_VAR(sdl_video_formats, f) {
   
		uint32_t id = f->id;
		if (id != SPA_VIDEO_FORMAT_UNKNOWN)
			spa_pod_builder_id(b, id);
	}
	spa_pod_builder_id(b, SPA_VIDEO_FORMAT_RGBA_F32);
	spa_pod_builder_pop(b, &f[1]);
	/* add size and framerate ranges */
	spa_pod_builder_add(b,
		SPA_FORMAT_VIDEO_size,      SPA_POD_CHOICE_RANGE_Rectangle(
							&SPA_RECTANGLE(WIDTH, HEIGHT),
							&SPA_RECTANGLE(1,1),
							&SPA_RECTANGLE(info->max_texture_width,
								      info->max_texture_height)),
		SPA_FORMAT_VIDEO_framerate, SPA_POD_CHOICE_RANGE_Fraction(
							&SPA_FRACTION(25,1),
							&SPA_FRACTION(0,1),
							&SPA_FRACTION(30,1)),
		0);
	return spa_pod_builder_pop(b, &f[0]);
}

screen_capture.c


#include <stdio.h>
#include <unistd.h>
#include <signal.h>

#include <pipewire/pipewire.h>

#include <gio/gunixfdlist.h>
#include <gio/gio.h>
#include <fcntl.h>
#include <linux/dma-buf.h>
#include <libdrm/drm_fourcc.h>
#include <spa/param/video/format-utils.h>
#include <spa/debug/format.h>
#include <spa/debug/types.h>
#include <spa/param/video/type-info.h>
#include <spa/utils/result.h>

#define WIDTH   1600
#define HEIGHT  1000

#define MAX_BUFFERS    64

#include "sdl.h"

struct pixel {
   
    float r, g, b, a;
};

enum portal_capture_type {
   
    PORTAL_CAPTURE_TYPE_MONITOR