TrigTauRecToolsConfig Namespace Reference

Functions
	trigTauVertexFinderCfg (flags, name='')
	trigTauTrackFinderCfg (flags, name='', TrackParticlesContainer='')
	tauVertexVariablesCfg (flags, name='')
	trigTauJetLVNNEvaluatorCfg (flags, tau_id='', use_taujet_rnnscore=True)
	trigTauJetONNXEvaluatorCfg (flags, str tau_id, str tau_container='', str hits_decoration_container='', str|None ptau_sfx=None, str|None pjet_sfx=None)
	trigTauWPDecoratorRNNCfg (flags, str tau_id, str precision_seq_name)
	trigTauWPDecoratorCfg (flags, str tau_id, str precision_seq_name, str tauContainerName)

Function Documentation

tauVertexVariablesCfg()

TrigTauRecToolsConfig.tauVertexVariablesCfg	(		flags,
			name = '' )

Vertex variables calculation

Definition at line 55 of file TrigTauRecToolsConfig.py.

def tauVertexVariablesCfg(flags, name=''):
    '''Vertex variables calculation'''
    acc = ComponentAccumulator()
 
    from TrkConfig.TrkVertexFittersConfig import TauAdaptiveVertexFitterCfg
    TauAdaptiveVertexFitter = acc.popToolsAndMerge(TauAdaptiveVertexFitterCfg(flags))
 
    from TrkConfig.TrkVertexSeedFinderToolsConfig import CrossDistancesSeedFinderCfg
    CrossDistancesSeedFinder = acc.popToolsAndMerge(CrossDistancesSeedFinderCfg(flags))
 
    acc.setPrivateTools(CompFactory.TauVertexVariables(
        name            = name,
        VertexFitter    = TauAdaptiveVertexFitter,
        SeedFinder      = CrossDistancesSeedFinder,
    ))
 
    return acc
    

trigTauJetLVNNEvaluatorCfg()

TrigTauRecToolsConfig.trigTauJetLVNNEvaluatorCfg	(		flags,
			tau_id = '',
			use_taujet_rnnscore = True )

TauJet identification inference based on LVNN models, for RNNs and DeepSets

Definition at line 73 of file TrigTauRecToolsConfig.py.

def trigTauJetLVNNEvaluatorCfg(flags, tau_id='', use_taujet_rnnscore=True):
    '''TauJet identification inference based on LVNN models, for RNNs and DeepSets'''
    acc = ComponentAccumulator()
 
    try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
    except NameError: raise ValueError(f'Invalid LVNN (RNN, DeepSet) TauID configuration: {tau_id}')
 
    # For legacy mediumRNN/tightRNN_tracktwoMVA/tracktwoLLP/trackLRT chains
    output_variable = 'RNNJetScore' if use_taujet_rnnscore else f'{tau_id}_Score'
 
    # This is to prevent clashes between the same ID running over the legacy chains, and the new decorator-based chains
    sfx = '_RNNJetScore' if use_taujet_rnnscore else ''
 
    acc.setPrivateTools(CompFactory.TauJetRNNEvaluator(
        name                = f'TrigTau_TauJetLVNNEvaluator_{tau_id}{sfx}',
        useTRT              = flags.Detector.EnableTRT,
 
        # Network config:
        NetworkFile0P       = id_flags.NetworkConfig[0],
        NetworkFile1P       = id_flags.NetworkConfig[1],
        NetworkFile3P       = id_flags.NetworkConfig[2],
        InputLayerScalar    = 'scalar',
        InputLayerTracks    = 'tracks',
        InputLayerClusters  = 'clusters',
        OutputLayer         = 'rnnid_output',
        OutputNode          = 'sig_prob',
 
        # Inputs:
        MaxTracks           = id_flags.MaxTracks, 
        MaxClusters         = id_flags.MaxClusters,
        MaxClusterDR        = 1.0,
        VertexCorrection    = False,
        TrackClassification = False,
 
        # Decorated TauJet variable names:
        OutputVarname       = output_variable,
    ))
 
    return acc
 

trigTauJetONNXEvaluatorCfg()

TrigTauRecToolsConfig.trigTauJetONNXEvaluatorCfg	(		flags,
		str	tau_id,
		str	tau_container = '',
		str	hits_decoration_container = '',
		str | None	ptau_sfx = None,
		str | None	pjet_sfx = None )

TauJet identification inference based on ONNX models, for GNNs, transformers, etc...

