InDetOutputConfig.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
 
'''@file InDetOutputConfig.py
@author M. Aparo
@date 13-03-20234
@brief CA-based python configurations to add the output stream of the AOD_IDTPM file 
'''
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
#from AthenaConfiguration.ComponentFactory import CompFactory
 
 
def InDetOutputCfg( flags, **kwargs ):
    '''
    Write output file in AOD_IDTPM format, on top of the
    histogram output file, for reprocessing
    '''
    acc = ComponentAccumulator()
 
    
 
    
 
    
    itemsToRecord = set()
 
    for trkAnaName in flags.PhysVal.IDTPM.trkAnaNames:
        
        offTracks = getattr( flags.PhysVal.IDTPM, trkAnaName+'.OfflineTrkKey' )
        trigTracks = getattr( flags.PhysVal.IDTPM, trkAnaName+'.TrigTrkKey' )
        itemsToRecord.update({
            'xAOD::TrackParticleContainer#'+offTracks,
            'xAOD::TrackParticleAuxContainer#'+offTracks+'Aux.',#+truthVars+'.'+objVars,
            'xAOD::TrackParticleContainer#'+trigTracks,
            'xAOD::TrackParticleAuxContainer#'+trigTracks+'Aux.',#+truthVars,
        })
 
        
        truthParticles = getattr( flags.PhysVal.IDTPM, trkAnaName+'.TruthPartKey' )
        itemsToRecord.update({
            'xAOD::TruthParticleContainer#'+truthParticles,
            'xAOD::TruthParticleAuxContainer#'+truthParticles+'Aux.',
        })
 
        
        objStr = getattr( flags.PhysVal.IDTPM, trkAnaName+".SelectOfflineObject" )
 
        
        if objStr == 'Electron':
            itemsToRecord.update({
                'xAOD::ElectronContainer#Electrons',
                # TODO - uncomment to store xAOD::Electron decorations
                #'xAOD::ElectronAuxContainer#ElectronsAux.',
            })
 
        
        if objStr == 'Muon':
            itemsToRecord.update({
                'xAOD::MuonContainer#Muons',
                # TODO - uncomment to store xAOD::Muon decorations
                #'xAOD::MuonAuxContainer#MuonsAux.',
            })
 
        
        if objStr == 'Tau':
            itemsToRecord.update({
                'xAOD::TauJetContainer#TauJets',
                # TODO - uncomment to store xAOD::TauJets decorations
                #'xAOD::TauJetAuxContainer#TauJetsAux.',
            })
 
        
        trkAnaInfo = "TrkAnaInfo"+getattr( flags.PhysVal.IDTPM, trkAnaName+'.anaTag' )
        itemsToRecord.update({
            'xAOD::BaseContainer#'+trkAnaInfo,
            'xAOD::AuxContainerBase#'+trkAnaInfo+'Aux.',
        })
 
 
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    acc.merge( OutputStreamCfg( flags, 'AOD_IDTPM',
                                ItemList = list( itemsToRecord ),
                                disableEventTag = True ) )
 
    return acc