/*
 * 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 <nvif/client.h>
#include <nvif/device.h>
#include <nvif/class.h>
#include <core/class.h>

#include <sys/time.h>

static struct nvif_device *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 nvif_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)
{
	nvif_object_fini(&item->object);
	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->name = signals[i];
		item->incr = 0;

		ret = nvif_object_init(nvif_object(device), NULL, 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 = nvif_exec(&item->object, NV_PERFCTR_SAMPLE,
					&args, sizeof(args));
			assert(ret == 0);
			sampled = true;
		}

		ret = nvif_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)
{
	const char *drv = NULL;
	const char *cfg = NULL;
	const char *dbg = "error";
	u64 dev = ~0ULL;
	struct nvif_client *client;
	struct nv_perfctr_query args = {};
	struct nvif_object object;
	int ret, c, k;
	int scan = 0;

	while ((c = getopt(argc, argv, "-a:b:c:d:s")) != -1) {
		switch (c) {
		case 'a': dev = strtoull(optarg, NULL, 0); break;
		case 'b': drv = optarg; break;
		case 'c': cfg = optarg; break;
		case 'd': dbg = optarg; break;
		case 's':
			scan = 1;
			break;
		case 1:
			return -EINVAL;
		}
	}

	ret = nvif_client_new(drv, argv[0], dev, cfg, dbg, &client);
	if (ret)
		return ret;

	ret = nvif_device_new(nvif_object(client), 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);
	nvif_client_ref(NULL, &client);
	if (ret)
		return ret;

	if (scan) {
		fprintf(stderr, "unimplemented\n");
		return 1;
	}

	ret = nvif_object_init(nvif_object(device), NULL, 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 = nvif_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 = nvif_exec(&object, NV_PERFCTR_QUERY,
					&args, sizeof(args));
			assert(ret == 0);
		}
	} while (args.iter != 0xffffffff);
	nvif_object_fini(&object);

	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);
	nvif_device_ref(NULL, &device);
	return 0;
}