Move view marks properties to container struct

Like border properties, this will be needed to implement layout saving
and restoring.

1 files changed, 148 insertions, 3 deletions

diff --git a/sway/tree/container.c b/sway/tree/container.c
index 322f2f67..458ed7ff 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -39,6 +39,7 @@ struct sway_container *container_create(struct sway_view *view) {
 		c->children = create_list();
 		c->current.children = create_list();
 	}
+	c->marks = create_list();
 	c->outputs = create_list();
 
 	wl_signal_init(&c->events.destroy);
@@ -66,6 +67,13 @@ void container_destroy(struct sway_container *con) {
 	list_free(con->current.children);
 	list_free(con->outputs);
 
+	list_foreach(con->marks, free);
+	list_free(con->marks);
+	wlr_texture_destroy(con->marks_focused);
+	wlr_texture_destroy(con->marks_focused_inactive);
+	wlr_texture_destroy(con->marks_unfocused);
+	wlr_texture_destroy(con->marks_urgent);
+
 	if (con->view) {
 		if (con->view->container == con) {
 			con->view->container = NULL;
@@ -996,9 +1004,7 @@ void container_discover_outputs(struct sway_container *con) {
 	double new_scale = new_output ? new_output->wlr_output->scale : -1;
 	if (old_scale != new_scale) {
 		container_update_title_textures(con);
-		if (con->view) {
-			view_update_marks_textures(con->view);
-		}
+		container_update_marks_textures(con);
 	}
 }
 
@@ -1218,3 +1224,142 @@ bool container_is_transient_for(struct sway_container *child,
 		child->view && ancestor->view &&
 		view_is_transient_for(child->view, ancestor->view);
 }
+
+static bool find_by_mark_iterator(struct sway_container *con, void *data) {
+	char *mark = data;
+	return container_has_mark(con, mark);
+}
+
+struct sway_container *container_find_mark(char *mark) {
+	return root_find_container(find_by_mark_iterator, mark);
+}
+
+bool container_find_and_unmark(char *mark) {
+	struct sway_container *con = root_find_container(
+		find_by_mark_iterator, mark);
+	if (!con) {
+		return false;
+	}
+
+	for (int i = 0; i < con->marks->length; ++i) {
+		char *con_mark = con->marks->items[i];
+		if (strcmp(con_mark, mark) == 0) {
+			free(con_mark);
+			list_del(con->marks, i);
+			container_update_marks_textures(con);
+			ipc_event_window(con, "mark");
+			return true;
+		}
+	}
+	return false;
+}
+
+void container_clear_marks(struct sway_container *con) {
+	list_foreach(con->marks, free);
+	con->marks->length = 0;
+	ipc_event_window(con, "mark");
+}
+
+bool container_has_mark(struct sway_container *con, char *mark) {
+	for (int i = 0; i < con->marks->length; ++i) {
+		char *item = con->marks->items[i];
+		if (strcmp(item, mark) == 0) {
+			return true;
+		}
+	}
+	return false;
+}
+
+void container_add_mark(struct sway_container *con, char *mark) {
+	list_add(con->marks, strdup(mark));
+	ipc_event_window(con, "mark");
+}
+
+static void update_marks_texture(struct sway_container *con,
+		struct wlr_texture **texture, struct border_colors *class) {
+	struct sway_output *output = container_get_effective_output(con);
+	if (!output) {
+		return;
+	}
+	if (*texture) {
+		wlr_texture_destroy(*texture);
+		*texture = NULL;
+	}
+	if (!con->marks->length) {
+		return;
+	}
+
+	size_t len = 0;
+	for (int i = 0; i < con->marks->length; ++i) {
+		char *mark = con->marks->items[i];
+		if (mark[0] != '_') {
+			len += strlen(mark) + 2;
+		}
+	}
+	char *buffer = calloc(len + 1, 1);
+	char *part = malloc(len + 1);
+
+	if (!sway_assert(buffer && part, "Unable to allocate memory")) {
+		free(buffer);
+		return;
+	}
+
+	for (int i = 0; i < con->marks->length; ++i) {
+		char *mark = con->marks->items[i];
+		if (mark[0] != '_') {
+			sprintf(part, "[%s]", mark);
+			strcat(buffer, part);
+		}
+	}
+	free(part);
+
+	double scale = output->wlr_output->scale;
+	int width = 0;
+	int height = con->title_height * scale;
+
+	cairo_t *c = cairo_create(NULL);
+	get_text_size(c, config->font, &width, NULL, NULL, scale, false,
+			"%s", buffer);
+	cairo_destroy(c);
+
+	cairo_surface_t *surface = cairo_image_surface_create(
+			CAIRO_FORMAT_ARGB32, width, height);
+	cairo_t *cairo = cairo_create(surface);
+	cairo_set_source_rgba(cairo, class->background[0], class->background[1],
+			class->background[2], class->background[3]);
+	cairo_paint(cairo);
+	PangoContext *pango = pango_cairo_create_context(cairo);
+	cairo_set_antialias(cairo, CAIRO_ANTIALIAS_BEST);
+	cairo_set_source_rgba(cairo, class->text[0], class->text[1],
+			class->text[2], class->text[3]);
+	cairo_move_to(cairo, 0, 0);
+
+	pango_printf(cairo, config->font, scale, false, "%s", buffer);
+
+	cairo_surface_flush(surface);
+	unsigned char *data = cairo_image_surface_get_data(surface);
+	int stride = cairo_format_stride_for_width(CAIRO_FORMAT_ARGB32, width);
+	struct wlr_renderer *renderer = wlr_backend_get_renderer(
+			output->wlr_output->backend);
+	*texture = wlr_texture_from_pixels(
+			renderer, WL_SHM_FORMAT_ARGB8888, stride, width, height, data);
+	cairo_surface_destroy(surface);
+	g_object_unref(pango);
+	cairo_destroy(cairo);
+	free(buffer);
+}
+
+void container_update_marks_textures(struct sway_container *con) {
+	if (!config->show_marks) {
+		return;
+	}
+	update_marks_texture(con, &con->marks_focused,
+			&config->border_colors.focused);
+	update_marks_texture(con, &con->marks_focused_inactive,
+			&config->border_colors.focused_inactive);
+	update_marks_texture(con, &con->marks_unfocused,
+			&config->border_colors.unfocused);
+	update_marks_texture(con, &con->marks_urgent,
+			&config->border_colors.urgent);
+	container_damage_whole(con);
+}