[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 <malloc.h>

#include <glib.h>
#include <gtk/gtk.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" },
};

struct gui {
	GtkWidget *window;
	GtkWidget *vbox;
	GtkWidget *topvbox;
	GtkWidget *topalign;
	GtkWidget *bottomalign;
	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 *scrolled_window;
	GtkWidget *textview;
	GtkWidget *error_info_bar;
	GtkWidget *error_label;
	GtkTextBuffer *text;
	pthread_t t;

	void *cookie;
	int nr_job_files;
	char **job_files;
} ui;

static void gfio_text_op(struct fio_client *client,
                FILE *f, __u16 pdu_len, const char *buf)
{
	GtkTextBuffer *buffer;
	GtkTextIter end;

	buffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui.textview));
	gdk_threads_enter();
	gtk_text_buffer_get_end_iter(buffer, &end);
	gtk_text_buffer_insert(buffer, &end, buf, -1);
	gdk_threads_leave();
	gtk_text_view_scroll_to_iter(GTK_TEXT_VIEW(ui.textview),
					&end, 0.0, FALSE, 0.0,0.0);
}

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 = {
	.text_op		= gfio_text_op,
	.disk_util		= gfio_disk_util_op,
	.thread_status		= gfio_thread_status_op,
	.group_stats		= gfio_group_stats_op,
	.eta			= gfio_eta_op,
	.probe			= gfio_probe_op,
	.thread_status_display	= 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 int send_job_files(struct gui *ui)
{
	int i, ret;

	for (i = 0; i < ui->nr_job_files; i++) {
		ret = fio_clients_send_ini(ui->job_files[i]);
		free(ui->job_files[i]);
		ui->job_files[i] = NULL;
		if (ret)
			return ret;
	}

	return 0;
}

static void start_job_thread(pthread_t *t, struct gui *ui)
{
	fio_clients_connect();

	if (send_job_files(ui)) {
		printf("Yeah, I didn't really like those options too much.\n");
		gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
		return;
	}
		
	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;

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

static void on_info_bar_response(GtkWidget *widget, gint response,
                                 gpointer data)
{
	if (response == GTK_RESPONSE_OK) {
		gtk_widget_destroy(widget);
		ui.error_info_bar = NULL;
	}
}

void report_error(GError* error)
{
	if (ui.error_info_bar == NULL) {
		ui.error_info_bar = gtk_info_bar_new_with_buttons(GTK_STOCK_OK,
		                                               GTK_RESPONSE_OK,
		                                               NULL);
		g_signal_connect(ui.error_info_bar, "response", G_CALLBACK(on_info_bar_response), NULL);
		gtk_info_bar_set_message_type(GTK_INFO_BAR(ui.error_info_bar),
		                              GTK_MESSAGE_ERROR);
		
		ui.error_label = gtk_label_new(error->message);
		GtkWidget *container = gtk_info_bar_get_content_area(GTK_INFO_BAR(ui.error_info_bar));
		gtk_container_add(GTK_CONTAINER(container), ui.error_label);
		
		gtk_box_pack_start(GTK_BOX(ui.vbox), ui.error_info_bar, FALSE, FALSE, 0);
		gtk_widget_show_all(ui.vbox);
	} else {
		char buffer[256];
		snprintf(buffer, sizeof(buffer), "Failed to open file.");
		gtk_label_set(GTK_LABEL(ui.error_label), buffer);
	}
}

static void file_open(GtkWidget *w, gpointer data)
{
	GtkWidget *dialog;
	GSList *filenames, *fn_glist;
	GtkFileFilter *filter;

	dialog = gtk_file_chooser_dialog_new("Open File",
		GTK_WINDOW(ui.window),
		GTK_FILE_CHOOSER_ACTION_OPEN,
		GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
		GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
		NULL);
	gtk_file_chooser_set_select_multiple(GTK_FILE_CHOOSER(dialog), TRUE);

	filter = gtk_file_filter_new();
	gtk_file_filter_add_pattern(filter, "*.fio");
	gtk_file_filter_add_pattern(filter, "*.job");
	gtk_file_filter_add_mime_type(filter, "text/fio");
	gtk_file_filter_set_name(filter, "Fio job file");
	gtk_file_chooser_set_filter(GTK_FILE_CHOOSER(dialog), filter);

	if (gtk_dialog_run(GTK_DIALOG(dialog)) != GTK_RESPONSE_ACCEPT) {
		gtk_widget_destroy(dialog);
		return;
	}

	fn_glist = gtk_file_chooser_get_filenames(GTK_FILE_CHOOSER(dialog));
	filenames = fn_glist;
	while (filenames != NULL) {
		const char *hostname;

		ui.job_files = realloc(ui.job_files, (ui.nr_job_files + 1) * sizeof(char *));
		ui.job_files[ui.nr_job_files] = strdup(filenames->data);
		ui.nr_job_files++;

		hostname = gtk_entry_get_text(GTK_ENTRY(ui.hostname_entry));
		fio_client_add(hostname, &ui.cookie);
#if 0
		if (error) {
			report_error(error);
			g_error_free(error);
			error = NULL;
		}
#endif
			
		g_free(filenames->data);
		filenames = g_slist_next(filenames);
	}
	g_slist_free(fn_glist);
	gtk_widget_destroy(dialog);
}

static void file_save(GtkWidget *w, gpointer data)
{
	GtkWidget *dialog;

	dialog = gtk_file_chooser_dialog_new("Save File",
		GTK_WINDOW(ui.window),
		GTK_FILE_CHOOSER_ACTION_SAVE,
		GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
		GTK_STOCK_SAVE, GTK_RESPONSE_ACCEPT,
		NULL);

	gtk_file_chooser_set_do_overwrite_confirmation(GTK_FILE_CHOOSER(dialog), TRUE);
	gtk_file_chooser_set_current_name(GTK_FILE_CHOOSER(dialog), "Untitled document");

	if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT) {
		char *filename;

		filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog));
		// save_job_file(filename);
		g_free(filename);
	}
	gtk_widget_destroy(dialog);
}

