SiliconPreProcessing.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.Enums import Format
 
def InDetRecPreProcessingSiliconCfg(flags):
    acc = ComponentAccumulator()
    # ------------------------------------------------------------
    #
    # ----------- Data-Preparation stage
    #
    # ------------------------------------------------------------
    #
    # ----------- PrepRawData creation from Raw Data Objects
    #
 
    #
    # --- Slim BCM RDOs by zero-suppressing
    #
    if flags.Detector.EnableBCM:
        if flags.Input.Format is Format.BS:
            from BCM_RawDataByteStreamCnv.BCM_RawDataByteStreamCnvConfig import (
                BCM_RawDataProviderAlgCfg)
            acc.merge(BCM_RawDataProviderAlgCfg(flags))
        from InDetConfig.BCM_ZeroSuppressionConfig import BCM_ZeroSuppressionCfg
        acc.merge(BCM_ZeroSuppressionCfg(flags))
 
    #
    # -- Pixel Clusterization
    #
    if (flags.Detector.EnablePixel and
        (flags.Input.Format is Format.BS
         or 'PixelRDOs' in flags.Input.Collections
         or f'{flags.Overlay.BkgPrefix}PixelRDOs' in flags.Input.Collections)):
        #
        # --- PixelClusterization algorithm
        #
        from InDetConfig.InDetPrepRawDataFormationConfig import (
            PixelClusterizationCfg)
        acc.merge(PixelClusterizationCfg(flags))
        if flags.InDet.doSplitReco:
            from InDetConfig.InDetPrepRawDataFormationConfig import (
                PixelClusterizationPUCfg)
            acc.merge(PixelClusterizationPUCfg(flags))
    #
    # --- SCT Clusterization
    #
    if (flags.Detector.EnableSCT and
        (flags.Input.Format is Format.BS
         or 'SCT_RDOs' in flags.Input.Collections
         or f'{flags.Overlay.BkgPrefix}SCT_RDOs' in flags.Input.Collections)):
 
        #
        # --- SCT_Clusterization algorithm
        #
        from InDetConfig.InDetPrepRawDataFormationConfig import (
            SCTClusterizationCfg)
        acc.merge(SCTClusterizationCfg(flags))
        if flags.InDet.doSplitReco:
            from InDetConfig.InDetPrepRawDataFormationConfig import (
                SCTClusterizationPUCfg)
            acc.merge(SCTClusterizationPUCfg(flags))
 
    #
    # ----------- form SpacePoints from clusters in SCT and Pixels
    #
    #
    from InDetConfig.SiSpacePointFormationConfig import (
        InDetSiTrackerSpacePointFinderCfg)
    acc.merge(InDetSiTrackerSpacePointFinderCfg(flags))
 
    # this truth must only be done if you do PRD and SpacePointformation
    # If you only do the latter (== running on ESD) then the needed input (simdata)
    # is not in ESD but the resulting truth (clustertruth) is already there ...
    if (flags.InDet.doTruth and
        (not flags.Detector.EnableSCT or
         'SCT_SDO_Map' in flags.Input.Collections) and
        (not flags.Detector.EnablePixel or
         'PixelSDO_Map' in flags.Input.Collections)):
 
        from InDetConfig.InDetTruthAlgsConfig import (
            InDetPRD_MultiTruthMakerSiCfg)
        acc.merge(InDetPRD_MultiTruthMakerSiCfg(flags))
        if flags.InDet.doSplitReco:
            from InDetConfig.InDetTruthAlgsConfig import (
                InDetPRD_MultiTruthMakerSiPUCfg)
            acc.merge(InDetPRD_MultiTruthMakerSiPUCfg(flags))
 
    return acc
 
 
