ActsPostIncludes.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
def PersistifyActsEDMCfg(flags) -> ComponentAccumulator:
    acc = ComponentAccumulator()
 
    toAOD = []
 
    if flags.Acts.EDM.PersistifyClusters or flags.Acts.EDM.PersistifySpacePoints:
        pixel_cluster_shortlist = ['-pixelClusterLink']
        strip_cluster_shortlist = ['-sctClusterLink']
        
        pixel_cluster_variables = '.'.join(pixel_cluster_shortlist)
        strip_cluster_variables = '.'.join(strip_cluster_shortlist)
 
        toAOD += ['xAOD::PixelClusterContainer#ITkPixelClusters',
                  'xAOD::PixelClusterAuxContainer#ITkPixelClustersAux.' + pixel_cluster_variables,
                  'xAOD::StripClusterContainer#ITkStripClusters',
                  'xAOD::StripClusterAuxContainer#ITkStripClustersAux.' + strip_cluster_variables]
 
        if flags.Reco.EnableHGTDExtension:
            hgtd_cluster_shortlist = ['-hgtdClusterLink']
 
            hgtd_cluster_variables = '.'.join(hgtd_cluster_shortlist)
            
            toAOD += ['xAOD::HGTDClusterContainer#HGTD_Clusters',
                      'xAOD::HGTDClusterAuxContainer#HGTD_ClustersAux.' + hgtd_cluster_variables]
 
        if flags.Acts.doITkConversion:
            toAOD += ['xAOD::StripClusterContainer#ITkConversionStripClusters',
                      'xAOD::StripClusterAuxContainer#ITkConversionStripClustersAux.' + strip_cluster_variables]
 
    if flags.Acts.EDM.PersistifySpacePoints:
        pixel_spacepoint_shortlist = ['-measurements',
                                      '-pixelSpacePointLink']
        strip_spacepoint_shortlist = ['topHalfStripLength', 
                                      'bottomHalfStripLength', 
                                      'topStripDirection',
                                      'bottomStripDirection',
                                      'stripCenterDistance',
                                      'topStripCenter',
                                      'measurementLink']
 
        pixel_spacepoint_variables = '.'.join(pixel_spacepoint_shortlist)
        strip_spacepoint_variables = '.'.join(strip_spacepoint_shortlist)
        
        toAOD += ['xAOD::SpacePointContainer#ITkPixelSpacePoints',
                  'xAOD::SpacePointAuxContainer#ITkPixelSpacePointsAux.' + pixel_spacepoint_variables,
                  'xAOD::SpacePointContainer#ITkStripSpacePoints',
                  'xAOD::SpacePointAuxContainer#ITkStripSpacePointsAux.' + strip_spacepoint_variables,
                  'xAOD::SpacePointContainer#ITkStripOverlapSpacePoints',
                  'xAOD::SpacePointAuxContainer#ITkStripOverlapSpacePointsAux.' + strip_spacepoint_variables]
 
    if flags.Acts.EDM.PersistifyTracks:
        trackPrefixes = ['Acts', 'ActsResolved',
                         'ActsLargeRadius', 'ActsLargeRadiusResolved',
                         'ActsConversion', 'ActsConversionResolved',
                         'ActsHeavyIon', 'ActsHeavyIonResolved']
        for prefix in trackPrefixes:
            toAOD +=  [f"xAOD::TrackSummaryContainer#{prefix}TrackSummary",
                       f"xAOD::TrackSummaryAuxContainer#{prefix}TrackSummaryAux.",
                       f"xAOD::TrackStateContainer#{prefix}TrackStates",
                       f"xAOD::TrackStateAuxContainer#{prefix}TrackStatesAux.-uncalibratedMeasurement",
                       f"xAOD::TrackParametersContainer#{prefix}TrackParameters",
                       f"xAOD::TrackParametersAuxContainer#{prefix}TrackParametersAux.",
                       f"xAOD::TrackJacobianContainer#{prefix}TrackJacobians",
                       f"xAOD::TrackJacobianAuxContainer#{prefix}TrackJacobiansAux.",
                       f"xAOD::TrackMeasurementContainer#{prefix}TrackMeasurements",
                       f"xAOD::TrackMeasurementAuxContainer#{prefix}TrackMeasurementsAux.",
                       f"xAOD::TrackSurfaceContainer#{prefix}TrackStateSurfaces",
                       f"xAOD::TrackSurfaceAuxContainer#{prefix}TrackStateSurfacesAux.",
                       f"xAOD::TrackSurfaceContainer#{prefix}TrackSurfaces",
                       f"xAOD::TrackSurfaceAuxContainer#{prefix}TrackSurfacesAux."]
 
    # add track particles created by the Acts TrackToTrackParticleCnvAlg to the AOD
    trackCnvPrefixes = ["Acts"]
    trackparticles_shortlist = [] if flags.Acts.EDM.PersistifyTracks else ['-actsTrack']
    trackparticles_variables = ".".join(trackparticles_shortlist)
    for prefix in trackCnvPrefixes:
        toAOD += [f"xAOD::TrackParticleContainer#InDet{prefix}TrackParticles",
                  f"xAOD::TrackParticleAuxContainer#InDet{prefix}TrackParticlesAux." + trackparticles_variables]
 
    # If there is nothing to persistify, returns an empty CA
    if len(toAOD) == 0:
        return acc
 
    from OutputStreamAthenaPool.OutputStreamConfig import addToAOD    
    acc.merge(addToAOD(flags, toAOD))
    return acc
 
