1 files changed, 66 insertions, 0 deletions

diff --git a/sway/scene_descriptor.c b/sway/scene_descriptor.c
new file mode 100644
index 00000000..a30d4664
--- /dev/null
+++ b/sway/scene_descriptor.c
@@ -0,0 +1,66 @@
+#include <stdlib.h>
+#include <wlr/util/addon.h>
+#include "log.h"
+#include "sway/scene_descriptor.h"
+
+struct scene_descriptor {
+        void *data;
+        struct wlr_addon addon;
+};
+
+static const struct wlr_addon_interface addon_interface;
+
+static struct scene_descriptor *scene_node_get_descriptor(
+                struct wlr_scene_node *node, enum sway_scene_descriptor_type type) {
+        struct wlr_addon *addon = wlr_addon_find(&node->addons, (void *)type, &addon_interface);
+        if (!addon) {
+                return NULL;
+        }
+
+        struct scene_descriptor *desc = wl_container_of(addon, desc, addon);
+        return desc;
+}
+
+static void descriptor_destroy(struct scene_descriptor *desc) {
+        wlr_addon_finish(&desc->addon);
+        free(desc);
+}
+
+void *scene_descriptor_try_get(struct wlr_scene_node *node,
+                enum sway_scene_descriptor_type type) {
+        struct scene_descriptor *desc = scene_node_get_descriptor(node, type);
+        if (!desc) {
+                return NULL;
+        }
+
+        return desc->data;
+}
+
+void scene_descriptor_destroy(struct wlr_scene_node *node,
+                enum sway_scene_descriptor_type type) {
+        struct scene_descriptor *desc = scene_node_get_descriptor(node, type);
+        descriptor_destroy(desc);
+}
+
+static void addon_handle_destroy(struct wlr_addon *addon) {
+        struct scene_descriptor *desc = wl_container_of(addon, desc, addon);
+        descriptor_destroy(desc);
+}
+
+static const struct wlr_addon_interface addon_interface = {
+        .name = "sway_scene_descriptor",
+        .destroy = addon_handle_destroy,
+};
+
+bool scene_descriptor_assign(struct wlr_scene_node *node,
+                enum sway_scene_descriptor_type type, void *data) {
+        struct scene_descriptor *desc = calloc(1, sizeof(*desc));
+        if (!desc) {
+                sway_log(SWAY_ERROR, "Could not allocate a scene descriptor");
+                return false;
+        }
+
+        wlr_addon_init(&desc->addon, &node->addons, (void *)type, &addon_interface);
+        desc->data = data;
+        return true;
+}