nv_perfmon: initial tool

Various fixes by Emil Velikov have been squashed into this commit.

Signed-off-by: Ben Skeggs <bskeggs@redhat.com>

2 files changed, 698 insertions, 2 deletions

diff --git a/bin/Makefile.am b/bin/Makefile.am
index 58f2201a4..a32acd71b 100644
--- a/bin/Makefile.am
+++ b/bin/Makefile.am
@@ -4,12 +4,12 @@ bin_PROGRAMS = \
 	       nv_ri08 nv_ri16 nv_ri32 nv_wi08 nv_wi16 nv_wi32 \
 	       nv_rv08 nv_rv16 nv_rv32 nv_wv08 nv_wv16 nv_wv32 \
 	       nv_rs08 nv_rs16 nv_rs32 nv_ws08 nv_ws16 nv_ws32 \
-	       nv_disp nv_i2c nv_aux nv_init nv_rdfuc
+	       nv_disp nv_i2c nv_aux nv_init nv_rdfuc nv_perfmon
 
 AM_CPPFLAGS = -I$(top_srcdir)/nvkm/include \
 	      -I$(top_srcdir)/nvkm \
 	      -I$(top_srcdir)/lib/
 
-LDADD = -lpciaccess \
+LDADD = -lpciaccess -lncurses -lmenu -lform \
 	$(top_srcdir)/lib/libpciaccessos.la \
 	$(top_srcdir)/nvkm/libnouveau.la
