Arrange windows on desktop

8 files changed, 253 insertions, 6 deletions

diff --git a/include/sway/container.h b/include/sway/container.h
index 96e4db69..1a173f3e 100644
--- a/include/sway/container.h
+++ b/include/sway/container.h
@@ -132,4 +132,6 @@ swayc_t *new_output(struct sway_output *sway_output);
 swayc_t *new_workspace(swayc_t *output, const char *name);
 swayc_t *new_view(swayc_t *sibling, struct sway_view *sway_view);
 
+swayc_t *swayc_parent_by_type(swayc_t *container, enum swayc_types type);
+
 #endif
diff --git a/include/sway/layout.h b/include/sway/layout.h
index 6356ad00..505036a0 100644
--- a/include/sway/layout.h
+++ b/include/sway/layout.h
@@ -7,5 +7,6 @@ void init_layout(void);
 void add_child(struct sway_container *parent, struct sway_container *child);
 enum swayc_layouts default_layout(struct sway_container *output);
 void sort_workspaces(struct sway_container *output);
+void arrange_windows(struct sway_container *container, double width, double height);
 
 #endif
diff --git a/include/sway/output.h b/include/sway/output.h
index e2f81bcb..895cb07d 100644
--- a/include/sway/output.h
+++ b/include/sway/output.h
@@ -13,6 +13,7 @@ struct sway_output {
         struct sway_server *server;
         struct timespec last_frame;
         struct wl_listener frame;
+        struct wl_listener resolution;
 };
 
 #endif
diff --git a/include/sway/view.h b/include/sway/view.h
index 2675a6f5..9d503674 100644
--- a/include/sway/view.h
+++ b/include/sway/view.h
@@ -52,6 +52,8 @@ struct sway_view {
         struct {
                 const char *(*get_prop)(struct sway_view *view,
                                 enum sway_view_prop prop);
+                void (*set_dimensions)(struct sway_view *view,
+                                int width, int height);
         } iface;
 };
 
diff --git a/sway/desktop/output.c b/sway/desktop/output.c
index 9e0c18e4..033a4c44 100644
--- a/sway/desktop/output.c
+++ b/sway/desktop/output.c
@@ -8,6 +8,7 @@
 #include <wlr/render/matrix.h>
 #include "log.h"
 #include "sway/container.h"
+#include "sway/layout.h"
 #include "sway/output.h"
 #include "sway/server.h"
 #include "sway/view.h"
@@ -25,10 +26,9 @@ static void output_frame_view(swayc_t *view, void *data) {
                 return;
         }
         // TODO
-        // - Force sway's resolution
         // - Deal with wlr_output_layout
-        int width = surface->current->width;
-        int height = surface->current->height;
+        int width = sway_view->swayc->width;
+        int height = sway_view->swayc->height;
         int render_width = width * wlr_output->scale;
         int render_height = height * wlr_output->scale;
         double ox = view->x, oy = view->y;
@@ -101,6 +101,12 @@ static void output_frame_notify(struct wl_listener *listener, void *data) {
         soutput->last_frame = now;
 }
 
