browser.c 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598
  1. #include "../util.h"
  2. #include "../cache.h"
  3. #include "../../perf.h"
  4. #include "libslang.h"
  5. #include <newt.h>
  6. #include "ui.h"
  7. #include "util.h"
  8. #include <linux/compiler.h>
  9. #include <linux/list.h>
  10. #include <linux/rbtree.h>
  11. #include <stdlib.h>
  12. #include <sys/ttydefaults.h>
  13. #include "browser.h"
  14. #include "helpline.h"
  15. #include "keysyms.h"
  16. #include "../color.h"
  17. static int ui_browser__percent_color(struct ui_browser *browser,
  18. double percent, bool current)
  19. {
  20. if (current && (!browser->use_navkeypressed || browser->navkeypressed))
  21. return HE_COLORSET_SELECTED;
  22. if (percent >= MIN_RED)
  23. return HE_COLORSET_TOP;
  24. if (percent >= MIN_GREEN)
  25. return HE_COLORSET_MEDIUM;
  26. return HE_COLORSET_NORMAL;
  27. }
  28. void ui_browser__set_color(struct ui_browser *self __used, int color)
  29. {
  30. SLsmg_set_color(color);
  31. }
  32. void ui_browser__set_percent_color(struct ui_browser *self,
  33. double percent, bool current)
  34. {
  35. int color = ui_browser__percent_color(self, percent, current);
  36. ui_browser__set_color(self, color);
  37. }
  38. void ui_browser__gotorc(struct ui_browser *self, int y, int x)
  39. {
  40. SLsmg_gotorc(self->y + y, self->x + x);
  41. }
  42. static struct list_head *
  43. ui_browser__list_head_filter_entries(struct ui_browser *browser,
  44. struct list_head *pos)
  45. {
  46. do {
  47. if (!browser->filter || !browser->filter(browser, pos))
  48. return pos;
  49. pos = pos->next;
  50. } while (pos != browser->entries);
  51. return NULL;
  52. }
  53. static struct list_head *
  54. ui_browser__list_head_filter_prev_entries(struct ui_browser *browser,
  55. struct list_head *pos)
  56. {
  57. do {
  58. if (!browser->filter || !browser->filter(browser, pos))
  59. return pos;
  60. pos = pos->prev;
  61. } while (pos != browser->entries);
  62. return NULL;
  63. }
  64. void ui_browser__list_head_seek(struct ui_browser *self, off_t offset, int whence)
  65. {
  66. struct list_head *head = self->entries;
  67. struct list_head *pos;
  68. if (self->nr_entries == 0)
  69. return;
  70. switch (whence) {
  71. case SEEK_SET:
  72. pos = ui_browser__list_head_filter_entries(self, head->next);
  73. break;
  74. case SEEK_CUR:
  75. pos = self->top;
  76. break;
  77. case SEEK_END:
  78. pos = ui_browser__list_head_filter_prev_entries(self, head->prev);
  79. break;
  80. default:
  81. return;
  82. }
  83. assert(pos != NULL);
  84. if (offset > 0) {
  85. while (offset-- != 0)
  86. pos = ui_browser__list_head_filter_entries(self, pos->next);
  87. } else {
  88. while (offset++ != 0)
  89. pos = ui_browser__list_head_filter_prev_entries(self, pos->prev);
  90. }
  91. self->top = pos;
  92. }
  93. void ui_browser__rb_tree_seek(struct ui_browser *self, off_t offset, int whence)
  94. {
  95. struct rb_root *root = self->entries;
  96. struct rb_node *nd;
  97. switch (whence) {
  98. case SEEK_SET:
  99. nd = rb_first(root);
  100. break;
  101. case SEEK_CUR:
  102. nd = self->top;
  103. break;
  104. case SEEK_END:
  105. nd = rb_last(root);
  106. break;
  107. default:
  108. return;
  109. }
  110. if (offset > 0) {
  111. while (offset-- != 0)
  112. nd = rb_next(nd);
  113. } else {
  114. while (offset++ != 0)
  115. nd = rb_prev(nd);
  116. }
  117. self->top = nd;
  118. }
  119. unsigned int ui_browser__rb_tree_refresh(struct ui_browser *self)
  120. {
  121. struct rb_node *nd;
  122. int row = 0;
  123. if (self->top == NULL)
  124. self->top = rb_first(self->entries);
  125. nd = self->top;
  126. while (nd != NULL) {
  127. ui_browser__gotorc(self, row, 0);
  128. self->write(self, nd, row);
  129. if (++row == self->height)
  130. break;
  131. nd = rb_next(nd);
  132. }
  133. return row;
  134. }
  135. bool ui_browser__is_current_entry(struct ui_browser *self, unsigned row)
  136. {
  137. return self->top_idx + row == self->index;
  138. }
  139. void ui_browser__refresh_dimensions(struct ui_browser *self)
  140. {
  141. self->width = SLtt_Screen_Cols - 1;
  142. self->height = SLtt_Screen_Rows - 2;
  143. self->y = 1;
  144. self->x = 0;
  145. }
  146. void ui_browser__handle_resize(struct ui_browser *browser)
  147. {
  148. ui__refresh_dimensions(false);
  149. ui_browser__show(browser, browser->title, ui_helpline__current);
  150. ui_browser__refresh(browser);
  151. }
  152. int ui_browser__warning(struct ui_browser *browser, int timeout,
  153. const char *format, ...)
  154. {
  155. va_list args;
  156. char *text;
  157. int key = 0, err;
  158. va_start(args, format);
  159. err = vasprintf(&text, format, args);
  160. va_end(args);
  161. if (err < 0) {
  162. va_start(args, format);
  163. ui_helpline__vpush(format, args);
  164. va_end(args);
  165. } else {
  166. while ((key == ui__question_window("Warning!", text,
  167. "Press any key...",
  168. timeout)) == K_RESIZE)
  169. ui_browser__handle_resize(browser);
  170. free(text);
  171. }
  172. return key;
  173. }
  174. int ui_browser__help_window(struct ui_browser *browser, const char *text)
  175. {
  176. int key;
  177. while ((key = ui__help_window(text)) == K_RESIZE)
  178. ui_browser__handle_resize(browser);
  179. return key;
  180. }
  181. bool ui_browser__dialog_yesno(struct ui_browser *browser, const char *text)
  182. {
  183. int key;
  184. while ((key = ui__dialog_yesno(text)) == K_RESIZE)
  185. ui_browser__handle_resize(browser);
  186. return key == K_ENTER || toupper(key) == 'Y';
  187. }
  188. void ui_browser__reset_index(struct ui_browser *self)
  189. {
  190. self->index = self->top_idx = 0;
  191. self->seek(self, 0, SEEK_SET);
  192. }
  193. void __ui_browser__show_title(struct ui_browser *browser, const char *title)
  194. {
  195. SLsmg_gotorc(0, 0);
  196. ui_browser__set_color(browser, NEWT_COLORSET_ROOT);
  197. slsmg_write_nstring(title, browser->width + 1);
  198. }
  199. void ui_browser__show_title(struct ui_browser *browser, const char *title)
  200. {
  201. pthread_mutex_lock(&ui__lock);
  202. __ui_browser__show_title(browser, title);
  203. pthread_mutex_unlock(&ui__lock);
  204. }
  205. int ui_browser__show(struct ui_browser *self, const char *title,
  206. const char *helpline, ...)
  207. {
  208. int err;
  209. va_list ap;
  210. ui_browser__refresh_dimensions(self);
  211. pthread_mutex_lock(&ui__lock);
  212. __ui_browser__show_title(self, title);
  213. self->title = title;
  214. free(self->helpline);
  215. self->helpline = NULL;
  216. va_start(ap, helpline);
  217. err = vasprintf(&self->helpline, helpline, ap);
  218. va_end(ap);
  219. if (err > 0)
  220. ui_helpline__push(self->helpline);
  221. pthread_mutex_unlock(&ui__lock);
  222. return err ? 0 : -1;
  223. }
  224. void ui_browser__hide(struct ui_browser *browser __used)
  225. {
  226. pthread_mutex_lock(&ui__lock);
  227. ui_helpline__pop();
  228. pthread_mutex_unlock(&ui__lock);
  229. }
  230. static void ui_browser__scrollbar_set(struct ui_browser *browser)
  231. {
  232. int height = browser->height, h = 0, pct = 0,
  233. col = browser->width,
  234. row = browser->y - 1;
  235. if (browser->nr_entries > 1) {
  236. pct = ((browser->index * (browser->height - 1)) /
  237. (browser->nr_entries - 1));
  238. }
  239. SLsmg_set_char_set(1);
  240. while (h < height) {
  241. ui_browser__gotorc(browser, row++, col);
  242. SLsmg_write_char(h == pct ? SLSMG_DIAMOND_CHAR : SLSMG_CKBRD_CHAR);
  243. ++h;
  244. }
  245. SLsmg_set_char_set(0);
  246. }
  247. static int __ui_browser__refresh(struct ui_browser *browser)
  248. {
  249. int row;
  250. int width = browser->width;
  251. row = browser->refresh(browser);
  252. ui_browser__set_color(browser, HE_COLORSET_NORMAL);
  253. if (!browser->use_navkeypressed || browser->navkeypressed)
  254. ui_browser__scrollbar_set(browser);
  255. else
  256. width += 1;
  257. SLsmg_fill_region(browser->y + row, browser->x,
  258. browser->height - row, width, ' ');
  259. return 0;
  260. }
  261. int ui_browser__refresh(struct ui_browser *browser)
  262. {
  263. pthread_mutex_lock(&ui__lock);
  264. __ui_browser__refresh(browser);
  265. pthread_mutex_unlock(&ui__lock);
  266. return 0;
  267. }
  268. /*
  269. * Here we're updating nr_entries _after_ we started browsing, i.e. we have to
  270. * forget about any reference to any entry in the underlying data structure,
  271. * that is why we do a SEEK_SET. Think about 'perf top' in the hists browser
  272. * after an output_resort and hist decay.
  273. */
  274. void ui_browser__update_nr_entries(struct ui_browser *browser, u32 nr_entries)
  275. {
  276. off_t offset = nr_entries - browser->nr_entries;
  277. browser->nr_entries = nr_entries;
  278. if (offset < 0) {
  279. if (browser->top_idx < (u64)-offset)
  280. offset = -browser->top_idx;
  281. browser->index += offset;
  282. browser->top_idx += offset;
  283. }
  284. browser->top = NULL;
  285. browser->seek(browser, browser->top_idx, SEEK_SET);
  286. }
  287. int ui_browser__run(struct ui_browser *self, int delay_secs)
  288. {
  289. int err, key;
  290. while (1) {
  291. off_t offset;
  292. pthread_mutex_lock(&ui__lock);
  293. err = __ui_browser__refresh(self);
  294. SLsmg_refresh();
  295. pthread_mutex_unlock(&ui__lock);
  296. if (err < 0)
  297. break;
  298. key = ui__getch(delay_secs);
  299. if (key == K_RESIZE) {
  300. ui__refresh_dimensions(false);
  301. ui_browser__refresh_dimensions(self);
  302. __ui_browser__show_title(self, self->title);
  303. ui_helpline__puts(self->helpline);
  304. continue;
  305. }
  306. if (self->use_navkeypressed && !self->navkeypressed) {
  307. if (key == K_DOWN || key == K_UP ||
  308. key == K_PGDN || key == K_PGUP ||
  309. key == K_HOME || key == K_END ||
  310. key == ' ') {
  311. self->navkeypressed = true;
  312. continue;
  313. } else
  314. return key;
  315. }
  316. switch (key) {
  317. case K_DOWN:
  318. if (self->index == self->nr_entries - 1)
  319. break;
  320. ++self->index;
  321. if (self->index == self->top_idx + self->height) {
  322. ++self->top_idx;
  323. self->seek(self, +1, SEEK_CUR);
  324. }
  325. break;
  326. case K_UP:
  327. if (self->index == 0)
  328. break;
  329. --self->index;
  330. if (self->index < self->top_idx) {
  331. --self->top_idx;
  332. self->seek(self, -1, SEEK_CUR);
  333. }
  334. break;
  335. case K_PGDN:
  336. case ' ':
  337. if (self->top_idx + self->height > self->nr_entries - 1)
  338. break;
  339. offset = self->height;
  340. if (self->index + offset > self->nr_entries - 1)
  341. offset = self->nr_entries - 1 - self->index;
  342. self->index += offset;
  343. self->top_idx += offset;
  344. self->seek(self, +offset, SEEK_CUR);
  345. break;
  346. case K_PGUP:
  347. if (self->top_idx == 0)
  348. break;
  349. if (self->top_idx < self->height)
  350. offset = self->top_idx;
  351. else
  352. offset = self->height;
  353. self->index -= offset;
  354. self->top_idx -= offset;
  355. self->seek(self, -offset, SEEK_CUR);
  356. break;
  357. case K_HOME:
  358. ui_browser__reset_index(self);
  359. break;
  360. case K_END:
  361. offset = self->height - 1;
  362. if (offset >= self->nr_entries)
  363. offset = self->nr_entries - 1;
  364. self->index = self->nr_entries - 1;
  365. self->top_idx = self->index - offset;
  366. self->seek(self, -offset, SEEK_END);
  367. break;
  368. default:
  369. return key;
  370. }
  371. }
  372. return -1;
  373. }
  374. unsigned int ui_browser__list_head_refresh(struct ui_browser *self)
  375. {
  376. struct list_head *pos;
  377. struct list_head *head = self->entries;
  378. int row = 0;
  379. if (self->top == NULL || self->top == self->entries)
  380. self->top = ui_browser__list_head_filter_entries(self, head->next);
  381. pos = self->top;
  382. list_for_each_from(pos, head) {
  383. if (!self->filter || !self->filter(self, pos)) {
  384. ui_browser__gotorc(self, row, 0);
  385. self->write(self, pos, row);
  386. if (++row == self->height)
  387. break;
  388. }
  389. }
  390. return row;
  391. }
  392. static struct ui_browser__colorset {
  393. const char *name, *fg, *bg;
  394. int colorset;
  395. } ui_browser__colorsets[] = {
  396. {
  397. .colorset = HE_COLORSET_TOP,
  398. .name = "top",
  399. .fg = "red",
  400. .bg = "default",
  401. },
  402. {
  403. .colorset = HE_COLORSET_MEDIUM,
  404. .name = "medium",
  405. .fg = "green",
  406. .bg = "default",
  407. },
  408. {
  409. .colorset = HE_COLORSET_NORMAL,
  410. .name = "normal",
  411. .fg = "default",
  412. .bg = "default",
  413. },
  414. {
  415. .colorset = HE_COLORSET_SELECTED,
  416. .name = "selected",
  417. .fg = "black",
  418. .bg = "lightgray",
  419. },
  420. {
  421. .colorset = HE_COLORSET_CODE,
  422. .name = "code",
  423. .fg = "blue",
  424. .bg = "default",
  425. },
  426. {
  427. .name = NULL,
  428. }
  429. };
  430. static int ui_browser__color_config(const char *var, const char *value,
  431. void *data __used)
  432. {
  433. char *fg = NULL, *bg;
  434. int i;
  435. /* same dir for all commands */
  436. if (prefixcmp(var, "colors.") != 0)
  437. return 0;
  438. for (i = 0; ui_browser__colorsets[i].name != NULL; ++i) {
  439. const char *name = var + 7;
  440. if (strcmp(ui_browser__colorsets[i].name, name) != 0)
  441. continue;
  442. fg = strdup(value);
  443. if (fg == NULL)
  444. break;
  445. bg = strchr(fg, ',');
  446. if (bg == NULL)
  447. break;
  448. *bg = '\0';
  449. while (isspace(*++bg));
  450. ui_browser__colorsets[i].bg = bg;
  451. ui_browser__colorsets[i].fg = fg;
  452. return 0;
  453. }
  454. free(fg);
  455. return -1;
  456. }
  457. void ui_browser__argv_seek(struct ui_browser *browser, off_t offset, int whence)
  458. {
  459. switch (whence) {
  460. case SEEK_SET:
  461. browser->top = browser->entries;
  462. break;
  463. case SEEK_CUR:
  464. browser->top = browser->top + browser->top_idx + offset;
  465. break;
  466. case SEEK_END:
  467. browser->top = browser->top + browser->nr_entries + offset;
  468. break;
  469. default:
  470. return;
  471. }
  472. }
  473. unsigned int ui_browser__argv_refresh(struct ui_browser *browser)
  474. {
  475. unsigned int row = 0, idx = browser->top_idx;
  476. char **pos;
  477. if (browser->top == NULL)
  478. browser->top = browser->entries;
  479. pos = (char **)browser->top;
  480. while (idx < browser->nr_entries) {
  481. if (!browser->filter || !browser->filter(browser, *pos)) {
  482. ui_browser__gotorc(browser, row, 0);
  483. browser->write(browser, pos, row);
  484. if (++row == browser->height)
  485. break;
  486. }
  487. ++idx;
  488. ++pos;
  489. }
  490. return row;
  491. }
  492. void ui_browser__init(void)
  493. {
  494. int i = 0;
  495. perf_config(ui_browser__color_config, NULL);
  496. while (ui_browser__colorsets[i].name) {
  497. struct ui_browser__colorset *c = &ui_browser__colorsets[i++];
  498. sltt_set_color(c->colorset, c->name, c->fg, c->bg);
  499. }
  500. }