MuonSelectorToolsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
  
#********************************************************************
# MuonsSelectionToolConfig.py 
# Configures muon selection tool which is used to select muons 
# for use in physics analysis
#********************************************************************
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import LHCPeriod
 
def MuonLoosenedNonCalibratedSelectionToolCfg(flags, name="MuonLoosenedNonCalibratedSelectionTool", **kwargs):
    """Configure the muon selection tool"""
    kwargs.setdefault("DisablePtCuts", True)
    kwargs.setdefault("TurnOffMomCorr", True)
    return MuonSelectionToolCfg(flags, name, **kwargs)
 
 
#Pretty much copy of the above but without expert flags on disabling pt cuts and turning off momentum corrections. This is for use in physics analysis
def MuonSelectionToolCfg(flags, name="MuonSelectionTool", **kwargs):
    """Configure the muon selection tool"""
    acc = ComponentAccumulator()
    if flags.Muon.scheduleActsReco:
        acc.setPrivateTools(CompFactory.CP.MuonPhaseIISelectionTool(name))
        return acc
 
    # Configure the Onnx tool FIRST
    from AthOnnxComps.OnnxRuntimeFlags import OnnxRuntimeType
    from AthOnnxComps.OnnxRuntimeInferenceConfig import OnnxRuntimeInferenceToolCfg
 
    model_fname = "MuonSelectorTools/TightNN_Experimental_18062025/model_DNN3norm_MC20ade.onnx"
    if flags.GeoModel.Run >= LHCPeriod.Run3:
        model_fname = "MuonSelectorTools/TightNN_Experimental_18062025/model_DNN3norm_MC23ad.onnx"
 
    execution_provider = OnnxRuntimeType.CPU
    # Set defaults AFTER ort_tool is available
    kwargs.setdefault("IsRun3Geo", flags.GeoModel.Run >= LHCPeriod.Run3)
    kwargs.setdefault("ORTInferenceTool", 
                      acc.popToolsAndMerge(OnnxRuntimeInferenceToolCfg(flags, model_fname, execution_provider, name=f"{name}_ORTInferenceTool")))
 
 
    # Now construct the tool with all kwargs set
    the_tool = CompFactory.CP.MuonSelectionTool(name, **kwargs)
    acc.setPrivateTools(the_tool)
    return acc