#define _POSIX_C_SOURCE 200809L
#include <scenefx/render/fx_renderer/fx_effect_framebuffers.h>
#include <scenefx/render/fx_renderer/fx_renderer.h>
#include <scenefx/render/pass.h>
#include <stdlib.h>
#include <strings.h>
#include <time.h>
#include <wayland-server-core.h>
#include <wlr/config.h>
#include <wlr/render/wlr_renderer.h>
#include <wlr/types/wlr_buffer.h>
#include <wlr/types/wlr_damage_ring.h>
#include <wlr/types/wlr_matrix.h>
#include <wlr/types/wlr_output_layout.h>
#include <wlr/types/wlr_output.h>
#include <wlr/types/wlr_compositor.h>
#include <wlr/util/region.h>
#include "config.h"
#include "sway/config.h"
#include "sway/input/input-manager.h"
#include "sway/input/seat.h"
#include "sway/layers.h"
#include "sway/output.h"
#include "sway/server.h"
#include "sway/tree/arrange.h"
#include "sway/tree/container.h"
#include "sway/tree/root.h"
#include "sway/tree/view.h"
#include "sway/tree/workspace.h"

static void transform_output_damage(pixman_region32_t *damage, struct wlr_output *output) {
	int ow, oh;
	wlr_output_transformed_resolution(output, &ow, &oh);
	enum wl_output_transform transform =
		wlr_output_transform_invert(output->transform);
	wlr_region_transform(damage, damage, transform, ow, oh);
}

static void transform_output_box(struct wlr_box *box, struct wlr_output *output) {
	int ow, oh;
	wlr_output_transformed_resolution(output, &ow, &oh);
	enum wl_output_transform transform =
		wlr_output_transform_invert(output->transform);
	wlr_box_transform(box, box, transform, ow, oh);
}

// TODO: Remove this ugly abomination with a complete border rework...
static void transform_corner_location(enum corner_location *corner_location, struct wlr_output *output) {
	if (*corner_location == ALL) {
		return;
	}
	switch (wlr_output_transform_invert(output->transform)) {
		case WL_OUTPUT_TRANSFORM_NORMAL:
			return;
		case WL_OUTPUT_TRANSFORM_90:
			*corner_location = (*corner_location + 1) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_180:
			*corner_location = (*corner_location + 2) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_270:
			*corner_location = (*corner_location + 3) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_FLIPPED:
			*corner_location = (*corner_location + (1 - 2 * (*corner_location % 2))) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_FLIPPED_90:
			*corner_location = (*corner_location + (4 - 2 * (*corner_location % 2))) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_FLIPPED_180:
			*corner_location = (*corner_location + (3 - 2 * (*corner_location % 2))) % 4;
			return;
		case WL_OUTPUT_TRANSFORM_FLIPPED_270:
			*corner_location = (*corner_location + (2 - 2 * (*corner_location % 2))) % 4;
			return;
		default:
			return;
	}
}

struct decoration_data get_undecorated_decoration_data() {
	return (struct decoration_data) {
		.alpha = 1.0f,
		.dim = 0.0f,
		.dim_color = config->dim_inactive_colors.unfocused,
		.corner_radius = 0,
		.saturation = 1.0f,
		.has_titlebar = false,
		.blur = false,
		.discard_transparent = false,
		.shadow = false,
	};
}

// Adjust the box position when switching the workspace
static void adjust_box_to_workspace_offset(struct wlr_box *box,
		bool on_focused_workspace, struct sway_output *output) {
	float scroll_percent = output->workspace_scroll.percent;

	int ws_dimen;
	int *box_coord = NULL;
	switch (output->workspace_scroll.direction) {
	case SWIPE_GESTURE_DIRECTION_NONE:
		return;
	case SWIPE_GESTURE_DIRECTION_HORIZONTAL:
		ws_dimen = output->width;
		box_coord = &box->x;
		break;
	case SWIPE_GESTURE_DIRECTION_VERTICAL:
		ws_dimen = output->height;
		box_coord = &box->y;
		break;
	}

	if (box_coord == NULL || ws_dimen == 0) {
		return;
	}

	*box_coord -= ws_dimen * scroll_percent;
	if (!on_focused_workspace) {
		if (scroll_percent > 0) {
			*box_coord += ws_dimen;
		} else if (scroll_percent < 0) {
			*box_coord -= ws_dimen;
		}
	}
}

// Clips the rendered damage to the workspace region.
// Fixes containers being rendered across workspaces while switching.
static void adjust_damage_to_workspace_bounds(pixman_region32_t *damage,
		bool on_focused_workspace, struct sway_output *output) {
	float scale = output->wlr_output->scale;
	float scroll_percent = output->workspace_scroll.percent;
	int x = 0, y = 0;

	int ws_dimen = 0;
	int *coord = NULL;
	switch (output->workspace_scroll.direction) {
	case SWIPE_GESTURE_DIRECTION_NONE:
		return;
	case SWIPE_GESTURE_DIRECTION_HORIZONTAL:
		ws_dimen = output->width;
		coord = &x;
		break;
	case SWIPE_GESTURE_DIRECTION_VERTICAL:
		ws_dimen = output->height;
		coord = &y;
		break;
	}

	if (coord == NULL || ws_dimen == 0) {
		return;
	}

	*coord = round(-ws_dimen * scroll_percent);
	if (!on_focused_workspace) {
		if (scroll_percent > 0) {
			*coord += ws_dimen;
		} else if (scroll_percent < 0) {
			*coord -= ws_dimen;
		}
	}

	int width, height;
	wlr_output_transformed_resolution(output->wlr_output, &width, &height);
	pixman_region32_intersect_rect(damage, damage,
			0, 0, width, height);
	pixman_region32_translate(damage, x * scale, y * scale);
}

/**
 * Apply scale to a width or height.
 *
 * One does not simply multiply the width by the scale. We allow fractional
 * scaling, which means the resulting scaled width might be a decimal.
 * So we round it.
 *
 * But even this can produce undesirable results depending on the X or Y offset
 * of the box. For example, with a scale of 1.5, a box with width=1 should not
 * scale to 2px if its X coordinate is 1, because the X coordinate would have
 * scaled to 2px.
 */
static int scale_length(int length, int offset, float scale) {
	return roundf((offset + length) * scale) - roundf(offset * scale);
}

static enum wlr_scale_filter_mode get_scale_filter(struct sway_output *output) {
	switch (output->scale_filter) {
	case SCALE_FILTER_LINEAR:
		return WLR_SCALE_FILTER_BILINEAR;
	case SCALE_FILTER_NEAREST:
		return WLR_SCALE_FILTER_NEAREST;
	default:
		abort(); // unreachable
	}
}

static void render_texture(struct fx_render_context *ctx, struct wlr_texture *texture,
		const struct wlr_fbox *_src_box, const struct wlr_box *dst_box,
		const struct wlr_box *_clip_box, enum wl_output_transform transform,
		struct decoration_data deco_data) {
	struct sway_output *output = ctx->output;

	struct wlr_box proj_box = *dst_box;

	struct wlr_fbox src_box = {0};
	if (_src_box) {
		src_box = *_src_box;
	}

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, proj_box.x, proj_box.y,
		proj_box.width, proj_box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);

	struct wlr_box clip_box = {0};
	if (_clip_box) {
		pixman_region32_intersect_rect(&damage, &damage,
				_clip_box->x, _clip_box->y, _clip_box->width, _clip_box->height);

		clip_box = *_clip_box;
	}

	bool damaged = pixman_region32_not_empty(&damage);
	if (!damaged) {
		goto damage_finish;
	}

	transform_output_box(&proj_box, output->wlr_output);
	transform_output_box(&clip_box, output->wlr_output);
	transform_output_damage(&damage, output->wlr_output);
	transform = wlr_output_transform_compose(transform, output->wlr_output->transform);

	fx_render_pass_add_texture(ctx->pass, &(struct fx_render_texture_options) {
		.base = {
			.texture = texture,
			.src_box = src_box,
			.dst_box = proj_box,
			.transform = transform,
			.alpha = &deco_data.alpha,
			.clip = &damage,
			.filter_mode = get_scale_filter(output),
		},
		.clip_box = &clip_box,
		.corner_radius = deco_data.corner_radius,
		.has_titlebar = deco_data.has_titlebar,
		.dim = deco_data.dim,
		.dim_color = {
			.r = deco_data.dim_color[0],
			.g = deco_data.dim_color[1],
			.b = deco_data.dim_color[2],
			.a = deco_data.dim_color[3],
		}
	});

damage_finish:
	pixman_region32_fini(&damage);
}

