// SPDX-License-Identifier: GPL-2.0
#include "gtk.h"
#include "../evlist.h"
#include "../callchain.h"
#include "../evsel.h"
#include "../sort.h"
#include "../hist.h"
#include "../helpline.h"
#include "../string2.h"
#include <signal.h>
#include <stdlib.h>
#include <linux/string.h>

#define MAX_COLUMNS   32

static int __percent_color_snprintf(struct perf_hpp *hpp, const char *fmt, ...)
{
 int ret = 0;
 int len;
 va_list args;
 double percent;
 const char *markup;
 char *buf = hpp->buf;
 size_t size = hpp->size;

 va_start(args, fmt);
 len = va_arg(args, int);
 percent = va_arg(args, double);
 va_end(args);

 markup = perf_gtk__get_percent_color(percent);
 if (markup)
  ret += scnprintf(buf, size, markup);

 ret += scnprintf(buf + ret, size - ret, fmt, len, percent);

 if (markup)
  ret += scnprintf(buf + ret, size - ret, "</span>");

 return ret;
}

#define __HPP_COLOR_PERCENT_FN(_type, _field)     \
static u64 he_get_##_field(struct hist_entry *he)    \
{          \
 return he->stat._field;       \
}          \
          \
static int perf_gtk__hpp_color_##_type(struct perf_hpp_fmt *fmt,  \
           struct perf_hpp *hpp,   \
           struct hist_entry *he)   \
{          \
 return hpp__fmt(fmt, hpp, he, he_get_##_field, " %*.2f%%",  \
   __percent_color_snprintf, true);   \
}

#define __HPP_COLOR_ACC_PERCENT_FN(_type, _field)    \
static u64 he_get_acc_##_field(struct hist_entry *he)    \
{          \
 return he->stat_acc->_field;      \
}          \
          \
static int perf_gtk__hpp_color_##_type(struct perf_hpp_fmt *fmt,  \
           struct perf_hpp *hpp,   \
           struct hist_entry *he)   \
{          \
 return hpp__fmt_acc(fmt, hpp, he, he_get_acc_##_field, " %*.2f%%",  \
       __percent_color_snprintf, true);   \
}

__HPP_COLOR_PERCENT_FN(overhead, period)
__HPP_COLOR_PERCENT_FN(overhead_sys, period_sys)
__HPP_COLOR_PERCENT_FN(overhead_us, period_us)
__HPP_COLOR_PERCENT_FN(overhead_guest_sys, period_guest_sys)
__HPP_COLOR_PERCENT_FN(overhead_guest_us, period_guest_us)
__HPP_COLOR_ACC_PERCENT_FN(overhead_acc, period)

#undef __HPP_COLOR_PERCENT_FN


void perf_gtk__init_hpp(void)
{
 perf_hpp__format[PERF_HPP__OVERHEAD].color =
    perf_gtk__hpp_color_overhead;
 perf_hpp__format[PERF_HPP__OVERHEAD_SYS].color =
    perf_gtk__hpp_color_overhead_sys;
 perf_hpp__format[PERF_HPP__OVERHEAD_US].color =
    perf_gtk__hpp_color_overhead_us;
 perf_hpp__format[PERF_HPP__OVERHEAD_GUEST_SYS].color =
    perf_gtk__hpp_color_overhead_guest_sys;
 perf_hpp__format[PERF_HPP__OVERHEAD_GUEST_US].color =
    perf_gtk__hpp_color_overhead_guest_us;
 perf_hpp__format[PERF_HPP__OVERHEAD_ACC].color =
    perf_gtk__hpp_color_overhead_acc;
}

