1 // SPDX-License-Identifier: GPL-2.0
2 #include "../util/string2.h"
3 #include "../util/config.h"
7 #include <linux/compiler.h>
8 #include <linux/list.h>
9 #include <linux/rbtree.h>
10 #include <linux/string.h>
12 #include <sys/ttydefaults.h>
16 #include "../util/color.h"
17 #include <linux/ctype.h>
18 #include <linux/zalloc.h>
20 static int ui_browser__percent_color(struct ui_browser
*browser
,
21 double percent
, bool current
)
23 if (current
&& (!browser
->use_navkeypressed
|| browser
->navkeypressed
))
24 return HE_COLORSET_SELECTED
;
25 if (percent
>= MIN_RED
)
26 return HE_COLORSET_TOP
;
27 if (percent
>= MIN_GREEN
)
28 return HE_COLORSET_MEDIUM
;
29 return HE_COLORSET_NORMAL
;
32 int ui_browser__set_color(struct ui_browser
*browser
, int color
)
34 int ret
= browser
->current_color
;
35 browser
->current_color
= color
;
36 SLsmg_set_color(color
);
40 void ui_browser__set_percent_color(struct ui_browser
*browser
,
41 double percent
, bool current
)
43 int color
= ui_browser__percent_color(browser
, percent
, current
);
44 ui_browser__set_color(browser
, color
);
47 void ui_browser__gotorc_title(struct ui_browser
*browser
, int y
, int x
)
49 SLsmg_gotorc(browser
->y
+ y
, browser
->x
+ x
);
52 void ui_browser__gotorc(struct ui_browser
*browser
, int y
, int x
)
54 SLsmg_gotorc(browser
->y
+ y
+ browser
->extra_title_lines
, browser
->x
+ x
);
57 void ui_browser__write_nstring(struct ui_browser
*browser __maybe_unused
, const char *msg
,
60 SLsmg_write_nstring(msg
, width
);
63 void ui_browser__vprintf(struct ui_browser
*browser __maybe_unused
, const char *fmt
, va_list args
)
65 SLsmg_vprintf(fmt
, args
);
68 void ui_browser__printf(struct ui_browser
*browser __maybe_unused
, const char *fmt
, ...)
73 ui_browser__vprintf(browser
, fmt
, args
);
77 static struct list_head
*
78 ui_browser__list_head_filter_entries(struct ui_browser
*browser
,
79 struct list_head
*pos
)
82 if (!browser
->filter
|| !browser
->filter(browser
, pos
))
85 } while (pos
!= browser
->entries
);
90 static struct list_head
*
91 ui_browser__list_head_filter_prev_entries(struct ui_browser
*browser
,
92 struct list_head
*pos
)
95 if (!browser
->filter
|| !browser
->filter(browser
, pos
))
98 } while (pos
!= browser
->entries
);
103 void ui_browser__list_head_seek(struct ui_browser
*browser
, off_t offset
, int whence
)
105 struct list_head
*head
= browser
->entries
;
106 struct list_head
*pos
;
108 if (browser
->nr_entries
== 0)
113 pos
= ui_browser__list_head_filter_entries(browser
, head
->next
);
119 pos
= ui_browser__list_head_filter_prev_entries(browser
, head
->prev
);
128 while (offset
-- != 0)
129 pos
= ui_browser__list_head_filter_entries(browser
, pos
->next
);
131 while (offset
++ != 0)
132 pos
= ui_browser__list_head_filter_prev_entries(browser
, pos
->prev
);
138 void ui_browser__rb_tree_seek(struct ui_browser
*browser
, off_t offset
, int whence
)
140 struct rb_root
*root
= browser
->entries
;
158 while (offset
-- != 0)
161 while (offset
++ != 0)
168 unsigned int ui_browser__rb_tree_refresh(struct ui_browser
*browser
)
173 if (browser
->top
== NULL
)
174 browser
->top
= rb_first(browser
->entries
);
179 ui_browser__gotorc(browser
, row
, 0);
180 browser
->write(browser
, nd
, row
);
181 if (++row
== browser
->rows
)
189 bool ui_browser__is_current_entry(struct ui_browser
*browser
, unsigned row
)
191 return browser
->top_idx
+ row
== browser
->index
;
194 void ui_browser__refresh_dimensions(struct ui_browser
*browser
)
196 browser
->width
= SLtt_Screen_Cols
- 1;
197 browser
->height
= browser
->rows
= SLtt_Screen_Rows
- 2;
198 browser
->rows
-= browser
->extra_title_lines
;
203 void ui_browser__handle_resize(struct ui_browser
*browser
)
205 ui__refresh_dimensions(false);
206 ui_browser__show(browser
, browser
->title
?: "", ui_helpline__current
);
207 ui_browser__refresh(browser
);
210 int ui_browser__warning(struct ui_browser
*browser
, int timeout
,
211 const char *format
, ...)
217 va_start(args
, format
);
218 err
= vasprintf(&text
, format
, args
);
222 va_start(args
, format
);
223 ui_helpline__vpush(format
, args
);
226 while ((key
= ui__question_window("Warning!", text
,
228 timeout
)) == K_RESIZE
)
229 ui_browser__handle_resize(browser
);
236 int ui_browser__help_window(struct ui_browser
*browser
, const char *text
)
240 while ((key
= ui__help_window(text
)) == K_RESIZE
)
241 ui_browser__handle_resize(browser
);
246 bool ui_browser__dialog_yesno(struct ui_browser
*browser
, const char *text
)
250 while ((key
= ui__dialog_yesno(text
)) == K_RESIZE
)
251 ui_browser__handle_resize(browser
);
253 return key
== K_ENTER
|| toupper(key
) == 'Y';
256 void ui_browser__reset_index(struct ui_browser
*browser
)
258 browser
->index
= browser
->top_idx
= 0;
259 browser
->seek(browser
, 0, SEEK_SET
);
262 void __ui_browser__show_title(struct ui_browser
*browser
, const char *title
)
265 ui_browser__set_color(browser
, HE_COLORSET_ROOT
);
266 ui_browser__write_nstring(browser
, title
, browser
->width
+ 1);
269 void ui_browser__show_title(struct ui_browser
*browser
, const char *title
)
271 mutex_lock(&ui__lock
);
272 __ui_browser__show_title(browser
, title
);
273 mutex_unlock(&ui__lock
);
276 int ui_browser__show(struct ui_browser
*browser
, const char *title
,
277 const char *helpline
, ...)
282 if (browser
->refresh_dimensions
== NULL
)
283 browser
->refresh_dimensions
= ui_browser__refresh_dimensions
;
285 browser
->refresh_dimensions(browser
);
287 mutex_lock(&ui__lock
);
288 __ui_browser__show_title(browser
, title
);
290 free(browser
->title
);
291 browser
->title
= strdup(title
);
292 zfree(&browser
->helpline
);
294 va_start(ap
, helpline
);
295 err
= vasprintf(&browser
->helpline
, helpline
, ap
);
298 ui_helpline__push(browser
->helpline
);
299 mutex_unlock(&ui__lock
);
303 void ui_browser__hide(struct ui_browser
*browser
)
305 mutex_lock(&ui__lock
);
307 zfree(&browser
->helpline
);
308 zfree(&browser
->title
);
309 mutex_unlock(&ui__lock
);
312 static void ui_browser__scrollbar_set(struct ui_browser
*browser
)
314 int height
= browser
->height
, h
= 0, pct
= 0,
315 col
= browser
->width
,
318 if (browser
->nr_entries
> 1) {
319 pct
= ((browser
->index
* (browser
->height
- 1)) /
320 (browser
->nr_entries
- 1));
323 SLsmg_set_char_set(1);
326 ui_browser__gotorc(browser
, row
++, col
);
327 SLsmg_write_char(h
== pct
? SLSMG_DIAMOND_CHAR
: SLSMG_CKBRD_CHAR
);
331 SLsmg_set_char_set(0);
334 static int __ui_browser__refresh(struct ui_browser
*browser
)
337 int width
= browser
->width
;
339 row
= browser
->refresh(browser
);
340 ui_browser__set_color(browser
, HE_COLORSET_NORMAL
);
342 if (!browser
->use_navkeypressed
|| browser
->navkeypressed
)
343 ui_browser__scrollbar_set(browser
);
347 SLsmg_fill_region(browser
->y
+ row
+ browser
->extra_title_lines
, browser
->x
,
348 browser
->rows
- row
, width
, ' ');
350 if (browser
->nr_entries
== 0 && browser
->no_samples_msg
)
351 __ui__info_window(NULL
, browser
->no_samples_msg
, NULL
);
355 int ui_browser__refresh(struct ui_browser
*browser
)
357 mutex_lock(&ui__lock
);
358 __ui_browser__refresh(browser
);
359 mutex_unlock(&ui__lock
);
365 * Here we're updating nr_entries _after_ we started browsing, i.e. we have to
366 * forget about any reference to any entry in the underlying data structure,
367 * that is why we do a SEEK_SET. Think about 'perf top' in the hists browser
368 * after an output_resort and hist decay.
370 void ui_browser__update_nr_entries(struct ui_browser
*browser
, u32 nr_entries
)
372 off_t offset
= nr_entries
- browser
->nr_entries
;
374 browser
->nr_entries
= nr_entries
;
377 if (browser
->top_idx
< (u64
)-offset
)
378 offset
= -browser
->top_idx
;
380 browser
->index
+= offset
;
381 browser
->top_idx
+= offset
;
385 browser
->seek(browser
, browser
->top_idx
, SEEK_SET
);
388 int ui_browser__run(struct ui_browser
*browser
, int delay_secs
)
395 mutex_lock(&ui__lock
);
396 err
= __ui_browser__refresh(browser
);
398 mutex_unlock(&ui__lock
);
402 key
= ui__getch(delay_secs
);
404 if (key
== K_RESIZE
) {
405 ui__refresh_dimensions(false);
406 browser
->refresh_dimensions(browser
);
407 __ui_browser__show_title(browser
, browser
->title
);
408 ui_helpline__puts(browser
->helpline
);
412 if (browser
->use_navkeypressed
&& !browser
->navkeypressed
) {
413 if (key
== K_DOWN
|| key
== K_UP
||
414 (browser
->columns
&& (key
== K_LEFT
|| key
== K_RIGHT
)) ||
415 key
== K_PGDN
|| key
== K_PGUP
||
416 key
== K_HOME
|| key
== K_END
||
418 browser
->navkeypressed
= true;
426 if (browser
->index
== browser
->nr_entries
- 1)
429 if (browser
->index
== browser
->top_idx
+ browser
->rows
) {
431 browser
->seek(browser
, +1, SEEK_CUR
);
435 if (browser
->index
== 0)
438 if (browser
->index
< browser
->top_idx
) {
440 browser
->seek(browser
, -1, SEEK_CUR
);
444 if (!browser
->columns
)
446 if (browser
->horiz_scroll
< browser
->columns
- 1)
447 ++browser
->horiz_scroll
;
450 if (!browser
->columns
)
452 if (browser
->horiz_scroll
!= 0)
453 --browser
->horiz_scroll
;
457 if (browser
->top_idx
+ browser
->rows
> browser
->nr_entries
- 1)
460 offset
= browser
->rows
;
461 if (browser
->index
+ offset
> browser
->nr_entries
- 1)
462 offset
= browser
->nr_entries
- 1 - browser
->index
;
463 browser
->index
+= offset
;
464 browser
->top_idx
+= offset
;
465 browser
->seek(browser
, +offset
, SEEK_CUR
);
468 if (browser
->top_idx
== 0)
471 if (browser
->top_idx
< browser
->rows
)
472 offset
= browser
->top_idx
;
474 offset
= browser
->rows
;
476 browser
->index
-= offset
;
477 browser
->top_idx
-= offset
;
478 browser
->seek(browser
, -offset
, SEEK_CUR
);
481 ui_browser__reset_index(browser
);
484 offset
= browser
->rows
- 1;
485 if (offset
>= browser
->nr_entries
)
486 offset
= browser
->nr_entries
- 1;
488 browser
->index
= browser
->nr_entries
- 1;
489 browser
->top_idx
= browser
->index
- offset
;
490 browser
->seek(browser
, -offset
, SEEK_END
);
500 unsigned int ui_browser__list_head_refresh(struct ui_browser
*browser
)
502 struct list_head
*pos
;
503 struct list_head
*head
= browser
->entries
;
506 if (browser
->top
== NULL
|| browser
->top
== browser
->entries
)
507 browser
->top
= ui_browser__list_head_filter_entries(browser
, head
->next
);
511 list_for_each_from(pos
, head
) {
512 if (!browser
->filter
|| !browser
->filter(browser
, pos
)) {
513 ui_browser__gotorc(browser
, row
, 0);
514 browser
->write(browser
, pos
, row
);
515 if (++row
== browser
->rows
)
523 static struct ui_browser_colorset
{
524 const char *name
, *fg
, *bg
;
526 } ui_browser__colorsets
[] = {
528 .colorset
= HE_COLORSET_TOP
,
534 .colorset
= HE_COLORSET_MEDIUM
,
540 .colorset
= HE_COLORSET_NORMAL
,
546 .colorset
= HE_COLORSET_SELECTED
,
552 .colorset
= HE_COLORSET_JUMP_ARROWS
,
553 .name
= "jump_arrows",
558 .colorset
= HE_COLORSET_ADDR
,
564 .colorset
= HE_COLORSET_ROOT
,
575 static int ui_browser__color_config(const char *var
, const char *value
,
576 void *data __maybe_unused
)
578 char *fg
= NULL
, *bg
;
581 /* same dir for all commands */
582 if (!strstarts(var
, "colors.") != 0)
585 for (i
= 0; ui_browser__colorsets
[i
].name
!= NULL
; ++i
) {
586 const char *name
= var
+ 7;
588 if (strcmp(ui_browser__colorsets
[i
].name
, name
) != 0)
595 bg
= strchr(fg
, ',');
600 bg
= skip_spaces(bg
+ 1);
601 ui_browser__colorsets
[i
].bg
= bg
;
602 ui_browser__colorsets
[i
].fg
= fg
;
610 void ui_browser__argv_seek(struct ui_browser
*browser
, off_t offset
, int whence
)
614 browser
->top
= browser
->entries
;
617 browser
->top
= (char **)browser
->top
+ offset
;
620 browser
->top
= (char **)browser
->entries
+ browser
->nr_entries
- 1 + offset
;
625 assert((char **)browser
->top
< (char **)browser
->entries
+ browser
->nr_entries
);
626 assert((char **)browser
->top
>= (char **)browser
->entries
);
629 unsigned int ui_browser__argv_refresh(struct ui_browser
*browser
)
631 unsigned int row
= 0, idx
= browser
->top_idx
;
634 if (browser
->top
== NULL
)
635 browser
->top
= browser
->entries
;
637 pos
= (char **)browser
->top
;
638 while (idx
< browser
->nr_entries
&&
639 row
< (unsigned)SLtt_Screen_Rows
- 1) {
640 assert(pos
< (char **)browser
->entries
+ browser
->nr_entries
);
641 if (!browser
->filter
|| !browser
->filter(browser
, *pos
)) {
642 ui_browser__gotorc(browser
, row
, 0);
643 browser
->write(browser
, pos
, row
);
644 if (++row
== browser
->rows
)
655 void __ui_browser__vline(struct ui_browser
*browser
, unsigned int column
,
658 SLsmg_set_char_set(1);
659 ui_browser__gotorc(browser
, start
, column
);
660 SLsmg_draw_vline(end
- start
+ 1);
661 SLsmg_set_char_set(0);
664 void ui_browser__write_graph(struct ui_browser
*browser __maybe_unused
,
667 SLsmg_set_char_set(1);
668 SLsmg_write_char(graph
);
669 SLsmg_set_char_set(0);
672 static void __ui_browser__line_arrow_up(struct ui_browser
*browser
,
676 unsigned int row
, end_row
;
678 SLsmg_set_char_set(1);
680 if (start
< browser
->top_idx
+ browser
->rows
) {
681 row
= start
- browser
->top_idx
;
682 ui_browser__gotorc(browser
, row
, column
);
683 SLsmg_write_char(SLSMG_LLCORN_CHAR
);
684 ui_browser__gotorc(browser
, row
, column
+ 1);
690 row
= browser
->rows
- 1;
692 if (end
> browser
->top_idx
)
693 end_row
= end
- browser
->top_idx
;
697 ui_browser__gotorc(browser
, end_row
, column
);
698 SLsmg_draw_vline(row
- end_row
+ 1);
700 ui_browser__gotorc(browser
, end_row
, column
);
701 if (end
>= browser
->top_idx
) {
702 SLsmg_write_char(SLSMG_ULCORN_CHAR
);
703 ui_browser__gotorc(browser
, end_row
, column
+ 1);
704 SLsmg_write_char(SLSMG_HLINE_CHAR
);
705 ui_browser__gotorc(browser
, end_row
, column
+ 2);
706 SLsmg_write_char(SLSMG_RARROW_CHAR
);
709 SLsmg_set_char_set(0);
712 static void __ui_browser__line_arrow_down(struct ui_browser
*browser
,
716 unsigned int row
, end_row
;
718 SLsmg_set_char_set(1);
720 if (start
>= browser
->top_idx
) {
721 row
= start
- browser
->top_idx
;
722 ui_browser__gotorc(browser
, row
, column
);
723 SLsmg_write_char(SLSMG_ULCORN_CHAR
);
724 ui_browser__gotorc(browser
, row
, column
+ 1);
732 if (end
>= browser
->top_idx
+ browser
->rows
)
733 end_row
= browser
->rows
- 1;
735 end_row
= end
- browser
->top_idx
;
737 ui_browser__gotorc(browser
, row
, column
);
738 SLsmg_draw_vline(end_row
- row
+ 1);
740 ui_browser__gotorc(browser
, end_row
, column
);
741 if (end
< browser
->top_idx
+ browser
->rows
) {
742 SLsmg_write_char(SLSMG_LLCORN_CHAR
);
743 ui_browser__gotorc(browser
, end_row
, column
+ 1);
744 SLsmg_write_char(SLSMG_HLINE_CHAR
);
745 ui_browser__gotorc(browser
, end_row
, column
+ 2);
746 SLsmg_write_char(SLSMG_RARROW_CHAR
);
749 SLsmg_set_char_set(0);
752 void __ui_browser__line_arrow(struct ui_browser
*browser
, unsigned int column
,
756 __ui_browser__line_arrow_up(browser
, column
, start
, end
);
758 __ui_browser__line_arrow_down(browser
, column
, start
, end
);
761 void ui_browser__mark_fused(struct ui_browser
*browser
, unsigned int column
,
762 unsigned int row
, int diff
, bool arrow_down
)
769 SLsmg_set_char_set(1);
772 if (row
+ diff
<= browser
->top_idx
)
775 end_row
= row
+ diff
- browser
->top_idx
;
776 ui_browser__gotorc(browser
, end_row
, column
- 1);
777 SLsmg_write_char(SLSMG_LTEE_CHAR
);
779 while (--end_row
>= 0 && end_row
> (int)(row
- browser
->top_idx
)) {
780 ui_browser__gotorc(browser
, end_row
, column
- 1);
784 end_row
= (int)(row
- browser
->top_idx
);
786 ui_browser__gotorc(browser
, end_row
, column
- 1);
787 SLsmg_write_char(SLSMG_ULCORN_CHAR
);
788 ui_browser__gotorc(browser
, end_row
, column
);
792 if (row
< browser
->top_idx
)
795 end_row
= row
- browser
->top_idx
;
796 ui_browser__gotorc(browser
, end_row
, column
- 1);
797 SLsmg_write_char(SLSMG_LTEE_CHAR
);
798 ui_browser__gotorc(browser
, end_row
, column
);
802 SLsmg_set_char_set(0);
805 void ui_browser__init(void)
809 perf_config(ui_browser__color_config
, NULL
);
811 while (ui_browser__colorsets
[i
].name
) {
812 struct ui_browser_colorset
*c
= &ui_browser__colorsets
[i
++];
813 SLtt_set_color(c
->colorset
, c
->name
, c
->fg
, c
->bg
);