void render_blur(struct fx_render_context *ctx, struct wlr_texture *texture,
		const struct wlr_fbox *src_box, const struct wlr_box *dst_box,
		bool optimized_blur, pixman_region32_t *opaque_region,
		struct decoration_data deco_data) {
	struct sway_output *output = ctx->output;

	struct wlr_box proj_box = *dst_box;

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, proj_box.x, proj_box.y,
		proj_box.width, proj_box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);

	if (!pixman_region32_not_empty(&damage)) {
		goto damage_finish;
	}

	transform_output_box(&proj_box, output->wlr_output);
	transform_output_damage(&damage, output->wlr_output);

	struct fx_render_blur_pass_options blur_options = {
		.tex_options = {
			.base = {
				.texture = texture,
				.src_box = *src_box,
				.dst_box = proj_box,
				.transform = WL_OUTPUT_TRANSFORM_NORMAL,
				.alpha = &deco_data.alpha,
				.clip = &damage,
				.filter_mode = WLR_SCALE_FILTER_BILINEAR,
			},
			.clip_box = &proj_box,
			.corner_radius = deco_data.corner_radius,
			.discard_transparent = false,
		},
		.opaque_region = opaque_region,
		.use_optimized_blur = optimized_blur,
		.blur_data = &config->blur_params,
		.ignore_transparent = deco_data.discard_transparent,
	};
	// Render the actual blur behind the surface
	fx_render_pass_add_blur(ctx->pass, &blur_options);

damage_finish:
	pixman_region32_fini(&damage);
}

// _box.x and .y are expected to be layout-local
// _box.width and .height are expected to be output-buffer-local
void render_box_shadow(struct fx_render_context *ctx, const struct wlr_box *_box,
		const float color[static 4], float blur_sigma, float corner_radius,
		float offset_x, float offset_y) {
	struct wlr_output *wlr_output = ctx->output->wlr_output;

	struct wlr_box box;
	memcpy(&box, _box, sizeof(struct wlr_box));

	// Extend the size of the box while also considering the shadow offset
	struct wlr_box shadow_box;
	memcpy(&shadow_box, _box, sizeof(struct wlr_box));
	shadow_box.x -= blur_sigma - offset_x;
	shadow_box.y -= blur_sigma - offset_y;
	shadow_box.width += blur_sigma * 2;
	shadow_box.height += blur_sigma * 2;

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, shadow_box.x, shadow_box.y,
			shadow_box.width, shadow_box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);
	if (!pixman_region32_not_empty(&damage)) {
		goto damage_finish;
	}

	transform_output_damage(&damage, wlr_output);
	transform_output_box(&box, wlr_output);
	transform_output_box(&shadow_box, wlr_output);

	struct shadow_data shadow_data = {
		.enabled = true,
		.offset_x = offset_x,
		.offset_y = offset_y,
		.color = {
			.r = color[0],
			.g = color[1],
			.b = color[2],
			.a = color[3],
		},
		.blur_sigma = blur_sigma,
	};
	struct fx_render_box_shadow_options shadow_options = {
		.shadow_box = shadow_box,
		.clip_box = box,
		.clip = &damage,
		.shadow_data = &shadow_data,
		.corner_radius = corner_radius,
	};
	fx_render_pass_add_box_shadow(ctx->pass, &shadow_options);

damage_finish:
	pixman_region32_fini(&damage);
}

// _box.x and .y are expected to be layout-local
// _box.width and .height are expected to be output-buffer-local
void render_rounded_border_corner(struct fx_render_context *ctx, const struct wlr_box *_box,
			const float color[static 4], int corner_radius, int border_thickness,
			enum corner_location location) {
	struct wlr_output *wlr_output = ctx->output->wlr_output;

	struct wlr_box box = *_box;
	const int size = MAX(box.width, box.height);
	box.width = size;
	box.height = size;
	box.x -= ctx->output->lx * wlr_output->scale;
	box.y -= ctx->output->ly * wlr_output->scale;

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, box.x, box.y, box.width, box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);
	if (!pixman_region32_not_empty(&damage)) {
		goto damage_finish;
	}

	transform_output_damage(&damage, wlr_output);
	transform_output_box(&box, wlr_output);
	transform_corner_location(&location, wlr_output);

	struct fx_render_rounded_border_corner_options border_corner_options = {
		.base = {
			.box = box,
			.color = {
				.r = color[0],
				.g = color[1],
				.b = color[2],
				.a = color[3],
			},
			.clip = &damage, // Render with the original extended clip region
		},
		.corner_radius = corner_radius,
		.border_thickness = border_thickness,
		.corner_location = location
	};
	fx_render_pass_add_rounded_border_corner(ctx->pass, &border_corner_options);

damage_finish:
	pixman_region32_fini(&damage);
}

static void render_surface_iterator(struct sway_output *output,
		struct sway_view *_view, struct wlr_surface *surface,
		struct wlr_box *_box, void *_data) {
	struct render_data *data = _data;
	struct wlr_output *wlr_output = output->wlr_output;

	struct wlr_texture *texture = wlr_surface_get_texture(surface);
	if (!texture) {
		return;
	}

	struct wlr_fbox src_box;
	wlr_surface_get_buffer_source_box(surface, &src_box);

	struct wlr_box dst_box = *_box;
	struct wlr_box clip_box = *_box;
	if (data->clip_box != NULL) {
		clip_box.width = fmin(dst_box.width, data->clip_box->width);
		clip_box.height = fmin(dst_box.height, data->clip_box->height);
		clip_box.x = fmax(dst_box.x, data->clip_box->x);
		clip_box.y = fmax(dst_box.y, data->clip_box->y);
 	}

	if (data->view) {
		adjust_box_to_workspace_offset(&dst_box, data->on_focused_workspace, output);
		adjust_box_to_workspace_offset(&clip_box, data->on_focused_workspace, output);
	}

	scale_box(&dst_box, wlr_output->scale);
	scale_box(&clip_box, wlr_output->scale);

	struct decoration_data deco_data = data->deco_data;
	deco_data.corner_radius *= wlr_output->scale;

	// render blur
	struct sway_view *view = data->view;
	bool is_subsurface = view ? view->surface != surface : false;
	if (deco_data.blur && config_should_parameters_blur() && !is_subsurface) {
		pixman_region32_t opaque_region;
		pixman_region32_init(&opaque_region);

		bool has_alpha = false;
		if (deco_data.alpha < 1.0 || deco_data.dim_color[3] < 1.0) {
			has_alpha = true;
			pixman_region32_union_rect(&opaque_region, &opaque_region, 0, 0, 0, 0);
		} else {
			has_alpha = !surface->opaque;
			pixman_region32_copy(&opaque_region, &surface->opaque_region);
		}

		if (has_alpha) {
			bool should_optimize_blur = view ?
				!container_is_floating_or_child(view->container) || config->blur_xray
				: false;
			render_blur(data->ctx, texture, &src_box, &clip_box,
					should_optimize_blur, &opaque_region, deco_data);
		}

		pixman_region32_fini(&opaque_region);
	}

	deco_data.discard_transparent = false;

	render_texture(data->ctx, texture,
		&src_box, &dst_box, &clip_box, surface->current.transform, deco_data);

	wlr_presentation_surface_textured_on_output(server.presentation, surface,
		wlr_output);
}

// view will be NULL every time
static void render_layer_iterator(struct sway_output *output,
		struct sway_view *view, struct wlr_surface *surface,
		struct wlr_box *_box, void *_data) {
	// render the layer's surface
	render_surface_iterator(output, view, surface, _box, _data);

	struct render_data *data = _data;
	struct decoration_data deco_data = data->deco_data;

	// Ignore effects if this is a subsurface
	if (!wlr_layer_surface_v1_try_from_wlr_surface(surface)) {
		deco_data = get_undecorated_decoration_data();
	}

	// render shadow
	if (deco_data.shadow && config_should_parameters_shadow()) {
		float output_scale = output->wlr_output->scale;
		struct wlr_box box = *_box;
		scale_box(&box, output_scale);
		render_box_shadow(data->ctx, &box, config->shadow_color, config->shadow_blur_sigma * output_scale,
			deco_data.corner_radius * output_scale, config->shadow_offset_x * output_scale, config->shadow_offset_y * output_scale);
	}
}

static void render_layer_toplevel(struct fx_render_context *ctx, struct wl_list *layer_surfaces) {
	struct render_data data = {
		.deco_data = get_undecorated_decoration_data(),
		.on_focused_workspace = true,
 		.ctx = ctx,
	};
	output_layer_for_each_toplevel_surface(ctx->output, layer_surfaces,
		render_layer_iterator, &data);
}

 static void render_layer_popups(struct fx_render_context *ctx, struct wl_list *layer_surfaces) {
	struct render_data data = {
		.deco_data = get_undecorated_decoration_data(),
		.on_focused_workspace = true,
		.ctx = ctx,
	};
	output_layer_for_each_popup_surface(ctx->output, layer_surfaces,
		render_layer_iterator, &data);
}

#if HAVE_XWAYLAND
static void render_unmanaged(struct fx_render_context *ctx, struct wl_list *unmanaged) {
	struct render_data data = {
		.deco_data = get_undecorated_decoration_data(),
		.on_focused_workspace = true,
		.ctx = ctx,
	};
	output_unmanaged_for_each_surface(ctx->output, unmanaged,
		render_surface_iterator, &data);
}
#endif

