NavConverterConfig.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.ComponentFactory import CompFactory
# from AthenaCommon.Constants import DEBUG  # uncomment for the easy usage while testing with DEBUG
from AthenaCommon.Logging import logging
log = logging.getLogger("NavConverterConfig")
 
 
def NavConverterCfg(flags, chainsList = [], runTheChecker = False):
    """Configures Run 1/2 to Run 3 navigation conversion algorithm for all triggers"""
    if len(chainsList) == 0:
        log.info("Chains list is empty, the conversion will happen for all chains, but no validation will be performed")
        log.info("The later is enabled when an explicit chain list is provided")
 
    acc = ComponentAccumulator()
 
    if flags.Trigger.EDMVersion >= 3:
        return acc
 
    if not flags.Trigger.doEDMVersionConversion:
        return acc
 
    from TrigDecisionTool.TrigDecisionToolConfig import TrigDecisionToolCfg, getRun3NavigationContainerFromInput
    tdt = acc.getPrimaryAndMerge(TrigDecisionToolCfg(flags))
 
    r2ToR3OutputName = getRun3NavigationContainerFromInput(flags)
 
    cnvAlg = CompFactory.Run2ToRun3TrigNavConverterV2("TrigRun2ToRun3NavConverter") #, OutputLevel = 2)
    cnvAlg.TrigDecisionTool = tdt
    cnvAlg.TrigNavReadKey = ""
    cnvAlg.TrigConfigSvc = tdt.TrigConfigSvc
    cnvAlg.OutputNavKey = r2ToR3OutputName
    from OutputStreamAthenaPool.OutputStreamConfig import addToAOD, addToESD
    collections = [f"xAOD::TrigCompositeContainer#{r2ToR3OutputName}", f"xAOD::TrigCompositeAuxContainer#{r2ToR3OutputName}Aux."]
    acc.merge(addToAOD(flags, collections))
    acc.merge(addToESD(flags, collections))
 
    cnvAlg.Rois = ["initialRoI","forID","forID1","forID2","forMS","forSA","forTB","forMT","forCB"]
 
    from TrigEDMConfig.TriggerEDM import getTriggerEDMList
    edm = getTriggerEDMList(flags, "AODCONV")
 
    def cleanKey(type, name):
        return name.replace("HLT_", "", 1).replace(type.replace("::", "__"), "", 1).replace("_", "", 1)
    types = []
    for type, keys in edm.items():
        
        for key in keys:            
            types.append(type+"#"+cleanKey(type, key))
    
    log.info("Assuming these collections are relevant for trigger: %s", " ".join(types))
    cnvAlg.Collections = types
    cnvAlg.Chains = chainsList
    cnvAlg.doCompression = True # set True for compression
    acc.addEventAlgo(cnvAlg)
 
    if runTheChecker:
        checker = CompFactory.Trig.NavigationTesterAlg(FailOnDifference = False, TrigDecisionTool = tdt) #, OutputLevel = 2)
        checker.RetrievalToolRun2Nav = CompFactory.Trig.IParticleRetrievalTool()
        
        # in conversion job  Run2 TDT is setup as default, we need to setup an alternative to access Run 3 format
        run3tdt = CompFactory.Trig.TrigDecisionTool("Run3TrigDecisionTool",
                                                    HLTSummary = r2ToR3OutputName,
                                                    NavigationFormat = 'TrigComposite',
                                                    AcceptMultipleInstance=True,
                                                    TrigConfigSvc = tdt.TrigConfigSvc) #, OutputLevel = 2)
        acc.addPublicTool(run3tdt)
        checker.RetrievalToolRun3Nav = CompFactory.Trig.R3IParticleRetrievalTool(TrigDecisionTool = run3tdt) #, OutputLevel = 2)
        checker.Chains = chainsList
        checker.TrigDecisionToolRun3 = run3tdt
        checker.TrigDecisionToolRun2 = tdt
 
        acc.addEventAlgo(checker)
 
    return acc
 
 
if __name__ == "__main__":
    # this is only config test, actual tests are in share/testTrigR2... scripts
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.AOD_RUN2_DATA
    flags.Trigger.doEDMVersionConversion = True
    flags.lock()
 
    acc = NavConverterCfg(flags)
    acc.printConfig(withDetails=True, summariseProps=True)
    acc.wasMerged()