def ACTSClusterPostInclude(flags) -> ComponentAccumulator:
    # Schedule ACTS Data Preparation and Measurement persistification
    # This is used for technical efficiencies studies of tracking pipelines
    if flags.Tracking.PRDInfo.KeepOnlyOnTrackMeasurements:
        raise ValueError("The ACTSClusterPostInclude is to be used for technical efficiency computation, however the " \
                         f"config flag 'Tracking.PRDInfo.KeepOnlyOnTrackMeasurements' is set to {flags.Tracking.PRDInfo.KeepOnlyOnTrackMeasurements}, " \
                         "which is incompatible with this purpose")
    
    acc = ComponentAccumulator()
    
    # Schedule Full Scan RoI
    from ActsConfig.ActsRegionsOfInterestConfig import ActsMainRegionsOfInterestCreatorAlgCfg
    acc.merge(ActsMainRegionsOfInterestCreatorAlgCfg(flags,
                                                     name = "ActsOfflineRegionsOfInterestCreatorAlg",
                                                     RoIs = "ActsOfflineRegionOfInterest"))
    
    # Cluster formation
    # This includes Pixel and Strip
    clusteringKwargs = dict()
    clusteringKwargs["PixelClusterizationAlg.name"] = "ActsOfflinePixelClusterizationAlg"
    clusteringKwargs["PixelClusterizationAlg.ClustersKey"] = "ITkOfflinePixelClusters"
    clusteringKwargs["StripClusterizationAlg.name"] = "ActsOfflineStripClusterizationAlg"
    clusteringKwargs["StripClusterizationAlg.ClustersKey"] = "ITkOfflineStripClusters"
    from ActsConfig.ActsClusterizationConfig import ActsMainClusterizationCfg
    acc.merge(ActsMainClusterizationCfg(flags,
                                        RoIs = "ActsOfflineRegionOfInterest",
                                        processHGTD = False,
                                        **clusteringKwargs))
 
    # Attach truth to clusters
    if flags.Tracking.doTruth:
        truthAssociationKwargs = dict()
        truthAssociationKwargs["PixelClusterToTruthAssociationAlg.name"] = "ActsOfflinePixelClusterToTruthAssociationAlg"
        truthAssociationKwargs["PixelClusterToTruthAssociationAlg.Measurements"] = "ITkOfflinePixelClusters"
        truthAssociationKwargs["PixelClusterToTruthAssociationAlg.AssociationMapOut"] = "ITkOfflinePixelClustersToTruthParticles"
        truthAssociationKwargs["StripClusterToTruthAssociationAlg.name"] = "ActsOfflineStripClusterToTruthAssociationAlg"
        truthAssociationKwargs["StripClusterToTruthAssociationAlg.Measurements"] = "ITkOfflineStripClusters"
        truthAssociationKwargs["StripClusterToTruthAssociationAlg.AssociationMapOut"] = "ITkOfflineStripClustersToTruthParticles"
 
        from ActsConfig.ActsTruthConfig import ActsTruthAssociationAlgCfg, ActsTruthParticleHitCountAlgCfg
        acc.merge(ActsTruthAssociationAlgCfg(flags,
                                             **truthAssociationKwargs))
        acc.merge(ActsTruthParticleHitCountAlgCfg(flags,
                                                  name = "ActsOfflineTruthParticleHitCountAlg",
                                                  PixelClustersToTruthAssociationMap = "ITkOfflinePixelClustersToTruthParticles",
                                                  StripClustersToTruthAssociationMap = "ITkOfflineStripClustersToTruthParticles",
                                                  TruthParticleHitCountsOut = "OfflineTruthParticleHitCounts"))
        
        from InDetConfig.InDetPrepRawDataToxAODConfig import TruthParticleIndexDecoratorAlgCfg
        acc.merge( TruthParticleIndexDecoratorAlgCfg(flags) )
 
        from ActsConfig.ActsObjectDecorationConfig import ActsPixelClusterTruthDecoratorAlgCfg,ActsStripClusterTruthDecoratorAlgCfg
        acc.merge(ActsPixelClusterTruthDecoratorAlgCfg(flags,
                                                       name = "ActsOfflinePixelClusterTruthDecoratorAlgCfg",
                                                       ClusterContainer = "ITkOfflinePixelClusters",
                                                       AssociationMapOut = "ITkOfflinePixelClustersToTruthParticles",
                                                       MeasurementContainer = "ITkPixelMeasurements_offl"))
        acc.merge(ActsStripClusterTruthDecoratorAlgCfg(flags,
                                                       name = "ActsOfflineStripClusterTruthDecoratorAlgCfg",
                                                       ClusterContainer = "ITkOfflineStripClusters",
                                                       AssociationMapOut = "ITkOfflineStripClustersToTruthParticles",
                                                       MeasurementContainer = "ITkStripMeasurements_offl"))
 
        
        toAOD = [
            'xAOD::TrackMeasurementValidationContainer#ITkPixelMeasurements_offl',
            'xAOD::TrackMeasurementValidationAuxContainer#ITkPixelMeasurements_offlAux.',
            'xAOD::TrackMeasurementValidationContainer#ITkStripMeasurements_offl',
            'xAOD::TrackMeasurementValidationAuxContainer#ITkStripMeasurements_offlAux.'
        ]
        
        from OutputStreamAthenaPool.OutputStreamConfig import addToAOD
        acc.merge( addToAOD( flags, toAOD ) )
 
    return acc