ATLAS Offline Software
Loading...
Searching...
No Matches
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
virtual StatusCode onFinalize (ModuleData &data) override
 actions just before algorithms are finalized
virtual StatusCode onCloseInputFile (ModuleData &data) override
 actions before closing an input file
virtual StatusCode onNewInputFile (ModuleData &data) override
 actions after opening a new input file
virtual StatusCode onFileExecute (ModuleData &data) override
 actions just before fileExecute is called on algorithms
virtual StatusCode onExecute (ModuleData &data) override
 actions just before execute is called on algorithms
virtual StatusCode firstInitialize (ModuleData &data)
 action at the the very beginning of the worker job
virtual StatusCode preFileInitialize (ModuleData &data)
 action before opening the first file in the worker job
virtual StatusCode processInputs (ModuleData &data, IInputModuleActions &actions)
 process all input files
virtual StatusCode postFirstEvent (ModuleData &data)
 action after processing first event
virtual StatusCode onFirstInputFile (ModuleData &data)
 actions after opening a the first input file
virtual StatusCode onNextInputFile (ModuleData &data)
 actions after opening an input file after the first one
virtual StatusCode postCloseInputFile (ModuleData &data)
 actions after CloseInputFile is called on the algorithms
virtual StatusCode postFinalize (ModuleData &data)
 actions after algorithms have been finalized
virtual StatusCode onWorkerEnd (ModuleData &data)
 action at the end of the worker job
virtual StatusCode postFileClose (ModuleData &data)
 action at end of the worker job
virtual void reportInputFailure (ModuleData &data)
 report that we failed to open our input file

Private Attributes

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

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::AlgorithmMemoryModule, EL::Detail::AlgorithmTimerModule, EL::Detail::MemoryMonitorModule, 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 97 of file AlgorithmStateModule.cxx.

99 {
100 return forAllAlgorithms (msg(), data, "endInputFile", [&] (AlgorithmData& alg) {
101 return alg->endInputFile ();});
102 }
char data[hepevt_bytes_allocation_ATLAS]
Definition HepEvt.cxx:11
MsgStream & msg
Definition testRead.cxx:32

◆ 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

Reimplemented from EL::Detail::Module.

Definition at line 135 of file AlgorithmStateModule.cxx.

137 {
138 data.m_skipEvent = false;
139 bool sequenceSkip = false;
140 const EventContext& ctx = Gaudi::Hive::currentContext();
141 for (auto& algData : data.m_algs)
142 {
143 try
144 {
145 if (algData.m_sequenceStart)
146 sequenceSkip = false;
147 else if (sequenceSkip)
148 {
149 algData.m_wasSkipped = true;
150 continue;
151 }
152
153 algData.m_executeCount += 1;
154 if (algData.m_algorithm->execute(ctx) == StatusCode::FAILURE)
155 {
156 ANA_MSG_ERROR ("while calling execute() on algorithm " << algData.m_algorithm->getName());
157 return StatusCode::FAILURE;
158 }
159
160 if (data.m_skipEvent)
161 {
162 algData.m_skipCount += 1;
163 algData.m_wasSkipped = true;
164 sequenceSkip = true;
165 data.m_skipEvent = false;
166 }
167 } catch (...)
168 {
169 Detail::report_exception (std::current_exception());
170 ANA_MSG_ERROR ("while calling execute() on algorithm " << algData.m_algorithm->getName());
171 return StatusCode::FAILURE;
172 }
173 }
174
175 for (auto& algData : data.m_algs)
176 {
177 try
178 {
179 // This will skip `postExecute` for all algorithms that called
180 // `setFilterPassed(false)` or that were skipped because of a
181 // prior algorithm calling `setFilterPassed(false)`.
182 if (algData.m_wasSkipped)
183 {
184 algData.m_wasSkipped = false;
185 continue;
186 }
187 if (algData.m_algorithm->postExecute() == StatusCode::FAILURE)
188 {
189 ANA_MSG_ERROR ("while calling postExecute() on algorithm " << algData.m_algorithm->getName());
190 return StatusCode::FAILURE;
191 }
192 } catch (...)
193 {
194 Detail::report_exception (std::current_exception());
195 ANA_MSG_ERROR ("while calling postExecute() on algorithm " << algData.m_algorithm->getName());
196 return StatusCode::FAILURE;
197 }
198 }
199
200 return StatusCode::SUCCESS;
201 }
#define ANA_MSG_ERROR(xmsg)
Macro printing error messages.
void report_exception(std::exception_ptr eptr)
print out the currently evaluated exception

