trace-view: Added saving of filters

Add "Save filters" to "File" menu, where it will allow the user to save
the filters to a file in XML format.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

8 files changed, 323 insertions, 5 deletions

diff --git a/Makefile b/Makefile
index 7402ae1..cd4db4a 100644
--- a/Makefile
+++ b/Makefile
@@ -96,7 +96,7 @@ plugin_dir_SQ = $(subst ','\'',$(plugin_dir))
 LIBS = -L. -ltracecmd -ldl
 LIB_FILE = libtracecmd.a
 
-PACKAGES= gtk+-2.0
+PACKAGES= gtk+-2.0 libxml-2.0
 
 ifndef BUILDGUI
  BUILDGUI = 0
@@ -213,13 +213,15 @@ $(obj)/%.o: $(src)/%.c
 %.o: $(src)/%.c
         $(Q)$(call check_gui)
 
-TRACE_GUI_OBJS = trace-filter.o trace-compat.o trace-hash.o trace-dialog.o
+TRACE_GUI_OBJS = trace-filter.o trace-compat.o trace-hash.o trace-dialog.o \
+                trace-xml.o
 TRACE_CMD_OBJS = trace-cmd.o trace-usage.o trace-read.o trace-split.o trace-listen.o
 TRACE_VIEW_OBJS = trace-view.o trace-view-store.o
 TRACE_GRAPH_OBJS = trace-graph.o trace-plot.o trace-plot-cpu.o trace-plot-task.o
 TRACE_VIEW_MAIN_OBJS = trace-view-main.o $(TRACE_VIEW_OBJS) $(TRACE_GUI_OBJS)
 TRACE_GRAPH_MAIN_OBJS = trace-graph-main.o $(TRACE_GRAPH_OBJS) $(TRACE_GUI_OBJS)
-KERNEL_SHARK_OBJS = $(TRACE_VIEW_OBJS) $(TRACE_GRAPH_OBJS) $(TRACE_GUI_OBJS) kernel-shark.o
+KERNEL_SHARK_OBJS = $(TRACE_VIEW_OBJS) $(TRACE_GRAPH_OBJS) $(TRACE_GUI_OBJS) \
+        kernel-shark.o
 
 PEVENT_LIB_OBJS = parse-events.o trace-seq.o parse-filter.o
 TCMD_LIB_OBJS = $(PEVENT_LIB_OBJS) trace-util.o trace-input.o trace-ftrace.o \
diff --git a/trace-filter.c b/trace-filter.c
index 15d59c0..904c9cb 100644
--- a/trace-filter.c
+++ b/trace-filter.c
@@ -2158,3 +2158,54 @@ void trace_filter_convert_char_to_filter(struct event_filter *filter,
 
         pevent_filter_free(copy);
 }
+
+int trace_filter_save_events(struct tracecmd_xml_handle *handle,
+                             struct event_filter *filter)
+{
+        struct event_format *event;
+        char **systems;
+        gint *event_ids;
+        char *str;
+        int i;
+
+        trace_filter_convert_filter_to_names(filter, &systems,
+                                             &event_ids);
+
+        for (i = 0; systems && systems[i]; i++)
+                tracecmd_xml_write_element(handle, "System", systems[i]);
+
+        for (i = 0; event_ids && event_ids[i] > 0; i++) {
+                str = pevent_filter_make_string(filter, event_ids[i]);
+                if (!str)
+                        continue;
+
+                event = pevent_find_event(filter->pevent, event_ids[i]);
+                if (event) {
+
+                        /* skip not filtered items */
+                        if (strcmp(str, "FALSE") == 0) {
+                                free(str);
+                                continue;
+                        }
+
+                        tracecmd_xml_start_sub_system(handle, "Event");
+                        tracecmd_xml_write_element(handle, "System", event->system);
+                        tracecmd_xml_write_element(handle, "Name", event->name);
+                        /* If this is has an advanced filter, include that too */
+                        if (strcmp(str, "TRUE") != 0) {
+                                tracecmd_xml_write_element(handle, "Advanced",
+                                                           str);
+                        }
+                        tracecmd_xml_end_sub_system(handle);
+                }
+                free(str);
+        }
+
+        return 0;
+}
+
+int trace_filter_save_tasks(struct tracecmd_xml_handle *handle,
+                            struct filter_task *filter)
+{
+        return 0;
+}
diff --git a/trace-filter.h b/trace-filter.h
index c68e8f0..7575244 100644
--- a/trace-filter.h
+++ b/trace-filter.h
@@ -23,6 +23,8 @@
 
 #include <gtk/gtk.h>
 