static void render_drag_icons(struct fx_render_context *ctx, struct wl_list *drag_icons) {
	struct render_data data = {
		.deco_data = get_undecorated_decoration_data(),
		.on_focused_workspace = true,
		.ctx = ctx,
	};
 	output_drag_icons_for_each_surface(ctx->output, drag_icons,
		render_surface_iterator, &data);
}

// _box.x and .y are expected to be layout-local
// _box.width and .height are expected to be output-buffer-local
void render_rect(struct fx_render_context *ctx, const struct wlr_box *_box,
		float color[static 4]) {
	struct wlr_output *wlr_output = ctx->output->wlr_output;

	struct wlr_box box = *_box;
	box.x -= ctx->output->lx * wlr_output->scale;
	box.y -= ctx->output->ly * wlr_output->scale;

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, box.x, box.y,
		box.width, box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);
	bool damaged = pixman_region32_not_empty(&damage);
	if (!damaged) {
		goto damage_finish;
	}

	transform_output_damage(&damage, wlr_output);
	transform_output_box(&box, wlr_output);

	fx_render_pass_add_rect(ctx->pass, &(struct fx_render_rect_options){
		.base = {
			.box = box,
			.color = {
				.r = color[0],
				.g = color[1],
				.b = color[2],
				.a = color[3],
			},
			.clip = &damage,
		},
	});

damage_finish:
	pixman_region32_fini(&damage);
}

void render_rounded_rect(struct fx_render_context *ctx, const struct wlr_box *_box,
		float color[static 4], int corner_radius, enum corner_location corner_location) {
	if (!corner_radius) {
		render_rect(ctx, _box, color);
		return;
	}

	struct wlr_output *wlr_output = ctx->output->wlr_output;

	struct wlr_box box = *_box;
	box.x -= ctx->output->lx * wlr_output->scale;
	box.y -= ctx->output->ly * wlr_output->scale;

	pixman_region32_t damage;
	pixman_region32_init_rect(&damage, box.x, box.y,
		box.width, box.height);
	pixman_region32_intersect(&damage, &damage, ctx->output_damage);
	bool damaged = pixman_region32_not_empty(&damage);
	if (!damaged) {
		goto damage_finish;
	}

	transform_output_damage(&damage, wlr_output);
	transform_output_box(&box, wlr_output);
	transform_corner_location(&corner_location, wlr_output);

	fx_render_pass_add_rounded_rect(ctx->pass, &(struct fx_render_rounded_rect_options){
		.base = {
			.box = box,
			.color = {
				.r = color[0],
				.g = color[1],
				.b = color[2],
				.a = color[3],
			},
			.clip = &damage,
		},
		.corner_radius = corner_radius,
		.corner_location = corner_location
	});

damage_finish:
	pixman_region32_fini(&damage);
}

void premultiply_alpha(float color[4], float opacity) {
	color[3] *= opacity;
	color[0] *= color[3];
	color[1] *= color[3];
	color[2] *= color[3];
}

static void render_view_toplevels(struct fx_render_context *ctx,
		struct sway_view *view, struct decoration_data deco_data, bool on_focused_workspace) {
	struct render_data data = {
		.deco_data = deco_data,
		.on_focused_workspace = on_focused_workspace,
		.view = view,
		.ctx = ctx,
	};
	// Clip the window to its view size, ignoring CSD
	struct wlr_box clip_box;
	struct sway_container_state state = view->container->current;
	clip_box.x = floor(state.x) - ctx->output->lx;
	clip_box.y = floor(state.y) - ctx->output->ly;
	clip_box.width = state.width;
	clip_box.height = state.height;

	bool smart = config->hide_edge_borders_smart == ESMART_ON ||
		(config->hide_edge_borders_smart == ESMART_NO_GAPS &&
		!gaps_to_edge(view));

	if (state.fullscreen_mode == FULLSCREEN_NONE
			&& (state.border == B_PIXEL || state.border == B_NORMAL)
			&& !smart) {
		clip_box.x += state.border_thickness;
		clip_box.width -= state.border_thickness * 2;

		if (deco_data.has_titlebar) {
			// Shift the box downward to compensate for the titlebar
			int titlebar_thickness = container_titlebar_height();
			clip_box.y += titlebar_thickness;
			clip_box.height -= state.border_thickness + titlebar_thickness;
		} else {
			// Regular border
			clip_box.y += state.border_thickness;
			clip_box.height -= state.border_thickness * 2;
		}
	}
	data.clip_box = &clip_box;

	// Render all toplevels without descending into popups
	double ox = view->container->surface_x -
		ctx->output->lx - view->geometry.x;
	double oy = view->container->surface_y -
		ctx->output->ly - view->geometry.y;
	output_surface_for_each_surface(ctx->output, view->surface, ox, oy,
			render_surface_iterator, &data);
}

static void render_view_popups(struct fx_render_context *ctx, struct sway_view *view,
		struct decoration_data deco_data) {
	struct render_data data = {
		.deco_data = deco_data,
		.on_focused_workspace = true,
		.view = view,
		.ctx = ctx,
	};
	output_view_for_each_popup_surface(ctx->output, view,
		render_surface_iterator, &data);
}

static void render_saved_view(struct fx_render_context *ctx, struct sway_view *view, struct decoration_data deco_data) {
	struct sway_output *output = ctx->output;
	struct wlr_output *wlr_output = output->wlr_output;

	if (wl_list_empty(&view->saved_buffers)) {
		return;
	}

	deco_data.corner_radius *= wlr_output->scale;

	struct sway_saved_buffer *saved_buf;
	wl_list_for_each(saved_buf, &view->saved_buffers, link) {
		if (!saved_buf->buffer->texture) {
			continue;
		}

		struct wlr_box proj_box = {
			.x = saved_buf->x - view->saved_geometry.x - output->lx,
			.y = saved_buf->y - view->saved_geometry.y - output->ly,
			.width = saved_buf->width,
			.height = saved_buf->height,
		};

		struct wlr_box output_box = {
			.width = output->width,
			.height = output->height,
		};

		struct wlr_box intersection;
		bool intersects = wlr_box_intersection(&intersection, &output_box, &proj_box);
		if (!intersects) {
			continue;
		}

		struct wlr_box dst_box = proj_box;
		struct wlr_box clip_box = proj_box;

		// Clip to actual geometry, clipping the CSD
		struct sway_container_state state = view->container->current;
		clip_box.x = state.x - output->lx;
		clip_box.y = state.y - output->ly;
		clip_box.width = state.width;
		clip_box.height = state.height;
		if (state.border == B_PIXEL || state.border == B_NORMAL) {
			clip_box.x += state.border_thickness;
			clip_box.width -= state.border_thickness * 2;

			if (deco_data.has_titlebar) {
				// Shift the box downward to compensate for the titlebar
				int titlebar_thickness = container_titlebar_height();
				clip_box.y += titlebar_thickness;
				clip_box.height -= state.border_thickness + titlebar_thickness;
			} else {
				// Regular border
				clip_box.y += state.border_thickness;
				clip_box.height -= state.border_thickness * 2;
			}
		}

		scale_box(&dst_box, wlr_output->scale);
		scale_box(&clip_box, wlr_output->scale);

		// render blur
		if (deco_data.blur && config_should_parameters_blur()) {
			struct fx_texture_attribs attribs;
			fx_texture_get_attribs(saved_buf->buffer->texture, &attribs);
			if (deco_data.alpha < 1.0 || attribs.has_alpha) {
				pixman_region32_t opaque_region;
				pixman_region32_init(&opaque_region);
				pixman_region32_union_rect(&opaque_region, &opaque_region, 0, 0, 0, 0);

				bool should_optimize_blur = !container_is_floating_or_child(view->container) || config->blur_xray;
				render_blur(ctx, saved_buf->buffer->texture,
						&saved_buf->source_box, &clip_box, should_optimize_blur,
						&opaque_region, deco_data);

				pixman_region32_fini(&opaque_region);
			}
		}

		deco_data.discard_transparent = false;

		render_texture(ctx, saved_buf->buffer->texture,
			&saved_buf->source_box, &dst_box, &clip_box, saved_buf->transform, deco_data);

		// FIXME: we should set the surface that this saved buffer originates from
		// as sampled here.
		// https://github.com/swaywm/sway/pull/4465#discussion_r321082059
	}
}

/**
 * Render a view's surface, shadow, and left/bottom/right borders.
 */
