[fio.git] / gfio.c

/*
 * gfio - gui front end for fio - the flexible io tester
 *
 * Copyright (C) 2012 Stephen M. Cameron <stephenmcameron@gmail.com> 
 *
 * The license below covers all files distributed with fio unless otherwise
 * noted in the file itself.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */
#include <locale.h>

#include <glib.h>
#include <gtk/gtk.h>

#include "fio_initialization.h"
#include "fio.h"

#define ARRAYSIZE(x) (sizeof((x)) / (sizeof((x)[0])))

typedef void (*clickfunction)(GtkWidget *widget, gpointer data);

static void quit_clicked(GtkWidget *widget, gpointer data);
static void start_job_clicked(GtkWidget *widget, gpointer data);

static struct button_spec {
	const char *buttontext;
	clickfunction f;
	const char *tooltiptext;
} buttonspeclist[] = {
#define START_JOB_BUTTON 0
	{ "Start Job",
		start_job_clicked,
		"Send current fio job to fio server to be executed" },
#define QUIT_BUTTON 1
	{ "Quit", quit_clicked, "Quit gfio" },
};

struct gui {
	GtkWidget *window;
	GtkWidget *vbox;
	GtkWidget *thread_status_pb;
	GtkWidget *buttonbox;
	GtkWidget *button[ARRAYSIZE(buttonspeclist)];
	GtkWidget *hostname_hbox;
	GtkWidget *hostname_label;
	GtkWidget *hostname_entry;
	GtkWidget *port_label;
	GtkWidget *port_entry;
	GtkWidget *hostname_combo_box; /* ipv4, ipv6 or socket */
	GtkWidget *jobfile_hbox;
	GtkWidget *jobfile_label;
	GtkWidget *jobfile_entry;
	pthread_t t;
} ui;

static void gfio_text_op(struct fio_client *client,
                FILE *f, __u16 pdu_len, const char *buf)
{
	printf("gfio_text_op called\n");
	fio_client_ops.text_op(client, f, pdu_len, buf);
}

static void gfio_disk_util_op(struct fio_client *client, struct fio_net_cmd *cmd)
{
	printf("gfio_disk_util_op called\n");
	fio_client_ops.disk_util(client, cmd);
}

static void gfio_thread_status_op(struct fio_net_cmd *cmd)
{
	printf("gfio_thread_status_op called\n");
	fio_client_ops.thread_status(cmd);
}

static void gfio_group_stats_op(struct fio_net_cmd *cmd)
{
	printf("gfio_group_stats_op called\n");
	fio_client_ops.group_stats(cmd);
}

static void gfio_eta_op(struct fio_client *client, struct fio_net_cmd *cmd)
{
	fio_client_ops.eta(client, cmd);
}

static void gfio_probe_op(struct fio_client *client, struct fio_net_cmd *cmd)
{
	printf("gfio_probe_op called\n");
	fio_client_ops.probe(client, cmd);
}

static void gfio_update_thread_status(char *status_message, double perc)
{
	static char message[100];
	const char *m = message;

	strncpy(message, status_message, sizeof(message) - 1);
	gtk_progress_bar_set_text(
		GTK_PROGRESS_BAR(ui.thread_status_pb), m);
	gtk_progress_bar_set_fraction(
		GTK_PROGRESS_BAR(ui.thread_status_pb), perc / 100.0);
	gdk_threads_enter();
	gtk_widget_queue_draw(ui.window);
	gdk_threads_leave();
}

struct client_ops gfio_client_ops = {
	gfio_text_op,
	gfio_disk_util_op,
	gfio_thread_status_op,
	gfio_group_stats_op,
	gfio_eta_op,
	gfio_probe_op,
	gfio_update_thread_status,
};

static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
                __attribute__((unused)) gpointer data)
{
        gtk_main_quit();
}

static void *job_thread(void *arg)
{
	struct gui *ui = arg;

	fio_handle_clients(&gfio_client_ops);
	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
	return NULL;
}

static void start_job_thread(pthread_t *t, struct gui *ui)
{
	pthread_create(t, NULL, job_thread, ui);
}

static void start_job_clicked(__attribute__((unused)) GtkWidget *widget,
                gpointer data)
{
	struct gui *ui = data;

	printf("Start job button was clicked.\n");
	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 0);
	start_job_thread(&ui->t, ui);
}

static void add_button(struct gui *ui, int i, GtkWidget *buttonbox,
			struct button_spec *buttonspec)
{
	ui->button[i] = gtk_button_new_with_label(buttonspec->buttontext);
	g_signal_connect(ui->button[i], "clicked", G_CALLBACK (buttonspec->f), ui);
	gtk_box_pack_start(GTK_BOX (ui->buttonbox), ui->button[i], TRUE, TRUE, 0);
	gtk_widget_set_tooltip_text(ui->button[i], buttonspeclist[i].tooltiptext);
}

