Tools/HitAnalysis/python/PostIncludes.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
# Adding SiHitValidation for whichever parts of ITk are running
def ITkHitAnalysis(flags):
    from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
    from HitAnalysis.HitAnalysisConfig import ITkPixelHitAnalysisCfg, ITkStripHitAnalysisCfg, PLR_HitAnalysisCfg
 
    result = ComponentAccumulator()
 
    if flags.Detector.EnableITkPixel:
        result.merge(ITkPixelHitAnalysisCfg(flags))
 
    if flags.Detector.EnableITkStrip:
        result.merge(ITkStripHitAnalysisCfg(flags))
 
    if flags.Detector.EnablePLR:
        result.merge(PLR_HitAnalysisCfg(flags))
 
    return result
 
def HGTDHitAnalysis(flags):
    from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
    from HitAnalysis.HitAnalysisConfig import HGTD_HitAnalysisCfg
 
    result = ComponentAccumulator()
 
    if flags.Detector.EnableHGTD:
        result.merge(HGTD_HitAnalysisCfg(flags))
 
    result.getService("THistSvc").Output = [
        "HGTDHitAnalysis DATAFILE='HGTDHitValid.root' OPT='RECREATE'"]
 
    return result
 
 
def IDHitAnalysis(flags): 
 
    from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
    from HitAnalysis.HitAnalysisConfig import PixelHitAnalysisCfg, SCTHitAnalysisCfg, TRTHitAnalysisCfg
    result = ComponentAccumulator()
    if flags.Detector.EnablePixel:
        result.merge(PixelHitAnalysisCfg(flags))
    if flags.Detector.EnableSCT:
        result.merge(SCTHitAnalysisCfg(flags))
    if flags.Detector.EnableTRT:
        result.merge(TRTHitAnalysisCfg(flags))
    return result
 
def MuonHitAnalysis(flags):
    from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
    result = ComponentAccumulator()
    if flags.Muon.usePhaseIIGeoSetup:
        from HitAnalysis.HitAnalysisConfig import xMuonHitAnalysisCfg
 
        if flags.Detector.GeometryMDT:
            result.merge(xMuonHitAnalysisCfg(flags,
                                             name="MdtSimHitTester",
                                             InputKey="xMdtSimHits",
                                             HistPath="xMuonSimHit/MDT/Hits",
                                             techIndex=0))
        if flags.Detector.GeometryRPC:
            result.merge(xMuonHitAnalysisCfg(flags,
                                             name="RpcSimHitTester",
                                             InputKey="xRpcSimHits",
                                             HistPath="xMuonSimHit/RPC/Hits",
                                             techIndex=2))
        if flags.Detector.GeometryTGC:
            result.merge(xMuonHitAnalysisCfg(flags,
                                             name="TgcSimHitTester",
                                             InputKey="xTgcSimHits",
                                             HistPath="xMuonSimHit/TGC/Hits",
                                             techIndex=3))
        if flags.Detector.GeometrysTGC:
            result.merge(xMuonHitAnalysisCfg(flags,
                                             name="sTgcSimHitTester",
                                             InputKey="xStgcSimHits",
                                             HistPath="xMuonSimHit/sTGC/Hits",
                                             techIndex=4))
        if flags.Detector.GeometryMM:
            result.merge(xMuonHitAnalysisCfg(flags,
                                             name="MmSimHitTester",
                                             InputKey="xMmSimHits",
                                             HistPath="xMuonSimHit/MM/Hits",
                                             techIndex=5))
        from MuonPRDTestR4.MuonHitTestConfig import MuonHitTesterCfg
        result.merge(MuonHitTesterCfg(flags, outFile=flags.Output.HISTFileName))
    else:
        if flags.Detector.GeometryMDT:
            from HitAnalysis.HitAnalysisConfig import MDTHitAnalysisCfg
            result.merge(MDTHitAnalysisCfg(flags))
        if flags.Detector.GeometryRPC:
            from HitAnalysis.HitAnalysisConfig import RPCHitAnalysisCfg
            result.merge(RPCHitAnalysisCfg(flags))
        if flags.Detector.GeometryTGC:
            from HitAnalysis.HitAnalysisConfig import TGCHitAnalysisCfg
            result.merge(TGCHitAnalysisCfg(flags))
        if flags.Detector.GeometrysTGC:
            from HitAnalysis.HitAnalysisConfig import sTGCHitAnalysisCfg
            result.merge(sTGCHitAnalysisCfg(flags))
        if flags.Detector.GeometryMM:
            from HitAnalysis.HitAnalysisConfig import MMHitAnalysisCfg
            result.merge(MMHitAnalysisCfg(flags)) 
        if flags.Detector.GeometryCSC:
            from HitAnalysis.HitAnalysisConfig import CSCHitAnalysisCfg
            result.merge(CSCHitAnalysisCfg(flags))
        from MuonPRDTest.HitValAlgSim import HitValAlgSimCfg
        result.merge(HitValAlgSimCfg(flags, outFile=flags.Output.HISTFileName))
 
 
    return result