static void about_dialog(GtkWidget *w, gpointer data)
{
	gtk_show_about_dialog(NULL,
		"program-name", "gfio",
		"comments", "Gtk2 UI for fio",
		"license", "GPLv2",
		"version", fio_version_string,
		"copyright", "Jens Axboe <axboe@kernel.dk> 2012",
		"logo-icon-name", "fio",
		/* Must be last: */
		NULL, NULL,
		NULL);
}

static GtkActionEntry menu_items[] = {
        { "FileMenuAction", GTK_STOCK_FILE, "File", NULL, NULL, NULL},
        { "HelpMenuAction", GTK_STOCK_HELP, "Help", NULL, NULL, NULL},
	{ "OpenFile",       GTK_STOCK_OPEN, NULL,   "<Control>O", NULL, G_CALLBACK(file_open) },
        { "SaveFile",       GTK_STOCK_SAVE, NULL,   "<Control>S", NULL, G_CALLBACK(file_save) },
        { "Quit",           GTK_STOCK_QUIT, NULL,   "<Control>Q", NULL, G_CALLBACK(quit_clicked) },
	{ "About",          GTK_STOCK_ABOUT, NULL,  NULL, NULL, G_CALLBACK(about_dialog) },
};
static gint nmenu_items = sizeof (menu_items) / sizeof(menu_items[0]);

static const gchar *ui_string = " \
	<ui> \
		<menubar name=\"MainMenu\"> \
			<menu name=\"FileMenu\" action=\"FileMenuAction\"> \
				<menuitem name=\"Open\" action=\"OpenFile\" /> \
				<menuitem name=\"Save\" action=\"SaveFile\" /> \
				<separator name=\"Separator\"/> \
				<menuitem name=\"Quit\" action=\"Quit\" /> \
			</menu> \
			<menu name=\"Help\" action=\"HelpMenuAction\"> \
				<menuitem name=\"About\" action=\"About\" /> \
			</menu> \
		</menubar> \
	</ui> \
";

