From 4ca1e77fdbbf559a5bb28d4936afa6ade63985cd Mon Sep 17 00:00:00 2001
From: Drew DeVault <sir@cmpwn.com>
Date: Wed, 22 Nov 2017 21:06:08 -0500
Subject: [PATCH] Add views to tree and render them

---
 include/sway/container.h    |  4 ++
 include/sway/server.h       |  2 +
 include/sway/view.h         |  2 +
 sway/desktop/output.c       | 74 +++++++++++++++++++++++++++++++++++++
 sway/desktop/xdg_shell_v6.c | 11 ++++++
 sway/server.c               |  4 ++
 sway/tree/container.c       | 39 +++++++++++++++++++
 7 files changed, 136 insertions(+)

diff --git a/include/sway/container.h b/include/sway/container.h
index 2a96165f..96e4db69 100644
--- a/include/sway/container.h
+++ b/include/sway/container.h
@@ -125,7 +125,11 @@ struct sway_container {
 	list_t *marks;
 };
 
+void swayc_descendants_of_type(swayc_t *root, enum swayc_types type,
+		void (*func)(swayc_t *item, void *data), void *data);
+
 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);
 
 #endif
diff --git a/include/sway/server.h b/include/sway/server.h
index 60f35190..b0684d15 100644
--- a/include/sway/server.h
+++ b/include/sway/server.h
@@ -4,6 +4,7 @@
 #include <wayland-server.h>
 #include <wlr/backend.h>
 #include <wlr/backend/session.h>
+#include <wlr/types/wlr_compositor.h>
 #include <wlr/types/wlr_data_device.h>
 #include <wlr/types/wlr_xdg_shell_v6.h>
 #include <wlr/render.h>
