lttv/modules/text/batchAnalysis.c

   1 /* This file is part of the Linux Trace Toolkit viewer
   2  * Copyright (C) 2003-2004 Michel Dagenais
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License Version 2 as
   6  * published by the Free Software Foundation;
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program; if not, write to the Free Software
  15  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  16  * MA 02111-1307, USA.
  17  */
  18
  19 /* This module inserts a hook in the program main loop. This hook processes
  20    all the events in the main tracefile. */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <glib.h>
  27 #include <unistd.h>
  28 #include <lttv/lttv.h>
  29 #include <lttv/attribute.h>
  30 #include <lttv/hook.h>
  31 #include <lttv/option.h>
  32 #include <lttv/module.h>
  33 #include <lttv/traceset-process.h>
  34 #include <lttv/state.h>
  35 #ifdef BABEL_CLEANUP
  36 #include <lttv/stats.h>
  37 #include <lttv/filter.h>
  38 #endif
  39 #ifdef BABEL_CLEANUP_SYNC
  40 #include <lttv/sync/sync_chain_lttv.h>
  41 #endif
  42 #include <babeltrace/context.h>
  43
  44 static LttvTraceset *traceset;
  45
  46 static LttvHooks
  47   *before_traceset,
  48   *after_traceset,
  49   *before_trace,
  50   *after_trace,
  51   *before_tracefile,
  52   *after_tracefile,
  53   *event_hook,
  54   *main_hooks;
  55
  56 static char *trace_path;
  57
  58 static gboolean a_stats;
  59 static gboolean a_live;
  60 static int a_live_update_period;
  61
  62 #define DEFAULT_LIVE_UPDATE_PERIOD 1
  63
  64 void lttv_trace_option(void *hook_data)
  65 {
  66   //LttTrace *trace;
  67   //trace_collection *trace;
  68   //format *fmt = bt_lookup_format(g_quark_from_static_string("ctf"));
  69   /*
  70   if(a_live) {
  71     //trace = ltt_trace_open_live(a_trace);
  72   } else {
  73     bt_create_context();
  74     //trace = bt_add_trace(a_trace,"ctf");
  75   }
  76   if(trace == NULL) g_critical("cannot open trace %s", a_trace);
  77   lttv_traceset_add(traceset, lttv_trace_new(trace));*/
  78
  79   if(lttv_traceset_add_path(traceset, trace_path) < 0) {
  80     g_error("Cannot add trace %s", trace_path);
  81   }
  82 }
  83
  84
  85 static gboolean process_traceset(void *hook_data, void *call_data)
  86 {
  87   LttvAttributeValue value_expression, value_filter;
  88
  89   LttvIAttribute *attributes = LTTV_IATTRIBUTE(lttv_global_attributes());
  90
  91 #ifdef BABEL_CLEANUP
  92   LttvTracesetStats *tscs = NULL;
  93
  94   LttvTracesetState *tss;
  95
  96   LttvTracesetContext *tc;
  97 #endif
  98   LttTime start, end;
  99   gboolean retval;
 100
 101   g_info("BatchAnalysis begin process traceset");
 102 #ifdef BABEL_CLEANUP
 103   if (a_stats) {
 104     tscs = g_object_new(LTTV_TRACESET_STATS_TYPE, NULL);
 105     tss = &tscs->parent;
 106   } else {
 107     tss = g_object_new(LTTV_TRACESET_STATE_TYPE, NULL);
 108   }
 109   tc = &tss->parent;
 110
 111   g_info("BatchAnalysis initialize context");
 112
 113   lttv_context_init(tc, traceset);
 114
 115
 116   syncTraceset(tc);
 117
 118   lttv_state_add_event_hooks(tss);
 119   if(a_stats) lttv_stats_add_event_hooks(tscs);
 120
 121
 122   retval= lttv_iattribute_find_by_path(attributes, "filter/expression",
 123     LTTV_POINTER, &value_expression);
 124   g_assert(retval);
 125
 126   retval= lttv_iattribute_find_by_path(attributes, "filter/lttv_filter",
 127     LTTV_POINTER, &value_filter);
 128   g_assert(retval);
 129
 130   /* Repeat the search for the first element, the second search might have
 131    * moved the first element (by creating the second element)
 132    */
 133   retval= lttv_iattribute_find_by_path(attributes, "filter/expression",
 134     LTTV_POINTER, &value_expression);
 135   g_assert(retval);
 136
 137   *(value_filter.v_pointer) = lttv_filter_new();
 138   //g_debug("Filter string: %s",((GString*)*(value_expression.v_pointer))->str);
 139
 140   lttv_filter_append_expression(*(value_filter.v_pointer),((GString*)*(value_expression.v_pointer))->str);
 141 #endif
 142   //lttv_traceset_context_add_hooks(tc,
 143   //before_traceset, after_traceset, NULL, before_trace, after_trace,
 144   //NULL, before_tracefile, after_tracefile, NULL, before_event, after_event);
 145
 146   lttv_state_add_event_hooks(traceset);
 147   lttv_process_traceset_begin(traceset,
 148                               before_traceset,
 149                               before_trace,
 150                               event_hook);
 151
 152   start.tv_sec = 0;
 153   start.tv_nsec = 0;
 154   end.tv_sec = G_MAXULONG;
 155   end.tv_nsec = G_MAXULONG;
 156
 157   g_info("BatchAnalysis process traceset");
 158
 159   lttv_process_traceset_seek_time(traceset, start);
 160   /* Read as long a we do not reach the end (0) */
 161   unsigned int count;
 162   unsigned int updated_count;
 163   do {
 164           count = lttv_process_traceset_middle(traceset,
 165                                                           end,
 166                                                           G_MAXULONG,
 167                                                           NULL);
 168
 169 #ifdef BABEL_CLEANUP
 170           updated_count = lttv_process_traceset_update(tc);
 171 #endif
 172
 173           sleep(a_live_update_period);
 174   } while(count != 0
 175 #ifdef BABEL_CLEANUP
 176       || updated_count > 0
 177 #endif
 178       );
 179
 180
 181   //lttv_traceset_context_remove_hooks(tc,
 182   //before_traceset, after_traceset, NULL, before_trace, after_trace,
 183   //NULL, before_tracefile, after_tracefile, NULL, before_event, after_event);
 184   lttv_process_traceset_end(traceset,
 185                             after_traceset,
 186                             after_trace,
 187                             event_hook);
 188
 189   g_info("BatchAnalysis destroy context");
 190 #ifdef BABEL_CLEANUP
 191   lttv_filter_destroy(*(value_filter.v_pointer));
 192
 193   lttv_state_remove_event_hooks(tss);
 194   if(a_stats) lttv_stats_remove_event_hooks(tscs);
 195
 196   lttv_context_fini(tc);
 197   if (a_stats)
 198     g_object_unref(tscs);
 199   else
 200     g_object_unref(tss);
 201 #endif
 202   g_info("BatchAnalysis end process traceset");
 203   return FALSE;
 204 }
 205
 206
 207 static void init()
 208 {
 209   LttvAttributeValue value;
 210
 211   LttvIAttribute *attributes = LTTV_IATTRIBUTE(lttv_global_attributes());
 212   gboolean retval;
 213
 214   g_info("Init batchAnalysis.c");
 215
 216   lttv_option_add("trace", 't',
 217       "add a trace to the trace set to analyse",
 218       "pathname of the directory containing the trace",
 219       LTTV_OPT_STRING, &trace_path, lttv_trace_option, NULL);
 220
 221   a_stats = FALSE;
 222   lttv_option_add("stats", 's',
 223       "write the traceset and trace statistics",
 224       "",
 225       LTTV_OPT_NONE, &a_stats, NULL, NULL);
 226
 227   a_live = FALSE;
 228   lttv_option_add("live", 0,
 229       "define if the traceset is receiving live informations",
 230       "",
 231       LTTV_OPT_NONE, &a_live, NULL, NULL);
 232
 233   a_live_update_period = DEFAULT_LIVE_UPDATE_PERIOD;
 234   lttv_option_add("live-period", 0,
 235                   "period to update a live trace",
 236                   "in seconds",
 237                   LTTV_OPT_INT,
 238                   &a_live_update_period,
 239                   NULL, NULL);
 240
 241
 242   traceset = lttv_traceset_new();
 243
 244   before_traceset = lttv_hooks_new();
 245   after_traceset = lttv_hooks_new();
 246   before_trace = lttv_hooks_new();
 247   after_trace = lttv_hooks_new();
 248   before_tracefile = lttv_hooks_new();
 249   after_tracefile = lttv_hooks_new();
 250   //before_event = lttv_hooks_new();
 251   //after_event = lttv_hooks_new();
 252   event_hook = lttv_hooks_new();
 253
 254   retval= lttv_iattribute_find_by_path(attributes, "hooks/traceset/before",
 255     LTTV_POINTER, &value);
 256   g_assert(retval);
 257   *(value.v_pointer) = before_traceset;
 258   retval= lttv_iattribute_find_by_path(attributes, "hooks/traceset/after",
 259     LTTV_POINTER, &value);
 260   g_assert(retval);
 261   *(value.v_pointer) = after_traceset;
 262   retval= lttv_iattribute_find_by_path(attributes, "hooks/trace/before",
 263     LTTV_POINTER, &value);
 264   g_assert(retval);
 265   *(value.v_pointer) = before_trace;
 266   retval= lttv_iattribute_find_by_path(attributes, "hooks/trace/after",
 267     LTTV_POINTER, &value);
 268   g_assert(retval);
 269   *(value.v_pointer) = after_trace;
 270   retval= lttv_iattribute_find_by_path(attributes, "hooks/tracefile/before",
 271     LTTV_POINTER, &value);
 272   g_assert(retval);
 273   *(value.v_pointer) = before_tracefile;
 274   retval= lttv_iattribute_find_by_path(attributes, "hooks/tracefile/after",
 275     LTTV_POINTER, &value);
 276   g_assert(retval);
 277   *(value.v_pointer) = after_tracefile;
 278   //g_assert(lttv_iattribute_find_by_path(attributes, "hooks/event/before",
 279   //    LTTV_POINTER, &value));
 280   //*(value.v_pointer) = before_event;
 281   //g_assert(lttv_iattribute_find_by_path(attributes, "hooks/event/after",
 282   //    LTTV_POINTER, &value));
 283   //*(value.v_pointer) = after_event;
 284   retval= lttv_iattribute_find_by_path(attributes, "hooks/event",
 285     LTTV_POINTER, &value);
 286   g_assert(retval);
 287   *(value.v_pointer) = event_hook;
 288
 289   retval= lttv_iattribute_find_by_path(attributes, "hooks/main/before",
 290     LTTV_POINTER, &value);
 291   g_assert(retval);
 292   g_assert((main_hooks = *(value.v_pointer)) != NULL);
 293   lttv_hooks_add(main_hooks, process_traceset, NULL, LTTV_PRIO_DEFAULT);
 294 }
 295
 296 static void destroy()
 297 {
 298 #ifdef BABEL_CLEANUP
 299   guint i, nb;
 300
 301   LttvTrace *trace;
 302 #endif
 303
 304   g_info("Destroy batchAnalysis.c");
 305
 306   lttv_option_remove("trace");
 307   lttv_option_remove("stats");
 308   lttv_option_remove("live");
 309   lttv_option_remove("live-period");
 310
 311   lttv_hooks_destroy(before_traceset);
 312   lttv_hooks_destroy(after_traceset);
 313   lttv_hooks_destroy(before_trace);
 314   lttv_hooks_destroy(after_trace);
 315   lttv_hooks_destroy(before_tracefile);
 316   lttv_hooks_destroy(after_tracefile);
 317   //lttv_hooks_destroy(before_event);
 318   //lttv_hooks_destroy(after_event);
 319   lttv_hooks_destroy(event_hook);
 320   lttv_hooks_remove_data(main_hooks, process_traceset, NULL);
 321
 322 #ifdef BABEL_CLEANUP
 323   nb = lttv_traceset_number(traceset);
 324   for(i = 0 ; i < nb ; i++) {
 325     trace = lttv_traceset_get(traceset, i);
 326     ltt_trace_close(lttv_trace(trace));
 327     lttv_trace_destroy(trace);
 328   }
 329 #endif
 330   lttv_traceset_destroy(traceset);
 331 }
 332
 333 LTTV_MODULE("batchAnalysis", "Batch processing of a trace", \
 334     "Run through a trace calling all the registered hooks", \
 335     init, destroy, "state", "option")
 336 //TODO ybrosseau 2012-05-15 reenable textFilter, stats, sync