| 1 |
#!/usr/bin/env python
|
| 2 |
|
| 3 |
import sys
|
| 4 |
import copy
|
| 5 |
import optparse
|
| 6 |
import re
|
| 7 |
|
| 8 |
|
| 9 |
class Branch(object):
|
| 10 |
pass
|
| 11 |
|
| 12 |
|
| 13 |
def branchType(branch):
|
| 14 |
type = cmstools.ROOT.branchToClass(branch).GetName()
|
| 15 |
if "edm::Wrapper" in type:
|
| 16 |
type = type.replace("edm::Wrapper<","").rstrip(">")
|
| 17 |
return type
|
| 18 |
|
| 19 |
|
| 20 |
def dumpBranches(filename):
|
| 21 |
events = cmstools.EventTree (filename)
|
| 22 |
listOfBranches = events._tree.GetListOfBranches()
|
| 23 |
trailingDotRE = re.compile (r'\.$')
|
| 24 |
branches = []
|
| 25 |
regexList = []
|
| 26 |
# are we asked to filter this list?
|
| 27 |
for regexString in options.regex:
|
| 28 |
#print "adding", regexString
|
| 29 |
regexList.append( re.compile( regexString, re.IGNORECASE ) )
|
| 30 |
for branch in listOfBranches:
|
| 31 |
# print branch.GetName()
|
| 32 |
tmpBranch = Branch()
|
| 33 |
tmpBranch.fullname = branch.GetName()
|
| 34 |
tmpBranch.name = trailingDotRE.sub ('', tmpBranch.fullname)
|
| 35 |
if "EventAux" in tmpBranch.fullname:
|
| 36 |
continue
|
| 37 |
#print "found", tmpBranch.name
|
| 38 |
parts = tmpBranch.fullname.split("_")
|
| 39 |
tmpBranch.module = parts [1]
|
| 40 |
tmpBranch.label = parts [2]
|
| 41 |
tmpBranch.process = parts [3]
|
| 42 |
tmpBranch.type = branchType(branch)
|
| 43 |
tmpBranch.cpp = events.cppCode (tmpBranch.fullname)
|
| 44 |
# are there any matches to the regexList?
|
| 45 |
found = False
|
| 46 |
for regex in regexList:
|
| 47 |
# search the branch name
|
| 48 |
if regex.search (tmpBranch.name):
|
| 49 |
found = True
|
| 50 |
break
|
| 51 |
# search the type
|
| 52 |
if regex.search (tmpBranch.cpp):
|
| 53 |
found = True
|
| 54 |
break
|
| 55 |
# if there is a list and no matches
|
| 56 |
if regexList and not found:
|
| 57 |
continue
|
| 58 |
branches.append( copy.copy(tmpBranch) )
|
| 59 |
for branch in branches:
|
| 60 |
#continue
|
| 61 |
if options.name:
|
| 62 |
print branch.name
|
| 63 |
elif options.all:
|
| 64 |
print '%-30s %-20s %-10s %-15s : %s' % \
|
| 65 |
(branch.type,
|
| 66 |
'"' + branch.module + '"',
|
| 67 |
'"' + branch.label + '"',
|
| 68 |
'"' + branch.process + '"',
|
| 69 |
branch.name)
|
| 70 |
else:
|
| 71 |
print '%-30s %-20s %-10s %-15s' % \
|
| 72 |
(branch.type,
|
| 73 |
'"' + branch.module + '"',
|
| 74 |
'"' + branch.label + '"',
|
| 75 |
'"' + branch.process + '"')
|
| 76 |
|
| 77 |
|
| 78 |
|
| 79 |
if __name__ == "__main__":
|
| 80 |
|
| 81 |
parser = optparse.OptionParser \
|
| 82 |
("usage: %prog [options] templates.root" \
|
| 83 |
"\nPrints out info on edm file.")
|
| 84 |
parser.add_option ('--name', dest='name', action='store_true',
|
| 85 |
help='print out only branch names')
|
| 86 |
parser.add_option ('--all', dest='all', action='store_true',
|
| 87 |
help='Print out everything: type, module, label, '\
|
| 88 |
'process, and branch name')
|
| 89 |
parser.add_option ("--regex", dest='regex', action="append",
|
| 90 |
type="string", default=[],
|
| 91 |
help="Filter results based on regex")
|
| 92 |
options, args = parser.parse_args()
|
| 93 |
if not args:
|
| 94 |
print parser.print_usage()
|
| 95 |
sys.exit()
|
| 96 |
try:
|
| 97 |
import PhysicsTools.PythonAnalysis as cmstools
|
| 98 |
## # to let ROOT understand we are in batch mode
|
| 99 |
## sys.argv.append( '-b-' )
|
| 100 |
import ROOT
|
| 101 |
ROOT.gROOT.SetBatch() # setting batch mode
|
| 102 |
ROOT.gSystem.Load("libFWCoreFWLite.so")
|
| 103 |
ROOT.AutoLibraryLoader.enable()
|
| 104 |
filename = args[0]
|
| 105 |
dumpBranches (filename)
|
| 106 |
except:
|
| 107 |
"Could not read %s" %filename
|
| 108 |
|
| 109 |
|
