def getSample(pd, runName, lumi): sample = Sample.fromDirectory(name=(pd + '_' + runName), treeName="Events", texName=(pd + ' (' + runName + ')'), directory=dirs[pd + '_' + runName]) sample.lumi = lumi return sample
def getSample(base, pd, runName, lumi): sample = Sample.fromDirectory(name=(pd + '_' + runName), treeName="Events", texName=(pd + ' (' + runName + ')'), directory=map( lambda d: os.path.join(base, d), dirs[pd + '_' + runName])) sample.lumi = lumi return sample
def getDataSample(pd, runName, lumi, dirs, redirector=None, fromDPM=True): if fromDPM: sample = Sample.fromDPMDirectory(name=(pd + '_' + runName), treeName="Events", redirector=redirector, texName=(pd + ' (' + runName + ')'), directory=dirs[pd + '_' + runName], noCheckProxy=True) else: sample = Sample.fromDirectory(name=(pd + '_' + runName), treeName="Events", texName=(pd + ' (' + runName + ')'), directory=dirs[pd + '_' + runName]) sample.lumi = lumi return sample
def getMCSample(name, texName, directory, redirector=None, color=None, noCheckProxy=True, fromDPM=True): if fromDPM: sample = Sample.fromDPMDirectory(name=name, isData=False, color=color, treeName="Events", redirector=redirector, texName=texName, directory=directory, noCheckProxy=noCheckProxy) else: sample = Sample.fromDirectory(name=name, isData=False, color=color, treeName="Events", texName=texName, directory=directory) return sample
#dirs['QCD_EM+bcToE'] = ["QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_EM+bcToE'] = ["QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE']= ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] #dirs['QCD'] = ["QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200"] #dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] #dirs['WGToLNuG'] = ["WGToLNuG"] dirs['ZGTo2LG'] = ["ZGTo2LG"] #dirs['WGJets'] = ["WGJets"] dirs['ZGJets'] = ["ZGJets"] dirs['ZG'] = dirs['ZGTo2LG'] + dirs['ZGJets'] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY", directory=directories['DY']) DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO,HT)", directory=directories['DY_HT_LO']) TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets']) TTJets_Lep = Sample.fromDirectory(name="TTJets_Lep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}(lep)", directory=directories['TTJets_Lep']) TTJets_Singlelep= Sample.fromDirectory(name="TTJets_Singlelep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets_Singlelep']) TTJets_Singlelep_singleTop = Sample.fromDirectory(name="TTJets_Singlelep_singleTop", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/t (1l)", directory=directories['TTJets_Singlelep']+directories['singleTop']) TTJets_Dilep = Sample.fromDirectory(name="TTJets_Dilep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets_Dilep']) #TTJets_LO = Sample.fromDirectory(name="TTJets_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (LO)", directory=directories['TTJets_LO']) TTLep_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep,pow)", directory=directories['TTLep_pow']) TT_pow = Sample.fromDirectory(name="TT_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (powheg)", directory=directories['TT_pow']) Top = Sample.fromDirectory(name="Top", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t", directory=directories['Top']) Top_pow = Sample.fromDirectory(name="Top_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t(powHeg)", directory=directories['Top_pow']) Top_pow_incl = Sample.fromDirectory(name="Top_pow_incl", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t(pow incl)", directory=directories['Top_pow_incl']) Top_amc = Sample.fromDirectory(name="Top_amc", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t(amc@NLO)", directory=directories['Top_amc'])
import copy, os, sys from RootTools.core.Sample import Sample import ROOT from StopsDilepton.tools.user import data_directory from StopsDilepton.samples.color import color # Take post processing directory if defined in main module try: import sys postProcessing_directory = sys.modules['__main__'].postProcessing_directory except: postProcessing_directory = "postProcessed_Fall15_mAODv2/gen" dirs = {} dirs['TTZtoLLNuNu'] = ["TTZToLLNuNu"] dirs['TTGJets'] = ["TTGJets"] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} TTZtoLLNuNu = Sample.fromDirectory(name="TTZtoNuNu", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) TTG = Sample.fromDirectory(name="TTGJets", treeName="Events", isData=False, color=color.TTG, texName="t#bar{t}#gamma + Jets", directory=directories['TTGJets'])
] directories = { key: [ os.path.join(data_directory_, postProcessing_directory_, dir) for dir in dirs[key] ] for key in dirs.keys() } # # FullSim signals DisplacedStops_mStop_250_ctau_0p01 = Sample.fromDirectory( name="DisplacedStops_mStop_250_ctau_0p01", treeName="Events", isData=False, color=color.DY, texName="mStop250ctau0p01", directory=directories['DisplacedStops_mStop_250_ctau_0p01']) signals_displ = [ DisplacedStops_mStop_250_ctau_0p01, ] for s in signals_displ: #t1, mStop, t2, ctau = s.name.replace('Displaced_','').split('_') #s.mStop = int(mStop) #s.mNeu = int(ctau) s.isFastSim = True
#dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE']= ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] #dirs['QCD'] = ["QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200"] #dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] #dirs['WGToLNuG'] = ["WGToLNuG"] #dirs['ZGTo2LG'] = ["ZGTo2LG_ext"] #dirs['WGJets'] = ["WGJets"] #dirs['ZGJets'] = ["ZGJets"] #dirs['ZG'] = dirs['ZGTo2LG'] + dirs['ZGJets'] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} #DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY", directory=directories['DY']) #DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO) HT binned", directory=directories['DY_HT_LO']) DYnJets = Sample.fromDirectory(name="DYnJets", treeName="Events", isData=False, color=color.DY, texName="DY (LO) 1-4 Jets", directory=directories['DYnJets']) ##TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets']) #TTJets_Lep = Sample.fromDirectory(name="TTJets_Lep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}(lep)", directory=directories['TTJets_Lep']) #TTJets_Singlelep= Sample.fromDirectory(name="TTJets_Singlelep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets_Singlelep']) ##TTJets_Singlelep_singleTop = Sample.fromDirectory(name="TTJets_Singlelep_singleTop", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/t (1l)", directory=directories['TTJets_Singlelep']+directories['singleTop']) #TTJets_Dilep = Sample.fromDirectory(name="TTJets_Dilep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTJets_Dilep']) ##TTJets_LO = Sample.fromDirectory(name="TTJets_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (LO)", directory=directories['TTJets_LO']) TTLep_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep,pow)", directory=directories['TTLep_pow']) ##TT_pow = Sample.fromDirectory(name="TT_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (powheg)", directory=directories['TT_pow']) #FIXME dilep #Top = Sample.fromDirectory(name="Top", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t", directory=directories['Top']) #Top_pow = Sample.fromDirectory(name="Top_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t", directory=directories['Top_pow']) ##Top_pow_incl = Sample.fromDirectory(name="Top_pow_incl", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t(pow incl)", directory=directories['Top_pow_incl']) #FIXME dilep ##Top_amc = Sample.fromDirectory(name="Top_amc", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t(amc@NLO)", directory=directories['Top_amc']) # # ##TTJets_HT_LO = Sample.fromDirectory(name="TTJets_HT_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (HT,LO)", directory=directories['TTJets_HT_LO'])
dirs['ZZInclusive'] = ["ZZ"] dirs['ZZ'] = ["ZZTo2L2Q", "ZZTo2Q2Nu"] dirs['WZ'] = ["WZTo1L1Nu2Q", "WZTo2L2Q", "WZTo3LNu"] #dirs['triBoson'] = ["WWZ","WZZ","ZZZ"] # No Fall15 production for WWZ and ZZZ? dirs['WZZ'] = ["WZZ"] #dirs['QCD_HT'] = ["QCD_HT100to200", "QCD_HT200to300", "QCD_HT300to500", "QCD_HT500to700", "QCD_HT700to1000", "QCD_HT1000to1500", "QCD_HT1500to2000", "QCD_HT2000toInf"] dirs['QCD_Mu5'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5"] dirs['QCD_EM+bcToE'] = ["QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_EM+bcToE'] = ["QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] #dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY + Jets", directory=directories['DY']) #DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO)", directory=directories['DY_LO']) DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO,HT)", directory=directories['DY_HT_LO']) TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets", directory=directories['TTJets']) #TTJets_LO = Sample.fromDirectory(name="TTJets_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (LO)", directory=directories['TTJets_LO']) #TTLep_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep,pow)", directory=directories['TTLep_pow']) TTJets_Lep = Sample.fromDirectory(name="TTJets_Lep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep)", directory=directories['TTJets_Lep']) TTJets_Dilep = Sample.fromDirectory(name="TTJets_Dilep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (2 lep)", directory=directories['TTJets_Dilep']) TTJets_Singlelep= Sample.fromDirectory(name="TTJets_Singlelep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (1 lep)", directory=directories['TTJets_Singlelep']) #TTJets_HT_LO = Sample.fromDirectory(name="TTJets_HT_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (HT,LO)", directory=directories['TTJets_HT_LO']) singleTop = Sample.fromDirectory(name="singleTop", treeName="Events", isData=False, color=color.singleTop, texName="single top", directory=directories['singleTop']) TTX = Sample.fromDirectory(name="TTX", treeName="Events", isData=False, color=color.TTX, texName="t#bar{t}X", directory=directories['TTX']) TTXNoZ = Sample.fromDirectory(name="TTXNoZ", treeName="Events", isData=False, color=color.TTXNoZ, texName="t#bar{t}H/W, tZq", directory=directories['TTXNoZ']) TTH = Sample.fromDirectory(name="TTH", treeName="Events", isData=False, color=color.TTH, texName="t#bar{t}H", directory=directories['TTH']) TTW = Sample.fromDirectory(name="TTW", treeName="Events", isData=False, color=color.TTW, texName="t#bar{t}W", directory=directories['TTW']) TTZ = Sample.fromDirectory(name="TTZ", treeName="Events", isData=False, color=color.TTZ, texName="t#bar{t}Z", directory=directories['TTZ'])
dirs['TTGJets'] = ["TTGJets"] dirs['ZG'] = ["ZGTo2LG"] dirs['WG'] = ["WGToLNuG"] dirs['WWG'] = ["WWG"] directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } TTG = Sample.fromDirectory(name="TTGJets", treeName="Events", isData=False, color=color.TTG, texName="t#bar{t}#gamma + Jets", directory=directories['TTGJets']) ZG = Sample.fromDirectory(name="ZG", treeName="Events", isData=False, color=color.WZ, texName="ZG", directory=directories['ZG']) WG = Sample.fromDirectory(name="WG", treeName="Events", isData=False, color=color.WG, texName="WG", directory=directories['WG']) WWG = Sample.fromDirectory(name="WWG",
#dirs['pseudoDataPriv'] = dirs['ewkDM_ttZ_ll_noH'] + dirs["WZ"] + dirs['TTW'] + dirs['TTX'] + dirs['TZQ'] + dirs['rare'] + dirs['nonprompt'] dirs['background'] = dirs["WZ_amcatnlo"] + dirs['TTW'] + dirs['TTX'] + dirs[ 'TZQ'] + dirs['rare'] directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } pseudoData = Sample.fromDirectory(name="pseudoData", treeName="Events", isData=False, color=ROOT.kBlack, texName="pseudo-data", directory=directories['pseudoData']) #pseudoDataPriv = Sample.fromDirectory(name="pseudoDataPriv", treeName="Events", isData=False, color=ROOT.kBlack, texName="pseudo-data", directory=directories['pseudoDataPriv']) TTZtoLLNuNu = Sample.fromDirectory(name="TTZtoLLNuNu", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (l#bar{l}/#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) #TTZ_LO = Sample.fromDirectory(name="TTZ_LO", treeName="Events", isData=False, color=color.TTZtoLLNuNu+1, texName="t#bar{t}Z (LO)", directory=directories['TTZ_LO']) TTG = Sample.fromDirectory(name="TTG", treeName="Events", isData=False, color=color.TTG_signal, texName="t#bar{t}#gamma",
#ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2A_m0p25 = Sample.fromDirectory(name="ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2A_m0p25", treeName="Events", isData=False, color=1, texName="C_{1A/V}=0, C_{2A}=-0.25, C_{2V}=0", directory=directories['ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2A_m0p25']) #ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_0p25 = Sample.fromDirectory(name="ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_0p25", treeName="Events", isData=False, color=1, texName="C_{1A/V}=0, C_{2A}=0, C_{2V}=0.25", directory=directories['ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_0p25']) #ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_m0p25 = Sample.fromDirectory(name="ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_m0p25", treeName="Events", isData=False, color=1, texName="C_{1A/V}=0, C_{2A}=-0.25, C_{2V}=0", directory=directories['ttZ0j_ll_DC1A_0p60_DC1V_m0p24_DC2V_m0p25']) # #ttZ0j_ll_noH = Sample.fromDirectory(name="ttZ0j_ll_noH", treeName="Events", isData=False, color=1, texName="#DeltaC_{1,2;V,A}=0", directory=directories['ttZ0j_ll_noH']) #ttZ0j_ll_noH_DC2V_0p05 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_0p05", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=0.05", directory=directories['ttZ0j_ll_noH_DC2V_0p05']) #ttZ0j_ll_noH_DC2V_0p10 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_0p10", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=0.10", directory=directories['ttZ0j_ll_noH_DC2V_0p10']) #ttZ0j_ll_noH_DC2V_0p20 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_0p20", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=0.20", directory=directories['ttZ0j_ll_noH_DC2V_0p20']) #ttZ0j_ll_noH_DC2V_0p30 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_0p30", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=0.30", directory=directories['ttZ0j_ll_noH_DC2V_0p30']) #ttZ0j_ll_noH_DC2V_m0p15 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_m0p15", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=-0.15", directory=directories['ttZ0j_ll_noH_DC2V_m0p15']) #ttZ0j_ll_noH_DC2V_m0p25 = Sample.fromDirectory(name="ttZ0j_ll_noH_DC2V_m0p25", treeName="Events", isData=False, color=1, texName="#DeltaC_{1;V,A}=0, C_{2V}=-0.25", directory=directories['ttZ0j_ll_noH_DC2V_m0p25']) ewkDM_TTZToLL_LO = Sample.fromDirectory( name="ewkDM_TTZToLL_LO", treeName="Events", isData=False, color=1, texName="#DeltaC_{1,2;V,A}=0", directory=directories['ewkDM_TTZToLL_LO']) ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2 = Sample.fromDirectory( name="ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2", treeName="Events", isData=False, color=1, texName="C_{2A}=0.2, C_{2V}=0.2", directory=directories['ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2']) allSignals = [\ # ttZ0j_ll, # ttZ0j_ll_DC2A_0p20_DC2V_0p20, # ttZ0j_ll_DC1A_0p50_DC1V_0p50,
##dirs['QCD_Mu5+EM+bcToE']= ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] ##dirs['QCD_Mu5+EM+bcToE']= ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] ##dirs['QCD'] = ["QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200"] ##dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] directories = { key: [ os.path.join(data_directory_, postProcessing_directory_, dir) for dir in dirs[key] ] for key in dirs.keys() } DY_HT_LO_16 = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="Drell-Yan", directory=directories['DY_HT_LO']) Top_pow_16 = Sample.fromDirectory(name="Top_pow", treeName="Events", isData=False, color=color.Top_pow, texName="t#bar{t}/t", directory=directories['Top_pow']) TTLep_pow_16 = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTLep_pow']) TTSingleLep_pow_16 = Sample.fromDirectory(
except: postProcessing_directory = "postProcessed_Fall15_mAODv2/dilepTiny" logger.info("Loading data samples from directory %s", os.path.join(data_directory, postProcessing_directory)) dirs = {} dirs['DoubleEG'] = ["DoubleEG_Run2015D_16Dec"] dirs['DoubleMuon'] = ["DoubleMuon_Run2015D_16Dec"] dirs['MuonEG'] = ["MuonEG_Run2015D_16Dec"] dirs['SingleElectron'] = ["SingleElectron_Run2015D_16Dec"] dirs['SingleMuon'] = ["SingleMuon_Run2015D_16Dec"] for key in dirs: dirs[key] = [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] DoubleEG_Run2015D = Sample.fromDirectory(name="DoubleEG_Run2015D", treeName="Events", texName="DoubleEG (Run2015D)", directory=dirs["DoubleEG"]) DoubleMuon_Run2015D = Sample.fromDirectory(name="DoubleMuon_Run2015D", treeName="Events", texName="DoubleMuon (Run2015D)", directory=dirs["DoubleMuon"]) MuonEG_Run2015D = Sample.fromDirectory(name="MuonEG_Run2015D", treeName="Events", texName="MuonEG (Run2015D)", directory=dirs["MuonEG"]) SingleElectron_Run2015D = Sample.fromDirectory(name="SingleElectron_Run2015D", treeName="Events", texName="SingleElectron (Run2015D)", directory=dirs["SingleElectron"]) SingleMuon_Run2015D = Sample.fromDirectory(name="SingleMuon_Run2015D", treeName="Events", texName="SingleMuon (Run2015D)", directory=dirs["SingleMuon"]) DoubleEG_Run2015D .lumi = 1000*(2.165) DoubleMuon_Run2015D .lumi = 1000*(2.165) MuonEG_Run2015D .lumi = 1000*(2.137) SingleElectron_Run2015D.lumi = 1000*(2.165) SingleMuon_Run2015D .lumi = 1000*(2.165) allSamples_Data25ns = [DoubleEG_Run2015D, MuonEG_Run2015D, DoubleMuon_Run2015D, SingleElectron_Run2015D, SingleMuon_Run2015D] for s in allSamples_Data25ns: s.color = ROOT.kBlack s.isData = True
# RootTools from RootTools.core.Sample import Sample jets = Sample.fromDirectory("jets", "/data/rschoefbeck/JetMET/localJEC/jetTrees/", treeName="jets") maxN = -1 #qcd_AllChGood_Flat0to50 = Sample.fromCMGOutput("qcd_AllChGood_Flat0to50", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AllChlgoodAsymptFlat0to50bx25_AllChannelsGood_v0-v2_RunIISpring15MiniAODv2-74X", treeFilename='tree.root', maxN=maxN) #qcd_Flat0to50 = Sample.fromCMGOutput("qcd_Flat0to50", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AsymptFlat0to50bx25Reco_MCRUN2_74_V9-v3_RunIISpring15MiniAODv2-74X_3", treeFilename='tree.root', maxN=maxN) qcd_AllChGood_noPU = Sample.fromCMGOutput( "qcd_AllChGood_noPU", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AllChlgoodAsymptNoPUbx25_AllChannelsGood_v0-v2_RunIISpring15MiniAODv2-74X", treeFilename='tree.root', maxN=maxN) qcd_noPU = Sample.fromCMGOutput( "qcd_noPU", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AsymptNoPUbx25Reco_MCRUN2_74_V9-v3_RunIISpring15MiniAODv2-74X_3", treeFilename='tree.root', maxN=maxN) #Flag_HBHENoiseIsoFilter #Flag_EcalDeadCellTriggerPrimitiveFilter #Flag_trkPOG_manystripclus53X #Flag_ecalLaserCorrFilter #Flag_trkPOG_toomanystripclus53X #Flag_hcalLaserEventFilter #Flag_trkPOG_logErrorTooManyClusters #Flag_trkPOGFilters #Flag_trackingFailureFilter #Flag_CSCTightHaloFilter
import copy, os, sys from RootTools.core.Sample import Sample import ROOT from StopsDilepton.tools.user import data_directory from StopsDilepton.samples.color import color # Take post processing directory if defined in main module try: import sys postProcessing_directory = sys.modules['__main__'].postProcessing_directory except: postProcessing_directory = "postProcessed_Fall15_mAODv2/dilepTiny" dirs = {} dirs['TTGJets'] = ["TTGJets"] dirs['ZG'] = ["ZGTo2LG"] dirs['WG'] = ["WGToLNuG"] dirs['WWG'] = ["WWG"] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} TTG = Sample.fromDirectory(name="TTGJets", treeName="Events", isData=False, color=color.TTG, texName="t#bar{t}#gamma + Jets", directory=directories['TTGJets']) ZG = Sample.fromDirectory(name="ZG", treeName="Events", isData=False, color=color.WZ, texName="ZG", directory=directories['ZG']) WG = Sample.fromDirectory(name="WG", treeName="Events", isData=False, color=color.WG, texName="WG", directory=directories['WG']) WWG = Sample.fromDirectory(name="WWG", treeName="Events", isData=False, color=color.WWG, texName="WWG", directory=directories['WWG'])
#stg_sch_4WC_ref.reweight_pkl = os.path.join( gridpack_directory, "EFT", "dipoles", "medRefdim3", "stg_sch_rwgt", "stg_sch_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" ) #TT_4WC_ref = Sample.fromDirectory( name="TT_4WC_ref", treeName="Events", isData=False, color=color.TT, texName="t#bar{t}", directory=directories["tt01j_4WC"]) #TT_4WC_ref.reweight_pkl = os.path.join( gridpack_directory, "EFT", "dipoles", "medRefdim3", "tt01j_SemiLept_restrict_rwgt", "tt01j_SemiLept_restrict_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" ) #TT_SemiLep_4WC_ref = Sample.fromDirectory( name="TT_SemiLep_4WC_ref", treeName="Events", isData=False, color=color.TT, texName="t#bar{t}", directory=directories["tt01j_1l_4WC"]) #TT_SemiLep_4WC_ref.reweight_pkl = os.path.join( gridpack_directory, "EFT", "dipoles", "medRefdim3", "tt01j_SemiLept_restrict_rwgt", "tt01j_SemiLept_restrict_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" ) #TT_Dilep_4WC_ref = Sample.fromDirectory( name="TT_Dilep_4WC_ref", treeName="Events", isData=False, color=color.TT, texName="t#bar{t}", directory=directories["tt01j_2l_4WC"]) #TT_Dilep_4WC_ref.reweight_pkl = os.path.join( gridpack_directory, "EFT", "dipoles", "medRefdim3", "tt01j_Dilept_restrict_rwgt", "tt01j_Dilept_restrict_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" ) #ctZ TTG_2WC_ref = Sample.fromDirectory(name="TTG_2WC_ref", treeName="Events", isData=False, color=color.TTG, texName="tt#gamma", directory=directories["ttGamma_2WC"]) TTG_2WC_ref.reweight_pkl = os.path.join( gridpack_directory, "EFT", "dipoles", "singleOp", "ttGamma_SemiLept_restrict_rwgt_2WC", "ttGamma_SemiLept_restrict_rwgt_2WC_slc7_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" ) TTG_SemiLep_2WC_ref = Sample.fromDirectory( name="TTG_SemiLep_2WC_ref", treeName="Events", isData=False, color=color.TTG, texName="tt#gamma", directory=directories["ttGamma_1l_2WC"])
#dirs['pseudoData'] = dirs['TTZtoLLNuNu'] + dirs["WZ"] + dirs['TTW'] + dirs['TTX'] + dirs['TZQ'] + dirs['rare'] + dirs['nonprompt'] dirs['rare'] = ["WW", "ZZ"] # almost everything missing directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } TTZtoLLNuNu_17 = Sample.fromDirectory( name="TTZtoLLNuNu_17", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (l#bar{l}/#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) WW_17 = Sample.fromDirectory(name="WW_17", treeName="Events", isData=False, color=color.WW, texName="WW", directory=directories['WW']) WZ_17 = Sample.fromDirectory(name="WZ_17", treeName="Events", isData=False, color=color.WZ, texName="WZ", directory=directories['WZ'])
# Directories dirs = {} dirs["tWZ0j_rwgt_dim6top_GEN"] = ["tWZ0j_rwgt_dim6top"] dirs["tWZ01j_rwgt_dim6top_GEN"] = ["tWZ01j_rwgt_dim6top"] dirs["tWZ01j_rwgt_filter_yt_GEN_SIM"] = ["tWZ01j_rwgt_filter_2"] directories = { key: [ os.path.join(data_directory, postprocessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } #tWZ0j_rwgt_dim6top_GEN = Sample.fromDirectory(name="tWZ0j_rwgt_dim6top_GEN", treeName="Events", isData=False, color=color.VG1, texName="tWZ0j", directory=directories["tWZ0j_rwgt_dim6top_GEN"]) #tWZ0j_rwgt_dim6top_GEN.reweight_pkl = "/afs/hephy.at/data/rschoefbeck01/gridpacks/dim6top/tWZ0j_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" #tWZ01j_rwgt_dim6top_GEN = Sample.fromDirectory(name="tWZ01j_rwgt_dim6top_GEN", treeName="Events", isData=False, color=color.VG1, texName="tWZ01j", directory=directories["tWZ01j_rwgt_dim6top_GEN"]) #tWZ01j_rwgt_dim6top_GEN.reweight_pkl = "/afs/hephy.at/data/rschoefbeck01/gridpacks/dim6top/tWZ01j_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl" tWZ01j_rwgt_filter_yt_GEN_SIM = Sample.fromDirectory( name="tWZ01j_rwgt_filter_yt_GEN_SIM", treeName="Events", isData=False, color=color.VG1, texName="tWZ01j", directory=directories["tWZ01j_rwgt_filter_yt_GEN_SIM"]) tWZ01j_rwgt_filter_yt_GEN_SIM.reweight_pkl = "/afs/hephy.at/data/cms04/ttschida/gridpacks/Yt/tWZ01j_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl"
dirs['WZ'] = ["WZTo1L1Nu2Q", "WZTo2L2Q", "WZTo3LNu"] #dirs['triBoson'] = ["WWZ","WZZ","ZZZ"] # No Fall15 production for WWZ and ZZZ? dirs['WZZ'] = ["WZZ"] #dirs['QCD_HT'] = ["QCD_HT100to200", "QCD_HT200to300", "QCD_HT300to500", "QCD_HT500to700", "QCD_HT700to1000", "QCD_HT1000to1500", "QCD_HT1500to2000", "QCD_HT2000toInf"] #dirs['QCD_Mu5'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5"] dirs['QCD_Mu5'] = [ "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5"] #dirs['QCD_EM+bcToE'] = ["QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] dirs['QCD_EM+bcToE'] = ["QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} from StopsDilepton.samples.color import color DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY + Jets", directory=directories['DY']) #DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO)", directory=directories['DY_LO']) #DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO,HT)", directory=directories['DY_HT_LO']) TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets", directory=directories['TTJets']) #TTJets_LO = Sample.fromDirectory(name="TTJets_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (LO)", directory=directories['TTJets_LO']) #TTLep_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep,pow)", directory=directories['TTLep_pow']) TTJets_Lep = Sample.fromDirectory(name="TTJets_Lep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep)", directory=directories['TTJets_Lep']) #TTJets_HT_LO = Sample.fromDirectory(name="TTJets_HT_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (HT,LO)", directory=directories['TTJets_HT_LO']) singleTop = Sample.fromDirectory(name="singleTop", treeName="Events", isData=False, color=color.singleTop, texName="single top", directory=directories['singleTop']) #TTX = Sample.fromDirectory(name="TTX", treeName="Events", isData=False, color=color.TTX, texName="t#bar{t}X", directory=directories['TTX']) #TTXNoZ = Sample.fromDirectory(name="TTXNoZ", treeName="Events", isData=False, color=color.TTXNoZ, texName="t#bar{t}H/W, tZq", directory=directories['TTXNoZ']) #TTH = Sample.fromDirectory(name="TTH", treeName="Events", isData=False, color=color.TTH, texName="t#bar{t}H", directory=directories['TTH']) #TTW = Sample.fromDirectory(name="TTW", treeName="Events", isData=False, color=color.TTW, texName="t#bar{t}W", directory=directories['TTW']) #TTZ = Sample.fromDirectory(name="TTZ", treeName="Events", isData=False, color=color.TTZ, texName="t#bar{t}Z", directory=directories['TTZ']) #TTZtoLLNuNu = Sample.fromDirectory(name="TTZtoNuNu", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (l#bar{l}/#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) #TTZtoQQ = Sample.fromDirectory(name="TTZtoQQ", treeName="Events", isData=False, color=color.TTZtoQQ, texName="t#bar{t}Z (q#bar{q})", directory=directories['TTZtoQQ'])
] directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } #ewkDM_TTZToLL_LO = Sample.fromDirectory(name="ewkDM_TTZToLL_LO", treeName="Events", isData=False, color=1, texName="#DeltaC_{1,2;V,A}=0", directory=directories['ewkDM_TTZToLL_LO']) #ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2 = Sample.fromDirectory(name="ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2", treeName="Events", isData=False, color=1, texName="C_{2A}=0.2, C_{2V}=0.2", directory=directories['ewkDM_TTZToLL_LO_DC2A0p2_DC2V0p2']) ttZ0j_ll = Sample.fromDirectory(name="ttZ0j_ll", treeName="Events", isData=False, color=1, texName="SM", directory=directories['ttZ0j_ll']) #HEL_UFO #ttZ0j_ll_cuW_0p100000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_0p100000", treeName="Events", isData=False, color=1, texName="c_{uW}=0.1", directory=directories['ttZ0j_ll_cuW_0p100000']) #ttZ0j_ll_cuW_0p200000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_0p200000", treeName="Events", isData=False, color=1, texName="c_{uW}=0.2", directory=directories['ttZ0j_ll_cuW_0p200000']) #ttZ0j_ll_cuW_0p300000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_0p300000", treeName="Events", isData=False, color=1, texName="c_{uW}=0.3", directory=directories['ttZ0j_ll_cuW_0p300000']) #ttZ0j_ll_cuW_m0p100000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_m0p100000", treeName="Events", isData=False, color=1, texName="c_{uW}=0.1", directory=directories['ttZ0j_ll_cuW_m0p100000']) #ttZ0j_ll_cuW_m0p200000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_m0p200000", treeName="Events", isData=False, color=1, texName="c_{uW}=-0.2", directory=directories['ttZ0j_ll_cuW_m0p200000']) #ttZ0j_ll_cuW_m0p300000 = Sample.fromDirectory(name="ttZ0j_ll_cuW_m0p300000", treeName="Events", isData=False, color=1, texName="c_{uW}=-0.3", directory=directories['ttZ0j_ll_cuW_m0p300000']) # Markus' standard point ttZ0j_ll_DC2A_0p200000_DC2V_0p200000 = Sample.fromDirectory( name="ttZ0j_ll_DC2A_0p200000_DC2V_0p200000", treeName="Events",
#dirs['other'] += dirs['VV'] dirs['other'] += dirs['WW'] + dirs['WZ'] + dirs['ZZ'] dirs['other'] += dirs['GluGlu'] directories = { key: [ os.path.join(data_directory, postprocessing_directory2017, dir) for dir in dirs[key] ] for key in dirs.keys() } # Samples DY_LO_17 = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) TT_pow_17 = Sample.fromDirectory(name="TT_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TT_pow']) #TTbar_17 = Sample.fromDirectory(name="TTbar", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTbar']) singleTop_17 = Sample.fromDirectory(name="singleTop", treeName="Events", isData=False, color=color.singleTop, texName="single-t", directory=directories['singleTop'])
"QCD_Pt170to300_EMEnriched" ] #dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] #dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY + Jets", directory=directories['DY']) #DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO)", directory=directories['DY_LO']) DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO,HT)", directory=directories['DY_HT_LO']) TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets", directory=directories['TTJets'])
#dirs['allMC'] = dirs['DY_LO'] + dirs['Top_pow'] + dirs['TTZ'] + dirs['TTXNoZ'] + dirs['multiBoson'] directories = { key: [ os.path.join(data_directory_, postProcessing_directory_, dir) for dir in dirs[key] ] for key in dirs.keys() } #DY_HT_LO_18 = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY (HT, LO)", directory=directories['DY_HT_LO']) #DY_LO_18 = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) #DY_LO_M50_18 = Sample.fromDirectory(name="DY_LO_M50", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO_M50']) Top_pow_18 = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}/single-t", directory=directories['TTLep_pow']) Top_pow_1l_18 = Sample.fromDirectory(name="Top_pow_1l", treeName="Events", isData=False, color=color.TTJets_1l, texName="t#bar{t} (1l)", directory=directories['Top_pow_1l']) WJets_18 = Sample.fromDirectory(name="WJets", treeName="Events", isData=False, color=color.WJetsToLNu, texName="W+Jets", directory=directories['WJets']) #TTXNoZ_18 = Sample.fromDirectory(name="TTXNoZ", treeName="Events", isData=False, color=color.TTXNoZ, texName="t#bar{t}H/W, tZq", directory=directories['TTXNoZ'])
"DoubleEG_Run2016H_v2_backup"] + dirs["DoubleEG_Run2016H_v3_backup"] dirs["DoubleMuon_Run2016_backup"] = dirs[ "DoubleMuon_Run2016BCDEFG_backup"] + dirs[ "DoubleMuon_Run2016H_v2_backup"] + dirs["DoubleMuon_Run2016H_v3_backup"] dirs["MuonEG_Run2016_backup"] = dirs["MuonEG_Run2016BCDEFG_backup"] + dirs[ "MuonEG_Run2016H_v2_backup"] + dirs["MuonEG_Run2016H_v3_backup"] for key in dirs: dirs[key] = [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] DoubleEG_Run2016H_v3_backup = Sample.fromDirectory( name="DoubleEG_Run2016H_v3_backup", treeName="Events", texName="DoubleEG (Run2016H_v3)", directory=dirs["DoubleEG_Run2016H_v3_backup"]) DoubleEG_Run2016G_backup = Sample.fromDirectory( name="DoubleEG_Run2016G_backup", treeName="Events", texName="DoubleEG (Run2016G)", directory=dirs["DoubleEG_Run2016G_backup"]) DoubleEG_Run2016BCD_backup = Sample.fromDirectory( name="DoubleEG_Run2016BCD_backup", treeName="Events", texName="DoubleEG (Run2016BCD)", directory=dirs["DoubleEG_Run2016BCD_backup"]) DoubleMuon_Run2016BCD_backup = Sample.fromDirectory( name="DoubleMuon_Run2016BCD_backup",
import RootTools.core.logger as logger_rt logger_rt = logger_rt.get_logger("INFO", logFile=None) else: import logging logger = logging.getLogger(__name__) logger.info("Loading MC samples from directory %s", os.path.join(data_directory, postprocessing_directory)) # Directories dirs = {} dirs["ttW01j_rwgt_dim6top_GEN_SIM"] = ["ttW01j_rwgt_dim6top"] directories = { key: [ os.path.join(data_directory, postprocessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } ttW01j_rwgt_dim6top_GEN_SIM = Sample.fromDirectory( name="ttW01j_rwgt_dim6top_GEN_SIM", treeName="Events", isData=False, color=color.VG1, texName="ttW01j", directory=directories["ttW01j_rwgt_dim6top_GEN_SIM"]) ttW01j_rwgt_dim6top_GEN_SIM.reweight_pkl = "/afs/hephy.at/data/cms04/ttschida/gridpacks/dim6top/ttW01j_rwgt_slc6_amd64_gcc630_CMSSW_9_3_16_tarball.pkl"
def make_dirs( dirs ): return [ os.path.join( directory_, dir_ ) for dir_ in dirs ] dirs = {} # 2016 analysis ## tWZ_sample, TTZtoLLNuNu, TTX_rare_TWZ, TZQ, WZ_amcatnlo, rare, ZZ, nonprompt_TWZ_3l # diboson #dirs['diBosonInclusive'] = ["WW", "WZ", "ZZ"] # diboson exclusive #dirs['WZ'] = ["WZTo1L1Nu2Q", "WZTo1L3Nu", "WZTo2L2Q", "WZTo3LNu"] dirs['WZ'] = ["WZTo1L3Nu", "WZTo2L2Q", "WZTo3LNu_amcatnlo"] WZ = Sample.fromDirectory(name="WZ", treeName="Events", isData=False, color=color.WZ, texName="WZ", directory=make_dirs( dirs['WZ'])) dirs['ZZ'] = ["ZZTo2L2Q_redBy5", "ZZTo2Q2Nu", "ZZTo4L"] ZZ = Sample.fromDirectory(name="ZZ", treeName="Events", isData=False, color=color.ZZ, texName="ZZ", directory=make_dirs( dirs['ZZ'])) dirs['WW'] = ["WWToLNuQQ"] dirs['VVTo2L2Nu'] = ["VVTo2L2Nu"] # covered by VVTo2L2Nu: #dirs['WWTo2L2Nu'] = ['WWTo2L2Nu'] #dirs['ZZTo2L2Nu'] = ['ZZTo2L2Nu', "ZZTo2L2Q"] #dirs['ZZTo2L2Nu_2'] = ['ZZTo2L2Nu'] dirs['triBoson'] = ["WWW_4F","WWZ","WZZ","ZZZ"] triBoson = Sample.fromDirectory(name="triBoson", treeName="Events", isData=False, color=color.triBoson, texName="VVV", directory=make_dirs( dirs['triBoson']))
dirs['DY_LO'] = ["DYJetsToLL_M50_LO"] dirs['Top'] = ["TTLep_pow", "T_tWch_ext", "TBar_tWch_ext"] dirs['diboson'] = ['VVTo2L2Nu', 'WZTo3LNu', 'WZTo1L3Nu'] dirs['triboson'] = ['WWZ_4F', 'WZZ', 'ZZZ'] directories = { key: [ os.path.join(data_directory, postProcessing_directory, dir) for dir in dirs[key] ] for key in dirs.keys() } DY_LO_17 = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) Top_17 = Sample.fromDirectory(name="Top", treeName="Events", isData=False, color=color.TTJets, texName="t(#bar{t})", directory=directories['Top']) diboson_17 = Sample.fromDirectory(name="diboson", treeName="Events", isData=False, color=color.diboson, texName="diboson", directory=directories['diboson']) rare_17 = Sample.fromDirectory(name="rare",
dirs = {} # mu enriched QCD (few bins missing!) dirs['QCD_mu'] = [ "QCD_Mu_pt1000toInf_comb", "QCD_Mu_pt120to170", "QCD_Mu_pt15to20", "QCD_Mu_pt170to300_comb", "QCD_Mu_pt20to30", "QCD_Mu_pt300to470_comb", "QCD_Mu_pt30to50", "QCD_Mu_pt470to600_comb", "QCD_Mu_pt50to80", "QCD_Mu_pt600to800_comb", "QCD_Mu_pt800to1000_comb", "QCD_Mu_pt80to120_comb" ] QCD_mu = Sample.fromDirectory(name="QCD_mu", treeName="Events", isData=False, color=color.QCD, texName="QCD(#mu)", directory=make_dirs(locations.fakes_mu_2016, dirs['QCD_mu'])) # pt binned inclusive QCD (mu) #dirs['QCD_pt'] = ["QCD_pt15to30", "QCD_pt30to50", "QCD_pt50to80", "QCD_pt80to120_comb", "QCD_pt120to170_comb", "QCD_pt170to300_comb", "QCD_pt300to470_comb", "QCD_pt470to600", "QCD_pt600to800_comb", "QCD_pt1400to1800_comb", "QCD_pt1800to2400_comb", "QCD_pt2400to3200_comb",] #QCD_pt_mu = Sample.fromDirectory(name="QCD_pt_mu", treeName="Events", isData=False, color=color.QCD, texName="QCD p_{T} (#mu)", directory=make_dirs( locations.fakes_mu_2016, dirs['QCD_pt'])) # EM enriched QCD (bcToE missing!) dirs['QCD_ele'] = [ "QCD_Ele_pt20to30", "QCD_Ele_pt30to50_comb", "QCD_Ele_pt50to80_comb", "QCD_Ele_pt80to120_comb", "QCD_Ele_pt120to170_comb", "QCD_Ele_pt170to300", "QCD_Ele_pt300toInf" ] # "QCD_Ele_pt20to30"
# RootTools from RootTools.core.Sample import Sample jets = Sample.fromDirectory("jets", "/data/rschoefbeck/JetMET/localJEC/jetTrees/", treeName = "jets") maxN = -1 #qcd_AllChGood_Flat0to50 = Sample.fromCMGOutput("qcd_AllChGood_Flat0to50", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AllChlgoodAsymptFlat0to50bx25_AllChannelsGood_v0-v2_RunIISpring15MiniAODv2-74X", treeFilename='tree.root', maxN=maxN) #qcd_Flat0to50 = Sample.fromCMGOutput("qcd_Flat0to50", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AsymptFlat0to50bx25Reco_MCRUN2_74_V9-v3_RunIISpring15MiniAODv2-74X_3", treeFilename='tree.root', maxN=maxN) qcd_AllChGood_noPU = Sample.fromCMGOutput("qcd_AllChGood_noPU", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AllChlgoodAsymptNoPUbx25_AllChannelsGood_v0-v2_RunIISpring15MiniAODv2-74X", treeFilename='tree.root', maxN=maxN) qcd_noPU = Sample.fromCMGOutput("qcd_noPU", "/data/rschoefbeck/cmgTuples/MC25ns_v2_0l/QCD_Pt-15to7000_TuneCUETP8M1_Flat_13TeV_pythia8_schoef-crab_QCD_Pt-15to7000_AsymptNoPUbx25Reco_MCRUN2_74_V9-v3_RunIISpring15MiniAODv2-74X_3", treeFilename='tree.root', maxN=maxN) #Flag_HBHENoiseIsoFilter #Flag_EcalDeadCellTriggerPrimitiveFilter #Flag_trkPOG_manystripclus53X #Flag_ecalLaserCorrFilter #Flag_trkPOG_toomanystripclus53X #Flag_hcalLaserEventFilter #Flag_trkPOG_logErrorTooManyClusters #Flag_trkPOGFilters #Flag_trackingFailureFilter #Flag_CSCTightHaloFilter #Flag_HBHENoiseFilter #Flag_goodVertices #Flag_METFilters #Flag_eeBadScFilter #qcd_noPU.chain.Draw("met_rawPt>>h(100,0,2000)") #qcd_AllChGood_noPU.chain.Draw("met_rawPt","","same") #qcd_noPU.chain.Draw("met_genPt","","same") #qcd_AllChGood_noPU.chain.Draw("met_genPt","","same")
#dirs['other'] += dirs['WWW'] + dirs['WWZ'] + dirs['WZG'] + dirs['WZZ'] + dirs['ZZZ'] dirs['other'] += dirs['VV'] + dirs['WW'] + dirs['WZ'] + dirs['ZZ'] dirs['other'] += dirs['GluGlu'] directories = { key: [ os.path.join(data_directory, postprocessing_directory2016, dir) for dir in dirs[key] ] for key in dirs.keys() } # Samples DY_LO_16 = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) TT_pow_16 = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTLep_pow']) TTbar_16 = Sample.fromDirectory(name="TTbar", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTbar']) singleTop_16 = Sample.fromDirectory(name="singleTop",
#dirs['WZ'] = ["WZTo1L3Nu", "WZTo1L1Nu2Q", "WZTo2L2Q", "WZTo3LNu"] #dirs['ZZ'] = ["ZZTo2L2Nu", "ZZTo2L2Q"] #dirs['ZZ'] = ["ZZTo2L2Nu", "ZZTo2L2Q", "ZZTo2Q2Nu", "ZZTo4L"] #dirs['GluGlu'] = ["GluGluToContinToZZTo2e2mu", "GluGluToContinToZZTo2e2tau", "GluGluToContinToZZTo2mu2tau", "GluGluToContinToZZTo4e", "GluGluToContinToZZTo4mu", "GluGluToContinToZZTo4tau"] dirs['GluGlu'] = ["GluGluToContinToZZTo2e2mu", "GluGluToContinToZZTo2e2tau"] dirs['other'] = [] #dirs['other'] += dirs['TZQ'] + dirs['THQ'] + dirs['THW'] #+ dirs['TWZ'] #dirs['other'] += dirs['TTW'] + dirs['TTZ'] + dirs['TTH'] #dirs['other'] += dirs['TTWZ'] + dirs['TTZZ'] + dirs['TTWW'] #dirs['other'] += dirs['WWW'] + dirs['WWZ'] + dirs['WZG'] + dirs['WZZ'] + dirs['ZZZ'] #dirs['other'] += dirs['VV'] + dirs['WW'] + dirs['WZ'] + dirs['ZZ'] dirs['other'] += dirs['WW'] dirs['other'] += dirs['GluGlu'] directories = { key : [ os.path.join( data_directory, postprocessing_directory2018, dir) for dir in dirs[key] ] for key in dirs.keys() } # Samples DY_LO_18 = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY (LO)", directory=directories['DY_LO']) TT_pow_18 = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTLep_pow']) #TTbar_18 = Sample.fromDirectory(name="TTbar", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TTbar']) #singleTop_18 = Sample.fromDirectory(name="singleTop", treeName="Events", isData=False, color=color.singleTop, texName="single-t", directory=directories['singleTop']) #TTGLep_18 = Sample.fromDirectory(name="TTGLep", treeName="Events", isData=False, color=color.TTG, texName="t#bar{t}#gamma", directory=directories['TTGLep']) #TTG_18 = Sample.fromDirectory(name="TTGLep", treeName="Events", isData=False, color=color.TTG, texName="t#bar{t}#gamma", directory=directories['TTG']) #ZGTo2LG_18 = Sample.fromDirectory(name="ZGTo2LG", treeName="Events", isData=False, color=color.diBoson, texName="Z#gamma", directory=directories['ZGTo2LG'] ) #ZGToLLG_18 = Sample.fromDirectory(name="ZGToLLG", treeName="Events", isData=False, color=color.diBoson, texName="Z#gamma", directory=directories['ZGToLLG'] ) other_18 = Sample.fromDirectory(name="other", treeName="Events", isData=False, color=color.other, texName="other", directory=directories['other']) signals = []
dirs['ZZ'] = ["ZZTo2L2Q", "ZZTo2Q2Nu"] dirs['WZ'] = ["WZTo1L1Nu2Q", "WZTo2L2Q", "WZTo3LNu"] dirs['triBoson'] = ["WWZ","WZZ","ZZZ"] # No Fall15 production for WWZ and ZZZ? dirs['WZZ'] = ["WZZ"] dirs['QCD_HT'] = ["QCD_HT100to200", "QCD_HT200to300", "QCD_HT300to500", "QCD_HT500to700", "QCD_HT700to1000", "QCD_HT1000to1500", "QCD_HT1500to2000", "QCD_HT2000toInf"] dirs['QCD_Mu5'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5"] dirs['QCD_EM+bcToE'] = ["QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_EM+bcToE'] = ["QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] dirs['QCD_Mu5+EM+bcToE'] = ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched"] #dirs['QCD_Mu5+EM+bcToE']= ["QCD_Pt20to30_Mu5", "QCD_Pt50to80_Mu5", "QCD_Pt80to120_Mu5", "QCD_Pt120to170_Mu5", "QCD_Pt170to300_Mu5", "QCD_Pt300to470_Mu5", "QCD_Pt470to600_Mu5", "QCD_Pt600to800_Mu5", "QCD_Pt800to1000_Mu5", "QCD_Pt1000toInf_Mu5", "QCD_Pt_15to20_bcToE", "QCD_Pt_20to30_bcToE", "QCD_Pt_30to80_bcToE", "QCD_Pt_80to170_bcToE", "QCD_Pt_170to250_bcToE", "QCD_Pt_250toInf_bcToE", "QCD_Pt15to20_EMEnriched", "QCD_Pt20to30_EMEnriched", "QCD_Pt30to50_EMEnriched", "QCD_Pt50to80_EMEnriched", "QCD_Pt80to120_EMEnriched", "QCD_Pt120to170_EMEnriched", "QCD_Pt170to300_EMEnriched", "QCD_Pt300toInf_EMEnriched"] dirs['QCD'] = ["QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200"] #dirs['QCD'] = ["QCD_Pt10to15", "QCD_Pt15to30", "QCD_Pt30to50", "QCD_Pt50to80", "QCD_Pt80to120", "QCD_Pt120to170", "QCD_Pt170to300", "QCD_Pt300to470", "QCD_Pt470to600", "QCD_Pt600to800", "QCD_Pt800to1000", "QCD_Pt1000to1400", "QCD_Pt1400to1800", "QCD_Pt1800to2400", "QCD_Pt2400to3200", "QCD_Pt3200"] directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} DY = Sample.fromDirectory(name="DY", treeName="Events", isData=False, color=color.DY, texName="DY + Jets", directory=directories['DY']) DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO)", directory=directories['DY_LO']) DY_HT_LO = Sample.fromDirectory(name="DY_HT_LO", treeName="Events", isData=False, color=color.DY, texName="DY + Jets (LO,HT)", directory=directories['DY_HT_LO']) TTJets = Sample.fromDirectory(name="TTJets", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets", directory=directories['TTJets']) TTJets_LO = Sample.fromDirectory(name="TTJets_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (LO)", directory=directories['TTJets_LO']) TTLep_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep,pow)", directory=directories['TTLep_pow']) TTJets_Lep = Sample.fromDirectory(name="TTJets_Lep", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (lep)", directory=directories['TTJets_Lep']) TTJets_HT_LO = Sample.fromDirectory(name="TTJets_HT_LO", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t} + Jets (HT,LO)", directory=directories['TTJets_HT_LO']) singleTop = Sample.fromDirectory(name="singleTop", treeName="Events", isData=False, color=color.singleTop, texName="single top", directory=directories['singleTop']) TTX = Sample.fromDirectory(name="TTX", treeName="Events", isData=False, color=color.TTX, texName="t#bar{t}H/W/Z, tZq", directory=directories['TTX']) TTXNoZ = Sample.fromDirectory(name="TTXNoZ", treeName="Events", isData=False, color=color.TTXNoZ, texName="t#bar{t}H/W, tZq", directory=directories['TTXNoZ']) TTH = Sample.fromDirectory(name="TTH", treeName="Events", isData=False, color=color.TTH, texName="t#bar{t}H", directory=directories['TTH']) TTW = Sample.fromDirectory(name="TTW", treeName="Events", isData=False, color=color.TTW, texName="t#bar{t}W", directory=directories['TTW']) TTZ = Sample.fromDirectory(name="TTZ", treeName="Events", isData=False, color=color.TTZ, texName="t#bar{t}Z", directory=directories['TTZ']) TTZtoLLNuNu = Sample.fromDirectory(name="TTZtoNuNu", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (l#bar{l}/#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) TTZtoQQ = Sample.fromDirectory(name="TTZtoQQ", treeName="Events", isData=False, color=color.TTZtoQQ, texName="t#bar{t}Z (q#bar{q})", directory=directories['TTZtoQQ'])
#dirs['WJets_LO'] = ['WJetsToLNu_LO'] #dirs['WJets'] = ['WJetsToLNu'] #dirs['pseudoData'] = dirs['TTZtoLLNuNu'] + dirs["WZ"] + dirs['TTW'] + dirs['TTX'] + dirs['TZQ'] + dirs['rare'] + dirs['nonprompt'] #dirs['pseudoDataPriv'] = dirs['ewkDM_ttZ_ll_noH'] + dirs["WZ"] + dirs['TTW'] + dirs['TTX'] + dirs['TZQ'] + dirs['rare'] + dirs['nonprompt'] #dirs['background'] = dirs["WZ"] + dirs['TTW'] + dirs['TTX'] + dirs['TZQ'] + dirs['rare'] #dirs['nonprompt'] = ['DYJetsToLL_M50', 'TTJets'] # no v4 version atm directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()} #pseudoData = Sample.fromDirectory(name="pseudoData", treeName="Events", isData=False, color=ROOT.kBlack, texName="pseudo-data", directory=directories['pseudoData']) #pseudoDataPriv = Sample.fromDirectory(name="pseudoDataPriv", treeName="Events", isData=False, color=ROOT.kBlack, texName="pseudo-data", directory=directories['pseudoDataPriv']) #TTZtoLLNuNu = Sample.fromDirectory(name="TTZtoLLNuNu", treeName="Events", isData=False, color=color.TTZtoLLNuNu, texName="t#bar{t}Z (l#bar{l}/#nu#bar{#nu})", directory=directories['TTZtoLLNuNu']) #TTZ_LO = Sample.fromDirectory(name="TTZ_LO", treeName="Events", isData=False, color=color.TTZtoLLNuNu+1, texName="t#bar{t}Z (LO)", directory=directories['TTZ_LO']) DY_LO = Sample.fromDirectory(name="DY_LO", treeName="Events", isData=False, color=color.DY, texName="DYToLL (LO)", directory=directories['DY_LO']) WZ = Sample.fromDirectory(name="WZ", treeName="Events", isData=False, color=color.WZ, texName="WZ", directory=directories['WZ']) WW = Sample.fromDirectory(name="WW", treeName="Events", isData=False, color=color.WW, texName="WW", directory=directories['WW']) ZZ = Sample.fromDirectory(name="ZZ", treeName="Events", isData=False, color=color.ZZ, texName="ZZ", directory=directories['ZZ']) #WJets_LO = Sample.fromDirectory(name="WJets_LO", treeName="Events", isData=False, color=color.WJetsToLNu, texName="W+jets (LO)", directory=directories['WJets_LO']) #WJets = Sample.fromDirectory(name="WJets", treeName="Events", isData=False, color=color.WJetsToLNu+2, texName="W+jets (NLO)", directory=directories['WJets']) #TTX = Sample.fromDirectory(name="TTX", treeName="Events", isData=False, color=ROOT.kRed-10, texName="t(t)X", directory=directories['TTX']) #TTW = Sample.fromDirectory(name="TTW", treeName="Events", isData=False, color=color.TTX, texName="t#bar{t}W", directory=directories['TTW']) #TZQ = Sample.fromDirectory(name="TZQ", treeName="Events", isData=False, color=ROOT.kOrange+7, texName="tZq", directory=directories['TZQ']) #rare = Sample.fromDirectory(name="rare", treeName="Events", isData=False, color=color.rare, texName="rare", directory=directories['rare']) nonprompt = Sample.fromDirectory(name="nonprompt", treeName="Events", isData=False, color=color.nonprompt, texName="nonprompt", directory=directories['nonprompt']) TT_pow = Sample.fromDirectory(name="TTLep_pow", treeName="Events", isData=False, color=color.TTJets, texName="t#bar{t}", directory=directories['TT_pow']) #background = Sample.fromDirectory(name="background", treeName="Events", isData=False, color=color.nonprompt, texName="background", directory=directories['background'])
def getSample(pd, runName, lumi): sample = Sample.fromDirectory(name=(pd + '_' + runName + '_backup'), treeName="Events", texName=(pd + ' (' + runName + ')'), directory=dirs[pd + '_' + runName + '_backup']) sample.lumi = lumi return sample