SUSYToolsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
#==============================================================================
# Provides configs for the tools used for SUSY Derivations
#==============================================================================
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
# SUSY20 TightBad flags, copied and adjusted from ​JetCommonConfig.py
def SUSY20EventCleaningToolCfg(flags, cleaningLevel = "TightBad"):
  
  """Configure the event/jet cleaning tool for SUSY20 derivations"""
 
  acc = ComponentAccumulator()
 
  from JetSelectorTools.JetSelectorToolsConfig import EventCleaningToolCfg, JetCleaningToolCfg
 
  JetCleaningTool = acc.popToolsAndMerge(JetCleaningToolCfg(
    flags, 
    name           = "SUSY20JetCleaningTool_" + cleaningLevel + "_EMTopo",
    jetdef         = "AntiKt4EMTopoJets",
    cleaningLevel  = cleaningLevel, 
    useDecorations = False
  ))
  acc.addPublicTool(JetCleaningTool)
  
  EventCleaningTool = acc.popToolsAndMerge(EventCleaningToolCfg(
    flags,
    name          = "SUSY20EventCleaningTool_" + cleaningLevel + "_EMTopo", 
    cleaningLevel = cleaningLevel
  ))
  EventCleaningTool.JetCleanPrefix  = "DFCommonJets_"
  EventCleaningTool.OrDecorator     = "passOR_EMTopo"
  EventCleaningTool.JetContainer    = "AntiKt4EMTopoJets"
  EventCleaningTool.JetCleaningTool = JetCleaningTool
  EventCleaningTool.DoDecorations   = True
  acc.addPublicTool(EventCleaningTool)
     
  EventCleanAlg = CompFactory.EventCleaningTestAlg(
    name              = "SUSY20EventCleaningTestAlg_" + cleaningLevel + "_EMTopo",
    EventCleaningTool = EventCleaningTool,
    JetCollectionName = "AntiKt4EMTopoJets",
    EventCleanPrefix  = "DFCommonJets_",
    CleaningLevel     = cleaningLevel,
    doEvent           = True
  ) 
  acc.addEventAlgo(EventCleanAlg)
  
  return acc
 
# SUSY20 trigger skimming
def SUSY20DTTriggerSkimmingToolCfg(flags, name, **kwargs):
 
  """Configure the DT trigger skimming tool for SUSY20 derivations"""
  
  # Trigger API for Run-1 and Run-2
  if flags.Trigger.EDMVersion <= 2:
  
    from TriggerMenuMT.TriggerAPI.TriggerAPI import TriggerAPI
    from TriggerMenuMT.TriggerAPI.TriggerEnums import TriggerPeriod, TriggerType
    
    allperiods = TriggerPeriod.y2015 | TriggerPeriod.y2016 | TriggerPeriod.y2017 | TriggerPeriod.y2018 | TriggerPeriod.future2e34
  
    TriggerAPI.setConfigFlags(flags)
    el_trig = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType = TriggerType.el_single, livefraction = 0.8)
    mu_trig = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType = TriggerType.mu_single, livefraction = 0.8)
    g_trig  = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType = TriggerType.g_single,  livefraction = 0.8)
    xe_trig = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType = TriggerType.xe,        livefraction = 0.8) 
 
    # Additional unprescaled met triggers for 2015
    xe_trig += ['HLT_xe70', 'HLT_xe70_tc_lcw', 'HLT_noalg_L1J400', 'HLT_noalg_L1J420', 'HLT_noalg_L1J450']
 
    # Merge and remove duplicates
    triggers = el_trig + mu_trig + g_trig + xe_trig
    triggers = sorted(list(set(triggers)))
 
  # Run-3 and beyond
  else:
 
    from TriggerMenuMT.TriggerAPI.TriggerEnums import TriggerType
    from DerivationFrameworkPhys.TriggerListsHelper import getTapisSession
 
    # TriggerAPI Session based trigger lists
    session  = getTapisSession(flags)
    triggers = set()
    triggers = session.getLowestUnprescaled(triggerType = TriggerType.el_single, livefraction = 0.8).union(triggers)
    triggers = session.getLowestUnprescaled(triggerType = TriggerType.mu_single, livefraction = 0.8).union(triggers)
    triggers = session.getLowestUnprescaled(triggerType = TriggerType.g_single,  livefraction = 0.8).union(triggers)
    triggers = session.getLowestUnprescaled(triggerType = TriggerType.xe,        livefraction = 0.8).union(triggers)
 
    # Additional unprescaled triggers
    triggers.update( ['HLT_e26_lhtight_ivarloose_L1EM22VHI', 'HLT_e60_lhmedium_L1EM22VHI', 'HLT_e140_lhloose_L1EM22VHI', 'HLT_e300_etcut_L1EM22VHI'] )
    triggers.update( ['HLT_g140_loose_L1EM22VHI', 'HLT_g300_etcut_L1EM22VHI'] )
    triggers.update( ['HLT_mu60_L1MU14FCH'] )
    triggers.update( ['HLT_xe55_cell_xe70_tcpufit_xe90_pfsum_vssk_L1XE50', 'HLT_xe55_cell_xe70_tcpufit_xe95_pfsum_cssk_L1XE50', 'HLT_xe60_cell_xe95_pfsum_cssk_L1XE50', 'HLT_xe65_cell_xe100_mhtpufit_pf_L1XE50', 'HLT_xe65_cell_xe105_mhtpufit_em_L1XE50', 'HLT_xe75_cell_xe65_tcpufit_xe90_trkmht_L1XE50'] )
 
    # Merge and remove duplicates
    triggers = sorted(list(set(list(triggers))))
 
  acc = ComponentAccumulator()
  TriggerSkimmingTool = CompFactory.DerivationFramework.TriggerSkimmingTool
  acc.addPublicTool(TriggerSkimmingTool
    (
      name,
      TriggerListAND = [],
      TriggerListOR  = triggers,
      **kwargs
    ),
    primary = True
  )     
  
  return acc