Accumulate damage from subsurfaces

2 files changed, 145 insertions, 0 deletions

diff --git a/include/sway/tree/view.h b/include/sway/tree/view.h
index d4dace4a..f8e41652 100644
--- a/include/sway/tree/view.h
+++ b/include/sway/tree/view.h
@@ -45,6 +45,12 @@ struct sway_view {
                 struct wlr_xwayland_surface *wlr_xwayland_surface;
                 struct wlr_wl_shell_surface *wlr_wl_shell_surface;
         };
+
+        struct {
+                struct wl_signal unmap;
+        } events;
+
+        struct wl_listener surface_new_subsurface;
 };
 
 struct sway_xdg_shell_v6_view {
@@ -95,6 +101,27 @@ struct sway_wl_shell_view {
         int pending_width, pending_height;
 };
 
+struct sway_view_child;
+
+struct sway_view_child_impl {
+        void (*destroy)(struct sway_view_child *child);
+};
+
+/**
+ * A view child is a surface in the view tree, such as a subsurface or a popup.
+ */
+struct sway_view_child {
+        const struct sway_view_child_impl *impl;
+
+        struct sway_view *view;
+        struct wlr_surface *surface;
+
+        struct wl_listener surface_commit;
+        struct wl_listener surface_new_subsurface;
+        struct wl_listener surface_destroy;
+        struct wl_listener view_unmap;
+};
+
 const char *view_get_title(struct sway_view *view);
 
 const char *view_get_app_id(struct sway_view *view);
@@ -129,4 +156,10 @@ void view_update_position(struct sway_view *view, double ox, double oy);
 
 void view_update_size(struct sway_view *view, int width, int height);
 
+void view_child_init(struct sway_view_child *child,
+        const struct sway_view_child_impl *impl, struct sway_view *view,
+        struct wlr_surface *surface);
+
+void view_child_destroy(struct sway_view_child *child);
+
 #endif
diff --git a/sway/tree/view.c b/sway/tree/view.c
index 3927c195..85fd3093 100644
--- a/sway/tree/view.c
+++ b/sway/tree/view.c
@@ -11,6 +11,7 @@ void view_init(struct sway_view *view, enum sway_view_type type,
                 const struct sway_view_impl *impl) {
         view->type = type;
         view->impl = impl;
+        wl_signal_init(&view->events.unmap);
 }
 
 void view_destroy(struct sway_view *view) {
@@ -123,6 +124,20 @@ static void view_update_outputs(struct sway_view *view,
         }
 }
 
+static void view_subsurface_create(struct sway_view *view,
+        struct wlr_subsurface *subsurface);
+
+static void view_init_subsurfaces(struct sway_view *view,
+        struct wlr_surface *surface);
+
+static void view_handle_surface_new_subsurface(struct wl_listener *listener,
+                void *data) {
+        struct sway_view *view =
+                wl_container_of(listener, view, surface_new_subsurface);
+        struct wlr_subsurface *subsurface = data;
+        view_subsurface_create(view, subsurface);
+}
+
 void view_map(struct sway_view *view, struct wlr_surface *wlr_surface) {
         if (!sway_assert(view->surface == NULL, "cannot map mapped view")) {
                 return;
@@ -136,11 +151,18 @@ void view_map(struct sway_view *view, struct wlr_surface *wlr_surface) {
         view->surface = wlr_surface;
         view->swayc = cont;
 
+        view_init_subsurfaces(view, wlr_surface);
+        wl_signal_add(&wlr_surface->events.new_subsurface,
+                &view->surface_new_subsurface);
+        view->surface_new_subsurface.notify = view_handle_surface_new_subsurface;
+
         arrange_windows(cont->parent, -1, -1);
         input_manager_set_focus(input_manager, cont);
 
         view_damage_whole(view);
         view_update_outputs(view, NULL);
+
+        // TODO: create view children for subsurfaces
 }
 
 void view_unmap(struct sway_view *view) {
@@ -148,6 +170,8 @@ void view_unmap(struct sway_view *view) {
                 return;
         }
 
+        wl_signal_emit(&view->events.unmap, view);
+
         view_damage_whole(view);
 
         struct sway_container *parent = container_destroy(view->swayc);
@@ -185,3 +209,91 @@ void view_update_size(struct sway_view *view, int width, int height) {
         view_update_outputs(view, &box);
         view_damage_whole(view);
 }
+
+
+static void view_subsurface_create(struct sway_view *view,
+                struct wlr_subsurface *subsurface) {
+        struct sway_view_child *child = calloc(1, sizeof(struct sway_view_child));
+        if (child == NULL) {
+                wlr_log(L_ERROR, "Allocation failed");
+                return;
+        }
+        view_child_init(child, NULL, view, subsurface->surface);
+}
+
+static void view_child_handle_surface_commit(struct wl_listener *listener,
+                void *data) {
+        struct sway_view_child *child =
+                wl_container_of(listener, child, surface_commit);
+        // TODO: only accumulate damage from the child
+        view_damage_from(child->view);
+}
+
+static void view_child_handle_surface_new_subsurface(
+                struct wl_listener *listener, void *data) {
+        struct sway_view_child *child =
+                wl_container_of(listener, child, surface_new_subsurface);
+        struct wlr_subsurface *subsurface = data;
+        view_subsurface_create(child->view, subsurface);
+}
+
+static void view_child_handle_surface_destroy(struct wl_listener *listener,
+                void *data) {
+        struct sway_view_child *child =
+                wl_container_of(listener, child, surface_destroy);
+        view_child_destroy(child);
+}
+
+static void view_child_handle_view_unmap(struct wl_listener *listener,
+                void *data) {
+        struct sway_view_child *child =
+                wl_container_of(listener, child, view_unmap);
+        view_child_destroy(child);
+}
+
+static void view_init_subsurfaces(struct sway_view *view,
+                struct wlr_surface *surface) {
+        struct wlr_subsurface *subsurface;
+        wl_list_for_each(subsurface, &surface->subsurface_list, parent_link) {
+                view_subsurface_create(view, subsurface);
+        }
+}
+
+void view_child_init(struct sway_view_child *child,
+                const struct sway_view_child_impl *impl, struct sway_view *view,
+                struct wlr_surface *surface) {
+        child->impl = impl;
+        child->view = view;
+        child->surface = surface;
+
+        wl_signal_add(&surface->events.commit, &child->surface_commit);
+        child->surface_commit.notify = view_child_handle_surface_commit;
+        wl_signal_add(&surface->events.new_subsurface,
+                &child->surface_new_subsurface);
+        child->surface_new_subsurface.notify =
+                view_child_handle_surface_new_subsurface;
+        wl_signal_add(&surface->events.destroy, &child->surface_destroy);
+        child->surface_destroy.notify = view_child_handle_surface_destroy;
+        wl_signal_add(&view->events.unmap, &child->view_unmap);
+        child->view_unmap.notify = view_child_handle_view_unmap;
+
+        view_init_subsurfaces(child->view, surface);
+
+        // TODO: only damage the whole child
+        view_damage_whole(child->view);
+}
+
+void view_child_destroy(struct sway_view_child *child) {
+        // TODO: only damage the whole child
+        view_damage_whole(child->view);
+
+        wl_list_remove(&child->surface_commit.link);
+        wl_list_remove(&child->surface_destroy.link);
+        wl_list_remove(&child->view_unmap.link);
+
+        if (child->impl && child->impl->destroy) {
+                child->impl->destroy(child);
+        } else {
+                free(child);
+        }
+}