static GtkWidget *get_menubar_menu(GtkWidget *window, GtkUIManager *ui_manager)
{
	GtkActionGroup *action_group = gtk_action_group_new("Menu");
	GError *error = 0;

	action_group = gtk_action_group_new("Menu");
	gtk_action_group_add_actions(action_group, menu_items, nmenu_items, 0);

	gtk_ui_manager_insert_action_group(ui_manager, action_group, 0);
	gtk_ui_manager_add_ui_from_string(GTK_UI_MANAGER(ui_manager), ui_string, -1, &error);

	gtk_window_add_accel_group(GTK_WINDOW(window), gtk_ui_manager_get_accel_group(ui_manager));
	return gtk_ui_manager_get_widget(ui_manager, "/MainMenu");
}

void gfio_ui_setup(GtkSettings *settings, GtkWidget *menubar,
                   GtkWidget *vbox, GtkUIManager *ui_manager)
{
        gtk_box_pack_start(GTK_BOX(vbox), menubar, FALSE, FALSE, 0);
}

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

	memset(ui, 0, sizeof(*ui));

	/* 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);
	settings = gtk_settings_get_default();
	gtk_settings_set_long_property(settings, "gtk_tooltip_timeout", 10, "gfio setting");
	g_type_init();
	
	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);

	uimanager = gtk_ui_manager_new();
	menu = get_menubar_menu(ui->window, uimanager);
	gfio_ui_setup(settings, menu, ui->vbox, uimanager);

	/*
	 * Set up alignments for widgets at the top of ui, 
	 * align top left, expand horizontally but not vertically
	 */
	ui->topalign = gtk_alignment_new(0, 0, 1, 0);
	ui->topvbox = gtk_vbox_new(FALSE, 0);
	gtk_container_add(GTK_CONTAINER(ui->topalign), ui->topvbox);
	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->topalign, FALSE, FALSE, 0);

	/*
	 * 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->topvbox), ui->hostname_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->topvbox), ui->thread_status_pb);

	/*
	 * Add a text box for text op messages 
	 */
	ui->textview = gtk_text_view_new();
	ui->text = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui->textview));
	gtk_text_buffer_set_text(ui->text, "", -1);
	gtk_text_view_set_editable(GTK_TEXT_VIEW(ui->textview), FALSE);
	gtk_text_view_set_cursor_visible(GTK_TEXT_VIEW(ui->textview), FALSE);
	ui->scrolled_window = gtk_scrolled_window_new(NULL, NULL);
	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(ui->scrolled_window),
					GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
	gtk_container_add(GTK_CONTAINER(ui->scrolled_window), ui->textview);
	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->scrolled_window,
			TRUE, TRUE, 0);

	/*
	 * Set up alignments for widgets at the bottom of ui, 
	 * align bottom left, expand horizontally but not vertically
	 */
	ui->bottomalign = gtk_alignment_new(0, 1, 1, 0);
	ui->buttonbox = gtk_hbox_new(FALSE, 0);
	gtk_container_add(GTK_CONTAINER(ui->bottomalign), ui->buttonbox);
	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->bottomalign,
					FALSE, FALSE, 0);

	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 (fio_init_options())
		return 1;

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

	gdk_threads_enter();
	gtk_main();
	gdk_threads_leave();
	return 0;
}
Commit	Line	Data
ff1f3280 SC	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	*/
8232e285	23	#include <locale.h>
60f6b330	24	#include <malloc.h>
8232e285	25
5b7573ab	26	#include <glib.h>
ff1f3280 SC	27	#include <gtk/gtk.h>
ff1f3280 SC	28
8232e285 SC	29	#include "fio.h"
8232e285 SC	30
f3074008 SC	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" },
f3074008 SC	47	};
f3074008 SC	48
ff1f3280 SC	49	struct gui {
ff1f3280 SC	50	GtkWidget *window;
5b7573ab	51	GtkWidget *vbox;
c36f98d9 SC	52	GtkWidget *topvbox;
	53	GtkWidget *topalign;
	54	GtkWidget *bottomalign;
04cc6b77	55	GtkWidget *thread_status_pb;
f3074008 SC	56	GtkWidget *buttonbox;
f3074008 SC	57	GtkWidget *button[ARRAYSIZE(buttonspeclist)];
45032dd8 SC	58	GtkWidget *hostname_hbox;
	59	GtkWidget *hostname_label;
	60	GtkWidget *hostname_entry;
	61	GtkWidget *port_label;
	62	GtkWidget *port_entry;
	63	GtkWidget hostname_combo_box; / ipv4, ipv6 or socket */
736f2dff SC	64	GtkWidget *scrolled_window;
736f2dff SC	65	GtkWidget *textview;
0420ba6a JA	66	GtkWidget *error_info_bar;
0420ba6a JA	67	GtkWidget *error_label;
736f2dff	68	GtkTextBuffer *text;
25927259	69	pthread_t t;
0420ba6a JA	70
	71	void *cookie;
	72	int nr_job_files;
	73	char **job_files;
5b7573ab	74	} ui;
ff1f3280	75
a1820207 SC	76	static void gfio_text_op(struct fio_client *client,
	77	FILE f, __u16 pdu_len, const char buf)
	78	{
736f2dff SC	79	GtkTextBuffer *buffer;
	80	GtkTextIter end;
	81
	82	buffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui.textview));
	83	gdk_threads_enter();
	84	gtk_text_buffer_get_end_iter(buffer, &end);
	85	gtk_text_buffer_insert(buffer, &end, buf, -1);
	86	gdk_threads_leave();
	87	gtk_text_view_scroll_to_iter(GTK_TEXT_VIEW(ui.textview),
	88	&end, 0.0, FALSE, 0.0,0.0);
