background = [['WW', 10524.477792], ['WWW', 79.037588], ['WWZ', 63.230267],
['WZZ', 17.997588], ['WZ_2l', 909.986306],
['WZ_3l', 1853.675094], ['ZJets_AMC', 1137600.914372],
['ZZZ', 3.428108], ['ZZ_2l_2q', 722.244018],
['ZZ_2l_2v', 212.112867], ['ZZ_4l', 946.472171],
['tW_neg_1', 9563.665061], ['tW_neg_2', 9548.841351],
['tW_pos_1', 9502.605773], ['tW_pos_2', 9546.889180],
['tZq', 85.976629], ['ttW_1', 196.426646], ['ttW_2', 195.472703],
['ttZ', 246.881972], ['tt_ll_AMC', 97340.321489],
['tt_ll_MG_1', 98109.617801], ['tt_ll_MG_2', 97747.879187]]
for s in signal:
c.add_signal(s[0], inFileDir + s[0] + ".root", s[1])
for b in background:
c.add_background(b[0], inFileDir + b[0] + ".root", b[1])
# c.add_signal('H2Mu_gg', inFileDir+"H2Mu_gg.root", 0.006343) # label, path, weight
# c.add_signal('H2Mu_VBF', inFileDir+"H2Mu_VBF.root", 0.000495)
# c.add_background('dy', inFileDir+"ZJets_AMC.root", 29.853717)
c.set_tree_path(treePath)
c.add_variable("muPairs.pt",
1) #second argument is the number of objects considered
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.dEta", 1) #for example, one muon pair
c.add_variable("muPairs.dPhi", 1)
c.add_variable("met.pt", 1)
c.add_variable("nJetsCent", 1)
c.add_variable("nJetsFwd", 1)
# ['ZJets_MG_HT_200_400_B', "/DYJetsToLL_M-50_HT-200to400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_200_400_B/180913_192001/0000/tuple_*root", 0.96*50.4177 ],
# ['ZJets_MG_HT_400_600_A', "/DYJetsToLL_M-50_HT-400to600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_400_600_A/180913_192022/0000/tuple_*root", 0.96*6.98394 ],
# ['ZJets_MG_HT_600_800', "/DYJetsToLL_M-50_HT-600to800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_600_800/180913_192118/0000/tuple_*root", 0.96*1.68141 ],
# ['ZJets_MG_HT_800_1200', "/DYJetsToLL_M-50_HT-800to1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_800_1200/180913_192145/0000/tuple_*root", 0.96*0.775392 ],
# ['ZJets_MG_HT_1200_2500', "/DYJetsToLL_M-50_HT-1200to2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_1200_2500/180913_192208/0000/tuple_*root", 0.96*0.186222 ],
# ['ZJets_MG_HT_2500_inf', "/DYJetsToLL_M-50_HT-2500toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_2500_inf/180913_192230/0000/tuple_*root", 0.96*0.004385 ],
# ['ZJets_hiM', "/DYJetsToLL_M-100to200_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8/ZJets_hiM/180913_191802/0000/tuple_*root", 5765.4*1.235 ]
]
for s in signal:
c.add_signal(s[0], mc_path+s[1], s[2])
for b in background:
c.add_background(b[0], bkg_path+b[1], b[2])
c.set_tree_path(treePath)
c.add_variable("muPairs.pt", 1) #second argument is the number of objects considered
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.phi", 1)
c.add_variable("muPairs.dEta", 1)
c.add_variable("muPairs.dPhi", 1)
c.add_variable("muons.eta", 2)
c.add_variable("met.pt", 1)
c.add_variable("nJets", 1)
c.add_variable("nJetsCent", 1)
# c.add_variable("nJetsFwd", 1)
c.add_variable("nBMed", 1)
# c.add_variable("jets.eta", 2)