# Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).

 # All rights reserved.

 # This component and the accompanying materials are made available

 # under the terms of "Eclipse Public License v1.0"

 # which accompanies this distribution, and is available

 # at the URL "http://www.eclipse.org/legal/epl-v10.html".

 #

 # Initial Contributors:

 # Nokia Corporation - initial contribution.

 #

 # Contributors:

 #

 # Description:

 # Groups an ASCII TracViewer trace by the contents of a named parameter. 	

 #

 import fileinput, string, re, sys, threadname

 # Utility function to group a list by a given regular expression.

 # returns a dictionary indexed by parameter 1 of the passed in pattern.

 def groupby(pattern, data):

 	r = {}

 	for entry in data:

 		matched = re.search(pattern, entry)

 		if matched:

 			r.setdefault(matched.group(1), []).append(entry)

 	return r

 # Show the usage if the parameters are not as expected

 if len(sys.argv) != 3:

 	print "Usage: fbsgroup <param|-t> <input>"

 	print "Where:"

 	print "<param> is a parameter to group by. (-t groups by thread id)"

 	print "<input> is the ASCII TraceViewer file to be parsed"

 	sys.exit(1)

 if sys.argv[1] == "-t":

 	pattern = "(Thread ID:0x.*$)"

 else:

 	pattern = "("+sys.argv[1]+"=*\w*);+"

 del sys.argv[1]

 # Add thread names to the raw trace

 rawinput = threadname.addnames(fileinput.input())

 # Group by the parameter supplied on the command line...

 results = groupby(pattern, rawinput)

 for group, entries in results.items():

 		print '\n'+group

 		# Show a count of the number of CFbsBitmap::xxx function calls

 		functions = groupby("(CFbsBitmap::\S+:)", entries)

 		for name, function in functions.items():

 			print "\t%s %s" % (name, len(function))

 		# Show a count of the number of CFbClient::xxx function calls	

 		functions = groupby("(CFbClient::\S+:)", entries)

 		for name, function in functions.items():

 			print "\t%s %s" % (name, len(function))

 		# Show a count of the number of RFbsSession::xxx function calls	

 		functions = groupby("(RFbsSession::\S+:)", entries)

 		for name, function in functions.items():

 			print "\t%s %s" % (name, len(function))

 		# Show the matching entries for this group

 		for entry in entries:

 			print "\t%s" % entry.strip()