+#include "trace-xml.h"
+
 struct event_filter_list {
         struct event_filter_list        *next;
         struct event                    *event;
@@ -140,4 +142,9 @@ int id_cmp(const void *a, const void *b);
 
 void trace_array_add(gint **array, gint *count, gint val);
 
+int trace_filter_save_events(struct tracecmd_xml_handle *handle,
+                             struct event_filter *filter);
+int trace_filter_save_tasks(struct tracecmd_xml_handle *handle,
+                            struct filter_task *filter);
+
 #endif /* _TRACE_FILTER_H */
diff --git a/trace-view-main.c b/trace-view-main.c
index 7f1461b..52fb4f8 100644
--- a/trace-view-main.c
+++ b/trace-view-main.c
@@ -28,6 +28,7 @@
 
 #include "trace-cmd.h"
 #include "trace-view.h"
+#include "trace-xml.h"
 #include "trace-gui.h"
 
 #define version "0.1.1"
@@ -71,6 +72,29 @@ load_clicked (gpointer data)
         g_free(filename);
 }
 
+/* Callback for the clicked signal of the Save State button */
+static void
+save_filters_clicked (gpointer data)
+{
+        struct trace_tree_info *info = data;
+        GtkTreeView *trace_tree = GTK_TREE_VIEW(info->trace_tree);
+        struct tracecmd_xml_handle *handle;
+        gchar *filename;
+
+        filename = trace_get_file_dialog("Save State");
+        if (!filename)
+                return;
+
+        handle = tracecmd_xml_create(filename);
+        if (!handle)
+                warning("Could not create save state %s", filename);
+        g_free(filename);
+
+        trace_view_save_filters(handle, trace_tree);
+
+        tracecmd_xml_close(handle);
+}
+
 /* Callback for the clicked signal of the Exit button */
 static void
 exit_clicked (GtkWidget *widget, gpointer data)
@@ -249,12 +273,11 @@ void trace_view(int argc, char **argv)
 
         /* --- File - Load Option --- */
 
-        sub_item = gtk_menu_item_new_with_label("Load info");
+        sub_item = gtk_menu_item_new_with_label("Load data");
 
         /* Add them to the menu */
         gtk_menu_shell_append(GTK_MENU_SHELL (menu), sub_item);
 
-        /* We can attach the Quit menu item to our exit function */
         g_signal_connect_swapped (G_OBJECT (sub_item), "activate",
                                   G_CALLBACK (load_clicked),
                                   (gpointer) &tree_info);
@@ -263,6 +286,21 @@ void trace_view(int argc, char **argv)
         gtk_widget_show(sub_item);
 
 
+        /* --- File - Save State Option --- */
+
+        sub_item = gtk_menu_item_new_with_label("Save filters");
+
+        /* Add them to the menu */
+        gtk_menu_shell_append(GTK_MENU_SHELL (menu), sub_item);
+
+        g_signal_connect_swapped (G_OBJECT (sub_item), "activate",
+                                  G_CALLBACK (save_filters_clicked),
+                                  (gpointer) &tree_info);
+
+        /* We do need to show menu items */
+        gtk_widget_show(sub_item);
+
+
         /* --- File - Quit Option --- */
 
         sub_item = gtk_menu_item_new_with_label("Quit");
