| 1 |
import ROOT
|
| 2 |
import sys, re, math
|
| 3 |
sys.path.append("/afs/naf.desy.de/user/n/nowaf/UserCode/nowaf/PythonScripts/")
|
| 4 |
import definitions as Def
|
| 5 |
import styles as Style
|
| 6 |
|
| 7 |
class Eff:
|
| 8 |
def __init__( self, u , l ):
|
| 9 |
self.upper = u
|
| 10 |
self.lower = l
|
| 11 |
self.eff = 0.
|
| 12 |
if not self.lower == 0.:
|
| 13 |
self.eff = self.upper/self.lower
|
| 14 |
pass
|
| 15 |
self.errUp, self.errDwn = self.getErr()
|
| 16 |
pass
|
| 17 |
|
| 18 |
def getErr( self ):
|
| 19 |
h1 = ROOT.TH1F( "h1", "h1", 1, 0, 1 )
|
| 20 |
h2 = ROOT.TH1F( "h2", "h2", 1, 0, 1 )
|
| 21 |
|
| 22 |
for i in range( 0, int( self.upper ) ):
|
| 23 |
h1.Fill( 0.5 )
|
| 24 |
pass
|
| 25 |
for i in range( 0, int( self.lower ) ):
|
| 26 |
h2.Fill( 0.5 )
|
| 27 |
pass
|
| 28 |
|
| 29 |
g = ROOT.TGraphAsymmErrors()
|
| 30 |
g.BayesDivide( h1, h2 )
|
| 31 |
errUp = g.GetErrorYhigh( 0 )
|
| 32 |
errDwn = g.GetErrorYlow( 0 )
|
| 33 |
return errUp, errDwn
|
| 34 |
|
| 35 |
pass
|
| 36 |
|
| 37 |
class ReadIn:
|
| 38 |
"""
|
| 39 |
Standard container class for reading in standard samples
|
| 40 |
"""
|
| 41 |
def __init__( self, fileDir, fileName, lumi = 1000., rebinDict = {}, externalScalingFactor = 1. ):
|
| 42 |
self.sampleList = []
|
| 43 |
self.sampleList.append( "TTbar" )
|
| 44 |
self.sampleList.append( "GVJets" )
|
| 45 |
self.sampleList.append( "WWJets" )
|
| 46 |
self.sampleList.append( "Zinv" )
|
| 47 |
self.sampleList.append( "ZJets" )
|
| 48 |
self.sampleList.append( "WJets" )
|
| 49 |
self.sampleList.append( "QCDFlat" )
|
| 50 |
self.sampleList.append( "Data" )
|
| 51 |
|
| 52 |
self.fileDict = {}
|
| 53 |
self.fileDir = fileDir
|
| 54 |
self.fileName = fileName
|
| 55 |
#for sample in self.sampleList:
|
| 56 |
# self.fileDict[ sample ] = fileDir + sample + fileName + ".root"
|
| 57 |
# pass
|
| 58 |
|
| 59 |
#self.aliasDict = {}
|
| 60 |
#self.aliasDict[ "GVJets" ] = "GV+Jets"
|
| 61 |
#self.aliasDict[ "VVJets" ] = "VV+Jets"
|
| 62 |
#self.aliasDict[ "ZJets" ] = "Z+Jets"
|
| 63 |
#self.aliasDict[ "QCDFlat" ] = "Multijet"
|
| 64 |
|
| 65 |
self.histList = []
|
| 66 |
self.histList.append( "MHT" )
|
| 67 |
|
| 68 |
self.scalingFactors = {}
|
| 69 |
#self.scalingFactors[ "Multijet" ] = lumi/100.
|
| 70 |
#self.scalingFactors[ "ZJets" ] = lumi/100.
|
| 71 |
#self.scalingFactors[ "VVJets" ] = lumi/100.
|
| 72 |
#self.scalingFactors[ "WJets" ] = lumi * 31300. / ( 15110974 * 0.98 )
|
| 73 |
#self.scalingFactors[ "TTbar" ] = lumi * 165 / ( 1164208. * 0.92 )
|
| 74 |
#self.scalingFactors[ "GVJets" ] = lumi * 173 / ( 1102193. * 0.97 )
|
| 75 |
#self.scalingFactors[ "Zinv" ] = lumi * 5760 / ( 2165002. * 0.96 )
|
| 76 |
#self.scalingFactors[ "Data" ] = 1.
|
| 77 |
|
| 78 |
self.lumi = lumi
|
| 79 |
self.externalScalingFactor = externalScalingFactor
|
| 80 |
|
| 81 |
#### Summer11
|
| 82 |
self.scalingFactors[ "QCDFlat" ] = lumi/1000.
|
| 83 |
self.scalingFactors[ "ZJets" ] = lumi * 3048/( 36277961 *1. )
|
| 84 |
##print "Warning: Wrong scaling Factor for: VVJets"
|
| 85 |
self.scalingFactors[ "VVJets" ] = lumi/100.
|
| 86 |
self.scalingFactors[ "WWJets" ] = lumi * 43. / ( 1197558. * 1. )
|
| 87 |
self.scalingFactors[ "WJets" ] = lumi * 31314. / ( 81352581 )
|
| 88 |
|
| 89 |
#self.scalingFactors[ "WJets" ] = lumi * 31314. / ( 81352581* 0.93 )
|
| 90 |
|
| 91 |
#self.scalingFactors[ "WJets" ] = lumi * 31300. / ( 81352581 * 451./483 )
|
| 92 |
self.scalingFactors[ "TTbar" ] = lumi * 165 / ( 3701947. * 1. )
|
| 93 |
self.scalingFactors[ "GVJets" ] = lumi * 173 / ( 1067879. * 1. )
|
| 94 |
##print "Warning: Wrong scaling Factor for: Zinv"
|
| 95 |
self.scalingFactors[ "Zinv" ] = lumi * 32.92 / ( 3067017. )
|
| 96 |
self.scalingFactors[ "Data" ] = 1.
|
| 97 |
self.scalingFactors[ "LM13" ] = lumi * 6.899 / ( 77000. * 1. ) ### skim
|
| 98 |
#self.scalingFactors[ "LM13" ] = lumi * 6.899 / ( 517000 * 1 ) ### sie haben nachproduziert!!
|
| 99 |
self.scalingFactors[ "LM3" ] = lumi * 3.438 / ( 36475. * 1. )
|
| 100 |
self.scalingFactors[ "LM8" ] = lumi * 0.73 / ( 10595. * 1. )
|
| 101 |
#self.scalingFactors[ "LM2" ] = lumi * 0.6027 / ( 11000. * 1. ) ### skim
|
| 102 |
self.scalingFactors[ "LM2" ] = lumi * 0.6027 / ( 445060 * 1. ) ### the new, unpublished skim
|
| 103 |
|
| 104 |
#self.scalingFactors[ "Zinv" ] = lumi * 32.92 / ( 3067017. * 0.84 ) ### full_iii
|
| 105 |
#self.scalingFactors[ "WWJets" ] = lumi * 43. / ( 1197558. * 0.08 ) ### full_iii
|
| 106 |
#self.scalingFactors[ "LM13" ] = lumi * 6.899 / ( 517000 * 1 ) ### full_iii
|
| 107 |
|
| 108 |
#### overwrite for full_ii
|
| 109 |
#self.scalingFactors[ "LM13" ] = lumi * 6.899 / ( 517000 * 1 ) ### sie haben nachproduziert!!
|
| 110 |
#self.scalingFactors[ "LM2" ] = lumi * 0.6027 / ( 445060 * 1. )
|
| 111 |
#self.scalingFactors[ "WWJets" ] = lumi * 43. / ( 1197558. * 0.9 )
|
| 112 |
#self.scalingFactors[ "Zinv" ] = lumi * 32.92 / ( 3067017. * 0.26 )
|
| 113 |
|
| 114 |
self.doScaling = True
|
| 115 |
self.normalize = False
|
| 116 |
|
| 117 |
self.rebinDict = rebinDict
|
| 118 |
|
| 119 |
pass
|
| 120 |
|
| 121 |
def changeScalingFactor( self, sample, factor ):
|
| 122 |
self.scalingFactors[ sample ] = factor
|
| 123 |
pass
|
| 124 |
|
| 125 |
def dontScale( self ):
|
| 126 |
self.doScaling = False
|
| 127 |
pass
|
| 128 |
|
| 129 |
def doNormalization( self ):
|
| 130 |
self.normalize = True
|
| 131 |
pass
|
| 132 |
|
| 133 |
def getHists( self, histDir ):
|
| 134 |
|
| 135 |
for sample in self.sampleList:
|
| 136 |
if sample not in [ "LM2", "LM13" ]:
|
| 137 |
self.fileDict[ sample ] = self.fileDir + sample + self.fileName + ".root"
|
| 138 |
else:
|
| 139 |
self.fileDict[ sample ] = self.fileDir + sample + "_cxviii" + ".root"
|
| 140 |
pass
|
| 141 |
|
| 142 |
histDict = {}
|
| 143 |
print "..... reading in ....."
|
| 144 |
for hist in self.histList:
|
| 145 |
print hist, ":"
|
| 146 |
histDict[ hist ] = {}
|
| 147 |
for file in self.sampleList:
|
| 148 |
print " from sample ", file
|
| 149 |
rootfile = ROOT.TFile.Open( self.fileDict[ file ] )
|
| 150 |
histDict[ hist ][ file ] = rootfile.Get( histDir + hist )
|
| 151 |
histDict[ hist ][ file ].SetDirectory( 0 )
|
| 152 |
if hist in self.rebinDict.keys():
|
| 153 |
histDict[ hist ][ file ].Rebin( self.rebinDict[ hist ] )
|
| 154 |
pass
|
| 155 |
if self.doScaling:
|
| 156 |
histDict[ hist ][ file ].Scale( self.scalingFactors[ file ] )
|
| 157 |
histDict[ hist ][ file ].Scale( self.externalScalingFactor )
|
| 158 |
pass
|
| 159 |
if self.normalize:
|
| 160 |
if self.doScaling:
|
| 161 |
print "Warning! Hists will be normalized, scaling has no effect."
|
| 162 |
pass
|
| 163 |
norm = histDict[ hist ][ file ].Integral()
|
| 164 |
if not norm == 0:
|
| 165 |
histDict[ hist ][ file ].Scale( 1/norm )
|
| 166 |
else:
|
| 167 |
print "Not normalizing because norm == 0."
|
| 168 |
pass
|
| 169 |
pass
|
| 170 |
pass
|
| 171 |
|
| 172 |
return histDict
|
| 173 |
|
| 174 |
def addHists( self, histList ):
|
| 175 |
for h in histList:
|
| 176 |
if not h in self.histList:
|
| 177 |
self.histList.append( h )
|
| 178 |
pass
|
| 179 |
pass
|
| 180 |
pass
|
| 181 |
|
| 182 |
def removeHists( self, histList ):
|
| 183 |
for h in histList:
|
| 184 |
self.histList.pop( self.histList.index( h ) )
|
| 185 |
pass
|
| 186 |
pass
|
| 187 |
|
| 188 |
#def setAlias( self, sample, alias ):
|
| 189 |
# self.aliasDict[ sample ] = alias
|
| 190 |
# pass
|
| 191 |
|
| 192 |
#def removeAlias( self, sample ):
|
| 193 |
# del self.aliasDict[ sample ]
|
| 194 |
# pass
|
| 195 |
|
| 196 |
def removeSamples( self, removalList ):
|
| 197 |
for r in removalList:
|
| 198 |
self.sampleList.pop( self.sampleList.index( r ) )
|
| 199 |
pass
|
| 200 |
pass
|
| 201 |
|
| 202 |
def addSamples( self, addList ):
|
| 203 |
for a in addList:
|
| 204 |
self.sampleList.append( a )
|
| 205 |
pass
|
| 206 |
pass
|
| 207 |
pass
|
| 208 |
|
| 209 |
class AddUp:
|
| 210 |
def __init__( self, histDict ):
|
| 211 |
self.histDict = histDict
|
| 212 |
pass
|
| 213 |
def add( self ):
|
| 214 |
newHist = None
|
| 215 |
for sample in self.histDict.keys():
|
| 216 |
if not newHist == None:
|
| 217 |
newHist.Add( self.histDict[ sample ] )
|
| 218 |
else:
|
| 219 |
newHist = self.histDict[ sample ].Clone()
|
| 220 |
newHist.SetDirectory( 0 )
|
| 221 |
pass
|
| 222 |
pass
|
| 223 |
return newHist
|
| 224 |
pass
|
| 225 |
|
| 226 |
class StackErr:
|
| 227 |
def __init__( self, histDict, histDictUnscaled ):
|
| 228 |
self.histDict = histDict
|
| 229 |
self.histDictUnscaled = histDictUnscaled
|
| 230 |
#self.scalingFactors = scalingFactors
|
| 231 |
pass
|
| 232 |
def getStackErr( self ):
|
| 233 |
errDict = {}
|
| 234 |
errDictBins = {}
|
| 235 |
errDictBins[ "All" ] = {}
|
| 236 |
sum = 0.
|
| 237 |
for sample in self.histDict.keys():
|
| 238 |
#print self.histDict[ sample ].GetName()
|
| 239 |
#print self.histDict[ sample ].Integral()
|
| 240 |
binsum = 0
|
| 241 |
binsumUnscaled = 0
|
| 242 |
errDictBins[ sample ] = {}
|
| 243 |
for bin in range( 1, self.histDict[ sample ].GetNbinsX() + 1 ):
|
| 244 |
binsumUnscaled += self.histDictUnscaled[ sample ].GetBinContent( bin )
|
| 245 |
binsum += self.histDict[ sample ].GetBinContent( bin )
|
| 246 |
averageBinWeight = 1.
|
| 247 |
if not self.histDictUnscaled[ sample ].GetBinContent( bin ) == 0:
|
| 248 |
averageBinWeight = self.histDict[ sample ].GetBinContent( bin )/\
|
| 249 |
self.histDictUnscaled[ sample ].GetBinContent( bin )
|
| 250 |
pass
|
| 251 |
errDictBins[ sample ][ bin ] = math.sqrt( self.histDictUnscaled[ sample ].GetBinContent( bin ) ) *\
|
| 252 |
averageBinWeight
|
| 253 |
if not bin in errDictBins[ "All" ].keys():
|
| 254 |
errDictBins[ "All" ][ bin ] = 0.
|
| 255 |
pass
|
| 256 |
errDictBins[ "All" ][ bin ] += errDictBins[ sample ][ bin ]**2
|
| 257 |
pass
|
| 258 |
|
| 259 |
averageWeight = 1
|
| 260 |
if not binsumUnscaled == 0.:
|
| 261 |
averageWeight = binsum / binsumUnscaled
|
| 262 |
|
| 263 |
errDict[ sample ] = math.sqrt( binsumUnscaled ) * averageWeight
|
| 264 |
sum += errDict[ sample ]**2
|
| 265 |
pass
|
| 266 |
|
| 267 |
errDict[ "All" ] = math.sqrt( sum )
|
| 268 |
|
| 269 |
### warning! errDictBins[ "All" ][ bin ] is squared!!
|
| 270 |
### errDictBins[ sample ][ bin ] is not!
|
| 271 |
return errDict, errDictBins
|
| 272 |
def getStackErrQCD( self ):
|
| 273 |
errDict = {}
|
| 274 |
errDictBins = {}
|
| 275 |
errDictBins[ "All" ] = {}
|
| 276 |
sum = 0.
|
| 277 |
for sample in self.histDict.keys():
|
| 278 |
errDictBins[ sample ] = {}
|
| 279 |
err = ROOT.Double( 0 )
|
| 280 |
integral = self.histDict[ sample ].IntegralAndError( 0, self.histDict[ sample ].GetNbinsX() + 1,
|
| 281 |
err )
|
| 282 |
errDict[ sample ] = err
|
| 283 |
sum += err**2
|
| 284 |
|
| 285 |
for bin in range( 1, self.histDict[ sample ].GetNbinsX() + 1 ):
|
| 286 |
errDictBins[ sample ][ bin ] = self.histDict[ sample ].GetBinError( bin )
|
| 287 |
if not bin in errDictBins[ "All" ].keys():
|
| 288 |
errDictBins[ "All" ][ bin ] = 0
|
| 289 |
pass
|
| 290 |
errDictBins[ "All" ][ bin ] += errDictBins[ sample ][ bin ]**2
|
| 291 |
pass
|
| 292 |
pass
|
| 293 |
|
| 294 |
errDict[ "All" ] = math.sqrt( sum )
|
| 295 |
|
| 296 |
return errDict, errDictBins
|
| 297 |
pass
|
| 298 |
|
| 299 |
|
| 300 |
class RealTauPred:
|
| 301 |
def __init__( self,
|
| 302 |
histDict,
|
| 303 |
reco={},
|
| 304 |
prompt={},
|
| 305 |
notau={},
|
| 306 |
hadrBR = 1 - 0.352,
|
| 307 |
hadrBRErr = 0.,
|
| 308 |
eff={},
|
| 309 |
):
|
| 310 |
self.histDict = histDict
|
| 311 |
#self.acc = acc
|
| 312 |
self.reco = reco
|
| 313 |
self.prompt = prompt
|
| 314 |
self.notau = notau
|
| 315 |
self.hadrBR = hadrBR
|
| 316 |
self.hadrBRErr = hadrBRErr
|
| 317 |
self.eff = eff
|
| 318 |
self.predDict = self.getPred()
|
| 319 |
self.errUpDict = self.getErrDirection( "Up" )
|
| 320 |
self.errDwnDict = self.getErrDirection( "Dwn" )
|
| 321 |
self.predHistDict = self.getPredHist()
|
| 322 |
#self.PredHistDictErrUp =
|
| 323 |
pass
|
| 324 |
|
| 325 |
def getPred( self, varyReco="No", varyPrompt="No", varyNotau="No", varyEff="No" ):
|
| 326 |
predDict = {}
|
| 327 |
for hist in self.histDict.keys():
|
| 328 |
predDict[ hist ] = {}
|
| 329 |
sum = 0.
|
| 330 |
for sample in self.histDict[ hist ].keys():
|
| 331 |
|
| 332 |
#acc = 1.
|
| 333 |
#if sample in self.acc.keys():
|
| 334 |
# acc = self.acc[ sample ].eff
|
| 335 |
# if varyAcc == "Up":
|
| 336 |
# acc += self.acc[ sample ].errUp
|
| 337 |
# elif varyAcc == "Dwn":
|
| 338 |
# acc -= self.acc[ sample ].errDwn
|
| 339 |
# pass
|
| 340 |
# pass
|
| 341 |
reco = 1.
|
| 342 |
if sample in self.reco.keys():
|
| 343 |
reco = self.reco[ sample ].eff
|
| 344 |
if varyReco == "Up":
|
| 345 |
reco += self.reco[ sample ].errUp
|
| 346 |
elif varyReco == "Dwn":
|
| 347 |
reco -= self.reco[ sample ].errDwn
|
| 348 |
pass
|
| 349 |
pass
|
| 350 |
prompt = 1.
|
| 351 |
if sample in self.prompt.keys():
|
| 352 |
prompt = self.prompt[ sample ].eff
|
| 353 |
if varyPrompt == "Up":
|
| 354 |
prompt += self.prompt[ sample ].errUp
|
| 355 |
elif varyPrompt == "Dwn":
|
| 356 |
prompt -= self.prompt[ sample ].errDwn
|
| 357 |
pass
|
| 358 |
pass
|
| 359 |
notau = 1.
|
| 360 |
if sample in self.notau.keys():
|
| 361 |
reco = self.notau[ sample ].eff
|
| 362 |
if varyNotau == "Up":
|
| 363 |
notau += self.notau[ sample ].errUp
|
| 364 |
elif varyNotau == "Dwn":
|
| 365 |
notau -= self.notau[ sample ].errDwn
|
| 366 |
pass
|
| 367 |
pass
|
| 368 |
eff = 1.
|
| 369 |
if sample in self.eff.keys():
|
| 370 |
eff = self.eff[ sample ].eff
|
| 371 |
if varyEff == "Up":
|
| 372 |
eff += self.eff[ sample ].errUp
|
| 373 |
elif varyEff == "Dwn":
|
| 374 |
eff -= self.eff[ sample ].errDwn
|
| 375 |
pass
|
| 376 |
pass
|
| 377 |
|
| 378 |
predDict[ hist ][ sample ] = self.histDict[ hist ][ sample ].Integral()
|
| 379 |
|
| 380 |
#if( varyPrompt == "No" and varyReco == "No" and varyNotau == "No" and varyEff == "No" ):
|
| 381 |
# print "Prediction ", sample, ": used acc=", acc, " eff=", eff, " hadrBR=", self.hadrBR
|
| 382 |
# print "used nEvts=",predDict[ hist ][ sample ]
|
| 383 |
# pass
|
| 384 |
|
| 385 |
if not reco == 0.:
|
| 386 |
predDict[ hist ][ sample ] *= 1/reco
|
| 387 |
pass
|
| 388 |
predDict[ hist ][ sample ] *= prompt
|
| 389 |
predDict[ hist ][ sample ] *= notau
|
| 390 |
predDict[ hist ][ sample ] *= self.hadrBR
|
| 391 |
predDict[ hist ][ sample ] *= eff
|
| 392 |
|
| 393 |
|
| 394 |
sum += predDict[ hist ][ sample ]
|
| 395 |
pass
|
| 396 |
|
| 397 |
predDict[ hist ][ "All" ] = sum
|
| 398 |
pass
|
| 399 |
|
| 400 |
return predDict
|
| 401 |
|
| 402 |
def getPredHist( self, varyReco="No", varyPrompt="No", varyNotau="No", varyEff="No" ):
|
| 403 |
|
| 404 |
predDict = {}
|
| 405 |
for hist in self.histDict.keys():
|
| 406 |
predDict[ hist ] = {}
|
| 407 |
for sample in self.histDict[ hist ].keys():
|
| 408 |
|
| 409 |
#acc = 1.
|
| 410 |
#if sample in self.acc.keys():
|
| 411 |
# acc = self.acc[ sample ].eff
|
| 412 |
# if varyAcc == "Up":
|
| 413 |
# acc += self.acc[ sample ].errUp
|
| 414 |
# elif varyAcc == "Dwn":
|
| 415 |
# acc -= self.acc[ sample ].errDwn
|
| 416 |
# pass
|
| 417 |
# pass
|
| 418 |
reco = 1.
|
| 419 |
if sample in self.reco.keys():
|
| 420 |
reco = self.reco[ sample ].eff
|
| 421 |
if varyReco == "Up":
|
| 422 |
reco += self.reco[ sample ].errUp
|
| 423 |
elif varyReco == "Dwn":
|
| 424 |
reco -= self.reco[ sample ].errDwn
|
| 425 |
pass
|
| 426 |
pass
|
| 427 |
prompt = 1.
|
| 428 |
if sample in self.prompt.keys():
|
| 429 |
prompt = self.prompt[ sample ].eff
|
| 430 |
if varyPrompt == "Up":
|
| 431 |
prompt += self.prompt[ sample ].errUp
|
| 432 |
elif varyPrompt == "Dwn":
|
| 433 |
prompt -= self.prompt[ sample ].errDwn
|
| 434 |
pass
|
| 435 |
pass
|
| 436 |
notau = 1.
|
| 437 |
if sample in self.notau.keys():
|
| 438 |
reco = self.notau[ sample ].eff
|
| 439 |
if varyNotau == "Up":
|
| 440 |
notau += self.notau[ sample ].errUp
|
| 441 |
elif varyNotau == "Dwn":
|
| 442 |
notau -= self.notau[ sample ].errDwn
|
| 443 |
pass
|
| 444 |
pass
|
| 445 |
eff = 1.
|
| 446 |
if sample in self.eff.keys():
|
| 447 |
eff = self.eff[ sample ].eff
|
| 448 |
if varyEff == "Up":
|
| 449 |
eff += self.eff[ sample ].errUp
|
| 450 |
elif varyEff == "Dwn":
|
| 451 |
eff -= self.eff[ sample ].errDwn
|
| 452 |
pass
|
| 453 |
pass
|
| 454 |
|
| 455 |
predDict[ hist ][ sample ] = self.histDict[ hist ][ sample ].Clone()
|
| 456 |
#if not acc == 0.:
|
| 457 |
# predDict[ hist ][ sample ].Scale( 1/acc )
|
| 458 |
# pass
|
| 459 |
if not reco == 0.:
|
| 460 |
predDict[ hist ][ sample ].Scale( 1./reco )
|
| 461 |
pass
|
| 462 |
predDict[ hist ][ sample ].Scale( prompt )
|
| 463 |
predDict[ hist ][ sample ].Scale( notau )
|
| 464 |
predDict[ hist ][ sample ].Scale( self.hadrBR )
|
| 465 |
predDict[ hist ][ sample ].Scale( eff )
|
| 466 |
|
| 467 |
#if( varyAcc == "No" and varyEff == "No" ):
|
| 468 |
# print "Prediction hist ", sample, ": used acc=", acc, " eff=", eff, " hadrBR=", self.hadrBR
|
| 469 |
# pass
|
| 470 |
pass
|
| 471 |
pass
|
| 472 |
return predDict
|
| 473 |
|
| 474 |
def getErrDirectionHist( self, direction ):
|
| 475 |
|
| 476 |
pass
|
| 477 |
|
| 478 |
|
| 479 |
def getErrDirection( self, direction ):
|
| 480 |
"""
|
| 481 |
Get the prediction for varying the acceptans and the efficiency up
|
| 482 |
"""
|
| 483 |
#predDictAccDir = self.getPred( varyAcc=direction )
|
| 484 |
predDictRecoDir = self.getPred( varyReco=direction )
|
| 485 |
predDictPromptDir = self.getPred( varyPrompt=direction )
|
| 486 |
predDictNotauDir = self.getPred( varyNotau=direction )
|
| 487 |
predDictEffDir = self.getPred( varyEff=direction )
|
| 488 |
|
| 489 |
errDirDict = {}
|
| 490 |
for hist in self.histDict.keys():
|
| 491 |
errDirDict[ hist ] = {}
|
| 492 |
for sample in self.histDict[ hist ].keys():
|
| 493 |
#errAccDir = abs( self.predDict[ hist ][ sample ] - predDictAccDir[ hist ][ sample ] )
|
| 494 |
errRecoDir = abs( self.predDict[ hist ][ sample ] - predDictRecoDir[ hist ][ sample ] )
|
| 495 |
errPromptDir = abs( self.predDict[ hist ][ sample ] - predDictPromptDir[ hist ][ sample ] )
|
| 496 |
errNotauDir = abs( self.predDict[ hist ][ sample ] - predDictNotauDir[ hist ][ sample ] )
|
| 497 |
errEffDir = abs( self.predDict[ hist ][ sample ] - predDictEffDir[ hist ][ sample ] )
|
| 498 |
##errDir = math.sqrt( errAccDir**2 + errEffDir**2 )
|
| 499 |
errDir = math.sqrt( errRecoDir**2 + errPromptDir**2 + errNotauDir**2 + errEffDir**2 )
|
| 500 |
errDirDict[ hist ][ sample ] = errDir
|
| 501 |
pass
|
| 502 |
#errAccDir = abs( self.predDict[ hist ][ "All" ] - predDictAccDir[ hist ][ "All" ] )
|
| 503 |
errRecoDir = abs( self.predDict[ hist ][ "All" ] - predDictRecoDir[ hist ][ "All" ] )
|
| 504 |
errPromptDir = abs( self.predDict[ hist ][ "All" ] - predDictPromptDir[ hist ][ "All" ] )
|
| 505 |
errNotauDir = abs( self.predDict[ hist ][ "All" ] - predDictNotauDir[ hist ][ "All" ] )
|
| 506 |
errEffDir = abs( self.predDict[ hist ][ "All" ] - predDictEffDir[ hist ][ "All" ] )
|
| 507 |
#errDir = math.sqrt( errAccDir**2 + errEffDir**2 )
|
| 508 |
errDir = math.sqrt( errRecoDir**2 + errPromptDir**2 + errNotauDir**2 + errEffDir**2 )
|
| 509 |
errDirDict[ hist ][ "All" ] = errDir
|
| 510 |
pass
|
| 511 |
return errDirDict
|
| 512 |
pass
|
| 513 |
|
| 514 |
class RatioData:
|
| 515 |
def __init__( self, dataHist, fakeHist, realHist, max=2.48, min=0.02, mcHistErrDict={}, colorErr=1,
|
| 516 |
xmin=None, xmax=None, title=None, markerStyle=20 ):
|
| 517 |
self.allHist = fakeHist.Clone()
|
| 518 |
self.allHist.Add( realHist )
|
| 519 |
|
| 520 |
#if not xmin == None and not xmax == None:
|
| 521 |
# self.allHist.GetXaxis().SetRangeUser( xmin, xmax )
|
| 522 |
# dataHist.GetXaxis().SetRangeUser( xmin, xmax )
|
| 523 |
# pass
|
| 524 |
|
| 525 |
self.Ratio = Ratio( dataHist=dataHist, mcHist=self.allHist, max=max, min=min, mcHistErrDict=mcHistErrDict,
|
| 526 |
colorUp=colorErr, xmin=xmin, xmax=xmax, title=title, markerStyle=markerStyle )
|
| 527 |
pass
|
| 528 |
|
| 529 |
def drawRatio( self, pad ):
|
| 530 |
self.Ratio.drawRatio( pad )
|
| 531 |
pass
|
| 532 |
|
| 533 |
class Ratio:
|
| 534 |
|
| 535 |
def __init__( self, dataHist, mcHist,max=1.98, min=0.02, mcHistErrDict = {}, colorUp=ROOT.kCyan + 2,
|
| 536 |
xmax=None, xmin=None, title=None, markerStyle=22, xtitle=None ):
|
| 537 |
self.dataHist = dataHist
|
| 538 |
self.mcHist = mcHist
|
| 539 |
self.max = max
|
| 540 |
self.min = min
|
| 541 |
self.mcHistErrDict = mcHistErrDict
|
| 542 |
self.colorUp = colorUp
|
| 543 |
self.xmax = xmax
|
| 544 |
self.xmin = xmin
|
| 545 |
self.title = title
|
| 546 |
self.markerStyle = markerStyle
|
| 547 |
self.xtitle=xtitle
|
| 548 |
self.graph, self.histErrUp, self.histErrDwn = self.getRatioGraph( )
|
| 549 |
pass
|
| 550 |
|
| 551 |
def getRatioGraph( self ):
|
| 552 |
|
| 553 |
print self.dataHist.ClassName()
|
| 554 |
|
| 555 |
if self.dataHist.ClassName() == "TH1F":
|
| 556 |
return self.getRatioGraphHists()
|
| 557 |
elif self.dataHist.ClassName() == "TGraphAsymmErrors":
|
| 558 |
return self.getRatioGraphGraphs()
|
| 559 |
else:
|
| 560 |
print "Unknown histogram type.Doing nothing."
|
| 561 |
pass
|
| 562 |
return None
|
| 563 |
|
| 564 |
|
| 565 |
def getRatioGraphGraphs( self ):
|
| 566 |
#newHist = self.dataHist.Clone()
|
| 567 |
|
| 568 |
newHist = ROOT.TGraphAsymmErrors()
|
| 569 |
for point in range( 0, self.dataHist.GetN() ):
|
| 570 |
#print "-----",point
|
| 571 |
xdata = ROOT.Double( 0 )
|
| 572 |
ydata = ROOT.Double( 0 )
|
| 573 |
self.dataHist.GetPoint( point, xdata,ydata )
|
| 574 |
xmc = ROOT.Double( 0 )
|
| 575 |
ymc = ROOT.Double( 0 )
|
| 576 |
self.mcHist.GetPoint( point, xmc,ymc )
|
| 577 |
#print ymc
|
| 578 |
if not ymc == 0:
|
| 579 |
newHist.SetPoint( point, xdata, ydata/ymc )
|
| 580 |
ydataErrUp = self.dataHist.GetErrorYhigh( point )
|
| 581 |
ydataErrLow = self.dataHist.GetErrorYlow( point )
|
| 582 |
ymcErrUp = self.mcHist.GetErrorYhigh( point )
|
| 583 |
ymcErrLow = self.mcHist.GetErrorYlow( point )
|
| 584 |
xdataErrUp = self.dataHist.GetErrorXhigh( point )
|
| 585 |
xdataErrLow = self.dataHist.GetErrorXlow( point )
|
| 586 |
errUp = math.sqrt( ( ydataErrUp/ymc )**2 +
|
| 587 |
( ( ymcErrUp * ydata )/ymc**2 )**2 )
|
| 588 |
errLow = math.sqrt( ( ydataErrLow/ymc )**2 +
|
| 589 |
( ( ymcErrLow * ydata )/ymc**2 )**2 )
|
| 590 |
|
| 591 |
newHist.SetPointError( point, xdataErrLow, xdataErrUp, errLow, errUp )
|
| 592 |
pass
|
| 593 |
pass
|
| 594 |
|
| 595 |
newHist.GetYaxis().SetTitle( "data/mc" )
|
| 596 |
newHist.GetYaxis().CenterTitle()
|
| 597 |
newHist.UseCurrentStyle()
|
| 598 |
newHist.SetMarkerStyle( self.markerStyle )
|
| 599 |
newHist.GetXaxis().SetLabelSize( 0.10 )
|
| 600 |
newHist.GetYaxis().SetLabelSize( 0.06 )
|
| 601 |
newHist.GetXaxis().SetTitleSize( 0.1 )
|
| 602 |
newHist.GetYaxis().SetTitleSize( 0.06 )
|
| 603 |
newHist.GetYaxis().SetTitleOffset( 0.7 )
|
| 604 |
newHist.GetXaxis().SetTitle( self.xtitle )
|
| 605 |
newHist.SetMarkerColor( self.colorUp )
|
| 606 |
#newHist.Divide( self.mcHist )
|
| 607 |
|
| 608 |
histErrUp = None
|
| 609 |
histErrDwn = None
|
| 610 |
return newHist, histErrUp, histErrDwn
|
| 611 |
|
| 612 |
def getRatioGraphHists( self ):
|
| 613 |
|
| 614 |
newHist = self.dataHist.Clone()
|
| 615 |
newHist.GetYaxis().SetTitle( "data/mc" )
|
| 616 |
if not self.title == None:
|
| 617 |
newHist.GetYaxis().SetTitle( self.title )
|
| 618 |
pass
|
| 619 |
newHist.GetYaxis().CenterTitle()
|
| 620 |
newHist.Sumw2()
|
| 621 |
newHist.SetDirectory( 0 )
|
| 622 |
newHist.UseCurrentStyle()
|
| 623 |
newHist.SetMarkerStyle( self.markerStyle )
|
| 624 |
newHist.GetXaxis().SetLabelSize( 0.10 )
|
| 625 |
newHist.GetYaxis().SetLabelSize( 0.06 )
|
| 626 |
newHist.GetXaxis().SetTitleSize( 0.1 )
|
| 627 |
newHist.GetYaxis().SetTitleSize( 0.06 )
|
| 628 |
newHist.GetYaxis().SetTitleOffset( 0.7 )
|
| 629 |
newHist.Divide( self.mcHist )
|
| 630 |
#### error for "data"
|
| 631 |
for i in range( 1, newHist.GetNbinsX() + 1 ):
|
| 632 |
if not self.mcHist.GetBinContent( i ) == 0.:
|
| 633 |
newHist.SetBinError( i, self.dataHist.GetBinError( i ) / self.mcHist.GetBinContent( i ) )
|
| 634 |
else:
|
| 635 |
newHist.SetBinError( i, 0 )
|
| 636 |
pass
|
| 637 |
pass
|
| 638 |
### error for "mc"
|
| 639 |
histErrUp = ROOT.TH1F( "errUp" + self.mcHist.GetName(),
|
| 640 |
"errUp" + self.mcHist.GetName(),
|
| 641 |
self.mcHist.GetNbinsX(), self.mcHist.GetXaxis().GetXmin(),\
|
| 642 |
self.mcHist.GetXaxis().GetXmax() )
|
| 643 |
histErrDwn = ROOT.TH1F( "errDwn" + self.mcHist.GetName(),
|
| 644 |
"errDwn" + self.mcHist.GetName(),
|
| 645 |
self.mcHist.GetNbinsX(), self.mcHist.GetXaxis().GetXmin(),\
|
| 646 |
self.mcHist.GetXaxis().GetXmax() )
|
| 647 |
|
| 648 |
histErrUp.GetYaxis().SetTitle( "data/mc" )
|
| 649 |
if not self.title == None:
|
| 650 |
histErrUp.GetYaxis().SetTitle( self.title )
|
| 651 |
pass
|
| 652 |
histErrUp.GetYaxis().CenterTitle()
|
| 653 |
histErrUp.SetDirectory( 0 )
|
| 654 |
histErrUp.UseCurrentStyle()
|
| 655 |
histErrUp.GetXaxis().SetLabelSize( 0.10 )
|
| 656 |
histErrUp.GetYaxis().SetLabelSize( 0.06 )
|
| 657 |
histErrUp.GetXaxis().SetTitleSize( 0.1 )
|
| 658 |
histErrUp.GetYaxis().SetTitleSize( 0.06 )
|
| 659 |
histErrUp.GetYaxis().SetTitleOffset( 0.7 )
|
| 660 |
histErrUp.GetXaxis().SetTitle( self.dataHist.GetXaxis().GetTitle() )
|
| 661 |
histErrUp.SetTitle( "" )
|
| 662 |
histErrDwn.GetYaxis().SetTitle( "data/mc" )
|
| 663 |
if not self.title == None:
|
| 664 |
histErrDwn.GetYaxis().SetTitle( self.title )
|
| 665 |
pass
|
| 666 |
histErrDwn.GetYaxis().CenterTitle()
|
| 667 |
histErrDwn.SetDirectory( 0 )
|
| 668 |
histErrDwn.UseCurrentStyle()
|
| 669 |
histErrDwn.GetXaxis().SetLabelSize( 0.10 )
|
| 670 |
histErrDwn.GetYaxis().SetLabelSize( 0.06 )
|
| 671 |
histErrDwn.GetXaxis().SetTitleSize( 0.1 )
|
| 672 |
histErrDwn.GetYaxis().SetTitleSize( 0.06 )
|
| 673 |
histErrDwn.GetYaxis().SetTitleOffset( 0.7 )
|
| 674 |
histErrDwn.GetXaxis().SetTitle( self.dataHist.GetXaxis().GetTitle() )
|
| 675 |
histErrDwn.SetTitle( "" )
|
| 676 |
|
| 677 |
print "mc hist nbins=", self.mcHist.GetNbinsX()
|
| 678 |
|
| 679 |
if not self.mcHistErrDict == {}:
|
| 680 |
for i in range( 1, self.mcHist.GetNbinsX() + 1 ):
|
| 681 |
#print "bin ", i, "histDict content= ", self.mcHistErrDict[ i ]
|
| 682 |
errmc = math.sqrt( self.mcHistErrDict[ i ] )
|
| 683 |
if not self.mcHist.GetBinContent( i ) == 0.:
|
| 684 |
err = errmc * self.dataHist.GetBinContent( i ) / ( self.mcHist.GetBinContent( i ) )**2
|
| 685 |
#err = errmc / ( self.mcHist.GetBinContent( i ) )**2
|
| 686 |
#print "bin ", i, " : ", err
|
| 687 |
histErrUp.SetBinContent( i, 1 + err )
|
| 688 |
histErrDwn.SetBinContent( i, 1 - err )
|
| 689 |
pass
|
| 690 |
else:
|
| 691 |
histErrDwn.SetBinContent( i, 1 )
|
| 692 |
pass
|
| 693 |
pass
|
| 694 |
pass
|
| 695 |
|
| 696 |
#histErrUp.UseCurrentStyle()
|
| 697 |
#histErrDwn.UseCurrentStyle()
|
| 698 |
|
| 699 |
newHist.GetYaxis().SetRangeUser( self.min, self.max )
|
| 700 |
histErrUp.GetYaxis().SetRangeUser( self.min, self.max )
|
| 701 |
histErrDwn.GetYaxis().SetRangeUser( self.min, self.max )
|
| 702 |
histErrUp.SetFillStyle( 3354 )
|
| 703 |
histErrUp.SetFillColor( self.colorUp )
|
| 704 |
histErrUp.SetLineColor( self.colorUp )
|
| 705 |
histErrDwn.SetFillColor( 10 )
|
| 706 |
histErrDwn.SetLineColor( self.colorUp )
|
| 707 |
histErrDwn.SetFillStyle( 1001 )
|
| 708 |
|
| 709 |
return newHist, histErrUp, histErrDwn
|
| 710 |
|
| 711 |
def getRatioGraphInvert( self ):
|
| 712 |
|
| 713 |
newHist = self.mcHist.Clone()
|
| 714 |
newHist.GetYaxis().SetTitle( "mc/data" )
|
| 715 |
if not self.title == None:
|
| 716 |
newHist.GetYaxis().SetTitle( self.title )
|
| 717 |
pass
|
| 718 |
newHist.GetYaxis().CenterTitle()
|
| 719 |
newHist.Sumw2()
|
| 720 |
newHist.SetDirectory( 0 )
|
| 721 |
newHist.UseCurrentStyle()
|
| 722 |
newHist.SetMarkerStyle( 22 )
|
| 723 |
newHist.GetXaxis().SetLabelSize( 0.10 )
|
| 724 |
newHist.GetYaxis().SetLabelSize( 0.06 )
|
| 725 |
newHist.GetXaxis().SetTitleSize( 0.1 )
|
| 726 |
newHist.GetYaxis().SetTitleSize( 0.06 )
|
| 727 |
newHist.GetYaxis().SetTitleOffset( 0.7 )
|
| 728 |
newHist.Divide( self.dataHist )
|
| 729 |
#for i in range( 1, newHist.GetNbinsX() + 1 ):
|
| 730 |
# if not self.mcHist.GetBinContent( i ) == 0.:
|
| 731 |
# newHist.SetBinError( i, self.dataHist.GetBinError( i ) / self.mcHist.GetBinContent( i ) )
|
| 732 |
# else:
|
| 733 |
# newHist.SetBinError( i, 0 )
|
| 734 |
# pass
|
| 735 |
# pass
|
| 736 |
|
| 737 |
newHist.GetYaxis().SetRangeUser( self.min, self.max )
|
| 738 |
|
| 739 |
return newHist
|
| 740 |
|
| 741 |
|
| 742 |
def drawRatio( self, pad ):
|
| 743 |
"""
|
| 744 |
Adds a pad to the existing, where the ratio should be drawn
|
| 745 |
"""
|
| 746 |
|
| 747 |
logx = pad.GetLogx()
|
| 748 |
left = pad.GetLeftMargin()
|
| 749 |
right = pad.GetRightMargin()
|
| 750 |
|
| 751 |
pad.SetBottomMargin( 0.36 )
|
| 752 |
pad.SetRightMargin(right)
|
| 753 |
pad.SetLeftMargin(left)
|
| 754 |
pad.SetBorderMode(0)
|
| 755 |
pad.SetBorderSize(0)
|
| 756 |
|
| 757 |
self.rPad = ROOT.TPad("rPad","",0,0,1,.361)
|
| 758 |
self.rPad.Draw()
|
| 759 |
self.rPad.cd()
|
| 760 |
self.rPad.SetLogy(0)
|
| 761 |
self.rPad.SetLogx(logx)
|
| 762 |
self.rPad.SetTicky(1)
|
| 763 |
self.rPad.SetTopMargin(0.01)
|
| 764 |
self.rPad.SetBottomMargin(0.45)
|
| 765 |
self.rPad.SetRightMargin(right)
|
| 766 |
self.rPad.SetLeftMargin(left)
|
| 767 |
self.rPad.SetBorderSize(0)
|
| 768 |
self.rPad.SetBorderMode(0)
|
| 769 |
|
| 770 |
#self.graphComb.Draw( "E2" )
|
| 771 |
option = ""
|
| 772 |
if not self.histErrUp == None:
|
| 773 |
if not self.xmax == None and not self.xmin == None:
|
| 774 |
self.histErrUp.GetXaxis().SetRangeUser( self.xmin, self.xmax )
|
| 775 |
self.histErrDwn.GetXaxis().SetRangeUser( self.xmin, self.xmax )
|
| 776 |
pass
|
| 777 |
|
| 778 |
self.histErrUp.GetYaxis().SetNdivisions( 505 )
|
| 779 |
self.histErrUp.GetYaxis().SetTickLength( ROOT.gStyle.GetTickLength("Y")/0.8 )
|
| 780 |
self.histErrUp.GetYaxis().SetLabelSize( 0.1 )
|
| 781 |
self.histErrUp.GetYaxis().SetTitleOffset( 0.6 )
|
| 782 |
self.histErrUp.GetYaxis().SetTitleSize( 0.1 )
|
| 783 |
|
| 784 |
if not self.mcHistErrDict == {}:
|
| 785 |
self.histErrUp.Draw( )
|
| 786 |
option = "same"
|
| 787 |
self.histErrDwn.Draw( option )
|
| 788 |
self.histErrUp.Draw( "axissame" )
|
| 789 |
#option = "same"
|
| 790 |
pass
|
| 791 |
pass
|
| 792 |
else:
|
| 793 |
option = "A"
|
| 794 |
self.graph.Draw( "ep" + option )
|
| 795 |
if not self.xmax == None and not self.xmin == None:
|
| 796 |
self.graph.GetXaxis().SetRangeUser( self.xmin, self.xmax )
|
| 797 |
pass
|
| 798 |
|
| 799 |
self.rPad.Update()
|
| 800 |
### draw a line at 1:
|
| 801 |
#line = TLine( self.dataHist.GetXaxis().GetXmin(), self.dataHist.GetXaxis().GetXmax() ,1,1 )
|
| 802 |
self.line = ROOT.TLine( self.dataHist.GetXaxis().GetXmin(), 1, self.dataHist.GetXaxis().GetXmax(), 1 )
|
| 803 |
if not self.xmin == None and not self.xmax == None:
|
| 804 |
self.line = ROOT.TLine( self.xmin, 1, self.xmax, 1 )
|
| 805 |
self.line.SetLineColor( 1 )
|
| 806 |
self.line.Draw( "same" )
|
| 807 |
self.rPad.Update()
|
| 808 |
|
| 809 |
pass
|
| 810 |
|
| 811 |
pass
|
| 812 |
|
| 813 |
class OneBinHist:
|
| 814 |
def __init__( self, dict ):
|
| 815 |
self.histDict = dict
|
| 816 |
pass
|
| 817 |
|
| 818 |
def getOneBinHists( self ):
|
| 819 |
dict = {}
|
| 820 |
for key in self.histDict.keys():
|
| 821 |
nBins = self.histDict[ key ].GetNbinsX()
|
| 822 |
dict[ key ] = self.histDict[ key ].Clone()
|
| 823 |
dict[ key ].Rebin( nBins )
|
| 824 |
pass
|
| 825 |
return dict
|
| 826 |
pass
|
| 827 |
|
| 828 |
class StatErrorHists:
|
| 829 |
def __init__( self, histDict, histErrDictUp,histErrDictDwn, colorUp = 1, colorDwn = 0 ):
|
| 830 |
self.histDict = histDict
|
| 831 |
self.histErrDictUp = histErrDictUp
|
| 832 |
self.histErrDictDwn = histErrDictDwn
|
| 833 |
self.colorUp = colorUp
|
| 834 |
self.colorDwn = colorDwn
|
| 835 |
pass
|
| 836 |
|
| 837 |
def getErrHists( self, histname ):
|
| 838 |
histUpDict = {}
|
| 839 |
histDwnDict = {}
|
| 840 |
name = "All"
|
| 841 |
#for hist in self.histDict.keys():
|
| 842 |
#print "NbinsX=", self.histDict.GetNbinsX(), " min=", self.histDict.GetXaxis().GetXmin(), " max=", \
|
| 843 |
# self.histDict.GetXaxis().GetXmax()
|
| 844 |
histUpDict[ name ] = ROOT.TH1F( name + histname + "Up", name + histname + "Up", self.histDict.GetNbinsX(),
|
| 845 |
self.histDict.GetXaxis().GetXmin(), self.histDict.GetXaxis().GetXmax() )
|
| 846 |
histUpDict[ name ].SetDirectory( 0 )
|
| 847 |
histUpDict[ name ].UseCurrentStyle()
|
| 848 |
histUpDict[ name ].SetFillStyle( 3354 )
|
| 849 |
histUpDict[ name ].SetFillColor( self.colorUp )
|
| 850 |
histUpDict[ name ].SetLineColor( self.colorUp )
|
| 851 |
histDwnDict[ name ] = ROOT.TH1F( name + histname + "Dwn", name + histname + "Dwn", self.histDict.GetNbinsX(),
|
| 852 |
self.histDict.GetXaxis().GetXmin(), self.histDict.GetXaxis().GetXmax() )
|
| 853 |
histDwnDict[ name ].SetDirectory( 0 )
|
| 854 |
histDwnDict[ name ].UseCurrentStyle()
|
| 855 |
histDwnDict[ name ].SetFillStyle( 1001 )
|
| 856 |
histDwnDict[ name ].SetFillColor( self.colorDwn )
|
| 857 |
histDwnDict[ name ].SetLineColor( self.colorUp )
|
| 858 |
for bin in self.histErrDictUp[ name ].keys():
|
| 859 |
#print bin, " : ", self.histDict.GetBinCenter( int( bin ) ), "=", self.histDict.GetBinContent( int( bin ) )
|
| 860 |
#print self.histDict.GetBinCenter( int( bin ) )
|
| 861 |
#xval = self.histDict.GetBinCenter( int( bin ) )
|
| 862 |
yval = self.histDict.GetBinContent( int( bin ) )
|
| 863 |
#errxval = self.histDict.GetBinWidth( int( bin ) )
|
| 864 |
erryvalUp = math.sqrt( self.histErrDictUp[ name ][ int( bin ) ] )
|
| 865 |
erryvalDwn = math.sqrt( self.histErrDictDwn[ name ][ int( bin ) ] )
|
| 866 |
#print "err=", erryval, " Up=", yval + erryval, " Dwn=", yval - erryval
|
| 867 |
histUpDict[ name ].SetBinContent( bin, yval + erryvalUp )
|
| 868 |
histDwnDict[ name ].SetBinContent( bin, yval - erryvalDwn )
|
| 869 |
pass
|
| 870 |
return histUpDict, histDwnDict
|
| 871 |
|
| 872 |
pass
|
| 873 |
|
| 874 |
pass
|
| 875 |
|
| 876 |
class MakeErrGraph:
|
| 877 |
#def __init__( self, fakeDict, fakeDictU, realDict, realDictU, systDict={} ):
|
| 878 |
def __init__( self, fakeDict, realDict, errDict={} ):
|
| 879 |
self.fakeDict = fakeDict
|
| 880 |
self.realDict = realDict
|
| 881 |
#self.fakeDictU = fakeDictU
|
| 882 |
#self.realDictU = realDictU
|
| 883 |
#self.binErrDict, self.binMeanDict, self.binXDict, self.binWidthDict = self.getBinErrs()
|
| 884 |
self.binMeanDict, self.binXDict, self.binWidthDict = self.getBinErrs()
|
| 885 |
self.errDict = errDict
|
| 886 |
pass
|
| 887 |
|
| 888 |
def getBinErrs( self ):
|
| 889 |
#binErrDict = {}
|
| 890 |
binMeanDict = {}
|
| 891 |
binXDict = {}
|
| 892 |
binWidthDict = {}
|
| 893 |
for hist in self.fakeDict.keys():
|
| 894 |
#binErrDict[ hist ] = {}
|
| 895 |
binMeanDict[ hist ] = {}
|
| 896 |
binXDict[ hist ] = {}
|
| 897 |
binWidthDict[ hist ] = {}
|
| 898 |
for sample in self.fakeDict[ hist ].keys():
|
| 899 |
for bin in range( 1, self.fakeDict[ hist ][ sample ].GetNbinsX() + 1 ):
|
| 900 |
### initialize bin error with 0
|
| 901 |
#if not bin in binErrDict[ hist ].keys():
|
| 902 |
# binErrDict[ hist ][ bin ] = 0
|
| 903 |
# pass
|
| 904 |
|
| 905 |
rName = hist
|
| 906 |
if hist == "JetPt":
|
| 907 |
rName = "Tau1Pt"
|
| 908 |
pass
|
| 909 |
elif hist == "JetPhi":
|
| 910 |
rName = "Tau1Phi"
|
| 911 |
pass
|
| 912 |
elif hist == "JetEta":
|
| 913 |
rName = "Tau1Eta"
|
| 914 |
pass
|
| 915 |
elif hist == "nJets":
|
| 916 |
rName = "NumberOfJets"
|
| 917 |
pass
|
| 918 |
|
| 919 |
#binContentFakeU = self.fakeDictU[ hist + "Unweighted" ][ sample ].GetBinContent( bin )
|
| 920 |
binContentFake = self.fakeDict[ hist ][ sample ].GetBinContent( bin )
|
| 921 |
#binContentRealU = self.realDictU[ rName + "Unscaled" ][ sample ].GetBinContent( bin )
|
| 922 |
binContentReal = self.realDict[ rName ][ sample ].GetBinContent( bin )
|
| 923 |
|
| 924 |
binMeanDict[ hist ][ bin ] = binContentFake + binContentReal
|
| 925 |
binXDict[ hist ][ bin ] = self.fakeDict[ hist ][ sample ].GetBinCenter( bin )
|
| 926 |
binWidthDict[ hist ][ bin ] = self.fakeDict[ hist ][ sample ].GetBinWidth( bin )
|
| 927 |
|
| 928 |
#binErrFake = 0
|
| 929 |
#if not binContentFakeU == 0:
|
| 930 |
# binErrFake = math.sqrt( binContentFakeU ) * binContentFake/binContentFakeU
|
| 931 |
# pass
|
| 932 |
#binErrReal = 0
|
| 933 |
#if not binContentRealU == 0:
|
| 934 |
# binErrReal = math.sqrt( binContentRealU ) * binContentReal/binContentRealU
|
| 935 |
# pass
|
| 936 |
|
| 937 |
#binErr = math.sqrt( binErrFake**2 + binErrReal**2 )
|
| 938 |
#binErrDict[ hist ][ bin ] = binErr
|
| 939 |
pass
|
| 940 |
pass
|
| 941 |
pass
|
| 942 |
#return binErrDict, binMeanDict, binXDict, binWidthDict
|
| 943 |
return binMeanDict, binXDict, binWidthDict
|
| 944 |
|
| 945 |
#def addSystErr( self ):
|
| 946 |
# for hist in self.fakeDict.keys():
|
| 947 |
# if hist in self.systDict.keys():
|
| 948 |
# for bin in self.binErrDict[ hist ].keys():
|
| 949 |
# self.binErrDict[ hist ][ bin ] = math.sqrt( self.binErrDict[ hist ][ bin ]**2 +\
|
| 950 |
# self.systDict[ hist ][ bin ] )
|
| 951 |
# pass
|
| 952 |
# pass
|
| 953 |
# pass
|
| 954 |
|
| 955 |
|
| 956 |
def getErrGraphs( self ):
|
| 957 |
graphDict = {}
|
| 958 |
#for hist in self.binErrDict.keys():
|
| 959 |
for hist in self.errDict[ "Up" ].keys():
|
| 960 |
graphDict[ hist ] = ROOT.TGraphAsymmErrors()
|
| 961 |
#graphDict[ hist ].SetDirectory( 0 )
|
| 962 |
#for bin in self.binErrDict[ hist ].keys():
|
| 963 |
rName = hist
|
| 964 |
if hist == "Tau1Pt":
|
| 965 |
rName = "JetPt"
|
| 966 |
pass
|
| 967 |
elif hist == "Tau1Phi":
|
| 968 |
rName = "JetPhi"
|
| 969 |
pass
|
| 970 |
elif hist == "Tau1Eta":
|
| 971 |
rName = "JetEta"
|
| 972 |
pass
|
| 973 |
elif hist == "NumberOfJets":
|
| 974 |
rName = "nJets"
|
| 975 |
pass
|
| 976 |
for bin in self.errDict[ "Up" ][ hist ][ "All" ].keys():
|
| 977 |
### note: hist binning starts at 1,
|
| 978 |
### graph binning at 0 !
|
| 979 |
graphDict[ hist ].SetPoint( bin - 1,
|
| 980 |
self.binXDict[ rName ][ bin ],
|
| 981 |
self.binMeanDict[ rName ][ bin ] )
|
| 982 |
graphDict[ hist ].SetPointError( bin - 1,
|
| 983 |
self.binWidthDict[ rName ][ bin ] / 2,
|
| 984 |
self.binWidthDict[ rName ][ bin ] / 2,
|
| 985 |
math.sqrt( self.errDict[ "Dwn" ][ hist ][ "All" ][ bin ] ),
|
| 986 |
math.sqrt( self.errDict[ "Up" ][ hist ][ "All" ][ bin ] )
|
| 987 |
)
|
| 988 |
pass
|
| 989 |
pass
|
| 990 |
|
| 991 |
return graphDict
|
| 992 |
|
| 993 |
pass
|
| 994 |
|
| 995 |
class SystErr:
|
| 996 |
def __init__( self, hist, theSystDict={} ):
|
| 997 |
self.hist = hist
|
| 998 |
self.theSystDict = theSystDict
|
| 999 |
pass
|
| 1000 |
|
| 1001 |
def getGlobalSystErr( self ):
|
| 1002 |
errUp2 = 0.
|
| 1003 |
errDwn2 = 0.
|
| 1004 |
entries = self.hist.Integral()
|
| 1005 |
for syst in self.theSystDict.keys():
|
| 1006 |
errUp2 += ( self.theSystDict[ syst ][ "Up" ] * entries )**2
|
| 1007 |
errDwn2 += ( self.theSystDict[ syst ][ "Dwn" ] * entries )**2
|
| 1008 |
pass
|
| 1009 |
return ( errUp2, errDwn2 )
|
| 1010 |
|
| 1011 |
def getBinSystErr( self ):
|
| 1012 |
err2Dict = {}
|
| 1013 |
for bin in range( 1, self.hist.GetNbinsX() + 1 ):
|
| 1014 |
err2Dict[ bin ] = {}
|
| 1015 |
err2Dict[ bin ][ "Up" ] = 0.
|
| 1016 |
err2Dict[ bin ][ "Dwn" ] = 0.
|
| 1017 |
for syst in self.theSystDict.keys():
|
| 1018 |
err2Dict[ bin ][ "Up" ] += ( self.theSystDict[ syst ][ "Up" ] *
|
| 1019 |
self.hist.GetBinContent( bin ) )**2
|
| 1020 |
err2Dict[ bin ][ "Dwn" ] += ( self.theSystDict[ syst ][ "Dwn" ] *
|
| 1021 |
self.hist.GetBinContent( bin ) )**2
|
| 1022 |
pass
|
| 1023 |
pass
|
| 1024 |
return err2Dict
|
| 1025 |
|
| 1026 |
def getSystErr( self ):
|
| 1027 |
return ( self.getGlobalSystErr(), self.getBinSystErr() )
|
| 1028 |
pass
|
| 1029 |
|
| 1030 |
class SystFromHist:
|
| 1031 |
def __init__( self, dict ):
|
| 1032 |
self.histDict = dict
|
| 1033 |
pass
|
| 1034 |
|
| 1035 |
def getSystDict( self ):
|
| 1036 |
systDict = {}
|
| 1037 |
for hist in self.histDict.keys():
|
| 1038 |
systDict[ hist ] = 0.
|
| 1039 |
firstKey = self.histDict[ hist ].keys()[ 0 ]
|
| 1040 |
for bin in range( 1, self.histDict[ hist ][ firstKey ].GetNbinsX() + 1 ):
|
| 1041 |
linEntry = 0.
|
| 1042 |
for sample in self.histDict[ hist ].keys():
|
| 1043 |
entry = self.histDict[ hist ][ sample ].GetBinContent( bin )
|
| 1044 |
linEntry += entry
|
| 1045 |
pass
|
| 1046 |
systDict[ hist ] += linEntry**2
|
| 1047 |
pass
|
| 1048 |
pass
|
| 1049 |
return systDict
|
