Exemplo n.º 1
0
def GetHistosFromLooper(year, ecut = '(t.TStatus == 1 or t.TStatus == 22)', processes = ['Charge flips','Prompt SS']):
  #l = looper(path=path[year], nSlots = 6, treeName = 'MiniTree', options = 'merge', nEvents = 1000, outpath=outpath+'tempfiles/')
  l = looper(path=path[year], nSlots = 20, treeName = 'MiniTree', options = 'merge', outpath = outpath+'tempfiles/')
  l.AddHeader('from framework.mva import ModelPredict\n')
  l.AddInit('      self.pd1 = ModelPredict("%s")\n'%model)

  for p in processes: l.AddSample(p,  processDic[year][p])

  syst = 'MuonEff, ElecEff, Trig, JES, JER, MuonES, Uncl, Btag, MisTag, PU, TopPt'
  if year != 2018: syst += ', Pref'
  systlist = [x+y for x in syst.replace(' ','').split(',') for y in ['Up','Down']]
  l.AddSyst(systlist)
  l.AddLoopCode(loopcode)

  l.AddExpr('deltaphi', 'TDeltaPhi', 'abs(TDeltaPhi)/3.141592')
  l.AddExpr('deltaeta', 'TDeltaEta', 'abs(TDeltaEta)')
  l.AddExpr('weight1', [], '1')
  l.AddExpr('vpd', '', 'prob1', True) # True because prob1 has to be defined before
  l.AddExpr('vmet', ['TMET'], 'TMET')
  l.AddExpr('vmt2', ['TMT2'], 'TMT2')
  l.AddExpr('vht',  ['THT' ], 'THT')
  l.AddExpr('vmll', ['TMll'], 'TMll')
  l.AddExpr('exprW', ['TWeight'], 'TWeight')

  l.AddSelection(selection)
  weight = 'exprW'
  cut = 'TMET > 50 and TMT2 > 80 and TNJets >= 2 and TNBtags >= 1 and TPassDilep and %s'%ecut
  l.AddCut(cut, ['TMET', 'TMT2', 'TNJets', 'TNBtags', 'TPassDilep'])
  cut = ''

  l.AddHisto('vpd',   'dnn',  20, 0, 1,   weight = weight, cut = '')
  l.AddHisto('TMll', 'mll', 30, 0, 300, weight = weight, cut = '')
  l.AddHisto('TMET', 'met', 25, 50, 300, weight = weight, cut = '')
  l.AddHisto('TMT2', 'mt2', 8, 80, 160, weight = weight, cut = '')
  l.AddHisto('TDilep_Pt', 'dileppt', 30, 0, 300, weight = weight, cut = '')
  l.AddHisto('TLep0Pt', 'lep0pt', 15, 0, 150, weight = weight, cut = '')
  l.AddHisto('TLep1Pt', 'lep1pt', 30, 0, 300, weight = weight, cut = '')
  l.AddHisto('TLep0Eta', 'lep0eta', 30, -2.4, 2.4, weight = weight, cut = '')
  l.AddHisto('TLep1Eta', 'lep1eta', 30, -2.4, 2.4, weight = weight, cut = '')
  l.AddHisto('deltaphi', 'deltaphi', 30, 0, 1, weight = weight, cut = '')
  l.AddHisto('deltaeta', 'deltaeta', 30, 0, 5, weight = weight, cut = '')
  l.AddHisto('TJet0Pt', 'jet0pt', 30, 0, 450, weight = weight, cut = '')
  l.AddHisto('TJet1Pt', 'jet1pt', 30, 0, 300, weight = weight, cut = '')
  l.AddHisto('TJet0Eta', 'jet0eta', 30, -2.4, 2.4, weight = weight, cut = '')
  l.AddHisto('TJet1Eta', 'jet1eta', 30, -2.4, 2.4, weight = weight, cut = '')
  l.AddHisto('THT', 'ht', 40, 0, 800, weight = weight, cut = '')
  l.AddHisto('TNJets', 'njets', 6, 1.5, 7.5, weight = weight, cut = '')
  l.AddHisto('TNBtags', 'nbtags', 3, 0.5, 3.5, weight = weight, cut = '')

  out = l.Run()
  return out
Exemplo n.º 2
0
region = args.region
if args.BS: region = 'BS'
if args.SR: region = 'SR'

# Set constants
outpath = baseoutpath + '/Unc_FSR/%s/%s/%i/mass%i_%i/' % (region, ch, year, ms,
                                                          ml)