static void render_view(struct fx_render_context *ctx, struct sway_container *con,
		struct border_colors *colors, struct decoration_data deco_data,
		bool on_focused_workspace) {
	struct sway_view *view = con->view;

	if (!wl_list_empty(&view->saved_buffers)) {
		render_saved_view(ctx, view, deco_data);
	} else if (view->surface) {
		render_view_toplevels(ctx, view, deco_data, on_focused_workspace);
	}

	struct sway_container_state *state = &con->current;

	if (state->border == B_CSD && !config->shadows_on_csd_enabled) {
		return;
	}

	struct wlr_box box;
	int corner_radius = deco_data.corner_radius;
	float output_scale = ctx->output->wlr_output->scale;

	// render shadow
	if (con->shadow_enabled && config_should_parameters_shadow()) {
		box.x = floor(state->x) - ctx->output->lx;
		box.y = floor(state->y) - ctx->output->ly;
		box.width = state->width;
		box.height = state->height;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
		scale_box(&box, output_scale);
		int shadow_corner_radius = corner_radius == 0 ? 0 : corner_radius + state->border_thickness;
		float* shadow_color = view_is_urgent(view) || state->focused ?
			config->shadow_color : config->shadow_inactive_color;

		render_box_shadow(ctx, &box, shadow_color, config->shadow_blur_sigma * output_scale,
			shadow_corner_radius * output_scale, config->shadow_offset_x * output_scale,
			config->shadow_offset_y * output_scale);
	}

	if (state->border == B_NONE || state->border == B_CSD) {
		return;
	}

	float color[4];

	if (state->border_left) {
		memcpy(&color, colors->child_border, sizeof(float) * 4);
		premultiply_alpha(color, con->alpha);
		box.x = floor(state->x);
		box.y = floor(state->content_y);
		box.width = state->border_thickness;
		box.height = state->content_height - corner_radius;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
		if (corner_radius && !deco_data.has_titlebar) {
			box.y += corner_radius;
			box.height -= corner_radius;
		}
		scale_box(&box, output_scale);
		render_rect(ctx, &box, color);
	}

	list_t *siblings = container_get_current_siblings(con);
	enum sway_container_layout layout =
		container_current_parent_layout(con);

	if (state->border_right) {
		if (!container_is_current_floating(con) && siblings->length == 1 && layout == L_HORIZ) {
			memcpy(&color, colors->indicator, sizeof(float) * 4);
		} else {
			memcpy(&color, colors->child_border, sizeof(float) * 4);
		}
		premultiply_alpha(color, con->alpha);
		box.x = floor(state->content_x + state->content_width);
		box.y = floor(state->content_y);
		box.width = state->border_thickness;
		box.height = state->content_height - corner_radius;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
		if (corner_radius && !deco_data.has_titlebar) {
			box.y += corner_radius;
			box.height -= corner_radius;
		}
		scale_box(&box, output_scale);
		render_rect(ctx, &box, color);
	}

	if (state->border_bottom) {
		if (!container_is_current_floating(con) && siblings->length == 1 && layout == L_VERT) {
			memcpy(&color, colors->indicator, sizeof(float) * 4);
		} else {
			memcpy(&color, colors->child_border, sizeof(float) * 4);
		}
		premultiply_alpha(color, con->alpha);
		box.x = floor(state->x);
		box.y = floor(state->content_y + state->content_height);
		box.width = state->width;
		box.height = state->border_thickness;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);

		// adjust sizing for rounded border corners
		if (deco_data.corner_radius) {
			box.x += corner_radius + state->border_thickness;
			box.width -= 2 * (corner_radius + state->border_thickness);
		}
		scale_box(&box, output_scale);
		render_rect(ctx, &box, color);

		if (corner_radius && state->border_thickness > 0) {
			int size = 2 * (corner_radius + state->border_thickness);
			int scaled_corner_radius = corner_radius * output_scale;
			int scaled_border_thickness = state->border_thickness * output_scale;
			if (state->border_left) {
				box.x = floor(state->x);
				box.y = floor(state->y + state->height - size);
				adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
				box.width = size;
				box.height = size;
				scale_box(&box, output_scale);
				render_rounded_border_corner(ctx, &box, color, scaled_corner_radius,
					scaled_border_thickness, BOTTOM_LEFT);
			}
			if (state->border_right) {
				box.x = floor(state->x + state->width - size);
				box.y = floor(state->y + state->height - size);
				adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
				scale_box(&box, output_scale);
				render_rounded_border_corner(ctx, &box, color, scaled_corner_radius,
					scaled_border_thickness, BOTTOM_RIGHT);
			}
		}
	}
}

/**
 * Render a titlebar.
 *
 * Care must be taken not to render over the same pixel multiple times,
 * otherwise the colors will be incorrect when using opacity.
 *
 * The height is: 1px border, 3px padding, font height, 3px padding, 1px border
 * The left side is: 1px border, 2px padding, title
 */