Definition at line 113 of file TrigTauRecToolsConfig.py.

):
    '''TauJet identification inference based on ONNX models, for GNNs, transformers, etc...'''
    acc = ComponentAccumulator()
 
    try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
    except NameError: raise ValueError(f'Invalid ONNX TauID configuration: {tau_id}')
 
    if isinstance(id_flags.ONNXConfig, str):
        network_config = {'NetworkFileInclusive': id_flags.ONNXConfig}
    elif isinstance(id_flags.ONNXConfig, (list, tuple)) and len(id_flags.ONNXConfig) == 3:
        network_config = {
            'NetworkFile0P': id_flags.ONNXConfig[0],
            'NetworkFile1P': id_flags.ONNXConfig[1],
            'NetworkFile3P': id_flags.ONNXConfig[2],
        }
    else:
        raise ValueError(f'Invalid {tau_id} ONNX network config file')
 
    name = f'TrigTau_TauJetONNXEvaluator_{tau_id}'
    if tau_container: name += f'_{tau_container}'
    if hits_decoration_container: name += f'_{hits_decoration_container}'
 
    ptau_decor = id_flags.OutputPTau if hasattr(id_flags, 'OutputPTau') else f'{tau_id}_ProbTau'
    if ptau_sfx: ptau_decor += f'_{ptau_sfx}'
    pjet_decor = id_flags.OutputPJet if hasattr(id_flags, 'OutputPJet') else f'{tau_id}_ProbJet'
    if pjet_sfx: pjet_decor += f'_{pjet_sfx}'
 
    acc.setPrivateTools(CompFactory.TauGNNEvaluator(
        name                = name,
        useTRT              = flags.Detector.EnableTRT,
 
        # Network config:
        **network_config,
        InputLayerScalar    = 'tau_vars',
        InputLayerTracks    = 'track_vars',
        InputLayerClusters  = 'cluster_vars',
        InputLayerHits      = 'hit_vars',
        NodeNameTau         = id_flags.NodeNameTau if hasattr(id_flags, 'NodeNameTau') else 'pTau',
        NodeNameJet         = id_flags.NodeNameJet if hasattr(id_flags, 'NodeNameJet') else 'pJet',
 
        # Inputs:
        MaxTracks           = id_flags.MaxTracks if hasattr(id_flags, 'MaxTracks') else 0,
        MaxClusters         = id_flags.MaxClusters if hasattr(id_flags, 'MaxClusters') else 0,
        MaxClusterDR        = 1.0,
        MaxHits             = id_flags.MaxHits if hasattr(id_flags, 'MaxHits') else 0,
        TauContainerName    = tau_container,
        HitsHandleKey       = hits_decoration_container,
        VertexCorrection    = False,
        TrackClassification = False,
 
        # Decorated TauJet variable names:
        OutputVarname       = f'{tau_id}_Score',
        OutputDiscriminant  = id_flags.OutputDiscriminant,
        OutputPTau          = ptau_decor,
        OutputPJet          = pjet_decor,
 
        # Tau prongness selection
        MinProngTrackPt     = id_flags.MinProngTrackPt if hasattr(id_flags, 'MinProngTrackPt') else 0,
    ))
 
    return acc
 

trigTauTrackFinderCfg()

TrigTauRecToolsConfig.trigTauTrackFinderCfg	(		flags,
			name = '',
			TrackParticlesContainer = '' )

Tau track association

Definition at line 22 of file TrigTauRecToolsConfig.py.

def trigTauTrackFinderCfg(flags, name='', TrackParticlesContainer=''):
    '''Tau track association'''
    acc = ComponentAccumulator()
 
    from TrkConfig.TrkVertexFitterUtilsConfig import AtlasTrackToVertexIPEstimatorCfg
    AtlasTrackToVertexIPEstimator = acc.popToolsAndMerge(AtlasTrackToVertexIPEstimatorCfg(flags))
 
    from TrackToVertex.TrackToVertexConfig import TrackToVertexCfg
    TrackToVertexTool = acc.popToolsAndMerge(TrackToVertexCfg(flags))
 
    from TrackToCalo.TrackToCaloConfig import ParticleCaloExtensionToolCfg
    ParticleCaloExtensionTool = acc.popToolsAndMerge(ParticleCaloExtensionToolCfg(flags))
 
    from InDetConfig.InDetTrackSelectorToolConfig import TrigTauInDetTrackSelectorToolCfg
    TrigTauInDetTrackSelectorTool = acc.popToolsAndMerge(TrigTauInDetTrackSelectorToolCfg(flags))
 
    acc.setPrivateTools(CompFactory.TauTrackFinder(
        name                            = name,
        MaxJetDrTau                     = 0.2,
        MaxJetDrWide                    = 0.4,
        TrackSelectorToolTau            = TrigTauInDetTrackSelectorTool,
        TrackToVertexTool               = TrackToVertexTool,
        Key_trackPartInputContainer     = TrackParticlesContainer,
        maxDeltaZ0wrtLeadTrk            = 0.75*mm,
        removeTracksOutsideZ0wrtLeadTrk = True,
        ParticleCaloExtensionTool       = ParticleCaloExtensionTool,
        BypassExtrapolator              = True,
        tauParticleCache                = "",
        TrackToVertexIPEstimator        = AtlasTrackToVertexIPEstimator,
    ))
 
    return acc
 

trigTauVertexFinderCfg()

TrigTauRecToolsConfig.trigTauVertexFinderCfg	(		flags,
			name = '' )

Algorithm that overwrites numTrack() and charge() of tauJets in container