if not os.path.isdir(outpath): os.system("mkdir -p %s" % outpath)

# Create the looper, set readOutput to true to read previous temporary rootfiles created for each sample
path = {}
path[year] = pathToTrees(year, ch, region)
pathToFiles = path[year]
l = looper(path=path[year] if region == 'SR' else pathBS[year],
           nSlots=nSlots,
           treeName='MiniTree',
           options='merge',
           outpath=outpath + 'tempfiles/',
           readOutput=True)  #, sendJobs=sendJobs)

# Add processes
processDic = {2016: {'tt': 'TT_PSWeights'}}
processes = ['tt']
if argprocess != '': processes = argprocess
for p in processes:
    l.AddSample(p, processDic[year][p])

# Systematic uncertainties
syst = 'FSR, ISR'
systlist = [
    x + y for x in syst.replace(' ', '').split(',') for y in ['Up', 'Down']
]
Exemplo n.º 3
0
def GetHistosFromLooper(year,
                        ecut='(t.TStatus == 1 or t.TStatus == 22)',
                        processes=['Chargeflips', 'PromptSS'],
                        outfolder='/'):
    #l = looper(path=path[year], nSlots = 6, treeName = 'MiniTree', options = 'merge', nEvents = 1000, outpath=outpath+'tempfiles/')
    l = looper(path=path[year] if not region == 'BS' else pathBS[year],
               nSlots=4,
               treeName='MiniTree',
               options='merge',
               outpath=GetOutPath(year, region) + '/' + outfolder +
               '/tempfiles/',
               readOutput=True)
    l.AddHeader('from framework.mva import ModelPredict\n')
    l.AddInit('      self.pd1 = ModelPredict("%s")\n' % model)

    for p in processes:
        l.AddSample(p, processDic[year][p])

    l.AddSyst(systlist)
    l.AddLoopCode(loopcode)

    l.AddExpr('deltaphi', 'TDeltaPhi', 'abs(TDeltaPhi)/3.141592')
    l.AddExpr('deltaeta', 'TDeltaEta', 'abs(TDeltaEta)')
    l.AddExpr('weight1', [], '1')
    l.AddExpr('vpd', '', 'prob1',
              True)  # True because prob1 has to be defined before
    l.AddExpr('vmet', ['TMET'], 'TMET')
    l.AddExpr('vmt2', ['TMT2'], 'TMT2')
    l.AddExpr('vht', ['THT'], 'THT')
    l.AddExpr('vmll', ['TMll'], 'TMll')
    l.AddExpr('exprW', ['TWeight'], 'TWeight')

    l.AddSelection(selection)
    weight = 'exprW'
    #cut = 'TMET > 50 and TMT2 > 80 and TNJets >= 2 and TNBtags >= 1 and TPassDilep and %s'%ecut
    #l.AddCut(cut, ['TMET', 'TMT2', 'TNJets', 'TNBtags', 'TPassDilep'])

    l.AddCut('TPassDilep == 1', 'TPassDilep')
    if region != 'BS':
        l.AddCut('TMET >= 50', 'TMET')
        l.AddCut('TMT2 >= 80', 'TMT2')
    l.AddCut('TNJets >= 2', 'TNJets')
    l.AddCut('TNBtags >= 1', 'TNBtags')
    l.AddCut(ecut, [])
    cut = ''

    l.AddHisto('vpd', 'dnn_5', 5, 0, 1, weight=weight, cut='')
    l.AddHisto('vpd', 'dnn_10', 10, 0, 1, weight=weight, cut='')
    l.AddHisto('vpd', 'dnn', 20, 0, 1, weight=weight, cut='')
    l.AddHisto('vpd', 'count', 1, 0, 2, weight=weight, cut='')
    l.AddHisto('TMll', 'mll', 10, 0, 300, weight=weight, cut='')
    l.AddHisto('TMET', 'met', 10, 50, 300, weight=weight, cut='')
    l.AddHisto('TMT2', 'mt2', 8, 80, 160, weight=weight, cut='')
    l.AddHisto('TMT2', 'mt2_4bins', 4, 80, 160, weight=weight, cut='')
    l.AddHisto('TDilep_Pt', 'dileppt', 10, 0, 300, weight=weight, cut='')
    l.AddHisto('TLep0Pt', 'lep0pt', 5, 0, 150, weight=weight, cut='')
    l.AddHisto('TLep1Pt', 'lep1pt', 10, 0, 300, weight=weight, cut='')
    l.AddHisto('TLep0Eta', 'lep0eta', 8, -2.4, 2.4, weight=weight, cut='')
    l.AddHisto('TLep1Eta', 'lep1eta', 8, -2.4, 2.4, weight=weight, cut='')
    l.AddHisto('deltaphi', 'deltaphi', 8, 0, 1, weight=weight, cut='')
    l.AddHisto('deltaeta', 'deltaeta', 8, 0, 5, weight=weight, cut='')
    l.AddHisto('TJet0Pt', 'jet0pt', 10, 0, 450, weight=weight, cut='')
    l.AddHisto('TJet1Pt', 'jet1pt', 6, 0, 300, weight=weight, cut='')
    l.AddHisto('TJet0Eta', 'jet0eta', 8, -2.4, 2.4, weight=weight, cut='')
    l.AddHisto('TJet1Eta', 'jet1eta', 8, -2.4, 2.4, weight=weight, cut='')
    l.AddHisto('THT', 'ht', 8, 0, 800, weight=weight, cut='')
    l.AddHisto('TNJets', 'njets', 6, 1.5, 7.5, weight=weight, cut='')
    l.AddHisto('TNBtags', 'nbtags', 3, 0.5, 3.5, weight=weight, cut='')

    out = l.Run()
    return out