static void render_titlebar(struct fx_render_context *ctx, struct sway_container *con,
		int x, int y, int width, struct border_colors *colors, int corner_radius,
		enum corner_location corner_location, struct wlr_texture *title_texture,
		struct wlr_texture *marks_texture, bool on_focused_workspace) {
	struct wlr_box box;
	float color[4];
	struct sway_output *output = ctx->output;
	float output_scale = output->wlr_output->scale;
	double output_x = output->lx;
	double output_y = output->ly;
	int titlebar_border_thickness = config->titlebar_border_thickness;
	int titlebar_h_padding = config->titlebar_h_padding;
	int titlebar_v_padding = config->titlebar_v_padding;
	enum alignment title_align = config->title_align;
	// value by which all heights should be adjusted to counteract removed bottom border
	int bottom_border_compensation = config->titlebar_separator ? 0 : titlebar_border_thickness;
	corner_radius *= output_scale;

	// Single pixel bar above title
	memcpy(&color, colors->border, sizeof(float) * 4);
	premultiply_alpha(color, con->alpha);
	box.x = x;
	box.y = y;
	box.width = width;
	box.height = titlebar_border_thickness;
	if (corner_radius) {
		if (corner_location != TOP_RIGHT) {
			box.x += corner_radius;
		}
		if (corner_location == ALL) {
			box.width -= corner_radius * 2;
		} else {
			box.width -= corner_radius;
		}
	}
	adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
	scale_box(&box, output_scale);
	render_rect(ctx, &box, color);

	// Single pixel bar below title
	if (!bottom_border_compensation) {
		box.x = x;
		box.y = y + container_titlebar_height() - titlebar_border_thickness;
		box.width = width;
		box.height = titlebar_border_thickness;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
		scale_box(&box, output_scale);
		render_rect(ctx, &box, color);
	}

	// Single pixel left edge
	box.x = x;
	box.y = y;
	box.width = titlebar_border_thickness;
	box.height = container_titlebar_height() - titlebar_border_thickness + bottom_border_compensation;
	if (corner_radius && corner_location != TOP_RIGHT) {
		box.height -= corner_radius;
		box.y += corner_radius;
	}
	adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
	scale_box(&box, output_scale);
	render_rect(ctx, &box, color);

	// Single pixel right edge
	box.x = x + width - titlebar_border_thickness;
	box.y = y;
	box.width = titlebar_border_thickness;
	box.height = container_titlebar_height() - titlebar_border_thickness + bottom_border_compensation;
	if (corner_radius && corner_location != TOP_LEFT) {
		box.height -= corner_radius;
		box.y += corner_radius;
	}
	adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
	scale_box(&box, output_scale);
	render_rect(ctx, &box, color);

	// if corner_radius: single pixel corners
	if (corner_radius) {
		// left corner
		if (corner_location != TOP_RIGHT) {
			box.x = x;
			box.y = y;
			box.width = corner_radius * 2;
			box.height = corner_radius * 2;
			adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
			scale_box(&box, output_scale);
			render_rounded_border_corner(ctx, &box, color, corner_radius,
					titlebar_border_thickness * output_scale, TOP_LEFT);
		}

		// right corner
		if (corner_location != TOP_LEFT) {
			box.x = x + width - corner_radius * 2;
			box.y = y;
			box.width = corner_radius * 2;
			box.height = corner_radius * 2;
			adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
			scale_box(&box, output_scale);
			render_rounded_border_corner(ctx, &box, color, corner_radius,
					titlebar_border_thickness * output_scale, TOP_RIGHT);
		}
	}

	int inner_x = x - output_x + titlebar_h_padding;
	int bg_y = y + titlebar_border_thickness;
	size_t inner_width = width - titlebar_h_padding * 2;

	// output-buffer local
	int ob_inner_x = roundf(inner_x * output_scale);
	int ob_inner_width = scale_length(inner_width, inner_x, output_scale);
	int ob_bg_height = scale_length(
			(titlebar_v_padding - titlebar_border_thickness) * 2 +
			config->font_height, bg_y, output_scale);

	// title marks textures should have no eyecandy
	struct decoration_data deco_data = get_undecorated_decoration_data();
	deco_data.alpha = con->alpha;

	// Marks
	int ob_marks_x = 0; // output-buffer-local
	int ob_marks_width = 0; // output-buffer-local
	if (config->show_marks && marks_texture) {
		struct wlr_box texture_box = {
			.width = marks_texture->width,
			.height = marks_texture->height,
		};
		ob_marks_width = texture_box.width;

		// The marks texture might be shorter than the config->font_height, in
		// which case we need to pad it as evenly as possible above and below.
		int ob_padding_total = ob_bg_height - texture_box.height;
		int ob_padding_above = floor(ob_padding_total / 2.0);
		int ob_padding_below = ceil(ob_padding_total / 2.0);

		// Render texture. If the title is on the right, the marks will be on
		// the left. Otherwise, they will be on the right.
		if (title_align == ALIGN_RIGHT || texture_box.width > ob_inner_width) {
			texture_box.x = ob_inner_x;
		} else {
			texture_box.x = ob_inner_x + ob_inner_width - texture_box.width;
		}
		ob_marks_x = texture_box.x;

		texture_box.y = round((bg_y - output_y) * output_scale) +
			ob_padding_above;

		struct wlr_box clip_box = texture_box;
		if (ob_inner_width < clip_box.width) {
			clip_box.width = ob_inner_width;
		}
		adjust_box_to_workspace_offset(&texture_box, on_focused_workspace, output);
		render_texture(ctx, marks_texture,
			NULL, &texture_box, &clip_box, WL_OUTPUT_TRANSFORM_NORMAL, deco_data);

		// Padding above
		memcpy(&color, colors->background, sizeof(float) * 4);
		premultiply_alpha(color, con->alpha);
		box.x = clip_box.x + round(output_x * output_scale);
		box.y = roundf((y + titlebar_border_thickness) * output_scale);
		box.width = clip_box.width;
		box.height = ob_padding_above;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
		render_rect(ctx, &box, color);

		// Padding below
		box.y += ob_padding_above + clip_box.height;
		box.height = ob_padding_below + bottom_border_compensation;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
		render_rect(ctx, &box, color);
	}

	// Title text
	int ob_title_x = 0;  // output-buffer-local
	int ob_title_width = 0; // output-buffer-local
	if (title_texture) {
		struct wlr_box texture_box = {
			.width = title_texture->width,
			.height = title_texture->height,
		};

		// The effective output may be NULL when con is not on any output.
		// This can happen because we render all children of containers,
		// even those that are out of the bounds of any output.
		struct sway_output *effective = container_get_effective_output(con);
		float title_scale = effective ? effective->wlr_output->scale : output_scale;
		texture_box.width = texture_box.width * output_scale / title_scale;
		texture_box.height = texture_box.height * output_scale / title_scale;
		ob_title_width = texture_box.width;

		// The title texture might be shorter than the config->font_height,
		// in which case we need to pad it above and below.
		int ob_padding_above = roundf((titlebar_v_padding -
					titlebar_border_thickness) * output_scale);
		int ob_padding_below = ob_bg_height - ob_padding_above -
			texture_box.height;

		// Render texture
		if (texture_box.width > ob_inner_width - ob_marks_width) {
			texture_box.x = (title_align == ALIGN_RIGHT && ob_marks_width)
				? ob_marks_x + ob_marks_width : ob_inner_x;
		} else if (title_align == ALIGN_LEFT) {
			texture_box.x = ob_inner_x;
		} else if (title_align == ALIGN_CENTER) {
			// If there are marks visible, center between the edge and marks.
			// Otherwise, center in the inner area.
			if (ob_marks_width) {
				texture_box.x = (ob_inner_x + ob_marks_x) / 2
					- texture_box.width / 2;
			} else {
				texture_box.x = ob_inner_x + ob_inner_width / 2
					- texture_box.width / 2;
			}
		} else {
			texture_box.x = ob_inner_x + ob_inner_width - texture_box.width;
		}
		ob_title_x = texture_box.x;

		texture_box.y =
			round((bg_y - output_y) * output_scale) + ob_padding_above;

		struct wlr_box clip_box = texture_box;
		if (ob_inner_width - ob_marks_width < clip_box.width) {
			clip_box.width = ob_inner_width - ob_marks_width;
		}

		adjust_box_to_workspace_offset(&texture_box, on_focused_workspace, output);
		adjust_box_to_workspace_offset(&clip_box, on_focused_workspace, output);
		render_texture(ctx, title_texture,
			NULL, &texture_box, &clip_box, WL_OUTPUT_TRANSFORM_NORMAL, deco_data);

		// Padding above
		memcpy(&color, colors->background, sizeof(float) * 4);
		premultiply_alpha(color, con->alpha);
		box.x = clip_box.x + round(output_x * output_scale);
		box.y = roundf((y + titlebar_border_thickness) * output_scale);
		box.width = clip_box.width;
		box.height = ob_padding_above;
		// adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
		render_rect(ctx, &box, color);

		// Padding below
		box.y += ob_padding_above + clip_box.height;
		box.height = ob_padding_below + bottom_border_compensation;
		render_rect(ctx, &box, color);
	}

	// Determine the left + right extends of the textures (output-buffer local)
	int ob_left_x, ob_left_width, ob_right_x, ob_right_width;
	if (ob_title_width == 0 && ob_marks_width == 0) {
		ob_left_x = ob_inner_x;
		ob_left_width = 0;
		ob_right_x = ob_inner_x;
		ob_right_width = 0;
	} else if (ob_title_x < ob_marks_x) {
		ob_left_x = ob_title_x;
		ob_left_width = ob_title_width;
		ob_right_x = ob_marks_x;
		ob_right_width = ob_marks_width;
	} else {
		ob_left_x = ob_marks_x;
		ob_left_width = ob_marks_width;
		ob_right_x = ob_title_x;
		ob_right_width = ob_title_width;
	}
	if (ob_left_x < ob_inner_x) {
		ob_left_x = ob_inner_x;
	} else if (ob_left_x + ob_left_width > ob_right_x + ob_right_width) {
		ob_right_x = ob_left_x;
		ob_right_width = ob_left_width;
	}

	// Filler between title and marks
	box.width = ob_right_x - ob_left_x - ob_left_width;
	if (box.width > 0) {
		box.x = ob_left_x + ob_left_width + round(output_x * output_scale);
		box.y = roundf(bg_y * output_scale);
		box.height = ob_bg_height + bottom_border_compensation;
		adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
		render_rect(ctx, &box, color);
	}

	// Padding on left side
	box.x = x + titlebar_border_thickness;
	box.y = y + titlebar_border_thickness;
	box.width = titlebar_h_padding - titlebar_border_thickness;
	box.height = (titlebar_v_padding - titlebar_border_thickness) * 2 +
		config->font_height + bottom_border_compensation;
	scale_box(&box, output_scale);
	int left_x = ob_left_x + round(output_x * output_scale);
	if (box.x + box.width < left_x) {
		box.width += left_x - box.x - box.width;
	}
	adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
	if (corner_radius && (corner_location == TOP_LEFT || corner_location == ALL)) {
		render_rounded_rect(ctx, &box, color, corner_radius, TOP_LEFT);
	} else {
		render_rect(ctx, &box, color);
	}

	// Padding on right side
	box.x = x + width - titlebar_h_padding;
	box.y = y + titlebar_border_thickness;
	box.width = titlebar_h_padding - titlebar_border_thickness;
	box.height = (titlebar_v_padding - titlebar_border_thickness) * 2 +
		config->font_height + bottom_border_compensation;
	scale_box(&box, output_scale);
	int right_rx = ob_right_x + ob_right_width + round(output_x * output_scale);
	if (right_rx < box.x) {
		box.width += box.x - right_rx;
		box.x = right_rx;
	}
	adjust_box_to_workspace_offset(&box, on_focused_workspace, output);
	if (corner_radius && (corner_location == TOP_RIGHT || corner_location == ALL)) {
		render_rounded_rect(ctx, &box, color, corner_radius, TOP_RIGHT);
	} else {
		render_rect(ctx, &box, color);
	}
}

/**
 * Render the top border line for a view using "border pixel".
 */
static void render_top_border(struct fx_render_context *ctx, struct sway_container *con,
		struct border_colors *colors, int corner_radius, bool on_focused_workspace) {
	struct sway_container_state *state = &con->current;
	if (!state->border_top) {
		return;
	}
	struct wlr_box box;
	float color[4];
	float output_scale = ctx->output->wlr_output->scale;

	// Child border - top edge
	memcpy(&color, colors->child_border, sizeof(float) * 4);
	premultiply_alpha(color, con->alpha);
	box.x = floor(state->x) + corner_radius;
	box.y = floor(state->y);
	box.width = state->width - (2 * corner_radius);
	box.height = state->border_thickness;
	adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
	scale_box(&box, output_scale);
	render_rect(ctx, &box, color);

	if (corner_radius && state->border_thickness > 0) {
		int size = 2 * (corner_radius + state->border_thickness);
		int scaled_corner_radius = corner_radius * output_scale;
		int scaled_border_thickness = state->border_thickness * output_scale;
		if (state->border_left) {
			box.x = floor(state->x);
			box.y = floor(state->y);
			adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
			box.width = size;
			box.height = size;
			scale_box(&box, output_scale);
			render_rounded_border_corner(ctx, &box, color, scaled_corner_radius,
				scaled_border_thickness, TOP_LEFT);
		}
		if (state->border_right) {
			box.x = floor(state->x + state->width - size);
			box.y = floor(state->y);
			adjust_box_to_workspace_offset(&box, on_focused_workspace, ctx->output);
			box.width = size;
			box.height = size;
			scale_box(&box, output_scale);
			render_rounded_border_corner(ctx, &box, color, scaled_corner_radius,
				scaled_border_thickness, TOP_RIGHT);
		}
	}
}

