BTagJetAugmenterAlgConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
def BTagJetAugmenterAlgCfg(ConfigFlags, BTagCollection, Associator,  TrackCollection, doFlipTagger=False):
 
    name = (BTagCollection + 'augment').lower()
    flip_config = 'FLIP_SIGN' if doFlipTagger else 'STANDARD'
    name+=("_"+flip_config)
 
    useIpxd = bool({'IP3D','IP2D'} & set(ConfigFlags.BTagging.taggerList))
 
    btagAug = CompFactory.FlavorTagDiscriminants.BTagAugmenterTool(
        name=name,
        flipTagConfig=flip_config,
        trackAssociator=Associator,
        useIpxd=useIpxd,
    )
 
    decorAlg = CompFactory.FlavorTagDiscriminants.BTagDecoratorAlg(
        name=f'{name}_alg',
        container=BTagCollection,
        constituentContainer=TrackCollection,
        decorator=btagAug,
    )
    # -- create the augmentation algorithm
    acc = ComponentAccumulator()
    acc.addEventAlgo(decorAlg)
 
    return acc