def ITkRecPreProcessingSiliconCfg(flags,
                                  *,
                                  previousActsExtension: str = None):
    acc = ComponentAccumulator()
    # ------------------------------------------------------------
    #
    # ----------- Data-Preparation stage
    #
    # ------------------------------------------------------------
    #
    # ----------- PrepRawData creation from Raw Data Objects
    #
 
    #
    # --- Slim BCM RDOs by zero-suppressing
    #
    if flags.Detector.EnableBCMPrime:
        from InDetConfig.BCM_ZeroSuppressionConfig import BCM_ZeroSuppressionCfg
        acc.merge(BCM_ZeroSuppressionCfg(flags))
    #
    # -- Clusterization Algorithms
    #
    if flags.Tracking.ActiveConfig.doAthenaCluster:
        from InDetConfig.InDetPrepRawDataFormationConfig import (
            AthenaTrkClusterizationCfg)
        acc.merge(AthenaTrkClusterizationCfg(flags))
        # if we require HGTD clusters in the ACTS Track reconstruction we have to create the clusters here
        if flags.Acts.useHGTDClusterInTrackFinding and flags.Detector.EnableHGTD:
            from HGTD_Config.HGTD_PrepRawDataFormationConfig import PadClusterizationCfg
            acc.merge(PadClusterizationCfg(flags))
        
    if flags.Tracking.ActiveConfig.doActsCluster:
        # If ACTS clusterization is activated, then schedule RoI creator
        from ActsConfig.ActsRegionsOfInterestConfig import ActsRegionsOfInterestCreatorAlgCfg
        acc.merge(ActsRegionsOfInterestCreatorAlgCfg(flags,
                                                     name=f"{flags.Tracking.ActiveConfig.extension}RegionsOfInterestCreatorAlg"))
 
        # If running FPGA tracking, use ACTS clusterization + FPGA pass-through
        if flags.Tracking.ActiveConfig.doFPGATrack:
            from EFTrackingFPGAIntegration.EFClusterizationConfig import EFPassThroughClusterizationCfg
            acc.merge(EFPassThroughClusterizationCfg(flags))
        else:
            from ActsConfig.ActsClusterizationConfig import ActsClusterizationCfg
            acc.merge(ActsClusterizationCfg(flags, previousActsExtension=previousActsExtension))
 
    #
    # ---  Cluster EDM converters
    #
    if flags.Tracking.ActiveConfig.doAthenaToActsCluster:
        #
        # --- InDet -> xAOD Cluster EDM converter
        #
        from InDetConfig.InDetPrepRawDataFormationConfig import (
            ITkInDetToXAODClusterConversionCfg)
        acc.merge(ITkInDetToXAODClusterConversionCfg(flags))
        # if we need HGTD clusters, we convert them as well
        if flags.Acts.useHGTDClusterInTrackFinding and flags.Detector.EnableHGTD:
            from InDetConfig.InDetPrepRawDataFormationConfig import HGTDInDetToXAODClusterConversionCfg
            acc.merge(HGTDInDetToXAODClusterConversionCfg(flags))
 
        
    if flags.Tracking.ActiveConfig.doActsToAthenaCluster:
        #
        # --- xAOD -> InDet Cluster EDM converter
        #
        from InDetConfig.InDetPrepRawDataFormationConfig import (
            ITkXAODToInDetClusterConversionCfg)
        acc.merge(ITkXAODToInDetClusterConversionCfg(flags))
 
    #
    # ----------- form SpacePoints from clusters in SCT and Pixels
    #
    if flags.Tracking.ActiveConfig.doAthenaSpacePoint:
        if flags.Tracking.doITkFastTracking:
            from InDetConfig.SiSpacePointFormationConfig import (
                ITkFastSiTrackerSpacePointFinderCfg)
            acc.merge(ITkFastSiTrackerSpacePointFinderCfg(flags))
        else:
            from InDetConfig.SiSpacePointFormationConfig import (
                ITkSiTrackerSpacePointFinderCfg)
            acc.merge(ITkSiTrackerSpacePointFinderCfg(flags))
 
    if flags.Tracking.ActiveConfig.doFPGASpacePoint:
        from EFTrackingFPGAIntegration.EFSpacePointFormationConfig import (
            EFSpacePointFormationCfg)
        acc.merge(EFSpacePointFormationCfg(flags, previousActsExtension=previousActsExtension))
 
    if flags.Tracking.ActiveConfig.doActsSpacePoint:
        from ActsConfig.ActsSpacePointFormationConfig import (
            ActsSpacePointFormationCfg)
        acc.merge(ActsSpacePointFormationCfg(flags, previousActsExtension=previousActsExtension))
 
    #
    # --- Space Point EDM converters
    #
    if flags.Tracking.ActiveConfig.doAthenaToActsSpacePoint:
        #
        # --- InDet -> xAOD Space Point EDM converter
        #
        from InDetConfig.SiSpacePointFormationConfig import (
            InDetToXAODSpacePointConversionCfg)
        acc.merge(InDetToXAODSpacePointConversionCfg(flags))
 
    # this truth must only be done if you do PRD and SpacePointformation
    # If you only do the latter (== running on ESD) then the needed input (simdata)
    # is not in ESD but the resulting truth (clustertruth) is already there ...
    if flags.Tracking.doTruth:
        if flags.Tracking.ActiveConfig.doAthenaCluster or flags.Tracking.ActiveConfig.doActsToAthenaCluster:
            from InDetConfig.InDetTruthAlgsConfig import ITkPRD_MultiTruthMakerSiCfg
            acc.merge(ITkPRD_MultiTruthMakerSiCfg(flags))
 
        if flags.Tracking.ActiveConfig.doActsCluster or flags.Tracking.ActiveConfig.doAthenaToActsCluster:
            from ActsConfig.ActsTruthConfig import ActsTruthAssociationAlgCfg, ActsTruthParticleHitCountAlgCfg
            acc.merge(ActsTruthAssociationAlgCfg(flags))
            acc.merge(ActsTruthParticleHitCountAlgCfg(flags))
 
 
    return acc
 
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
 
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags.Input.Files = defaultTestFiles.RDO_RUN2
 
    flags.Tracking.doPixelClusterSplitting = True
 
    numThreads = 1
    flags.Concurrency.NumThreads = numThreads
    flags.Concurrency.NumConcurrentEvents = numThreads
 
    flags.lock()
    flags.dump()
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    top_acc = MainServicesCfg(flags)
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    top_acc.merge(PoolReadCfg(flags))
 
    top_acc.merge(InDetRecPreProcessingSiliconCfg(flags))
 
    iovsvc = top_acc.getService('IOVDbSvc')
    iovsvc.OutputLevel = 5
 
    top_acc.printConfig()
    top_acc.run(25)
    top_acc.store(open("test_SiliconPreProcessing.pkl", "wb"))