?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Add three new copy-mode commands - select-line, copy-line,
[tmux-openbsd.git] / window-copy.c
blob4ee3c3e389bb1dc7e151f451b8fcddf3bc33a5c4
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 session *, int);
30 int window_copy_key_input(struct window_pane *, int);
31 int window_copy_key_numeric_prefix(struct window_pane *, int);
32 void window_copy_mouse(
33 struct window_pane *, struct session *, struct mouse_event *);
34
35 void window_copy_redraw_lines(struct window_pane *, u_int, u_int);
36 void window_copy_redraw_screen(struct window_pane *);
37 void window_copy_write_line(
38 struct window_pane *, struct screen_write_ctx *, u_int);
39 void window_copy_write_lines(
40 struct window_pane *, struct screen_write_ctx *, u_int, u_int);
41
42 void window_copy_scroll_to(struct window_pane *, u_int, u_int);
43 int window_copy_search_compare(
44 struct grid *, u_int, u_int, struct grid *, u_int);
45 int window_copy_search_lr(
46 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
47 int window_copy_search_rl(
48 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
49 void window_copy_search_up(struct window_pane *, const char *);
50 void window_copy_search_down(struct window_pane *, const char *);
51 void window_copy_goto_line(struct window_pane *, const char *);
52 void window_copy_update_cursor(struct window_pane *, u_int, u_int);
53 void window_copy_start_selection(struct window_pane *);
54 int window_copy_update_selection(struct window_pane *);
55 void window_copy_copy_selection(struct window_pane *);
56 void window_copy_clear_selection(struct window_pane *);
57 void window_copy_copy_line(
58 struct window_pane *, char **, size_t *, u_int, u_int, u_int);
59 int window_copy_in_set(struct window_pane *, u_int, u_int, const char *);
60 u_int window_copy_find_length(struct window_pane *, u_int);
61 void window_copy_cursor_start_of_line(struct window_pane *);
62 void window_copy_cursor_back_to_indentation(struct window_pane *);
63 void window_copy_cursor_end_of_line(struct window_pane *);
64 void window_copy_cursor_left(struct window_pane *);
65 void window_copy_cursor_right(struct window_pane *);
66 void window_copy_cursor_up(struct window_pane *, int);
67 void window_copy_cursor_down(struct window_pane *, int);
68 void window_copy_cursor_jump(struct window_pane *);
69 void window_copy_cursor_jump_back(struct window_pane *);
70 void window_copy_cursor_next_word(struct window_pane *, const char *);
71 void window_copy_cursor_next_word_end(struct window_pane *, const char *);
72 void window_copy_cursor_previous_word(struct window_pane *, const char *);
73 void window_copy_scroll_up(struct window_pane *, u_int);
74 void window_copy_scroll_down(struct window_pane *, u_int);
75 void window_copy_rectangle_toggle(struct window_pane *);
76
77 const struct window_mode window_copy_mode = {
78 window_copy_init,
79 window_copy_free,
80 window_copy_resize,
81 window_copy_key,
82 window_copy_mouse,
83 NULL,
84 };
85
86 enum window_copy_input_type {
87 WINDOW_COPY_OFF,
88 WINDOW_COPY_NUMERICPREFIX,
89 WINDOW_COPY_SEARCHUP,
90 WINDOW_COPY_SEARCHDOWN,
91 WINDOW_COPY_JUMPFORWARD,
92 WINDOW_COPY_JUMPBACK,
93 WINDOW_COPY_GOTOLINE,
94 };
95
96 /*
97 * Copy-mode's visible screen (the "screen" field) is filled from one of
98 * two sources: the original contents of the pane (used when we
99 * actually enter via the "copy-mode" command, to copy the contents of
100 * the current pane), or else a series of lines containing the output
101 * from an output-writing tmux command (such as any of the "show-*" or
102 * "list-*" commands).
103 *
104 * In either case, the full content of the copy-mode grid is pointed at
105 * by the "backing" field, and is copied into "screen" as needed (that
106 * is, when scrolling occurs). When copy-mode is backed by a pane,
107 * backing points directly at that pane's screen structure (&wp->base);
108 * when backed by a list of output-lines from a command, it points at
109 * a newly-allocated screen structure (which is deallocated when the
110 * mode ends).
111 */
112 struct window_copy_mode_data {
113 struct screen screen;
114
115 struct screen *backing;
116 int backing_written; /* backing display has started */
117
118 struct mode_key_data mdata;
119
120 u_int oy;
121
122 u_int selx;
123 u_int sely;
124
125 u_int rectflag; /* are we in rectangle copy mode? */
126
127 u_int cx;
128 u_int cy;
129
130 u_int lastcx; /* position in last line with content */
131 u_int lastsx; /* size of last line with content */
132
133 enum window_copy_input_type inputtype;
134 const char *inputprompt;
135 char *inputstr;
136
137 u_int numprefix;
138
139 enum window_copy_input_type searchtype;
140 char *searchstr;
141
142 enum window_copy_input_type jumptype;
143 char jumpchar;
144 };
145
146 struct screen *
147 window_copy_init(struct window_pane *wp)
148 {
149 struct window_copy_mode_data *data;
150 struct screen *s;
151 int keys;
152
153 wp->modedata = data = xmalloc(sizeof *data);
154 data->oy = 0;
155 data->cx = 0;
156 data->cy = 0;
157
158 data->lastcx = 0;
159 data->lastsx = 0;
160
161 data->backing_written = 0;
162
163 data->rectflag = 0;
164
165 data->inputtype = WINDOW_COPY_OFF;
166 data->inputprompt = NULL;
167 data->inputstr = xstrdup("");
168 data->numprefix = 0;
169
170 data->searchtype = WINDOW_COPY_OFF;
171 data->searchstr = NULL;
172
173 wp->flags |= PANE_FREEZE;
174 if (wp->fd != -1)
175 bufferevent_disable(wp->event, EV_READ|EV_WRITE);
176
177 data->jumptype = WINDOW_COPY_OFF;
178 data->jumpchar = '\0';
179
180 s = &data->screen;
181 screen_init(s, screen_size_x(&wp->base), screen_size_y(&wp->base), 0);
182 if (options_get_number(&wp->window->options, "mode-mouse"))
183 s->mode |= MODE_MOUSE_STANDARD;
184
185 keys = options_get_number(&wp->window->options, "mode-keys");
186 if (keys == MODEKEY_EMACS)
187 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
188 else
189 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
190
191 data->backing = NULL;
192
193 return (s);
194 }
195
196 void
197 window_copy_init_from_pane(struct window_pane *wp)
198 {
199 struct window_copy_mode_data *data = wp->modedata;
200 struct screen *s = &data->screen;
201 struct screen_write_ctx ctx;
202 u_int i;
203
204 if (wp->mode != &window_copy_mode)
205 fatalx("not in copy mode");
206
207 data->backing = &wp->base;
208 data->cx = data->backing->cx;
209 data->cy = data->backing->cy;
210
211 s->cx = data->cx;
212 s->cy = data->cy;
213
214 screen_write_start(&ctx, NULL, s);
215 for (i = 0; i < screen_size_y(s); i++)
216 window_copy_write_line(wp, &ctx, i);
217 screen_write_cursormove(&ctx, data->cx, data->cy);
218 screen_write_stop(&ctx);
219 }
220
221 void
222 window_copy_init_for_output(struct window_pane *wp)
223 {
224 struct window_copy_mode_data *data = wp->modedata;
225
226 data->backing = xmalloc(sizeof *data->backing);
227 screen_init(data->backing, screen_size_x(&wp->base),
228 screen_size_y(&wp->base), UINT_MAX);
229 data->backing->mode &= ~MODE_WRAP;
230 }
231
232 void
233 window_copy_free(struct window_pane *wp)
234 {
235 struct window_copy_mode_data *data = wp->modedata;
236
237 wp->flags &= ~PANE_FREEZE;
238 if (wp->fd != -1)
239 bufferevent_enable(wp->event, EV_READ|EV_WRITE);
240
241 if (data->searchstr != NULL)
242 xfree(data->searchstr);
243 xfree(data->inputstr);
244
245 if (data->backing != &wp->base) {
246 screen_free(data->backing);
247 xfree(data->backing);
248 }
249 screen_free(&data->screen);
250
251 xfree(data);
252 }
253
254 void
255 window_copy_add(struct window_pane *wp, const char *fmt, ...)
256 {
257 va_list ap;
258
259 va_start(ap, fmt);
260 window_copy_vadd(wp, fmt, ap);
261 va_end(ap);
262 }
263
264 void
265 window_copy_vadd(struct window_pane *wp, const char *fmt, va_list ap)
266 {
267 struct window_copy_mode_data *data = wp->modedata;
268 struct screen *backing = data->backing;
269 struct screen_write_ctx back_ctx, ctx;
270 struct grid_cell gc;
271 int utf8flag;
272 u_int old_hsize;
273
274 if (backing == &wp->base)
275 return;
276
277 utf8flag = options_get_number(&wp->window->options, "utf8");
278 memcpy(&gc, &grid_default_cell, sizeof gc);
279
280 old_hsize = screen_hsize(data->backing);
281 screen_write_start(&back_ctx, NULL, backing);
282 if (data->backing_written) {
283 /*
284 * On the second or later line, do a CRLF before writing
285 * (so it's on a new line).
286 */
287 screen_write_carriagereturn(&back_ctx);
288 screen_write_linefeed(&back_ctx, 0);
289 } else
290 data->backing_written = 1;
291 screen_write_vnputs(&back_ctx, 0, &gc, utf8flag, fmt, ap);
292 screen_write_stop(&back_ctx);
293
294 data->oy += screen_hsize(data->backing) - old_hsize;
295
296 screen_write_start(&ctx, wp, &data->screen);
297
298 /*
299 * If the history has changed, draw the top line.
300 * (If there's any history at all, it has changed.)
301 */
302 if (screen_hsize(data->backing))
303 window_copy_redraw_lines(wp, 0, 1);
304
305 /* Write the line, if it's visible. */
306 if (backing->cy + data->oy < screen_size_y(backing))
307 window_copy_redraw_lines(wp, backing->cy, 1);
308
309 screen_write_stop(&ctx);
310 }
311
312 void
313 window_copy_pageup(struct window_pane *wp)
314 {
315 struct window_copy_mode_data *data = wp->modedata;
316 struct screen *s = &data->screen;
317 u_int n;
318
319 n = 1;
320 if (screen_size_y(s) > 2)
321 n = screen_size_y(s) - 2;
322 if (data->oy + n > screen_hsize(data->backing))
323 data->oy = screen_hsize(data->backing);
324 else
325 data->oy += n;
326 window_copy_update_selection(wp);
327 window_copy_redraw_screen(wp);
328 }
329
330 void
331 window_copy_resize(struct window_pane *wp, u_int sx, u_int sy)
332 {
333 struct window_copy_mode_data *data = wp->modedata;
334 struct screen *s = &data->screen;
335 struct screen_write_ctx ctx;
336
337 screen_resize(s, sx, sy);
338 if (data->backing != &wp->base)
339 screen_resize(data->backing, sx, sy);
340
341 if (data->cy > sy - 1)
342 data->cy = sy - 1;
343 if (data->cx > sx)
344 data->cx = sx;
345 if (data->oy > screen_hsize(data->backing))
346 data->oy = screen_hsize(data->backing);
347
348 window_copy_clear_selection(wp);
349
350 screen_write_start(&ctx, NULL, s);
351 window_copy_write_lines(wp, &ctx, 0, screen_size_y(s) - 1);
352 screen_write_stop(&ctx);
353
354 window_copy_redraw_screen(wp);
355 }
356
357 void
358 window_copy_key(struct window_pane *wp, struct session *sess, int key)
359 {
360 const char *word_separators;
361 struct window_copy_mode_data *data = wp->modedata;
362 struct screen *s = &data->screen;
363 u_int n, np;
364 int keys;
365 enum mode_key_cmd cmd;
366
367 np = data->numprefix;
368 if (np == 0)
369 np = 1;
370
371 if (data->inputtype == WINDOW_COPY_JUMPFORWARD ||
372 data->inputtype == WINDOW_COPY_JUMPBACK) {
373 /* Ignore keys with modifiers. */
374 if ((key & KEYC_MASK_MOD) == 0) {
375 data->jumpchar = key;
376 if (data->inputtype == WINDOW_COPY_JUMPFORWARD) {
377 for (; np != 0; np--)
378 window_copy_cursor_jump(wp);
379 } else {
380 for (; np != 0; np--)
381 window_copy_cursor_jump_back(wp);
382 }
383 }
384 data->jumptype = data->inputtype;
385 data->inputtype = WINDOW_COPY_OFF;
386 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
387 return;
388 } else if (data->inputtype == WINDOW_COPY_NUMERICPREFIX) {
389 if (window_copy_key_numeric_prefix(wp, key) == 0)
390 return;
391 data->inputtype = WINDOW_COPY_OFF;
392 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
393 } else if (data->inputtype != WINDOW_COPY_OFF) {
394 if (window_copy_key_input(wp, key) != 0)
395 goto input_off;
396 return;
397 }
398
399 cmd = mode_key_lookup(&data->mdata, key);
400 switch (cmd) {
401 case MODEKEYCOPY_CANCEL:
402 window_pane_reset_mode(wp);
403 return;
404 case MODEKEYCOPY_LEFT:
405 for (; np != 0; np--)
406 window_copy_cursor_left(wp);
407 break;
408 case MODEKEYCOPY_RIGHT:
409 for (; np != 0; np--)
410 window_copy_cursor_right(wp);
411 break;
412 case MODEKEYCOPY_UP:
413 for (; np != 0; np--)
414 window_copy_cursor_up(wp, 0);
415 break;
416 case MODEKEYCOPY_DOWN:
417 for (; np != 0; np--)
418 window_copy_cursor_down(wp, 0);
419 break;
420 case MODEKEYCOPY_SCROLLUP:
421 for (; np != 0; np--)
422 window_copy_cursor_up(wp, 1);
423 break;
424 case MODEKEYCOPY_SCROLLDOWN:
425 for (; np != 0; np--)
426 window_copy_cursor_down(wp, 1);
427 break;
428 case MODEKEYCOPY_PREVIOUSPAGE:
429 for (; np != 0; np--)
430 window_copy_pageup(wp);
431 break;
432 case MODEKEYCOPY_NEXTPAGE:
433 n = 1;
434 if (screen_size_y(s) > 2)
435 n = screen_size_y(s) - 2;
436 for (; np != 0; np--) {
437 if (data->oy < n)
438 data->oy = 0;
439 else
440 data->oy -= n;
441 }
442 window_copy_update_selection(wp);
443 window_copy_redraw_screen(wp);
444 break;
445 case MODEKEYCOPY_HALFPAGEUP:
446 n = screen_size_y(s) / 2;
447 for (; np != 0; np--) {
448 if (data->oy + n > screen_hsize(data->backing))
449 data->oy = screen_hsize(data->backing);
450 else
451 data->oy += n;
452 }
453 window_copy_update_selection(wp);
454 window_copy_redraw_screen(wp);
455 break;
456 case MODEKEYCOPY_HALFPAGEDOWN:
457 n = screen_size_y(s) / 2;
458 for (; np != 0; np--) {
459 if (data->oy < n)
460 data->oy = 0;
461 else
462 data->oy -= n;
463 }
464 window_copy_update_selection(wp);
465 window_copy_redraw_screen(wp);
466 break;
467 case MODEKEYCOPY_TOPLINE:
468 data->cx = 0;
469 data->cy = 0;
470 window_copy_update_selection(wp);
471 window_copy_redraw_screen(wp);
472 break;
473 case MODEKEYCOPY_MIDDLELINE:
474 data->cx = 0;
475 data->cy = (screen_size_y(s) - 1) / 2;
476 window_copy_update_selection(wp);
477 window_copy_redraw_screen(wp);
478 break;
479 case MODEKEYCOPY_BOTTOMLINE:
480 data->cx = 0;
481 data->cy = screen_size_y(s) - 1;
482 window_copy_update_selection(wp);
483 window_copy_redraw_screen(wp);
484 break;
485 case MODEKEYCOPY_HISTORYTOP:
486 data->cx = 0;
487 data->cy = 0;
488 data->oy = screen_hsize(data->backing);
489 window_copy_update_selection(wp);
490 window_copy_redraw_screen(wp);
491 break;
492 case MODEKEYCOPY_HISTORYBOTTOM:
493 data->cx = 0;
494 data->cy = screen_size_y(s) - 1;
495 data->oy = 0;
496 window_copy_update_selection(wp);
497 window_copy_redraw_screen(wp);
498 break;
499 case MODEKEYCOPY_STARTSELECTION:
500 window_copy_start_selection(wp);
501 window_copy_redraw_screen(wp);
502 break;
503 case MODEKEYCOPY_COPYLINE:
504 case MODEKEYCOPY_SELECTLINE:
505 window_copy_cursor_start_of_line(wp);
506 /* FALLTHROUGH */
507 case MODEKEYCOPY_COPYENDOFLINE:
508 window_copy_start_selection(wp);
509 for (; np > 1; np--)
510 window_copy_cursor_down(wp, 0);
511 window_copy_cursor_end_of_line(wp);
512 window_copy_redraw_screen(wp);
513
514 /* If a copy command then copy the selection and exit. */
515 if (sess != NULL &&
516 (cmd == MODEKEYCOPY_COPYLINE ||
517 cmd == MODEKEYCOPY_COPYENDOFLINE)) {
518 window_copy_copy_selection(wp);
519 window_pane_reset_mode(wp);
520 return;
521 }
522 break;
523 case MODEKEYCOPY_CLEARSELECTION:
524 window_copy_clear_selection(wp);
525 window_copy_redraw_screen(wp);
526 break;
527 case MODEKEYCOPY_COPYSELECTION:
528 if (sess != NULL) {
529 window_copy_copy_selection(wp);
530 window_pane_reset_mode(wp);
531 return;
532 }
533 break;
534 case MODEKEYCOPY_STARTOFLINE:
535 window_copy_cursor_start_of_line(wp);
536 break;
537 case MODEKEYCOPY_BACKTOINDENTATION:
538 window_copy_cursor_back_to_indentation(wp);
539 break;
540 case MODEKEYCOPY_ENDOFLINE:
541 window_copy_cursor_end_of_line(wp);
542 break;
543 case MODEKEYCOPY_NEXTSPACE:
544 for (; np != 0; np--)
545 window_copy_cursor_next_word(wp, " ");
546 break;
547 case MODEKEYCOPY_NEXTSPACEEND:
548 for (; np != 0; np--)
549 window_copy_cursor_next_word_end(wp, " ");
550 break;
551 case MODEKEYCOPY_NEXTWORD:
552 word_separators =
553 options_get_string(&wp->window->options, "word-separators");
554 for (; np != 0; np--)
555 window_copy_cursor_next_word(wp, word_separators);
556 break;
557 case MODEKEYCOPY_NEXTWORDEND:
558 word_separators =
559 options_get_string(&wp->window->options, "word-separators");
560 for (; np != 0; np--)
561 window_copy_cursor_next_word_end(wp, word_separators);
562 break;
563 case MODEKEYCOPY_PREVIOUSSPACE:
564 for (; np != 0; np--)
565 window_copy_cursor_previous_word(wp, " ");
566 break;
567 case MODEKEYCOPY_PREVIOUSWORD:
568 word_separators =
569 options_get_string(&wp->window->options, "word-separators");
570 for (; np != 0; np--)
571 window_copy_cursor_previous_word(wp, word_separators);
572 break;
573 case MODEKEYCOPY_JUMP:
574 data->inputtype = WINDOW_COPY_JUMPFORWARD;
575 data->inputprompt = "Jump Forward";
576 *data->inputstr = '\0';
577 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
578 return; /* skip numprefix reset */
579 case MODEKEYCOPY_JUMPAGAIN:
580 if (data->jumptype == WINDOW_COPY_JUMPFORWARD) {
581 for (; np != 0; np--)
582 window_copy_cursor_jump(wp);
583 } else if (data->jumptype == WINDOW_COPY_JUMPBACK) {
584 for (; np != 0; np--)
585 window_copy_cursor_jump_back(wp);
586 }
587 break;
588 case MODEKEYCOPY_JUMPREVERSE:
589 if (data->jumptype == WINDOW_COPY_JUMPFORWARD) {
590 for (; np != 0; np--)
591 window_copy_cursor_jump_back(wp);
592 } else if (data->jumptype == WINDOW_COPY_JUMPBACK) {
593 for (; np != 0; np--)
594 window_copy_cursor_jump(wp);
595 }
596 break;
597 case MODEKEYCOPY_JUMPBACK:
598 data->inputtype = WINDOW_COPY_JUMPBACK;
599 data->inputprompt = "Jump Back";
600 *data->inputstr = '\0';
601 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
602 return; /* skip numprefix reset */
603 case MODEKEYCOPY_SEARCHUP:
604 data->inputtype = WINDOW_COPY_SEARCHUP;
605 data->inputprompt = "Search Up";
606 goto input_on;
607 case MODEKEYCOPY_SEARCHDOWN:
608 data->inputtype = WINDOW_COPY_SEARCHDOWN;
609 data->inputprompt = "Search Down";
610 goto input_on;
611 case MODEKEYCOPY_SEARCHAGAIN:
612 case MODEKEYCOPY_SEARCHREVERSE:
613 switch (data->searchtype) {
614 case WINDOW_COPY_OFF:
615 case WINDOW_COPY_GOTOLINE:
616 case WINDOW_COPY_JUMPFORWARD:
617 case WINDOW_COPY_JUMPBACK:
618 case WINDOW_COPY_NUMERICPREFIX:
619 break;
620 case WINDOW_COPY_SEARCHUP:
621 if (cmd == MODEKEYCOPY_SEARCHAGAIN) {
622 for (; np != 0; np--) {
623 window_copy_search_up(
624 wp, data->searchstr);
625 }
626 } else {
627 for (; np != 0; np--) {
628 window_copy_search_down(
629 wp, data->searchstr);
630 }
631 }
632 break;
633 case WINDOW_COPY_SEARCHDOWN:
634 if (cmd == MODEKEYCOPY_SEARCHAGAIN) {
635 for (; np != 0; np--) {
636 window_copy_search_down(
637 wp, data->searchstr);
638 }
639 } else {
640 for (; np != 0; np--) {
641 window_copy_search_up(
642 wp, data->searchstr);
643 }
644 }
645 break;
646 }
647 break;
648 case MODEKEYCOPY_GOTOLINE:
649 data->inputtype = WINDOW_COPY_GOTOLINE;
650 data->inputprompt = "Goto Line";
651 *data->inputstr = '\0';
652 goto input_on;
653 case MODEKEYCOPY_STARTNUMBERPREFIX:
654 key &= KEYC_MASK_KEY;
655 if (key >= '0' && key <= '9') {
656 data->inputtype = WINDOW_COPY_NUMERICPREFIX;
657 data->numprefix = 0;
658 window_copy_key_numeric_prefix(wp, key);
659 return;
660 }
661 break;
662 case MODEKEYCOPY_RECTANGLETOGGLE:
663 window_copy_rectangle_toggle(wp);
664 break;
665 default:
666 break;
667 }
668
669 data->numprefix = 0;
670 return;
671
672 input_on:
673 keys = options_get_number(&wp->window->options, "mode-keys");
674 if (keys == MODEKEY_EMACS)
675 mode_key_init(&data->mdata, &mode_key_tree_emacs_edit);
676 else
677 mode_key_init(&data->mdata, &mode_key_tree_vi_edit);
678
679 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
680 return;
681
682 input_off:
683 keys = options_get_number(&wp->window->options, "mode-keys");
684 if (keys == MODEKEY_EMACS)
685 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
686 else
687 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
688
689 data->inputtype = WINDOW_COPY_OFF;
690 data->inputprompt = NULL;
691
692 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
693 }
694
695 int
696 window_copy_key_input(struct window_pane *wp, int key)
697 {
698 struct window_copy_mode_data *data = wp->modedata;
699 struct screen *s = &data->screen;
700 size_t inputlen;
701 u_int np;
702
703 switch (mode_key_lookup(&data->mdata, key)) {
704 case MODEKEYEDIT_CANCEL:
705 data->numprefix = 0;
706 return (-1);
707 case MODEKEYEDIT_BACKSPACE:
708 inputlen = strlen(data->inputstr);
709 if (inputlen > 0)
710 data->inputstr[inputlen - 1] = '\0';
711 break;
712 case MODEKEYEDIT_DELETELINE:
713 *data->inputstr = '\0';
714 break;
715 case MODEKEYEDIT_ENTER:
716 np = data->numprefix;
717 if (np == 0)
718 np = 1;
719
720 switch (data->inputtype) {
721 case WINDOW_COPY_OFF:
722 case WINDOW_COPY_JUMPFORWARD:
723 case WINDOW_COPY_JUMPBACK:
724 case WINDOW_COPY_NUMERICPREFIX:
725 break;
726 case WINDOW_COPY_SEARCHUP:
727 for (; np != 0; np--)
728 window_copy_search_up(wp, data->inputstr);
729 data->searchtype = data->inputtype;
730 data->searchstr = xstrdup(data->inputstr);
731 break;
732 case WINDOW_COPY_SEARCHDOWN:
733 for (; np != 0; np--)
734 window_copy_search_down(wp, data->inputstr);
735 data->searchtype = data->inputtype;
736 data->searchstr = xstrdup(data->inputstr);
737 break;
738 case WINDOW_COPY_GOTOLINE:
739 window_copy_goto_line(wp, data->inputstr);
740 *data->inputstr = '\0';
741 break;
742 }
743 data->numprefix = 0;
744 return (1);
745 case MODEKEY_OTHER:
746 if (key < 32 || key > 126)
747 break;
748 inputlen = strlen(data->inputstr) + 2;
749
750 data->inputstr = xrealloc(data->inputstr, 1, inputlen);
751 data->inputstr[inputlen - 2] = key;
752 data->inputstr[inputlen - 1] = '\0';
753 break;
754 default:
755 break;
756 }
757
758 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
759 return (0);
760 }
761
762 int
763 window_copy_key_numeric_prefix(struct window_pane *wp, int key)
764 {
765 struct window_copy_mode_data *data = wp->modedata;
766 struct screen *s = &data->screen;
767
768 key &= KEYC_MASK_KEY;
769 if (key < '0' || key > '9')
770 return 1;
771
772 if (data->numprefix >= 100) /* no more than three digits */
773 return 0;
774 data->numprefix = data->numprefix * 10 + key - '0';
775
776 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
777 return 0;
778 }
779
780 /* ARGSUSED */
781 void
782 window_copy_mouse(
783 struct window_pane *wp, struct session *sess, struct mouse_event *m)
784 {
785 struct window_copy_mode_data *data = wp->modedata;
786 struct screen *s = &data->screen;
787 u_int i, old_cy;
788
789 if (m->x >= screen_size_x(s))
790 return;
791 if (m->y >= screen_size_y(s))
792 return;
793
794 /* If mouse wheel (buttons 4 and 5), scroll. */
795 if ((m->b & MOUSE_45)) {
796 if ((m->b & MOUSE_BUTTON) == MOUSE_1) {
797 for (i = 0; i < 5; i++)
798 window_copy_cursor_up(wp, 0);
799 } else if ((m->b & MOUSE_BUTTON) == MOUSE_2) {
800 old_cy = data->cy;
801 for (i = 0; i < 5; i++)
802 window_copy_cursor_down(wp, 0);
803 if (old_cy == data->cy)
804 goto reset_mode;
805 }
806 return;
807 }
808
809 /*
810 * If already reading motion, move the cursor while buttons are still
811 * pressed, or stop the selection on their release.
812 */
813 if (s->mode & MODE_MOUSE_BUTTON) {
814 if ((m->b & MOUSE_BUTTON) != MOUSE_UP) {
815 window_copy_update_cursor(wp, m->x, m->y);
816 if (window_copy_update_selection(wp))
817 window_copy_redraw_screen(wp);
818 return;
819 }
820 goto reset_mode;
821 }
822
823 /* Otherwise if other buttons pressed, start selection and motion. */
824 if ((m->b & MOUSE_BUTTON) != MOUSE_UP) {
825 s->mode &= ~MODE_MOUSE_STANDARD;
826 s->mode |= MODE_MOUSE_BUTTON;
827
828 window_copy_update_cursor(wp, m->x, m->y);
829 window_copy_start_selection(wp);
830 window_copy_redraw_screen(wp);
831 }
832
833 return;
834
835 reset_mode:
836 s->mode &= ~MODE_MOUSE_BUTTON;
837 s->mode |= MODE_MOUSE_STANDARD;
838 if (sess != NULL) {
839 window_copy_copy_selection(wp);
840 window_pane_reset_mode(wp);
841 }
842 }
843
844 void
845 window_copy_scroll_to(struct window_pane *wp, u_int px, u_int py)
846 {
847 struct window_copy_mode_data *data = wp->modedata;
848 struct grid *gd = data->backing->grid;
849 u_int offset, gap;
850
851 data->cx = px;
852
853 gap = gd->sy / 4;
854 if (py < gd->sy) {
855 offset = 0;
856 data->cy = py;
857 } else if (py > gd->hsize + gd->sy - gap) {
858 offset = gd->hsize;
859 data->cy = py - gd->hsize;
860 } else {
861 offset = py + gap - gd->sy;
862 data->cy = py - offset;
863 }
864 data->oy = gd->hsize - offset;
865
866 window_copy_update_selection(wp);
867 window_copy_redraw_screen(wp);
868 }
869
870 int
871 window_copy_search_compare(
872 struct grid *gd, u_int px, u_int py, struct grid *sgd, u_int spx)
873 {
874 const struct grid_cell *gc, *sgc;
875 const struct grid_utf8 *gu, *sgu;
876
877 gc = grid_peek_cell(gd, px, py);
878 sgc = grid_peek_cell(sgd, spx, 0);
879
880 if ((gc->flags & GRID_FLAG_UTF8) != (sgc->flags & GRID_FLAG_UTF8))
881 return (0);
882
883 if (gc->flags & GRID_FLAG_UTF8) {
884 gu = grid_peek_utf8(gd, px, py);
885 sgu = grid_peek_utf8(sgd, spx, 0);
886 if (grid_utf8_compare(gu, sgu))
887 return (1);
888 } else {
889 if (gc->data == sgc->data)
890 return (1);
891 }
892 return (0);
893 }
894
895 int
896 window_copy_search_lr(struct grid *gd,
897 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
898 {
899 u_int ax, bx, px;
900
901 for (ax = first; ax < last; ax++) {
902 if (ax + sgd->sx >= gd->sx)
903 break;
904 for (bx = 0; bx < sgd->sx; bx++) {
905 px = ax + bx;
906 if (!window_copy_search_compare(gd, px, py, sgd, bx))
907 break;
908 }
909 if (bx == sgd->sx) {
910 *ppx = ax;
911 return (1);
912 }
913 }
914 return (0);
915 }
916
917 int
918 window_copy_search_rl(struct grid *gd,
919 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
920 {
921 u_int ax, bx, px;
922
923 for (ax = last + 1; ax > first; ax--) {
924 if (gd->sx - (ax - 1) < sgd->sx)
925 continue;
926 for (bx = 0; bx < sgd->sx; bx++) {
927 px = ax - 1 + bx;
928 if (!window_copy_search_compare(gd, px, py, sgd, bx))
929 break;
930 }
931 if (bx == sgd->sx) {
932 *ppx = ax - 1;
933 return (1);
934 }
935 }
936 return (0);
937 }
938
939 void
940 window_copy_search_up(struct window_pane *wp, const char *searchstr)
941 {
942 struct window_copy_mode_data *data = wp->modedata;
943 struct screen *s = data->backing, ss;
944 struct screen_write_ctx ctx;
945 struct grid *gd = s->grid, *sgd;
946 struct grid_cell gc;
947 size_t searchlen;
948 u_int i, last, fx, fy, px;
949 int utf8flag, n, wrapped;
950
951 if (*searchstr == '\0')
952 return;
953 utf8flag = options_get_number(&wp->window->options, "utf8");
954 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
955
956 screen_init(&ss, searchlen, 1, 0);
957 screen_write_start(&ctx, NULL, &ss);
958 memcpy(&gc, &grid_default_cell, sizeof gc);
959 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
960 screen_write_stop(&ctx);
961
962 fx = data->cx;
963 fy = gd->hsize - data->oy + data->cy;
964
965 if (fx == 0) {
966 if (fy == 0)
967 return;
968 fx = gd->sx - 1;
969 fy--;
970 } else
971 fx--;
972 n = wrapped = 0;
973
974 retry:
975 sgd = ss.grid;
976 for (i = fy + 1; i > 0; i--) {
977 last = screen_size_x(s);
978 if (i == fy + 1)
979 last = fx;
980 n = window_copy_search_rl(gd, sgd, &px, i - 1, 0, last);
981 if (n) {
982 window_copy_scroll_to(wp, px, i - 1);
983 break;
984 }
985 }
986 if (!n && !wrapped) {
987 fx = gd->sx - 1;
988 fy = gd->hsize + gd->sy - 1;
989 wrapped = 1;
990 goto retry;
991 }
992
993 screen_free(&ss);
994 }
995
996 void
997 window_copy_search_down(struct window_pane *wp, const char *searchstr)
998 {
999 struct window_copy_mode_data *data = wp->modedata;
1000 struct screen *s = data->backing, ss;
1001 struct screen_write_ctx ctx;
1002 struct grid *gd = s->grid, *sgd;
1003 struct grid_cell gc;
1004 size_t searchlen;
1005 u_int i, first, fx, fy, px;
1006 int utf8flag, n, wrapped;
1007
1008 if (*searchstr == '\0')
1009 return;
1010 utf8flag = options_get_number(&wp->window->options, "utf8");
1011 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
1012
1013 screen_init(&ss, searchlen, 1, 0);
1014 screen_write_start(&ctx, NULL, &ss);
1015 memcpy(&gc, &grid_default_cell, sizeof gc);
1016 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
1017 screen_write_stop(&ctx);
1018
1019 fx = data->cx;
1020 fy = gd->hsize - data->oy + data->cy;
1021
1022 if (fx == gd->sx - 1) {
1023 if (fy == gd->hsize + gd->sy)
1024 return;
1025 fx = 0;
1026 fy++;
1027 } else
1028 fx++;
1029 n = wrapped = 0;
1030
1031 retry:
1032 sgd = ss.grid;
1033 for (i = fy + 1; i < gd->hsize + gd->sy; i++) {
1034 first = 0;
1035 if (i == fy + 1)
1036 first = fx;
1037 n = window_copy_search_lr(gd, sgd, &px, i - 1, first, gd->sx);
1038 if (n) {
1039 window_copy_scroll_to(wp, px, i - 1);
1040 break;
1041 }
1042 }
1043 if (!n && !wrapped) {
1044 fx = 0;
1045 fy = 0;
1046 wrapped = 1;
1047 goto retry;
1048 }
1049
1050 screen_free(&ss);
1051 }
1052
1053 void
1054 window_copy_goto_line(struct window_pane *wp, const char *linestr)
1055 {
1056 struct window_copy_mode_data *data = wp->modedata;
1057 const char *errstr;
1058 u_int lineno;
1059
1060 lineno = strtonum(linestr, 0, screen_hsize(data->backing), &errstr);
1061 if (errstr != NULL)
1062 return;
1063
1064 data->oy = lineno;
1065 window_copy_update_selection(wp);
1066 window_copy_redraw_screen(wp);
1067 }
1068
1069 void
1070 window_copy_write_line(
1071 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py)
1072 {
1073 struct window_copy_mode_data *data = wp->modedata;
1074 struct screen *s = &data->screen;
1075 struct options *oo = &wp->window->options;
1076 struct grid_cell gc;
1077 char hdr[32];
1078 size_t last, xoff = 0, size = 0;
1079
1080 memcpy(&gc, &grid_default_cell, sizeof gc);
1081 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
1082 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
1083 gc.attr |= options_get_number(oo, "mode-attr");
1084
1085 last = screen_size_y(s) - 1;
1086 if (py == 0) {
1087 size = xsnprintf(hdr, sizeof hdr,
1088 "[%u/%u]", data->oy, screen_hsize(data->backing));
1089 if (size > screen_size_x(s))
1090 size = screen_size_x(s);
1091 screen_write_cursormove(ctx, screen_size_x(s) - size, 0);
1092 screen_write_puts(ctx, &gc, "%s", hdr);
1093 } else if (py == last && data->inputtype != WINDOW_COPY_OFF) {
1094 if (data->inputtype == WINDOW_COPY_NUMERICPREFIX) {
1095 xoff = size = xsnprintf(hdr, sizeof hdr,
1096 "Repeat: %u", data->numprefix);
1097 } else {
1098 xoff = size = xsnprintf(hdr, sizeof hdr,
1099 "%s: %s", data->inputprompt, data->inputstr);
1100 }
1101 screen_write_cursormove(ctx, 0, last);
1102 screen_write_puts(ctx, &gc, "%s", hdr);
1103 } else
1104 size = 0;
1105
1106 screen_write_cursormove(ctx, xoff, py);
1107 screen_write_copy(ctx, data->backing, xoff,
1108 (screen_hsize(data->backing) - data->oy) + py,
1109 screen_size_x(s) - size, 1);
1110
1111 if (py == data->cy && data->cx == screen_size_x(s)) {
1112 memcpy(&gc, &grid_default_cell, sizeof gc);
1113 screen_write_cursormove(ctx, screen_size_x(s) - 1, py);
1114 screen_write_putc(ctx, &gc, '$');
1115 }
1116 }
1117
1118 void
1119 window_copy_write_lines(
1120 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py, u_int ny)
1121 {
1122 u_int yy;
1123
1124 for (yy = py; yy < py + ny; yy++)
1125 window_copy_write_line(wp, ctx, py);
1126 }
1127
1128 void
1129 window_copy_redraw_lines(struct window_pane *wp, u_int py, u_int ny)
1130 {
1131 struct window_copy_mode_data *data = wp->modedata;
1132 struct screen_write_ctx ctx;
1133 u_int i;
1134
1135 screen_write_start(&ctx, wp, NULL);
1136 for (i = py; i < py + ny; i++)
1137 window_copy_write_line(wp, &ctx, i);
1138 screen_write_cursormove(&ctx, data->cx, data->cy);
1139 screen_write_stop(&ctx);
1140 }
1141
1142 void
1143 window_copy_redraw_screen(struct window_pane *wp)
1144 {
1145 struct window_copy_mode_data *data = wp->modedata;
1146
1147 window_copy_redraw_lines(wp, 0, screen_size_y(&data->screen));
1148 }
1149
1150 void
1151 window_copy_update_cursor(struct window_pane *wp, u_int cx, u_int cy)
1152 {
1153 struct window_copy_mode_data *data = wp->modedata;
1154 struct screen *s = &data->screen;
1155 struct screen_write_ctx ctx;
1156 u_int old_cx, old_cy;
1157
1158 old_cx = data->cx; old_cy = data->cy;
1159 data->cx = cx; data->cy = cy;
1160 if (old_cx == screen_size_x(s))
1161 window_copy_redraw_lines(wp, old_cy, 1);
1162 if (data->cx == screen_size_x(s))
1163 window_copy_redraw_lines(wp, data->cy, 1);
1164 else {
1165 screen_write_start(&ctx, wp, NULL);
1166 screen_write_cursormove(&ctx, data->cx, data->cy);
1167 screen_write_stop(&ctx);
1168 }
1169 }
1170
1171 void
1172 window_copy_start_selection(struct window_pane *wp)
1173 {
1174 struct window_copy_mode_data *data = wp->modedata;
1175 struct screen *s = &data->screen;
1176
1177 data->selx = data->cx;
1178 data->sely = screen_hsize(data->backing) + data->cy - data->oy;
1179
1180 s->sel.flag = 1;
1181 window_copy_update_selection(wp);
1182 }
1183
1184 int
1185 window_copy_update_selection(struct window_pane *wp)
1186 {
1187 struct window_copy_mode_data *data = wp->modedata;
1188 struct screen *s = &data->screen;
1189 struct options *oo = &wp->window->options;
1190 struct grid_cell gc;
1191 u_int sx, sy, ty, cy;
1192
1193 if (!s->sel.flag)
1194 return (0);
1195
1196 /* Set colours. */
1197 memcpy(&gc, &grid_default_cell, sizeof gc);
1198 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
1199 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
1200 gc.attr |= options_get_number(oo, "mode-attr");
1201
1202 /* Find top of screen. */
1203 ty = screen_hsize(data->backing) - data->oy;
1204
1205 /* Adjust the selection. */
1206 sx = data->selx;
1207 sy = data->sely;
1208 if (sy < ty) { /* above screen */
1209 if (!data->rectflag)
1210 sx = 0;
1211 sy = 0;
1212 } else if (sy > ty + screen_size_y(s) - 1) { /* below screen */
1213 if (!data->rectflag)
1214 sx = screen_size_x(s) - 1;
1215 sy = screen_size_y(s) - 1;
1216 } else
1217 sy -= ty;
1218 sy = screen_hsize(s) + sy;
1219
1220 screen_set_selection(s,
1221 sx, sy, data->cx, screen_hsize(s) + data->cy, data->rectflag, &gc);
1222
1223 if (data->rectflag) {
1224 /*
1225 * Can't rely on the caller to redraw the right lines for
1226 * rectangle selection - find the highest line and the number
1227 * of lines, and redraw just past that in both directions
1228 */
1229 cy = data->cy;
1230 if (sy < cy)
1231 window_copy_redraw_lines(wp, sy, cy - sy + 1);
1232 else
1233 window_copy_redraw_lines(wp, cy, sy - cy + 1);
1234 }
1235
1236 return (1);
1237 }
1238
1239 void
1240 window_copy_copy_selection(struct window_pane *wp)
1241 {
1242 struct window_copy_mode_data *data = wp->modedata;
1243 struct screen *s = &data->screen;
1244 char *buf;
1245 size_t off;
1246 u_int i, xx, yy, sx, sy, ex, ey, limit;
1247 u_int firstsx, lastex, restex, restsx;
1248 int keys;
1249
1250 if (!s->sel.flag)
1251 return;
1252
1253 buf = xmalloc(1);
1254 off = 0;
1255
1256 *buf = '\0';
1257
1258 /*
1259 * The selection extends from selx,sely to (adjusted) cx,cy on
1260 * the base screen.
1261 */
1262
1263 /* Find start and end. */
1264 xx = data->cx;
1265 yy = screen_hsize(data->backing) + data->cy - data->oy;
1266 if (yy < data->sely || (yy == data->sely && xx < data->selx)) {
1267 sx = xx; sy = yy;
1268 ex = data->selx; ey = data->sely;
1269 } else {
1270 sx = data->selx; sy = data->sely;
1271 ex = xx; ey = yy;
1272 }
1273
1274 /* Trim ex to end of line. */
1275 xx = window_copy_find_length(wp, ey);
1276 if (ex > xx)
1277 ex = xx;
1278
1279 /*
1280 * Deal with rectangle-copy if necessary; four situations: start of
1281 * first line (firstsx), end of last line (lastex), start (restsx) and
1282 * end (restex) of all other lines.
1283 */
1284 xx = screen_size_x(s);
1285
1286 /*
1287 * Behave according to mode-keys. If it is emacs, copy like emacs,
1288 * keeping the top-left-most character, and dropping the
1289 * bottom-right-most, regardless of copy direction. If it is vi, also
1290 * keep bottom-right-most character.
1291 */
1292 keys = options_get_number(&wp->window->options, "mode-keys");
1293 if (data->rectflag) {
1294 /*
1295 * Need to ignore the column with the cursor in it, which for
1296 * rectangular copy means knowing which side the cursor is on.
1297 */
1298 if (data->selx < data->cx) {
1299 /* Selection start is on the left. */
1300 if (keys == MODEKEY_EMACS) {
1301 lastex = data->cx;
1302 restex = data->cx;
1303 }
1304 else {
1305 lastex = data->cx + 1;
1306 restex = data->cx + 1;
1307 }
1308 firstsx = data->selx;
1309 restsx = data->selx;
1310 } else {
1311 /* Cursor is on the left. */
1312 lastex = data->selx + 1;
1313 restex = data->selx + 1;
1314 firstsx = data->cx;
1315 restsx = data->cx;
1316 }
1317 } else {
1318 if (keys == MODEKEY_EMACS)
1319 lastex = ex;
1320 else
1321 lastex = ex + 1;
1322 restex = xx;
1323 firstsx = sx;
1324 restsx = 0;
1325 }
1326
1327 /* Copy the lines. */
1328 if (sy == ey)
1329 window_copy_copy_line(wp, &buf, &off, sy, firstsx, lastex);
1330 else {
1331 window_copy_copy_line(wp, &buf, &off, sy, firstsx, restex);
1332 if (ey - sy > 1) {
1333 for (i = sy + 1; i < ey; i++) {
1334 window_copy_copy_line(
1335 wp, &buf, &off, i, restsx, restex);
1336 }
1337 }
1338 window_copy_copy_line(wp, &buf, &off, ey, restsx, lastex);
1339 }
1340
1341 /* Don't bother if no data. */
1342 if (off == 0) {
1343 xfree(buf);
1344 return;
1345 }
1346 off--; /* remove final \n */
1347
1348 /* Add the buffer to the stack. */
1349 limit = options_get_number(&global_options, "buffer-limit");
1350 paste_add(&global_buffers, buf, off, limit);
1351 }
1352
1353 void
1354 window_copy_copy_line(struct window_pane *wp,
1355 char **buf, size_t *off, u_int sy, u_int sx, u_int ex)
1356 {
1357 struct window_copy_mode_data *data = wp->modedata;
1358 struct grid *gd = data->backing->grid;
1359 const struct grid_cell *gc;
1360 const struct grid_utf8 *gu;
1361 struct grid_line *gl;
1362 u_int i, xx, wrapped = 0;
1363 size_t size;
1364
1365 if (sx > ex)
1366 return;
1367
1368 /*
1369 * Work out if the line was wrapped at the screen edge and all of it is
1370 * on screen.
1371 */
1372 gl = &gd->linedata[sy];
1373 if (gl->flags & GRID_LINE_WRAPPED && gl->cellsize <= gd->sx)
1374 wrapped = 1;
1375
1376 /* If the line was wrapped, don't strip spaces (use the full length). */
1377 if (wrapped)
1378 xx = gl->cellsize;
1379 else
1380 xx = window_copy_find_length(wp, sy);
1381 if (ex > xx)
1382 ex = xx;
1383 if (sx > xx)
1384 sx = xx;
1385
1386 if (sx < ex) {
1387 for (i = sx; i < ex; i++) {
1388 gc = grid_peek_cell(gd, i, sy);
1389 if (gc->flags & GRID_FLAG_PADDING)
1390 continue;
1391 if (!(gc->flags & GRID_FLAG_UTF8)) {
1392 *buf = xrealloc(*buf, 1, (*off) + 1);
1393 (*buf)[(*off)++] = gc->data;
1394 } else {
1395 gu = grid_peek_utf8(gd, i, sy);
1396 size = grid_utf8_size(gu);
1397 *buf = xrealloc(*buf, 1, (*off) + size);
1398 *off += grid_utf8_copy(gu, *buf + *off, size);
1399 }
1400 }
1401 }
1402
1403 /* Only add a newline if the line wasn't wrapped. */
1404 if (!wrapped || ex != xx) {
1405 *buf = xrealloc(*buf, 1, (*off) + 1);
1406 (*buf)[(*off)++] = '\n';
1407 }
1408 }
1409
1410 void
1411 window_copy_clear_selection(struct window_pane *wp)
1412 {
1413 struct window_copy_mode_data *data = wp->modedata;
1414 u_int px, py;
1415
1416 screen_clear_selection(&data->screen);
1417
1418 py = screen_hsize(data->backing) + data->cy - data->oy;
1419 px = window_copy_find_length(wp, py);
1420 if (data->cx > px)
1421 window_copy_update_cursor(wp, px, data->cy);
1422 }
1423
1424 int
1425 window_copy_in_set(struct window_pane *wp, u_int px, u_int py, const char *set)
1426 {
1427 struct window_copy_mode_data *data = wp->modedata;
1428 const struct grid_cell *gc;
1429
1430 gc = grid_peek_cell(data->backing->grid, px, py);
1431 if (gc->flags & (GRID_FLAG_PADDING|GRID_FLAG_UTF8))
1432 return (0);
1433 if (gc->data == 0x00 || gc->data == 0x7f)
1434 return (0);
1435 return (strchr(set, gc->data) != NULL);
1436 }
1437
1438 u_int
1439 window_copy_find_length(struct window_pane *wp, u_int py)
1440 {
1441 struct window_copy_mode_data *data = wp->modedata;
1442 struct screen *s = data->backing;
1443 const struct grid_cell *gc;
1444 u_int px;
1445
1446 /*
1447 * If the pane has been resized, its grid can contain old overlong
1448 * lines. grid_peek_cell does not allow accessing cells beyond the
1449 * width of the grid, and screen_write_copy treats them as spaces, so
1450 * ignore them here too.
1451 */
1452 px = s->grid->linedata[py].cellsize;
1453 if (px > screen_size_x(s))
1454 px = screen_size_x(s);
1455 while (px > 0) {
1456 gc = grid_peek_cell(s->grid, px - 1, py);
1457 if (gc->flags & GRID_FLAG_UTF8)
1458 break;
1459 if (gc->data != ' ')
1460 break;
1461 px--;
1462 }
1463 return (px);
1464 }
1465
1466 void
1467 window_copy_cursor_start_of_line(struct window_pane *wp)
1468 {
1469 struct window_copy_mode_data *data = wp->modedata;
1470 struct screen *back_s = data->backing;
1471 struct grid *gd = back_s->grid;
1472 u_int py;
1473
1474 if (data->cx == 0) {
1475 py = screen_hsize(back_s) + data->cy - data->oy;
1476 while (py > 0 && gd->linedata[py-1].flags & GRID_LINE_WRAPPED) {
1477 window_copy_cursor_up(wp, 0);
1478 py = screen_hsize(back_s) + data->cy - data->oy;
1479 }
1480 }
1481 window_copy_update_cursor(wp, 0, data->cy);
1482 if (window_copy_update_selection(wp))
1483 window_copy_redraw_lines(wp, data->cy, 1);
1484 }
1485
1486 void
1487 window_copy_cursor_back_to_indentation(struct window_pane *wp)
1488 {
1489 struct window_copy_mode_data *data = wp->modedata;
1490 u_int px, py, xx;
1491 const struct grid_cell *gc;
1492
1493 px = 0;
1494 py = screen_hsize(data->backing) + data->cy - data->oy;
1495 xx = window_copy_find_length(wp, py);
1496
1497 while (px < xx) {
1498 gc = grid_peek_cell(data->backing->grid, px, py);
1499 if (gc->flags & GRID_FLAG_UTF8)
1500 break;
1501 if (gc->data != ' ')
1502 break;
1503 px++;
1504 }
1505
1506 window_copy_update_cursor(wp, px, data->cy);
1507 if (window_copy_update_selection(wp))
1508 window_copy_redraw_lines(wp, data->cy, 1);
1509 }
1510
1511 void
1512 window_copy_cursor_end_of_line(struct window_pane *wp)
1513 {
1514 struct window_copy_mode_data *data = wp->modedata;
1515 struct screen *back_s = data->backing;
1516 struct grid *gd = back_s->grid;
1517 u_int px, py;
1518
1519 py = screen_hsize(back_s) + data->cy - data->oy;
1520 px = window_copy_find_length(wp, py);
1521
1522 if (data->cx == px) {
1523 if (data->screen.sel.flag && data->rectflag)
1524 px = screen_size_x(back_s);
1525 if (gd->linedata[py].flags & GRID_LINE_WRAPPED) {
1526 while (py < gd->sy + gd->hsize &&
1527 gd->linedata[py].flags & GRID_LINE_WRAPPED) {
1528 window_copy_cursor_down(wp, 0);
1529 py = screen_hsize(back_s)
1530 + data->cy - data->oy;
1531 }
1532 px = window_copy_find_length(wp, py);
1533 }
1534 }
1535 window_copy_update_cursor(wp, px, data->cy);
1536
1537 if (window_copy_update_selection(wp))
1538 window_copy_redraw_lines(wp, data->cy, 1);
1539 }
1540
1541 void
1542 window_copy_cursor_left(struct window_pane *wp)
1543 {
1544 struct window_copy_mode_data *data = wp->modedata;
1545
1546 if (data->cx == 0) {
1547 window_copy_cursor_up(wp, 0);
1548 window_copy_cursor_end_of_line(wp);
1549 } else {
1550 window_copy_update_cursor(wp, data->cx - 1, data->cy);
1551 if (window_copy_update_selection(wp))
1552 window_copy_redraw_lines(wp, data->cy, 1);
1553 }
1554 }
1555
1556 void
1557 window_copy_cursor_right(struct window_pane *wp)
1558 {
1559 struct window_copy_mode_data *data = wp->modedata;
1560 u_int px, py;
1561
1562 if (data->screen.sel.flag && data->rectflag)
1563 px = screen_size_x(&data->screen);
1564 else {
1565 py = screen_hsize(data->backing) + data->cy - data->oy;
1566 px = window_copy_find_length(wp, py);
1567 }
1568
1569 if (data->cx >= px) {
1570 window_copy_cursor_start_of_line(wp);
1571 window_copy_cursor_down(wp, 0);
1572 } else {
1573 window_copy_update_cursor(wp, data->cx + 1, data->cy);
1574 if (window_copy_update_selection(wp))
1575 window_copy_redraw_lines(wp, data->cy, 1);
1576 }
1577 }
1578
1579 void
1580 window_copy_cursor_up(struct window_pane *wp, int scroll_only)
1581 {
1582 struct window_copy_mode_data *data = wp->modedata;
1583 struct screen *s = &data->screen;
1584 u_int ox, oy, px, py;
1585
1586 oy = screen_hsize(data->backing) + data->cy - data->oy;
1587 ox = window_copy_find_length(wp, oy);
1588 if (ox != 0) {
1589 data->lastcx = data->cx;
1590 data->lastsx = ox;
1591 }
1592
1593 data->cx = data->lastcx;
1594 if (scroll_only || data->cy == 0) {
1595 window_copy_scroll_down(wp, 1);
1596 if (scroll_only) {
1597 if (data->cy == screen_size_y(s) - 1)
1598 window_copy_redraw_lines(wp, data->cy, 1);
1599 else
1600 window_copy_redraw_lines(wp, data->cy, 2);
1601 }
1602 } else {
1603 window_copy_update_cursor(wp, data->cx, data->cy - 1);
1604 if (window_copy_update_selection(wp)) {
1605 if (data->cy == screen_size_y(s) - 1)
1606 window_copy_redraw_lines(wp, data->cy, 1);
1607 else
1608 window_copy_redraw_lines(wp, data->cy, 2);
1609 }
1610 }
1611
1612 if (!data->screen.sel.flag || !data->rectflag) {
1613 py = screen_hsize(data->backing) + data->cy - data->oy;
1614 px = window_copy_find_length(wp, py);
1615 if ((data->cx >= data->lastsx && data->cx != px) ||
1616 data->cx > px)
1617 window_copy_cursor_end_of_line(wp);
1618 }
1619 }
1620
1621 void
1622 window_copy_cursor_down(struct window_pane *wp, int scroll_only)
1623 {
1624 struct window_copy_mode_data *data = wp->modedata;
1625 struct screen *s = &data->screen;
1626 u_int ox, oy, px, py;
1627
1628 oy = screen_hsize(data->backing) + data->cy - data->oy;
1629 ox = window_copy_find_length(wp, oy);
1630 if (ox != 0) {
1631 data->lastcx = data->cx;
1632 data->lastsx = ox;
1633 }
1634
1635 data->cx = data->lastcx;
1636 if (scroll_only || data->cy == screen_size_y(s) - 1) {
1637 window_copy_scroll_up(wp, 1);
1638 if (scroll_only && data->cy > 0)
1639 window_copy_redraw_lines(wp, data->cy - 1, 2);
1640 } else {
1641 window_copy_update_cursor(wp, data->cx, data->cy + 1);
1642 if (window_copy_update_selection(wp))
1643 window_copy_redraw_lines(wp, data->cy - 1, 2);
1644 }
1645
1646 if (!data->screen.sel.flag || !data->rectflag) {
1647 py = screen_hsize(data->backing) + data->cy - data->oy;
1648 px = window_copy_find_length(wp, py);
1649 if ((data->cx >= data->lastsx && data->cx != px) ||
1650 data->cx > px)
1651 window_copy_cursor_end_of_line(wp);
1652 }
1653 }
1654
1655 void
1656 window_copy_cursor_jump(struct window_pane *wp)
1657 {
1658 struct window_copy_mode_data *data = wp->modedata;
1659 struct screen *back_s = data->backing;
1660 const struct grid_cell *gc;
1661 u_int px, py, xx;
1662
1663 px = data->cx + 1;
1664 py = screen_hsize(back_s) + data->cy - data->oy;
1665 xx = window_copy_find_length(wp, py);
1666
1667 while (px < xx) {
1668 gc = grid_peek_cell(back_s->grid, px, py);
1669 if ((gc->flags & (GRID_FLAG_PADDING|GRID_FLAG_UTF8)) == 0
1670 && gc->data == data->jumpchar) {
1671
1672 window_copy_update_cursor(wp, px, data->cy);
1673 if (window_copy_update_selection(wp))
1674 window_copy_redraw_lines(wp, data->cy, 1);
1675 return;
1676 }
1677 px++;
1678 }
1679 }
1680
1681 void
1682 window_copy_cursor_jump_back(struct window_pane *wp)
1683 {
1684 struct window_copy_mode_data *data = wp->modedata;
1685 struct screen *back_s = data->backing;
1686 const struct grid_cell *gc;
1687 u_int px, py;
1688
1689 px = data->cx;
1690 py = screen_hsize(back_s) + data->cy - data->oy;
1691
1692 if (px > 0)
1693 px--;
1694
1695 for (;;) {
1696 gc = grid_peek_cell(back_s->grid, px, py);
1697 if ((gc->flags & (GRID_FLAG_PADDING|GRID_FLAG_UTF8)) == 0
1698 && gc->data == data->jumpchar) {
1699
1700 window_copy_update_cursor(wp, px, data->cy);
1701 if (window_copy_update_selection(wp))
1702 window_copy_redraw_lines(wp, data->cy, 1);
1703 return;
1704 }
1705 if (px == 0)
1706 break;
1707 px--;
1708 }
1709 }
1710
1711 void
1712 window_copy_cursor_next_word(struct window_pane *wp, const char *separators)
1713 {
1714 struct window_copy_mode_data *data = wp->modedata;
1715 struct screen *back_s = data->backing;
1716 u_int px, py, xx, yy;
1717 int expected = 0;
1718
1719 px = data->cx;
1720 py = screen_hsize(back_s) + data->cy - data->oy;
1721 xx = window_copy_find_length(wp, py);
1722 yy = screen_hsize(back_s) + screen_size_y(back_s) - 1;
1723
1724 /*
1725 * First skip past any nonword characters and then any word characters.
1726 *
1727 * expected is initially set to 0 for the former and then 1 for the
1728 * latter.
1729 */
1730 do {
1731 while (px > xx ||
1732 window_copy_in_set(wp, px, py, separators) == expected) {
1733 /* Move down if we're past the end of the line. */
1734 if (px > xx) {
1735 if (py == yy)
1736 return;
1737 window_copy_cursor_down(wp, 0);
1738 px = 0;
1739
1740 py = screen_hsize(back_s) + data->cy - data->oy;
1741 xx = window_copy_find_length(wp, py);
1742 } else
1743 px++;
1744 }
1745 expected = !expected;
1746 } while (expected == 1);
1747
1748 window_copy_update_cursor(wp, px, data->cy);
1749 if (window_copy_update_selection(wp))
1750 window_copy_redraw_lines(wp, data->cy, 1);
1751 }
1752
1753 void
1754 window_copy_cursor_next_word_end(struct window_pane *wp, const char *separators)
1755 {
1756 struct window_copy_mode_data *data = wp->modedata;
1757 struct screen *back_s = data->backing;
1758 u_int px, py, xx, yy;
1759 int expected = 1;
1760
1761 px = data->cx;
1762 py = screen_hsize(back_s) + data->cy - data->oy;
1763 xx = window_copy_find_length(wp, py);
1764 yy = screen_hsize(back_s) + screen_size_y(back_s) - 1;
1765
1766 /*
1767 * First skip past any word characters, then any nonword characters.
1768 *
1769 * expected is initially set to 1 for the former and then 0 for the
1770 * latter.
1771 */
1772 do {
1773 while (px > xx ||
1774 window_copy_in_set(wp, px, py, separators) == expected) {
1775 /* Move down if we're past the end of the line. */
1776 if (px > xx) {
1777 if (py == yy)
1778 return;
1779 window_copy_cursor_down(wp, 0);
1780 px = 0;
1781
1782 py = screen_hsize(back_s) + data->cy - data->oy;
1783 xx = window_copy_find_length(wp, py);
1784 } else
1785 px++;
1786 }
1787 expected = !expected;
1788 } while (expected == 0);
1789
1790 window_copy_update_cursor(wp, px, data->cy);
1791 if (window_copy_update_selection(wp))
1792 window_copy_redraw_lines(wp, data->cy, 1);
1793 }
1794
1795 /* Move to the previous place where a word begins. */
1796 void
1797 window_copy_cursor_previous_word(struct window_pane *wp, const char *separators)
1798 {
1799 struct window_copy_mode_data *data = wp->modedata;
1800 u_int px, py;
1801
1802 px = data->cx;
1803 py = screen_hsize(data->backing) + data->cy - data->oy;
1804
1805 /* Move back to the previous word character. */
1806 for (;;) {
1807 if (px > 0) {
1808 px--;
1809 if (!window_copy_in_set(wp, px, py, separators))
1810 break;
1811 } else {
1812 if (data->cy == 0 &&
1813 (screen_hsize(data->backing) == 0 ||
1814 data->oy >= screen_hsize(data->backing) - 1))
1815 goto out;
1816 window_copy_cursor_up(wp, 0);
1817
1818 py = screen_hsize(data->backing) + data->cy - data->oy;
1819 px = window_copy_find_length(wp, py);
1820 }
1821 }
1822
1823 /* Move back to the beginning of this word. */
1824 while (px > 0 && !window_copy_in_set(wp, px - 1, py, separators))
1825 px--;
1826
1827 out:
1828 window_copy_update_cursor(wp, px, data->cy);
1829 if (window_copy_update_selection(wp))
1830 window_copy_redraw_lines(wp, data->cy, 1);
1831 }
1832
1833 void
1834 window_copy_scroll_up(struct window_pane *wp, u_int ny)
1835 {
1836 struct window_copy_mode_data *data = wp->modedata;
1837 struct screen *s = &data->screen;
1838 struct screen_write_ctx ctx;
1839
1840 if (data->oy < ny)
1841 ny = data->oy;
1842 if (ny == 0)
1843 return;
1844 data->oy -= ny;
1845
1846 screen_write_start(&ctx, wp, NULL);
1847 screen_write_cursormove(&ctx, 0, 0);
1848 screen_write_deleteline(&ctx, ny);
1849 window_copy_write_lines(wp, &ctx, screen_size_y(s) - ny, ny);
1850 window_copy_write_line(wp, &ctx, 0);
1851 if (screen_size_y(s) > 1)
1852 window_copy_write_line(wp, &ctx, 1);
1853 if (screen_size_y(s) > 3)
1854 window_copy_write_line(wp, &ctx, screen_size_y(s) - 2);
1855 if (s->sel.flag && screen_size_y(s) > ny) {
1856 window_copy_update_selection(wp);
1857 window_copy_write_line(wp, &ctx, screen_size_y(s) - ny - 1);
1858 }
1859 screen_write_cursormove(&ctx, data->cx, data->cy);
1860 window_copy_update_selection(wp);
1861 screen_write_stop(&ctx);
1862 }
1863
1864 void
1865 window_copy_scroll_down(struct window_pane *wp, u_int ny)
1866 {
1867 struct window_copy_mode_data *data = wp->modedata;
1868 struct screen *s = &data->screen;
1869 struct screen_write_ctx ctx;
1870
1871 if (ny > screen_hsize(data->backing))
1872 return;
1873
1874 if (data->oy > screen_hsize(data->backing) - ny)
1875 ny = screen_hsize(data->backing) - data->oy;
1876 if (ny == 0)
1877 return;
1878 data->oy += ny;
1879
1880 screen_write_start(&ctx, wp, NULL);
1881 screen_write_cursormove(&ctx, 0, 0);
1882 screen_write_insertline(&ctx, ny);
1883 window_copy_write_lines(wp, &ctx, 0, ny);
1884 if (s->sel.flag && screen_size_y(s) > ny) {
1885 window_copy_update_selection(wp);
1886 window_copy_write_line(wp, &ctx, ny);
1887 } else if (ny == 1) /* nuke position */
1888 window_copy_write_line(wp, &ctx, 1);
1889 screen_write_cursormove(&ctx, data->cx, data->cy);
1890 window_copy_update_selection(wp);
1891 screen_write_stop(&ctx);
1892 }
1893
1894 void
1895 window_copy_rectangle_toggle(struct window_pane *wp)
1896 {
1897 struct window_copy_mode_data *data = wp->modedata;
1898 u_int px, py;
1899
1900 data->rectflag = !data->rectflag;
1901
1902 py = screen_hsize(data->backing) + data->cy - data->oy;
1903 px = window_copy_find_length(wp, py);
1904 if (data->cx > px)
1905 window_copy_update_cursor(wp, px, data->cy);
1906
1907 window_copy_update_selection(wp);
1908 window_copy_redraw_screen(wp);
1909 }