python.HLT.Bjet.BjetFlavourTaggingConfig Namespace Reference

Functions
def	flavourTaggingCfg (flags, inputJets, inputVertex, inputTracks, BTagName, inputMuons="")
def	fastFlavourTaggingCfg (flags, inputJets, inputVertex, inputTracks, isPFlow=False, fastDipsMinimumPt=None, doXbbtagLargeRJet=False)
def	OnlineBeamspotIpAugmenterCfg (cfgFlags, tracks, vertices='', trackIpPrefix='simpleIp_')

Function Documentation

fastFlavourTaggingCfg()

def python.HLT.Bjet.BjetFlavourTaggingConfig.fastFlavourTaggingCfg	(		flags,
			inputJets,
			inputVertex,
			inputTracks,
			isPFlow = False,
			fastDipsMinimumPt = None,
			doXbbtagLargeRJet = False
	)

This function tags jets directly: there is no  b-tagging object

Definition at line 66 of file BjetFlavourTaggingConfig.py.

def fastFlavourTaggingCfg( flags, inputJets, inputVertex, inputTracks, isPFlow=False, fastDipsMinimumPt=None, doXbbtagLargeRJet = False):
    """
    This function tags jets directly: there is no  b-tagging object
    """
 
    ca = ComponentAccumulator()
 
    ca.merge(JetTagCalibCfg(flags))
    
    # first add the track augmentation
    jet_name = inputJets
    if isPFlow:
        ca.merge(
            BTagTrackAugmenterAlgCfg(
                flags,
                TrackCollection=inputTracks,
                PrimaryVertexCollectionName=inputVertex,
            )
        )
    else:
        trackIpPrefix='simpleIp_'
        ca.merge(
            OnlineBeamspotIpAugmenterCfg(
            flags,
            tracks=inputTracks,
            vertices=inputVertex,
            trackIpPrefix=trackIpPrefix,
            )
        )
        if inputVertex:
            ca.merge(
            BTagTrackAugmenterAlgCfg(
                flags,
                TrackCollection=inputTracks,
                PrimaryVertexCollectionName=inputVertex,
            )
        )
 
    # now we associate the tracks to the jet
    
    tracksOnJetDecoratorName = "TracksForMinimalJetTag"
    pass_flag = f'{tracksOnJetDecoratorName}_isValid'
    ca.merge(
        JetParticleAssociationAlgCfg(
            flags,
            JetCollection=jet_name,
            InputParticleCollection=inputTracks,
            OutputParticleDecoration=tracksOnJetDecoratorName,
            MinimumJetPt=fastDipsMinimumPt,
            MinimumJetPtFlag=pass_flag
        )
    )
 
    # Now we have to add an algorithm that tags the jets with dips
    # The input and output remapping is handled via a map in DL2.
    #
    # The file above adds fastDIPSnoPV20220211_p*, we'll call them
    # fastDips_p* on the jet.
    if isPFlow:
        if doXbbtagLargeRJet:
            dl2_configs = [
                [
                    'BTagging/20230705/gn2xv01/antikt10ufo/network.onnx',
                    {
                        'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    }
                ],
            ]
        else: 
            dl2_configs=[
                [
                    'BTagging/20211216trig/dips/AntiKt4EMPFlow/network.json',
                    {
                        'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    }
                ],
                [
                    'BTagging/20211215trig/fastDips/antikt4empflow/network.json',
                    {
                        'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    },
                ],
 
                [
                    'BTagging/20230331trig/gn1/antikt4empflow/network.onnx',
                    {
                        'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    },
                ],
                                [
                 'BTagging/20240122trig/gn2/antikt4empflow/Small.onnx',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    **{f'p{x}': f'tlaGN220240122_p{x}' for x in 'cub'},
                }
                ],   
            ]
    else:
        dl2_configs=[
            [
                'BTagging/20220211trig/fastDips/antikt4empflow/network.json',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    **{f'fastDIPSnoPV20220211_p{x}': f'fastDips_p{x}' for x in 'cub'},
                    'btagIp_': trackIpPrefix,
                }
            ],
            [
                'BTagging/20230327trig/gn1/antikt4emtopo/network.onnx',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    **{f'GN120230327_p{x}': f'fastGN120230327_p{x}' for x in 'cub'},
                    'btagIp_': trackIpPrefix,
                }
            ],
            [
                'BTagging/20231122trig/dipz/antikt4emtopo/network.json',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    'btagIp_': trackIpPrefix,
                }
            ],
 
        ]
        if inputVertex: 
            dl2_configs += [
                [
                 'BTagging/20230331trig/gn1/antikt4empflow/network.onnx',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    **{f'GN120230331_p{x}': f'fastGN120230331_p{x}' for x in 'cub'}
                },   
                ],
                [
                 'BTagging/20240122trig/gn2/antikt4empflow/Small.onnx',
                {
                    'BTagTrackToJetAssociator': tracksOnJetDecoratorName,
                    **{f'p{x}': f'fastGN220240122_p{x}' for x in 'cub'},
                }
                ],
 
                [ 'BTagging/20240216trig/gn2/antikt4emtopo/GNTau.onnx',
                    { 'BTagTrackToJetAssociator': tracksOnJetDecoratorName
                    , 'GN2_ptau' : 'fastGNTau20240216_ptau'
                    , 'GN2_pu' : 'fastGNTau20240216_pu'
                    }
                ]   
            ]
 
    # not all the keys that the NN requests are declaired. This will
    # cause an algorithm stall if we don't explicetly tell it that it
    # can ignore some of them.
    missingKeys = getStaticTrackVars(inputTracks)
 
    for nnFile, variableRemapping in dl2_configs:
        nnAlgo = nnFile.replace('/','_').split('.')
        nnAlgoKey = nnAlgo[0]
        nnAlgoext = nnAlgo[1]
        toolDict = {
            "json": CompFactory.FlavorTagDiscriminants.DL2Tool,
            "onnx": CompFactory.FlavorTagDiscriminants.GNNTool
        }
        ca.addEventAlgo(
            CompFactory.FlavorTagDiscriminants.JetTagConditionalDecoratorAlg(
                name='_'.join([
                    'simpleJetTagAlg',
                    jet_name,
                    inputTracks,
                    nnAlgoKey,
                ]),
                container=jet_name,
                constituentContainer=inputTracks,
                undeclaredReadDecorKeys=missingKeys,
                tagFlag=pass_flag,
                decorator=toolDict[nnAlgoext](
                    name='_'.join([
                        'simpleDipsToJet',
                        nnAlgoKey,
                    ]),
                    nnFile=nnFile,
                    variableRemapping=variableRemapping,
                    # note that the tracks are associated to the jet as
                    # and IParticle container.
                    trackLinkType='IPARTICLE',
                    defaultOutputValue=0
                ),
            )
        )
    return ca
 

