Add linux_dmabuf to registry

emersion · Sep 13, 2020 · 2122757 · 2122757
1 parent b2cb766
commit 2122757
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/include/screencast_common.h b/include/screencast_common.h
@@ -5,6 +5,7 @@
 #include <spa/param/video/format-utils.h>
 #include <wayland-client-protocol.h>
 
+#include "linux-dmabuf-unstable-v1-client-protocol.h"
 #include <gbm.h>
 
 // this seems to be right based on
@@ -79,6 +80,7 @@ struct xdpw_screencast_context {
 	struct zxdg_output_manager_v1* xdg_output_manager;
 	struct wl_shm *shm;
 	struct gbm_device *gbm;
+	struct zwp_linux_dmabuf_v1 *linux_dmabuf;
 
 	// cli options
 	const char *output_name;

diff --git a/include/wlr_screencast.h b/include/wlr_screencast.h
@@ -9,6 +9,8 @@
 
 #define XDG_OUTPUT_MANAGER_VERSION 3
 
+#define LINUX_DMABUF_VERSION 3
+
 struct xdpw_state;
 
 int xdpw_wlr_screencast_init(struct xdpw_state *state);

diff --git a/src/screencast/wlr_screencast.c b/src/screencast/wlr_screencast.c
@@ -21,6 +21,8 @@
 #include "xdpw.h"
 #include "logger.h"
 
+#include "linux-dmabuf-unstable-v1-client-protocol.h"
+
 void xdpw_wlr_frame_free(struct xdpw_screencast_instance *cast) {
 	xdpw_wlr_screencopy_frame_free(cast);
 
@@ -170,6 +172,10 @@ static void wlr_registry_handle_add(void *data, struct wl_registry *reg,
 		ctx->xdg_output_manager =
 			wl_registry_bind(reg, id, &zxdg_output_manager_v1_interface, XDG_OUTPUT_MANAGER_VERSION);
 	}
+	if (strcmp(interface, zwp_linux_dmabuf_v1_interface.name) == 0) {
+		logprint(DEBUG, "wlroots: |-- registered to interface %s (Version %u)", interface, LINUX_DMABUF_VERSION);
+		ctx->linux_dmabuf = wl_registry_bind(reg, id, &zwp_linux_dmabuf_v1_interface, LINUX_DMABUF_VERSION);
+	}
 }
 
 static void wlr_registry_handle_remove(void *data, struct wl_registry *reg,
@@ -245,6 +251,9 @@ void xdpw_wlr_screencast_finish(struct xdpw_screencast_context *ctx) {
 	if (ctx->shm) {
 		wl_shm_destroy(ctx->shm);
 	}
+	if (ctx->linux_dmabuf) {
+		zwp_linux_dmabuf_v1_destroy(ctx->linux_dmabuf);
+	}
 	if (ctx->gbm) {
 		destroy_gbm_device(ctx->gbm);
 	}