a1820207 SC	89	}
	90
	91	static void gfio_disk_util_op(struct fio_client client, struct fio_net_cmd cmd)
	92	{
	93	printf("gfio_disk_util_op called\n");
	94	fio_client_ops.disk_util(client, cmd);
	95	}
	96
	97	static void gfio_thread_status_op(struct fio_net_cmd *cmd)
	98	{
	99	printf("gfio_thread_status_op called\n");
	100	fio_client_ops.thread_status(cmd);
	101	}
	102
	103	static void gfio_group_stats_op(struct fio_net_cmd *cmd)
	104	{
	105	printf("gfio_group_stats_op called\n");
	106	fio_client_ops.group_stats(cmd);
	107	}
	108
	109	static void gfio_eta_op(struct fio_client client, struct fio_net_cmd cmd)
	110	{
a1820207 SC	111	fio_client_ops.eta(client, cmd);
	112	}
	113
	114	static void gfio_probe_op(struct fio_client client, struct fio_net_cmd cmd)
	115	{
	116	printf("gfio_probe_op called\n");
	117	fio_client_ops.probe(client, cmd);
	118	}
	119
04cc6b77	120	static void gfio_update_thread_status(char *status_message, double perc)
5b7573ab SC	121	{
	122	static char message[100];
	123	const char *m = message;
	124
	125	strncpy(message, status_message, sizeof(message) - 1);
04cc6b77 SC	126	gtk_progress_bar_set_text(
	127	GTK_PROGRESS_BAR(ui.thread_status_pb), m);
	128	gtk_progress_bar_set_fraction(
	129	GTK_PROGRESS_BAR(ui.thread_status_pb), perc / 100.0);
5b7573ab SC	130	gdk_threads_enter();
	131	gtk_widget_queue_draw(ui.window);
	132	gdk_threads_leave();
	133	}
	134
a1820207	135	struct client_ops gfio_client_ops = {
0420ba6a JA	136	.text_op = gfio_text_op,
	137	.disk_util = gfio_disk_util_op,
	138	.thread_status = gfio_thread_status_op,
	139	.group_stats = gfio_group_stats_op,
	140	.eta = gfio_eta_op,
	141	.probe = gfio_probe_op,
	142	.thread_status_display = gfio_update_thread_status,
a1820207 SC	143	};
a1820207 SC	144
ff1f3280 SC	145	static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
	146	__attribute__((unused)) gpointer data)
	147	{
	148	gtk_main_quit();
	149	}
	150
25927259 SC	151	static void job_thread(void arg)
	152	{
	153	struct gui *ui = arg;
	154
	155	fio_handle_clients(&gfio_client_ops);
	156	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
	157	return NULL;
	158	}
	159