flavourTaggingCfg()

def python.HLT.Bjet.BjetFlavourTaggingConfig.flavourTaggingCfg	(		flags,
			inputJets,
			inputVertex,
			inputTracks,
			BTagName,
			inputMuons = ""
	)

Definition at line 16 of file BjetFlavourTaggingConfig.py.

def flavourTaggingCfg( flags, inputJets, inputVertex, inputTracks, BTagName,
                       inputMuons = ""):
 
    # because Cfg functions internally re-append the 'Jets' string
    inputJetsPrefix = inputJets.replace("bJets","b")
 
    acc = ComponentAccumulator()
 
    acc.merge(JetTagCalibCfg(flags))
    
    #Track Augmenter
    acc.merge(BTagTrackAugmenterAlgCfg(
        flags,
        TrackCollection=inputTracks,
        PrimaryVertexCollectionName=inputVertex
    ))
 
    #Run new Run3 taggers, i.e. DL1, RNNIP, DL1r
    nnList = [
 
        # These are trigger-specific trainings
        #
        # R22 retraining for DIPS, provides dips20211116 with a loose
        # track selection
        'BTagging/20211216trig/dips/AntiKt4EMPFlow/network.json',
        # R22 retraining with the above DIPS, provides DL1d20211216
        'BTagging/20211216trig/dl1d/AntiKt4EMPFlow/network.json',
        # Trigger GN1 training
        'BTagging/20220813trig/gn1/antikt4empflow/network.onnx',
        # Trigger DL1dbb training
        'BTagging/20230314trig/dl1dbb/antikt4empflow/network.json',
        #Trigger GN2 training
        'BTagging/20240122trig/gn2/antikt4empflow/SmallPrec.onnx'
    ]
 
 
    acc.merge(BTagAlgsCfg(
        inputFlags=flags,
        JetCollection=inputJetsPrefix,
        nnList=nnList,
        trackCollection=inputTracks,
        muons=inputMuons,
        primaryVertices=inputVertex,
        BTagCollection=BTagName,
        renameTrackJets=False,
        AddedJetSuffix='Jets'
    ))
 
    return acc
 

