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