struct parent_data {
	enum sway_container_layout layout;
	struct wlr_box box;
	list_t *children;
	bool focused;
	struct sway_container *active_child;

	// Indicates whether the target is on the focused workspace or not.
	bool on_focused_workspace;
};

static void render_container(struct fx_render_context *ctx,
	struct sway_container *con, bool parent_focused, bool is_current_ws);

// TODO: no rounding top corners when rendering with titlebar
/**
 * Render a container's children using a L_HORIZ or L_VERT layout.
 *
 * Wrap child views in borders and leave child containers borderless because
 * they'll apply their own borders to their children.
 */
static void render_containers_linear(struct fx_render_context *ctx, struct parent_data *parent) {
	for (int i = 0; i < parent->children->length; ++i) {
		struct sway_container *child = parent->children->items[i];

		if (container_is_scratchpad_hidden(child)) {
			continue;
		}

		if (child->view) {
			struct sway_view *view = child->view;
			struct border_colors *colors;
			struct wlr_texture *title_texture;
			struct wlr_texture *marks_texture;
			struct sway_container_state *state = &child->current;

			if (view_is_urgent(view)) {
				colors = &config->border_colors.urgent;
				title_texture = child->title_urgent;
				marks_texture = child->marks_urgent;
			} else if (state->focused || parent->focused) {
				colors = &config->border_colors.focused;
				title_texture = child->title_focused;
				marks_texture = child->marks_focused;
			} else if (child == parent->active_child) {
				colors = &config->border_colors.focused_inactive;
				title_texture = child->title_focused_inactive;
				marks_texture = child->marks_focused_inactive;
			} else {
				colors = &config->border_colors.unfocused;
				title_texture = child->title_unfocused;
				marks_texture = child->marks_unfocused;
			}

			bool has_titlebar = state->border == B_NORMAL;

			struct decoration_data deco_data = {
				.alpha = child->alpha,
				.dim_color = view_is_urgent(view)
						? config->dim_inactive_colors.urgent
						: config->dim_inactive_colors.unfocused,
				.dim = child->current.focused || parent->focused ? 0.0f : child->dim,
				// no corner radius if no gaps (allows smart_gaps to work as expected)
				.corner_radius = config->smart_corner_radius &&
						ctx->output->current.active_workspace->current_gaps.top == 0
						? 0 : child->corner_radius,
				.saturation = child->saturation,
				.has_titlebar = has_titlebar,
				.blur = child->blur_enabled,
				.discard_transparent = false,
				.shadow = child->shadow_enabled,
			};
			render_view(ctx, child, colors, deco_data, parent->on_focused_workspace);
			if (has_titlebar) {
				render_titlebar(ctx, child, floor(state->x), floor(state->y),
						state->width, colors, deco_data.corner_radius,
						ALL, title_texture, marks_texture, parent->on_focused_workspace);
			} else if (state->border == B_PIXEL) {
				render_top_border(ctx, child, colors, deco_data.corner_radius, parent->on_focused_workspace);
			}
		} else {
			render_container(ctx, child,
					parent->focused || child->current.focused,
					parent->on_focused_workspace);
		}
	}
}

static bool container_is_focused(struct sway_container *con, void *data) {
	return con->current.focused;
}

static bool container_has_focused_child(struct sway_container *con) {
	return container_find_child(con, container_is_focused, NULL);
}

/**
 * Render a container's children using the L_TABBED layout.
 */
static void render_containers_tabbed(struct fx_render_context *ctx, struct parent_data *parent) {
	if (!parent->children->length) {
		return;
	}
	struct sway_container *current = parent->active_child;
	struct border_colors *current_colors = &config->border_colors.unfocused;
	int tab_width = parent->box.width / parent->children->length;

	struct decoration_data deco_data = {
		.alpha = current->alpha,
		.dim_color = current->view && view_is_urgent(current->view)
				? config->dim_inactive_colors.urgent
				: config->dim_inactive_colors.unfocused,
		.dim = current->current.focused || parent->focused ? 0.0f : current->dim,
		// no corner radius if no gaps (allows smart_gaps to work as expected)
		.corner_radius = config->smart_corner_radius &&
				ctx->output->current.active_workspace->current_gaps.top == 0
				? 0 : current->corner_radius,
		.saturation = current->saturation,
		.has_titlebar = true,
		.blur = current->blur_enabled,
		.discard_transparent = false,
		.shadow = current->shadow_enabled,
	};

	// Render tabs
	for (int i = 0; i < parent->children->length; ++i) {
		struct sway_container *child = parent->children->items[i];
		struct sway_view *view = child->view;
		struct sway_container_state *cstate = &child->current;
		struct border_colors *colors;
		struct wlr_texture *title_texture;
		struct wlr_texture *marks_texture;
		bool urgent = view ?
			view_is_urgent(view) : container_has_urgent_child(child);

		if (urgent) {
			colors = &config->border_colors.urgent;
			title_texture = child->title_urgent;
			marks_texture = child->marks_urgent;
		} else if (cstate->focused || parent->focused) {
			colors = &config->border_colors.focused;
			title_texture = child->title_focused;
			marks_texture = child->marks_focused;
		} else if (config->has_focused_tab_title && container_has_focused_child(child)) {
			colors = &config->border_colors.focused_tab_title;
			title_texture = child->title_focused_tab_title;
			marks_texture = child->marks_focused_tab_title;
		} else if (child == parent->active_child) {
			colors = &config->border_colors.focused_inactive;
			title_texture = child->title_focused_inactive;
			marks_texture = child->marks_focused_inactive;
		} else {
			colors = &config->border_colors.unfocused;
			title_texture = child->title_unfocused;
			marks_texture = child->marks_unfocused;
		}

		int x = floor(cstate->x + tab_width * i);

		// Make last tab use the remaining width of the parent
		if (i == parent->children->length - 1) {
			tab_width = parent->box.width - tab_width * i;
		}

		// only round outer corners
		int corner_radius = deco_data.corner_radius;
		enum corner_location corner_location = ALL;
		if (parent->children->length > 1) {
			if (i == 0) {
				corner_location = TOP_LEFT;
			} else if (i == parent->children->length - 1) {
				corner_location = TOP_RIGHT;
			} else {
				corner_radius = 0;
			}
		}

		render_titlebar(ctx, child, x, parent->box.y, tab_width, colors,
			corner_radius, corner_location, title_texture, marks_texture,
			parent->on_focused_workspace);

		if (child == current) {
			current_colors = colors;
		}
	}

	// Render surface and left/right/bottom borders
	if (current->view) {
		render_view(ctx, current, current_colors, deco_data,
				parent->on_focused_workspace);
	} else {
		render_container(ctx, current,
				parent->focused || current->current.focused,
				parent->on_focused_workspace);
	}
}

/**
 * Render a container's children using the L_STACKED layout.
 */
static void render_containers_stacked(struct fx_render_context *ctx, struct parent_data *parent) {
	if (!parent->children->length) {
		return;
	}
	struct sway_container *current = parent->active_child;
	struct border_colors *current_colors = &config->border_colors.unfocused;
	size_t titlebar_height = container_titlebar_height();

	struct decoration_data deco_data = {
		.alpha = current->alpha,
		.dim_color = current->view && view_is_urgent(current->view)
				? config->dim_inactive_colors.urgent
				: config->dim_inactive_colors.unfocused,
		.dim = current->current.focused || parent->focused ? 0.0f : current->dim,
		.saturation = current->saturation,
		.corner_radius = config->smart_corner_radius &&
				ctx->output->current.active_workspace->current_gaps.top == 0
				? 0 : current->corner_radius,
		.has_titlebar = true,
		.blur = current->blur_enabled,
		.discard_transparent = false,
		.shadow = current->shadow_enabled,
	};

	// Render titles
	for (int i = 0; i < parent->children->length; ++i) {
		struct sway_container *child = parent->children->items[i];
		struct sway_view *view = child->view;
		struct sway_container_state *cstate = &child->current;
		struct border_colors *colors;
		struct wlr_texture *title_texture;
		struct wlr_texture *marks_texture;
		bool urgent = view ?
			view_is_urgent(view) : container_has_urgent_child(child);

		if (urgent) {
			colors = &config->border_colors.urgent;
			title_texture = child->title_urgent;
			marks_texture = child->marks_urgent;
		} else if (cstate->focused || parent->focused) {
			colors = &config->border_colors.focused;
			title_texture = child->title_focused;
			marks_texture = child->marks_focused;
		} else if (config->has_focused_tab_title && container_has_focused_child(child)) {
			colors = &config->border_colors.focused_tab_title;
			title_texture = child->title_focused_tab_title;
			marks_texture = child->marks_focused_tab_title;
		 } else if (child == parent->active_child) {
			colors = &config->border_colors.focused_inactive;
			title_texture = child->title_focused_inactive;
			marks_texture = child->marks_focused_inactive;
		} else {
			colors = &config->border_colors.unfocused;
			title_texture = child->title_unfocused;
			marks_texture = child->marks_unfocused;
		}

		int y = parent->box.y + titlebar_height * i;
		int corner_radius = i != 0 ? 0 : deco_data.corner_radius;
		render_titlebar(ctx, child, parent->box.x, y, parent->box.width, colors,
			corner_radius, ALL, title_texture, marks_texture, parent->on_focused_workspace);

		if (child == current) {
			current_colors = colors;
		}
	}

	// Render surface and left/right/bottom borders
	if (current->view) {
		render_view(ctx, current, current_colors, deco_data,
				parent->on_focused_workspace);
	} else {
		render_container(ctx, current,
				parent->focused || current->current.focused,
				parent->on_focused_workspace);
	}
}