diff --git a/bin/nv_perfmon.c b/bin/nv_perfmon.c
new file mode 100644
index 000000000..9285d7dbe
--- /dev/null
+++ b/bin/nv_perfmon.c
@@ -0,0 +1,696 @@
+/*
+ * Copyright 2012 Red Hat Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ *
+ * Authors: Ben Skeggs
+ */
+
+#include <curses.h>
+#include <signal.h>
+#include <ctype.h>
+#include <menu.h>
+#include <form.h>
+
+#include <core/object.h>
+#include <core/device.h>
+#include <core/class.h>
+
+#include <sys/time.h>
+
+static struct nouveau_object *client;
+static struct nouveau_object *device;
+static char **signals;
+static int nr_signals;
+
+#define SEC_US  1000000
+#define REFRESH_PERIOD SEC_US
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+enum ui_colour {
+	UI_DEFAULT = 0,
+	UI_BLACK_GREEN
+};
+
+struct ui_window {
+	WINDOW *win;
+	enum ui_colour colour;
+	const char *status;
+	int  (*create)(struct ui_window *, int c, int r);
+	void (*destroy)(struct ui_window *);
+	void (*redraw)(struct ui_window *);
+	bool (*driver)(struct ui_window *, int k);
+};
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+struct ui_table {
+	struct ui_window w;
+	FIELD **data;
+	FORM   *form;
+	int y, r, c;
+	char *(*layout)(struct ui_table *);
+	void  (*change)(struct ui_table *, FIELD *, int x, int y);
+	void  (*action)(struct ui_table *, FIELD *, int x, int y);
+	void  (*redraw)(struct ui_table *);
+	bool  (*driver)(struct ui_table *, int k);
+};
+
+static void
+ui_field_change(FORM *f)
+{
+	struct ui_table *t = form_userptr(f);
+	if (t->change) {
+		FIELD *f = current_field(t->form);
+		int i = field_index(f);
+		int y = i / t->c;
+		int x = i % t->c;
+		t->change(t, f, x, y);
+	}
+}
+
+static void
+ui_table_destroy(struct ui_window *w)
+{
+	struct ui_table *t = (void *)w;
+	int i = -1;
+
+	unpost_form(t->form);
+	free_form(t->form);
+
+	while (t->data[++i])
+		free_field(t->data[i]);
+	free(t->data);
+}
+
+static int
+ui_table_create(struct ui_window *win, int c, int r)
+{
+	struct ui_table *t = (void *)win;
+	char *layout = t->layout(t);
+	int f = layout[0];
+	int s = layout[1];
+	int l = layout[2];
+	int x, y, n, p;
+	FIELD **v;
+
+	if (s) {
+		n  = c - (f + 1) - 1;
+		n /= (s + 1) + l;
+		l *= n;
+	} else {
+		l = max(l, c - (f + 1));
+		n = 0;
+	}
+
+	t->c = 2 + n;
+	t->r = r;
+
+	t->data = malloc(sizeof(*t->data) * ((t->c * t->r) + 1));
+	for (y = 0; y < t->r; y++) {
+		t->data[(y * t->c) + 0] = new_field(1, f, y, 0, 0, 0);
+		for (x = 1, p = (f + 1); x < n + 1; x++, p += (s + 1))
+			t->data[(y * t->c) + x] = new_field(1, s, y, p, 0, 0);
+		t->data[(y * t->c) + x] = new_field(1, l, y, p, 0, 0);
+	}
+	t->data[y * t->c] = NULL;
+
+	v = t->data - 1;
+	while (*(++v))
+		field_opts_off(*v, O_ACTIVE);
+
+	t->form = new_form(t->data);
+	set_form_sub(t->form, win->win);
+	set_form_userptr(t->form, t);
+	set_field_term(t->form, ui_field_change);
+	post_form(t->form);
+	return 0;
+}
+
+static void
+ui_table_redraw(struct ui_window *w)
+{
+	struct ui_table *t = (void *)w;
+	t->redraw(t);
+}
+
+static bool
+ui_table_driver(struct ui_window *w, int k)
+{
+	struct ui_table *t = (void *)w;
+	FIELD *v = current_field(t->form);
+	int i = field_index(v);
+	int y = i / t->c;
+	int x = i % t->c;
+
+	switch (k) {
+	case ' ':
+		t->y = 0;
+		return true;
+	case KEY_UP:
+// XXX: buggy until we move the initial cursor to the correct location
+		if (y == 2) {
+			if (t->y > 0)
+				t->y--;
+			return true;
+		}
+		form_driver(t->form, REQ_UP_FIELD);
+		break;
+	case KEY_PPAGE:
+		t->y -= (t->r - 2);
+		if (t->y < 0)
+			t->y = 0;
+		return true;
+	case KEY_HOME:
+// TODO: move cursor to top entry
+		t->y = 0;
+		return true;
+	case KEY_DOWN:
+		if ((t->y + (x - 2)) >= (nr_signals - 1))
+			break;
+		if (y == (t->r - 1)) {
+			t->y++;
+			return true;
+		}
+		form_driver(t->form, REQ_DOWN_FIELD);
+		break;
+	case KEY_NPAGE:
+		if ((t->y + (x - 2) + (t->r - 2)) > (nr_signals - 1))
+			break;
+		t->y += (t->r - 2);
+		return true;
+	case KEY_END:
+// TODO: move cursor to top entry
+		t->y = nr_signals - 1;
+		return true;
+	case KEY_LEFT:
+		form_driver(t->form, REQ_LEFT_FIELD);
+		break;
+	case KEY_RIGHT:
+		form_driver(t->form, REQ_RIGHT_FIELD);
+		break;
+	case KEY_ENTER:
+	case '\r':
+	case '\n':
+		if (t->action)
+			t->action(t, v, x, y);
+		break;
+	default:
+		if (!t->driver || t->driver(t, k))
+			form_driver(t->form, k);
+		break;
+	}
+
+	return false;
+}
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+static void
+ui_menu_redraw(struct ui_window *w)
+{
+	const char *name = "nVPerfMon";
+	mvwaddstr(w->win, 0, getmaxx(w->win) - strlen(name) - 1, name);
+}
+
+static struct ui_window
+ui_menu_win = {
+	.colour = UI_BLACK_GREEN,
+	.redraw = ui_menu_redraw,
+	.driver = NULL,
+};
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+struct ui_main {
+	struct list_head head;
+	u32 handle;
+	struct nouveau_object *object;
+	const char *name;
+	u32 clk;
+	u32 ctr;
+	u64 incr;
+};
+
+static struct list_head ui_main_list = LIST_HEAD_INIT(ui_main_list);
+static u32 ui_main_handle = 0xc0000000;
+
+static void
+ui_main_remove(struct ui_main *item)
+{
+	int ret = nouveau_object_del(client, 0x00000000, item->handle);
+	list_del(&item->head);
+	free(item);
+}
+
+static void
+ui_main_select(void)
+{
+	struct ui_main *item, *temp;
+	int ret, i;
+
+	list_for_each_entry_safe(item, temp, &ui_main_list, head) {
+		ui_main_remove(item);
+	}
+
+	for (i = 0; i < nr_signals; i++) {
+		item = calloc(1, sizeof(*item));
+		item->handle = ui_main_handle++;
+		item->object = NULL;
+		item->name = signals[i];
+		item->incr = 0;
+
+		ret = nouveau_object_new(client, 0x00000000, item->handle,
+					 NV_PERFCTR_CLASS,
+					 &(struct nv_perfctr_class) {
+						.logic_op = 0xaaaa,
+						.signal[0].name =
+							(char *)item->name,
+						.signal[0].size =
+							strlen(item->name)
+					 }, sizeof(struct nv_perfctr_class),
+					 &item->object);
+		assert(ret == 0);
+		list_add_tail(&item->head, &ui_main_list);
+	}
+}
+
+static void
+ui_main_alarm_handler(int signal)
+{
+	struct ui_main *item;
+	bool sampled = false;
+
+	if (list_empty(&ui_main_list))
+		ui_main_select();
+
+	list_for_each_entry(item, &ui_main_list, head) {
+		struct nv_perfctr_read args;
+		int ret;
+
+		if (!sampled) {
+			struct nv_perfctr_sample args;
+			ret = nv_exec(item->object, NV_PERFCTR_SAMPLE,
+				     &args, sizeof(args));
+			assert(ret == 0);
+			sampled = true;
+		}
+
+		ret = nv_exec(item->object, NV_PERFCTR_READ, &args, sizeof(args));
+		assert(ret == 0 || ret == -EAGAIN);
+
+		if (ret == 0) {
+			item->clk = args.clk;
+			item->ctr = args.ctr;
+		}
+		item->incr += item->ctr;
+	}
+}
+
+static struct sigaction
+ui_main_alarm = {
+	.sa_handler = ui_main_alarm_handler,
+};
+
+static char *
+ui_main_layout(struct ui_table *t)
+{
+	return "\x10\x00\x01";
+}
+
+static void
+ui_main_action(struct ui_table *t, FIELD *f, int x, int y)
+{
+	void *priv = field_userptr(f);
+	(void)priv;
+}
+
+static void
+ui_main_redraw(struct ui_table *t)
+{
+	struct ui_main *item;
+	FIELD **f = t->data;
+	char b[128];
+	int y;
+
+	set_field_buffer(f[1], 0, "   Samples      Count      %            Total");
+	f += t->c * 2;
+// TODO: move cursor to top entry
+
+	y = -1;
+	list_for_each_entry(item, &ui_main_list, head) {
+		if (++y >= t->y)
+			break;
+	}
+
+	y = 2;
+	list_for_each_entry_from(item, &ui_main_list, head) {
+		set_field_buffer(f[0], 0, item->name);
+		set_field_userptr(f[0], item);
+		field_opts_on(f[0], O_VISIBLE | O_ACTIVE);
+
+		snprintf(b, sizeof(b), "%10u %10u %6.2f %16"PRIu64,
+			    item->clk, item->ctr,
+			    (float)item->ctr * 100.0 / item->clk, item->incr);
+		set_field_buffer(f[1], 0, b);
+		field_opts_on(f[1], O_VISIBLE);
+
+		if (++y == t->r)
+			break;
+		f += t->c;
+	}
+
+	for (; y < t->r; y++) {
+		field_opts_off(f[0], O_VISIBLE | O_ACTIVE);
+		field_opts_off(f[1], O_VISIBLE);
+		set_field_userptr(f[0], NULL);
+		f += t->c;
+	}
+}
+
+static bool
+ui_main_driver(struct ui_window *w, int k)
+{
+	struct ui_table *t = (void *)w;
+	struct ui_main *item, *temp;
+	FIELD *f;
+
+	switch (k) {
+	case KEY_DC:
+	case KEY_DL:
+		f = current_field(t->form);
+		item = field_userptr(f);
+		if (item) {
+			ui_main_remove(item);
+			return true;
+		}
+		break;
+	case 'x':
+		list_for_each_entry_safe(item, temp, &ui_main_list, head) {
+			if (item->incr == 0)
+				ui_main_remove(item);
+		}
+		break;
+	default:
+		return ui_table_driver(w, k);
+	}
+
+	return false;
+}
+
+static int
+ui_main_create(struct ui_window *win, int c, int r)
+{
+	int ret = ui_table_create(win, c, r);
+	if (ret == 0) {
+		sigaction(SIGALRM, &ui_main_alarm, 0);
+		ualarm(1, 999999);
+	}
+	return ret;
+}
+
+static struct ui_table
+ui_main_f = {
+	.w.status = "main",
+	.w.colour = UI_DEFAULT,
+	.w.create = ui_main_create,
+	.w.destroy = ui_table_destroy,
+	.w.redraw = ui_table_redraw,
+	.w.driver = ui_main_driver,
+	.layout = ui_main_layout,
+	.action = ui_main_action,
+	.redraw = ui_main_redraw,
+};
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+static void
+ui_stat_redraw(struct ui_window *w)
+{
+}
+
+static struct ui_window
+ui_stat_win = {
+	.colour = UI_BLACK_GREEN,
+	.redraw = ui_stat_redraw,
+	.driver = NULL,
+};
+
+/*******************************************************************************
+ *
+ ******************************************************************************/
+
+static struct ui_layout {
+	struct ui_window *w;
+	int parent;
+	int horiz;
+	int percent;
+	int adjust;
+	int border;
+} ui[] = {
+	{ &ui_menu_win, -1, 0,   0, +1, 0 },
+	{ &ui_main_f.w, -1, 0, 100, -2, 0 },
+	{ &ui_stat_win, -1, 0,   0, +1, 0 },
+	{}
+}, *active = &ui[1];
+
+static void
+ui_redraw_win(struct ui_window *w)
+{
+	wclear(w->win);
+	if (w->redraw)
+		w->redraw(w);
+	wrefresh(w->win);
+}
+
+static void
+ui_active(int adjust)
+{
+	do {
+		if ((active += adjust)->w == NULL) {
+			active = &ui[0];
+			adjust = 1;
+		}
+	} while(!active->w->driver);
+	wrefresh(active->w->win);
+}
+
+static void
+ui_redraw(void)
+{
+	struct ui_layout *l = ui - 1;
+	struct ui_window *w;
+
+	clear();
+	refresh();
+
+	while ((w = (++l)->w))
+		ui_redraw_win(w);
+	ui_active(0);
+}
+
+static void
+ui_create(void)
+{
+	struct ui_layout *l = ui - 1;
+	struct ui_window *w;
+	WINDOW *p = stdscr;
+	int mc = COLS, mr = LINES;
+	int ac = 0, ar = 0;
+	int x, y, c, r;
+	int lp = -1;
+
+	refresh();
+
+	while ((w = (++l)->w)) {
+		if (l->parent != lp) {
+			 p = ui[l->parent].w->win;
+			mc = getmaxx(p);
+			mr = getmaxy(p);
+			lp = l->parent;
+			ac = ar = 0;
+		}
+
+		if (l->horiz) {
+			if ((c = l->percent)) {
+				if ((l + 1)->parent == l->parent)
+					c = ((mc * l->percent) / 100);
+				else
+					c = mc - ac;
+			}
+			c = c + l->adjust;
+			r = mr + (l->adjust * l->border);
+			y = 0;
+			x = ac; ac += c;
+		} else {
+			c = mc + (l->adjust * l->border);
+			if ((r = l->percent)) {
+				if ((l + 1)->parent == l->parent)
+					r = ((mr * l->percent) / 100);
+				else
+					r = mr - ar;
+			}
+			r = r + l->adjust;
+			x = 0;
+			y = ar; ar += r;
+		}
+
+		x += l->border;
+		y += l->border;
+
+		w->win = derwin(p, r, c, y, x);
+		wbkgd(w->win, COLOR_PAIR(w->colour));
+		if (w->create)
+			w->create(w, c, r);
+	}
+
+	ui_redraw();
+}
+
+static void
+ui_destroy(void)
+{
+	struct ui_layout *l = ui + ARRAY_SIZE(ui);
+	struct ui_window *w;
+
+	while ((w = (--l)->w)) {
+		if (w->destroy)
+			w->destroy(w);
+		delwin(w->win);
+		w->win = NULL;
+	}
+}
+
+static void
+ui_resize(void)
+{
+	ui_destroy();
+	ui_create();
+}
+
+int
+main(int argc, char **argv)
+{
+	struct nv_perfctr_query args = {};
+	struct nouveau_object *object;
+	int ret, c, k;
+	int scan = 0;
+
+	while ((c = getopt(argc, argv, "-s")) != -1) {
+		switch (c) {
+		case 's':
+			scan = 1;
+			break;
+		case 1:
+			return -EINVAL;
+		}
+	}
+
+
+	ret = os_client_new(NULL, "error", argc, argv, &client);
+	if (ret)
+		return ret;
+
+	ret = nouveau_object_new(client, 0xffffffff, 0x00000000,
+				 NV_DEVICE_CLASS, &(struct nv_device_class) {
+					.device = ~0ULL,
+					.disable = ~(NV_DEVICE_DISABLE_MMIO |
+						     NV_DEVICE_DISABLE_VBIOS |
+						     NV_DEVICE_DISABLE_CORE |
+						     NV_DEVICE_DISABLE_IDENTIFY),
+					.debug0 = ~((1ULL << NVDEV_SUBDEV_TIMER) |
+						    (1ULL << NVDEV_ENGINE_PERFMON)),
+				}, sizeof(struct nv_device_class), &device);
+	if (ret)
+		return ret;
+
+	if (scan) {
+		fprintf(stderr, "unimplemented\n");
+		return 1;
+	}
+
+	ret = nouveau_object_new(client, 0x00000000, 0xdeadbeef,
+				 NV_PERFCTR_CLASS, &(struct nv_perfctr_class) {
+				 }, sizeof(struct nv_perfctr_class), &object);
+	assert(ret == 0);
+	do {
+		u32 prev_iter = args.iter;
+
+		args.name = NULL;
+		ret = nv_exec(object, NV_PERFCTR_QUERY, &args, sizeof(args));
+		assert(ret == 0);
+
+		if (prev_iter) {
+			nr_signals++;
+			signals = realloc(signals, nr_signals * sizeof(char*));
+			signals[nr_signals - 1] = malloc(args.size);
+
+			args.iter = prev_iter;
+			args.name = signals[nr_signals - 1];
+			ret = nv_exec(object, NV_PERFCTR_QUERY,
+				     &args, sizeof(args));
+			assert(ret == 0);
+		}
+	} while (args.iter != 0xffffffff);
+	nouveau_object_del(client, 0x00000000, 0xdeadbeef);
+
+	initscr();
+	keypad(stdscr, TRUE);
+	nonl();
+	cbreak();
+	noecho();
+
+	if (has_colors())
+		start_color();
+	init_pair(UI_BLACK_GREEN, COLOR_BLACK, COLOR_GREEN);
+
+	ui_create();
+
+	while ((k = getch()) != '\x1b') {
+		switch (k) {
+		case KEY_RESIZE:
+			ui_resize();
+			break;
+		default:
+			active->w->driver(active->w, k);
+			break;
+		}
+
+		active->w->redraw(active->w);
+	}
+
+	ui_destroy();
+	endwin();
+
+	while (nr_signals--)
+		free(signals[nr_signals]);
+	free(signals);
+	return 0;
+}