From f8c1d1562a662a1fae481abf872b18639e18d659 Mon Sep 17 00:00:00 2001 From: Timo Horstschaefer Date: Tue, 18 Apr 2017 18:26:43 +0200 Subject: [PATCH] add script to count events in rosbag --- dvs_file_writer/scripts/count_events_in_bag.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100755 dvs_file_writer/scripts/count_events_in_bag.py diff --git a/dvs_file_writer/scripts/count_events_in_bag.py b/dvs_file_writer/scripts/count_events_in_bag.py new file mode 100755 index 0000000..93300a6 --- /dev/null +++ b/dvs_file_writer/scripts/count_events_in_bag.py @@ -0,0 +1,21 @@ +#!/usr/bin/env python + +import rosbag +import dvs_msgs +import argparse + +parser = argparse.ArgumentParser(description='Count number of events in a rosbag') +parser.add_argument('--topic', default='/dvs/events') +parser.add_argument('bag') + +args = parser.parse_args() + +n_events = 0 + +for topic, msg, t in rosbag.Bag(args.bag).read_messages(): + # This also replaces tf timestamps under the assumption + # that all transforms in the message share the same timestamp + if topic == args.topic: + n_events += len(msg.events) + print "# events: ", n_events + \ No newline at end of file