static void render_containers(struct fx_render_context *ctx, struct parent_data *parent) {
	if (config->hide_lone_tab && parent->children->length == 1) {
		struct sway_container *child = parent->children->items[0];
		if (child->view) {
			render_containers_linear(ctx, parent);
			return;
		}
	}

	switch (parent->layout) {
	case L_NONE:
	case L_HORIZ:
	case L_VERT:
		render_containers_linear(ctx, parent);
		break;
	case L_STACKED:
		render_containers_stacked(ctx, parent);
		break;
	case L_TABBED:
		render_containers_tabbed(ctx, parent);
		break;
	}
}

static void render_container(struct fx_render_context *ctx,
		struct sway_container *con, bool focused, bool is_focused_ws) {
	struct parent_data data = {
		.layout = con->current.layout,
		.box = {
			.x = floor(con->current.x),
			.y = floor(con->current.y),
			.width = con->current.width,
			.height = con->current.height,
		},
		.children = con->current.children,
		.focused = focused,
		.active_child = con->current.focused_inactive_child,
		.on_focused_workspace = is_focused_ws,
	};
	render_containers(ctx, &data);
}

static void render_workspace(struct fx_render_context *ctx,
		struct sway_workspace *ws, bool focused, bool on_focused_workspace) {
	struct parent_data data = {
		.layout = ws->current.layout,
		.box = {
			.x = floor(ws->current.x),
			.y = floor(ws->current.y),
			.width = ws->current.width,
			.height = ws->current.height,
		},
		.children = ws->current.tiling,
		.focused = focused,
		.active_child = ws->current.focused_inactive_child,
		.on_focused_workspace = on_focused_workspace,
	};
	render_containers(ctx, &data);
}

static void render_floating_container(struct fx_render_context *ctx,
		struct sway_container *con, bool is_focused_ws) {
	struct sway_container_state *state = &con->current;
	if (con->view) {
		struct sway_view *view = con->view;
		struct border_colors *colors;
		struct wlr_texture *title_texture;
		struct wlr_texture *marks_texture;

		if (view_is_urgent(view)) {
			colors = &config->border_colors.urgent;
			title_texture = con->title_urgent;
			marks_texture = con->marks_urgent;
		} else if (state->focused) {
			colors = &config->border_colors.focused;
			title_texture = con->title_focused;
			marks_texture = con->marks_focused;
		} else {
			colors = &config->border_colors.unfocused;
			title_texture = con->title_unfocused;
			marks_texture = con->marks_unfocused;
		}

		bool has_titlebar = state->border == B_NORMAL;
		struct decoration_data deco_data = {
			.alpha = con->alpha,
			.dim_color = view_is_urgent(view)
					? config->dim_inactive_colors.urgent
					: config->dim_inactive_colors.unfocused,
			.dim = con->current.focused ? 0.0f : con->dim,
			.saturation = con->saturation,
			.corner_radius = con->corner_radius,
			.has_titlebar = has_titlebar,
			.blur = con->blur_enabled,
			.discard_transparent = false,
			.shadow = con->shadow_enabled,
		};
		render_view(ctx, con, colors, deco_data, is_focused_ws);
		if (has_titlebar) {
			render_titlebar(ctx, con, floor(con->current.x), floor(con->current.y), con->current.width,
					colors, deco_data.corner_radius, ALL, title_texture, marks_texture, is_focused_ws);
		} else if (state->border == B_PIXEL) {
			render_top_border(ctx, con, colors, deco_data.corner_radius, is_focused_ws);
		}
	} else {
		render_container(ctx, con, state->focused, is_focused_ws);
	}
}

static void render_floating(struct fx_render_context *ctx,
		struct sway_workspace *other_ws, bool has_fullscreen) {
	for (int i = 0; i < root->outputs->length; ++i) {
		struct sway_output *output = root->outputs->items[i];

		// Don't render floating windows across outputs when switching workspaces
		if (output->workspace_scroll.percent != 0 && output != ctx->output) {
			continue;
		}

		struct sway_workspace *visible_ws = output->current.active_workspace;
		for (int j = 0; j < output->current.workspaces->length; ++j) {
			struct sway_workspace *ws = output->current.workspaces->items[j];

			// Only render affected workspaces
			if ((ws != other_ws && ws != visible_ws) || 
					(workspace_is_visible(ws) && has_fullscreen)) {
				continue;
			}

			for (int k = 0; k < ws->current.floating->length; ++k) {
				struct sway_container *floater = ws->current.floating->items[k];
				if (floater->current.fullscreen_mode != FULLSCREEN_NONE) {
					continue;
				}
				render_floating_container(ctx, floater, ws == visible_ws);
			}
		}
	}
}

static void render_seatops(struct fx_render_context *ctx) {
	struct sway_seat *seat;
	wl_list_for_each(seat, &server.input->seats, link) {
		seatop_render(seat, ctx);
	}
}

static void render_fullscreen_con(struct fx_render_context *ctx,
		pixman_region32_t *transformed_damage, struct sway_container *fullscreen_con,
		struct sway_workspace *workspace, bool clear_whole_screen) {
	struct wlr_output *wlr_output = ctx->output->wlr_output;

	bool on_focused_workspace = workspace == ctx->output->current.active_workspace;

	// Only clear the transformed fullscreen bounds
	pixman_region32_t dmg;
	pixman_region32_init(&dmg);
	if (!clear_whole_screen) {
		pixman_region32_copy(&dmg, transformed_damage);
		adjust_damage_to_workspace_bounds(&dmg, on_focused_workspace, ctx->output);
		transformed_damage = &dmg;
	}


	fx_render_pass_add_rect(ctx->pass, &(struct fx_render_rect_options){
		.base = {
			.box = { .width = wlr_output->width, .height = wlr_output->height },
			.color = { .r = 0, .g = 0, .b = 0, .a = 1 },
			.clip = transformed_damage,
		},
	});

	if (fullscreen_con->view) {
		struct decoration_data deco_data = get_undecorated_decoration_data();
		deco_data.saturation = fullscreen_con->saturation;
		if (!wl_list_empty(&fullscreen_con->view->saved_buffers)) {
			render_saved_view(ctx, fullscreen_con->view, deco_data);
		} else if (fullscreen_con->view->surface) {
			render_view_toplevels(ctx, fullscreen_con->view, deco_data, on_focused_workspace);
		}
	} else {
		render_container(ctx, fullscreen_con, fullscreen_con->current.focused, on_focused_workspace);
	}

	for (int i = 0; i < workspace->current.floating->length; ++i) {
		struct sway_container *floater =
			workspace->current.floating->items[i];
		if (container_is_transient_for(floater, fullscreen_con)) {
			render_floating_container(ctx, floater, on_focused_workspace);
		}
	}
#if HAVE_XWAYLAND
	render_unmanaged(ctx, &root->xwayland_unmanaged);
#endif

	pixman_region32_fini(&dmg);
}

