Add outputs to the tree

8 files changed, 229 insertions, 4 deletions

diff --git a/include/sway/container.h b/include/sway/container.h
new file mode 100644
index 00000000..35d1c146
--- /dev/null
+++ b/include/sway/container.h
@@ -0,0 +1,130 @@
+#ifndef _SWAY_CONTAINER_H
+#define _SWAY_CONTAINER_H
+#include <stdint.h>
+#include <sys/types.h>
+#include <wlr/types/wlr_box.h>
+#include "list.h"
+
+typedef struct sway_container swayc_t;
+
+extern swayc_t root_container;
+
+struct sway_view;
+
+/**
+ * Different kinds of containers.
+ *
+ * This enum is in order. A container will never be inside of a container below
+ * it on this list.
+ */
+enum swayc_types {
+        C_ROOT,      /**< The root container. Only one of these ever exists. */
+        C_OUTPUT,    /**< An output (aka monitor, head, etc). */
+        C_WORKSPACE, /**< A workspace. */
+        C_CONTAINER, /**< A manually created container. */
+        C_VIEW,      /**< A view (aka window). */
+
+        C_TYPES,
+};
+
+/**
+ * Different ways to arrange a container.
+ */
+enum swayc_layouts {
+        L_NONE,     /**< Used for containers that have no layout (views, root) */
+        L_HORIZ,
+        L_VERT,
+        L_STACKED,
+        L_TABBED,
+        L_FLOATING, /**< A psuedo-container, removed from the tree, to hold floating windows */
+
+        /* Awesome/Monad style auto layouts */
+        L_AUTO_LEFT,
+        L_AUTO_RIGHT,
+        L_AUTO_TOP,
+        L_AUTO_BOTTOM,
+
+        L_AUTO_FIRST = L_AUTO_LEFT,
+        L_AUTO_LAST = L_AUTO_BOTTOM,
+
+        // Keep last
+        L_LAYOUTS,
+};
+
+enum swayc_border_types {
+        B_NONE,   /**< No border */
+        B_PIXEL,  /**< 1px border */
+        B_NORMAL, /**< Normal border with title bar */
+};
+
+struct sway_output;
+struct sway_view;
+struct wlr_output_layout;
+
+/**
+ * Stores information about a container.
+ *
+ * The tree is made of these. Views are containers that cannot have children.
+ */
+struct sway_container {
+        union {
+                // TODO: Encapsulate state for other node types as well like C_CONTAINER
+                struct wlr_output_layout *output_layout; // C_ROOT
+                struct sway_output *sway_output;         // C_OUTPUT
+                struct sway_view *sway_view;             // C_VIEW
+        };
+
+        /**
+         * A unique ID to identify this container. Primarily used in the
+         * get_tree JSON output.
+         */
+        size_t id;
+
+        char *name;
+
+        enum swayc_types type;
+        enum swayc_layouts layout;
+        enum swayc_layouts prev_layout;
+        enum swayc_layouts workspace_layout;
+
+        /**
+         * The coordinates that this view appear at, relative to the output they
+         * are located on (output containers have absolute coordinates).
+         */
+        double x, y;
+
+        /**
+         * Width and height of this container, without borders or gaps.
+         */
+        double width, height;
+
+        list_t *children;
+
+        /**
+         * The parent of this container. NULL for the root container.
+         */
+        struct sway_container *parent;
+        /**
+         * Which of this container's children has focus.
+         */
+        struct sway_container *focused;
+
+        /**
+         * Number of master views in auto layouts.
+         */
+        size_t nb_master;
+
+        /**
+         * Number of slave groups (e.g. columns) in auto layouts.
+         */
+        size_t nb_slave_groups;
+
+        /**
+         * Marks applied to the container, list_t of char*.
+         */
+        list_t *marks;
+};
+
+swayc_t *new_output(struct sway_output *sway_output);
+
+#endif
diff --git a/include/sway/layout.h b/include/sway/layout.h
new file mode 100644
index 00000000..7e7a9c35
--- /dev/null
+++ b/include/sway/layout.h
@@ -0,0 +1,9 @@
+#ifndef _SWAY_LAYOUT_H
+#define _SWAY_LAYOUT_H
+
+struct sway_container;
+
+void init_layout(void);
+void add_child(struct sway_container *parent, struct sway_container *child);
+
+#endif
diff --git a/include/sway/output.h b/include/sway/output.h
index ffc6708d..e2f81bcb 100644
--- a/include/sway/output.h
+++ b/include/sway/output.h
@@ -5,9 +5,11 @@
 #include <wlr/types/wlr_output.h>
 
 struct sway_server;
