1 #include "vterm_internal.h"
2 
3 // vim: set sw=2 :
4 #include <stdio.h>
5 #include <string.h>
6 
7 #include "rect.h"
8 #include "utf8.h"
9 
10 #define UNICODE_SPACE 0x20
11 #define UNICODE_LINEFEED 0x0a
12 
13 /* State of the pen at some moment in time, also used in a cell */
14 typedef struct
15 {
16   /* After the bitfield */
17   VTermColor   fg, bg;
18 
19   unsigned int bold      : 1;
20   unsigned int underline : 2;
21   unsigned int italic    : 1;
22   unsigned int blink     : 1;
23   unsigned int reverse   : 1;
24   unsigned int conceal   : 1;
25   unsigned int strike    : 1;
26   unsigned int font      : 4; /* 0 to 9 */
27 
28   /* Extra state storage that isn't strictly pen-related */
29   unsigned int protected_cell : 1;
30   unsigned int dwl            : 1; /* on a DECDWL or DECDHL line */
31   unsigned int dhl            : 2; /* on a DECDHL line (1=top 2=bottom) */
32 } ScreenPen;
33 
34 /* Internal representation of a screen cell */
35 typedef struct
36 {
37   uint32_t chars[VTERM_MAX_CHARS_PER_CELL];
38   ScreenPen pen;
39 } ScreenCell;
40 
41 struct VTermScreen
42 {
43   VTerm *vt;
44   VTermState *state;
45 
46   const VTermScreenCallbacks *callbacks;
47   void *cbdata;
48 
49   VTermDamageSize damage_merge;
50   /* start_row == -1 => no damage */
51   VTermRect damaged;
52   VTermRect pending_scrollrect;
53   int pending_scroll_downward, pending_scroll_rightward;
54 
55   int rows;
56   int cols;
57   int global_reverse;
58 
59   /* Primary and Altscreen. buffers[1] is lazily allocated as needed */
60   ScreenCell *buffers[2];
61 
62   /* buffer will == buffers[0] or buffers[1], depending on altscreen */
63   ScreenCell *buffer;
64 
65   /* buffer for a single screen row used in scrollback storage callbacks */
66   VTermScreenCell *sb_buffer;
67 
68   ScreenPen pen;
69 };
70 
clearcell(const VTermScreen * screen,ScreenCell * cell)71 static void clearcell(const VTermScreen *screen, ScreenCell *cell)
72 {
73   cell->chars[0] = 0;
74   cell->pen = screen->pen;
75 }
76 
getcell(const VTermScreen * screen,int row,int col)77 static ScreenCell *getcell(const VTermScreen *screen, int row, int col)
78 {
79   if(row < 0 || row >= screen->rows)
80     return NULL;
81   if(col < 0 || col >= screen->cols)
82     return NULL;
83   if (screen->buffer == NULL)
84     return NULL;
85   return screen->buffer + (screen->cols * row) + col;
86 }
87 
alloc_buffer(VTermScreen * screen,int rows,int cols)88 static ScreenCell *alloc_buffer(VTermScreen *screen, int rows, int cols)
89 {
90   ScreenCell *new_buffer = vterm_allocator_malloc(screen->vt, sizeof(ScreenCell) * rows * cols);
91   int row;
92   int col;
93 
94   for(row = 0; row < rows; row++) {
95     for(col = 0; col < cols; col++) {
96       clearcell(screen, &new_buffer[row * cols + col]);
97     }
98   }
99 
100   return new_buffer;
101 }
102 
damagerect(VTermScreen * screen,VTermRect rect)103 static void damagerect(VTermScreen *screen, VTermRect rect)
104 {
105   VTermRect emit;
106 
107   switch(screen->damage_merge) {
108   case VTERM_DAMAGE_CELL:
109     /* Always emit damage event */
110     emit = rect;
111     break;
112 
113   case VTERM_DAMAGE_ROW:
114     /* Emit damage longer than one row. Try to merge with existing damage in
115      * the same row */
116     if(rect.end_row > rect.start_row + 1) {
117       // Bigger than 1 line - flush existing, emit this
118       vterm_screen_flush_damage(screen);
119       emit = rect;
120     }
121     else if(screen->damaged.start_row == -1) {
122       // None stored yet
123       screen->damaged = rect;
124       return;
125     }
126     else if(rect.start_row == screen->damaged.start_row) {
127       // Merge with the stored line
128       if(screen->damaged.start_col > rect.start_col)
129         screen->damaged.start_col = rect.start_col;
130       if(screen->damaged.end_col < rect.end_col)
131         screen->damaged.end_col = rect.end_col;
132       return;
133     }
134     else {
135       // Emit the currently stored line, store a new one
136       emit = screen->damaged;
137       screen->damaged = rect;
138     }
139     break;
140 
141   case VTERM_DAMAGE_SCREEN:
142   case VTERM_DAMAGE_SCROLL:
143     /* Never emit damage event */
144     if(screen->damaged.start_row == -1)
145       screen->damaged = rect;
146     else {
147       rect_expand(&screen->damaged, &rect);
148     }
149     return;
150 
151   default:
152     DEBUG_LOG1("TODO: Maybe merge damage for level %d\n", screen->damage_merge);
153     return;
154   }
155 
156   if(screen->callbacks && screen->callbacks->damage)
157     (*screen->callbacks->damage)(emit, screen->cbdata);
158 }
159 
damagescreen(VTermScreen * screen)160 static void damagescreen(VTermScreen *screen)
161 {
162   VTermRect rect = {0,0,0,0};
163   rect.end_row = screen->rows;
164   rect.end_col = screen->cols;
165 
166   damagerect(screen, rect);
167 }
168 
putglyph(VTermGlyphInfo * info,VTermPos pos,void * user)169 static int putglyph(VTermGlyphInfo *info, VTermPos pos, void *user)
170 {
171   int i;
172   int col;
173   VTermRect rect;
174 
175   VTermScreen *screen = user;
176   ScreenCell *cell = getcell(screen, pos.row, pos.col);
177 
178   if(!cell)
179     return 0;
180 
181   for(i = 0; i < VTERM_MAX_CHARS_PER_CELL && info->chars[i]; i++) {
182     cell->chars[i] = info->chars[i];
183     cell->pen = screen->pen;
184   }
185   if(i < VTERM_MAX_CHARS_PER_CELL)
186     cell->chars[i] = 0;
187 
188   for(col = 1; col < info->width; col++)
189   {
190     ScreenCell *onecell = getcell(screen, pos.row, pos.col + col);
191     if (onecell == NULL)
192       break;
193     onecell->chars[0] = (uint32_t)-1;
194   }
195 
196   rect.start_row = pos.row;
197   rect.end_row   = pos.row+1;
198   rect.start_col = pos.col;
199   rect.end_col   = pos.col+info->width;
200 
201   cell->pen.protected_cell = info->protected_cell;
202   cell->pen.dwl            = info->dwl;
203   cell->pen.dhl            = info->dhl;
204 
205   damagerect(screen, rect);
206 
207   return 1;
208 }
209 
sb_pushline_from_row(VTermScreen * screen,int row)210 static void sb_pushline_from_row(VTermScreen *screen, int row)
211 {
212   VTermPos pos;
213   pos.row = row;
214   for(pos.col = 0; pos.col < screen->cols; pos.col++)
215     vterm_screen_get_cell(screen, pos, screen->sb_buffer + pos.col);
216 
217   (screen->callbacks->sb_pushline)(screen->cols, screen->sb_buffer, screen->cbdata);
218 }
219 
moverect_internal(VTermRect dest,VTermRect src,void * user)220 static int moverect_internal(VTermRect dest, VTermRect src, void *user)
221 {
222   VTermScreen *screen = user;
223 
224   if(screen->callbacks && screen->callbacks->sb_pushline &&
225      dest.start_row == 0 && dest.start_col == 0 &&        // starts top-left corner
226      dest.end_col == screen->cols &&                      // full width
227      screen->buffer == screen->buffers[BUFIDX_PRIMARY]) { // not altscreen
228     int row;
229     for(row = 0; row < src.start_row; row++)
230       sb_pushline_from_row(screen, row);
231   }
232 
233   {
234     int cols = src.end_col - src.start_col;
235     int downward = src.start_row - dest.start_row;
236     int init_row, test_row, inc_row;
237     int row;
238 
239     if(downward < 0) {
240       init_row = dest.end_row - 1;
241       test_row = dest.start_row - 1;
242       inc_row  = -1;
243     }
244     else {
245       init_row = dest.start_row;
246       test_row = dest.end_row;
247       inc_row  = +1;
248     }
249 
250     for(row = init_row; row != test_row; row += inc_row)
251       memmove(getcell(screen, row, dest.start_col),
252 	      getcell(screen, row + downward, src.start_col),
253 	      cols * sizeof(ScreenCell));
254   }
255 
256   return 1;
257 }
258 
moverect_user(VTermRect dest,VTermRect src,void * user)259 static int moverect_user(VTermRect dest, VTermRect src, void *user)
260 {
261   VTermScreen *screen = user;
262 
263   if(screen->callbacks && screen->callbacks->moverect) {
264     if(screen->damage_merge != VTERM_DAMAGE_SCROLL)
265       // Avoid an infinite loop
266       vterm_screen_flush_damage(screen);
267 
268     if((*screen->callbacks->moverect)(dest, src, screen->cbdata))
269       return 1;
270   }
271 
272   damagerect(screen, dest);
273 
274   return 1;
275 }
276 
erase_internal(VTermRect rect,int selective,void * user)277 static int erase_internal(VTermRect rect, int selective, void *user)
278 {
279   VTermScreen *screen = user;
280   int row, col;
281 
282   for(row = rect.start_row; row < screen->state->rows && row < rect.end_row; row++) {
283     const VTermLineInfo *info = vterm_state_get_lineinfo(screen->state, row);
284 
285     for(col = rect.start_col; col < rect.end_col; col++) {
286       ScreenCell *cell = getcell(screen, row, col);
287 
288       if (cell == NULL)
289       {
290         DEBUG_LOG2("libvterm: erase_internal() position invalid: %d / %d",
291 								     row, col);
292 	return 1;
293       }
294       if(selective && cell->pen.protected_cell)
295         continue;
296 
297       cell->chars[0] = 0;
298       cell->pen = screen->pen;
299       cell->pen.dwl = info->doublewidth;
300       cell->pen.dhl = info->doubleheight;
301     }
302   }
303 
304   return 1;
305 }
306 
erase_user(VTermRect rect,int selective UNUSED,void * user)307 static int erase_user(VTermRect rect, int selective UNUSED, void *user)
308 {
309   VTermScreen *screen = user;
310 
311   damagerect(screen, rect);
312 
313   return 1;
314 }
315 
erase(VTermRect rect,int selective,void * user)316 static int erase(VTermRect rect, int selective, void *user)
317 {
318   erase_internal(rect, selective, user);
319   return erase_user(rect, 0, user);
320 }
321 
scrollrect(VTermRect rect,int downward,int rightward,void * user)322 static int scrollrect(VTermRect rect, int downward, int rightward, void *user)
323 {
324   VTermScreen *screen = user;
325 
326   if(screen->damage_merge != VTERM_DAMAGE_SCROLL) {
327     vterm_scroll_rect(rect, downward, rightward,
328         moverect_internal, erase_internal, screen);
329 
330     vterm_screen_flush_damage(screen);
331 
332     vterm_scroll_rect(rect, downward, rightward,
333         moverect_user, erase_user, screen);
334 
335     return 1;
336   }
337 
338   if(screen->damaged.start_row != -1 &&
339      !rect_intersects(&rect, &screen->damaged)) {
340     vterm_screen_flush_damage(screen);
341   }
342 
343   if(screen->pending_scrollrect.start_row == -1) {
344     screen->pending_scrollrect = rect;
345     screen->pending_scroll_downward  = downward;
346     screen->pending_scroll_rightward = rightward;
347   }
348   else if(rect_equal(&screen->pending_scrollrect, &rect) &&
349      ((screen->pending_scroll_downward  == 0 && downward  == 0) ||
350       (screen->pending_scroll_rightward == 0 && rightward == 0))) {
351     screen->pending_scroll_downward  += downward;
352     screen->pending_scroll_rightward += rightward;
353   }
354   else {
355     vterm_screen_flush_damage(screen);
356 
357     screen->pending_scrollrect = rect;
358     screen->pending_scroll_downward  = downward;
359     screen->pending_scroll_rightward = rightward;
360   }
361 
362   vterm_scroll_rect(rect, downward, rightward,
363       moverect_internal, erase_internal, screen);
364 
365   if(screen->damaged.start_row == -1)
366     return 1;
367 
368   if(rect_contains(&rect, &screen->damaged)) {
369     /* Scroll region entirely contains the damage; just move it */
370     vterm_rect_move(&screen->damaged, -downward, -rightward);
371     rect_clip(&screen->damaged, &rect);
372   }
373   /* There are a number of possible cases here, but lets restrict this to only
374    * the common case where we might actually gain some performance by
375    * optimising it. Namely, a vertical scroll that neatly cuts the damage
376    * region in half.
377    */
378   else if(rect.start_col <= screen->damaged.start_col &&
379           rect.end_col   >= screen->damaged.end_col &&
380           rightward == 0) {
381     if(screen->damaged.start_row >= rect.start_row &&
382        screen->damaged.start_row  < rect.end_row) {
383       screen->damaged.start_row -= downward;
384       if(screen->damaged.start_row < rect.start_row)
385         screen->damaged.start_row = rect.start_row;
386       if(screen->damaged.start_row > rect.end_row)
387         screen->damaged.start_row = rect.end_row;
388     }
389     if(screen->damaged.end_row >= rect.start_row &&
390        screen->damaged.end_row  < rect.end_row) {
391       screen->damaged.end_row -= downward;
392       if(screen->damaged.end_row < rect.start_row)
393         screen->damaged.end_row = rect.start_row;
394       if(screen->damaged.end_row > rect.end_row)
395         screen->damaged.end_row = rect.end_row;
396     }
397   }
398   else {
399     DEBUG_LOG2("TODO: Just flush and redo damaged=" STRFrect " rect=" STRFrect "\n",
400         ARGSrect(screen->damaged), ARGSrect(rect));
401   }
402 
403   return 1;
404 }
405 
movecursor(VTermPos pos,VTermPos oldpos,int visible,void * user)406 static int movecursor(VTermPos pos, VTermPos oldpos, int visible, void *user)
407 {
408   VTermScreen *screen = user;
409 
410   if(screen->callbacks && screen->callbacks->movecursor)
411     return (*screen->callbacks->movecursor)(pos, oldpos, visible, screen->cbdata);
412 
413   return 0;
414 }
415 
setpenattr(VTermAttr attr,VTermValue * val,void * user)416 static int setpenattr(VTermAttr attr, VTermValue *val, void *user)
417 {
418   VTermScreen *screen = user;
419 
420   switch(attr) {
421   case VTERM_ATTR_BOLD:
422     screen->pen.bold = val->boolean;
423     return 1;
424   case VTERM_ATTR_UNDERLINE:
425     screen->pen.underline = val->number;
426     return 1;
427   case VTERM_ATTR_ITALIC:
428     screen->pen.italic = val->boolean;
429     return 1;
430   case VTERM_ATTR_BLINK:
431     screen->pen.blink = val->boolean;
432     return 1;
433   case VTERM_ATTR_REVERSE:
434     screen->pen.reverse = val->boolean;
435     return 1;
436   case VTERM_ATTR_CONCEAL:
437     screen->pen.conceal = val->boolean;
438     return 1;
439   case VTERM_ATTR_STRIKE:
440     screen->pen.strike = val->boolean;
441     return 1;
442   case VTERM_ATTR_FONT:
443     screen->pen.font = val->number;
444     return 1;
445   case VTERM_ATTR_FOREGROUND:
446     screen->pen.fg = val->color;
447     return 1;
448   case VTERM_ATTR_BACKGROUND:
449     screen->pen.bg = val->color;
450     return 1;
451 
452   case VTERM_N_ATTRS:
453     return 0;
454   }
455 
456   return 0;
457 }
458 
settermprop(VTermProp prop,VTermValue * val,void * user)459 static int settermprop(VTermProp prop, VTermValue *val, void *user)
460 {
461   VTermScreen *screen = user;
462 
463   switch(prop) {
464   case VTERM_PROP_ALTSCREEN:
465     if(val->boolean && !screen->buffers[BUFIDX_ALTSCREEN])
466       return 0;
467 
468     screen->buffer = val->boolean ? screen->buffers[BUFIDX_ALTSCREEN] : screen->buffers[BUFIDX_PRIMARY];
469     /* only send a damage event on disable; because during enable there's an
470      * erase that sends a damage anyway
471      */
472     if(!val->boolean)
473       damagescreen(screen);
474     break;
475   case VTERM_PROP_REVERSE:
476     screen->global_reverse = val->boolean;
477     damagescreen(screen);
478     break;
479   default:
480     ; /* ignore */
481   }
482 
483   if(screen->callbacks && screen->callbacks->settermprop)
484     return (*screen->callbacks->settermprop)(prop, val, screen->cbdata);
485 
486   return 1;
487 }
488 
bell(void * user)489 static int bell(void *user)
490 {
491   VTermScreen *screen = user;
492 
493   if(screen->callbacks && screen->callbacks->bell)
494     return (*screen->callbacks->bell)(screen->cbdata);
495 
496   return 0;
497 }
498 
resize_buffer(VTermScreen * screen,int bufidx,int new_rows,int new_cols,int active,VTermStateFields * statefields)499 static void resize_buffer(VTermScreen *screen, int bufidx, int new_rows, int new_cols, int active, VTermStateFields *statefields)
500 {
501   int old_rows = screen->rows;
502   int old_cols = screen->cols;
503 
504   ScreenCell *old_buffer = screen->buffers[bufidx];
505   ScreenCell *new_buffer = vterm_allocator_malloc(screen->vt, sizeof(ScreenCell) * new_rows * new_cols);
506 
507   // Find the final row of old buffer content
508   int old_row = old_rows - 1;
509   int new_row = new_rows - 1;
510   int col;
511 
512   while(new_row >= 0 && old_row >= 0) {
513     for(col = 0; col < old_cols && col < new_cols; col++)
514       new_buffer[new_row * new_cols + col] = old_buffer[old_row * old_cols + col];
515     for( ; col < new_cols; col++)
516       clearcell(screen, &new_buffer[new_row * new_cols + col]);
517 
518     old_row--;
519     new_row--;
520 
521     if(new_row < 0 && old_row >= 0 &&
522         new_buffer[(new_rows - 1) * new_cols].chars[0] == 0 &&
523         (!active || statefields->pos.row < (new_rows - 1))) {
524       int moverows = new_rows - 1;
525       memmove(&new_buffer[1 * new_cols], &new_buffer[0], moverows * new_cols * sizeof(ScreenCell));
526 
527       new_row++;
528     }
529   }
530 
531   if(old_row >= 0 && bufidx == BUFIDX_PRIMARY) {
532     /* Push spare lines to scrollback buffer */
533     int row;
534     for(row = 0; row <= old_row; row++)
535       sb_pushline_from_row(screen, row);
536     if(active)
537       statefields->pos.row -= (old_row + 1);
538   }
539   if(new_row >= 0 && bufidx == BUFIDX_PRIMARY &&
540       screen->callbacks && screen->callbacks->sb_popline) {
541     /* Try to backfill rows by popping scrollback buffer */
542     while(new_row >= 0) {
543       VTermPos pos;
544       if(!(screen->callbacks->sb_popline(old_cols, screen->sb_buffer, screen->cbdata)))
545         break;
546 
547       pos.row = new_row;
548       for(pos.col = 0; pos.col < old_cols && pos.col < new_cols; pos.col += screen->sb_buffer[pos.col].width) {
549         VTermScreenCell *src = &screen->sb_buffer[pos.col];
550         ScreenCell *dst = &new_buffer[pos.row * new_cols + pos.col];
551 	int i;
552 
553         for(i = 0; i < VTERM_MAX_CHARS_PER_CELL; i++) {
554           dst->chars[i] = src->chars[i];
555           if(!src->chars[i])
556             break;
557         }
558 
559         dst->pen.bold      = src->attrs.bold;
560         dst->pen.underline = src->attrs.underline;
561         dst->pen.italic    = src->attrs.italic;
562         dst->pen.blink     = src->attrs.blink;
563         dst->pen.reverse   = src->attrs.reverse ^ screen->global_reverse;
564         dst->pen.conceal   = src->attrs.conceal;
565         dst->pen.strike    = src->attrs.strike;
566         dst->pen.font      = src->attrs.font;
567 
568         dst->pen.fg = src->fg;
569         dst->pen.bg = src->bg;
570 
571         if(src->width == 2 && pos.col < (new_cols-1))
572           (dst + 1)->chars[0] = (uint32_t) -1;
573       }
574       for( ; pos.col < new_cols; pos.col++)
575         clearcell(screen, &new_buffer[pos.row * new_cols + pos.col]);
576       new_row--;
577 
578       if(active)
579         statefields->pos.row++;
580     }
581   }
582 
583   if(new_row >= 0) {
584     /* Scroll new rows back up to the top and fill in blanks at the bottom */
585     int moverows = new_rows - new_row - 1;
586     memmove(&new_buffer[0], &new_buffer[(new_row + 1) * new_cols], moverows * new_cols * sizeof(ScreenCell));
587 
588     for(new_row = moverows; new_row < new_rows; new_row++)
589       for(col = 0; col < new_cols; col++)
590         clearcell(screen, &new_buffer[new_row * new_cols + col]);
591   }
592 
593   vterm_allocator_free(screen->vt, old_buffer);
594   screen->buffers[bufidx] = new_buffer;
595 
596   return;
597 
598   /* REFLOW TODO:
599    *   Handle delta. Probably needs to be a full cursorpos that we edit
600    */
601 }
602 
resize(int new_rows,int new_cols,VTermStateFields * fields,void * user)603 static int resize(int new_rows, int new_cols, VTermStateFields *fields, void *user)
604 {
605   VTermScreen *screen = user;
606 
607   int altscreen_active = (screen->buffers[BUFIDX_ALTSCREEN] && screen->buffer == screen->buffers[BUFIDX_ALTSCREEN]);
608 
609   int old_cols = screen->cols;
610 
611   if(new_cols > old_cols) {
612     /* Ensure that ->sb_buffer is large enough for a new or and old row */
613     if(screen->sb_buffer)
614       vterm_allocator_free(screen->vt, screen->sb_buffer);
615 
616     screen->sb_buffer = vterm_allocator_malloc(screen->vt, sizeof(VTermScreenCell) * new_cols);
617   }
618 
619   resize_buffer(screen, 0, new_rows, new_cols, !altscreen_active, fields);
620   if(screen->buffers[BUFIDX_ALTSCREEN])
621     resize_buffer(screen, 1, new_rows, new_cols, altscreen_active, fields);
622 
623   screen->buffer = altscreen_active ? screen->buffers[BUFIDX_ALTSCREEN] : screen->buffers[BUFIDX_PRIMARY];
624 
625   screen->rows = new_rows;
626   screen->cols = new_cols;
627 
628   if(new_cols <= old_cols) {
629     if(screen->sb_buffer)
630       vterm_allocator_free(screen->vt, screen->sb_buffer);
631 
632     screen->sb_buffer = vterm_allocator_malloc(screen->vt, sizeof(VTermScreenCell) * new_cols);
633   }
634 
635   /* TODO: Maaaaybe we can optimise this if there's no reflow happening */
636   damagescreen(screen);
637 
638   if(screen->callbacks && screen->callbacks->resize)
639     return (*screen->callbacks->resize)(new_rows, new_cols, screen->cbdata);
640 
641   return 1;
642 }
643 
setlineinfo(int row,const VTermLineInfo * newinfo,const VTermLineInfo * oldinfo,void * user)644 static int setlineinfo(int row, const VTermLineInfo *newinfo, const VTermLineInfo *oldinfo, void *user)
645 {
646   VTermScreen *screen = user;
647   int col;
648   VTermRect rect;
649 
650   if(newinfo->doublewidth != oldinfo->doublewidth ||
651      newinfo->doubleheight != oldinfo->doubleheight) {
652     for(col = 0; col < screen->cols; col++) {
653       ScreenCell *cell = getcell(screen, row, col);
654       if (cell == NULL)
655       {
656         DEBUG_LOG2("libvterm: setlineinfo() position invalid: %d / %d",
657 								     row, col);
658 	return 1;
659       }
660       cell->pen.dwl = newinfo->doublewidth;
661       cell->pen.dhl = newinfo->doubleheight;
662     }
663 
664     rect.start_row = row;
665     rect.end_row   = row + 1;
666     rect.start_col = 0;
667     rect.end_col   = newinfo->doublewidth ? screen->cols / 2 : screen->cols;
668     damagerect(screen, rect);
669 
670     if(newinfo->doublewidth) {
671       rect.start_col = screen->cols / 2;
672       rect.end_col   = screen->cols;
673 
674       erase_internal(rect, 0, user);
675     }
676   }
677 
678   return 1;
679 }
680 
681 static VTermStateCallbacks state_cbs = {
682   &putglyph, // putglyph
683   &movecursor, // movecursor
684   &scrollrect, // scrollrect
685   NULL, // moverect
686   &erase, // erase
687   NULL, // initpen
688   &setpenattr, // setpenattr
689   &settermprop, // settermprop
690   &bell, // bell
691   &resize, // resize
692   &setlineinfo // setlineinfo
693 };
694 
695 /*
696  * Allocate a new screen and return it.
697  * Return NULL when out of memory.
698  */
screen_new(VTerm * vt)699 static VTermScreen *screen_new(VTerm *vt)
700 {
701   VTermState *state = vterm_obtain_state(vt);
702   VTermScreen *screen;
703   int rows, cols;
704 
705   if (state == NULL)
706     return NULL;
707   screen = vterm_allocator_malloc(vt, sizeof(VTermScreen));
708   if (screen == NULL)
709     return NULL;
710 
711   vterm_get_size(vt, &rows, &cols);
712 
713   screen->vt = vt;
714   screen->state = state;
715 
716   screen->damage_merge = VTERM_DAMAGE_CELL;
717   screen->damaged.start_row = -1;
718   screen->pending_scrollrect.start_row = -1;
719 
720   screen->rows = rows;
721   screen->cols = cols;
722 
723   screen->callbacks = NULL;
724   screen->cbdata    = NULL;
725 
726   screen->buffers[BUFIDX_PRIMARY] = alloc_buffer(screen, rows, cols);
727 
728   screen->buffer = screen->buffers[BUFIDX_PRIMARY];
729 
730   screen->sb_buffer = vterm_allocator_malloc(screen->vt, sizeof(VTermScreenCell) * cols);
731   if (screen->buffer == NULL || screen->sb_buffer == NULL)
732   {
733     vterm_screen_free(screen);
734     return NULL;
735   }
736 
737   vterm_state_set_callbacks(screen->state, &state_cbs, screen);
738 
739   return screen;
740 }
741 
vterm_screen_free(VTermScreen * screen)742 INTERNAL void vterm_screen_free(VTermScreen *screen)
743 {
744   vterm_allocator_free(screen->vt, screen->buffers[BUFIDX_PRIMARY]);
745   if(screen->buffers[BUFIDX_ALTSCREEN])
746     vterm_allocator_free(screen->vt, screen->buffers[BUFIDX_ALTSCREEN]);
747 
748   vterm_allocator_free(screen->vt, screen->sb_buffer);
749 
750   vterm_allocator_free(screen->vt, screen);
751 }
752 
vterm_screen_reset(VTermScreen * screen,int hard)753 void vterm_screen_reset(VTermScreen *screen, int hard)
754 {
755   screen->damaged.start_row = -1;
756   screen->pending_scrollrect.start_row = -1;
757   vterm_state_reset(screen->state, hard);
758   vterm_screen_flush_damage(screen);
759 }
760 
_get_chars(const VTermScreen * screen,const int utf8,void * buffer,size_t len,const VTermRect rect)761 static size_t _get_chars(const VTermScreen *screen, const int utf8, void *buffer, size_t len, const VTermRect rect)
762 {
763   size_t outpos = 0;
764   int padding = 0;
765   int row, col;
766 
767 #define PUT(c)                                             \
768   if(utf8) {                                               \
769     size_t thislen = utf8_seqlen(c);                       \
770     if(buffer && outpos + thislen <= len)                  \
771       outpos += fill_utf8((c), (char *)buffer + outpos);   \
772     else                                                   \
773       outpos += thislen;                                   \
774   }                                                        \
775   else {                                                   \
776     if(buffer && outpos + 1 <= len)                        \
777       ((uint32_t*)buffer)[outpos++] = (c);                 \
778     else                                                   \
779       outpos++;                                            \
780   }
781 
782   for(row = rect.start_row; row < rect.end_row; row++) {
783     for(col = rect.start_col; col < rect.end_col; col++) {
784       ScreenCell *cell = getcell(screen, row, col);
785       int i;
786 
787       if (cell == NULL)
788       {
789         DEBUG_LOG2("libvterm: _get_chars() position invalid: %d / %d",
790 								     row, col);
791 	return 1;
792       }
793       if(cell->chars[0] == 0)
794         // Erased cell, might need a space
795         padding++;
796       else if(cell->chars[0] == (uint32_t)-1)
797         // Gap behind a double-width char, do nothing
798         ;
799       else {
800         while(padding) {
801           PUT(UNICODE_SPACE);
802           padding--;
803         }
804         for(i = 0; i < VTERM_MAX_CHARS_PER_CELL && cell->chars[i]; i++) {
805           PUT(cell->chars[i]);
806         }
807       }
808     }
809 
810     if(row < rect.end_row - 1) {
811       PUT(UNICODE_LINEFEED);
812       padding = 0;
813     }
814   }
815 
816   return outpos;
817 }
818 
vterm_screen_get_chars(const VTermScreen * screen,uint32_t * chars,size_t len,const VTermRect rect)819 size_t vterm_screen_get_chars(const VTermScreen *screen, uint32_t *chars, size_t len, const VTermRect rect)
820 {
821   return _get_chars(screen, 0, chars, len, rect);
822 }
823 
vterm_screen_get_text(const VTermScreen * screen,char * str,size_t len,const VTermRect rect)824 size_t vterm_screen_get_text(const VTermScreen *screen, char *str, size_t len, const VTermRect rect)
825 {
826   return _get_chars(screen, 1, str, len, rect);
827 }
828 
829 /* Copy internal to external representation of a screen cell */
vterm_screen_get_cell(const VTermScreen * screen,VTermPos pos,VTermScreenCell * cell)830 int vterm_screen_get_cell(const VTermScreen *screen, VTermPos pos, VTermScreenCell *cell)
831 {
832   ScreenCell *intcell = getcell(screen, pos.row, pos.col);
833   int i;
834 
835   if(!intcell)
836     return 0;
837 
838   for(i = 0; i < VTERM_MAX_CHARS_PER_CELL; i++) {
839     cell->chars[i] = intcell->chars[i];
840     if(!intcell->chars[i])
841       break;
842   }
843 
844   cell->attrs.bold      = intcell->pen.bold;
845   cell->attrs.underline = intcell->pen.underline;
846   cell->attrs.italic    = intcell->pen.italic;
847   cell->attrs.blink     = intcell->pen.blink;
848   cell->attrs.reverse   = intcell->pen.reverse ^ screen->global_reverse;
849   cell->attrs.conceal   = intcell->pen.conceal;
850   cell->attrs.strike    = intcell->pen.strike;
851   cell->attrs.font      = intcell->pen.font;
852 
853   cell->attrs.dwl = intcell->pen.dwl;
854   cell->attrs.dhl = intcell->pen.dhl;
855 
856   cell->fg = intcell->pen.fg;
857   cell->bg = intcell->pen.bg;
858 
859   if(vterm_get_special_pty_type() == 2) {
860     // Get correct cell width from cell information contained in line buffer
861     if(pos.col < (screen->cols - 1) &&
862        getcell(screen, pos.row, pos.col + 1)->chars[0] == (uint32_t)-1) {
863       if(getcell(screen, pos.row, pos.col)->chars[0] == 0x20) {
864         getcell(screen, pos.row, pos.col)->chars[0] = 0;
865         cell->width = 2;
866       } else if(getcell(screen, pos.row, pos.col)->chars[0] == 0) {
867         getcell(screen, pos.row, pos.col + 1)->chars[0] = 0;
868         cell->width = 1;
869       } else {
870         cell->width = 2;
871       }
872     } else
873       cell->width = 1;
874   } else {
875     if(pos.col < (screen->cols - 1) &&
876        getcell(screen, pos.row, pos.col + 1)->chars[0] == (uint32_t)-1)
877       cell->width = 2;
878     else
879       cell->width = 1;
880   }
881 
882   return 1;
883 }
884 
vterm_screen_is_eol(const VTermScreen * screen,VTermPos pos)885 int vterm_screen_is_eol(const VTermScreen *screen, VTermPos pos)
886 {
887   /* This cell is EOL if this and every cell to the right is black */
888   for(; pos.col < screen->cols; pos.col++) {
889     ScreenCell *cell = getcell(screen, pos.row, pos.col);
890     if(cell->chars[0] != 0)
891       return 0;
892   }
893 
894   return 1;
895 }
896 
vterm_obtain_screen(VTerm * vt)897 VTermScreen *vterm_obtain_screen(VTerm *vt)
898 {
899   if(!vt->screen)
900     vt->screen = screen_new(vt);
901   return vt->screen;
902 }
903 
vterm_screen_enable_altscreen(VTermScreen * screen,int altscreen)904 void vterm_screen_enable_altscreen(VTermScreen *screen, int altscreen)
905 {
906   if(!screen->buffers[BUFIDX_ALTSCREEN] && altscreen) {
907     int rows, cols;
908     vterm_get_size(screen->vt, &rows, &cols);
909 
910     screen->buffers[BUFIDX_ALTSCREEN] = alloc_buffer(screen, rows, cols);
911   }
912 }
913 
vterm_screen_set_callbacks(VTermScreen * screen,const VTermScreenCallbacks * callbacks,void * user)914 void vterm_screen_set_callbacks(VTermScreen *screen, const VTermScreenCallbacks *callbacks, void *user)
915 {
916   screen->callbacks = callbacks;
917   screen->cbdata = user;
918 }
919 
vterm_screen_get_cbdata(VTermScreen * screen)920 void *vterm_screen_get_cbdata(VTermScreen *screen)
921 {
922   return screen->cbdata;
923 }
924 
vterm_screen_set_unrecognised_fallbacks(VTermScreen * screen,const VTermStateFallbacks * fallbacks,void * user)925 void vterm_screen_set_unrecognised_fallbacks(VTermScreen *screen, const VTermStateFallbacks *fallbacks, void *user)
926 {
927   vterm_state_set_unrecognised_fallbacks(screen->state, fallbacks, user);
928 }
929 
vterm_screen_get_unrecognised_fbdata(VTermScreen * screen)930 void *vterm_screen_get_unrecognised_fbdata(VTermScreen *screen)
931 {
932   return vterm_state_get_unrecognised_fbdata(screen->state);
933 }
934 
vterm_screen_flush_damage(VTermScreen * screen)935 void vterm_screen_flush_damage(VTermScreen *screen)
936 {
937   if(screen->pending_scrollrect.start_row != -1) {
938     vterm_scroll_rect(screen->pending_scrollrect, screen->pending_scroll_downward, screen->pending_scroll_rightward,
939         moverect_user, erase_user, screen);
940 
941     screen->pending_scrollrect.start_row = -1;
942   }
943 
944   if(screen->damaged.start_row != -1) {
945     if(screen->callbacks && screen->callbacks->damage)
946       (*screen->callbacks->damage)(screen->damaged, screen->cbdata);
947 
948     screen->damaged.start_row = -1;
949   }
950 }
951 
vterm_screen_set_damage_merge(VTermScreen * screen,VTermDamageSize size)952 void vterm_screen_set_damage_merge(VTermScreen *screen, VTermDamageSize size)
953 {
954   vterm_screen_flush_damage(screen);
955   screen->damage_merge = size;
956 }
957 
attrs_differ(VTermAttrMask attrs,ScreenCell * a,ScreenCell * b)958 static int attrs_differ(VTermAttrMask attrs, ScreenCell *a, ScreenCell *b)
959 {
960   if((attrs & VTERM_ATTR_BOLD_MASK)       && (a->pen.bold != b->pen.bold))
961     return 1;
962   if((attrs & VTERM_ATTR_UNDERLINE_MASK)  && (a->pen.underline != b->pen.underline))
963     return 1;
964   if((attrs & VTERM_ATTR_ITALIC_MASK)     && (a->pen.italic != b->pen.italic))
965     return 1;
966   if((attrs & VTERM_ATTR_BLINK_MASK)      && (a->pen.blink != b->pen.blink))
967     return 1;
968   if((attrs & VTERM_ATTR_REVERSE_MASK)    && (a->pen.reverse != b->pen.reverse))
969     return 1;
970   if((attrs & VTERM_ATTR_CONCEAL_MASK)    && (a->pen.conceal != b->pen.conceal))
971     return 1;
972   if((attrs & VTERM_ATTR_STRIKE_MASK)     && (a->pen.strike != b->pen.strike))
973     return 1;
974   if((attrs & VTERM_ATTR_FONT_MASK)       && (a->pen.font != b->pen.font))
975     return 1;
976   if((attrs & VTERM_ATTR_FOREGROUND_MASK) && !vterm_color_is_equal(&a->pen.fg, &b->pen.fg))
977     return 1;
978   if((attrs & VTERM_ATTR_BACKGROUND_MASK) && !vterm_color_is_equal(&a->pen.bg, &b->pen.bg))
979     return 1;
980 
981   return 0;
982 }
983 
vterm_screen_get_attrs_extent(const VTermScreen * screen,VTermRect * extent,VTermPos pos,VTermAttrMask attrs)984 int vterm_screen_get_attrs_extent(const VTermScreen *screen, VTermRect *extent, VTermPos pos, VTermAttrMask attrs)
985 {
986   int col;
987 
988   ScreenCell *target = getcell(screen, pos.row, pos.col);
989 
990   // TODO: bounds check
991   extent->start_row = pos.row;
992   extent->end_row   = pos.row + 1;
993 
994   if(extent->start_col < 0)
995     extent->start_col = 0;
996   if(extent->end_col < 0)
997     extent->end_col = screen->cols;
998 
999   for(col = pos.col - 1; col >= extent->start_col; col--)
1000     if(attrs_differ(attrs, target, getcell(screen, pos.row, col)))
1001       break;
1002   extent->start_col = col + 1;
1003 
1004   for(col = pos.col + 1; col < extent->end_col; col++)
1005     if(attrs_differ(attrs, target, getcell(screen, pos.row, col)))
1006       break;
1007   extent->end_col = col - 1;
1008 
1009   return 1;
1010 }
1011 
vterm_screen_convert_color_to_rgb(const VTermScreen * screen,VTermColor * col)1012 void vterm_screen_convert_color_to_rgb(const VTermScreen *screen, VTermColor *col)
1013 {
1014   vterm_state_convert_color_to_rgb(screen->state, col);
1015 }
1016