fio: capture thread status display

[fio.git] / gfio.c

diff --git a/gfio.c b/gfio.c
index 36e27f71112b46926d1daf413c2ebce87dff9437..d736a261aab8705a92a4fd6da2bf13802a62262f 100644 (file)
--- a/gfio.c
+++ b/gfio.c
@@ -20,10 +20,84 @@
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  *
  */
+#include <locale.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 *buttonbox;
+       GtkWidget *button[ARRAYSIZE(buttonspeclist)];
+       pthread_t t;
+};
+
+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)
+{
+       printf("gfio_eta_op called\n");
+       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);
+}
+
+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,
 };
 
 static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
@@ -32,6 +106,51 @@ static void quit_clicked(__attribute__((unused)) GtkWidget *widget,
         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->window), 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)
 {
        gtk_init(argc, argv);
@@ -43,13 +162,20 @@ static void init_ui(int *argc, char **argv[], struct gui *ui)
        g_signal_connect(ui->window, "delete-event", G_CALLBACK (quit_clicked), NULL);
        g_signal_connect(ui->window, "destroy", G_CALLBACK (quit_clicked), NULL);
 
+       add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist));
        gtk_widget_show_all(ui->window);
 }
 
-int main(int argc, char *argv[])
+int main(int argc, char *argv[], char *envp[])
 {
        struct gui ui;
 
+       if (initialize_fio(envp))
+               return 1;
+
+       if (parse_options(argc, argv))
+               return 1;
+
        init_ui(&argc, &argv, &ui);
        gtk_main();
        return 0;