0420ba6a	160	static int send_job_files(struct gui *ui)
60f6b330	161	{
0420ba6a JA	162	int i, ret;
	163
	164	for (i = 0; i < ui->nr_job_files; i++) {
	165	ret = fio_clients_send_ini(ui->job_files[i]);
	166	free(ui->job_files[i]);
	167	ui->job_files[i] = NULL;
	168	if (ret)
	169	return ret;
	170	}
	171
	172	return 0;
60f6b330 SC	173	}
60f6b330 SC	174
25927259 SC	175	static void start_job_thread(pthread_t t, struct gui ui)
25927259 SC	176	{
0420ba6a JA	177	fio_clients_connect();
	178
	179	if (send_job_files(ui)) {
60f6b330	180	printf("Yeah, I didn't really like those options too much.\n");
60f6b330 SC	181	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
	182	return;
	183	}
0420ba6a	184
25927259 SC	185	pthread_create(t, NULL, job_thread, ui);
	186	}
	187
f3074008	188	static void start_job_clicked(__attribute__((unused)) GtkWidget *widget,
25927259	189	gpointer data)
f3074008	190	{
25927259 SC	191	struct gui *ui = data;
25927259 SC	192
f3074008	193	printf("Start job button was clicked.\n");
25927259 SC	194	gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 0);
25927259 SC	195	start_job_thread(&ui->t, ui);
f3074008 SC	196	}
	197
	198	static void add_button(struct gui ui, int i, GtkWidget buttonbox,
	199	struct button_spec *buttonspec)
	200	{
	201	ui->button[i] = gtk_button_new_with_label(buttonspec->buttontext);
	202	g_signal_connect(ui->button[i], "clicked", G_CALLBACK (buttonspec->f), ui);
	203	gtk_box_pack_start(GTK_BOX (ui->buttonbox), ui->button[i], TRUE, TRUE, 0);
	204	gtk_widget_set_tooltip_text(ui->button[i], buttonspeclist[i].tooltiptext);
	205	}
	206
	207	static void add_buttons(struct gui *ui,
	208	struct button_spec *buttonlist,
	209	int nbuttons)
	210	{
	211	int i;
	212
f3074008 SC	213	for (i = 0; i < nbuttons; i++)
	214	add_button(ui, i, ui->buttonbox, &buttonlist[i]);
	215	}
	216
