[net-next-2.6.git] / tools / perf / util / ui / browsers / annotate.c

#include "../browser.h"
#include "../helpline.h"
#include "../libslang.h"
#include "../../hist.h"
#include "../../sort.h"
#include "../../symbol.h"

static void ui__error_window(const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	newtWinMessagev((char *)"Error", (char *)"Ok", (char *)fmt, ap);
	va_end(ap);
}

struct annotate_browser {
	struct ui_browser b;
	struct rb_root	  entries;
	struct rb_node	  *curr_hot;
};

struct objdump_line_rb_node {
	struct rb_node	rb_node;
	double		percent;
	u32		idx;
};

static inline
struct objdump_line_rb_node *objdump_line__rb(struct objdump_line *self)
{
	return (struct objdump_line_rb_node *)(self + 1);
}

static void annotate_browser__write(struct ui_browser *self, void *entry, int row)
{
	struct objdump_line *ol = rb_entry(entry, struct objdump_line, node);
	bool current_entry = ui_browser__is_current_entry(self, row);
	int width = self->width;

	if (ol->offset != -1) {
		struct objdump_line_rb_node *olrb = objdump_line__rb(ol);
		int color = ui_browser__percent_color(olrb->percent, current_entry);
		SLsmg_set_color(color);
		slsmg_printf(" %7.2f ", olrb->percent);
		if (!current_entry)
			SLsmg_set_color(HE_COLORSET_CODE);
	} else {
		int color = ui_browser__percent_color(0, current_entry);
		SLsmg_set_color(color);
		slsmg_write_nstring(" ", 9);
	}

	SLsmg_write_char(':');
	slsmg_write_nstring(" ", 8);
	if (!*ol->line)
		slsmg_write_nstring(" ", width - 18);
	else
		slsmg_write_nstring(ol->line, width - 18);
}

static double objdump_line__calc_percent(struct objdump_line *self,
					 struct list_head *head,
					 struct symbol *sym)
{
	double percent = 0.0;

	if (self->offset != -1) {
		int len = sym->end - sym->start;
		unsigned int hits = 0;
		struct sym_priv *priv = symbol__priv(sym);
		struct sym_ext *sym_ext = priv->ext;
		struct sym_hist *h = priv->hist;
		s64 offset = self->offset;
		struct objdump_line *next = objdump__get_next_ip_line(head, self);


		while (offset < (s64)len &&
		       (next == NULL || offset < next->offset)) {
			if (sym_ext) {
				percent += sym_ext[offset].percent;
			} else
				hits += h->ip[offset];

			++offset;
		}

		if (sym_ext == NULL && h->sum)
			percent = 100.0 * hits / h->sum;
	}

	return percent;
}

static void objdump__insert_line(struct rb_root *self,
				 struct objdump_line_rb_node *line)
{
	struct rb_node **p = &self->rb_node;
	struct rb_node *parent = NULL;
	struct objdump_line_rb_node *l;

	while (*p != NULL) {
		parent = *p;
		l = rb_entry(parent, struct objdump_line_rb_node, rb_node);
		if (line->percent < l->percent)
			p = &(*p)->rb_left;
		else
			p = &(*p)->rb_right;
	}
	rb_link_node(&line->rb_node, parent, p);
	rb_insert_color(&line->rb_node, self);
}

static void annotate_browser__set_top(struct annotate_browser *self,
				      struct rb_node *nd)
{
	struct objdump_line_rb_node *rbpos;
	struct objdump_line *pos;
	unsigned back;

	ui_browser__refresh_dimensions(&self->b);
	back = self->b.height / 2;
	rbpos = rb_entry(nd, struct objdump_line_rb_node, rb_node);
	pos = ((struct objdump_line *)rbpos) - 1;
	self->b.top_idx = self->b.index = rbpos->idx;

	while (self->b.top_idx != 0 && back != 0) {
		pos = list_entry(pos->node.prev, struct objdump_line, node);

		--self->b.top_idx;
		--back;
	}

	self->b.top = pos;
	self->curr_hot = nd;
}

static int annotate_browser__run(struct annotate_browser *self,
				 struct newtExitStruct *es)
{
	struct rb_node *nd;
	struct hist_entry *he = self->b.priv;

