Babeltrace python binding
[babeltrace.git] / bindings / python / examples / eventcountlist.py
1 # eventcountlist.py
2 #
3 # Babeltrace event count list example script
4 #
5 # Copyright 2012 EfficiOS Inc.
6 #
7 # Author: Danny Serres <danny.serres@efficios.com>
8 #
9 # Permission is hereby granted, free of charge, to any person obtaining a copy
10 # of this software and associated documentation files (the "Software"), to deal
11 # in the Software without restriction, including without limitation the rights
12 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
13 # copies of the Software, and to permit persons to whom the Software is
14 # furnished to do so, subject to the following conditions:
15 #
16 # The above copyright notice and this permission notice shall be included in
17 # all copies or substantial portions of the Software.
18
19 # The script prints a count and rate of events.
20 # It also outputs a bar graph of count per event, using the cairoplot module.
21
22 import sys
23 from babeltrace import *
24 from output_format_modules import cairoplot
25 from output_format_modules.pprint_table import pprint_table as pprint
26
27 # Check for path arg:
28 if len(sys.argv) < 2:
29 raise TypeError("Usage: python eventcountlist.py path/to/trace")
30
31 ctx = Context()
32 ret = ctx.add_trace(sys.argv[1], "ctf")
33 if ret is None:
34 raise IOError("Error adding trace")
35
36 # Events and their assossiated count
37 # will be stored as a dict:
38 events_count = {}
39
40 # Setting iterator:
41 bp = IterPos(SEEK_BEGIN)
42 ctf_it = ctf.Iterator(ctx,bp)
43
44 prev_event = None
45 event = ctf_it.read_event()
46
47 start_time = event.get_timestamp()
48
49 # Reading events:
50 while(event is not None):
51 if event.get_name() in events_count:
52 events_count[event.get_name()] += 1
53 else:
54 events_count[event.get_name()] = 1
55
56 ret = ctf_it.next()
57 if ret < 0:
58 break
59 else:
60 prev_event = event
61 event = ctf_it.read_event()
62
63 if event:
64 total_time = event.get_timestamp() - start_time
65 else:
66 total_time = prev_event.get_timestamp() - start_time
67
68 del ctf_it
69
70 # Printing encountered events with respective count and rate:
71 print("Total time: {} ns".format(total_time))
72 table = [["EVENT", "COUNT", "RATE (Hz)"]]
73 for item in sorted(events_count.iterkeys()):
74 tmp = [item, events_count[item],
75 events_count[item]/(total_time/1000000000.0)]
76 table.append(tmp)
77 pprint(table)
78
79 # Exporting data as bar graph
80 cairoplot.vertical_bar_plot ( 'eventcountlist.svg', events_count, 50+85*len(events_count),
81 800, border = 20, display_values = True, grid = True,
82 rounded_corners = True,
83 x_labels = sorted(events_count.keys()) )
This page took 0.031818 seconds and 4 git commands to generate.