Exemplo n.º 4
0
 if   self.outname == 'tt':
   if t.TJERindex != 0: return 
 elif self.outname == 'ttgaus':
   if t.TJERindex != 1: return 
 elif self.outname == 'ttnongaus':
   if t.TJERindex != 2:  return 
'''%('\n'+stopcuts+'\n')


loopcode = '''
values = [%i,  %i, t.TDilep_Pt, t.TDeltaPhi, t.TDeltaEta, t.TLep0Pt, t.TLep0Eta, t.TLep1Pt, vmet, t.TLep1Eta, vmll, vmt2, vht]
prob1 = self.pd1.GetProb(values)
'''%(ms, ml)

if doSaveHistos:
  l = looper(path=path[year] if not region=='BS' else pathBS[year], nSlots = 4, treeName = 'MiniTree', options = 'merge', outpath = GetOutPath(year, region, chan, ms, ml)+'/tempfiles/', readOutput=True)
  l.AddHeader('from framework.mva import ModelPredict\n')
  l.AddInit('      self.pd1 = ModelPredict("%s")\n'%model)
  
  processes = processDic[year].keys() #['tt']
  for p in processes: l.AddSample(p,  processDic[year][p])
  
  l.AddSyst(systlist)
  l.AddLoopCode(loopcode)
  
  l.AddExpr('deltaphi', 'TDeltaPhi', 'abs(TDeltaPhi)/3.141592')
  l.AddExpr('deltaeta', 'TDeltaEta', 'abs(TDeltaEta)')
  l.AddExpr('weight1', [], '1')
  l.AddExpr('vpd', '', 'prob1', True) # True because prob1 has to be defined before
  l.AddExpr('vmet', ['TMET'], 'TMET')
  l.AddExpr('vmt2', ['TMT2'], 'TMT2')
Exemplo n.º 5
0
ms = 275
ml = 100

# Set constants
outpath = './Unc/SR/mass%i_%i/' % (ms, ml)
os.system("mkdir -p %s" % outpath)
model = '/nfs/fanae/user/juanr/CMSSW_10_2_5/src/xuAnalysis/TopPlots/DrawMiniTrees/NNtotal_model2.h5'

# Set year
year = 2018

# Create the looper, set readOutput to true to read previous temporary rootfiles created for each sample
#l = looper(path=path[year], nSlots = 6, treeName = 'MiniTree', options = 'merge', nEvents = 1000, outpath='tempfiles/')
l = looper(path=path[year],
           nSlots=5,
           treeName='MiniTree',
           options='merge',
           outpath=outpath + 'tempfiles/',
           readOutput=True)

# Add processes
processes = processDic[year].keys()  #['tt']
for p in processes:
    l.AddSample(p, processDic[year][p])

# Systematic uncertainties
syst = 'MuonEff, ElecEff, Trig, JES, JER, MuonES, Uncl, Btag, MisTag, PU, TopPt'
if year != 2018: syst += ', Pref'
systlist = [
    x + y for x in syst.replace(' ', '').split(',') for y in ['Up', 'Down']
]