ATLAS Offline Software
Public Member Functions | Private Attributes | List of all members
EL::Detail::AlgorithmStateModule Class Referencefinal

a Module managing the state of the algorithms on the workers More...

#include <AlgorithmStateModule.h>

Inheritance diagram for EL::Detail::AlgorithmStateModule:
Collaboration diagram for EL::Detail::AlgorithmStateModule:

Public Member Functions

virtual StatusCode onInitialize (ModuleData &data) override
 action just before algorithms are initialized More...
 
virtual StatusCode onFinalize (ModuleData &data) override
 actions just before algorithms are finalized More...
 
virtual StatusCode onCloseInputFile (ModuleData &data) override
 actions before closing an input file More...
 
virtual StatusCode onNewInputFile (ModuleData &data) override
 actions after opening a new input file More...
 
virtual StatusCode onFileExecute (ModuleData &data) override
 actions just before fileExecute is called on algorithms More...
 
virtual StatusCode onExecute (ModuleData &data) override
 actions just before execute is called on algorithms More...
 
virtual StatusCode firstInitialize (ModuleData &data)
 action at the the very beginning of the worker job More...
 
virtual StatusCode preFileInitialize (ModuleData &data)
 action before opening the first file in the worker job More...
 
virtual StatusCode processInputs (ModuleData &data, IInputModuleActions &actions)
 process all input files More...
 
virtual StatusCode postFirstEvent (ModuleData &data)
 action after processing first event More...
 
virtual StatusCode postCloseInputFile (ModuleData &data)
 actions after CloseInputFile is called on the algorithms More...
 
virtual StatusCode postFinalize (ModuleData &data)
 actions after algorithms have been finalized More...
 
virtual StatusCode onWorkerEnd (ModuleData &data)
 action at the end of the worker job More...
 
virtual StatusCode postFileClose (ModuleData &data)
 action at end of the worker job More...
 
virtual void reportInputFailure (ModuleData &data)
 report that we failed to open our input file More...
 

Private Attributes

bool m_initialized {false}
 whether Algorithm::initialize has been called More...
 

Detailed Description

a Module managing the state of the algorithms on the workers

\warn This module ought to be the last module loaded. That ensures that the algorithms get to do their thing after the modules. If a module needs to act after the algorithms, a new callback should be added for that.

The algorithms require actually a fairly complicated state machine to maintain. Mostly because the functions of the Algorithm class have to be called in a specific order, and each needs to have a specific set of pre-conditions fulfilled. This logic used to be (before 19 Jan 19) in the Worker class itself, but I decided it is better to factor it out into a separate module.

Definition at line 36 of file AlgorithmStateModule.h.

Member Function Documentation

◆ firstInitialize()

StatusCode EL::Detail::Module::firstInitialize ( ModuleData data)
virtualinherited

action at the the very beginning of the worker job

This gets called as early as possible in the worker initialization. Essentially all that should happen before this is to load all the modules. The main purpose is to start any benchmarks that are meant to capture the initialization process as well.

Reimplemented in EL::Detail::MemoryMonitorModule, EL::Detail::AlgorithmMemoryModule, EL::Detail::AlgorithmTimerModule, and EL::Detail::StopwatchModule.

Definition at line 25 of file Module.cxx.

27  {
28  return ::StatusCode::SUCCESS;
29  }

◆ onCloseInputFile()

StatusCode EL::Detail::AlgorithmStateModule::onCloseInputFile ( ModuleData data)
overridevirtual

actions before closing an input file

Reimplemented from EL::Detail::Module.

Definition at line 95 of file AlgorithmStateModule.cxx.

97  {
98  return forAllAlgorithms (msg(), data, "endInputFile", [&] (AlgorithmData& alg) {
99  return alg->endInputFile ();});
100  }

◆ onExecute()

StatusCode EL::Detail::AlgorithmStateModule::onExecute ( ModuleData data)
overridevirtual

actions just before execute is called on algorithms

For now that is mostly used to point input modules to the right event

rationale: this will make sure that the post-processing runs for all algorithms for which the regular processing was run

Reimplemented from EL::Detail::Module.

Definition at line 134 of file AlgorithmStateModule.cxx.

