===================================================================
RCS file: /cvsrepo/anoncvs/cvs/src/usr.bin/tmux/window-tree.c,v
retrieving revision 1.5
retrieving revision 1.6
diff -u -r1.5 -r1.6
--- src/usr.bin/tmux/window-tree.c	2017/06/09 16:01:39	1.5
+++ src/usr.bin/tmux/window-tree.c	2017/06/30 22:24:08	1.6
@@ -1,4 +1,4 @@
-/* $OpenBSD: window-tree.c,v 1.5 2017/06/09 16:01:39 nicm Exp $ */
+/* $OpenBSD: window-tree.c,v 1.6 2017/06/30 22:24:08 nicm Exp $ */
 
 /*
  * Copyright (c) 2017 Nicholas Marriott <nicholas.marriott@gmail.com>
@@ -408,6 +408,140 @@
 	}
 }
 
+static void
+window_tree_draw_session(struct session *s, struct screen_write_ctx *ctx,
+    u_int sx, u_int sy)
+{
+	struct options		*oo = s->options;
+	struct winlink		*wl;
+	struct window		*w;
+	u_int			 i, n, each, width, left;
+	struct grid_cell	 gc;
+	int			 colour, active_colour;
+	char			*label;
+	size_t			 len;
+
+	if (sx < 6)
+		return;
+	n = winlink_count(&s->windows);
+
+	memcpy(&gc, &grid_default_cell, sizeof gc);
+	colour = options_get_number(oo, "display-panes-colour");
+	active_colour = options_get_number(oo, "display-panes-active-colour");
+
+	each = sx / n;
+	if (each < 24) {
+		n = (sx - 6) / 24;
+		each = (sx - 6) / n;
+		left = sx - (n * each);
+
+		screen_write_cursormove(ctx, sx - left, 0);
+		screen_write_vline(ctx, sy, 0, 0);
+		screen_write_cursormove(ctx, sx - left + left / 2, sy / 2);
+		screen_write_puts(ctx, &grid_default_cell, "...");
+
+		if (sx == left)
+			return;
+		left = 0;
+	} else
+		left = sx - (n * each);
+
+	wl = RB_MIN(winlinks, &s->windows);
+	for (i = 0; i < n; i++) {
+		if (wl == s->curw)
+			gc.fg = active_colour;
+		else
+			gc.fg = colour;
+		if (i == n - 1)
+			width = each + left;
+		else
+			width = each - 1;
+		w = wl->window;
+
+		screen_write_cursormove(ctx, i * each, 0);
+		screen_write_preview(ctx, &w->active->base, width, sy);
+
+		xasprintf(&label, " %u:%s ", wl->idx, w->name);
+		if (strlen(label) > width)
+			xasprintf(&label, " %u ", wl->idx);
+		len = strlen(label) / 2;
+		screen_write_cursormove(ctx, i * each + each / 2 - len, sy / 2);
+		screen_write_puts(ctx, &gc, "%s", label);
+		free(label);
+
+		if (i != n - 1) {
+			screen_write_cursormove(ctx, i * each + width, 0);
+			screen_write_vline(ctx, sy, 0, 0);
+		}
+		wl = RB_NEXT(winlinks, &s->windows, wl);
+	}
+}
+
+static void
+window_tree_draw_window(struct session *s, struct window *w,
+    struct screen_write_ctx *ctx, u_int sx, u_int sy)
+{
+	struct options		*oo = s->options;
+	struct window_pane	*wp;
+	u_int			 i, n, each, width, left;
+	struct grid_cell	 gc;
+	int			 colour, active_colour;
+	char			*label;
+	size_t			 len;
+
+	if (sx < 6)
+		return;
+	n = window_count_panes(w);
+
+	memcpy(&gc, &grid_default_cell, sizeof gc);
+	colour = options_get_number(oo, "display-panes-colour");
+	active_colour = options_get_number(oo, "display-panes-active-colour");
+
+	each = sx / n;
+	if (each < 24) {
+		n = (sx - 6) / 24;
+		each = (sx - 6) / n;
+		left = sx - (n * each);
+
+		screen_write_cursormove(ctx, sx - left, 0);
+		screen_write_vline(ctx, sy, 0, 0);
+		screen_write_cursormove(ctx, sx - left + left / 2, sy / 2);
+		screen_write_puts(ctx, &grid_default_cell, "...");
+
+		if (sx == left)
+			return;
+		left = 0;
+	} else
+		left = sx - (n * each);
+
+	wp = TAILQ_FIRST(&w->panes);
+	for (i = 0; i < n; i++) {
+		if (wp == w->active)
+			gc.fg = active_colour;
+		else
+			gc.fg = colour;
+		if (i == n - 1)
+			width = each + left;
+		else
+			width = each - 1;
+
+		screen_write_cursormove(ctx, i * each, 0);
+		screen_write_preview(ctx, &wp->base, width, sy);
+
+		xasprintf(&label, " %u ", i);
+		len = strlen(label) / 2;
+		screen_write_cursormove(ctx, i * each + each / 2 - len, sy / 2);
+		screen_write_puts(ctx, &gc, "%s", label);
+		free(label);
+
+		if (i != n - 1) {
+			screen_write_cursormove(ctx, i * each + width, 0);
+			screen_write_vline(ctx, sy, 0, 0);
+		}
+		wp = TAILQ_NEXT(wp, entry);
+	}
+}
+
 static struct screen *
 window_tree_draw(__unused void *modedata, void *itemdata, u_int sx, u_int sy)
 {
@@ -423,10 +557,21 @@
 		return (NULL);
 
 	screen_init(&s, sx, sy, 0);
-
 	screen_write_start(&ctx, NULL, &s);
 
-	screen_write_preview(&ctx, &wp->base, sx, sy);
+	switch (item->type) {
+	case WINDOW_TREE_NONE:
+		return (0);
+	case WINDOW_TREE_SESSION:
+		window_tree_draw_session(sp, &ctx, sx, sy);
+		break;
+	case WINDOW_TREE_WINDOW:
+		window_tree_draw_window(sp, wlp->window, &ctx, sx, sy);
+		break;
+	case WINDOW_TREE_PANE:
+		screen_write_preview(&ctx, &wp->base, sx, sy);
+		break;
+	}
 
 	screen_write_stop(&ctx);
 	return (&s);