c.multiclass = True
c.dy_label = "ZJets_aMC"
c.tt_label = "tt_ll_POW"
c.ggh_label = "H2Mu_gg"
c.vbf_label = "H2Mu_VBF"
##################### Input samples #######################
c.add_data(SingleMu2017B.name, SingleMu2017B.path, SingleMu2017B.lumi)
c.add_data(SingleMu2017C.name, SingleMu2017C.path, SingleMu2017C.lumi)
c.add_data(SingleMu2017D.name, SingleMu2017D.path, SingleMu2017D.lumi)
c.add_data(SingleMu2017E.name, SingleMu2017E.path, SingleMu2017E.lumi)
c.add_data(SingleMu2017F.name, SingleMu2017F.path, SingleMu2017F.lumi)
c.add_category(c.ggh_label, True)
c.add_dir_to_category(ggH_2017_powheg.name, ggH_2017_powheg.path,
ggH_2017_powheg.xSec, c.ggh_label)
c.add_category(c.vbf_label, True)
c.add_dir_to_category(VBF_2017_powheg.name, VBF_2017_powheg.path,
VBF_2017_powheg.xSec, c.vbf_label)
c.add_category(c.dy_label, False)
c.add_dir_to_category(ZJets_aMC_2017_hiStat.name, ZJets_aMC_2017_hiStat.path,
ZJets_aMC_2017_hiStat.xSec, c.dy_label)
c.add_category(c.tt_label, False)
c.add_dir_to_category(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
tt_ll_POW_2017.xSec, c.tt_label)
c.set_year("2017")
c.multiclass = False
c.sig_label = "sig"
c.bkg_label = "bkg"
##################### Input samples #######################
c.add_data(SingleMu2017B.name, SingleMu2017B.path, SingleMu2017B.lumi)
c.add_data(SingleMu2017C.name, SingleMu2017C.path, SingleMu2017C.lumi)
c.add_data(SingleMu2017D.name, SingleMu2017D.path, SingleMu2017D.lumi)
c.add_data(SingleMu2017E.name, SingleMu2017E.path, SingleMu2017E.lumi)
c.add_data(SingleMu2017F.name, SingleMu2017F.path, SingleMu2017F.lumi)
c.add_category(c.sig_label, True)
c.add_dir_to_category(ggH_2017_powheg.name, ggH_2017_powheg.path,
ggH_2017_powheg.xSec, c.sig_label)
c.add_dir_to_category(VBF_2017_powheg.name, VBF_2017_powheg.path,
VBF_2017_powheg.xSec, c.sig_label)
c.add_category(c.bkg_label, False)
c.add_dir_to_category(ZJets_aMC_2017.name, ZJets_aMC_2017.path,
ZJets_aMC_2017.xSec, c.bkg_label)
c.add_dir_to_category(tt_ll_POW_2017.name, tt_ll_POW_2017.path,
tt_ll_POW_2017.xSec, c.bkg_label)
##########################################################
### ------ Raffaele's variables ------ ###
c.add_variable("muPairs.pt", 1)
print comment
treePath = 'dimuons/tree'
# path = '/tmp/Hmumu_ntuples/updated/'
path = '/scratch/browngpu/dkondra/2016_ntuples_updated/'
c.add_data('SingleMu_2016B' , path+'/SingleMu_2016B/' , 5788)
c.add_data('SingleMu_2016C' , path+'/SingleMu_2016C/' , 2573)
c.add_data('SingleMu_2016D' , path+'/SingleMu_2016D/' , 4248)
c.add_data('SingleMu_2016E' , path+'/SingleMu_2016E/' , 4009)
c.add_data('SingleMu_2016F_1' , path+'/SingleMu_2016F_1/' , 3102)
c.add_data('SingleMu_2016G' , path+'/SingleMu_2016G/' , 7540)
c.add_data('SingleMu_2016H_1' , path+'/SingleMu_2016H_1/' , 8392)
c.add_data('SingleMu_2016H_2' , path+'/SingleMu_2016H_2/' , 214)
c.add_category('H2Mu_gg', True)
c.add_dir_to_category('H2Mu_gg', path+'/H2Mu_ggH/', 0.009618, 'H2Mu_gg')
c.add_category('H2Mu_VBF', True)
c.add_dir_to_category('H2Mu_VBF', path+'/H2Mu_VBF/', 0.0008208, 'H2Mu_VBF')
c.add_category('ZJets_MG', False)
c.add_dir_to_category('ZJets_MG', path+'/ZJets_MG/', 5765.4, 'ZJets_MG')
c.add_category('tt_ll_AMC', False)
c.add_dir_to_category('tt_ll_AMC', path+'/tt_ll_AMC/', 85.656*0.9, 'tt_ll_AMC')
c.set_tree_path(treePath)
c.add_variable("muPairs.pt", 1)
c.add_variable("muPairs.eta", 1)
# path = '/tmp/Hmumu_ntuples/updated/'
# path = '/scratch/browngpu/dkondra/2016_ntuples_updated/'
path = '/scratch/gilbreth/dkondra/'
# path = "/mnt/hadoop/store/user/dkondrat/"
# c.add_data('SingleMu_2016B' , path+'/SingleMu_2016B/' , 5788)
# c.add_data('SingleMu_2016C' , path+'/SingleMu_2016C/' , 2573)
# c.add_data('SingleMu_2016D' , path+'/SingleMu_2016D/' , 4248)
# c.add_data('SingleMu_2016E' , path+'/SingleMu_2016E/' , 4009)
# c.add_data('SingleMu_2016F_1' , path+'/SingleMu_2016F_1/' , 3102)
# c.add_data('SingleMu_2016G' , path+'/SingleMu_2016G/' , 7540)
# c.add_data('SingleMu_2016H_1' , path+'/SingleMu_2016H_1/' , 8392)
# c.add_data('SingleMu_2016H_2' , path+'/SingleMu_2016H_2/' , 214)
c.add_category('H2Mu_gg', True)
c.add_dir_to_category(
'H2Mu_gg', path +
'/GluGluHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_gg_powheg/190323_193526/0000/',
0.009618, 'H2Mu_gg')
c.add_category('H2Mu_VBF', True)
c.add_dir_to_category(
'H2Mu_VBF', path +
'/VBFHToMuMu_M-125_TuneCP5_PSweights_13TeV_powheg_pythia8/H2Mu_VBF_powheg/190322_195711/0000/',
0.0008208, 'H2Mu_VBF')
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/',