136  {
137  RCU_CHANGE_INVARIANT (this);
138 
139  data.m_skipEvent = false;
140  auto iter = data.m_algs.begin();
141  try
142  {
143  for (auto end = data.m_algs.end();
144  iter != end; ++ iter)
145  {
146  iter->m_executeCount += 1;
147  if (iter->m_algorithm->execute() == StatusCode::FAILURE)
148  {
149  ANA_MSG_ERROR ("while calling execute() on algorithm " << iter->m_algorithm->getName());
150  return StatusCode::FAILURE;
151  }
152 
153  if (data.m_skipEvent)
154  {
155  iter->m_skipCount += 1;
156  return StatusCode::SUCCESS;
157  }
158  }
159  } catch (...)
160  {
161  Detail::report_exception (std::current_exception());
162  ANA_MSG_ERROR ("while calling execute() on algorithm " << iter->m_algorithm->getName());
163  return StatusCode::FAILURE;
164  }
165 
168  try
169  {
170  for (auto jter = data.m_algs.begin(), end = iter;
171  jter != end && !data.m_skipEvent; ++ jter)
172  {
173  if (jter->m_algorithm->postExecute() == StatusCode::FAILURE)
174  {
175  ANA_MSG_ERROR ("while calling postExecute() on algorithm " << iter->m_algorithm->getName());
176  return StatusCode::FAILURE;
177  }
178  }
179  } catch (...)
180  {
181  Detail::report_exception (std::current_exception());
182  ANA_MSG_ERROR ("while calling postExecute() on algorithm " << iter->m_algorithm->getName());
183  return StatusCode::FAILURE;
184  }
185 
186  return StatusCode::SUCCESS;
187  }

◆ onFileExecute()

StatusCode EL::Detail::AlgorithmStateModule::onFileExecute ( ModuleData data)
overridevirtual

actions just before fileExecute is called on algorithms

Reimplemented from EL::Detail::Module.

Definition at line 125 of file AlgorithmStateModule.cxx.

127  {
128  return forAllAlgorithms (msg(), data, "fileExecute", [&] (AlgorithmData& alg) {
129  return alg->fileExecute ();});
130  }

◆ onFinalize()

StatusCode EL::Detail::AlgorithmStateModule::onFinalize ( ModuleData data)
overridevirtual

actions just before algorithms are finalized

Reimplemented from EL::Detail::Module.

Definition at line 82 of file AlgorithmStateModule.cxx.

84  {
85  if (!m_initialized)
86  return StatusCode::SUCCESS;
87  if (forAllAlgorithms (msg(), data, "finalize", [&] (AlgorithmData& alg) {
88  return alg->finalize ();}).isFailure())
89  return StatusCode::FAILURE;
90  return StatusCode::SUCCESS;
91  }

◆ onInitialize()

StatusCode EL::Detail::AlgorithmStateModule::onInitialize ( ModuleData data)
overridevirtual

action just before algorithms are initialized

This is typically used for any setup that this module needs to do.

Reimplemented from EL::Detail::Module.

Definition at line 61 of file AlgorithmStateModule.cxx.

63  {
64  if (m_initialized)
65  {
66  ANA_MSG_ERROR ("getting second initialize call");
67  return StatusCode::FAILURE;
68  }
69  m_initialized = true;
70  AlgorithmWorkerData workerData;
71  workerData.m_histogramWorker = data.m_worker;
72  workerData.m_treeWorker = data.m_worker;
73  workerData.m_filterWorker = data.m_worker;
74  workerData.m_wk = data.m_worker;
75  workerData.m_evtStore = data.m_evtStore;
76  return forAllAlgorithms (msg(), data, "initialize", [&] (AlgorithmData& alg) {
77  return alg->initialize (workerData);});
78  }

◆ onNewInputFile()

StatusCode EL::Detail::AlgorithmStateModule::onNewInputFile ( ModuleData data)
overridevirtual

actions after opening a new input file

Reimplemented from EL::Detail::Module.

Definition at line 104 of file AlgorithmStateModule.cxx.

106  {
107  if (!m_initialized)
108  {
109  ANA_MSG_ERROR ("algorithms have not been initialized yet");
110  return StatusCode::FAILURE;
111  }
112 
113  if (data.m_inputTree == nullptr ||
114  data.m_inputTree->GetEntries() == 0)
115  return StatusCode::SUCCESS;
116 
117  if (forAllAlgorithms (msg(), data, "changeInput", [&] (AlgorithmData& alg) {
118  return alg->beginInputFile ();}).isFailure())
119  return StatusCode::FAILURE;
120  return StatusCode::SUCCESS;
121  }

◆ onWorkerEnd()

StatusCode EL::Detail::Module::onWorkerEnd ( ModuleData data)
virtualinherited

action at the end of the worker job

This is mostly meant/used to print job summary statements at the very end of worker job, and actually have them show up at or at least near the end of any log file.

