Merge branch 'master' into gfio
[fio.git] / gfio.c
CommitLineData
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>
28
8232e285
SC
29#include "fio_initialization.h"
30#include "fio.h"
31
f3074008
SC
32#define ARRAYSIZE(x) (sizeof((x)) / (sizeof((x)[0])))
33
34typedef void (*clickfunction)(GtkWidget *widget, gpointer data);
35
36static void quit_clicked(GtkWidget *widget, gpointer data);
37static void start_job_clicked(GtkWidget *widget, gpointer data);
38
39static struct button_spec {
40 const char *buttontext;
41 clickfunction f;
42 const char *tooltiptext;
43} buttonspeclist[] = {
44#define START_JOB_BUTTON 0
45 { "Start Job",
46 start_job_clicked,
47 "Send current fio job to fio server to be executed" },
48#define QUIT_BUTTON 1
49 { "Quit", quit_clicked, "Quit gfio" },
50};
51
ff1f3280 52struct gui {
60f6b330
SC
53 int argc;
54 char **argv;
ff1f3280 55 GtkWidget *window;
5b7573ab 56 GtkWidget *vbox;
c36f98d9
SC
57 GtkWidget *topvbox;
58 GtkWidget *topalign;
59 GtkWidget *bottomalign;
04cc6b77 60 GtkWidget *thread_status_pb;
f3074008
SC
61 GtkWidget *buttonbox;
62 GtkWidget *button[ARRAYSIZE(buttonspeclist)];
45032dd8
SC
63 GtkWidget *hostname_hbox;
64 GtkWidget *hostname_label;
65 GtkWidget *hostname_entry;
66 GtkWidget *port_label;
67 GtkWidget *port_entry;
68 GtkWidget *hostname_combo_box; /* ipv4, ipv6 or socket */
47066341
SC
69 GtkWidget *jobfile_hbox;
70 GtkWidget *jobfile_label;
71 GtkWidget *jobfile_entry;
736f2dff
SC
72 GtkWidget *scrolled_window;
73 GtkWidget *textview;
74 GtkTextBuffer *text;
25927259 75 pthread_t t;
5b7573ab 76} ui;
ff1f3280 77
a1820207
SC
78static void gfio_text_op(struct fio_client *client,
79 FILE *f, __u16 pdu_len, const char *buf)
80{
736f2dff
SC
81 GtkTextBuffer *buffer;
82 GtkTextIter end;
83
84 buffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui.textview));
85 gdk_threads_enter();
86 gtk_text_buffer_get_end_iter(buffer, &end);
87 gtk_text_buffer_insert(buffer, &end, buf, -1);
88 gdk_threads_leave();
89 gtk_text_view_scroll_to_iter(GTK_TEXT_VIEW(ui.textview),
90 &end, 0.0, FALSE, 0.0,0.0);
a1820207
SC
91}
92
93static void gfio_disk_util_op(struct fio_client *client, struct fio_net_cmd *cmd)
94{
95 printf("gfio_disk_util_op called\n");
96 fio_client_ops.disk_util(client, cmd);
97}
98
99static void gfio_thread_status_op(struct fio_net_cmd *cmd)
100{
101 printf("gfio_thread_status_op called\n");
102 fio_client_ops.thread_status(cmd);
103}
104
105static void gfio_group_stats_op(struct fio_net_cmd *cmd)
106{
107 printf("gfio_group_stats_op called\n");
108 fio_client_ops.group_stats(cmd);
109}
110
111static void gfio_eta_op(struct fio_client *client, struct fio_net_cmd *cmd)
112{
a1820207
SC
113 fio_client_ops.eta(client, cmd);
114}
115
116static void gfio_probe_op(struct fio_client *client, struct fio_net_cmd *cmd)
117{
118 printf("gfio_probe_op called\n");
119 fio_client_ops.probe(client, cmd);
120}
121
04cc6b77 122static void gfio_update_thread_status(char *status_message, double perc)
5b7573ab
SC
123{
124 static char message[100];
125 const char *m = message;
126
127 strncpy(message, status_message, sizeof(message) - 1);
04cc6b77
SC
128 gtk_progress_bar_set_text(
129 GTK_PROGRESS_BAR(ui.thread_status_pb), m);
130 gtk_progress_bar_set_fraction(
131 GTK_PROGRESS_BAR(ui.thread_status_pb), perc / 100.0);
5b7573ab
SC
132 gdk_threads_enter();
133 gtk_widget_queue_draw(ui.window);
134 gdk_threads_leave();
135}
136
a1820207
SC
137struct client_ops gfio_client_ops = {
138 gfio_text_op,
139 gfio_disk_util_op,
140 gfio_thread_status_op,
141 gfio_group_stats_op,
142 gfio_eta_op,
143 gfio_probe_op,
5b7573ab 144 gfio_update_thread_status,
a1820207
SC
145};
146
ff1f3280
SC
147static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
148 __attribute__((unused)) gpointer data)
149{
150 gtk_main_quit();
151}
152
60f6b330
SC
153static void add_arg(char **argv, int index, const char *value)
154{
155 argv[index] = malloc(strlen(value) + 1);
156 strcpy(argv[index], value);
157}
158
159static void free_args(int argc, char **argv)
160{
161 int i;
162
163 for (i = 0; i < argc; i++)
164 free(argv[i]);
165 free(argv);
166}
167
25927259
SC
168static void *job_thread(void *arg)
169{
170 struct gui *ui = arg;
171
172 fio_handle_clients(&gfio_client_ops);
173 gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
60f6b330 174 free_args(ui->argc, ui->argv);
25927259
SC
175 return NULL;
176}
177
60f6b330
SC
178static void construct_options(struct gui *ui, int *argc, char ***argv)
179{
180 const char *hostname, *hostname_type, *port, *jobfile;
181 char newarg[200];
182
183 hostname_type = gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(ui->hostname_combo_box)->entry));
184 hostname = gtk_entry_get_text(GTK_ENTRY(ui->hostname_entry));
185 port = gtk_entry_get_text(GTK_ENTRY(ui->port_entry));
186 jobfile = gtk_entry_get_text(GTK_ENTRY(ui->jobfile_entry));
187
188 *argc = 3;
189 *argv = malloc(*argc * sizeof(**argv));
190 add_arg(*argv, 0, "gfio");
191 snprintf(newarg, sizeof(newarg) - 1, "--client=%s", hostname);
192 add_arg(*argv, 1, newarg);
193 add_arg(*argv, 2, jobfile);
194}
195
25927259
SC
196static void start_job_thread(pthread_t *t, struct gui *ui)
197{
60f6b330
SC
198 construct_options(ui, &ui->argc, &ui->argv);
199 if (parse_options(ui->argc, ui->argv)) {
200 printf("Yeah, I didn't really like those options too much.\n");
201 free_args(ui->argc, ui->argv);
202 gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1);
203 return;
204 }
25927259
SC
205 pthread_create(t, NULL, job_thread, ui);
206}
207
f3074008 208static void start_job_clicked(__attribute__((unused)) GtkWidget *widget,
25927259 209 gpointer data)
f3074008 210{
25927259
SC
211 struct gui *ui = data;
212
f3074008 213 printf("Start job button was clicked.\n");
25927259
SC
214 gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 0);
215 start_job_thread(&ui->t, ui);
f3074008
SC
216}
217
218static void add_button(struct gui *ui, int i, GtkWidget *buttonbox,
219 struct button_spec *buttonspec)
220{
221 ui->button[i] = gtk_button_new_with_label(buttonspec->buttontext);
222 g_signal_connect(ui->button[i], "clicked", G_CALLBACK (buttonspec->f), ui);
223 gtk_box_pack_start(GTK_BOX (ui->buttonbox), ui->button[i], TRUE, TRUE, 0);
224 gtk_widget_set_tooltip_text(ui->button[i], buttonspeclist[i].tooltiptext);
225}
226
227static void add_buttons(struct gui *ui,
228 struct button_spec *buttonlist,
229 int nbuttons)
230{
231 int i;
232
f3074008
SC
233 for (i = 0; i < nbuttons; i++)
234 add_button(ui, i, ui->buttonbox, &buttonlist[i]);
235}
236
ff1f3280
SC
237static void init_ui(int *argc, char **argv[], struct gui *ui)
238{
45032dd8
SC
239 GList *hostname_type_list = NULL;
240 char portnum[20];
241
2839f0c6 242 /* Magical g*thread incantation, you just need this thread stuff.
04cc6b77 243 * Without it, the update that happens in gfio_update_thread_status
2839f0c6
SC
244 * doesn't really happen in a timely fashion, you need expose events
245 */
246 if (!g_thread_supported ())
247 g_thread_init(NULL);
248 gdk_threads_init();
249
ff1f3280
SC
250 gtk_init(argc, argv);
251
252 ui->window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
253 gtk_window_set_title(GTK_WINDOW(ui->window), "fio");
254 gtk_window_set_default_size(GTK_WINDOW(ui->window), 700, 500);
255
256 g_signal_connect(ui->window, "delete-event", G_CALLBACK (quit_clicked), NULL);
257 g_signal_connect(ui->window, "destroy", G_CALLBACK (quit_clicked), NULL);
258
5b7573ab
SC
259 ui->vbox = gtk_vbox_new(FALSE, 0);
260 gtk_container_add(GTK_CONTAINER (ui->window), ui->vbox);
04cc6b77 261
c36f98d9
SC
262 /*
263 * Set up alignments for widgets at the top of ui,
264 * align top left, expand horizontally but not vertically
265 */
266 ui->topalign = gtk_alignment_new(0, 0, 1, 0);
267 ui->topvbox = gtk_vbox_new(FALSE, 0);
268 gtk_container_add(GTK_CONTAINER(ui->topalign), ui->topvbox);
e164534f 269 gtk_box_pack_start(GTK_BOX(ui->vbox), ui->topalign, FALSE, FALSE, 0);
c36f98d9 270
45032dd8
SC
271 /*
272 * Set up hostname label + entry, port label + entry,
273 */
274 ui->hostname_hbox = gtk_hbox_new(FALSE, 0);
275 ui->hostname_label = gtk_label_new("Host:");
276 ui->hostname_entry = gtk_entry_new();
277 gtk_entry_set_text(GTK_ENTRY(ui->hostname_entry), "localhost");
278 ui->port_label = gtk_label_new("Port:");
279 ui->port_entry = gtk_entry_new();
280 snprintf(portnum, sizeof(portnum) - 1, "%d", FIO_NET_PORT);
281 gtk_entry_set_text(GTK_ENTRY(ui->port_entry), (gchar *) portnum);
282
283 /*
284 * Set up combo box for address type
285 */
286 ui->hostname_combo_box = gtk_combo_new();
287 gtk_entry_set_text(GTK_ENTRY (GTK_COMBO(ui->hostname_combo_box)->entry), "IPv4");
288 hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv4");
289 hostname_type_list = g_list_append(hostname_type_list, (gpointer) "local socket");
290 hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv6");
291 gtk_combo_set_popdown_strings (GTK_COMBO (ui->hostname_combo_box), hostname_type_list);
292 g_list_free(hostname_type_list);
293
294 gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_label);
295 gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_entry);
296 gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_label);
297 gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_entry);
298 gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_combo_box);
c36f98d9 299 gtk_container_add(GTK_CONTAINER (ui->topvbox), ui->hostname_hbox);
45032dd8 300
47066341
SC
301 /*
302 * Set up jobfile text entry (temporary until gui really works)
303 */
304 ui->jobfile_hbox = gtk_hbox_new(FALSE, 0);
305 ui->jobfile_label = gtk_label_new("Job file:");
306 ui->jobfile_entry = gtk_entry_new();
307 gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_label);
308 gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_entry);
c36f98d9 309 gtk_container_add(GTK_CONTAINER (ui->topvbox), ui->jobfile_hbox);
47066341 310
04cc6b77
SC
311 /*
312 * Set up thread status progress bar
313 */
314 ui->thread_status_pb = gtk_progress_bar_new();
315 gtk_progress_bar_set_fraction(
316 GTK_PROGRESS_BAR(ui->thread_status_pb), 0.0);
317 gtk_progress_bar_set_text(
318 GTK_PROGRESS_BAR(ui->thread_status_pb), "No jobs running");
c36f98d9 319 gtk_container_add(GTK_CONTAINER (ui->topvbox), ui->thread_status_pb);
5b7573ab 320
736f2dff
SC
321 /*
322 * Add a text box for text op messages
323 */
324 ui->textview = gtk_text_view_new();
325 ui->text = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui->textview));
326 gtk_text_buffer_set_text(ui->text, "", -1);
327 gtk_text_view_set_editable(GTK_TEXT_VIEW(ui->textview), FALSE);
328 gtk_text_view_set_cursor_visible(GTK_TEXT_VIEW(ui->textview), FALSE);
329 ui->scrolled_window = gtk_scrolled_window_new(NULL, NULL);
330 gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(ui->scrolled_window),
331 GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
332 gtk_container_add(GTK_CONTAINER(ui->scrolled_window), ui->textview);
e164534f
SC
333 gtk_box_pack_start(GTK_BOX(ui->vbox), ui->scrolled_window,
334 TRUE, TRUE, 0);
736f2dff 335
c36f98d9
SC
336 /*
337 * Set up alignments for widgets at the bottom of ui,
338 * align bottom left, expand horizontally but not vertically
339 */
340 ui->bottomalign = gtk_alignment_new(0, 1, 1, 0);
341 ui->buttonbox = gtk_hbox_new(FALSE, 0);
342 gtk_container_add(GTK_CONTAINER(ui->bottomalign), ui->buttonbox);
e164534f
SC
343 gtk_box_pack_start(GTK_BOX(ui->vbox), ui->bottomalign,
344 FALSE, FALSE, 0);
c36f98d9 345
f3074008 346 add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist));
ff1f3280
SC
347 gtk_widget_show_all(ui->window);
348}
349
8232e285 350int main(int argc, char *argv[], char *envp[])
ff1f3280 351{
8232e285
SC
352 if (initialize_fio(envp))
353 return 1;
a1820207 354
ff1f3280 355 init_ui(&argc, &argv, &ui);
5b7573ab 356
2839f0c6 357 gdk_threads_enter();
ff1f3280 358 gtk_main();
2839f0c6 359 gdk_threads_leave();
ff1f3280
SC
360 return 0;
361}