FullScanInDetConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.AccumulatorCache import AccumulatorCache
from AthenaConfiguration.AthConfigFlags import AthConfigFlags
 
from TrigInDetConfig.utils import getFlagsForActiveConfig
from TrigInDetConfig.TrigInDetConfig import trigInDetFastTrackingCfg, trigInDetLRTCfg
from InDetConfig.InDetPriVxFinderConfig import InDetTrigPriVxFinderCfg
 
from AthenaCommon.Logging import logging
from AthenaCommon.CFElements import parOR
 
logging.getLogger().info("Importing %s",__name__)
log = logging.getLogger(__name__)
 
from .FullScanDefs import trkFSRoI
 
@AccumulatorCache
def commonInDetFullScanCfg(flags: AthConfigFlags) -> ComponentAccumulator:
    acc = ComponentAccumulator()
    seqname='TrigInDetFullScan'
    acc.addSequence(parOR(seqname),primary=True)
 
    flagsWithTrk = getFlagsForActiveConfig(flags, 'fullScan', log)
    acc.merge(
        trigInDetFastTrackingCfg(
            flagsWithTrk,
            trkFSRoI,
            signatureName='fullScan',
            in_view=False
        ),
        seqname
    )
 
    vxkwargs = {
        "inputTracks": flagsWithTrk.Tracking.ActiveConfig.tracks_FTF,
        "outputVtx": flagsWithTrk.Tracking.ActiveConfig.vertex_jet,
    }
    if flags.Trigger.FSTrk.doJetRestrictedVertexSort:
        from TrkConfig.TrkVertexToolsConfig import JetRestrictedSumPt2VertexCollectionSortingToolCfg
        from TrkConfig.TrkVertexWeightCalculatorsConfig import JetRestrictedSumPt2VertexWeightCalculatorCfg
        jetcalccfg = JetRestrictedSumPt2VertexWeightCalculatorCfg(
            flags,
            JetContainer='HLT_AntiKt4EMTopoJets_subjesIS',
            TrackParticleLocation=flagsWithTrk.Tracking.ActiveConfig.tracks_FTF,
            PlainSumPtDecor='UnrestrictedSumPt'
        )
        jetcalc = jetcalccfg.popPrivateTools()
        acc.merge(jetcalccfg)
        vxsortcfg = JetRestrictedSumPt2VertexCollectionSortingToolCfg(
            flags,
            VertexWeightCalculator=jetcalc
        )
        vxsort = acc.popToolsAndMerge(vxsortcfg)
        vxkwargs["VertexCollectionSortingTool"] = vxsort
 
        from TrkConfig.TrkVertexToolsConfig import SumPt2VertexCollectionSortingToolCfg
        default_sumpt_vxsort = SumPt2VertexCollectionSortingToolCfg(flags)
        from InDetPriVxFinder.ResortVerticesConfig import ResortVerticesCfg
        acc.merge(ResortVerticesCfg(
            flags,
            flagsWithTrk.Tracking.ActiveConfig.vertex_jet,
            "HLT_IDVertex_FS_origsumpt",
            default_sumpt_vxsort,
        ))
 
    acc.merge(
        InDetTrigPriVxFinderCfg(
            flagsWithTrk,
            **vxkwargs,
        ),
        seqname
    )
 
    return acc
 
 
def commonInDetLRTCfg(flags    : AthConfigFlags, 
                      flagsLRT : AthConfigFlags, 
                      rois     : str = trkFSRoI) -> ComponentAccumulator:
    
    acc = ComponentAccumulator()
    seqname = 'TrigInDetLRT_'+flagsLRT.Tracking.ActiveConfig.name
    acc.addSequence(parOR(seqname),primary=True)
 
    acc.merge(
        trigInDetLRTCfg(
            flagsLRT,
            flags.Tracking.ActiveConfig.trkTracks_FTF,
            rois,
            in_view=False
        ),
        seqname
    )
 
    return acc