Exemplo n.º 1
0
   jet_maps = pickle.load(infile)

print 'build jet by jet map from flat tree'
flat_map = {}
inputFile = io.root_open('trees/CombinedSV_ALL.root')
flat_tree = inputFile.tree

for entry in flat_tree:
   evtid = (entry.run, entry.lumi, entry.evt)
   if evtid not in jet_maps:
      continue
   if (entry.jetPt, entry.jetEta) not in jet_maps[evtid]:
      continue

   if evtid not in flat_map: flat_map[evtid] = {}
   flat_map[evtid][(entry.jetPt, entry.jetEta)] = Struct.from_entry(entry)


handle = Handle('std::vector<pat::Jet>')
vtx_handle = Handle('vector<reco::Vertex>')
print 'analyzing pat output'
same_mva, same_inputs = 0, 0
n_analyzed_jets = 0
different_jets = {}
for evt in events:
   evtid = (evt.eventAuxiliary().run(), evt.eventAuxiliary().luminosityBlock(), evt.eventAuxiliary().event())

   evt.getByLabel('selectedPatJets', handle)
   jets = handle.product()
   ext_jets = []
   for jet in jets: