X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=ltt%2Fbranches%2Fpoly%2Flttv%2Fmodules%2Fgui%2Flttvwindow%2Flttvwindow%2Finit_module.c;h=a6914fff40f1bf8b37d9834a3859890fdb043099;hb=313bd6fc2be6b0b81b4dde2c9355e1e2edc21291;hp=4428cb4d11b7acd4937cfccca3369318c78cc4c0;hpb=6395d57c1887bfba11a50d1d1905dc0f01f6a5a5;p=lttv.git

diff --git a/ltt/branches/poly/lttv/modules/gui/lttvwindow/lttvwindow/init_module.c b/ltt/branches/poly/lttv/modules/gui/lttvwindow/lttvwindow/init_module.c
index 4428cb4d..a6914fff 100644
--- a/ltt/branches/poly/lttv/modules/gui/lttvwindow/lttvwindow/init_module.c
+++ b/ltt/branches/poly/lttv/modules/gui/lttvwindow/lttvwindow/init_module.c
@@ -66,6 +66,8 @@ LttvTraceInfo LTTV_TRACES,
        LTTV_AFTER_REQUEST,
        LTTV_EVENT_HOOK,
        LTTV_EVENT_HOOK_BY_ID,
+       LTTV_HOOK_ADDER,
+       LTTV_HOOK_REMOVER,
        LTTV_IN_PROGRESS,
        LTTV_READY;
 
@@ -158,6 +160,8 @@ static void init() {
   LTTV_AFTER_REQUEST = g_quark_from_string("after_request");
   LTTV_EVENT_HOOK = g_quark_from_string("event_hook");
   LTTV_EVENT_HOOK_BY_ID = g_quark_from_string("event_hook_by_id");
+  LTTV_HOOK_ADDER = g_quark_from_string("hook_adder");
+  LTTV_HOOK_REMOVER = g_quark_from_string("hook_remover");
   LTTV_IN_PROGRESS = g_quark_from_string("in_progress");
   LTTV_READY = g_quark_from_string("ready");
 
@@ -174,6 +178,16 @@ static void init() {
 
   lttv_hooks_add(main_hooks, window_creation_hook, NULL, LTTV_PRIO_DEFAULT);
 
+  LttvHooks *hook_adder = lttv_hooks_new();
+  lttv_hooks_add(hook_adder, lttv_state_save_hook_add_event_hooks, NULL,
+                 LTTV_PRIO_DEFAULT);
+  LttvHooks *hook_remover = lttv_hooks_new();
+  lttv_hooks_add(hook_remover, lttv_state_save_hook_remove_event_hooks,
+                                  NULL, LTTV_PRIO_DEFAULT);
+  /* Add state computation background hook adder to attributes */
+  lttvwindowtraces_register_computation_hooks(g_quark_from_string("state"),
+      NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
+      hook_adder, hook_remover);
 }
 
 void
@@ -196,6 +210,9 @@ static void destroy() {
   LttvTrace *trace;
   GSList *iter = NULL;
   
+  lttvwindowtraces_unregister_requests(g_quark_from_string("state"));
+  lttvwindowtraces_unregister_computation_hooks(g_quark_from_string("state"));
+
   lttv_option_remove("trace");
 
   lttv_hooks_remove_data(main_hooks, window_creation_hook, NULL);