+static void output_resolution_notify(struct wl_listener *listener, void *data) {
+        struct sway_output *soutput = wl_container_of(
+                        listener, soutput, resolution);
+        arrange_windows(soutput->swayc, -1, -1);
+}
+
 void output_add_notify(struct wl_listener *listener, void *data) {
         struct sway_server *server = wl_container_of(listener, server, output_add);
         struct wlr_output *wlr_output = data;
@@ -113,6 +119,9 @@ void output_add_notify(struct wl_listener *listener, void *data) {
 
         output->frame.notify = output_frame_notify;
         wl_signal_add(&wlr_output->events.frame, &output->frame);
+
+        output->resolution.notify = output_resolution_notify;
+        wl_signal_add(&wlr_output->events.resolution, &output->resolution);
 }
 
 void output_remove_notify(struct wl_listener *listener, void *data) {
diff --git a/sway/desktop/xdg_shell_v6.c b/sway/desktop/xdg_shell_v6.c
index e2a61ebf..8f1885c1 100644
--- a/sway/desktop/xdg_shell_v6.c
+++ b/sway/desktop/xdg_shell_v6.c
@@ -1,15 +1,21 @@
 #define _POSIX_C_SOURCE 199309L
+#include <stdbool.h>
 #include <stdlib.h>
 #include <wayland-server.h>
 #include <wlr/types/wlr_xdg_shell_v6.h>
 #include "sway/container.h"
+#include "sway/layout.h"
 #include "sway/server.h"
 #include "sway/view.h"
 #include "log.h"
 
+static bool assert_xdg(struct sway_view *view) {
+        return sway_assert(view->type == SWAY_XDG_SHELL_V6_VIEW,
+                        "Expected xdg shell v6 view!");
+}
+
 static const char *get_prop(struct sway_view *view, enum sway_view_prop prop) {
-        if (!sway_assert(view->type == SWAY_XDG_SHELL_V6_VIEW,
-                                "xdg get_prop for non-xdg view!")) {
+        if (!assert_xdg(view)) {
                 return NULL;
         }
         switch (prop) {
@@ -22,6 +28,12 @@ static const char *get_prop(struct sway_view *view, enum sway_view_prop prop) {
         }
 }
 
+static void set_dimensions(struct sway_view *view, int width, int height) {
+        if (assert_xdg(view)) {
+                wlr_xdg_toplevel_v6_set_size(view->wlr_xdg_surface_v6, width, height);
+        }
+}
+
 void handle_xdg_shell_v6_surface(struct wl_listener *listener, void *data) {
         struct sway_server *server = wl_container_of(
                         listener, server, xdg_shell_v6_surface);
@@ -48,13 +60,13 @@ void handle_xdg_shell_v6_surface(struct wl_listener *listener, void *data) {
         }
         sway_view->type = SWAY_XDG_SHELL_V6_VIEW;
         sway_view->iface.get_prop = get_prop;
+        sway_view->iface.set_dimensions = set_dimensions;
         sway_view->wlr_xdg_surface_v6 = xdg_surface;
         sway_view->sway_xdg_surface_v6 = sway_surface;
         sway_view->surface = xdg_surface->surface;
         sway_surface->view = sway_view;
         
         // TODO:
-        // - Add to tree
         // - Wire up listeners
         // - Handle popups
         // - Look up pid and open on appropriate workspace
@@ -67,4 +79,6 @@ void handle_xdg_shell_v6_surface(struct wl_listener *listener, void *data) {
 
         swayc_t *cont = new_view(parent, sway_view);
         sway_view->swayc = cont;
+
+        arrange_windows(cont->parent, -1, -1);
 }
diff --git a/sway/tree/container.c b/sway/tree/container.c
index d3931612..a83c0f6b 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -108,3 +108,16 @@ swayc_t *new_view(swayc_t *sibling, struct sway_view *sway_view) {
         }
         return swayc;
 }
+
+swayc_t *swayc_parent_by_type(swayc_t *container, enum swayc_types type) {
+        if (!sway_assert(container, "container is NULL")) {
+                return NULL;
+        }
+        if (!sway_assert(type < C_TYPES && type >= C_ROOT, "invalid type")) {
+                return NULL;
+        }
+        do {
+                container = container->parent;
+        } while (container && container->type != type);
+        return container;
+}
diff --git a/sway/tree/layout.c b/sway/tree/layout.c
index 5a70c570..3d6b404d 100644
--- a/sway/tree/layout.c
+++ b/sway/tree/layout.c
@@ -1,10 +1,14 @@
 #define _POSIX_C_SOURCE 200809L
 #include <ctype.h>
+#include <math.h>
 #include <stdbool.h>
 #include <stdlib.h>
 #include <string.h>
+#include <wlr/types/wlr_output.h>
 #include <wlr/types/wlr_output_layout.h>
 #include "sway/container.h"
+#include "sway/output.h"
+#include "sway/view.h"
 #include "list.h"
 #include "log.h"
 
@@ -70,3 +74,204 @@ static int sort_workspace_cmp_qsort(const void *_a, const void *_b) {
 void sort_workspaces(swayc_t *output) {
         list_stable_sort(output->children, sort_workspace_cmp_qsort);
 }
+
+static void apply_horiz_layout(swayc_t *container, const double x,
+                                const double y, const double width,
+                                const double height, const int start,
+                                const int end);
+
+static void apply_vert_layout(swayc_t *container, const double x,
+                                const double y, const double width,
+                                const double height, const int start,
+                                const int end);
+
+void arrange_windows(swayc_t *container, double width, double height) {
+        int i;
+        if (width == -1 || height == -1) {
+                width = container->width;
+                height = container->height;
+        }
+        // pixels are indivisible. if we don't round the pixels, then the view
+        // calculations will be off (e.g. 50.5 + 50.5 = 101, but in reality it's
+        // 50 + 50 = 100). doing it here cascades properly to all width/height/x/y.
+        width = floor(width);
+        height = floor(height);
+
+        sway_log(L_DEBUG, "Arranging layout for %p %s %fx%f+%f,%f", container,
+                 container->name, container->width, container->height, container->x,
+                 container->y);
+
+        double x = 0, y = 0;
+        switch (container->type) {
+        case C_ROOT:
+                // TODO: wlr_output_layout probably
+                for (i = 0; i < container->children->length; ++i) {
+                        swayc_t *output = container->children->items[i];
+                        sway_log(L_DEBUG, "Arranging output '%s' at %f,%f",
+                                        output->name, output->x, output->y);
+                        arrange_windows(output, -1, -1);
+                }
+                return;
+        case C_OUTPUT:
+                {
+                        int _width, _height;
+                        wlr_output_effective_resolution(
+                                        container->sway_output->wlr_output, &_width, &_height);
+                        width = container->width = _width;
+                        height = container->height = _height;
+                }
+                // arrange all workspaces:
+                for (i = 0; i < container->children->length; ++i) {
+                        swayc_t *child = container->children->items[i];
+                        arrange_windows(child, -1, -1);
+                }
+                return;
+        case C_WORKSPACE:
+                {
+                        swayc_t *output = swayc_parent_by_type(container, C_OUTPUT);
+                        width = output->width, height = output->height;
+                        container->x = x;
+                        container->y = y;
+                        width = container->width;
+                        height = container->height;
+                        sway_log(L_DEBUG, "Arranging workspace '%s' at %f, %f",
+                                        container->name, container->x, container->y);
+                }
+                // children are properly handled below
+                break;
+        case C_VIEW:
+                {
+                        container->width = width;
+                        container->height = height;
+                        container->sway_view->iface.set_dimensions(container->sway_view,
+                                        container->width, container->height);
+                        sway_log(L_DEBUG, "Set view to %.f x %.f @ %.f, %.f",
+                                        container->width, container->height,
+                                        container->x, container->y);
+                }
+                return;
+        default:
+                container->width = width;
+                container->height = height;
+                x = container->x;
+                y = container->y;
+                break;
+        }
+
+        switch (container->layout) {
+        case L_HORIZ:
+                apply_horiz_layout(container, x, y, width, height, 0,
+                        container->children->length);
+                break;
+        case L_VERT:
+                apply_vert_layout(container, x, y, width, height, 0,
+                        container->children->length);
+                break;
+        default:
+                sway_log(L_DEBUG, "TODO: arrange layout type %d", container->layout);
+                apply_horiz_layout(container, x, y, width, height, 0,
+                        container->children->length);
+                break;
+        }
+}
+
+static void apply_horiz_layout(swayc_t *container,
+                const double x, const double y,
+                const double width, const double height,
+                const int start, const int end) {
+        double scale = 0;
+        // Calculate total width
+        for (int i = start; i < end; ++i) {
+                double *old_width = &((swayc_t *)container->children->items[i])->width;
+                if (*old_width <= 0) {
+                        if (end - start > 1) {
+                                *old_width = width / (end - start - 1);
+                        } else {
+                                *old_width = width;
+                        }
+                }
+                scale += *old_width;
+        }
+        scale = width / scale;
+
+        // Resize windows
+        double child_x = x;
+        if (scale > 0.1) {
+                sway_log(L_DEBUG, "Arranging %p horizontally", container);
+                for (int i = start; i < end; ++i) {
+                        swayc_t *child = container->children->items[i];
+                        sway_log(L_DEBUG,
+                                 "Calculating arrangement for %p:%d (will scale %f by %f)",
+                                 child, child->type, width, scale);
+                        child->x = child_x;
+                        child->y = y;
+
+                        if (i == end - 1) {
+                                double remaining_width = x + width - child_x;
+                                arrange_windows(child, remaining_width, height);
+                        } else {
+                                arrange_windows(child, child->width * scale, height);
+                        }
+                        child_x += child->width;
+                }
+
+                // update focused view border last because it may
+                // depend on the title bar geometry of its siblings.
+                /* TODO WLR
+                if (focused && container->children->length > 1) {
+                        update_container_border(focused);
+                }
+                */
+        }
+}
+
+void apply_vert_layout(swayc_t *container,
+                const double x, const double y,
+                const double width, const double height, const int start,
+                const int end) {
+        int i;
+        double scale = 0;
+        // Calculate total height
+        for (i = start; i < end; ++i) {
+                double *old_height = &((swayc_t *)container->children->items[i])->height;
+                if (*old_height <= 0) {
+                        if (end - start > 1) {
+                                *old_height = height / (end - start - 1);
+                        } else {
+                                *old_height = height;
+                        }
+                }
+                scale += *old_height;
+        }
+        scale = height / scale;
+
+        // Resize
+        double child_y = y;
+        if (scale > 0.1) {
+                sway_log(L_DEBUG, "Arranging %p vertically", container);
+                for (i = start; i < end; ++i) {
+                        swayc_t *child = container->children->items[i];
+                        sway_log(L_DEBUG,
+                                 "Calculating arrangement for %p:%d (will scale %f by %f)",
+                                 child, child->type, height, scale);
+                        child->x = x;
+                        child->y = child_y;
+
+                        if (i == end - 1) {
+                                double remaining_height = y + height - child_y;
+                                arrange_windows(child, width, remaining_height);
+                        } else {
+                                arrange_windows(child, width, child->height * scale);
+                        }
+                        child_y += child->height;
+                }
+
+                // update focused view border last because it may
+                // depend on the title bar geometry of its siblings.
+                /* TODO WLR
+                if (focused && container->children->length > 1) {
+                        update_container_border(focused);
+                }
+                */
+        }
+}