File indexing completed on 2024-11-25 02:29:53
0001 import ROOT
0002 import pprint
0003 import sys
0004 from DataFormats.FWLite import Events, Handle
0005 ROOT.gROOT.SetBatch()
0006 from argparse import ArgumentParser
0007
0008 parser = ArgumentParser()
0009 parser.add_argument('file')
0010 parser.add_argument('collections', default=['slimmedJets'], nargs='*')
0011 args = parser.parse_args()
0012
0013 events = Events(args.file)
0014 jet_labels = args.collections
0015 tested_discriminators = ['pfCombinedCvsLJetTags', 'pfCombinedCvsBJetTags']
0016
0017 evt = next(events.__iter__())
0018 handle = Handle('std::vector<pat::Jet>')
0019 for label in jet_labels:
0020 evt.getByLabel(label, handle)
0021 jets = handle.product()
0022 jet = jets.at(0)
0023 available = set([i.first for i in jet.getPairDiscri()])
0024 for test in tested_discriminators:
0025 print("%s in %s: %s" % (test, label, test in available))