0420ba6a JA	217	static void on_info_bar_response(GtkWidget *widget, gint response,
	218	gpointer data)
	219	{
	220	if (response == GTK_RESPONSE_OK) {
	221	gtk_widget_destroy(widget);
	222	ui.error_info_bar = NULL;
	223	}
	224	}
	225
	226	void report_error(GError* error)
	227	{
	228	if (ui.error_info_bar == NULL) {
	229	ui.error_info_bar = gtk_info_bar_new_with_buttons(GTK_STOCK_OK,
	230	GTK_RESPONSE_OK,
	231	NULL);
	232	g_signal_connect(ui.error_info_bar, "response", G_CALLBACK(on_info_bar_response), NULL);
	233	gtk_info_bar_set_message_type(GTK_INFO_BAR(ui.error_info_bar),
	234	GTK_MESSAGE_ERROR);
	235
	236	ui.error_label = gtk_label_new(error->message);
	237	GtkWidget *container = gtk_info_bar_get_content_area(GTK_INFO_BAR(ui.error_info_bar));
	238	gtk_container_add(GTK_CONTAINER(container), ui.error_label);
	239
	240	gtk_box_pack_start(GTK_BOX(ui.vbox), ui.error_info_bar, FALSE, FALSE, 0);
	241	gtk_widget_show_all(ui.vbox);
	242	} else {
	243	char buffer[256];
	244	snprintf(buffer, sizeof(buffer), "Failed to open file.");
	245	gtk_label_set(GTK_LABEL(ui.error_label), buffer);
	246	}
	247	}
	248
	249	static void file_open(GtkWidget *w, gpointer data)
	250	{
	251	GtkWidget *dialog;
	252	GSList filenames, fn_glist;
	253	GtkFileFilter *filter;
	254
	255	dialog = gtk_file_chooser_dialog_new("Open File",
	256	GTK_WINDOW(ui.window),
	257	GTK_FILE_CHOOSER_ACTION_OPEN,
	258	GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
	259	GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
	260	NULL);
	261	gtk_file_chooser_set_select_multiple(GTK_FILE_CHOOSER(dialog), TRUE);
	262
	263	filter = gtk_file_filter_new();
	264	gtk_file_filter_add_pattern(filter, "*.fio");
	265	gtk_file_filter_add_pattern(filter, "*.job");
	266	gtk_file_filter_add_mime_type(filter, "text/fio");
	267	gtk_file_filter_set_name(filter, "Fio job file");
	268	gtk_file_chooser_set_filter(GTK_FILE_CHOOSER(dialog), filter);
	269
	270	if (gtk_dialog_run(GTK_DIALOG(dialog)) != GTK_RESPONSE_ACCEPT) {
	271	gtk_widget_destroy(dialog);
	272	return;
	273	}
	274
	275	fn_glist = gtk_file_chooser_get_filenames(GTK_FILE_CHOOSER(dialog));
	276	filenames = fn_glist;
	277	while (filenames != NULL) {
	278	const char *hostname;
	279
	280	ui.job_files = realloc(ui.job_files, (ui.nr_job_files + 1) * sizeof(char *));
281	ui.job_files[ui.nr_job_files] = strdup(filenames->data);
282	ui.nr_job_files++;
283
284	hostname = gtk_entry_get_text(GTK_ENTRY(ui.hostname_entry));
285	fio_client_add(hostname, &ui.cookie);
286	#if 0
287	if (error) {
288	report_error(error);
289	g_error_free(error);
290	error = NULL;
291	}
292	#endif
293
294	g_free(filenames->data);
295	filenames = g_slist_next(filenames);
296	}
297	g_slist_free(fn_glist);
298	gtk_widget_destroy(dialog);
299	}
300
301	static void file_save(GtkWidget *w, gpointer data)
302	{
303	GtkWidget *dialog;
304
305	dialog = gtk_file_chooser_dialog_new("Save File",
306	GTK_WINDOW(ui.window),
307	GTK_FILE_CHOOSER_ACTION_SAVE,
308	GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
309	GTK_STOCK_SAVE, GTK_RESPONSE_ACCEPT,
310	NULL);
311
312	gtk_file_chooser_set_do_overwrite_confirmation(GTK_FILE_CHOOSER(dialog), TRUE);
313	gtk_file_chooser_set_current_name(GTK_FILE_CHOOSER(dialog), "Untitled document");
314
315	if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT) {
316	char *filename;
317
318	filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog));
319	// save_job_file(filename);
320	g_free(filename);
321	}
322	gtk_widget_destroy(dialog);
323	}
324
325	static void about_dialog(GtkWidget *w, gpointer data)
326	{
327	gtk_show_about_dialog(NULL,
328	"program-name", "gfio",
329	"comments", "Gtk2 UI for fio",
330	"license", "GPLv2",
331	"version", fio_version_string,
332	"copyright", "Jens Axboe <axboe@kernel.dk> 2012",
333	"logo-icon-name", "fio",
334	/* Must be last: */
335	NULL, NULL,
336	NULL);
337	}
338
339	static GtkActionEntry menu_items[] = {
340	{ "FileMenuAction", GTK_STOCK_FILE, "File", NULL, NULL, NULL},
341	{ "HelpMenuAction", GTK_STOCK_HELP, "Help", NULL, NULL, NULL},
342	{ "OpenFile", GTK_STOCK_OPEN, NULL, "<Control>O", NULL, G_CALLBACK(file_open) },
343	{ "SaveFile", GTK_STOCK_SAVE, NULL, "<Control>S", NULL, G_CALLBACK(file_save) },
344	{ "Quit", GTK_STOCK_QUIT, NULL, "<Control>Q", NULL, G_CALLBACK(quit_clicked) },
345	{ "About", GTK_STOCK_ABOUT, NULL, NULL, NULL, G_CALLBACK(about_dialog) },
346	};
347	static gint nmenu_items = sizeof (menu_items) / sizeof(menu_items[0]);
348
349	static const gchar *ui_string = " \
350	<ui> \
351	<menubar name=\"MainMenu\"> \
352	<menu name=\"FileMenu\" action=\"FileMenuAction\"> \
353	<menuitem name=\"Open\" action=\"OpenFile\" /> \
354	<menuitem name=\"Save\" action=\"SaveFile\" /> \
355	<separator name=\"Separator\"/> \
356	<menuitem name=\"Quit\" action=\"Quit\" /> \
357	</menu> \
358	<menu name=\"Help\" action=\"HelpMenuAction\"> \
359	<menuitem name=\"About\" action=\"About\" /> \
360	</menu> \
361	</menubar> \
362	</ui> \
363	";
364
365	static GtkWidget get_menubar_menu(GtkWidget window, GtkUIManager *ui_manager)
366	{
367	GtkActionGroup *action_group = gtk_action_group_new("Menu");
368	GError *error = 0;
369
370	action_group = gtk_action_group_new("Menu");
371	gtk_action_group_add_actions(action_group, menu_items, nmenu_items, 0);
372
373	gtk_ui_manager_insert_action_group(ui_manager, action_group, 0);
374	gtk_ui_manager_add_ui_from_string(GTK_UI_MANAGER(ui_manager), ui_string, -1, &error);
375
376	gtk_window_add_accel_group(GTK_WINDOW(window), gtk_ui_manager_get_accel_group(ui_manager));
377	return gtk_ui_manager_get_widget(ui_manager, "/MainMenu");
378	}
379
380	void gfio_ui_setup(GtkSettings settings, GtkWidget menubar,
381	GtkWidget vbox, GtkUIManager ui_manager)
382	{
383	gtk_box_pack_start(GTK_BOX(vbox), menubar, FALSE, FALSE, 0);
384	}
385
ff1f3280 SC	386	static void init_ui(int argc, char argv[], struct gui ui)
ff1f3280 SC	387	{
45032dd8 SC	388	GList *hostname_type_list = NULL;
45032dd8 SC	389	char portnum[20];
0420ba6a JA	390	GtkSettings *settings;
	391	GtkUIManager *uimanager;
	392	GtkWidget *menu;
	393
	394	memset(ui, 0, sizeof(*ui));
45032dd8	395
2839f0c6	396	/* Magical g*thread incantation, you just need this thread stuff.
04cc6b77	397	* Without it, the update that happens in gfio_update_thread_status
2839f0c6 SC	398	* doesn't really happen in a timely fashion, you need expose events
	399	*/
	400	if (!g_thread_supported ())
	401	g_thread_init(NULL);
	402	gdk_threads_init();
	403
ff1f3280	404	gtk_init(argc, argv);
0420ba6a JA	405	settings = gtk_settings_get_default();
	406	gtk_settings_set_long_property(settings, "gtk_tooltip_timeout", 10, "gfio setting");
	407	g_type_init();
ff1f3280 SC	408
	409	ui->window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
	410	gtk_window_set_title(GTK_WINDOW(ui->window), "fio");
	411	gtk_window_set_default_size(GTK_WINDOW(ui->window), 700, 500);
	412
0420ba6a JA	413	g_signal_connect(ui->window, "delete-event", G_CALLBACK(quit_clicked), NULL);
0420ba6a JA	414	g_signal_connect(ui->window, "destroy", G_CALLBACK(quit_clicked), NULL);
ff1f3280	415
5b7573ab SC	416	ui->vbox = gtk_vbox_new(FALSE, 0);
5b7573ab SC	417	gtk_container_add(GTK_CONTAINER (ui->window), ui->vbox);
04cc6b77	418
0420ba6a JA	419	uimanager = gtk_ui_manager_new();
	420	menu = get_menubar_menu(ui->window, uimanager);
	421	gfio_ui_setup(settings, menu, ui->vbox, uimanager);
	422
c36f98d9 SC	423	/*
	424	* Set up alignments for widgets at the top of ui,
	425	* align top left, expand horizontally but not vertically
	426	*/
	427	ui->topalign = gtk_alignment_new(0, 0, 1, 0);
	428	ui->topvbox = gtk_vbox_new(FALSE, 0);
	429	gtk_container_add(GTK_CONTAINER(ui->topalign), ui->topvbox);
e164534f	430	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->topalign, FALSE, FALSE, 0);
c36f98d9	431
45032dd8 SC	432	/*
	433	* Set up hostname label + entry, port label + entry,
	434	*/
	435	ui->hostname_hbox = gtk_hbox_new(FALSE, 0);
	436	ui->hostname_label = gtk_label_new("Host:");
	437	ui->hostname_entry = gtk_entry_new();
	438	gtk_entry_set_text(GTK_ENTRY(ui->hostname_entry), "localhost");
	439	ui->port_label = gtk_label_new("Port:");
	440	ui->port_entry = gtk_entry_new();
	441	snprintf(portnum, sizeof(portnum) - 1, "%d", FIO_NET_PORT);
	442	gtk_entry_set_text(GTK_ENTRY(ui->port_entry), (gchar *) portnum);
	443
	444	/*
	445	* Set up combo box for address type
	446	*/
	447	ui->hostname_combo_box = gtk_combo_new();