diff --git a/trace-view.c b/trace-view.c
index 59e7e1a..c183b19 100644
--- a/trace-view.c
+++ b/trace-view.c
@@ -29,6 +29,7 @@
 #include "trace-local.h"
 #include "trace-view.h"
 #include "trace-compat.h"
+#include "version.h"
 #include "cpu.h"
 #include "util.h"
 
@@ -870,3 +871,50 @@ void trace_view_search_setup(GtkBox *box, GtkTreeView *treeview)
                                   G_CALLBACK (search_tree),
                                   (gpointer) info);
 }
+
+int trace_view_save_filters(struct tracecmd_xml_handle *handle,
+                            GtkTreeView *trace_tree)
+{
+        struct event_filter *event_filter;
+        GtkTreeModel *model;
+        TraceViewStore *store;
+        gboolean all_events;
+
+        model = gtk_tree_view_get_model(trace_tree);
+        if (!model)
+                return -1;
+
+        store = TRACE_VIEW_STORE(model);
+
+        tracecmd_xml_start_system(handle, "TraceView", VERSION_STRING);
+
+        all_events = trace_view_store_get_all_events_enabled(store);
+        event_filter = trace_view_store_get_event_filter(store);
+
+        tracecmd_xml_start_sub_system(handle, "EventFilter");
+
+        if (all_events || !event_filter)
+                tracecmd_xml_write_element(handle, "FilterType", "all events");
+        else {
+                tracecmd_xml_write_element(handle, "FilterType", "filter");
+                trace_filter_save_events(handle, event_filter);
+        }
+
+        tracecmd_xml_end_sub_system(handle);
+
+        if (store->task_filter) {
+                tracecmd_xml_start_sub_system(handle, "TaskFilter");
+                trace_filter_save_tasks(handle, store->task_filter);
+                tracecmd_xml_end_sub_system(handle);
+        }
+
+        if (store->hide_tasks) {
+                tracecmd_xml_start_sub_system(handle, "HideTasks");
+                trace_filter_save_tasks(handle, store->hide_tasks);
+                tracecmd_xml_end_sub_system(handle);
+        }
+
+        tracecmd_xml_end_system(handle);
+
+        return 0;
+}
diff --git a/trace-view.h b/trace-view.h
index 182b285..5ad33bf 100644
--- a/trace-view.h
+++ b/trace-view.h
@@ -23,6 +23,7 @@
 
 #include "trace-view-store.h"
 #include "trace-filter.h"
