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/&TrkAnaName&/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( "-a", "--analyses", default="TrkAnaEF", help="Comma-separeted list of track analyses to process" )
parser.add_argument( "-o", "--outName", default="TrkAnaSummary_&TrkAnaName&.html", help="Name of the output html files" )
MyArgs = parser.parse_args()
anaList = MyArgs.analyses.strip().split(',')
 
if not MyArgs.testFile:
    print( "ERROR: input test file not provided" )
    sys.exit(1)
 
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()
 
 
outFile = MyArgs.outName.replace( "_&TrkAnaName&", "" )
if os.path.isfile( outFile ) :
    os.remove( outFile ) 
 
 
for anaName in anaList :
    
    data = {}
    index = []
    anaDirName = MyArgs.dirName.replace( "&TrkAnaName&", anaName )
    anaOutName = MyArgs.outName.replace( "&TrkAnaName&", anaName )
 
    
    processFile(
        inFileName  = MyArgs.testFile,
        dirName     = anaDirName,
        label       = MyArgs.testLabel,
        data        = data,
        index       = index
    )
 
    
    if MyArgs.refFile :
        processFile(
            inFileName  = MyArgs.refFile,
            dirName     = anaDirName,
            label       = MyArgs.refLabel,
            data        = data,
            index       = index,
            updateIndex=False
        )
 
    
    df = pd.DataFrame( data, index=index )
    titleStr = f"Summary for TrackAnalysis = {anaName}:"
    print( f"\n\n---------------\n{titleStr}" )
    print( df )
 
    
    with open( anaOutName, 'w' ) as f :
        print( df.to_html(), file=f )
 
    
    os.system( f"echo \"<br><b>{titleStr}</b><br>\" >> {outFile}" )
    os.system( f"cat {anaOutName} >> {outFile}" )
    os.remove( f"{anaOutName}" ) 
 
sys.exit(0)