1 files changed, 157 insertions, 0 deletions
diff --git a/types/scene/output_layout.c b/types/scene/output_layout.c
new file mode 100644
index 0000000..1d1484a
--- /dev/null
+++ b/types/scene/output_layout.c
@@ -0,0 +1,157 @@
+#include <stdlib.h>
+#include <wlr/types/wlr_output_layout.h>
+#include <wlr/types/wlr_scene.h>
+
+struct wlr_scene_output_layout {
+	struct wlr_output_layout *layout;
+	struct wlr_scene *scene;
+
+	struct wl_list outputs; // wlr_scene_output_layout_output.link
+
+	struct wl_listener layout_add;
+	struct wl_listener layout_change;
+	struct wl_listener layout_destroy;
+	struct wl_listener scene_destroy;
+};
+
+struct wlr_scene_output_layout_output {
+	struct wlr_output_layout_output *layout_output;
+	struct wlr_scene_output *scene_output;
+
+	struct wl_list link; // wlr_scene_output_layout.outputs
+
+	struct wl_listener layout_output_destroy;
+	struct wl_listener scene_output_destroy;
+};
+
+static void scene_output_layout_output_destroy(
+		struct wlr_scene_output_layout_output *solo) {
+	wl_list_remove(&solo->layout_output_destroy.link);
+	wl_list_remove(&solo->scene_output_destroy.link);
+	wl_list_remove(&solo->link);
+	wlr_scene_output_destroy(solo->scene_output);
+	free(solo);
+}
+
+static void scene_output_layout_output_handle_layout_output_destroy(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout_output *solo =
+		wl_container_of(listener, solo, layout_output_destroy);
+	scene_output_layout_output_destroy(solo);
+}
+
+static void scene_output_layout_output_handle_scene_output_destroy(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout_output *solo =
+		wl_container_of(listener, solo, scene_output_destroy);
+	solo->scene_output = NULL;
+	scene_output_layout_output_destroy(solo);
+}
+
+static void scene_output_layout_destroy(struct wlr_scene_output_layout *sol) {
+	struct wlr_scene_output_layout_output *solo, *tmp;
+	wl_list_for_each_safe(solo, tmp, &sol->outputs, link) {
+		scene_output_layout_output_destroy(solo);
+	}
+	wl_list_remove(&sol->layout_add.link);
+	wl_list_remove(&sol->layout_change.link);
+	wl_list_remove(&sol->layout_destroy.link);
+	wl_list_remove(&sol->scene_destroy.link);
+	free(sol);
+}
+
+static void scene_output_layout_handle_layout_change(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout *sol =
+		wl_container_of(listener, sol, layout_change);
+
+	struct wlr_scene_output_layout_output *solo;
+	wl_list_for_each(solo, &sol->outputs, link) {
+		wlr_scene_output_set_position(solo->scene_output,
+			solo->layout_output->x, solo->layout_output->y);
+	}
+}
+
+static void scene_output_layout_add(struct wlr_scene_output_layout *sol,
+		struct wlr_output_layout_output *lo) {
+	struct wlr_scene_output_layout_output *solo = calloc(1, sizeof(*solo));
+	if (solo == NULL) {
+		return;
+	}
+
+	solo->scene_output = wlr_scene_output_create(sol->scene, lo->output);
+	if (solo->scene_output == NULL) {
+		free(solo);
+		return;
+	}
+
+	solo->layout_output = lo;
+
+	solo->layout_output_destroy.notify =
+		scene_output_layout_output_handle_layout_output_destroy;
+	wl_signal_add(&lo->events.destroy, &solo->layout_output_destroy);
+
+	solo->scene_output_destroy.notify =
+		scene_output_layout_output_handle_scene_output_destroy;
+	wl_signal_add(&solo->scene_output->events.destroy,
+		&solo->scene_output_destroy);
+
+	wl_list_insert(&sol->outputs, &solo->link);
+
+	wlr_scene_output_set_position(solo->scene_output, lo->x, lo->y);
+}
+
+static void scene_output_layout_handle_layout_add(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout *sol =
+		wl_container_of(listener, sol, layout_add);
+	struct wlr_output_layout_output *lo = data;
+
+	scene_output_layout_add(sol, lo);
+}
+
+static void scene_output_layout_handle_layout_destroy(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout *sol =
+		wl_container_of(listener, sol, layout_destroy);
+	scene_output_layout_destroy(sol);
+}
+
+static void scene_output_layout_handle_scene_destroy(
+		struct wl_listener *listener, void *data) {
+	struct wlr_scene_output_layout *sol =
+		wl_container_of(listener, sol, scene_destroy);
+	scene_output_layout_destroy(sol);
+}
+
+bool wlr_scene_attach_output_layout(struct wlr_scene *scene,
+		struct wlr_output_layout *output_layout) {
+	struct wlr_scene_output_layout *sol = calloc(1, sizeof(*sol));
+	if (sol == NULL) {
+		return false;
+	}
+
+	sol->scene = scene;
+	sol->layout = output_layout;
+
+	wl_list_init(&sol->outputs);
+
+	sol->layout_destroy.notify = scene_output_layout_handle_layout_destroy;
+	wl_signal_add(&output_layout->events.destroy, &sol->layout_destroy);
+
+	sol->layout_change.notify = scene_output_layout_handle_layout_change;
+	wl_signal_add(&output_layout->events.change, &sol->layout_change);
+
+	sol->layout_add.notify = scene_output_layout_handle_layout_add;
+	wl_signal_add(&output_layout->events.add, &sol->layout_add);
+
+	sol->scene_destroy.notify = scene_output_layout_handle_scene_destroy;
+	wl_signal_add(&scene->tree.node.events.destroy, &sol->scene_destroy);
+
+	struct wlr_output_layout_output *lo;
+	wl_list_for_each(lo, &output_layout->outputs, link) {
+		scene_output_layout_add(sol, lo);
+	}
+
+	return true;
+}