static void perf_gtk__add_callchain_flat(struct rb_root *root, GtkTreeStore *store,
      GtkTreeIter *parent, int col, u64 total)
{
 struct rb_node *nd;
 bool has_single_node = (rb_first(root) == rb_last(root));

 for (nd = rb_first(root); nd; nd = rb_next(nd)) {
  struct callchain_node *node;
  struct callchain_list *chain;
  GtkTreeIter iter, new_parent;
  bool need_new_parent;

  node = rb_entry(nd, struct callchain_node, rb_node);

  new_parent = *parent;
  need_new_parent = !has_single_node;

  callchain_node__make_parent_list(node);

  list_for_each_entry(chain, &node->parent_val, list) {
   char buf[128];

   gtk_tree_store_append(store, &iter, &new_parent);

   callchain_node__scnprintf_value(node, buf, sizeof(buf), total);
   gtk_tree_store_set(store, &iter, 0, buf, -1);

   callchain_list__sym_name(chain, buf, sizeof(buf), false);
   gtk_tree_store_set(store, &iter, col, buf, -1);

   if (need_new_parent) {
    /*
 * Only show the top-most symbol in a callchain
 * if it's not the only callchain.
 */
    new_parent = iter;
    need_new_parent = false;
   }
  }

  list_for_each_entry(chain, &node->val, list) {
   char buf[128];

   gtk_tree_store_append(store, &iter, &new_parent);

   callchain_node__scnprintf_value(node, buf, sizeof(buf), total);
   gtk_tree_store_set(store, &iter, 0, buf, -1);

   callchain_list__sym_name(chain, buf, sizeof(buf), false);
   gtk_tree_store_set(store, &iter, col, buf, -1);

   if (need_new_parent) {
    /*
 * Only show the top-most symbol in a callchain
 * if it's not the only callchain.
 */
    new_parent = iter;
    need_new_parent = false;
   }
  }
 }
}

static void perf_gtk__add_callchain_folded(struct rb_root *root, GtkTreeStore *store,
        GtkTreeIter *parent, int col, u64 total)
{
 struct rb_node *nd;

 for (nd = rb_first(root); nd; nd = rb_next(nd)) {
  struct callchain_node *node;
  struct callchain_list *chain;
  GtkTreeIter iter;
  char buf[64];
  char *str, *str_alloc = NULL;
  bool first = true;

  node = rb_entry(nd, struct callchain_node, rb_node);

  callchain_node__make_parent_list(node);

  list_for_each_entry(chain, &node->parent_val, list) {
   char name[1024];

   callchain_list__sym_name(chain, name, sizeof(name), false);

   if (asprintf(&str, "%s%s%s",
         first ? "" : str_alloc,
         first ? "" : symbol_conf.field_sep ?: "; ",
         name) < 0)
    return;

   first = false;
   free(str_alloc);
   str_alloc = str;
  }

  list_for_each_entry(chain, &node->val, list) {
   char name[1024];

   callchain_list__sym_name(chain, name, sizeof(name), false);

   if (asprintf(&str, "%s%s%s",
         first ? "" : str_alloc,
         first ? "" : symbol_conf.field_sep ?: "; ",
         name) < 0)
    return;

   first = false;
   free(str_alloc);
   str_alloc = str;
  }

  gtk_tree_store_append(store, &iter, parent);

  callchain_node__scnprintf_value(node, buf, sizeof(buf), total);
  gtk_tree_store_set(store, &iter, 0, buf, -1);

  gtk_tree_store_set(store, &iter, col, str, -1);

  free(str_alloc);
 }
}

static void perf_gtk__add_callchain_graph(struct rb_root *root, GtkTreeStore *store,
       GtkTreeIter *parent, int col, u64 total)
{
 struct rb_node *nd;
 bool has_single_node = (rb_first(root) == rb_last(root));

 for (nd = rb_first(root); nd; nd = rb_next(nd)) {
  struct callchain_node *node;
  struct callchain_list *chain;
  GtkTreeIter iter, new_parent;
  bool need_new_parent;
  u64 child_total;

  node = rb_entry(nd, struct callchain_node, rb_node);

  new_parent = *parent;
  need_new_parent = !has_single_node && (node->val_nr > 1);

  list_for_each_entry(chain, &node->val, list) {
   char buf[128];

   gtk_tree_store_append(store, &iter, &new_parent);

   callchain_node__scnprintf_value(node, buf, sizeof(buf), total);
   gtk_tree_store_set(store, &iter, 0, buf, -1);

   callchain_list__sym_name(chain, buf, sizeof(buf), false);
   gtk_tree_store_set(store, &iter, col, buf, -1);

   if (need_new_parent) {
    /*
 * Only show the top-most symbol in a callchain
 * if it's not the only callchain.
 */
    new_parent = iter;
    need_new_parent = false;
   }
  }

  if (callchain_param.mode == CHAIN_GRAPH_REL)
   child_total = node->children_hit;
  else
   child_total = total;

  /* Now 'iter' contains info of the last callchain_list */
  perf_gtk__add_callchain_graph(&node->rb_root, store, &iter, col,
           child_total);
 }
}

