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

97 {
98 return forAllAlgorithms (msg(), data, "endInputFile", [&] (AlgorithmData& alg) {
99 return alg->endInputFile ();});
100 }
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

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 data.m_skipEvent = false;
138 auto iter = data.m_algs.begin();
139 try
140 {
141 for (auto end = data.m_algs.end();
142 iter != end; ++ iter)
143 {
144 iter->m_executeCount += 1;
145 if (iter->m_algorithm->execute() == StatusCode::FAILURE)
146 {
147 ANA_MSG_ERROR ("while calling execute() on algorithm " << iter->m_algorithm->getName());
148 return StatusCode::FAILURE;
149 }
150
151 if (data.m_skipEvent)
152 {
153 iter->m_skipCount += 1;
154 return StatusCode::SUCCESS;
155 }
156 }
157 } catch (...)
158 {
159 Detail::report_exception (std::current_exception());
160 ANA_MSG_ERROR ("while calling execute() on algorithm " << iter->m_algorithm->getName());
161 return StatusCode::FAILURE;
162 }
163
166 try
167 {
168 for (auto jter = data.m_algs.begin(), end = iter;
169 jter != end && !data.m_skipEvent; ++ jter)
170 {
171 if (jter->m_algorithm->postExecute() == StatusCode::FAILURE)
172 {
173 ANA_MSG_ERROR ("while calling postExecute() on algorithm " << iter->m_algorithm->getName());
174 return StatusCode::FAILURE;
175 }
176 }
177 } catch (...)
178 {
179 Detail::report_exception (std::current_exception());
180 ANA_MSG_ERROR ("while calling postExecute() on algorithm " << iter->m_algorithm->getName());
181 return StatusCode::FAILURE;
182 }
183
184 return StatusCode::SUCCESS;
185 }
#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 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 }
bool m_initialized
whether Algorithm::initialize has been called

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

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

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

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.

61{false};

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