void output_render(struct fx_render_context *ctx) {
	struct wlr_output *wlr_output = ctx->output->wlr_output;
	struct sway_output *output = ctx->output;
	pixman_region32_t *damage = ctx->output_damage;

	struct fx_effect_framebuffers *effect_fbos = ctx->pass->fx_effect_framebuffers;

	struct sway_workspace *workspace = output->current.active_workspace;
	if (workspace == NULL) {
		return;
	}

	// Get the sibling workspaces
	struct sway_workspace *other_ws = NULL;
	if (output->workspace_scroll.percent < 0) {
		other_ws = workspace_output_prev(workspace,
				config->workspace_gesture_wrap_around);
	} else if (output->workspace_scroll.percent > 0) {
		other_ws = workspace_output_next(workspace,
				config->workspace_gesture_wrap_around);
	}

	struct sway_container *fullscreen_con = root->fullscreen_global;
	if (!fullscreen_con) {
		fullscreen_con = workspace->current.fullscreen;
	}
	// Also check if the sibling workspace has a fullscreen container
	bool has_fullscreen = fullscreen_con != NULL;
	bool other_ws_has_fullscreen = false;
	if (other_ws) {
		other_ws_has_fullscreen = other_ws->current.fullscreen;
		if (!has_fullscreen) {
			has_fullscreen = other_ws_has_fullscreen;
		}
	}

	if (!pixman_region32_not_empty(damage)) {
		// Output isn't damaged but needs buffer swap
		return;
	}

	if (debug.damage == DAMAGE_HIGHLIGHT) {
		fx_render_pass_add_rect(ctx->pass, &(struct fx_render_rect_options){
			.base = {
				.box = { .width = wlr_output->width, .height = wlr_output->height },
				.color = { .r = 1, .g = 1, .b = 0, .a = 1 },
			},
		});
	}

	pixman_region32_t transformed_damage;
	pixman_region32_init(&transformed_damage);
	pixman_region32_copy(&transformed_damage, damage);
	transform_output_damage(&transformed_damage, wlr_output);

	if (server.session_lock.locked) {
		struct wlr_render_color clear_color = {
			.a = 1.0f
		};
		if (server.session_lock.lock == NULL) {
			// abandoned lock -> red BG
			clear_color.r = 1.f;
		}

		fx_render_pass_add_rect(ctx->pass, &(struct fx_render_rect_options){
			.base = {
				.box = { .width = wlr_output->width, .height = wlr_output->height },
				.color = clear_color,
				.clip = &transformed_damage,
			},
		});

		if (server.session_lock.lock != NULL) {
			struct render_data data = {
				.deco_data = get_undecorated_decoration_data(),
				.on_focused_workspace = true,
				.ctx = ctx,
			};

			struct wlr_session_lock_surface_v1 *lock_surface;
			wl_list_for_each(lock_surface, &server.session_lock.lock->surfaces, link) {
				if (lock_surface->output != wlr_output) {
					continue;
				}
				if (!lock_surface->surface->mapped) {
					continue;
				}

				output_surface_for_each_surface(output, lock_surface->surface,
					0.0, 0.0, render_surface_iterator, &data);
			}
		}
		goto renderer_end;
	}

	if (output_has_opaque_overlay_layer_surface(output)) {
		goto render_overlay;
	}

	if (fullscreen_con && output->workspace_scroll.percent == 0) {
		// Only draw fullscreen con if not transitioning between workspaces
		render_fullscreen_con(ctx, &transformed_damage, fullscreen_con, workspace, true);
	} else {
		int output_width, output_height;
		wlr_output_transformed_resolution(wlr_output, &output_width, &output_height);

		pixman_region32_t blur_region;
		pixman_region32_init(&blur_region);
		bool workspace_has_blur = workspace_get_blur_info(workspace, &blur_region);
		// Expand the damage to compensate for blur
		if (effect_fbos && workspace_has_blur) {
			// Skip the blur artifact prevention if damaging the whole viewport
			if (effect_fbos->blur_buffer_dirty) {
				// Needs to be extended before clearing
				pixman_region32_union_rect(damage, damage,
						0, 0, output_width, output_height);
			} else {
				// copy the surrounding content where the blur would display artifacts
				// and draw it above the artifacts

				// Expand the original damage to compensate for surrounding
				// blurred views to avoid sharp edges between damage regions
				wlr_region_expand(damage, damage, config_get_blur_size());

				pixman_region32_t extended_damage;
				pixman_region32_init(&extended_damage);
				pixman_region32_intersect(&extended_damage, damage, &blur_region);
				// Expand the region to compensate for blur artifacts
				wlr_region_expand(&extended_damage, &extended_damage, config_get_blur_size());
				// Limit to the monitors viewport
				pixman_region32_intersect_rect(&extended_damage, &extended_damage,
						0, 0, output_width, output_height);

				// capture the padding pixels around the blur where artifacts will be drawn
				pixman_region32_subtract(&effect_fbos->blur_padding_region,
						&extended_damage, damage);
				// Combine into the surface damage (we need to redraw the padding area as well)
				pixman_region32_union(damage, damage, &extended_damage);
				pixman_region32_fini(&extended_damage);

				// Copy the new extended damage into the transformed damage
				pixman_region32_copy(&transformed_damage, damage);
				transform_output_damage(&transformed_damage, wlr_output);

				// Capture the padding pixels before blur for later use
				fx_renderer_read_to_buffer(ctx->pass, &effect_fbos->blur_padding_region,
						effect_fbos->blur_saved_pixels_buffer,
						ctx->pass->buffer, true);
			}
		}
		pixman_region32_fini(&blur_region);

		fx_render_pass_add_rect(ctx->pass, &(struct fx_render_rect_options){
			.base = {
				.box = { .width = wlr_output->width, .height = wlr_output->height },
				.color = { .r = 0.25f, .g = 0.25f, .b = 0.25f, .a = 1 },
				.clip = &transformed_damage,
			},
		});

		render_layer_toplevel(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_BACKGROUND]);
		render_layer_toplevel(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM]);

		// Check if the background needs to be blurred.
		// Render optimized/x-ray blur
		if (effect_fbos && workspace_has_blur && effect_fbos->blur_buffer_dirty) {
			const float opacity = 1.0f;
			struct fx_render_blur_pass_options blur_options = {
				.tex_options = {
					.base = {
						.transform = WL_OUTPUT_TRANSFORM_NORMAL,
						.alpha = &opacity,
						.blend_mode = WLR_RENDER_BLEND_MODE_NONE,
					},
					.corner_radius = 0,
					.discard_transparent = false,
				},
				.blur_data = &config->blur_params,
			};
			fx_render_pass_add_optimized_blur(ctx->pass, &blur_options);
		}

		// Render both workspaces
		if (!other_ws_has_fullscreen && other_ws) {
			render_workspace(ctx, other_ws, false, false);
		}
		if (!fullscreen_con) {
			render_workspace(ctx, workspace, workspace->current.focused, true);
		}
		render_floating(ctx,
				!other_ws_has_fullscreen ? other_ws : NULL,
				fullscreen_con != NULL);
#if HAVE_XWAYLAND
		render_unmanaged(ctx, &root->xwayland_unmanaged);
#endif
		render_layer_toplevel(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_TOP]);

		render_layer_popups(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_BACKGROUND]);
		render_layer_popups(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM]);
		render_layer_popups(ctx,
			&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_TOP]);

		// Render the fullscreen containers on top
		if (has_fullscreen) {
			float clear_color[] = {0.25f, 0.25f, 0.25f, 1.0f};
			// Render a blank rect next to the fullscreen container if
			// there's no sibling workspace in the swipes direction
			if (!other_ws) {
				struct wlr_box mon_box = { 0, 0, output->width, output->height };
				adjust_box_to_workspace_offset(&mon_box, false, output);
				scale_box(&mon_box, output->wlr_output->scale);
				render_rect(ctx, &mon_box, clear_color);

				// Render a shadow to separate the edge and the fullscreen
				// container
				if (config_should_parameters_shadow()) {
					struct wlr_box shadow_box = { 0, 0, output->width, output->height };
					adjust_box_to_workspace_offset(&shadow_box, true, output);
					scale_box(&shadow_box, output->wlr_output->scale);
					// Render rect to fix minor pixel gaps between fullscreen
					// container and shadow
					render_rect(ctx, &shadow_box, clear_color);
					render_box_shadow(ctx, &shadow_box,
							config->shadow_color, config->shadow_blur_sigma, 0,
							0, 0);
				}
			} else {
				// Render sibling fullscreen container
				struct sway_container *f_con = other_ws->current.fullscreen;
				if (other_ws->current.fullscreen) {
					render_fullscreen_con(ctx, &transformed_damage, f_con, other_ws, false);
				}
			}
			// Render focused fullscreen container
			if (fullscreen_con) {
				render_fullscreen_con(ctx, &transformed_damage, fullscreen_con, workspace, false);
			}
		}
	}

	render_seatops(ctx);

	struct sway_seat *seat = input_manager_current_seat();
	struct sway_container *focus = seat_get_focused_container(seat);
	if (focus && focus->view) {
		struct decoration_data deco_data = {
			.alpha = focus->alpha,
			.dim_color = view_is_urgent(focus->view)
					? config->dim_inactive_colors.urgent
					: config->dim_inactive_colors.unfocused,
			.dim = focus->current.focused ? 0.0f : focus->dim,
			.corner_radius = 0,
			.saturation = focus->saturation,
			.has_titlebar = false,
			.blur = false,
			.discard_transparent = false,
			.shadow = false,
		};
		render_view_popups(ctx, focus->view, deco_data);
	}

render_overlay:
	render_layer_toplevel(ctx,
		&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_OVERLAY]);
	render_layer_popups(ctx,
		&output->layers[ZWLR_LAYER_SHELL_V1_LAYER_OVERLAY]);
	render_drag_icons(ctx, &root->drag_icons);

renderer_end:
	// Not needed if we damaged the whole viewport
	if (effect_fbos && !effect_fbos->blur_buffer_dirty) {
		// Render the saved pixels over the blur artifacts
		fx_renderer_read_to_buffer(ctx->pass, &effect_fbos->blur_padding_region,
				ctx->pass->buffer, effect_fbos->blur_saved_pixels_buffer, true);
	}

	pixman_region32_fini(&transformed_damage);
	wlr_output_add_software_cursors_to_render_pass(wlr_output, &ctx->pass->base, damage);
}