pyroot_unit_tests/met_making.py at master · qbuat/pyroot_unit_tests · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os

import ROOT
ROOT.gROOT.Macro('$ROOTCOREDIR/scripts/load_packages.C')
ROOT.xAOD.Init().ignore()
ROOT.xAOD.AuxContainerBase()

files = [
    '/cluster/data01/qbuat/xaod_skims/testinput/mc12_powhegpythia_ztautau/mc14_8TeV.147808.PowhegPythia8_AU2CT10_Ztautau.merge.AOD.e2372_s1933_s1911_r5591_r5625/AOD.01512482._001049.pool.root.1',
]

chain = ROOT.TChain('CollectionTree')
for f in files:
    chain.Add(f)

# TOOLS = ROOT.asg.ToolStore()


tree = ROOT.xAOD.MakeTransientTree(chain)
print tree

# from ROOT.met import METMaker
# met_making_tool = METMaker('met_maker_tool')
# met_making_tool.setProperty('double')('JetPtCut', 20000.)
# met_making_tool.setProperty('double')('JetJVFCut', 0.25)
# met_making_tool.setProperty('double')('JetMinEFrac', 0.5)
# met_making_tool.setProperty('double')('JetMinWeightedPt', 0.)
# met_making_tool.setProperty('bool')('DoJetJVFCut', True)
# met_making_tool.setProperty('bool')('CorrectJetPhi', False)
# sc = met_making_tool.initialize()

# from ROOT import METHandler
# met_tool = METHandler(False)

store_helper = ROOT.xAOD.StorePyHelper()
store = ROOT.xAOD.TPyStore()

for i, event in enumerate(tree):
    if i > 10:
        break
    mets = tree.MET_RefFinal
    met = mets['Final']


    mets_copy = store_helper.shallowCopyMissingETContainer(mets)
    met_copy = mets_copy['Final']
    jets_copy = store_helper.shallowCopyJetContainer(event.AntiKt4LCTopoJets)

    if met.met() > 0:
        print 20 * '-'
        print '######  EVENT {0} ######'.format(i)
        print 'MET (orig), (copy), (calib) = ', met.met(), met_copy.met()
    store.clear()