def ReadNonClosureHistograms(dirName):
if not dirName.endswith("/"):
dirName = dirName + "/"
# Run over each file (one per jet definition)
histos = {}
files = sorted(glob.glob(dirName + "*.root"))
for aFileName in files:
# Determine the jet definition
jetDef = ""
for aFileDef, aJetDef in jetDefDict.iteritems():
if aFileDef in aFileName:
jetDef = aJetDef
break
if jetDef == "":
print "Failed to determine jet definition for file:", aFileName
return None
histos[jetDef] = {}
# Read in the histogram from the file
inFile = TFile(aFileName, "READ")
for aName, aSystName in SystematicNameDictionary.iteritems():
systematicName = aSystName + "_" + jetDef
histo = inFile.Get(aName)
if histo is None:
print "Failed to get histogram:", systematicName
return None
histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
histo, systematicName)
histoSym.SetDirectory(0)
histos[jetDef][aSystName] = histoSym
# Also add a blank nonclosure histo for Pythia8
# Also add a blank Closeby histo for both AFII and Pythia8
etaBins = [
min(ProviderHistoHelpers.GetDefaultEtaBins()), 0,
max(ProviderHistoHelpers.GetDefaultEtaBins())
]
histos[jetDef][
"RelativeNonClosure_Pythia8"] = ProviderHistoHelpers.MakeProviderHisto(
"RelativeNonClosure_Pythia8",
ptBins=[15, 1000, 2500],
etaBins=etaBins)
histos[jetDef]["RelativeNonClosure_Pythia8"].SetDirectory(0)
histos[jetDef][
"CloseBy_Pythia8"] = ProviderHistoHelpers.MakeProviderHisto(
"CloseBy_Pythia8",
ptBins=[15, 1000, 2500],
etaBins=etaBins)
histos[jetDef]["CloseBy_Pythia8"].SetDirectory(0)
histos[jetDef][
"CloseBy_AFII"] = ProviderHistoHelpers.MakeProviderHisto(
"CloseBy_AFII", ptBins=[15, 1000, 2500], etaBins=etaBins)
histos[jetDef]["CloseBy_AFII"].SetDirectory(0)
# Done reading, close the file
inFile.Close()
return histos
def ReadCrossCalValidityHistograms(dirName):
if not dirName.endswith("/"):
dirName = dirName + "/"
# Run over each file (one per jet definition)
histos = {}
files = sorted(glob.glob(dirName + "*.root"))
for aFileName in files:
# Determine the jet definition
jetDef = ""
for aFileDef, aJetDef in jetDefDict.iteritems():
if aFileDef in aFileName:
jetDef = aJetDef
break
if jetDef == "":
print "Failed to determine jet definition for file:", aFileName
return None
histos[jetDef] = {}
# Read in the histograms of interest from the file
inFile = TFile(aFileName, "READ")
for histName in inFile.GetKeyNames():
if 'valid' not in histName:
continue
dictkey = histName.replace("valid_", "")
systematicName = "valid_" + SystematicNameDictionary[
dictkey] + "_" + jetDef
histo = inFile.Get(histName)
if histo is None:
print "Failed to get histogram:", systematicName
return None
histo.SetName(systematicName + "_old")
# Histo has different range, extend it
histoNew = ProviderHistoHelpers.ExtendPtRangeOfHisto(
histo, "valid_" + systematicName)
histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
histoNew, "valid_" + systematicName)
histoSym.SetDirectory(0)
histos[jetDef][SystematicNameDictionary[dictkey]] = histoSym
# Done reading, close the file
inFile.Close()
return histos
def ReadFlavourHistograms(dirName):
if not dirName.endswith("/"):
dirName = dirName + "/"
# Run over each file (one per jet definition for LQ/Gluon and one per jet definition for HvP)
histos = {}
files = sorted(glob.glob(dirName+"*.root"))
for aFileName in files:
# Determine the jet definition
jetDef = ""
for aFileDef,aJetDef in jetDefDict.iteritems():
if aFileDef in aFileName:
jetDef = aJetDef
break
if jetDef == "":
print "Failed to determine jet definition for file:",aFileName
return None
if jetDef not in histos:
histos[jetDef] = {}
# Read in the histograms from the file
# Note that we want all the histograms, but with this scheme not everything in the dict is in each file
# So instead loop over all histos in the file
inFile = TFile(aFileName,"READ")
for histName in inFile.GetKeyNames():
systematicName = SystematicNameDictionary[histName]+"_"+jetDef
histo = inFile.Get(histName)
if histo is None:
print "Failed to get histogram:",systematicName
return None
histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(histo,systematicName)
histoSym.SetDirectory(0)
histos[jetDef][SystematicNameDictionary[histName]] = histoSym
# Done reading, close the file
inFile.Close()
return histos
def ReadCrossCalHistograms(dirName,
scaleCrossCalVal=1.0,
freezeCrossCalAll=False,
freezeCrossCalBarrel=False,
freezeCrossCalCentral=False):
if not dirName.endswith("/"):
dirName = dirName + "/"
# Run over each file (one per jet definition)
histos = {}
files = sorted(glob.glob(dirName + "*.root"))
for aFileName in files:
# Determine the jet definition
jetDef = ""
for aFileDef, aJetDef in jetDefDict.iteritems():
if aFileDef in aFileName:
jetDef = aJetDef
break
if jetDef == "":
print "Failed to determine jet definition for file:", aFileName
return None
histos[jetDef] = {}
# Read in the histograms of interest from the file
inFile = TFile(aFileName, "READ")
for histName in inFile.GetKeyNames():
# Skip validity histograms.
if 'valid' in histName:
continue
systematicName = SystematicNameDictionary[histName] + "_" + jetDef
histo = inFile.Get(histName)
if histo is None:
print "Failed to get histogram:", systematicName
return None
histo.SetName(systematicName + "_old")
if freezeCrossCalAll or freezeCrossCalBarrel or freezeCrossCalCentral:
# Loop over pT bins.
for xbin in range(histo.GetNbinsX() + 2):
# Loop out in eta.
freezeval = 0
for ybin in range(histo.GetNbinsY() + 2):
#print "Bin at pT ",histo.GetXaxis().GetBinCenter(xbin)," and [ylow,y,yhigh] = ", binYLow, histo.GetYaxis().GetBinCenter(ybin), binYHigh
# Store bin contents as we go out: last one is one we want as frozen value.
eta = abs(histo.GetYaxis().GetBinCenter(ybin))
if eta < 0.8:
freezeval = histo.GetBinContent(xbin, ybin)
else:
if (eta < 2.5 and
freezeCrossCalBarrel) or freezeCrossCalAll:
histo.SetBinContent(
xbin, ybin, freezeval * scaleCrossCalVal)
else:
histo.SetBinContent(xbin, ybin, 0)
# Histo has different range, extend it
histoNew = ProviderHistoHelpers.ExtendPtRangeOfHisto(
histo, systematicName)
histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
histoNew, systematicName)
histoSym.SetDirectory(0)
# If this is the 50 vs 25 ns histogram, we need to separate it by eta region.
if '50nsVs25ns' in histName:
CentralHist = histo.Clone()
CentralHist.SetName("Central_holder")
ForwardHist = histo.Clone()
ForwardHist.SetName("Forward_holder")
# Loop over pT bins.
for xbin in range(histo.GetNbinsX() + 2):
# Loop out in eta.
freezeval = 0
for ybin in range(histo.GetNbinsY() + 2):
#print "Bin at pT ",histo.GetXaxis().GetBinCenter(xbin)," and [ylow,y,yhigh] = ", binYLow, histo.GetYaxis().GetBinCenter(ybin), binYHigh
# Store bin contents as we go out: last one is one we want as frozen value.
eta = abs(histo.GetYaxis().GetBinCenter(ybin))
if eta < 0.8:
freezeval = histo.GetBinContent(xbin, ybin)
ForwardHist.SetBinContent(xbin, ybin, 0.0)
CentralHist.SetBinContent(xbin, ybin, freezeval)
else:
ForwardHist.SetBinContent(xbin, ybin, freezeval)
CentralHist.SetBinContent(xbin, ybin, 0.0)
# Now do symmetrization for these hists and write to file.
NewCentral = ProviderHistoHelpers.ExtendPtRangeOfHisto(
CentralHist, systematicName + "holder")
CentralSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
NewCentral,
SystematicNameDictionary[histName] + "_Central_" + jetDef)
CentralSym.SetDirectory(0)
histos[jetDef][SystematicNameDictionary[histName] +
"_Central"] = CentralSym
NewForward = ProviderHistoHelpers.ExtendPtRangeOfHisto(
ForwardHist, systematicName + "holder")
ForwardSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
NewForward,
SystematicNameDictionary[histName] + "_Forward_" + jetDef)
ForwardSym.SetDirectory(0)
histos[jetDef][SystematicNameDictionary[histName] +
"_Forward"] = ForwardSym
# If this is the early data histogram, we need to deactivate it in the central region for 25 ns data.
if 'EarlyData' in histName:
ForwardHist = histo.Clone()
ForwardHist.SetName("EarlyData_Forward_holder")
# Loop over pT bins.
for xbin in range(histo.GetNbinsX() + 2):
# Loop out in eta.
for ybin in range(histo.GetNbinsY() + 2):
#print "Bin at pT ",histo.GetXaxis().GetBinCenter(xbin)," and [ylow,y,yhigh] = ", binYLow, histo.GetYaxis().GetBinCenter(ybin), binYHigh
# Store bin contents as we go out: last one is one we want as frozen value.
eta = abs(histo.GetYaxis().GetBinCenter(ybin))
if eta < 0.8:
ForwardHist.SetBinContent(xbin, ybin, 0.0)
# Now do symmetrization and write to file.
NewForward = ProviderHistoHelpers.ExtendPtRangeOfHisto(
ForwardHist, systematicName + "holder")
ForwardSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
NewForward,
SystematicNameDictionary[histName] + "_Forward_" + jetDef)
ForwardSym.SetDirectory(0)
histos[jetDef][SystematicNameDictionary[histName] +
"_Forward"] = ForwardSym
histos[jetDef][SystematicNameDictionary[histName]] = histoSym
else:
histos[jetDef][SystematicNameDictionary[histName]] = histoSym
# Done reading, close the file
inFile.Close()
return histos
def ReadNonClosureHistograms(dirName, freezepT=False):
if dirName.endswith(".root"):
files = [dirName]
elif not dirName.endswith("/"):
dirName = dirName + "/"
files = sorted(glob.glob(dirName + "*.root"))
# Run over each file (one per jet definition)
histos = {}
for aFileName in files:
# Determine the jet definition
jetDef = ""
for aFileDef, aJetDef in jetDefDict.iteritems():
if aFileDef in aFileName:
jetDef = aJetDef
break
if jetDef == "":
print "Failed to determine jet definition for file:", aFileName
return None
histos[jetDef] = {}
# Read in the histogram from the file
inFile = TFile(aFileName, "READ")
for aName, aSystName in SystematicNameDictionary.iteritems():
systematicName = aSystName + "_" + jetDef
histo = inFile.Get(aName)
if histo is None:
print "Failed to get histogram:", systematicName
return None
if freezepT:
# Loop over eta bins.
for ybin in range(histo.GetNbinsY() + 2):
# Loop out in pT.
freezeval = 0
for xbin in range(histo.GetNbinsX() + 2):
inbin = histo.GetBinContent(xbin, ybin)
#print "Bin at pT ",histo.GetXaxis().GetBinCenter(xbin)," and [ylow,y,yhigh] = ", binYLow, histo.GetYaxis().GetBinCenter(ybin), binYHigh
# Store bin contents as we go out: last one is one we want as frozen value.
if inbin > -0.5:
freezeval = inbin
else:
histo.SetBinContent(xbin, ybin, freezeval)
histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(
histo, systematicName)
histoSym.SetDirectory(0)
histos[jetDef][aSystName] = histoSym
# Also add a blank nonclosure histo for Pythia8
# Also add a blank Closeby histo for both AFII and Pythia8
etaBins = [
min(ProviderHistoHelpers.GetDefaultEtaBins()), 0,
max(ProviderHistoHelpers.GetDefaultEtaBins())
]
histos[jetDef][
"RelativeNonClosure_MC15"] = ProviderHistoHelpers.MakeProviderHisto(
"RelativeNonClosure_MC15",
ptBins=[15, 1000, 2500],
etaBins=etaBins)
histos[jetDef]["RelativeNonClosure_MC15"].SetDirectory(0)
histos[jetDef][
"CloseBy_Pythia8"] = ProviderHistoHelpers.MakeProviderHisto(
"CloseBy_MC15", ptBins=[15, 1000, 2500], etaBins=etaBins)
histos[jetDef]["CloseBy_Pythia8"].SetDirectory(0)
histos[jetDef][
"CloseBy_AFII"] = ProviderHistoHelpers.MakeProviderHisto(
"CloseBy_AFII", ptBins=[15, 1000, 2500], etaBins=etaBins)
histos[jetDef]["CloseBy_AFII"].SetDirectory(0)
# Done reading, close the file
inFile.Close()
return histos