@@ -18,6 +19,7 @@ struct sway_server {
 	struct wlr_backend *backend;
 	struct wlr_renderer *renderer;
 
+	struct wlr_compositor *compositor;
 	struct wlr_data_device_manager *data_device_manager;
 
 	struct sway_input *input;
diff --git a/include/sway/view.h b/include/sway/view.h
index 979b20a8..2675a6f5 100644
--- a/include/sway/view.h
+++ b/include/sway/view.h
@@ -1,6 +1,7 @@
 #ifndef _SWAY_VIEW_H
 #define _SWAY_VIEW_H
 #include <wayland-server.h>
+#include <wlr/types/wlr_surface.h>
 #include <wlr/types/wlr_xdg_shell_v6.h>
 
 struct sway_container;
@@ -38,6 +39,7 @@ struct sway_view {
 	struct wl_listener destroy;
 	enum sway_view_type type;
 	struct sway_container *swayc;
+	struct wlr_surface *surface;
 
 	union {
 		struct wlr_xdg_surface_v6 *wlr_xdg_surface_v6;
diff --git a/sway/desktop/output.c b/sway/desktop/output.c
index 27579c1b..9e0c18e4 100644
--- a/sway/desktop/output.c
+++ b/sway/desktop/output.c
@@ -3,11 +3,82 @@
 #include <time.h>
 #include <wayland-server.h>
 #include <wlr/types/wlr_output.h>
+#include <wlr/types/wlr_surface.h>
 #include <wlr/render.h>
+#include <wlr/render/matrix.h>
 #include "log.h"
 #include "sway/container.h"
 #include "sway/output.h"
 #include "sway/server.h"
+#include "sway/view.h"
+
+static inline int64_t timespec_to_msec(const struct timespec *a) {
+	return (int64_t)a->tv_sec * 1000 + a->tv_nsec / 1000000;
+}
+
+static void output_frame_view(swayc_t *view, void *data) {
+	struct sway_output *output = data;
+	struct wlr_output *wlr_output = output->wlr_output;
+	struct sway_view *sway_view = view->sway_view;
+	struct wlr_surface *surface = sway_view->surface;
+	if (!wlr_surface_has_buffer(surface)) {
+		return;
+	}
+	// TODO
+	// - Force sway's resolution
+	// - Deal with wlr_output_layout
+	int width = surface->current->width;
+	int height = surface->current->height;
+	int render_width = width * wlr_output->scale;
+	int render_height = height * wlr_output->scale;
+	double ox = view->x, oy = view->y;
+	// TODO
+	//wlr_output_layout_output_coords(desktop->layout, wlr_output, &ox, &oy);
+	ox *= wlr_output->scale;
+	oy *= wlr_output->scale;
+	// TODO
+	//if (wlr_output_layout_intersects(desktop->layout, wlr_output,
+	//		lx, ly, lx + render_width, ly + render_height)) {
+	//		return;
+	//}
+
+	// TODO
+	double rotation = 0;
+	float matrix[16];
+
+	float translate_origin[16];
+	wlr_matrix_translate(&translate_origin,
+		(int)ox + render_width / 2, (int)oy + render_height / 2, 0);
+
+	float rotate[16];
+	wlr_matrix_rotate(&rotate, rotation);
+
+	float translate_center[16];
+	wlr_matrix_translate(&translate_center, -render_width / 2,
+		-render_height / 2, 0);
+
+	float scale[16];
+	wlr_matrix_scale(&scale, render_width, render_height, 1);
+
+	float transform[16];
+	wlr_matrix_mul(&translate_origin, &rotate, &transform);
+	wlr_matrix_mul(&transform, &translate_center, &transform);
+	wlr_matrix_mul(&transform, &scale, &transform);
+	wlr_matrix_mul(&wlr_output->transform_matrix, &transform, &matrix);
+
+	wlr_render_with_matrix(output->server->renderer, surface->texture, &matrix);
+
+	// TODO: move into wlroots
+	struct timespec now;
+	clock_gettime(CLOCK_MONOTONIC, &now);
+
+	struct wlr_frame_callback *cb, *cnext;
+	wl_list_for_each_safe(cb, cnext,
+			&surface->current->frame_callback_list, link) {
+		wl_callback_send_done(cb->resource, timespec_to_msec(&now));
+		wl_resource_destroy(cb->resource);
+	}
+}
 
 static void output_frame_notify(struct wl_listener *listener, void *data) {
 	struct sway_output *soutput = wl_container_of(
@@ -21,6 +92,9 @@ static void output_frame_notify(struct wl_listener *listener, void *data) {
 	wlr_output_make_current(wlr_output);
 	wlr_renderer_begin(server->renderer, wlr_output);
 
+	swayc_descendants_of_type(
+			&root_container, C_VIEW, output_frame_view, soutput);
+
 	wlr_renderer_end(server->renderer);
 	wlr_output_swap_buffers(wlr_output);
 
diff --git a/sway/desktop/xdg_shell_v6.c b/sway/desktop/xdg_shell_v6.c
index b3dbcfbe..e2a61ebf 100644
--- a/sway/desktop/xdg_shell_v6.c
+++ b/sway/desktop/xdg_shell_v6.c
@@ -2,6 +2,7 @@
 #include <stdlib.h>
 #include <wayland-server.h>
 #include <wlr/types/wlr_xdg_shell_v6.h>
+#include "sway/container.h"
 #include "sway/server.h"
 #include "sway/view.h"
 #include "log.h"
@@ -47,6 +48,9 @@ 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->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:
@@ -56,4 +60,11 @@ void handle_xdg_shell_v6_surface(struct wl_listener *listener, void *data) {
 	// - Look up pid and open on appropriate workspace
 	// - Set new view to maximized so it behaves nicely
 	// - Criteria
+
+	// TODO: actual focus semantics
+	swayc_t *parent = root_container.children->items[0];
+	parent = parent->children->items[0]; // workspace
+
+	swayc_t *cont = new_view(parent, sway_view);
+	sway_view->swayc = cont;
 }
diff --git a/sway/server.c b/sway/server.c
index f78bcef1..6e66bc3c 100644
--- a/sway/server.c
+++ b/sway/server.c
@@ -6,6 +6,7 @@
 #include <wlr/backend/session.h>
 #include <wlr/render.h>
 #include <wlr/render/gles2.h>
+#include <wlr/types/wlr_compositor.h>
 // TODO WLR: make Xwayland optional
 #include <wlr/xwayland.h>
 #include "sway/server.h"
@@ -21,6 +22,9 @@ bool server_init(struct sway_server *server) {
 	server->renderer = wlr_gles2_renderer_create(server->backend);
 	wl_display_init_shm(server->wl_display);
 
+	server->compositor = wlr_compositor_create(
+			server->wl_display, server->renderer);
+
 	server->data_device_manager =
 		wlr_data_device_manager_create(server->wl_display);
 
diff --git a/sway/tree/container.c b/sway/tree/container.c
index ac79356a..d3931612 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -5,9 +5,23 @@
 #include "sway/container.h"
 #include "sway/layout.h"
 #include "sway/output.h"
+#include "sway/view.h"
 #include "sway/workspace.h"
 #include "log.h"
 
+void swayc_descendants_of_type(swayc_t *root, enum swayc_types type,
+		void (*func)(swayc_t *item, void *data), void *data) {
+	for (int i = 0; i < root->children->length; ++i) {
+		swayc_t *item = root->children->items[i];
+		if (item->type == type) {
+			func(item, data);
+		}
+		if (item->children && item->children->length) {
+			swayc_descendants_of_type(item, type, func, data);
+		}
+	}
+}
+
 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;
@@ -69,3 +83,28 @@ swayc_t *new_workspace(swayc_t *output, const char *name) {
 	sort_workspaces(output);
 	return workspace;
 }
+
+swayc_t *new_view(swayc_t *sibling, struct sway_view *sway_view) {
+	if (!sway_assert(sibling, "new_view called with NULL sibling/parent")) {
+		return NULL;
+	}
+	const char *title = sway_view->iface.get_prop(sway_view, VIEW_PROP_TITLE);
+	swayc_t *swayc = new_swayc(C_VIEW);
+	sway_log(L_DEBUG, "Adding new view %p:%s to container %p %d",
+		swayc, title, sibling, sibling ? sibling->type : 0);
+	// Setup values
+	swayc->sway_view = sway_view;
+	swayc->name = title ? strdup(title) : NULL;
+	swayc->width = 0;
+	swayc->height = 0;
+
+	if (sibling->type == C_WORKSPACE) {
+		// Case of focused workspace, just create as child of it
+		add_child(sibling, swayc);
+	} else {
+		// Regular case, create as sibling of current container
+		// TODO WLR
+		//add_sibling(sibling, swayc);
+	}
+	return swayc;
+}