PrintTrkAnaSummary.py

Go to the documentation of this file.

#!/usr/bin/env python
# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
import sys, os, argparse, ROOT
import pandas as pd
 
# Parsing arguments
commandName = os.path.basename( sys.argv[0] )
summaryDirDefault="InDetTrackPerfMonPlots/TrkAnaEF/Offline/Tracks/"
parser = argparse.ArgumentParser( description = commandName+" options:" )
parser.add_argument( "-t", "--testFile", help="Path to input TEST file" )
parser.add_argument( "-r", "--refFile", default="", help="Path to input REFERENCE file" )
parser.add_argument( "-T", "--testLabel", default="TEST", help="Label for TEST" )
parser.add_argument( "-R", "--refLabel", default="REF", help="Label for REFERENCE" )
parser.add_argument( "-d", "--dirName", default=summaryDirDefault, help="Name of the TDirectory path with plots" )
parser.add_argument( "-o", "--outName", default="TrkAnaSummary.html", help="Name of the output html file" )
MyArgs = parser.parse_args()
 
if not MyArgs.testFile:
    print( "ERROR: input test file not provided" )
    sys.exit(1)
 
data={}
index=[]
 
def processFile( inFileName, dirName, label, data, index, updateIndex=True ):
    sList = []
    inFile = ROOT.TFile.Open( inFileName, "READ" )
 
    
    hs = inFile.Get( dirName+"Multiplicities/summary" )
    if hs:
        for i in range( 1, hs.GetNbinsX()+1 ) :
            if updateIndex : index.append( hs.GetXaxis().GetBinLabel(i) )
            c = hs.GetBinContent(i)
            e = hs.GetBinError(i)
            sList.append( f"{c:.1f} +/- {e:.1f}" )
 
    
    he = inFile.Get( dirName+"Efficiencies/eff_vs_truth_inclusive" )
    if updateIndex : index.append( " " )
    sList.append( " " )
    if updateIndex : index.append( "Eff_vs_truth" )
    if he:
        c = 100*he.GetEfficiency(1)
        eu = 100*he.GetEfficiencyErrorUp(1)
        el = 100*he.GetEfficiencyErrorLow(1)
        sList.append( f"{c:.2f} +{eu:.2f}/-{el:.2f} %" )
    else : sList.append( "-" )
 
    
    het = inFile.Get( dirName+"Efficiencies/Technical/eff_vs_truth_inclusive" )
    if updateIndex : index.append( "Tech_eff_vs_truth" )
    if het:
        c = 100*het.GetEfficiency(1)
        eu = 100*het.GetEfficiencyErrorUp(1)
        el = 100*het.GetEfficiencyErrorLow(1)
        sList.append( f"{c:.2f} +{eu:.2f}/-{el:.2f} %" )
    else : sList.append( "-" )
 
 
    
    hrpt = inFile.Get( dirName+"Resolutions/resolution_pt_vs_truth_inclusive" )
    if updateIndex : index.append( " " )
    sList.append( " " )
    if updateIndex : index.append( "Resolution_pT_vs_truth" )
    if hrpt:
        c = hrpt.GetBinContent(1)
        e = hrpt.GetBinError(1)
        sList.append( f"{c:.2f} +/- {e:.2f} GeV" )
    else : sList.append( "-" )
 
 
    
    hrd0 = inFile.Get( dirName+"Resolutions/resolution_d0_vs_truth_inclusive" )
    if updateIndex : index.append( "Resolution_d0_vs_truth" )
    if hrd0:
        c = hrd0.GetBinContent(1)
        e = hrd0.GetBinError(1)
        sList.append( f"{c:.2f} +/- {e:.2f} mm" )
    else : sList.append( "-" )
 
 
    
    hrz0 = inFile.Get( dirName+"Resolutions/resolution_z0_vs_truth_inclusive" )
    if updateIndex : index.append( "Resolution_z0_vs_truth" )
    if hrz0:
        c = hrz0.GetBinContent(1)
        e = hrz0.GetBinError(1)
        sList.append( f"{c:.2f} +/- {e:.2f} mm" )
    else : sList.append( "-" )
 
 
    
    hf = inFile.Get( dirName+"FakeRates/fakerate_vs_offl_inclusive" )
    if updateIndex : index.append( " " )
    sList.append( " " )
    if updateIndex : index.append( "FakeRate_vs_reco" )
    if hf:
        c = 100*hf.GetEfficiency(1)
        eu = 100*hf.GetEfficiencyErrorUp(1)
        el = 100*hf.GetEfficiencyErrorLow(1)
        sList.append( f"{c:.2f} +{eu:.2f}/-{el:.2f} %" )
    else : sList.append( "-" )
 
 
    
    hd = inFile.Get( dirName+"Duplicates/duplrate_vs_truth_inclusive" )
    if updateIndex : index.append( "DuplicateRate_vs_truth" )
    if hd:
        c = 100*hd.GetEfficiency(1)
        eu = 100*hd.GetEfficiencyErrorUp(1)
        el = 100*hd.GetEfficiencyErrorLow(1)
        sList.append( f"{c:.2f} +{eu:.2f}/-{el:.2f} %" )
    else : sList.append( "-" )
 
 
    
    data.update( { label : sList } )
    inFile.Close()
 
 
 
processFile(
    inFileName  = MyArgs.testFile,
    dirName     = MyArgs.dirName,
    label       = MyArgs.testLabel,
    data        = data,
    index       = index
)
 
 
if MyArgs.refFile :
    processFile(
        inFileName  = MyArgs.refFile,
        dirName     = MyArgs.dirName,
        label       = MyArgs.refLabel,
        data        = data,
        index       = index,
        updateIndex=False
    )
 
 
df = pd.DataFrame( data, index=index )
print( df )
 
 
with open( MyArgs.outName, 'w' ) as f :
    print( df.to_html(), file=f )
 
sys.exit(0)