0420ba6a	448	gtk_entry_set_text(GTK_ENTRY(GTK_COMBO(ui->hostname_combo_box)->entry), "IPv4");
45032dd8 SC	449	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv4");
	450	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "local socket");
	451	hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv6");
0420ba6a	452	gtk_combo_set_popdown_strings(GTK_COMBO(ui->hostname_combo_box), hostname_type_list);
45032dd8 SC	453	g_list_free(hostname_type_list);
	454
	455	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_label);
	456	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_entry);
	457	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_label);
	458	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_entry);
	459	gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_combo_box);
c36f98d9	460	gtk_container_add(GTK_CONTAINER (ui->topvbox), ui->hostname_hbox);
45032dd8	461
04cc6b77 SC	462	/*
	463	* Set up thread status progress bar
	464	*/
	465	ui->thread_status_pb = gtk_progress_bar_new();
	466	gtk_progress_bar_set_fraction(
	467	GTK_PROGRESS_BAR(ui->thread_status_pb), 0.0);
	468	gtk_progress_bar_set_text(
	469	GTK_PROGRESS_BAR(ui->thread_status_pb), "No jobs running");
c36f98d9	470	gtk_container_add(GTK_CONTAINER (ui->topvbox), ui->thread_status_pb);
5b7573ab	471
736f2dff SC	472	/*
	473	* Add a text box for text op messages
	474	*/
	475	ui->textview = gtk_text_view_new();
	476	ui->text = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui->textview));
	477	gtk_text_buffer_set_text(ui->text, "", -1);
	478	gtk_text_view_set_editable(GTK_TEXT_VIEW(ui->textview), FALSE);
	479	gtk_text_view_set_cursor_visible(GTK_TEXT_VIEW(ui->textview), FALSE);
	480	ui->scrolled_window = gtk_scrolled_window_new(NULL, NULL);
	481	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(ui->scrolled_window),
	482	GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
	483	gtk_container_add(GTK_CONTAINER(ui->scrolled_window), ui->textview);