static void perf_gtk__add_callchain(struct rb_root *root, GtkTreeStore *store,
        GtkTreeIter *parent, int col, u64 total)
{
 if (callchain_param.mode == CHAIN_FLAT)
  perf_gtk__add_callchain_flat(root, store, parent, col, total);
 else if (callchain_param.mode == CHAIN_FOLDED)
  perf_gtk__add_callchain_folded(root, store, parent, col, total);
 else
  perf_gtk__add_callchain_graph(root, store, parent, col, total);
}

static void on_row_activated(GtkTreeView *view, GtkTreePath *path,
        GtkTreeViewColumn *col __maybe_unused,
        gpointer user_data __maybe_unused)
{
 bool expanded = gtk_tree_view_row_expanded(view, path);

 if (expanded)
  gtk_tree_view_collapse_row(view, path);
 else
  gtk_tree_view_expand_row(view, path, FALSE);
}

static void perf_gtk__show_hists(GtkWidget *window, struct hists *hists,
     float min_pcnt)
{
 struct perf_hpp_fmt *fmt;
 GType col_types[MAX_COLUMNS];
 GtkCellRenderer *renderer;
 GtkTreeStore *store;
 struct rb_node *nd;
 GtkWidget *view;
 int col_idx;
 int sym_col = -1;
 int nr_cols;
 char s[512];

 struct perf_hpp hpp = {
  .buf  = s,
  .size  = sizeof(s),
 };

 nr_cols = 0;

 hists__for_each_format(hists, fmt)
  col_types[nr_cols++] = G_TYPE_STRING;

 store = gtk_tree_store_newv(nr_cols, col_types);

 view = gtk_tree_view_new();

 renderer = gtk_cell_renderer_text_new();

 col_idx = 0;

 hists__for_each_format(hists, fmt) {
  if (perf_hpp__should_skip(fmt, hists))
   continue;

  /*
 * XXX no way to determine where symcol column is..
 *     Just use last column for now.
 */
  if (perf_hpp__is_sort_entry(fmt))
   sym_col = col_idx;

  gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(view),
           -1, fmt->name,
           renderer, "markup",
           col_idx++, NULL);
 }

 for (col_idx = 0; col_idx < nr_cols; col_idx++) {
  GtkTreeViewColumn *column;

  column = gtk_tree_view_get_column(GTK_TREE_VIEW(view), col_idx);
  gtk_tree_view_column_set_resizable(column, TRUE);

  if (col_idx == sym_col) {
   gtk_tree_view_set_expander_column(GTK_TREE_VIEW(view),
         column);
  }
 }

 gtk_tree_view_set_model(GTK_TREE_VIEW(view), GTK_TREE_MODEL(store));

 g_object_unref(GTK_TREE_MODEL(store));

 for (nd = rb_first_cached(&hists->entries); nd; nd = rb_next(nd)) {
  struct hist_entry *h = rb_entry(nd, struct hist_entry, rb_node);
  GtkTreeIter iter;
  u64 total = hists__total_period(h->hists);
  float percent;

  if (h->filtered)
   continue;

  percent = hist_entry__get_percent_limit(h);
  if (percent < min_pcnt)
   continue;

  gtk_tree_store_append(store, &iter, NULL);

  col_idx = 0;

  hists__for_each_format(hists, fmt) {
   if (perf_hpp__should_skip(fmt, h->hists))
    continue;

   if (fmt->color)
    fmt->color(fmt, &hpp, h);
   else
    fmt->entry(fmt, &hpp, h);

   gtk_tree_store_set(store, &iter, col_idx++, s, -1);
  }

  if (hist_entry__has_callchains(h) &&
      symbol_conf.use_callchain && hists__has(hists, sym)) {
   if (callchain_param.mode == CHAIN_GRAPH_REL)
    total = symbol_conf.cumulate_callchain ?
     h->stat_acc->period : h->stat.period;

   perf_gtk__add_callchain(&h->sorted_chain, store, &iter,
      sym_col, total);
  }
 }

 gtk_tree_view_set_rules_hint(GTK_TREE_VIEW(view), TRUE);

 g_signal_connect(view, "row-activated",
    G_CALLBACK(on_row_activated), NULL);
 gtk_container_add(GTK_CONTAINER(window), view);
}

