SCTTracksMonAlg.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
'''@file SCTTracksMonAlg.py
@author Ken Kreul
@date 2019-07-15
@brief Based on AthenaMonitoring/ExampleMonitorAlgorithm.py
'''
 
def SCTTracksMonAlgConfig(flags):
    '''Function to configures some algorithms in the monitoring system.'''
    
    from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
    result = ComponentAccumulator()
 
    # The following class will make a sequence, configure algorithms, and link
    # them to GenericMonitoringTools
    from AthenaMonitoring import AthMonitorCfgHelper
    helper = AthMonitorCfgHelper(flags, 'SCTTracksMonCfg')
 
 
    
    from AthenaConfiguration.ComponentFactory import CompFactory
    from TrkConfig.TrkResidualPullCalculatorConfig import (
        ResidualPullCalculatorCfg)
    myMonAlg = helper.addAlgorithm(CompFactory.SCTTracksMonAlg, 'SCTTracksMonAlg',
                                   ResPullCalc = result.popToolsAndMerge(
                                       ResidualPullCalculatorCfg(flags)))
 
    from AthenaConfiguration.Enums import BeamType
    if flags.Beam.Type is BeamType.Collisions:
        from AthenaMonitoring.FilledBunchFilterToolConfig import FilledBunchFilterToolCfg
        myMonAlg.FilterTools += [result.popToolsAndMerge(FilledBunchFilterToolCfg(flags))]
 
    doTrigger = False
    if not flags.Input.isMC:
        if flags.Trigger.doHLT:
            doTrigger = True
    myMonAlg.doTrigger = doTrigger
 
    # # If for some really obscure reason you need to instantiate an algorithm
    # # yourself, the AddAlgorithm method will still configure the base 
    # # properties and add the algorithm to the monitoring sequence.
    # helper.AddAlgorithm(myExistingAlg)
 
 
    
    myMonAlg.TriggerChain = ''
    # myMonAlg.RandomHist = True
 
    from TrkConfig.TrkTrackSummaryToolConfig import InDetTrackSummaryToolCfg
    myMonAlg.TrackSummaryTool = result.popToolsAndMerge(InDetTrackSummaryToolCfg(flags))
 
    
 
    # # Then, add a tool that doesn't have its own configuration function. In
    # # this example, no accumulator is returned, so no merge is necessary.
    # from MyDomainPackage.MyDomainPackageConf import MyDomainTool
    # myMonAlg.MyDomainTool = MyDomainTool()
 
    # Add a generic monitoring tool (a "group" in old language). The returned 
    # object here is the standard GenericMonitoringTool.
    myMonGroup = helper.addGroup(
        myMonAlg,
        "SCTTracksMonitor",
        "SCT/GENERAL/"
    )
 
    
    regionNames = ["EndCapC", "Barrel", "EndCapA"]
    N_REGIONS = len(regionNames)
    s_triggerNames = ["RNDM", "BPTX", "L1CAL", "TGC", "RPC", "MBTS", "COSM", "Calib"]
    N_TRIGGER_TYPES = len(s_triggerNames)
    N_HIT_BINS =  50
    FIRST_HIT_BIN =  0
    LAST_HIT_BIN  = N_HIT_BINS-FIRST_HIT_BIN-1 # This is already defined in SCT_MonitoringNumbers.h
    myMonGroup.defineHistogram(varname="trk_N", # ; means alias
                               type="TH1F",
                               title="Number of tracks"+";Number of Tracks",
                               path="tracks", # path cannot be "".
                               xbins=400, xmin=0., xmax=4000.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_chi2", # ; means alias
                               type="TH1F",
                               title="Track #chi^{2} div ndf"+";Number of track #chi^{2}/NDF",
                               path="tracks", # path cannot be "".
                               xbins=150, xmin=0., xmax=150.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_d0", # ; means alias
                               type="TH1F",
                               title="Track d0"+";d0 [mm]",
                               path="tracks", # path cannot be "".
                               xbins=160, xmin=-40., xmax=40.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_z0", # ; means alias
                               type="TH1F",
                               title="Track z0"+";z0 [mm]",
                               path="tracks", # path cannot be "".
                               xbins=200, xmin=-200., xmax=200.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_phi", # ; means alias
                               type="TH1F",
                               title="Track Phi"+";#phi [rad]",
                               path="tracks", # path cannot be "".
                               xbins=160, xmin=-4., xmax=4.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_pt", # ; means alias
                               type="TH1F",
                               title="Track P_{T}"+";P_{T} [GeV]",
                               path="tracks", # path cannot be "".
                               xbins=150, xmin=0., xmax=150.,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_sct_hits", # ; means alias
                               type="TH1F",
                               title="SCT HITS per single Track"+";Num of Hits",
                               path="tracks", # path cannot be "".
                               xbins=N_HIT_BINS, xmin=FIRST_HIT_BIN, xmax=LAST_HIT_BIN,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="trk_eta", # ; means alias
                               type="TH1F",
                               title="Track Eta"+";#eta",
                               path="tracks", # path cannot be "".
                               xbins=160, xmin=-4., xmax=4.,
                               opt='kAlwaysCreate')
                                           
    myMonGroup.defineHistogram(varname="trackTriggers", # ; means alias
                               type="TH1I",
                               title="Tracks for different trigger types",
                               path="tracks", # path cannot be "".
                               xbins=N_TRIGGER_TYPES, xmin=-0.5, xmax=7.5, xlabels=s_triggerNames,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="region"+","+"hitsRegion"+";"+"SCTTrackRate", # ; means alias
                               type="TProfile",
                               title="Track per event for SCT regions",
                               path="tracks", # path cannot be "".
                               xbins=3, xmin=0.0, xmax=3.0, xlabels=regionNames,
                               opt='kAlwaysCreate')
    myMonGroup.defineHistogram(varname="tracksPerRegion", # ; means alias
                               type="TH1F",
                               title="Number of tracks in eta regions",
                               path="tracks", # path cannot be "".
                               xbins=N_REGIONS, xmin=0, xmax=N_REGIONS, xlabels=regionNames,
                               opt='kAlwaysCreate')
 
 
    for region in regionNames:
        myMonGroup.defineHistogram(varname="total"+region+"Residual",
                                   title="Overall Residual Distribution for the "+region+";Residual [mm]",
                                   type="TH1F", path="tracks",
                                   xbins=100, xmin=-0.5, xmax=0.5)
        myMonGroup.defineHistogram(varname="total"+region+"Pull",
                                   title="Overall Pull Distribution for the "+region+";Pull",
                                   type="TH1F", path="tracks",
                                   xbins=100, xmin=-5, xmax=5)
    
    
    # # Otherwise, merge with result object and return
    result.merge(helper.result())
    return result
 
if __name__ == "__main__": 
    # Setup logs
    from AthenaCommon.Logging import log
    from AthenaCommon.Constants import INFO
    log.setLevel(INFO)
 
    # Set the Athena configuration flags
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
    flags.Input.Files = ["/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/RecExRecoTest/mc16_13TeV.361022.Pythia8EvtGen_A14NNPDF23LO_jetjet_JZ2W.recon.ESD.e3668_s3170_r10572_homeMade.pool.root"]
    flags.Input.isMC = True
    flags.Output.HISTFileName = 'SCTTracksMonOutput.root'
    flags.GeoModel.Align.Dynamic = False
    flags.Detector.GeometryID = True
    flags.Detector.GeometryPixel = True
    flags.Detector.GeometrySCT = True
    flags.Detector.GeometryTRT = True
    flags.lock()
 
    # Initialize configuration object, add accumulator, merge, and run.
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    cfg = MainServicesCfg(flags)
    cfg.merge(PoolReadCfg(flags))
 
    cfg.merge(SCTTracksMonAlgConfig(flags))
 
    cfg.run()