Definition at line 7 of file TrigTauRecToolsConfig.py.

def trigTauVertexFinderCfg(flags, name=''):
    '''Algorithm that overwrites numTrack() and charge() of tauJets in container'''
    acc = ComponentAccumulator()
 
    acc.setPrivateTools(CompFactory.TauVertexFinder(
        name                            = name,
        UseTJVA                         = False,
        AssociatedTracks                = 'GhostTrack',
        InDetTrackSelectionToolForTJVA  = '',
        Key_vertexInputContainer        = '',
        TVATool                         = '', 
    ))
 
    return acc
 

trigTauWPDecoratorCfg()

TrigTauRecToolsConfig.trigTauWPDecoratorCfg	(		flags,
		str	tau_id,
		str	precision_seq_name,
		str	tauContainerName )

TauJet signal transformed score and ID WPs decorator tool

Definition at line 222 of file TrigTauRecToolsConfig.py.

def trigTauWPDecoratorCfg(flags, tau_id: str, precision_seq_name: str, tauContainerName: str):
    '''TauJet signal transformed score and ID WPs decorator tool'''
    acc = ComponentAccumulator()
 
    try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
    except NameError: raise ValueError(f'Invalid TauID configuration: {tau_id}')
 
    if isinstance(id_flags.ScoreFlatteningConfig, str):
        cfg = {
            'flatteningFile0Prong': id_flags.ScoreFlatteningConfig,
            'flatteningFile1Prong': id_flags.ScoreFlatteningConfig,
            'flatteningFile3Prong': id_flags.ScoreFlatteningConfig,
        }
    elif isinstance(id_flags.ScoreFlatteningConfig, (list, tuple)) and len(id_flags.ScoreFlatteningConfig) == 3:
        cfg = {
            'flatteningFile0Prong': id_flags.ScoreFlatteningConfig[0],
            'flatteningFile1Prong': id_flags.ScoreFlatteningConfig[1],
            'flatteningFile3Prong': id_flags.ScoreFlatteningConfig[2],
        }
    else:
        raise ValueError(f'Invalid {tau_id} WP decorator flattening config')
 
 
    acc.setPrivateTools(CompFactory.TauWPDecorator(
        name=f'TrigTau_TauWPDecorator_{precision_seq_name}_{tau_id}',
        **cfg,
        TauContainerName=tauContainerName,
        DecorWPNames=[f'{tau_id}_{wp}' for wp in id_flags.TargetWPs],
        DecorWPCutEffs0P=[eff[0] for eff in id_flags.TargetWPs.values()],
        DecorWPCutEffs1P=[eff[1] for eff in id_flags.TargetWPs.values()],
        DecorWPCutEffs3P=[eff[2] for eff in id_flags.TargetWPs.values()],
        ScoreName=f'{tau_id}_Score',
        NewScoreName=f'{tau_id}_ScoreSigTrans',
        DefineWPs=True,
    ))
 
    return acc

trigTauWPDecoratorRNNCfg()

TrigTauRecToolsConfig.trigTauWPDecoratorRNNCfg	(		flags,
		str	tau_id,
		str	precision_seq_name )

TauJet signal transformed score and ID WPs decorator tool,
for the legacy mediumRNN/tightRNN_tracktwoMVA/tracktwoLLP/trackLRT chains ONLY!

Definition at line 182 of file TrigTauRecToolsConfig.py.

def trigTauWPDecoratorRNNCfg(flags, tau_id: str, precision_seq_name: str):
    '''
    TauJet signal transformed score and ID WPs decorator tool,
    for the legacy mediumRNN/tightRNN_tracktwoMVA/tracktwoLLP/trackLRT chains ONLY!
    '''
 
    acc = ComponentAccumulator()
 
    try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
    except NameError: raise ValueError(f'Invalid TauID configuration: {tau_id}')
 
    # In this version we store the WPs as flags accessable through
    # tau->isTau(xAOD::TauJetParameters::IsTauFlags::JetRNNSig...)
    import PyUtils.RootUtils as ru
    ROOT = ru.import_root()
    import cppyy
    cppyy.load_library('libxAODTau_cDict')
    WPEnumVals = [
        ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigVeryLoose,
        ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigLoose,
        ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigMedium,
        ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigTight,
    ]
 
    acc.setPrivateTools(CompFactory.TauWPDecorator(
        name=f'TrigTau_TauWPDecoratorRNN_{precision_seq_name}_{tau_id}',
        flatteningFile0Prong=id_flags.ScoreFlatteningConfig[0],
        flatteningFile1Prong=id_flags.ScoreFlatteningConfig[1],
        flatteningFile3Prong=id_flags.ScoreFlatteningConfig[2],
        CutEnumVals=WPEnumVals,
        SigEff0P=id_flags.TargetEff[0],
        SigEff1P=id_flags.TargetEff[1],
        SigEff3P=id_flags.TargetEff[2],
        ScoreName='RNNJetScore',
        NewScoreName='RNNJetScoreSigTrans',
        DefineWPs=True,
    ))
 
    return acc