perf ui browser: Handle K_RESIZE in dialog windows

Just provide wrappers for things like ui__warning, ui__dialog_yesno and
if they return K_RESIZE, refresh dimensions, redraw the entries, etc.

Cc: David Ahern <dsahern@gmail.com>
Cc: Frederic Weisbecker <fweisbec@gmail.com>
Cc: Mike Galbraith <efault@gmx.de>
Cc: Paul Mackerras <paulus@samba.org>
Cc: Peter Zijlstra <peterz@infradead.org>
Cc: Stephane Eranian <eranian@google.com>
Link: http://lkml.kernel.org/n/tip-3ih7hyk9weryxaxb501sfq4u@git.kernel.org
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

7 files changed, 87 insertions, 20 deletions

diff --git a/tools/perf/util/ui/browser.c b/tools/perf/util/ui/browser.c
index 8607efa13564..d2051be04f12 100644
--- a/tools/perf/util/ui/browser.c
+++ b/tools/perf/util/ui/browser.c
@@ -169,6 +169,46 @@ void ui_browser__refresh_dimensions(struct ui_browser *self)
         self->x = 0;
 }
 
+void ui_browser__handle_resize(struct ui_browser *browser)
+{
+        ui__refresh_dimensions(false);
+        ui_browser__show(browser, browser->title, ui_helpline__current);
+        ui_browser__refresh(browser);
+}
+
+int ui_browser__warning(struct ui_browser *browser, const char *format, ...)
+{
+        va_list args;
+        int key;
+
+        va_start(args, format);
+        while ((key = __ui__warning("Warning!", format, args)) == K_RESIZE)
+                ui_browser__handle_resize(browser);
+        va_end(args);
+
+        return key;
+}
+
+int ui_browser__help_window(struct ui_browser *browser, const char *text)
+{
+        int key;
+
+        while ((key = ui__help_window(text)) == K_RESIZE)
+                ui_browser__handle_resize(browser);
+
+        return key;
+}
+
+bool ui_browser__dialog_yesno(struct ui_browser *browser, const char *text)
+{
+        int key;
+
+        while ((key = ui__dialog_yesno(text)) == K_RESIZE)
+                ui_browser__handle_resize(browser);
+
+        return key == K_ENTER || toupper(key) == 'Y';
+}
+
 void ui_browser__reset_index(struct ui_browser *self)
 {
         self->index = self->top_idx = 0;
diff --git a/tools/perf/util/ui/browser.h b/tools/perf/util/ui/browser.h
index 81a8d2afaa9b..fb1c59883e6a 100644
--- a/tools/perf/util/ui/browser.h
+++ b/tools/perf/util/ui/browser.h
@@ -43,6 +43,11 @@ void ui_browser__hide(struct ui_browser *self);
 int ui_browser__refresh(struct ui_browser *self);
 int ui_browser__run(struct ui_browser *browser, int delay_secs);
 void ui_browser__update_nr_entries(struct ui_browser *browser, u32 nr_entries);
+void ui_browser__handle_resize(struct ui_browser *browser);
+
+int ui_browser__warning(struct ui_browser *browser, const char *format, ...);
+int ui_browser__help_window(struct ui_browser *browser, const char *text);
+bool ui_browser__dialog_yesno(struct ui_browser *browser, const char *text);
 
 void ui_browser__argv_seek(struct ui_browser *browser, off_t offset, int whence);
 unsigned int ui_browser__argv_refresh(struct ui_browser *browser);
diff --git a/tools/perf/util/ui/browsers/hists.c b/tools/perf/util/ui/browsers/hists.c
index 44210d471719..b8733c0770cd 100644
--- a/tools/perf/util/ui/browsers/hists.c
+++ b/tools/perf/util/ui/browsers/hists.c
@@ -17,6 +17,7 @@
 #include "../browser.h"
 #include "../helpline.h"
 #include "../util.h"
+#include "../ui.h"
 #include "map.h"
 
 struct hist_browser {
@@ -882,7 +883,7 @@ static int perf_evsel__hists_browse(struct perf_evsel *evsel, int nr_events,
                         goto out_free_stack;
                 case 'a':
                         if (!browser->has_symbols) {
-                                ui__warning(
+                                ui_browser__warning(&browser->b,
                         "Annotation is only available for symbolic views, "
                         "include \"sym\" in --sort to use it.");
                                 continue;
@@ -900,7 +901,8 @@ static int perf_evsel__hists_browse(struct perf_evsel *evsel, int nr_events,
                 case K_F1:
                 case 'h':
                 case '?':
-                        ui__help_window("h/?/F1        Show this window\n"
+                        ui_browser__help_window(&browser->b,
+                                        "h/?/F1        Show this window\n"
                                         "UP/DOWN/PGUP\n"
                                         "PGDN/SPACE    Navigate\n"
                                         "q/ESC/CTRL+C  Exit browser\n\n"
@@ -939,7 +941,8 @@ static int perf_evsel__hists_browse(struct perf_evsel *evsel, int nr_events,
                 }
                 case K_ESC:
                         if (!left_exits &&
-                            !ui__dialog_yesno("Do you really want to exit?"))
+                            !ui_browser__dialog_yesno(&browser->b,
+                                               "Do you really want to exit?"))
                                 continue;
                         /* Fall thru */
                 case 'q':
@@ -992,6 +995,7 @@ add_exit_option:
 
                 if (choice == annotate) {
                         struct hist_entry *he;
+                        int err;
 do_annotate:
                         he = hist_browser__selected_entry(browser);
                         if (he == NULL)
@@ -1000,10 +1004,12 @@ do_annotate:
                          * Don't let this be freed, say, by hists__decay_entry.
                          */
                         he->used = true;
-                        hist_entry__tui_annotate(he, evsel->idx, nr_events,
-                                                 timer, arg, delay_secs);
+                        err = hist_entry__tui_annotate(he, evsel->idx, nr_events,
+                                                       timer, arg, delay_secs);
                         he->used = false;
                         ui_browser__update_nr_entries(&browser->b, browser->hists->nr_entries);
+                        if (err)
+                                ui_browser__handle_resize(&browser->b);
                 } else if (choice == browse_map)
                         map__browse(browser->selection->map);
                 else if (choice == zoom_dso) {
@@ -1132,7 +1138,8 @@ browse_hists:
                                         pos = list_entry(pos->node.prev, struct perf_evsel, node);
                                 goto browse_hists;
                         case K_ESC:
-                                if (!ui__dialog_yesno("Do you really want to exit?"))
+                                if (!ui_browser__dialog_yesno(&menu->b,
+                                                "Do you really want to exit?"))
                                         continue;
                                 /* Fall thru */
                         case 'q':
@@ -1144,7 +1151,8 @@ browse_hists:
                 case K_LEFT:
                         continue;
                 case K_ESC:
-                        if (!ui__dialog_yesno("Do you really want to exit?"))
+                        if (!ui_browser__dialog_yesno(&menu->b,
+                                               "Do you really want to exit?"))
                                 continue;
                         /* Fall thru */
                 case 'q':
diff --git a/tools/perf/util/ui/helpline.c b/tools/perf/util/ui/helpline.c
index 600243d766c0..6ef3c5691762 100644
--- a/tools/perf/util/ui/helpline.c
+++ b/tools/perf/util/ui/helpline.c
@@ -1,6 +1,7 @@
 #define _GNU_SOURCE
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 
 #include "../debug.h"
 #include "helpline.h"
@@ -11,12 +12,17 @@ void ui_helpline__pop(void)
 {
 }
 
+char ui_helpline__current[512];
+
 void ui_helpline__push(const char *msg)
 {
+        const size_t sz = sizeof(ui_helpline__current);
+
         SLsmg_gotorc(SLtt_Screen_Rows - 1, 0);
         SLsmg_set_color(0);
         SLsmg_write_nstring((char *)msg, SLtt_Screen_Cols);
         SLsmg_refresh();
+        strncpy(ui_helpline__current, msg, sz)[sz - 1] = '\0';
 }
 
 void ui_helpline__vpush(const char *fmt, va_list ap)
diff --git a/tools/perf/util/ui/helpline.h b/tools/perf/util/ui/helpline.h
index fdcbc0270acd..7bab6b34e35e 100644
--- a/tools/perf/util/ui/helpline.h
+++ b/tools/perf/util/ui/helpline.h
@@ -11,4 +11,6 @@ void ui_helpline__vpush(const char *fmt, va_list ap);
 void ui_helpline__fpush(const char *fmt, ...);
 void ui_helpline__puts(const char *msg);
 
+extern char ui_helpline__current[];
+
 #endif /* _PERF_UI_HELPLINE_H_ */
diff --git a/tools/perf/util/ui/util.c b/tools/perf/util/ui/util.c
index ef9b79332fe8..45daa7c41dad 100644
--- a/tools/perf/util/ui/util.c
+++ b/tools/perf/util/ui/util.c
@@ -121,43 +121,48 @@ int ui__help_window(const char *text)
         return ui__question_window("Help", text, "Press any key...", 0);
 }
 
-bool ui__dialog_yesno(const char *msg)
+int ui__dialog_yesno(const char *msg)
 {
-        int answer = ui__question_window(NULL, msg, "Enter: Yes, ESC: No", 0);
-
-        return answer == K_ENTER;
+        return ui__question_window(NULL, msg, "Enter: Yes, ESC: No", 0);
 }
 
-static void __ui__warning(const char *title, const char *format, va_list args)
+int __ui__warning(const char *title, const char *format, va_list args)
 {
         char *s;
 
         if (use_browser > 0 && vasprintf(&s, format, args) > 0) {
+                int key;
+
                 pthread_mutex_lock(&ui__lock);
-                ui__question_window(title, s, "Press any key...", 0);
+                key = ui__question_window(title, s, "Press any key...", 0);
                 pthread_mutex_unlock(&ui__lock);
                 free(s);
-                return;
+                return key;
         }
 
         fprintf(stderr, "%s:\n", title);
         vfprintf(stderr, format, args);
+        return K_ESC;
 }
 
-void ui__warning(const char *format, ...)
+int ui__warning(const char *format, ...)
 {
+        int key;
         va_list args;
 
         va_start(args, format);
-        __ui__warning("Warning", format, args);
+        key = __ui__warning("Warning", format, args);
         va_end(args);
+        return key;
 }
 
-void ui__error(const char *format, ...)
+int ui__error(const char *format, ...)
 {
+        int key;
         va_list args;
 
         va_start(args, format);
-        __ui__warning("Error", format, args);
+        key = __ui__warning("Error", format, args);
         va_end(args);
+        return key;
 }
diff --git a/tools/perf/util/ui/util.h b/tools/perf/util/ui/util.h
index 9a25538d4735..2d1738bd71c8 100644
--- a/tools/perf/util/ui/util.h
+++ b/tools/perf/util/ui/util.h
@@ -1,13 +1,14 @@
 #ifndef _PERF_UI_UTIL_H_
 #define _PERF_UI_UTIL_H_ 1
 
-#include <stdbool.h>
+#include <stdarg.h>
 
 int ui__getch(int delay_secs);
 int ui__popup_menu(int argc, char * const argv[]);
 int ui__help_window(const char *text);
-bool ui__dialog_yesno(const char *msg);
+int ui__dialog_yesno(const char *msg);
 int ui__question_window(const char *title, const char *text,
                         const char *exit_msg, int delay_secs);
+int __ui__warning(const char *title, const char *format, va_list args);
 
 #endif /* _PERF_UI_UTIL_H_ */