pyroot_unit_tests/met_systematics.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
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)

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

from ROOT.met import METSystematicsTool

met_syst_tool = METSystematicsTool('met_maker_tool')
met_syst_tool.setProperty('std::string')('ConfigSoftTrkFile', 'TrackSoftTerms.config')
met_syst_tool.setProperty('std::string')('ConfigSoftCaloFile', 'METRefFinal.config')
met_syst_tool.initialize()

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

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

    if met.met() > 0:
        count_10_events +=1
    mets_copy = store_helper.shallowCopyMissingETContainer(mets)
    met_copy = mets_copy[0]

    if met.met() > 0:
        print 20 * '-'
        print '######  EVENT {0} ######'.format(i)
        print 'BEFORE Met: pt(orig), pt(copy) = ', met.met(), met_copy.met()
    # met_calib_tool.applyCalibration(met_copy)
        print 'AFTER Met: pt(orig), pt(copy) = ', met.met(), met_copy.met()

    store.clear()