Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include "PodioInput.h"
#include "TFile.h"
#include "TROOT.h"
#include "FWCore/DataWrapper.h"
#include "FWCore/PodioDataSvc.h"
DECLARE_COMPONENT(PodioInput)
PodioInput::PodioInput(const std::string& name, ISvcLocator* svcLoc) : GaudiAlgorithm(name, svcLoc) {}
StatusCode PodioInput::initialize() {
if (GaudiAlgorithm::initialize().isFailure()) return StatusCode::FAILURE;
// check whether we have the PodioEvtSvc active
m_podioDataSvc = dynamic_cast<PodioDataSvc*>(evtSvc().get());
if (nullptr == m_podioDataSvc) return StatusCode::FAILURE;
auto idTable = m_podioDataSvc->getCollectionIDs();
for (auto& name : m_collectionNames) {
debug() << "Finding collection " << name << " in collection registry." << endmsg;
if (!idTable->present(name)) {
error() << "Requested product " << name << " not found." << endmsg;
return StatusCode::FAILURE;
}
m_collectionIDs.push_back(idTable->collectionID(name));
}
return StatusCode::SUCCESS;
}
StatusCode PodioInput::execute() {
size_t cntr = 0;
// Re-create the collections from ROOT file
for (auto& id : m_collectionIDs) {
const std::string& collName = m_collectionNames.value().at(cntr++);
debug() << "Registering collection to read " << collName << " with id " << id << endmsg;
if (m_podioDataSvc->readCollection(collName, id).isFailure()) {
return StatusCode::FAILURE;
}
}
// Tell data service that we are done with requested collections
m_podioDataSvc->endOfRead();
return StatusCode::SUCCESS;
}
StatusCode PodioInput::finalize() {
if (GaudiAlgorithm::finalize().isFailure()) return StatusCode::FAILURE;
return StatusCode::SUCCESS;
}