+#include "trace-xml.h"
 
 void
 trace_view_load(GtkWidget *view, struct tracecmd_input *handle,
@@ -65,4 +66,7 @@ void trace_view_search_setup(GtkBox *box, GtkTreeView *treeview);
 
 gint trace_view_get_selected_row(GtkWidget *treeview);
 
+int trace_view_save_filters(struct tracecmd_xml_handle *handle,
+                            GtkTreeView *treeview);
+
 #endif /* _TRACE_VIEW_H */
diff --git a/trace-xml.c b/trace-xml.c
new file mode 100644
index 0000000..0bc130c
--- /dev/null
+++ b/trace-xml.c
@@ -0,0 +1,127 @@
+/*
+ * Copyright (C) 2009, 2010 Red Hat Inc, Steven Rostedt <srostedt@redhat.com>
+ *
+ * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation;
+ * version 2.1 of the License (not later!)
+ *
+ * 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser 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 <stdio.h>
+#include <stdlib.h>
+#include <stdarg.h>
+#include <string.h>
+
+#include <libxml/xmlwriter.h>
+#include <libxml/parser.h>
+#include <libxml/encoding.h>
+
+#include "trace-cmd.h"
+#include "trace-xml.h"
+
+struct tracecmd_xml_handle {
+        xmlTextWriterPtr        writer;
+};
+
+#define TRACE_ENCODING "UTF-8"
+
+int tracecmd_xml_write_element(struct tracecmd_xml_handle *handle,
+                               const char *obj,
+                               const char *fmt, ...)
+{
+        va_list ap;
+        int ret;
+
+        va_start(ap, fmt);
+        ret = xmlTextWriterWriteVFormatElement(handle->writer,
+                                               BAD_CAST obj, fmt, ap);
+        va_end(ap);
+
+        return ret;
+}
+
+struct tracecmd_xml_handle *tracecmd_xml_create(const char *name)
+{
+        struct tracecmd_xml_handle *handle;
+        int ret;
+
+        handle = malloc_or_die(sizeof(*handle));
+        memset(handle, 0, sizeof(*handle));
+
+        handle->writer = xmlNewTextWriterFilename(name, 0);
+        if (!handle->writer)
+                goto fail_free;
+
+        ret = xmlTextWriterStartDocument(handle->writer, NULL,
+                                         TRACE_ENCODING, NULL);
+        if (ret < 0)
+                goto fail_close;
+
+        return handle;
+
+ fail_close:
+        xmlFreeTextWriter(handle->writer);
+ fail_free:
+        free(handle);
+        return NULL;
+}
+
+int tracecmd_xml_start_system(struct tracecmd_xml_handle *handle,
+                              const char *system, const char *version)
+{
+        int ret;
+
+        ret = xmlTextWriterStartElement(handle->writer,
+                                        BAD_CAST system);
+
+        if (ret < 0)
+                return ret;
+
+        ret = tracecmd_xml_write_element(handle, "Version", "%s", version);
+        if (ret < 0)
+                return ret;
+
+        return 0;
+}
+
+int tracecmd_xml_start_sub_system(struct tracecmd_xml_handle *handle,
+                                  const char *subsystem)
+{
+        int ret;
+
+        ret = xmlTextWriterStartElement(handle->writer,
+                                        BAD_CAST subsystem);
+
+        return ret;
+}
+
+void tracecmd_xml_end_system(struct tracecmd_xml_handle *handle)
+{
+        xmlTextWriterEndElement(handle->writer);
+}
+
+void tracecmd_xml_end_sub_system(struct tracecmd_xml_handle *handle)
+{
+        xmlTextWriterEndElement(handle->writer);
+}
+
+void tracecmd_xml_close(struct tracecmd_xml_handle *handle)
+{
+        if (handle->writer) {
+                xmlTextWriterEndDocument(handle->writer);
+                xmlFreeTextWriter(handle->writer);
+        }
+
+        free(handle);
+}
diff --git a/trace-xml.h b/trace-xml.h
new file mode 100644
index 0000000..41f6193
--- /dev/null
+++ b/trace-xml.h
@@ -0,0 +1,41 @@
+/*
+ * Copyright (C) 2009, 2010 Red Hat Inc, Steven Rostedt <srostedt@redhat.com>
+ *
+ * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation;
+ * version 2.1 of the License (not later!)
+ *
+ * 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser 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
+ *
+ * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+ */
+#ifndef __TRACE_XML_H
+#define __TRACE_XML_H
+
+struct tracecmd_xml_handle;
+
+struct tracecmd_xml_handle *tracecmd_xml_create(const char *name);
+void tracecmd_xml_close(struct tracecmd_xml_handle *handle);
+
+int tracecmd_xml_start_system(struct tracecmd_xml_handle *handle,
+                              const char *system, const char *version);
+void tracecmd_xml_end_system(struct tracecmd_xml_handle *handle);
+
+int tracecmd_xml_start_sub_system(struct tracecmd_xml_handle *handle,
+                                  const char *subsystem);
+void tracecmd_xml_end_sub_system(struct tracecmd_xml_handle *handle);
+
+int tracecmd_xml_write_element(struct tracecmd_xml_handle *handle,
+                               const char *obj,
+                               const char *fmt, ...);
+
+#endif /* __TRACE_XML_H */