◆ 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 126 of file AlgorithmStateModule.cxx.

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

◆ onFinalize()

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

actions just before algorithms are finalized

Reimplemented from EL::Detail::Module.

Definition at line 84 of file AlgorithmStateModule.cxx.

86 {
87 if (!m_initialized)
88 return StatusCode::SUCCESS;
89 if (forAllAlgorithms (msg(), data, "finalize", [&] (AlgorithmData& alg) {
90 return alg->finalize ();}).isFailure())
91 return StatusCode::FAILURE;
92 return StatusCode::SUCCESS;
93 }
bool m_initialized
whether Algorithm::initialize has been called

◆ onFirstInputFile()

StatusCode EL::Detail::Module::onFirstInputFile ( ModuleData & data)
virtualinherited

actions after opening a the first input file

Reimplemented in EL::Detail::EventModule.

Definition at line 37 of file Module.cxx.

39 {
40 return ::StatusCode::SUCCESS;
41 }

◆ 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 63 of file AlgorithmStateModule.cxx.

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

◆ onNewInputFile()

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

actions after opening a new input file

Reimplemented from EL::Detail::Module.

Definition at line 106 of file AlgorithmStateModule.cxx.

108 {
109 if (!m_initialized)
110 {
111 ANA_MSG_ERROR ("algorithms have not been initialized yet");
112 return StatusCode::FAILURE;
113 }
114
115 // Check that there are events on input
116 if (!data.m_hasInputEvents) return StatusCode::SUCCESS;
117
118 if (forAllAlgorithms (msg(), data, "changeInput", [&] (AlgorithmData& alg) {
119 return alg->beginInputFile ();}).isFailure())
120 return StatusCode::FAILURE;
121 return StatusCode::SUCCESS;
122 }

◆ onNextInputFile()

StatusCode EL::Detail::Module::onNextInputFile ( ModuleData & data)
virtualinherited

actions after opening an input file after the first one

Reimplemented in EL::Detail::EventModule.

Definition at line 43 of file Module.cxx.

45 {
46 return ::StatusCode::SUCCESS;
47 }

◆ 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::LeakCheckModule, EL::Detail::MemoryMonitorModule, and EL::Detail::StopwatchModule.

Definition at line 113 of file Module.cxx.

115 {
116 return ::StatusCode::SUCCESS;
117 }

◆ 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 Event object from the input file.

Reimplemented in EL::Detail::EventModule.

Definition at line 61 of file Module.cxx.

63 {
64 return ::StatusCode::SUCCESS;
65 }

◆ 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 119 of file Module.cxx.

121 {
122 return ::StatusCode::SUCCESS;
123 }

◆ 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::EventCountModule, EL::Detail::EventModule, EL::Detail::FileExecutedModule, EL::Detail::LeakCheckModule, and EL::Detail::StopwatchModule.

Definition at line 107 of file Module.cxx.

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

◆ 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::LeakCheckModule, and EL::Detail::MemoryMonitorModule.

Definition at line 67 of file Module.cxx.

69 {
70 return ::StatusCode::SUCCESS;
71 }

◆ 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 95 of file Module.cxx.

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

◆ 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 73 of file Module.cxx.

75 {}

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.

61{false};

The documentation for this class was generated from the following files: