# ['ZJets_MG_HT_100_200_A', "/DYJetsToLL_M-50_HT-100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_100_200_A/180913_191844/0000/", 0.96*181.302 ],
# ['ZJets_MG_HT_100_200_B', "/DYJetsToLL_M-50_HT-100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_100_200_B/180913_191907/0000/", 0.96*181.302 ],
# ['ZJets_MG_HT_200_400_A', "/DYJetsToLL_M-50_HT-200to400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_200_400_A/180913_191940/0000/", 0.96*50.4177 ],
# ['ZJets_MG_HT_200_400_B', "/DYJetsToLL_M-50_HT-200to400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8/ZJets_MG_HT_200_400_B/180913_192001/0000/", 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/", 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/", 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/", 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/", 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/", 0.96*0.004385 ],
# ['ZJets_hiM', "/DYJetsToLL_M-100to200_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8/ZJets_hiM/180913_191802/0000/", 5765.4*1.235 ]
]
for s in signal:
c.add_signal_dir(s[0], mc_path+s[1], s[2])
for b in background:
c.add_background_dir(b[0], bkg_path+b[1], b[2])
c.set_tree_path(treePath)
c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.dEta", 1)
c.add_variable("muPairs.dPhi", 1)
c.add_variable("met.pt", 1)
c.add_variable("nJetsCent", 1)
c.add_variable("nJetsFwd", 1)
c.add_variable("nBMed", 1)
c.add_variable("jets.eta", 2)
c.add_comment(comment)
print comment
c.outDir = '/tmp/dkondrat/ML_output/'
treePath = 'dimuons/tree'
metadataPath = 'dimuons/metadata'
c.set_tree_path(treePath)
c.set_metadata_path(metadataPath)
c.set_year("2017")
c.ebe_weights = True
##################### Input samples #######################
c.add_signal_dir(ggH_2017_powheg.name, ggH_2017_powheg.path,
ggH_2017_powheg.xSec)
c.add_signal_dir(VBF_2017_powheg.name, VBF_2017_powheg.path,
VBF_2017_powheg.xSec)
c.add_background_dir(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path,
ZJets_aMC_2017_hiStat.xSec)
c.add_background_dir(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
tt_ll_POW_2017.xSec)
##########################################################
### ------ Raffaele's variables ------ ###
c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.cosThetaCS", 1)
c.add_variable("muPairs.phiCS", 1)
# c.add_category('ZJets_aMC', False)
# c.add_dir_to_category('ZJets_aMC', path+'/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/', 5765.4, 'ZJets_aMC')
# c.add_category('tt_ll_POW', False)
# c.add_dir_to_category('tt_ll_POW', path+'/TTTo2L2Nu_TuneCP5_PSweights_13TeV-powheg-pythia8/tt_ll_POW/190322_195819/0000/', 85.656, 'tt_ll_POW')
# c.add_signal_dir('H2Mu_gg', path+'/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/', 0.009618)
# c.add_signal_dir('H2Mu_VBF', path+'/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/', 0.0008208)
# c.add_background_dir('ZJets_aMC', path+'/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/', 5765.4)
# c.add_background_dir('tt_ll_POW', path+'/TTTo2L2Nu_TuneCP5_PSweights_13TeV-powheg-pythia8/tt_ll_POW/190322_195819/0000/', 85.656)
path = path = "/tmp/dkondrat/updated_2017_mc/"
c.add_signal_dir('H2Mu_gg', path+'/H2Mu_ggH/', 0.009618)
c.add_signal_dir('H2Mu_VBF', path+'/H2Mu_VBF/', 0.0008208)
c.add_background_dir('ZJets_aMC', path+'/ZJets_AMC/', 5765.4)
c.add_background_dir('tt_ll_POW', path+'/tt_ll_POW/', 85.656)
# c.add_signal_file('H2Mu_gg', path+'/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/tuple_1.root', 0.009618)
# c.add_signal_file('H2Mu_VBF', path+'/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/tuple_1.root', 0.0008208)
# c.add_background_file('ZJets_aMC', path+'/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/ZJets_AMC/190323_195010/0000/tuple_1.root', 5765.4)
# c.add_background_file('tt_ll_POW', path+'/TTTo2L2Nu_TuneCP5_PSweights_13TeV-powheg-pythia8/tt_ll_POW/190322_195819/0000/tuple_1.root', 85.656)
c.set_tree_path(treePath)
c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