OnlineBeamspotIpAugmenterCfg()

def python.HLT.Bjet.BjetFlavourTaggingConfig.OnlineBeamspotIpAugmenterCfg	(		cfgFlags,
			tracks,
			vertices = '',
			trackIpPrefix = 'simpleIp_'
	)

Definition at line 256 of file BjetFlavourTaggingConfig.py.

def OnlineBeamspotIpAugmenterCfg(cfgFlags, tracks, vertices='',
                                 trackIpPrefix='simpleIp_'):
    ca = ComponentAccumulator()
 
    pfx = 'online'
    i = 'EventInfo'
    x = f'{i}.{pfx}BeamPosX'
    y = f'{i}.{pfx}BeamPosY'
    z = f'{i}.{pfx}BeamPosZ'
    sig_x = f'{i}.{pfx}BeamPosSigmaX'
    sig_y = f'{i}.{pfx}BeamPosSigmaY'
    sig_z = f'{i}.{pfx}BeamPosSigmaZ'
    cov_xy = f'{i}.{pfx}BeamPosSigmaXY'
    tilt_XZ = f'{i}.{pfx}BeamTiltXZ'
    tilt_YZ = f'{i}.{pfx}BeamTiltYZ'
    status = f'{i}.{pfx}BeamStatus'
 
    ca.merge(BeamSpotCondAlgCfg(cfgFlags))
    ca.addEventAlgo(CompFactory.xAODMaker.EventInfoBeamSpotDecoratorAlg(
        name='_'.join([
                'EventInfoBeamSpotDecorator',
                tracks,
                vertices,
                trackIpPrefix,
            ]).replace('__','_').rstrip('_'),
        beamPosXKey=x,
        beamPosYKey=y,
        beamPosZKey=z,
        beamPosSigmaXKey=sig_x,
        beamPosSigmaYKey=sig_y,
        beamPosSigmaZKey=sig_z,
        beamPosSigmaXYKey=cov_xy,
        beamTiltXZKey=tilt_XZ,
        beamTiltYZKey=tilt_YZ,
        beamStatusKey=status,
    ))
 
    ca.addEventAlgo(
        CompFactory.FlavorTagDiscriminants.PoorMansIpAugmenterAlg(
            name='_'.join([
                'SimpleTrackAugmenter',
                tracks,
                vertices,
                trackIpPrefix,
            ]).replace('__','_').rstrip('_'),
            trackContainer=tracks,
            primaryVertexContainer=vertices,
            prefix=trackIpPrefix,
            beamspotSigmaX=sig_x,
            beamspotSigmaY=sig_y,
            beamspotSigmaZ=sig_z,
            beamspotCovarianceXY=cov_xy
        )
    )
    return ca