?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Missing Pp, from Tiago Cunha.
[tmux-openbsd.git] / window-copy.c
blobd6e026fff55db56f5cbc630c5431d147be4a3aec
1 /* $OpenBSD$ */
2
3 /*
4 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15 * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16 * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include <sys/types.h>
20
21 #include <stdlib.h>
22 #include <string.h>
23
24 #include "tmux.h"
25
26 struct screen *window_copy_init(struct window_pane *);
27 void window_copy_free(struct window_pane *);
28 void window_copy_resize(struct window_pane *, u_int, u_int);
29 void window_copy_key(struct window_pane *, struct client *, int);
30 int window_copy_key_input(struct window_pane *, int);
31 void window_copy_mouse(
32 struct window_pane *, struct client *, struct mouse_event *);
33
34 void window_copy_redraw_lines(struct window_pane *, u_int, u_int);
35 void window_copy_redraw_screen(struct window_pane *);
36 void window_copy_write_line(
37 struct window_pane *, struct screen_write_ctx *, u_int);
38 void window_copy_write_lines(
39 struct window_pane *, struct screen_write_ctx *, u_int, u_int);
40
41 void window_copy_scroll_to(struct window_pane *, u_int, u_int);
42 int window_copy_search_compare(
43 struct grid *, u_int, u_int, struct grid *, u_int);
44 int window_copy_search_lr(
45 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
46 int window_copy_search_rl(
47 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
48 void window_copy_search_up(struct window_pane *, const char *);
49 void window_copy_search_down(struct window_pane *, const char *);
50 void window_copy_goto_line(struct window_pane *, const char *);
51 void window_copy_update_cursor(struct window_pane *, u_int, u_int);
52 void window_copy_start_selection(struct window_pane *);
53 int window_copy_update_selection(struct window_pane *);
54 void window_copy_copy_selection(struct window_pane *, struct client *);
55 void window_copy_copy_line(
56 struct window_pane *, char **, size_t *, u_int, u_int, u_int);
57 int window_copy_is_space(struct window_pane *, u_int, u_int);
58 u_int window_copy_find_length(struct window_pane *, u_int);
59 void window_copy_cursor_start_of_line(struct window_pane *);
60 void window_copy_cursor_back_to_indentation(struct window_pane *);
61 void window_copy_cursor_end_of_line(struct window_pane *);
62 void window_copy_cursor_left(struct window_pane *);
63 void window_copy_cursor_right(struct window_pane *);
64 void window_copy_cursor_up(struct window_pane *, int);
65 void window_copy_cursor_down(struct window_pane *, int);
66 void window_copy_cursor_next_word(struct window_pane *);
67 void window_copy_cursor_previous_word(struct window_pane *);
68 void window_copy_scroll_up(struct window_pane *, u_int);
69 void window_copy_scroll_down(struct window_pane *, u_int);
70
71 const struct window_mode window_copy_mode = {
72 window_copy_init,
73 window_copy_free,
74 window_copy_resize,
75 window_copy_key,
76 window_copy_mouse,
77 NULL,
78 };
79
80 enum window_copy_input_type {
81 WINDOW_COPY_OFF,
82 WINDOW_COPY_SEARCHUP,
83 WINDOW_COPY_SEARCHDOWN,
84 WINDOW_COPY_GOTOLINE,
85 };
86
87 struct window_copy_mode_data {
88 struct screen screen;
89
90 struct mode_key_data mdata;
91
92 u_int oy;
93
94 u_int selx;
95 u_int sely;
96
97 u_int cx;
98 u_int cy;
99
100 u_int lastcx; /* position in last line with content */
101 u_int lastsx; /* size of last line with content */
102
103 enum window_copy_input_type inputtype;
104 const char *inputprompt;
105 char *inputstr;
106
107 enum window_copy_input_type searchtype;
108 char *searchstr;
109 };
110
111 struct screen *
112 window_copy_init(struct window_pane *wp)
113 {
114 struct window_copy_mode_data *data;
115 struct screen *s;
116 struct screen_write_ctx ctx;
117 u_int i;
118 int keys;
119
120 wp->modedata = data = xmalloc(sizeof *data);
121 data->oy = 0;
122 data->cx = wp->base.cx;
123 data->cy = wp->base.cy;
124
125 data->lastcx = 0;
126 data->lastsx = 0;
127
128 data->inputtype = WINDOW_COPY_OFF;
129 data->inputprompt = NULL;
130 data->inputstr = xstrdup("");
131
132 data->searchtype = WINDOW_COPY_OFF;
133 data->searchstr = NULL;
134
135 s = &data->screen;
136 screen_init(s, screen_size_x(&wp->base), screen_size_y(&wp->base), 0);
137 if (options_get_number(&wp->window->options, "mode-mouse"))
138 s->mode |= MODE_MOUSE;
139
140 keys = options_get_number(&wp->window->options, "mode-keys");
141 if (keys == MODEKEY_EMACS)
142 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
143 else
144 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
145
146 s->cx = data->cx;
147 s->cy = data->cy;
148
149 screen_write_start(&ctx, NULL, s);
150 for (i = 0; i < screen_size_y(s); i++)
151 window_copy_write_line(wp, &ctx, i);
152 screen_write_cursormove(&ctx, data->cx, data->cy);
153 screen_write_stop(&ctx);
154
155 return (s);
156 }
157
158 void
159 window_copy_free(struct window_pane *wp)
160 {
161 struct window_copy_mode_data *data = wp->modedata;
162
163 if (data->searchstr != NULL)
164 xfree(data->searchstr);
165 xfree(data->inputstr);
166
167 screen_free(&data->screen);
168
169 xfree(data);
170 }
171
172 void
173 window_copy_pageup(struct window_pane *wp)
174 {
175 struct window_copy_mode_data *data = wp->modedata;
176 struct screen *s = &data->screen;
177 u_int n;
178
179 n = 1;
180 if (screen_size_y(s) > 2)
181 n = screen_size_y(s) - 2;
182 if (data->oy + n > screen_hsize(&wp->base))
183 data->oy = screen_hsize(&wp->base);
184 else
185 data->oy += n;
186 window_copy_update_selection(wp);
187 window_copy_redraw_screen(wp);
188 }
189
190 void
191 window_copy_resize(struct window_pane *wp, u_int sx, u_int sy)
192 {
193 struct window_copy_mode_data *data = wp->modedata;
194 struct screen *s = &data->screen;
195 struct screen_write_ctx ctx;
196
197 screen_resize(s, sx, sy);
198
199 if (data->cy > sy - 1)
200 data->cy = sy - 1;
201 if (data->cx > sx)
202 data->cx = sx;
203
204 screen_clear_selection(&data->screen);
205
206 screen_write_start(&ctx, NULL, s);
207 window_copy_write_lines(wp, &ctx, 0, screen_size_y(s) - 1);
208 screen_write_stop(&ctx);
209
210 window_copy_redraw_screen(wp);
211 }
212
213 void
214 window_copy_key(struct window_pane *wp, struct client *c, int key)
215 {
216 struct window_copy_mode_data *data = wp->modedata;
217 struct screen *s = &data->screen;
218 u_int n;
219 int keys;
220
221 if (data->inputtype != WINDOW_COPY_OFF) {
222 if (window_copy_key_input(wp, key) != 0)
223 goto input_off;
224 return;
225 }
226
227 switch (mode_key_lookup(&data->mdata, key)) {
228 case MODEKEYCOPY_CANCEL:
229 window_pane_reset_mode(wp);
230 break;
231 case MODEKEYCOPY_LEFT:
232 window_copy_cursor_left(wp);
233 return;
234 case MODEKEYCOPY_RIGHT:
235 window_copy_cursor_right(wp);
236 return;
237 case MODEKEYCOPY_UP:
238 window_copy_cursor_up(wp, 0);
239 return;
240 case MODEKEYCOPY_DOWN:
241 window_copy_cursor_down(wp, 0);
242 return;
243 case MODEKEYCOPY_SCROLLUP:
244 window_copy_cursor_up(wp, 1);
245 break;
246 case MODEKEYCOPY_SCROLLDOWN:
247 window_copy_cursor_down(wp, 1);
248 break;
249 case MODEKEYCOPY_PREVIOUSPAGE:
250 window_copy_pageup(wp);
251 break;
252 case MODEKEYCOPY_NEXTPAGE:
253 n = 1;
254 if (screen_size_y(s) > 2)
255 n = screen_size_y(s) - 2;
256 if (data->oy < n)
257 data->oy = 0;
258 else
259 data->oy -= n;
260 window_copy_update_selection(wp);
261 window_copy_redraw_screen(wp);
262 break;
263 case MODEKEYCOPY_HALFPAGEUP:
264 n = screen_size_y(s) / 2;
265 if (data->oy + n > screen_hsize(&wp->base))
266 data->oy = screen_hsize(&wp->base);
267 else
268 data->oy += n;
269 window_copy_update_selection(wp);
270 window_copy_redraw_screen(wp);
271 break;
272 case MODEKEYCOPY_HALFPAGEDOWN:
273 n = screen_size_y(s) / 2;
274 if (data->oy < n)
275 data->oy = 0;
276 else
277 data->oy -= n;
278 window_copy_update_selection(wp);
279 window_copy_redraw_screen(wp);
280 break;
281 case MODEKEYCOPY_TOPLINE:
282 data->cx = 0;
283 data->cy = 0;
284 window_copy_update_selection(wp);
285 window_copy_redraw_screen(wp);
286 break;
287 case MODEKEYCOPY_MIDDLELINE:
288 data->cx = 0;
289 data->cy = (screen_size_y(s) - 1) / 2;
290 window_copy_update_selection(wp);
291 window_copy_redraw_screen(wp);
292 break;
293 case MODEKEYCOPY_BOTTOMLINE:
294 data->cx = 0;
295 data->cy = screen_size_y(s) - 1;
296 window_copy_update_selection(wp);
297 window_copy_redraw_screen(wp);
298 break;
299 case MODEKEYCOPY_STARTSELECTION:
300 window_copy_start_selection(wp);
301 window_copy_redraw_screen(wp);
302 break;
303 case MODEKEYCOPY_CLEARSELECTION:
304 screen_clear_selection(&data->screen);
305 window_copy_redraw_screen(wp);
306 break;
307 case MODEKEYCOPY_COPYSELECTION:
308 if (c != NULL && c->session != NULL) {
309 window_copy_copy_selection(wp, c);
310 window_pane_reset_mode(wp);
311 }
312 break;
313 case MODEKEYCOPY_STARTOFLINE:
314 window_copy_cursor_start_of_line(wp);
315 break;
316 case MODEKEYCOPY_BACKTOINDENTATION:
317 window_copy_cursor_back_to_indentation(wp);
318 break;
319 case MODEKEYCOPY_ENDOFLINE:
320 window_copy_cursor_end_of_line(wp);
321 break;
322 case MODEKEYCOPY_NEXTWORD:
323 window_copy_cursor_next_word(wp);
324 break;
325 case MODEKEYCOPY_PREVIOUSWORD:
326 window_copy_cursor_previous_word(wp);
327 break;
328 case MODEKEYCOPY_SEARCHUP:
329 data->inputtype = WINDOW_COPY_SEARCHUP;
330 data->inputprompt = "Search Up";
331 goto input_on;
332 case MODEKEYCOPY_SEARCHDOWN:
333 data->inputtype = WINDOW_COPY_SEARCHDOWN;
334 data->inputprompt = "Search Down";
335 goto input_on;
336 case MODEKEYCOPY_SEARCHAGAIN:
337 switch (data->searchtype) {
338 case WINDOW_COPY_OFF:
339 case WINDOW_COPY_GOTOLINE:
340 break;
341 case WINDOW_COPY_SEARCHUP:
342 window_copy_search_up(wp, data->searchstr);
343 break;
344 case WINDOW_COPY_SEARCHDOWN:
345 window_copy_search_down(wp, data->searchstr);
346 break;
347 }
348 break;
349 case MODEKEYCOPY_GOTOLINE:
350 data->inputtype = WINDOW_COPY_GOTOLINE;
351 data->inputprompt = "Goto Line";
352 *data->inputstr = '\0';
353 goto input_on;
354 default:
355 break;
356 }
357
358 return;
359
360 input_on:
361 keys = options_get_number(&wp->window->options, "mode-keys");
362 if (keys == MODEKEY_EMACS)
363 mode_key_init(&data->mdata, &mode_key_tree_emacs_edit);
364 else
365 mode_key_init(&data->mdata, &mode_key_tree_vi_edit);
366
367 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
368 return;
369
370 input_off:
371 keys = options_get_number(&wp->window->options, "mode-keys");
372 if (keys == MODEKEY_EMACS)
373 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
374 else
375 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
376
377 data->inputtype = WINDOW_COPY_OFF;
378 data->inputprompt = NULL;
379
380 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
381 }
382
383 int
384 window_copy_key_input(struct window_pane *wp, int key)
385 {
386 struct window_copy_mode_data *data = wp->modedata;
387 struct screen *s = &data->screen;
388 size_t inputlen;
389
390 switch (mode_key_lookup(&data->mdata, key)) {
391 case MODEKEYEDIT_CANCEL:
392 return (-1);
393 case MODEKEYEDIT_BACKSPACE:
394 inputlen = strlen(data->inputstr);
395 if (inputlen > 0)
396 data->inputstr[inputlen - 1] = '\0';
397 break;
398 case MODEKEYEDIT_DELETELINE:
399 *data->inputstr = '\0';
400 break;
401 case MODEKEYEDIT_ENTER:
402 switch (data->inputtype) {
403 case WINDOW_COPY_OFF:
404 break;
405 case WINDOW_COPY_SEARCHUP:
406 window_copy_search_up(wp, data->inputstr);
407 data->searchtype = data->inputtype;
408 data->searchstr = xstrdup(data->inputstr);
409 break;
410 case WINDOW_COPY_SEARCHDOWN:
411 window_copy_search_down(wp, data->inputstr);
412 data->searchtype = data->inputtype;
413 data->searchstr = xstrdup(data->inputstr);
414 break;
415 case WINDOW_COPY_GOTOLINE:
416 window_copy_goto_line(wp, data->inputstr);
417 *data->inputstr = '\0';
418 break;
419 }
420 return (1);
421 case MODEKEY_OTHER:
422 if (key < 32 || key > 126)
423 break;
424 inputlen = strlen(data->inputstr) + 2;
425
426 data->inputstr = xrealloc(data->inputstr, 1, inputlen);
427 data->inputstr[inputlen - 2] = key;
428 data->inputstr[inputlen - 1] = '\0';
429 break;
430 default:
431 break;
432 }
433
434 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
435 return (0);
436 }
437
438 /* ARGSUSED */
439 void
440 window_copy_mouse(
441 struct window_pane *wp, unused struct client *c, struct mouse_event *m)
442 {
443 struct window_copy_mode_data *data = wp->modedata;
444 struct screen *s = &data->screen;
445
446 if ((m->b & 3) == 3)
447 return;
448 if (m->x >= screen_size_x(s))
449 return;
450 if (m->y >= screen_size_y(s))
451 return;
452
453 window_copy_update_cursor(wp, m->x, m->y);
454 if (window_copy_update_selection(wp))
455 window_copy_redraw_screen(wp);
456 }
457
458 void
459 window_copy_scroll_to(struct window_pane *wp, u_int px, u_int py)
460 {
461 struct window_copy_mode_data *data = wp->modedata;
462 struct screen *s = &wp->base;
463 struct grid *gd = s->grid;
464 u_int offset, gap;
465
466 data->cx = px;
467
468 gap = gd->sy / 4;
469 if (py < gd->sy) {
470 offset = 0;
471 data->cy = py;
472 } else if (py > gd->hsize + gd->sy - gap) {
473 offset = gd->hsize;
474 data->cy = py - gd->hsize;
475 } else {
476 offset = py + gap - gd->sy;
477 data->cy = py - offset;
478 }
479 data->oy = gd->hsize - offset;
480
481 window_copy_redraw_screen(wp);
482 }
483
484 int
485 window_copy_search_compare(
486 struct grid *gd, u_int px, u_int py, struct grid *sgd, u_int spx)
487 {
488 const struct grid_cell *gc, *sgc;
489 const struct grid_utf8 *gu, *sgu;
490
491 gc = grid_peek_cell(gd, px, py);
492 sgc = grid_peek_cell(sgd, spx, 0);
493
494 if ((gc->flags & GRID_FLAG_UTF8) != (sgc->flags & GRID_FLAG_UTF8))
495 return (0);
496
497 if (gc->flags & GRID_FLAG_UTF8) {
498 gu = grid_peek_utf8(gd, px, py);
499 sgu = grid_peek_utf8(sgd, spx, 0);
500 if (grid_utf8_compare(gu, sgu))
501 return (1);
502 } else {
503 if (gc->data == sgc->data)
504 return (1);
505 }
506 return (0);
507 }
508
509 int
510 window_copy_search_lr(struct grid *gd,
511 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
512 {
513 u_int ax, bx, px;
514
515 for (ax = first; ax < last; ax++) {
516 if (ax + sgd->sx >= gd->sx)
517 break;
518 for (bx = 0; bx < sgd->sx; bx++) {
519 px = ax + bx;
520 if (!window_copy_search_compare(gd, px, py, sgd, bx))
521 break;
522 }
523 if (bx == sgd->sx) {
524 *ppx = ax;
525 return (1);
526 }
527 }
528 return (0);
529 }
530
531 int
532 window_copy_search_rl(struct grid *gd,
533 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
534 {
535 u_int ax, bx, px;
536
537 for (ax = last + 1; ax > first; ax--) {
538 if (gd->sx - (ax - 1) < sgd->sx)
539 continue;
540 for (bx = 0; bx < sgd->sx; bx++) {
541 px = ax - 1 + bx;
542 if (!window_copy_search_compare(gd, px, py, sgd, bx))
543 break;
544 }
545 if (bx == sgd->sx) {
546 *ppx = ax - 1;
547 return (1);
548 }
549 }
550 return (0);
551 }
552
553 void
554 window_copy_search_up(struct window_pane *wp, const char *searchstr)
555 {
556 struct window_copy_mode_data *data = wp->modedata;
557 struct screen *s = &wp->base, ss;
558 struct screen_write_ctx ctx;
559 struct grid *gd = s->grid, *sgd;
560 struct grid_cell gc;
561 size_t searchlen;
562 u_int i, last, fx, fy, px;
563 int utf8flag, n, wrapped;
564
565 if (*searchstr == '\0')
566 return;
567 utf8flag = options_get_number(&wp->window->options, "utf8");
568 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
569
570 screen_init(&ss, searchlen, 1, 0);
571 screen_write_start(&ctx, NULL, &ss);
572 memcpy(&gc, &grid_default_cell, sizeof gc);
573 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
574 screen_write_stop(&ctx);
575
576 fx = data->cx;
577 fy = gd->hsize - data->oy + data->cy;
578
579 if (fx == 0) {
580 if (fy == 0)
581 return;
582 fx = gd->sx - 1;
583 fy--;
584 } else
585 fx--;
586 n = wrapped = 0;
587
588 retry:
589 sgd = ss.grid;
590 for (i = fy + 1; i > 0; i--) {
591 last = screen_size_x(s);
592 if (i == fy + 1)
593 last = fx;
594 n = window_copy_search_rl(gd, sgd, &px, i - 1, 0, last);
595 if (n) {
596 window_copy_scroll_to(wp, px, i - 1);
597 break;
598 }
599 }
600 if (!n && !wrapped) {
601 fx = gd->sx - 1;
602 fy = gd->hsize + gd->sy - 1;
603 wrapped = 1;
604 goto retry;
605 }
606
607 screen_free(&ss);
608 }
609
610 void
611 window_copy_search_down(struct window_pane *wp, const char *searchstr)
612 {
613 struct window_copy_mode_data *data = wp->modedata;
614 struct screen *s = &wp->base, ss;
615 struct screen_write_ctx ctx;
616 struct grid *gd = s->grid, *sgd;
617 struct grid_cell gc;
618 size_t searchlen;
619 u_int i, first, fx, fy, px;
620 int utf8flag, n, wrapped;
621
622 if (*searchstr == '\0')
623 return;
624 utf8flag = options_get_number(&wp->window->options, "utf8");
625 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
626
627 screen_init(&ss, searchlen, 1, 0);
628 screen_write_start(&ctx, NULL, &ss);
629 memcpy(&gc, &grid_default_cell, sizeof gc);
630 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
631 screen_write_stop(&ctx);
632
633 fx = data->cx;
634 fy = gd->hsize - data->oy + data->cy;
635
636 if (fx == gd->sx - 1) {
637 if (fy == gd->hsize + gd->sy)
638 return;
639 fx = 0;
640 fy++;
641 } else
642 fx++;
643 n = wrapped = 0;
644
645 retry:
646 sgd = ss.grid;
647 for (i = fy + 1; i < gd->hsize + gd->sy; i++) {
648 first = 0;
649 if (i == fy + 1)
650 first = fx;
651 n = window_copy_search_lr(gd, sgd, &px, i - 1, first, gd->sx);
652 if (n) {
653 window_copy_scroll_to(wp, px, i - 1);
654 break;
655 }
656 }
657 if (!n && !wrapped) {
658 fx = 0;
659 fy = 0;
660 wrapped = 1;
661 goto retry;
662 }
663
664 screen_free(&ss);
665 }
666
667 void
668 window_copy_goto_line(struct window_pane *wp, const char *linestr)
669 {
670 struct window_copy_mode_data *data = wp->modedata;
671 const char *errstr;
672 u_int lineno;
673
674 lineno = strtonum(linestr, 0, screen_hsize(&wp->base), &errstr);
675 if (errstr != NULL)
676 return;
677
678 data->oy = lineno;
679 window_copy_redraw_screen(wp);
680 }
681
682 void
683 window_copy_write_line(
684 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py)
685 {
686 struct window_copy_mode_data *data = wp->modedata;
687 struct screen *s = &data->screen;
688 struct options *oo = &wp->window->options;
689 struct grid_cell gc;
690 char hdr[32];
691 size_t last, xoff = 0, size = 0;
692
693 memcpy(&gc, &grid_default_cell, sizeof gc);
694 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
695 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
696 gc.attr |= options_get_number(oo, "mode-attr");
697
698 last = screen_size_y(s) - 1;
699 if (py == 0) {
700 size = xsnprintf(hdr, sizeof hdr,
701 "[%u/%u]", data->oy, screen_hsize(&wp->base));
702 screen_write_cursormove(ctx, screen_size_x(s) - size, 0);
703 screen_write_puts(ctx, &gc, "%s", hdr);
704 } else if (py == last && data->inputtype != WINDOW_COPY_OFF) {
705 xoff = size = xsnprintf(hdr, sizeof hdr,
706 "%s: %s", data->inputprompt, data->inputstr);
707 screen_write_cursormove(ctx, 0, last);
708 screen_write_puts(ctx, &gc, "%s", hdr);
709 } else
710 size = 0;
711
712 screen_write_cursormove(ctx, xoff, py);
713 screen_write_copy(ctx, &wp->base, xoff, (screen_hsize(&wp->base) -
714 data->oy) + py, screen_size_x(s) - size, 1);
715
716 if (py == data->cy && data->cx == screen_size_x(s)) {
717 memcpy(&gc, &grid_default_cell, sizeof gc);
718 screen_write_cursormove(ctx, screen_size_x(s) - 1, py);
719 screen_write_putc(ctx, &gc, '$');
720 }
721 }
722
723 void
724 window_copy_write_lines(
725 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py, u_int ny)
726 {
727 u_int yy;
728
729 for (yy = py; yy < py + ny; yy++)
730 window_copy_write_line(wp, ctx, py);
731 }
732
733 void
734 window_copy_redraw_lines(struct window_pane *wp, u_int py, u_int ny)
735 {
736 struct window_copy_mode_data *data = wp->modedata;
737 struct screen_write_ctx ctx;
738 u_int i;
739
740 screen_write_start(&ctx, wp, NULL);
741 for (i = py; i < py + ny; i++)
742 window_copy_write_line(wp, &ctx, i);
743 screen_write_cursormove(&ctx, data->cx, data->cy);
744 screen_write_stop(&ctx);
745 }
746
747 void
748 window_copy_redraw_screen(struct window_pane *wp)
749 {
750 struct window_copy_mode_data *data = wp->modedata;
751
752 window_copy_redraw_lines(wp, 0, screen_size_y(&data->screen));
753 }
754
755 void
756 window_copy_update_cursor(struct window_pane *wp, u_int cx, u_int cy)
757 {
758 struct window_copy_mode_data *data = wp->modedata;
759 struct screen *s = &data->screen;
760 struct screen_write_ctx ctx;
761 u_int old_cx, old_cy;
762
763 old_cx = data->cx; old_cy = data->cy;
764 data->cx = cx; data->cy = cy;
765 if (old_cx == screen_size_x(s))
766 window_copy_redraw_lines(wp, old_cy, 1);
767 if (data->cx == screen_size_x(s))
768 window_copy_redraw_lines(wp, data->cy, 1);
769 else {
770 screen_write_start(&ctx, wp, NULL);
771 screen_write_cursormove(&ctx, data->cx, data->cy);
772 screen_write_stop(&ctx);
773 }
774 }
775
776 void
777 window_copy_start_selection(struct window_pane *wp)
778 {
779 struct window_copy_mode_data *data = wp->modedata;
780 struct screen *s = &data->screen;
781
782 data->selx = data->cx;
783 data->sely = screen_hsize(&wp->base) + data->cy - data->oy;
784
785 s->sel.flag = 1;
786 window_copy_update_selection(wp);
787 }
788
789 int
790 window_copy_update_selection(struct window_pane *wp)
791 {
792 struct window_copy_mode_data *data = wp->modedata;
793 struct screen *s = &data->screen;
794 struct options *oo = &wp->window->options;
795 struct grid_cell gc;
796 u_int sx, sy, ty;
797
798 if (!s->sel.flag)
799 return (0);
800
801 /* Set colours. */
802 memcpy(&gc, &grid_default_cell, sizeof gc);
803 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
804 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
805 gc.attr |= options_get_number(oo, "mode-attr");
806
807 /* Find top of screen. */
808 ty = screen_hsize(&wp->base) - data->oy;
809
810 /* Adjust the selection. */
811 sx = data->selx;
812 sy = data->sely;
813 if (sy < ty) { /* above screen */
814 sx = 0;
815 sy = 0;
816 } else if (sy > ty + screen_size_y(s) - 1) { /* below screen */
817 sx = screen_size_x(s) - 1;
818 sy = screen_size_y(s) - 1;
819 } else
820 sy -= ty;
821 sy = screen_hsize(s) + sy;
822
823 screen_set_selection(
824 s, sx, sy, data->cx, screen_hsize(s) + data->cy, &gc);
825 return (1);
826 }
827
828 void
829 window_copy_copy_selection(struct window_pane *wp, struct client *c)
830 {
831 struct window_copy_mode_data *data = wp->modedata;
832 struct screen *s = &data->screen;
833 char *buf;
834 size_t off;
835 u_int i, xx, yy, sx, sy, ex, ey, limit;
836
837 if (!s->sel.flag)
838 return;
839
840 buf = xmalloc(1);
841 off = 0;
842
843 *buf = '\0';
844
845 /*
846 * The selection extends from selx,sely to (adjusted) cx,cy on
847 * the base screen.
848 */
849
850 /* Find start and end. */
851 xx = data->cx;
852 yy = screen_hsize(&wp->base) + data->cy - data->oy;
853 if (yy < data->sely || (yy == data->sely && xx < data->selx)) {
854 sx = xx; sy = yy;
855 ex = data->selx; ey = data->sely;
856 } else {
857 sx = data->selx; sy = data->sely;
858 ex = xx; ey = yy;
859 }
860
861 /* Trim ex to end of line. */
862 xx = window_copy_find_length(wp, ey);
863 if (ex > xx)
864 ex = xx;
865
866 /* Copy the lines. */
867 if (sy == ey)
868 window_copy_copy_line(wp, &buf, &off, sy, sx, ex);
869 else {
870 xx = screen_size_x(s);
871 window_copy_copy_line(wp, &buf, &off, sy, sx, xx);
872 if (ey - sy > 1) {
873 for (i = sy + 1; i < ey; i++)
874 window_copy_copy_line(wp, &buf, &off, i, 0, xx);
875 }
876 window_copy_copy_line(wp, &buf, &off, ey, 0, ex);
877 }
878
879 /* Don't bother if no data. */
880 if (off == 0) {
881 xfree(buf);
882 return;
883 }
884 off--; /* remove final \n */
885
886 /* Add the buffer to the stack. */
887 limit = options_get_number(&c->session->options, "buffer-limit");
888 paste_add(&c->session->buffers, buf, off, limit);
889 }
890
891 void
892 window_copy_copy_line(struct window_pane *wp,
893 char **buf, size_t *off, u_int sy, u_int sx, u_int ex)
894 {
895 struct grid *gd = wp->base.grid;
896 const struct grid_cell *gc;
897 const struct grid_utf8 *gu;
898 struct grid_line *gl;
899 u_int i, xx, wrapped = 0;
900 size_t size;
901
902 if (sx > ex)
903 return;
904
905 /*
906 * Work out if the line was wrapped at the screen edge and all of it is
907 * on screen.
908 */
909 gl = &gd->linedata[sy];
910 if (gl->flags & GRID_LINE_WRAPPED && gl->cellsize <= gd->sx)
911 wrapped = 1;
912
913 /* If the line was wrapped, don't strip spaces (use the full length). */
914 if (wrapped)
915 xx = gl->cellsize;
916 else
917 xx = window_copy_find_length(wp, sy);
918 if (ex > xx)
919 ex = xx;
920 if (sx > xx)
921 sx = xx;
922
923 if (sx < ex) {
924 for (i = sx; i < ex; i++) {
925 gc = grid_peek_cell(gd, i, sy);
926 if (gc->flags & GRID_FLAG_PADDING)
927 continue;
928 if (!(gc->flags & GRID_FLAG_UTF8)) {
929 *buf = xrealloc(*buf, 1, (*off) + 1);
930 (*buf)[(*off)++] = gc->data;
931 } else {
932 gu = grid_peek_utf8(gd, i, sy);
933 size = grid_utf8_size(gu);
934 *buf = xrealloc(*buf, 1, (*off) + size);
935 *off += grid_utf8_copy(gu, *buf + *off, size);
936 }
937 }
938 }
939
940 /* Only add a newline if the line wasn't wrapped. */
941 if (!wrapped) {
942 *buf = xrealloc(*buf, 1, (*off) + 1);
943 (*buf)[(*off)++] = '\n';
944 }
945 }
946
947 int
948 window_copy_is_space(struct window_pane *wp, u_int px, u_int py)
949 {
950 const struct grid_cell *gc;
951 const char *spaces = " -_@";
952
953 gc = grid_peek_cell(wp->base.grid, px, py);
954 if (gc->flags & (GRID_FLAG_PADDING|GRID_FLAG_UTF8))
955 return (0);
956 if (gc->data == 0x00 || gc->data == 0x7f)
957 return (0);
958 return (strchr(spaces, gc->data) != NULL);
959 }
960
961 u_int
962 window_copy_find_length(struct window_pane *wp, u_int py)
963 {
964 const struct grid_cell *gc;
965 u_int px;
966
967 /*
968 * If the pane has been resized, its grid can contain old overlong
969 * lines. grid_peek_cell does not allow accessing cells beyond the
970 * width of the grid, and screen_write_copy treats them as spaces, so
971 * ignore them here too.
972 */
973 px = wp->base.grid->linedata[py].cellsize;
974 if (px > screen_size_x(&wp->base))
975 px = screen_size_x(&wp->base);
976 while (px > 0) {
977 gc = grid_peek_cell(wp->base.grid, px - 1, py);
978 if (gc->flags & GRID_FLAG_UTF8)
979 break;
980 if (gc->data != ' ')
981 break;
982 px--;
983 }
984 return (px);
985 }
986
987 void
988 window_copy_cursor_start_of_line(struct window_pane *wp)
989 {
990 struct window_copy_mode_data *data = wp->modedata;
991
992 window_copy_update_cursor(wp, 0, data->cy);
993 if (window_copy_update_selection(wp))
994 window_copy_redraw_lines(wp, data->cy, 1);
995 }
996
997 void
998 window_copy_cursor_back_to_indentation(struct window_pane *wp)
999 {
1000 struct window_copy_mode_data *data = wp->modedata;
1001 u_int px, py, xx;
1002 const struct grid_cell *gc;
1003
1004 px = 0;
1005 py = screen_hsize(&wp->base) + data->cy - data->oy;
1006 xx = window_copy_find_length(wp, py);
1007
1008 /*
1009 * Don't use window_copy_is_space because that treats some word
1010 * delimiters as spaces.
1011 */
1012 while (px < xx) {
1013 gc = grid_peek_cell(wp->base.grid, px, py);
1014 if (gc->flags & GRID_FLAG_UTF8)
1015 break;
1016 if (gc->data != ' ')
1017 break;
1018 px++;
1019 }
1020
1021 window_copy_update_cursor(wp, px, data->cy);
1022 if (window_copy_update_selection(wp))
1023 window_copy_redraw_lines(wp, data->cy, 1);
1024 }
1025
1026 void
1027 window_copy_cursor_end_of_line(struct window_pane *wp)
1028 {
1029 struct window_copy_mode_data *data = wp->modedata;
1030 u_int px, py;
1031
1032 py = screen_hsize(&wp->base) + data->cy - data->oy;
1033 px = window_copy_find_length(wp, py);
1034
1035 window_copy_update_cursor(wp, px, data->cy);
1036 if (window_copy_update_selection(wp))
1037 window_copy_redraw_lines(wp, data->cy, 1);
1038 }
1039
1040 void
1041 window_copy_cursor_left(struct window_pane *wp)
1042 {
1043 struct window_copy_mode_data *data = wp->modedata;
1044
1045 if (data->cx == 0) {
1046 window_copy_cursor_up(wp, 0);
1047 window_copy_cursor_end_of_line(wp);
1048 } else {
1049 window_copy_update_cursor(wp, data->cx - 1, data->cy);
1050 if (window_copy_update_selection(wp))
1051 window_copy_redraw_lines(wp, data->cy, 1);
1052 }
1053 }
1054
1055 void
1056 window_copy_cursor_right(struct window_pane *wp)
1057 {
1058 struct window_copy_mode_data *data = wp->modedata;
1059 u_int px, py;
1060
1061 py = screen_hsize(&wp->base) + data->cy - data->oy;
1062 px = window_copy_find_length(wp, py);
1063
1064 if (data->cx >= px) {
1065 window_copy_cursor_start_of_line(wp);
1066 window_copy_cursor_down(wp, 0);
1067 } else {
1068 window_copy_update_cursor(wp, data->cx + 1, data->cy);
1069 if (window_copy_update_selection(wp))
1070 window_copy_redraw_lines(wp, data->cy, 1);
1071 }
1072 }
1073
1074 void
1075 window_copy_cursor_up(struct window_pane *wp, int scroll_only)
1076 {
1077 struct window_copy_mode_data *data = wp->modedata;
1078 u_int ox, oy, px, py;
1079
1080 oy = screen_hsize(&wp->base) + data->cy - data->oy;
1081 ox = window_copy_find_length(wp, oy);
1082 if (ox != 0) {
1083 data->lastcx = data->cx;
1084 data->lastsx = ox;
1085 }
1086
1087 data->cx = data->lastcx;
1088 if (scroll_only || data->cy == 0) {
1089 window_copy_scroll_down(wp, 1);
1090 if (scroll_only)
1091 window_copy_redraw_lines(wp, data->cy, 2);
1092 } else {
1093 window_copy_update_cursor(wp, data->cx, data->cy - 1);
1094 if (window_copy_update_selection(wp))
1095 window_copy_redraw_lines(wp, data->cy, 2);
1096 }
1097
1098 py = screen_hsize(&wp->base) + data->cy - data->oy;
1099 px = window_copy_find_length(wp, py);
1100 if (data->cx >= data->lastsx || data->cx > px)
1101 window_copy_cursor_end_of_line(wp);
1102 }
1103
1104 void
1105 window_copy_cursor_down(struct window_pane *wp, int scroll_only)
1106 {
1107 struct window_copy_mode_data *data = wp->modedata;
1108 struct screen *s = &data->screen;
1109 u_int ox, oy, px, py;
1110
1111 oy = screen_hsize(&wp->base) + data->cy - data->oy;
1112 ox = window_copy_find_length(wp, oy);
1113 if (ox != 0) {
1114 data->lastcx = data->cx;
1115 data->lastsx = ox;
1116 }
1117
1118 data->cx = data->lastcx;
1119 if (scroll_only || data->cy == screen_size_y(s) - 1) {
1120 window_copy_scroll_up(wp, 1);
1121 if (scroll_only && data->cy > 0)
1122 window_copy_redraw_lines(wp, data->cy - 1, 2);
1123 } else {
1124 window_copy_update_cursor(wp, data->cx, data->cy + 1);
1125 if (window_copy_update_selection(wp))
1126 window_copy_redraw_lines(wp, data->cy - 1, 2);
1127 }
1128
1129 py = screen_hsize(&wp->base) + data->cy - data->oy;
1130 px = window_copy_find_length(wp, py);
1131 if (data->cx >= data->lastsx || data->cx > px)
1132 window_copy_cursor_end_of_line(wp);
1133 }
1134
1135 void
1136 window_copy_cursor_next_word(struct window_pane *wp)
1137 {
1138 struct window_copy_mode_data *data = wp->modedata;
1139 struct screen *s = &data->screen;
1140 u_int px, py, xx, skip;
1141
1142 px = data->cx;
1143 py = screen_hsize(&wp->base) + data->cy - data->oy;
1144 xx = window_copy_find_length(wp, py);
1145
1146 skip = 1;
1147 if (px < xx) {
1148 /* If currently on a space, skip space. */
1149 if (window_copy_is_space(wp, px, py))
1150 skip = 0;
1151 } else
1152 skip = 0;
1153 for (;;) {
1154 if (px >= xx) {
1155 if (skip) {
1156 px = xx;
1157 break;
1158 }
1159
1160 while (px >= xx) {
1161 if (data->cy == screen_size_y(s) - 1) {
1162 if (data->oy == 0)
1163 goto out;
1164 }
1165
1166 px = 0;
1167 window_copy_cursor_down(wp, 0);
1168
1169 py =screen_hsize(
1170 &wp->base) + data->cy - data->oy;
1171 xx = window_copy_find_length(wp, py);
1172 }
1173 }
1174
1175 if (skip) {
1176 /* Currently skipping non-space (until space). */
1177 if (window_copy_is_space(wp, px, py))
1178 break;
1179 } else {
1180 /* Currently skipping space (until non-space). */
1181 if (!window_copy_is_space(wp, px, py))
1182 skip = 1;
1183 }
1184
1185 px++;
1186 }
1187
1188 out:
1189 window_copy_update_cursor(wp, px, data->cy);
1190 if (window_copy_update_selection(wp))
1191 window_copy_redraw_lines(wp, data->cy, 1);
1192 }
1193
1194 /* Move to the previous place where a word begins. */
1195 void
1196 window_copy_cursor_previous_word(struct window_pane *wp)
1197 {
1198 struct window_copy_mode_data *data = wp->modedata;
1199 u_int px, py;
1200
1201 px = data->cx;
1202 py = screen_hsize(&wp->base) + data->cy - data->oy;
1203
1204 /* Move back to the previous word character. */
1205 for (;;) {
1206 if (px > 0) {
1207 px--;
1208 if (!window_copy_is_space(wp, px, py))
1209 break;
1210 } else {
1211 if (data->cy == 0 &&
1212 (screen_hsize(&wp->base) == 0 ||
1213 data->oy >= screen_hsize(&wp->base) - 1))
1214 goto out;
1215 window_copy_cursor_up(wp, 0);
1216
1217 py = screen_hsize(
1218 &wp->base) + data->cy - data->oy;
1219 px = window_copy_find_length(wp, py);
1220 }
1221 }
1222
1223 /* Move back to the beginning of this word. */
1224 while (px > 0 && !window_copy_is_space(wp, px - 1, py))
1225 px--;
1226
1227 out:
1228 window_copy_update_cursor(wp, px, data->cy);
1229 if (window_copy_update_selection(wp))
1230 window_copy_redraw_lines(wp, data->cy, 1);
1231 }
1232
1233 void
1234 window_copy_scroll_up(struct window_pane *wp, u_int ny)
1235 {
1236 struct window_copy_mode_data *data = wp->modedata;
1237 struct screen *s = &data->screen;
1238 struct screen_write_ctx ctx;
1239
1240 if (data->oy < ny)
1241 ny = data->oy;
1242 if (ny == 0)
1243 return;
1244 data->oy -= ny;
1245 window_copy_update_selection(wp);
1246
1247 screen_write_start(&ctx, wp, NULL);
1248 screen_write_cursormove(&ctx, 0, 0);
1249 screen_write_deleteline(&ctx, ny);
1250 window_copy_write_lines(wp, &ctx, screen_size_y(s) - ny, ny);
1251 window_copy_write_line(wp, &ctx, 0);
1252 if (screen_size_y(s) > 1)
1253 window_copy_write_line(wp, &ctx, 1);
1254 if (screen_size_y(s) > 3)
1255 window_copy_write_line(wp, &ctx, screen_size_y(s) - 2);
1256 if (s->sel.flag && screen_size_y(s) > ny)
1257 window_copy_write_line(wp, &ctx, screen_size_y(s) - ny - 1);
1258 screen_write_cursormove(&ctx, data->cx, data->cy);
1259 screen_write_stop(&ctx);
1260 }
1261
1262 void
1263 window_copy_scroll_down(struct window_pane *wp, u_int ny)
1264 {
1265 struct window_copy_mode_data *data = wp->modedata;
1266 struct screen *s = &data->screen;
1267 struct screen_write_ctx ctx;
1268
1269 if (ny > screen_hsize(&wp->base))
1270 return;
1271
1272 if (data->oy > screen_hsize(&wp->base) - ny)
1273 ny = screen_hsize(&wp->base) - data->oy;
1274 if (ny == 0)
1275 return;
1276 data->oy += ny;
1277 window_copy_update_selection(wp);
1278
1279 screen_write_start(&ctx, wp, NULL);
1280 screen_write_cursormove(&ctx, 0, 0);
1281 screen_write_insertline(&ctx, ny);
1282 window_copy_write_lines(wp, &ctx, 0, ny);
1283 if (s->sel.flag && screen_size_y(s) > ny)
1284 window_copy_write_line(wp, &ctx, ny);
1285 else if (ny == 1) /* nuke position */
1286 window_copy_write_line(wp, &ctx, 1);
1287 screen_write_cursormove(&ctx, data->cx, data->cy);
1288 screen_write_stop(&ctx);
1289 }