	if (ui_browser__show(&self->b, he->ms.sym->name,
			     "<- or ESC: exit, TAB/shift+TAB: cycle thru samples") < 0)
		return -1;

	newtFormAddHotKey(self->b.form, NEWT_KEY_LEFT);
	newtFormAddHotKey(self->b.form, NEWT_KEY_RIGHT);

	nd = self->curr_hot;
	if (nd) {
		newtFormAddHotKey(self->b.form, NEWT_KEY_TAB);
		newtFormAddHotKey(self->b.form, NEWT_KEY_UNTAB);
	}

	while (1) {
		ui_browser__run(&self->b, es);

		if (es->reason != NEWT_EXIT_HOTKEY)
			break;

		switch (es->u.key) {
		case NEWT_KEY_TAB:
			nd = rb_prev(nd);
			if (nd == NULL)
				nd = rb_last(&self->entries);
			annotate_browser__set_top(self, nd);
			break;
		case NEWT_KEY_UNTAB:
			nd = rb_next(nd);
			if (nd == NULL)
				nd = rb_first(&self->entries);
			annotate_browser__set_top(self, nd);
			break;
		default:
			goto out;
		}
	}
out:
	ui_browser__hide(&self->b);
	return es->u.key;
}

int hist_entry__tui_annotate(struct hist_entry *self)
{
	struct newtExitStruct es;
	struct objdump_line *pos, *n;
	struct objdump_line_rb_node *rbpos;
	LIST_HEAD(head);
	struct annotate_browser browser = {
		.b = {
			.entries = &head,
			.refresh = ui_browser__list_head_refresh,
			.seek	 = ui_browser__list_head_seek,
			.write	 = annotate_browser__write,
			.priv	 = self,
		},
	};
	int ret;

	if (self->ms.sym == NULL)
		return -1;

	if (self->ms.map->dso->annotate_warned)
		return -1;

	if (hist_entry__annotate(self, &head, sizeof(*rbpos)) < 0) {
		ui__error_window(ui_helpline__last_msg);
		return -1;
	}

	ui_helpline__push("Press <- or ESC to exit");

	list_for_each_entry(pos, &head, node) {
		size_t line_len = strlen(pos->line);
		if (browser.b.width < line_len)
			browser.b.width = line_len;
		rbpos = objdump_line__rb(pos);
		rbpos->idx = browser.b.nr_entries++;
		rbpos->percent = objdump_line__calc_percent(pos, &head, self->ms.sym);
		if (rbpos->percent < 0.01)
			continue;
		objdump__insert_line(&browser.entries, rbpos);
	}

	/*
	 * Position the browser at the hottest line.
	 */
	browser.curr_hot = rb_last(&browser.entries);
	if (browser.curr_hot)
		annotate_browser__set_top(&browser, browser.curr_hot);

	browser.b.width += 18; /* Percentage */
	ret = annotate_browser__run(&browser, &es);
	list_for_each_entry_safe(pos, n, &head, node) {
		list_del(&pos->node);
		objdump_line__free(pos);
	}
	return ret;
}
Commit	Line	Data
211ef127 ACM	1	#include "../browser.h"
	2	#include "../helpline.h"
	3	#include "../libslang.h"
	4	#include "../../hist.h"
	5	#include "../../sort.h"
	6	#include "../../symbol.h"
	7
	8	static void ui__error_window(const char *fmt, ...)
	9	{
	10	va_list ap;
	11
	12	va_start(ap, fmt);
	13	newtWinMessagev((char )"Error", (char )"Ok", (char *)fmt, ap);
	14	va_end(ap);
	15	}
	16
92221162 ACM	17	struct annotate_browser {
	18	struct ui_browser b;
	19	struct rb_root entries;
f1e9214c	20	struct rb_node *curr_hot;
92221162 ACM	21	};
	22
	23	struct objdump_line_rb_node {
	24	struct rb_node rb_node;
	25	double percent;
	26	u32 idx;
	27	};
	28
	29	static inline
	30	struct objdump_line_rb_node objdump_line__rb(struct objdump_line self)
	31	{
	32	return (struct objdump_line_rb_node *)(self + 1);
	33	}
	34