static void add_buttons(struct gui *ui,
				struct button_spec *buttonlist,
				int nbuttons)
{
	int i;

	ui->buttonbox = gtk_hbox_new(FALSE, 0);
	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->buttonbox);
	for (i = 0; i < nbuttons; i++)
		add_button(ui, i, ui->buttonbox, &buttonlist[i]);
}

static void init_ui(int *argc, char **argv[], struct gui *ui)
{
	GList *hostname_type_list = NULL;
	char portnum[20];

	/* Magical g*thread incantation, you just need this thread stuff.
	 * Without it, the update that happens in gfio_update_thread_status
	 * doesn't really happen in a timely fashion, you need expose events
	 */
	if (!g_thread_supported ())
		g_thread_init(NULL);
	gdk_threads_init();

	gtk_init(argc, argv);
	
	ui->window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
        gtk_window_set_title(GTK_WINDOW(ui->window), "fio");
	gtk_window_set_default_size(GTK_WINDOW(ui->window), 700, 500);

	g_signal_connect(ui->window, "delete-event", G_CALLBACK (quit_clicked), NULL);
	g_signal_connect(ui->window, "destroy", G_CALLBACK (quit_clicked), NULL);

	ui->vbox = gtk_vbox_new(FALSE, 0);
	gtk_container_add(GTK_CONTAINER (ui->window), ui->vbox);

	/*
	 * Set up hostname label + entry, port label + entry,
	 */
	ui->hostname_hbox = gtk_hbox_new(FALSE, 0);
	ui->hostname_label = gtk_label_new("Host:");
	ui->hostname_entry = gtk_entry_new();
	gtk_entry_set_text(GTK_ENTRY(ui->hostname_entry), "localhost");
	ui->port_label = gtk_label_new("Port:");
	ui->port_entry = gtk_entry_new();
	snprintf(portnum, sizeof(portnum) - 1, "%d", FIO_NET_PORT);
	gtk_entry_set_text(GTK_ENTRY(ui->port_entry), (gchar *) portnum);

	/*
	 * Set up combo box for address type
	 */
	ui->hostname_combo_box = gtk_combo_new();
	gtk_entry_set_text(GTK_ENTRY (GTK_COMBO(ui->hostname_combo_box)->entry), "IPv4");
	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv4"); 
	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "local socket"); 
	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv6"); 
	gtk_combo_set_popdown_strings (GTK_COMBO (ui->hostname_combo_box), hostname_type_list);
	g_list_free(hostname_type_list);

	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_label);
	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_entry);
	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_label);
	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_entry);
	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_combo_box);
	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->hostname_hbox);

	/*
	 * Set up jobfile text entry (temporary until gui really works)
	 */
	ui->jobfile_hbox = gtk_hbox_new(FALSE, 0);
	ui->jobfile_label = gtk_label_new("Job file:");
	ui->jobfile_entry = gtk_entry_new();
	gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_label);
	gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_entry);
	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->jobfile_hbox);

	/*
	 * Set up thread status progress bar
	 */
	ui->thread_status_pb = gtk_progress_bar_new();
	gtk_progress_bar_set_fraction(
		GTK_PROGRESS_BAR(ui->thread_status_pb), 0.0);
	gtk_progress_bar_set_text(
		GTK_PROGRESS_BAR(ui->thread_status_pb), "No jobs running");
	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->thread_status_pb);

	add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist));
	gtk_widget_show_all(ui->window);
}