static void perf_gtk__add_hierarchy_entries(struct hists *hists,
         struct rb_root_cached *root,
         GtkTreeStore *store,
         GtkTreeIter *parent,
         struct perf_hpp *hpp,
         float min_pcnt)
{
 int col_idx = 0;
 struct rb_node *node;
 struct hist_entry *he;
 struct perf_hpp_fmt *fmt;
 struct perf_hpp_list_node *fmt_node;
 u64 total = hists__total_period(hists);
 int size;

 for (node = rb_first_cached(root); node; node = rb_next(node)) {
  GtkTreeIter iter;
  float percent;
  char *bf;

  he = rb_entry(node, struct hist_entry, rb_node);
  if (he->filtered)
   continue;

  percent = hist_entry__get_percent_limit(he);
  if (percent < min_pcnt)
   continue;

  gtk_tree_store_append(store, &iter, parent);

  col_idx = 0;

  /* the first hpp_list_node is for overhead columns */
  fmt_node = list_first_entry(&hists->hpp_formats,
         struct perf_hpp_list_node, list);
  perf_hpp_list__for_each_format(&fmt_node->hpp, fmt) {
   if (fmt->color)
    fmt->color(fmt, hpp, he);
   else
    fmt->entry(fmt, hpp, he);

   gtk_tree_store_set(store, &iter, col_idx++, hpp->buf, -1);
  }

  bf = hpp->buf;
  size = hpp->size;
  perf_hpp_list__for_each_format(he->hpp_list, fmt) {
   int ret;

   if (fmt->color)
    ret = fmt->color(fmt, hpp, he);
   else
    ret = fmt->entry(fmt, hpp, he);

   snprintf(hpp->buf + ret, hpp->size - ret, "  ");
   advance_hpp(hpp, ret + 2);
  }

  gtk_tree_store_set(store, &iter, col_idx, strim(bf), -1);

  if (!he->leaf) {
   hpp->buf = bf;
   hpp->size = size;

   perf_gtk__add_hierarchy_entries(hists, &he->hroot_out,
       store, &iter, hpp,
       min_pcnt);

   if (!hist_entry__has_hierarchy_children(he, min_pcnt)) {
    char buf[32];
    GtkTreeIter child;

    snprintf(buf, sizeof(buf), "no entry >= %.2f%%",
      min_pcnt);

    gtk_tree_store_append(store, &child, &iter);
    gtk_tree_store_set(store, &child, col_idx, buf, -1);
   }
  }

  if (he->leaf && hist_entry__has_callchains(he) && symbol_conf.use_callchain) {
   if (callchain_param.mode == CHAIN_GRAPH_REL)
    total = symbol_conf.cumulate_callchain ?
     he->stat_acc->period : he->stat.period;

   perf_gtk__add_callchain(&he->sorted_chain, store, &iter,
      col_idx, total);
  }
 }

}

static void perf_gtk__show_hierarchy(GtkWidget *window, struct hists *hists,
         float min_pcnt)
{
 struct perf_hpp_fmt *fmt;
 struct perf_hpp_list_node *fmt_node;
 GType col_types[MAX_COLUMNS];
 GtkCellRenderer *renderer;
 GtkTreeStore *store;
 GtkWidget *view;
 int col_idx;
 int nr_cols = 0;
 char s[512];
 char buf[512];
 bool first_node, first_col;
 struct perf_hpp hpp = {
  .buf  = s,
  .size  = sizeof(s),
 };

 hists__for_each_format(hists, fmt) {
  if (perf_hpp__is_sort_entry(fmt) ||
      perf_hpp__is_dynamic_entry(fmt))
   break;

  col_types[nr_cols++] = G_TYPE_STRING;
 }
 col_types[nr_cols++] = G_TYPE_STRING;

 store = gtk_tree_store_newv(nr_cols, col_types);
 view = gtk_tree_view_new();
 renderer = gtk_cell_renderer_text_new();

 col_idx = 0;

 /* the first hpp_list_node is for overhead columns */
 fmt_node = list_first_entry(&hists->hpp_formats,
        struct perf_hpp_list_node, list);
 perf_hpp_list__for_each_format(&fmt_node->hpp, fmt) {
  gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(view),
           -1, fmt->name,
           renderer, "markup",
           col_idx++, NULL);
 }

 /* construct merged column header since sort keys share single column */
 buf[0] = '\0';
 first_node = true;
 list_for_each_entry_continue(fmt_node, &hists->hpp_formats, list) {
  if (!first_node)
   strcat(buf, " / ");
  first_node = false;

  first_col = true;
  perf_hpp_list__for_each_format(&fmt_node->hpp ,fmt) {
   if (perf_hpp__should_skip(fmt, hists))
    continue;

   if (!first_col)
    strcat(buf, "+");
   first_col = false;

   fmt->header(fmt, &hpp, hists, 0, NULL);
   strcat(buf, strim(hpp.buf));
  }
 }

 gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(view),
          -1, buf,
          renderer, "markup",
          col_idx++, NULL);

 for (col_idx = 0; col_idx < nr_cols; col_idx++) {
  GtkTreeViewColumn *column;

  column = gtk_tree_view_get_column(GTK_TREE_VIEW(view), col_idx);
  gtk_tree_view_column_set_resizable(column, TRUE);

  if (col_idx == 0) {
   gtk_tree_view_set_expander_column(GTK_TREE_VIEW(view),
         column);
  }
 }

 gtk_tree_view_set_model(GTK_TREE_VIEW(view), GTK_TREE_MODEL(store));
 g_object_unref(GTK_TREE_MODEL(store));

 perf_gtk__add_hierarchy_entries(hists, &hists->entries, store,
     NULL, &hpp, min_pcnt);

 gtk_tree_view_set_rules_hint(GTK_TREE_VIEW(view), TRUE);

 g_signal_connect(view, "row-activated",
    G_CALLBACK(on_row_activated), NULL);
 gtk_container_add(GTK_CONTAINER(window), view);
}

