ATLAS Offline Software
TriggerListsHelper.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
2 # TriggerListsHelper: helper class which retrieves the full set of triggers needed for
3 # trigger matching in the DAODs and can then return them when needed.
4 
5 from TriggerMenuMT.TriggerAPI import TriggerAPISession, TriggerType, TriggerAPI, TriggerPeriod
6 from PathResolver import PathResolver
7 from AthenaConfiguration.AutoConfigFlags import GetFileMD
8 
9 def read_trig_list_file(fname):
10  """Read a text file containing a list of triggers
11  Returns the list of triggers held in the file
12  """
13  triggers = []
14  with open(PathResolver.FindCalibFile(fname)) as fp:
15  for line in fp:
16  line = line.strip()
17  if line == "" or line.startswith("#"):
18  continue
19  triggers.append(line)
20  return triggers
21 
22 def read_trig_list_flags(flags):
23  trigger_names_notau = []
24  trigger_names_tau = []
25  for chain_name in flags.Trigger.derivationsExtraChains:
26  if "tau" in chain_name:
27  trigger_names_tau.append(chain_name)
28  else:
29  trigger_names_notau.append(chain_name)
30  return (trigger_names_notau, trigger_names_tau)
31 
32 def getTapisSession(flags):
33  from Campaigns.Utils import Campaign
34  from AthenaConfiguration.Enums import Format
35  yearStr = str(flags.Input.DataYear) if not flags.Input.isMC else ""
36  if flags.Input.isMC:
37  if flags.Input.MCCampaign == Campaign.MC16a or flags.Input.MCCampaign == Campaign.MC20a:
38  yearStr = "2015_2016"
39  elif flags.Input.MCCampaign == Campaign.MC16d or flags.Input.MCCampaign == Campaign.MC20d:
40  yearStr = "2017"
41  elif flags.Input.MCCampaign == Campaign.MC16e or flags.Input.MCCampaign == Campaign.MC20e:
42  yearStr = "2018"
43  elif flags.Input.MCCampaign == Campaign.MC23a or flags.Input.MCCampaign == Campaign.MC21a:
44  yearStr = "2022"
45  elif flags.Input.MCCampaign == Campaign.MC23c or flags.Input.MCCampaign == Campaign.MC23d:
46  yearStr = "2023"
47  elif flags.Input.MCCampaign == Campaign.MC23e:
48  yearStr = "2024"
49 
50  session_files = {
51  "2015": "TriggerAPISessions/tapis_data15_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns.json",
52  "2016": "TriggerAPISessions/tapis_data16_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns_WITH_IGNORES.json",
53  "2015_2016": "TriggerAPISessions/tapis_data15_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns_data16_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns_WITH_IGNORES.json",
54  "2017": "TriggerAPISessions/tapis_data17_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns_Triggerno17e33prim.json",
55  "2018": "TriggerAPISessions/tapis_data18_13TeV_20190708_PHYS_StandardGRL_All_Good_25ns_Triggerno17e33prim.json",
56  "2022": "TriggerAPISessions/tapis_data22_13p6TeV_20230207_PHYS_StandardGRL_All_Good_25ns.json",
57  "2023": "TriggerAPISessions/tapis_data23_13p6TeV_20230828_PHYS_StandardGRL_All_Good_25ns.json",
58  "2024": "TriggerAPISessions/tapis_data24_13p6TeV_20241118_PHYS_StandardGRL_All_Good_25ns.json",
59  }
60 
61  if yearStr in session_files:
62  return TriggerAPISession(json=session_files[yearStr])
63 
64  # Otherwise: data24, MC23e, phase-II. Base this on the menu from the AOD.
65  if flags.Input.Format == Format.POOL:
66  return TriggerAPISession(file=flags.Input.Files[0])
67 
68  from AthenaCommon.Logging import logging
69  logging.getLogger('TriggerListHelper::GetTriggerLists::getTapisSession').error('Failed to obtain a session.')
70  return None
71 
72 
73 class TriggerListsHelper:
74  def __init__(self, flags):
75  self.flags = flags
76  TriggerAPI.setConfigFlags(flags)
77  self.Run2TriggerNamesTau = []
78  self.Run2TriggerNamesNoTau = []
79  self.Run3TriggerNames = []
80  self.GetTriggerLists()
81 
82  def GetTriggerLists(self):
83 
84  md = GetFileMD(self.flags.Input.Files)
85  hlt_menu = md.get('TriggerMenu', {}).get('HLTChains', None)
86 
87  if self.flags.Trigger.EDMVersion <= 2:
88 
89  # Trigger API for Runs 1, 2
90  #====================================================================
91  # TRIGGER CONTENT
92  #====================================================================
93 
96  allperiods = TriggerPeriod.y2015 | TriggerPeriod.y2016 | TriggerPeriod.y2017 | TriggerPeriod.y2018 | TriggerPeriod.future2e34
97  trig_el = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.el, livefraction=0.8)
98  trig_mu = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.mu, livefraction=0.8)
99  trig_g = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.g, livefraction=0.8)
100  trig_tau = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.tau, livefraction=0.8)
101 
102  trig_em = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.el, additionalTriggerType=TriggerType.mu, livefraction=0.8)
103  trig_et = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.el, additionalTriggerType=TriggerType.tau, livefraction=0.8)
104  trig_mt = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.mu, additionalTriggerType=TriggerType.tau, livefraction=0.8)
105  # Note that this seems to pick up both isolated and non-isolated triggers already, so no need for extra grabs
106  trig_txe = TriggerAPI.getLowestUnprescaledAnyPeriod(allperiods, triggerType=TriggerType.tau, additionalTriggerType=TriggerType.xe, livefraction=0.8)
107 
108 
109  extra_file_notau = read_trig_list_file("DerivationFrameworkPhys/run2ExtraMatchingTriggers.txt")
110  extra_file_tau = read_trig_list_file("DerivationFrameworkPhys/run2ExtraMatchingTauTriggers.txt")
111 
112 
113  extra_flag_notau, extra_flag_tau = read_trig_list_flags(self.flags)
114 
115 
116  trigger_names_full_notau = list(set(trig_el+trig_mu+trig_g+trig_em+trig_et+trig_mt+extra_file_notau+extra_flag_notau))
117  trigger_names_full_tau = list(set(trig_tau+trig_txe+extra_file_tau+extra_flag_tau))
118 
119 
120  trigger_names_notau = []
121  trigger_names_tau = []
122 
123  if hlt_menu:
124  for chain_name in hlt_menu:
125  if chain_name in trigger_names_full_notau: trigger_names_notau.append(chain_name)
126  if chain_name in trigger_names_full_tau: trigger_names_tau.append(chain_name)
127  else: # No means to filter based on in-file metadata
128  trigger_names_notau = trigger_names_full_notau
129  trigger_names_tau = trigger_names_full_tau
130 
131  self.Run2TriggerNamesNoTau = trigger_names_notau
132  self.Run2TriggerNamesTau = trigger_names_tau
133 
134  else: # Run 3 and Run 4
135 
136  # TriggerAPI Session based trigger lists
137  session = getTapisSession(self.flags)
138  lf = 0.8 # Prescale weighted life fraction of the GRL's LBs
139  api_trigger_names = set()
140  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.el, livefraction=lf).union(api_trigger_names)
141  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.mu, livefraction=lf).union(api_trigger_names)
142  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.g, livefraction=lf).union(api_trigger_names)
143  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.tau, livefraction=lf).union(api_trigger_names)
144 
145  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.el, TriggerType.mu], livefraction=lf).union(api_trigger_names)
146  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.el, TriggerType.tau], livefraction=lf).union(api_trigger_names)
147  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.mu, TriggerType.tau], livefraction=lf).union(api_trigger_names)
148  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.tau, TriggerType.xe], livefraction=lf).union(api_trigger_names)
149 
150  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.el, TriggerType.g], livefraction=lf).union(api_trigger_names)
151  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.el, TriggerType.xe], livefraction=lf).union(api_trigger_names)
152  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.mu, TriggerType.g], livefraction=lf).union(api_trigger_names)
153  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.g, TriggerType.xe], livefraction=lf).union(api_trigger_names)
154  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.tau, TriggerType.g], livefraction=lf).union(api_trigger_names)
155 
156  # Add hadronic categories, new for Run 3
157  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.j, livefraction=lf).union(api_trigger_names)
158  api_trigger_names = session.getLowestUnprescaled(triggerType=TriggerType.bj, livefraction=lf).union(api_trigger_names)
159  # Add cross-triggers as well
160  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.el, TriggerType.j], livefraction=lf).union(api_trigger_names)
161  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.mu, TriggerType.j], livefraction=lf).union(api_trigger_names)
162  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.j, TriggerType.bj], livefraction=lf).union(api_trigger_names)
163  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.j, TriggerType.g], livefraction=lf).union(api_trigger_names)
164  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.j, TriggerType.tau], livefraction=lf).union(api_trigger_names)
165  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.j, TriggerType.xe], livefraction=lf).union(api_trigger_names)
166  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.j, TriggerType.ht], livefraction=lf).union(api_trigger_names)
167  api_trigger_names = session.getLowestUnprescaled(triggerType=[TriggerType.bj, TriggerType.xe], livefraction=lf).union(api_trigger_names)
168 
169 
170  extra_flag_notau, extra_flag_tau = read_trig_list_flags(self.flags)
171  extra_flag = extra_flag_notau + extra_flag_tau
172 
173 
174  extra_file = read_trig_list_file("DerivationFrameworkPhys/run3ExtraMatchingTriggers.txt")
175 
176 
177  self.Run3TriggerNames = list(set(extra_file + extra_flag + list(api_trigger_names)))
178 
179  return
python.TriggerListsHelper.TriggerListsHelper.flags
flags
Definition: TriggerListsHelper.py:75
python.AutoConfigFlags.GetFileMD
def GetFileMD(filenames, allowEmpty=True, maxLevel='peeker')
Definition: AutoConfigFlags.py:65
python.TriggerListsHelper.TriggerListsHelper.Run2TriggerNamesTau
Run2TriggerNamesTau
Definition: TriggerListsHelper.py:77
PathResolver::FindCalibFile
static std::string FindCalibFile(const std::string &logical_file_name)
Definition: PathResolver.h:108
python.TriggerListsHelper.TriggerListsHelper.Run2TriggerNamesNoTau
Run2TriggerNamesNoTau
See https://twiki.cern.ch/twiki/bin/view/Atlas/TriggerAPI Get single and multi mu,...
Definition: TriggerListsHelper.py:78
python.TriggerListsHelper.TriggerListsHelper.GetTriggerLists
def GetTriggerLists(self)
Definition: TriggerListsHelper.py:82
python.TriggerListsHelper.read_trig_list_flags
def read_trig_list_flags(flags)
Definition: TriggerListsHelper.py:22
histSizes.list
def list(name, path='/')
Definition: histSizes.py:38
CxxUtils::set
constexpr std::enable_if_t< is_bitmask_v< E >, E & > set(E &lhs, E rhs)
Convenience function to set bits in a class enum bitmask.
Definition: bitmask.h:232
python.TriggerListsHelper.read_trig_list_file
def read_trig_list_file(fname)
Definition: TriggerListsHelper.py:9
python.TriggerListsHelper.TriggerListsHelper
Definition: TriggerListsHelper.py:73
Trk::open
@ open
Definition: BinningType.h:40
get
T * get(TKey *tobj)
get a TObject* from a TKey* (why can't a TObject be a TKey?)
Definition: hcg.cxx:127
str
Definition: BTagTrackIpAccessor.cxx:11
python.TriggerListsHelper.getTapisSession
def getTapisSession(flags)
Definition: TriggerListsHelper.py:32
python.TriggerListsHelper.TriggerListsHelper.Run3TriggerNames
Run3TriggerNames
Add Run 2 cross-triggers for some sets.
Definition: TriggerListsHelper.py:79
error
Definition: IImpactPoint3dEstimator.h:70
python.TriggerListsHelper.TriggerListsHelper.__init__
def __init__(self, flags)
Definition: TriggerListsHelper.py:74
Generated on Wed Jan 8 2025 21:19:53 for ATLAS Offline Software by   doxygen 1.8.18