e164534f SC	484	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->scrolled_window,
e164534f SC	485	TRUE, TRUE, 0);
736f2dff	486
c36f98d9 SC	487	/*
	488	* Set up alignments for widgets at the bottom of ui,
	489	* align bottom left, expand horizontally but not vertically
	490	*/
	491	ui->bottomalign = gtk_alignment_new(0, 1, 1, 0);
	492	ui->buttonbox = gtk_hbox_new(FALSE, 0);
	493	gtk_container_add(GTK_CONTAINER(ui->bottomalign), ui->buttonbox);
e164534f SC	494	gtk_box_pack_start(GTK_BOX(ui->vbox), ui->bottomalign,
e164534f SC	495	FALSE, FALSE, 0);
c36f98d9	496
f3074008	497	add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist));
ff1f3280 SC	498	gtk_widget_show_all(ui->window);
	499	}
	500
8232e285	501	int main(int argc, char argv[], char envp[])
ff1f3280	502	{
8232e285 SC	503	if (initialize_fio(envp))
8232e285 SC	504	return 1;
0420ba6a JA	505	if (fio_init_options())
0420ba6a JA	506	return 1;
a1820207	507
ff1f3280	508	init_ui(&argc, &argv, &ui);
5b7573ab	509
2839f0c6	510	gdk_threads_enter();
ff1f3280	511	gtk_main();
2839f0c6	512	gdk_threads_leave();
ff1f3280 SC	513	return 0;
ff1f3280 SC	514	}