211ef127 ACM	35	static void annotate_browser__write(struct ui_browser self, void entry, int row)
	36	{
	37	struct objdump_line *ol = rb_entry(entry, struct objdump_line, node);
	38	bool current_entry = ui_browser__is_current_entry(self, row);
	39	int width = self->width;
	40
	41	if (ol->offset != -1) {
92221162 ACM	42	struct objdump_line_rb_node *olrb = objdump_line__rb(ol);
	43	int color = ui_browser__percent_color(olrb->percent, current_entry);
	44	SLsmg_set_color(color);
	45	slsmg_printf(" %7.2f ", olrb->percent);
	46	if (!current_entry)
	47	SLsmg_set_color(HE_COLORSET_CODE);
	48	} else {
	49	int color = ui_browser__percent_color(0, current_entry);
	50	SLsmg_set_color(color);
	51	slsmg_write_nstring(" ", 9);
	52	}
	53
	54	SLsmg_write_char(':');
	55	slsmg_write_nstring(" ", 8);
	56	if (!*ol->line)
	57	slsmg_write_nstring(" ", width - 18);
	58	else
	59	slsmg_write_nstring(ol->line, width - 18);
	60	}
	61
	62	static double objdump_line__calc_percent(struct objdump_line *self,
	63	struct list_head *head,
	64	struct symbol *sym)
	65	{
	66	double percent = 0.0;
	67
	68	if (self->offset != -1) {
	69	int len = sym->end - sym->start;
211ef127	70	unsigned int hits = 0;
211ef127 ACM	71	struct sym_priv *priv = symbol__priv(sym);
	72	struct sym_ext *sym_ext = priv->ext;
	73	struct sym_hist *h = priv->hist;
92221162 ACM	74	s64 offset = self->offset;
	75	struct objdump_line *next = objdump__get_next_ip_line(head, self);
	76
211ef127 ACM	77
	78	while (offset < (s64)len &&
	79	(next == NULL \|\| offset < next->offset)) {
	80	if (sym_ext) {
	81	percent += sym_ext[offset].percent;
	82	} else
	83	hits += h->ip[offset];
	84
	85	++offset;
	86	}
	87
	88	if (sym_ext == NULL && h->sum)
	89	percent = 100.0 * hits / h->sum;
211ef127 ACM	90	}
211ef127 ACM	91
92221162 ACM	92	return percent;
	93	}
	94
	95	static void objdump__insert_line(struct rb_root *self,
	96	struct objdump_line_rb_node *line)
	97	{
	98	struct rb_node **p = &self->rb_node;
	99	struct rb_node *parent = NULL;
	100	struct objdump_line_rb_node *l;
	101
	102	while (*p != NULL) {
	103	parent = *p;
	104	l = rb_entry(parent, struct objdump_line_rb_node, rb_node);
	105	if (line->percent < l->percent)
	106	p = &(*p)->rb_left;
	107	else
	108	p = &(*p)->rb_right;
	109	}
	110	rb_link_node(&line->rb_node, parent, p);
	111	rb_insert_color(&line->rb_node, self);
211ef127 ACM	112	}
211ef127 ACM	113
f1e9214c ACM	114	static void annotate_browser__set_top(struct annotate_browser *self,
	115	struct rb_node *nd)
	116	{
	117	struct objdump_line_rb_node *rbpos;
	118	struct objdump_line *pos;
	119	unsigned back;
	120
	121	ui_browser__refresh_dimensions(&self->b);
	122	back = self->b.height / 2;
	123	rbpos = rb_entry(nd, struct objdump_line_rb_node, rb_node);
	124	pos = ((struct objdump_line *)rbpos) - 1;
	125	self->b.top_idx = self->b.index = rbpos->idx;
	126
	127	while (self->b.top_idx != 0 && back != 0) {
	128	pos = list_entry(pos->node.prev, struct objdump_line, node);
	129
	130	--self->b.top_idx;
	131	--back;
	132	}
	133
	134	self->b.top = pos;
	135	self->curr_hot = nd;
	136	}
	137
	138	static int annotate_browser__run(struct annotate_browser *self,
	139	struct newtExitStruct *es)
	140	{
	141	struct rb_node *nd;
	142	struct hist_entry *he = self->b.priv;
	143
59e8fe32 ACM	144	if (ui_browser__show(&self->b, he->ms.sym->name,
59e8fe32 ACM	145	"<- or ESC: exit, TAB/shift+TAB: cycle thru samples") < 0)
f1e9214c ACM	146	return -1;
f1e9214c ACM	147
f1e9214c	148	newtFormAddHotKey(self->b.form, NEWT_KEY_LEFT);
e9184621	149	newtFormAddHotKey(self->b.form, NEWT_KEY_RIGHT);
f1e9214c ACM	150
	151	nd = self->curr_hot;
	152	if (nd) {
	153	newtFormAddHotKey(self->b.form, NEWT_KEY_TAB);
	154	newtFormAddHotKey(self->b.form, NEWT_KEY_UNTAB);
	155	}
	156
	157	while (1) {
	158	ui_browser__run(&self->b, es);
	159
	160	if (es->reason != NEWT_EXIT_HOTKEY)
	161	break;
	162
	163	switch (es->u.key) {
	164	case NEWT_KEY_TAB:
	165	nd = rb_prev(nd);
	166	if (nd == NULL)
	167	nd = rb_last(&self->entries);
	168	annotate_browser__set_top(self, nd);
	169	break;
	170	case NEWT_KEY_UNTAB:
	171	nd = rb_next(nd);
	172	if (nd == NULL)
	173	nd = rb_first(&self->entries);
	174	annotate_browser__set_top(self, nd);
	175	break;
	176	default:
	177	goto out;
	178	}
	179	}
	180	out:
59e8fe32	181	ui_browser__hide(&self->b);
e9184621	182	return es->u.key;
f1e9214c ACM	183	}
f1e9214c ACM	184
211ef127 ACM	185	int hist_entry__tui_annotate(struct hist_entry *self)
	186	{
	187	struct newtExitStruct es;
	188	struct objdump_line pos, n;
92221162	189	struct objdump_line_rb_node *rbpos;
211ef127	190	LIST_HEAD(head);
92221162 ACM	191	struct annotate_browser browser = {
	192	.b = {
	193	.entries = &head,
	194	.refresh = ui_browser__list_head_refresh,
	195	.seek = ui_browser__list_head_seek,
	196	.write = annotate_browser__write,
	197	.priv = self,
	198	},
211ef127 ACM	199	};
	200	int ret;
	201
	202	if (self->ms.sym == NULL)
	203	return -1;
	204
	205	if (self->ms.map->dso->annotate_warned)
	206	return -1;
	207
92221162	208	if (hist_entry__annotate(self, &head, sizeof(*rbpos)) < 0) {
1e6dd077	209	ui__error_window(ui_helpline__last_msg);
211ef127 ACM	210	return -1;
	211	}
	212
	213	ui_helpline__push("Press <- or ESC to exit");
	214
	215	list_for_each_entry(pos, &head, node) {
	216	size_t line_len = strlen(pos->line);
92221162 ACM	217	if (browser.b.width < line_len)
	218	browser.b.width = line_len;
	219	rbpos = objdump_line__rb(pos);
	220	rbpos->idx = browser.b.nr_entries++;
	221	rbpos->percent = objdump_line__calc_percent(pos, &head, self->ms.sym);
	222	if (rbpos->percent < 0.01)
	223	continue;
	224	objdump__insert_line(&browser.entries, rbpos);
	225	}
	226
	227	/*
	228	* Position the browser at the hottest line.
	229	*/
f1e9214c ACM	230	browser.curr_hot = rb_last(&browser.entries);
	231	if (browser.curr_hot)
	232	annotate_browser__set_top(&browser, browser.curr_hot);
211ef127	233
92221162	234	browser.b.width += 18; /* Percentage */
f1e9214c	235	ret = annotate_browser__run(&browser, &es);
211ef127 ACM	236	list_for_each_entry_safe(pos, n, &head, node) {
	237	list_del(&pos->node);
	238	objdump_line__free(pos);
	239	}
211ef127 ACM	240	return ret;
211ef127 ACM	241	}