Reimplemented in EL::Detail::MemoryMonitorModule, EL::Detail::StopwatchModule, and EL::Detail::LeakCheckModule.

Definition at line 101 of file Module.cxx.

103  {
104  return ::StatusCode::SUCCESS;
105  }

◆ postCloseInputFile()

StatusCode EL::Detail::Module::postCloseInputFile ( ModuleData data)
virtualinherited

actions after CloseInputFile is called on the algorithms

Right now that is only used to disconnect the TEvent object from the input file.

Reimplemented in EL::Detail::TEventModule.

Definition at line 49 of file Module.cxx.

51  {
52  return ::StatusCode::SUCCESS;
53  }

◆ postFileClose()

StatusCode EL::Detail::Module::postFileClose ( ModuleData data)
virtualinherited

action at end of the worker job

Executed just right before worker exit successfully at this stage all outputs have been created

Reimplemented in EL::Detail::GridReportingModule, and EL::Detail::PostClosedOutputsModule.

Definition at line 107 of file Module.cxx.

109  {
110  return ::StatusCode::SUCCESS;
111  }

◆ postFinalize()

StatusCode EL::Detail::Module::postFinalize ( ModuleData data)
virtualinherited

actions after algorithms have been finalized

This is usually used to calculate some job summary information that is to be stored it in the histogram output file.

Reimplemented in EL::Detail::TEventModule, EL::Detail::FileExecutedModule, EL::Detail::LeakCheckModule, EL::Detail::StopwatchModule, and EL::Detail::EventCountModule.

Definition at line 95 of file Module.cxx.

97  {
98  return ::StatusCode::SUCCESS;
99  }

◆ postFirstEvent()

StatusCode EL::Detail::Module::postFirstEvent ( ModuleData data)
virtualinherited

action after processing first event

This is mostly meant to set up benchmarks that record per-event performance. While a lot of initialization happens during initialize() there is a fair amount of initialization that happens on the first event, so when recording per-event performance the first event is sort of "special" and may need to be omitted.

Reimplemented in EL::Detail::MemoryMonitorModule, and EL::Detail::LeakCheckModule.

Definition at line 55 of file Module.cxx.

57  {
58  return ::StatusCode::SUCCESS;
59  }

◆ preFileInitialize()

StatusCode EL::Detail::Module::preFileInitialize ( ModuleData data)
virtualinherited

action before opening the first file in the worker job

This is mostly meant to allow loading the dictionaries before any files and associated information is loaded.

Definition at line 31 of file Module.cxx.

33  {
34  return ::StatusCode::SUCCESS;
35  }

◆ processInputs()

StatusCode EL::Detail::Module::processInputs ( ModuleData data,
IInputModuleActions actions 
)
virtualinherited

process all input files

This deviates slightly from the usual pattern for module functions in that I pass in the possible actions as an argument. See IInputModuleActions for details.

Reimplemented in EL::Detail::BatchInputModule, and EL::Detail::DirectInputModule.

Definition at line 83 of file Module.cxx.

85  {
86  return ::StatusCode::SUCCESS;
87  }

◆ reportInputFailure()

void EL::Detail::Module::reportInputFailure ( ModuleData data)
virtualinherited

report that we failed to open our input file

Reimplemented in EL::Detail::GridReportingModule.

Definition at line 61 of file Module.cxx.

63  {}

Member Data Documentation

◆ m_initialized

bool EL::Detail::AlgorithmStateModule::m_initialized {false}
private

whether Algorithm::initialize has been called

Definition at line 61 of file AlgorithmStateModule.h.


The documentation for this class was generated from the following files:
data
char data[hepevt_bytes_allocation_ATLAS]
Definition: HepEvt.cxx:11
SGout2dot.alg
alg
Definition: SGout2dot.py:243
ANA_MSG_ERROR
#define ANA_MSG_ERROR(xmsg)
Macro printing error messages.
Definition: Control/AthToolSupport/AsgMessaging/AsgMessaging/MessageCheck.h:294
mergePhysValFiles.end
end
Definition: DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py:93
EL::Detail::AlgorithmStateModule::m_initialized
bool m_initialized
whether Algorithm::initialize has been called
Definition: AlgorithmStateModule.h:61
RCU_CHANGE_INVARIANT
#define RCU_CHANGE_INVARIANT(x)
Definition: Assert.h:231
EL::Detail::report_exception
void report_exception(std::exception_ptr eptr)
print out the currently evaluated exception
Definition: PhysicsAnalysis/D3PDTools/EventLoop/Root/MessageCheck.cxx:27
python.AutoConfigFlags.msg
msg
Definition: AutoConfigFlags.py:7