+struct sway_container;
 
 struct sway_output {
         struct wlr_output *wlr_output;
+        struct sway_container *swayc;
         struct sway_server *server;
         struct timespec last_frame;
         struct wl_listener frame;
diff --git a/sway/CMakeLists.txt b/sway/CMakeLists.txt
index 73531a84..9a92466c 100644
--- a/sway/CMakeLists.txt
+++ b/sway/CMakeLists.txt
@@ -14,6 +14,9 @@ add_executable(sway
     desktop/output.c
     desktop/xdg_shell_v6.c
 
+    tree/layout.c
+    tree/container.c
+
         base64.c
         main.c
     server.c
diff --git a/sway/desktop/output.c b/sway/desktop/output.c
index 6ddcac3b..27579c1b 100644
--- a/sway/desktop/output.c
+++ b/sway/desktop/output.c
@@ -5,6 +5,7 @@
 #include <wlr/types/wlr_output.h>
 #include <wlr/render.h>
 #include "log.h"
+#include "sway/container.h"
 #include "sway/output.h"
 #include "sway/server.h"
 
@@ -34,11 +35,10 @@ void output_add_notify(struct wl_listener *listener, void *data) {
         struct sway_output *output = calloc(1, sizeof(struct sway_output));
         output->wlr_output = wlr_output;
         output->server = server;
+        output->swayc = new_output(output);
 
         output->frame.notify = output_frame_notify;
         wl_signal_add(&wlr_output->events.frame, &output->frame);
-
-        // TODO: Add to tree
 }
 
 void output_remove_notify(struct wl_listener *listener, void *data) {
diff --git a/sway/main.c b/sway/main.c
index 5710a099..42262b05 100644
--- a/sway/main.c
+++ b/sway/main.c
@@ -16,10 +16,11 @@
 #include <sys/prctl.h>
 #endif
 #include "sway/server.h"
+#include "sway/layout.h"
 #include "ipc-client.h"
+#include "log.h"
 #include "readline.h"
 #include "stringop.h"
-#include "log.h"
 #include "util.h"
 
 static bool terminate_request = false;
@@ -436,7 +437,7 @@ int main(int argc, char **argv) {
                 return 1;
         }
 
-        //init_layout();
+        init_layout();
         //ipc_init();
 
         //if (validate) {
diff --git a/sway/tree/container.c b/sway/tree/container.c
new file mode 100644
index 00000000..54bcf478
--- /dev/null
+++ b/sway/tree/container.c
@@ -0,0 +1,45 @@
+#define _POSIX_C_SOURCE 200809L
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include "sway/container.h"
+#include "sway/layout.h"
+#include "sway/output.h"
+
+static swayc_t *new_swayc(enum swayc_types type) {
+        // next id starts at 1 because 0 is assigned to root_container in layout.c
+        static size_t next_id = 1;
+        swayc_t *c = calloc(1, sizeof(swayc_t));
+        if (!c) {
+                return NULL;
+        }
+        c->id = next_id++;
+        c->layout = L_NONE;
+        c->workspace_layout = L_NONE;
+        c->type = type;
+        c->nb_master = 1;
+        c->nb_slave_groups = 1;
+        if (type != C_VIEW) {
+                c->children = create_list();
+        }
+        return c;
+}
+
+swayc_t *new_output(struct sway_output *sway_output) {
+        struct wlr_box size;
+        wlr_output_effective_resolution(sway_output->wlr_output,
+                        &size.width, &size.height);
+        const char *name = sway_output->wlr_output->name;
+
+        swayc_t *output = new_swayc(C_OUTPUT);
+        output->sway_output = sway_output;
+        output->name = name ? strdup(name) : NULL;
+        output->width = size.width;
+        output->height = size.width;
+
+        add_child(&root_container, output);
+
+        // TODO: Create workspace
+
+        return output;
+}
diff --git a/sway/tree/layout.c b/sway/tree/layout.c
new file mode 100644
index 00000000..06200bbf
--- /dev/null
+++ b/sway/tree/layout.c
@@ -0,0 +1,35 @@
+#define _POSIX_C_SOURCE 200809L
+#include <stdbool.h>
+#include <string.h>
+#include <wlr/types/wlr_output_layout.h>
+#include "sway/container.h"
+#include "list.h"
+#include "log.h"
+
+swayc_t root_container;
+
+void init_layout(void) {
+        root_container.id = 0; // normally assigned in new_swayc()
+        root_container.type = C_ROOT;
+        root_container.layout = L_NONE;
+        root_container.name = strdup("root");
+        root_container.children = create_list();
+        root_container.output_layout = wlr_output_layout_create();
+}
+
+void add_child(swayc_t *parent, swayc_t *child) {
+        sway_log(L_DEBUG, "Adding %p (%d, %fx%f) to %p (%d, %fx%f)",
+                        child, child->type, child->width, child->height,
+                        parent, parent->type, parent->width, parent->height);
+        list_add(parent->children, child);
+        child->parent = parent;
+        // set focus for this container
+        if (!parent->focused) {
+                parent->focused = child;
+        }
+        /* TODO WLR
+        if (parent->type == C_WORKSPACE && child->type == C_VIEW && (parent->workspace_layout == L_TABBED || parent->workspace_layout == L_STACKED)) {
+                child = new_container(child, parent->workspace_layout);
+        }
+        */
+}