#addAsBranch = True
addAsBranch = False
print 'w.getBranches()', w.getBranches()
if addAsBranch:
# add all new branches defined in the above module as output
# this links the above module with the SampleTree class
# which will call the processEvent() during the loop to
# fill the output branches
sampleTree.addOutputBranches(w.getBranches())
# output files
sampleTree.addOutputTree(
'/scratch/berger_p2/testWithWeight_MET_below_250.root',
cut='MET_Pt<250',
branches='*')
sampleTree.addOutputTree(
'/scratch/berger_p2/testWithWeight_MET_above_250.root',
cut='MET_Pt>250',
branches='*')
# loop over all events!
sampleTree.process()
else:
# process the first event only
w.processEvent(sampleTree.tree)
print('weight:', w._b('weight')[0])
print('weightF:', w._b('weightF')[0])
print('weightXS:', w._b('weightXS')[0])
'/store/group/phys_higgs/hbb/ntuples/VHbbPostNano/2017/V11/WplusH_HToBB_WToLNu_M125_13TeV_powheg_pythia8/adewit-crab_nano2017_WplusH_HT81/190606_065851/0000/tree_1.root'
],
treeName='Events',
xrootdRedirector="root://eoscms.cern.ch/")
# initialize module
w = JetSmearer("2017")
w.customInit({
'sampleTree': sampleTree,
'sample': sample,
'config': config
})
n = 0
#for event in sampleTree:
# w.processEvent(event)
# n=n+1
# if n==3: break
sampleTree.addOutputBranches(w.getBranches())
# output files
sampleTree.addOutputTree('/scratch/berger_p2/test.root',
cut='1',
branches='*')
sampleTree.addCallback('event', w.processEvent)
sampleTree.addCallback(
'event', lambda x: False if x.GetReadEntry() > 1000 else True)
# loop over all events!
sampleTree.process()