def DoROC(rateToEffMap, signalDataset, rocSaveName, bSaveAuxHistos=False):
p0 = m_plotter.Plotter(Verbose=False, BatchMode=True)
RateHistoList, EffHistoList = p0.GetROCHistoLists(rateToEffMap)
### Rates
p1 = m_plotter.Plotter(Verbose=False, BatchMode=True)
p1.SetBoolUseDatasetAsLegEntry(False)
p1.AddDataset("MinBias", datasetPaths["MinBias"])
p1.AddHisto(RateHistoList)
p1.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p1.SaveHistos(bSaveAuxHistos, savePath, saveFormats)
### Efficiencies
p2 = m_plotter.Plotter(Verbose=False, BatchMode=True)
p2.SetBoolUseDatasetAsLegEntry(False)
p2.AddDataset(signalDataset, datasetPaths[signalDataset])
p2.AddHisto(EffHistoList)
p2.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p2.SaveHistos(bSaveAuxHistos, savePath, saveFormats)
### ROCs
p3 = m_plotter.Plotter(Verbose=False, BatchMode=True)
p3.SetBoolUseDatasetAsLegEntry(False)
p3.ConvertToROC(p1.GetHistos(),
p2.GetHistos(),
decimals=3,
bDrawZValues=True,
**ROC)
p3.DrawMultigraph(rocSaveName, **ROC)
p3.SetTLegendHeader(signalDataset, "")
p3.SaveHistos(True, savePath, saveFormats)
return
def DoROC(rateToEffMap, signalDataset, saveNameExt):
p0 = m_plotter.Plotter(bVerbose, bBatchMode)
RateHistoList, EffHistoList = p0.GetROCHistoLists(rateToEffMap)
### Background Rate
p1 = m_plotter.Plotter(bVerbose, bBatchMode)
p1.AddDataset("MinBias:default", inputPath_def % ("nugun"))
p1.AddDataset("MinBias:3#sigma", inputPath_3s % ("nugun"))
p1.AddDataset("MinBias:5#sigma", inputPath_5s % ("nugun"))
p1.AddDataset("MinBias:10#sigma", inputPath_10s % ("nugun"))
p1.AddHisto(RateHistoList)
p1.SetBoolUseDatasetAsLegEntry(True)
p1.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p1.SetTLegendHeader("MinBias", "") #saveNameExt
p1.SaveHistos(True, mySavePath + signalDataset + "/", mySaveFormats)
### Per-Event Efficiencies
p2 = m_plotter.Plotter(bVerbose, bBatchMode)
if signalDataset == "HPlus160":
signalDatasetFileExt = "liteHiggs"
elif signalDataset == "HPlus200":
signalDatasetFileExt = "heavyHiggs"
else:
signalDatasetFileExt = signalDataset
p2.AddDataset(signalDataset + ":default",
inputPath_def % (signalDatasetFileExt))
p2.AddDataset(signalDataset + ":3#sigma",
inputPath_3s % (signalDatasetFileExt))
p2.AddDataset(signalDataset + ":5#sigma",
inputPath_5s % (signalDatasetFileExt))
p2.AddDataset(signalDataset + ":10#sigma",
inputPath_10s % (signalDatasetFileExt))
p2.SetBoolUseDatasetAsLegEntry(True)
p2.AddHisto(EffHistoList)
p2.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p2.SetTLegendHeader(signalDataset, "") #saveNameExt
p2.SaveHistos(True, mySavePath + signalDataset + "/", mySaveFormats)
### ROC Curves
p3 = m_plotter.Plotter(bVerbose, bBatchMode)
p3.AddDataset(signalDataset + ":default",
inputPath_def % (signalDatasetFileExt))
p3.AddDataset(signalDataset + ":3#sigma",
inputPath_3s % (signalDatasetFileExt))
p3.AddDataset(signalDataset + ":5#sigma",
inputPath_5s % (signalDatasetFileExt))
p3.AddDataset(signalDataset + ":10#sigma",
inputPath_10s % (signalDatasetFileExt))
p3.SetBoolUseDatasetAsLegEntry(True)
p3.ConvertToROC(p1.GetHistos(), p2.GetHistos(), decimals=3)
p3.DrawMultigraph("SingleTau_ROC_" + saveNameExt, **ROC)
p3.SetTLegendHeader(signalDataset, "") #saveNameExt
p3.SaveHistos(bSavePlots, mySavePath + signalDataset + "/", mySaveFormats)
return
def DoPlotsWithTF1(hList,
datasetList,
myF1Exp,
xMin,
xMax,
tauDecayMode="",
kwargs=None):
saveExt = ""
p = m_plotter.Plotter(bVerbose, bBatchMode)
for dataset in datasetList:
if tauDecayMode == "":
p.AddDataset(dataset + ":all", datasetPaths_all[dataset])
p.AddDataset(dataset + ":1pr", datasetPaths_1p[dataset])
p.AddDataset(dataset + ":3pr", datasetPaths_3p[dataset])
else:
saveExt = "_" + tauDecayMode
p.AddDataset(dataset + ":" + tauDecayMode,
datasetPaths_all[dataset])
p.SetBoolUseDatasetAsLegEntry(True)
p.AddHisto(hList)
p.AddTF1(myF1Exp, xMin, xMax, kwargs)
p.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p.SetTLegendHeader(dataset, tauDecayMode)
p.SaveHistos(bSavePlots, savePath, saveFormats, saveExt)
return
def PrintTreeCreationParameters():
inputPath = "/Users/attikis/my_work/cms/lxplus/hltaus/rootFiles/TTrees/CMSSW_6_2_0_SLHC12_patch1/TkTauFromCaloAnalyzer_v2/SigTk5_R01to04_TTI2023Upg14D_DefaultTrackingSequence_08Aug2014_164642/"
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
p.GetDatasets("TTI2023Upg14D", inputPath, ["minbias"])
p.PrintPSet("TkTauFromCaloAnalyzer/configInfo/parameterSet")
return
def PrintTreeCreationParameters():
inputPath = "/Users/attikis/hltaus/rootFiles/TTrees/CMSSW_6_2_0_SLHC12_patch1/TkTauFromCaloAnalyzer_v3/TTracks_TTPixelTracks_CaloTausCorr_FitParam5_PtMin2_TTI2023Upg14D_TestVersionWithPixelBugs_28Jan2015_144735/"
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
p.GetDatasets("TTI2023Upg14D", inputPath, ["minbias"])
p.PrintPSet("TkTauFromCaloNTupleMaker/configInfo/parameterSet")
return
def PrintTreePSet():
'''
'''
baseDir = "/Users/attikis/hltaus/rootFiles/TTrees/CMSSW_6_2_0_SLHC12_patch1/TkTauFromCaloAnalyzer_v3/"
inputPath = baseDir + "AllTracks_CaloTausCorr_FitParam4_TTI2023Upg14D_17Feb2015_165244/"
p = m_plotter.Plotter( Verbose=False, BatchMode=True )
p.GetDatasets("TTI2023Upg14D", inputPath, ["minbias"])
p.PrintPSet("TkTauFromCaloAnalyzer/configInfo/parameterSet")
return
def DoPlots(hList, datasetList, datasetPaths, bColourPalette=False, saveExt=""):
p = m_plotter.Plotter(False, True)
for dataset in datasetList:
p.AddDataset( dataset, datasetPaths[dataset] )
p.AddHisto(hList)
p.SetBoolUseDatasetAsLegEntry(True)
p.EnableColourPalette(bColourPalette)
p.Draw(THStackDrawOpt="nostack", bStackInclusive = False, bAddReferenceHisto = False)
p.GetTLegend().SetHeader( "" )
p.SaveHistos(True, mySavePath, mySaveFormats, saveExt)
return
def DoPlots(hList, datasetList):
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
for dataset in datasetList:
p.AddDataset(dataset, datasetPaths[dataset])
p.AddHisto(hList)
p.EnableColourPalette(True)
p.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p.GetTLegend().SetHeader(
m_datasets.DatasetClass().ConvertDatasetToLatex(dataset))
p.SaveHistos(True, savePath, saveFormats, "_" + tkWP)
return
def DoEfficiency(hList, datasetList, cutDirList, saveExt=""):
for cutDir in cutDirList:
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
for dataset in datasetList:
p.AddDataset(dataset, datasetPaths[dataset])
p.SetBoolUseDatasetAsLegEntry(True)
p.AddHisto(hList)
p.EnableColourPalette(False)
p.DrawEfficiency(cutDir, "binomial")
p.GetTLegend().SetHeader("L1TkTau")
p.SaveHistos(True, savePath, saveFormats, saveExt)
return
def DoPlots(hList, datasetList, bColourPalette=False, saveExt=""):
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
for dataset in datasetList:
p.AddDataset(dataset, datasetPaths[dataset])
p.AddHisto(hList)
p.EnableColourPalette(bColourPalette)
p.Draw(THStackDrawOpt="nostack",
bStackInclusive=False,
bAddReferenceHisto=False)
if len(hList) > 1:
p.SetTLegendHeader(datasetList[0], "")
p.SaveHistos(True, savePath, saveFormats, saveExt)
return
def DoEfficiency(hList, datasetList, cutDirList, saveExt=""):
for cutDir in cutDirList:
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
for dataset in datasetList:
p.AddDataset(dataset, datasetPaths[dataset])
p.AddHisto(hList)
p.EnableColourPalette(True)
p.DrawEfficiency(cutDir, "binomial")
p.GetTLegend().SetHeader(
m_datasets.DatasetClass().ConvertDatasetToLatex(dataset))
p.SaveHistos(True, savePath, saveFormats, "_" + tkWP)
return
def DoPlots(hList, datasetList, saveExt="", bLegHeader=None):
p = m_plotter.Plotter(Verbose=False, BatchMode=True)
for dataset in datasetList:
p.AddDataset(dataset, datasetPaths[dataset])
p.AddHisto(hList)
p.SetBoolUseDatasetAsLegEntry(False)
p.EnableColourPalette(True)
p.Draw(THStackDrawOpt="nostack",
bStackInclusive=False,
bAddReferenceHisto=False)
if bLegHeader != None:
p.GetTLegend().SetHeader(bLegHeader)
p.SaveHistos(True, savePath, saveFormats, saveExt)
return
def DoPlots(histoObjectList, dataset):
'''
'''
p = m_plotter.Plotter(bVerbose, bBatchMode)
p.EnableColourPalette(True)
p.AddDataset(dataset + ":default", inputPath_def % ("nugun"))
p.AddDataset(dataset + ":3#sigma", inputPath_3s % ("nugun"))
p.AddDataset(dataset + ":5#sigma", inputPath_5s % ("nugun"))
p.AddDataset(dataset + ":10#sigma", inputPath_10s % ("nugun"))
p.AddHisto(histoObjectList)
p.SetBoolUseDatasetAsLegEntry(True)
p.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p.GetTLegend().SetHeader(dataset)
p.SaveHistos(bSavePlots, mySavePath, mySaveFormats)
return
def DoPlots(hList, dataset, saveNameExt):
p = m_plotter.Plotter(bVerbose, bBatchMode)
if dataset == "HPlus160":
datasetFileExt = "liteHiggs"
elif dataset == "HPlus200":
datasetFileExt = "heavyHiggs"
else:
datasetFileExt = dataset
p.AddDataset(dataset + ":default", inputPath_def % (datasetFileExt))
p.AddDataset(dataset + ":3#sigma", inputPath_3s % (datasetFileExt))
p.AddDataset(dataset + ":5#sigma", inputPath_5s % (datasetFileExt))
p.AddDataset(dataset + ":10#sigma", inputPath_10s % (datasetFileExt))
p.SetBoolUseDatasetAsLegEntry(True)
p.AddHisto(hList)
p.Draw(THStackDrawOpt="nostack", bStackInclusive=False)
p.SetTLegendHeader(dataset, "")
#p.AppendToCanvasName("_" + saveNameExt)
p.SaveHistos(True, mySavePath + dataset + "/", mySaveFormats)
return
def DoProfile(hList, datasetList, tauDecayMode, profileAxis="x"):
saveExt = ""
p = m_plotter.Plotter(bVerbose, bBatchMode)
for dataset in datasetList:
if tauDecayMode == "":
p.AddDataset(dataset + ":all", datasetPaths_all[dataset])
p.AddDataset(dataset + ":1pr", datasetPaths_1p[dataset])
p.AddDataset(dataset + ":3pr", datasetPaths_3p[dataset])
else:
saveExt = "_" + tauDecayMode
p.AddDataset(dataset + ":" + tauDecayMode,
datasetPaths_all[dataset])
p.SetBoolUseDatasetAsLegEntry(True)
p.AddHisto(hList)
p.Draw2DProfile(THStackDrawOpt="nostack",
bStackInclusive=False,
ProfileAxis=profileAxis,
firstBin=1,
lastBin=-1)
p.SetTLegendHeader(dataset, "")
p.SaveHistos(bSavePlots, savePath, saveFormats, saveExt)
return
def DoEfficiency(hList, datasetList, cutDirList, tauDecayMode=""):
saveExt = ""
for histo in hList:
for cutDir in cutDirList:
p = m_plotter.Plotter(bVerbose, bBatchMode)
for dataset in datasetList:
if tauDecayMode == "":
p.AddDataset(dataset + ":all", datasetPaths_all[dataset])
p.AddDataset(dataset + ":1pr", datasetPaths_1p[dataset])
p.AddDataset(dataset + ":3pr", datasetPaths_3p[dataset])
else:
saveExt = "_" + tauDecayMode
p.AddDataset(dataset + ":" + tauDecayMode,
datasetPaths_all[dataset])
p.AddHisto([histo])
p.SetBoolUseDatasetAsLegEntry(True)
p.EnableColourPalette(True)
p.DrawEfficiency(cutDir, "binomial")
p.SetTLegendHeader(dataset, tauDecayMode)
p.SaveHistos(bSavePlots, savePath, saveFormats, saveExt)
return