int evlist__gtk_browse_hists(struct evlist *evlist, const char *help,
        struct hist_browser_timer *hbt __maybe_unused, float min_pcnt)
{
 struct evsel *pos;
 GtkWidget *vbox;
 GtkWidget *notebook;
 GtkWidget *info_bar;
 GtkWidget *statbar;
 GtkWidget *window;

 signal(SIGSEGV, perf_gtk__signal);
 signal(SIGFPE,  perf_gtk__signal);
 signal(SIGINT,  perf_gtk__signal);
 signal(SIGQUIT, perf_gtk__signal);
 signal(SIGTERM, perf_gtk__signal);

 window = gtk_window_new(GTK_WINDOW_TOPLEVEL);

 gtk_window_set_title(GTK_WINDOW(window), "perf report");

 g_signal_connect(window, "delete_event", gtk_main_quit, NULL);

 pgctx = perf_gtk__activate_context(window);
 if (!pgctx)
  return -1;

 vbox = gtk_vbox_new(FALSE, 0);

 notebook = gtk_notebook_new();

 gtk_box_pack_start(GTK_BOX(vbox), notebook, TRUE, TRUE, 0);

 info_bar = perf_gtk__setup_info_bar();
 if (info_bar)
  gtk_box_pack_start(GTK_BOX(vbox), info_bar, FALSE, FALSE, 0);

 statbar = perf_gtk__setup_statusbar();
 gtk_box_pack_start(GTK_BOX(vbox), statbar, FALSE, FALSE, 0);

 gtk_container_add(GTK_CONTAINER(window), vbox);

 evlist__for_each_entry(evlist, pos) {
  struct hists *hists = evsel__hists(pos);
  const char *evname = evsel__name(pos);
  GtkWidget *scrolled_window;
  GtkWidget *tab_label;
  char buf[512];
  size_t size = sizeof(buf);

  if (symbol_conf.event_group) {
   if (!evsel__is_group_leader(pos))
    continue;

   if (pos->core.nr_members > 1) {
    evsel__group_desc(pos, buf, size);
    evname = buf;
   }
  }

  scrolled_window = gtk_scrolled_window_new(NULL, NULL);

  gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolled_window),
       GTK_POLICY_AUTOMATIC,
       GTK_POLICY_AUTOMATIC);

  if (symbol_conf.report_hierarchy)
   perf_gtk__show_hierarchy(scrolled_window, hists, min_pcnt);
  else
   perf_gtk__show_hists(scrolled_window, hists, min_pcnt);

  tab_label = gtk_label_new(evname);

  gtk_notebook_append_page(GTK_NOTEBOOK(notebook), scrolled_window, tab_label);
 }

 gtk_widget_show_all(window);

 perf_gtk__resize_window(window);

 gtk_window_set_position(GTK_WINDOW(window), GTK_WIN_POS_CENTER);

 ui_helpline__push(help);

 gtk_main();

 perf_gtk__deactivate_context(&pgctx);

 return 0;
}