int main(int argc, char *argv[], char *envp[])
{
	if (initialize_fio(envp))
		return 1;

	if (parse_options(argc, argv))
		return 1;

	init_ui(&argc, &argv, &ui);

	gdk_threads_enter();
	gtk_main();
	gdk_threads_leave();
	return 0;
}
Commit	Line	Data
	1	/*
	2	* gfio - gui front end for fio - the flexible io tester
	3	*
	4	* Copyright (C) 2012 Stephen M. Cameron <stephenmcameron@gmail.com>
	5	*
	6	* The license below covers all files distributed with fio unless otherwise
	7	* noted in the file itself.
	8	*
	9	* This program is free software; you can redistribute it and/or modify
	10	* it under the terms of the GNU General Public License version 2 as
	11	* published by the Free Software Foundation.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	21	*
	22	*/
	23	#include <locale.h>
	24
	25	#include <glib.h>
	26	#include <gtk/gtk.h>
	27
	28	#include "fio_initialization.h"
	29	#include "fio.h"
	30
	31	#define ARRAYSIZE(x) (sizeof((x)) / (sizeof((x)[0])))
	32
	33	typedef void (clickfunction)(GtkWidget widget, gpointer data);
	34
	35	static void quit_clicked(GtkWidget *widget, gpointer data);
	36	static void start_job_clicked(GtkWidget *widget, gpointer data);
	37
	38	static struct button_spec {
	39	const char *buttontext;
	40	clickfunction f;
	41	const char *tooltiptext;
	42	} buttonspeclist[] = {
	43	#define START_JOB_BUTTON 0
	44	{ "Start Job",
	45	start_job_clicked,
	46	"Send current fio job to fio server to be executed" },
	47	#define QUIT_BUTTON 1
	48	{ "Quit", quit_clicked, "Quit gfio" },
	49	};
	50
	51	struct gui {
	52	GtkWidget *window;
	53	GtkWidget *vbox;
	54	GtkWidget *thread_status_pb;
	55	GtkWidget *buttonbox;
	56	GtkWidget *button[ARRAYSIZE(buttonspeclist)];
	57	GtkWidget *hostname_hbox;
	58	GtkWidget *hostname_label;
	59	GtkWidget *hostname_entry;
	60	GtkWidget *port_label;
	61	GtkWidget *port_entry;
	62	GtkWidget hostname_combo_box; / ipv4, ipv6 or socket */
	63	GtkWidget *jobfile_hbox;
	64	GtkWidget *jobfile_label;
	65	GtkWidget *jobfile_entry;
	66	pthread_t t;
	67	} ui;
	68
	69	static void gfio_text_op(struct fio_client *client,
	70	FILE f, __u16 pdu_len, const char buf)
	71	{
	72	printf("gfio_text_op called\n");
	73	fio_client_ops.text_op(client, f, pdu_len, buf);
	74	}
	75
	76	static void gfio_disk_util_op(struct fio_client client, struct fio_net_cmd cmd)
	77	{
	78	printf("gfio_disk_util_op called\n");
	79	fio_client_ops.disk_util(client, cmd);
	80	}
	81
	82	static void gfio_thread_status_op(struct fio_net_cmd *cmd)
	83	{
	84	printf("gfio_thread_status_op called\n");
	85	fio_client_ops.thread_status(cmd);
	86	}
	87
	88	static void gfio_group_stats_op(struct fio_net_cmd *cmd)
	89	{
	90	printf("gfio_group_stats_op called\n");
	91	fio_client_ops.group_stats(cmd);
	92	}
	93
	94	static void gfio_eta_op(struct fio_client client, struct fio_net_cmd cmd)
	95	{
	96	fio_client_ops.eta(client, cmd);
	97	}
	98
	99	static void gfio_probe_op(struct fio_client client, struct fio_net_cmd cmd)
	100	{
	101	printf("gfio_probe_op called\n");
	102	fio_client_ops.probe(client, cmd);
	103	}
	104
	105	static void gfio_update_thread_status(char *status_message, double perc)
	106	{
	107	static char message[100];
	108	const char *m = message;
	109
	110	strncpy(message, status_message, sizeof(message) - 1);
	111	gtk_progress_bar_set_text(
	112	GTK_PROGRESS_BAR(ui.thread_status_pb), m);
	113	gtk_progress_bar_set_fraction(
	114	GTK_PROGRESS_BAR(ui.thread_status_pb), perc / 100.0);
	115	gdk_threads_enter();
	116	gtk_widget_queue_draw(ui.window);
	117	gdk_threads_leave();
	118	}
	119
	120	struct client_ops gfio_client_ops = {
	121	gfio_text_op,
	122	gfio_disk_util_op,
	123	gfio_thread_status_op,
	124	gfio_group_stats_op,
	125	gfio_eta_op,
	126	gfio_probe_op,
	127	gfio_update_thread_status,
	128	};
	129
	130	static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
	131	__attribute__((unused)) gpointer data)
	132	{
	133	gtk_main_quit();
	134	}
	135
	136	static void job_thread(void arg)
	137	{
	138	struct gui *ui = arg;
	139
	140	fio_handle_clients(&gfio_client_ops);
	141	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
	142	return NULL;
	143	}
	144
	145	static void start_job_thread(pthread_t t, struct gui ui)
	146	{
	147	pthread_create(t, NULL, job_thread, ui);
	148	}
	149
	150	static void start_job_clicked(__attribute__((unused)) GtkWidget *widget,
	151	gpointer data)
	152	{
	153	struct gui *ui = data;
	154
	155	printf("Start job button was clicked.\n");
	156	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 0);
	157	start_job_thread(&ui->t, ui);
	158	}
	159
	160	static void add_button(struct gui ui, int i, GtkWidget buttonbox,
	161	struct button_spec *buttonspec)
	162	{
	163	ui->button[i] = gtk_button_new_with_label(buttonspec->buttontext);
	164	g_signal_connect(ui->button[i], "clicked", G_CALLBACK (buttonspec->f), ui);
	165	gtk_box_pack_start(GTK_BOX (ui->buttonbox), ui->button[i], TRUE, TRUE, 0);
	166	gtk_widget_set_tooltip_text(ui->button[i], buttonspeclist[i].tooltiptext);
	167	}
	168
	169	static void add_buttons(struct gui *ui,
	170	struct button_spec *buttonlist,
	171	int nbuttons)
	172	{
	173	int i;
	174
	175	ui->buttonbox = gtk_hbox_new(FALSE, 0);
	176	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->buttonbox);
	177	for (i = 0; i < nbuttons; i++)
	178	add_button(ui, i, ui->buttonbox, &buttonlist[i]);
	179	}
	180
	181	static void init_ui(int argc, char argv[], struct gui ui)
	182	{
	183	GList *hostname_type_list = NULL;
	184	char portnum[20];
	185
	186	/* Magical g*thread incantation, you just need this thread stuff.
	187	* Without it, the update that happens in gfio_update_thread_status
	188	* doesn't really happen in a timely fashion, you need expose events
	189	*/
	190	if (!g_thread_supported ())
	191	g_thread_init(NULL);
	192	gdk_threads_init();
	193
	194	gtk_init(argc, argv);
	195
	196	ui->window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
	197	gtk_window_set_title(GTK_WINDOW(ui->window), "fio");
	198	gtk_window_set_default_size(GTK_WINDOW(ui->window), 700, 500);
	199
	200	g_signal_connect(ui->window, "delete-event", G_CALLBACK (quit_clicked), NULL);
	201	g_signal_connect(ui->window, "destroy", G_CALLBACK (quit_clicked), NULL);
	202
	203	ui->vbox = gtk_vbox_new(FALSE, 0);
	204	gtk_container_add(GTK_CONTAINER (ui->window), ui->vbox);
	205
	206	/*
	207	* Set up hostname label + entry, port label + entry,
	208	*/
	209	ui->hostname_hbox = gtk_hbox_new(FALSE, 0);
	210	ui->hostname_label = gtk_label_new("Host:");
	211	ui->hostname_entry = gtk_entry_new();
	212	gtk_entry_set_text(GTK_ENTRY(ui->hostname_entry), "localhost");
	213	ui->port_label = gtk_label_new("Port:");
	214	ui->port_entry = gtk_entry_new();
	215	snprintf(portnum, sizeof(portnum) - 1, "%d", FIO_NET_PORT);
	216	gtk_entry_set_text(GTK_ENTRY(ui->port_entry), (gchar *) portnum);
	217
	218	/*
	219	* Set up combo box for address type
	220	*/
	221	ui->hostname_combo_box = gtk_combo_new();
	222	gtk_entry_set_text(GTK_ENTRY (GTK_COMBO(ui->hostname_combo_box)->entry), "IPv4");
	223	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv4");
	224	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "local socket");
	225	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv6");
	226	gtk_combo_set_popdown_strings (GTK_COMBO (ui->hostname_combo_box), hostname_type_list);
	227	g_list_free(hostname_type_list);
	228
	229	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_label);
	230	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_entry);
	231	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_label);
	232	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_entry);
	233	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_combo_box);
	234	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->hostname_hbox);
	235
	236	/*
	237	* Set up jobfile text entry (temporary until gui really works)
	238	*/
	239	ui->jobfile_hbox = gtk_hbox_new(FALSE, 0);
	240	ui->jobfile_label = gtk_label_new("Job file:");
	241	ui->jobfile_entry = gtk_entry_new();
	242	gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_label);
	243	gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_entry);
	244	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->jobfile_hbox);
	245
	246	/*
	247	* Set up thread status progress bar
	248	*/
	249	ui->thread_status_pb = gtk_progress_bar_new();
	250	gtk_progress_bar_set_fraction(
	251	GTK_PROGRESS_BAR(ui->thread_status_pb), 0.0);
	252	gtk_progress_bar_set_text(
	253	GTK_PROGRESS_BAR(ui->thread_status_pb), "No jobs running");
	254	gtk_container_add(GTK_CONTAINER (ui->vbox), ui->thread_status_pb);
	255
	256	add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist));
	257	gtk_widget_show_all(ui->window);
	258	}
	259
	260	int main(int argc, char argv[], char envp[])
	261	{
	262	if (initialize_fio(envp))
	263	return 1;
	264
	265	if (parse_options(argc, argv))
	266	return 1;
	267
	268	init_ui(&argc, &argv, &ui);
	269
	270	gdk_threads_enter();
	271	gtk_main();
	272	gdk_threads_leave();
	273	return 0;
	274	}