Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
Public Member Functions | Private Attributes | List of all members
TruthHelper::GenAccessIO Class Reference

#include <GenAccessIO.h>

Collaboration diagram for TruthHelper::GenAccessIO:

Public Member Functions

 GenAccessIO ()
 
StatusCode getMC (MCParticleCollection &mcParticles, const bool ifgen=false, const std::string &key="GEN_EVENT") const
 
StatusCode getDH (const McEventCollection *&dh) const
 
StatusCode getDH (const McEventCollection *&dh, const std::string &key) const
 
StatusCode store (McEventCollection *storee, const std::string &key) const
 
 GenAccessIO ()
 
StatusCode getMC (MCParticleCollection &mcParticles, const bool ifgen=false, const std::string &key="GEN_EVENT") const
 
StatusCode getDH (const McEventCollection *&dh) const
 
StatusCode getDH (const McEventCollection *&dh, const std::string &key) const
 
StatusCode store (McEventCollection *storee, const std::string &key) const
 

Private Attributes

ServiceHandle< StoreGateSvc > m_sgSvc {"StoreGateSvc", "GenAccessIO"}
 

Detailed Description

Definition at line 21 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

Constructor & Destructor Documentation

◆ GenAccessIO() [1/2]

TruthHelper::GenAccessIO::GenAccessIO ( )
inline

Definition at line 24 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

24  : m_sgSvc("StoreGateSvc", "GenAccessIO") {
25  if (m_sgSvc.retrieve().isFailure()) {
26  throw StatusCode::FAILURE;
27  }
28  }

◆ GenAccessIO() [2/2]

TruthHelper::GenAccessIO::GenAccessIO ( )
inline

Definition at line 27 of file PhysicsAnalysis/TruthParticleID/McParticleTools/src/GenAccessIO.h.

27  {
28  if (m_sgSvc.retrieve().isFailure()) {
29  throw StatusCode::FAILURE;
30  }
31  }

Member Function Documentation

◆ getDH() [1/4]

StatusCode TruthHelper::GenAccessIO::getDH ( const McEventCollection *&  dh) const
inline

Definition at line 70 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

70  {
71  return m_sgSvc->retrieve(dh);
72  }

◆ getDH() [2/4]

StatusCode TruthHelper::GenAccessIO::getDH ( const McEventCollection *&  dh) const
inline

Definition at line 73 of file PhysicsAnalysis/TruthParticleID/McParticleTools/src/GenAccessIO.h.

73  {
74  return m_sgSvc->retrieve(dh);
75  }

◆ getDH() [3/4]

StatusCode TruthHelper::GenAccessIO::getDH ( const McEventCollection *&  dh,
const std::string &  key 
) const
inline

Definition at line 73 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

73  {
74  return m_sgSvc->retrieve(dh, key);
75  }

◆ getDH() [4/4]

StatusCode TruthHelper::GenAccessIO::getDH ( const McEventCollection *&  dh,
const std::string &  key 
) const
inline

Definition at line 76 of file PhysicsAnalysis/TruthParticleID/McParticleTools/src/GenAccessIO.h.

76  {
77  return m_sgSvc->retrieve(dh, key);
78  }

◆ getMC() [1/2]

StatusCode TruthHelper::GenAccessIO::getMC ( MCParticleCollection &  mcParticles,
const bool  ifgen = false,
const std::string &  key = "GEN_EVENT" 
) const
inline

Definition at line 29 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

29  {
30  MsgStream log(Athena::getMessageSvc(), "GenAccessIO");
31 
32  // Retrieve iterators for McEventCollections objects
33  SG::ConstIterator<McEventCollection> firstMEC;
34  SG::ConstIterator<McEventCollection> lastMEC;
35  if ( (m_sgSvc->retrieve(firstMEC, lastMEC)).isFailure() ) {
36  log << MSG::ERROR << "Could not retrieve iterators for McEventCollections" << endmsg;
37  }
38  const McEventCollection& mcColl = *firstMEC;
39  int icount = 0;
40  for ( ; firstMEC!= lastMEC; ++firstMEC) icount++;
41  log << MSG::DEBUG << "Number of McEventCollections= "<< icount << endmsg;
42 
43  // If there is more than one then do the retrieve with the key
44  if (icount > 1) {
45  log << MSG::DEBUG << "Key = " << key << endmsg;
46  const McEventCollection* mcCollptr = nullptr;
47  return this->getDH(mcCollptr, key);
48  }
49 
50  if (icount > 0) {
51  // Iterate over all McEvent records
52  McEventCollection::const_iterator itr;
53  for (itr = mcColl.begin(); itr!=mcColl.end(); ++itr) {
54  // Access the HepMC record which is wrapped within McEvent
55  const HepMC::GenEvent* genEvt = (*itr);
56  if (genEvt == 0) return StatusCode::FAILURE;
57  if (ifgen) {
58  for (auto it: *genEvt) {
59  if (!HepMC::is_simulation_particle(it)) mcParticles.push_back(it);
60  }
61  } else {
62  for (auto it: *genEvt) {
63  mcParticles.push_back(it);
64  }
65  }
66  }
67  }
68  return StatusCode::SUCCESS;
69  }

◆ getMC() [2/2]

StatusCode TruthHelper::GenAccessIO::getMC ( MCParticleCollection &  mcParticles,
const bool  ifgen = false,
const std::string &  key = "GEN_EVENT" 
) const
inline

Definition at line 32 of file PhysicsAnalysis/TruthParticleID/McParticleTools/src/GenAccessIO.h.

32  {
33  MsgStream log(Athena::getMessageSvc(), "GenAccessIO");
34 
35  // Retrieve iterators for McEventCollections objects
36  SG::ConstIterator<McEventCollection> firstMEC;
37  SG::ConstIterator<McEventCollection> lastMEC;
38  if ( (m_sgSvc->retrieve(firstMEC, lastMEC)).isFailure() ) {
39  log << MSG::ERROR << "Could not retrieve iterators for McEventCollections" << endmsg;
40  }
41  const McEventCollection& mcColl = *firstMEC;
42  int icount = 0;
43  for ( ; firstMEC!= lastMEC; ++firstMEC) icount++;
44  log << MSG::DEBUG << "Number of McEventCollections= "<< icount << endmsg;
45 
46  // If there is more than one then do the retrieve with the key
47  if (icount > 1) {
48  log << MSG::DEBUG << "Key = " << key << endmsg;
49  const McEventCollection* mcCollptr = nullptr;
50  return this->getDH(mcCollptr, key);
51  }
52 
53  if (icount > 0) {
54  // Iterate over all McEvent records
55  McEventCollection::const_iterator itr;
56  for (itr = mcColl.begin(); itr!=mcColl.end(); ++itr) {
57  // Access the HepMC record which is wrapped within McEvent
58  const HepMC::GenEvent* genEvt = (*itr);
59  if (genEvt == 0) return StatusCode::FAILURE;
60  if (ifgen) {
61  for (auto it: *genEvt) {
62  if (!HepMC::is_simulation_particle(it)) mcParticles.push_back(it);
63  }
64  } else {
65  for (auto it: *genEvt) {
66  mcParticles.push_back(it);
67  }
68  }
69  }
70  }
71  return StatusCode::SUCCESS;
72  }

◆ store() [1/2]

StatusCode TruthHelper::GenAccessIO::store ( McEventCollection *  storee,
const std::string &  key 
) const
inline

Definition at line 76 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

76  {
77  return m_sgSvc->record(storee, key);
78  }

◆ store() [2/2]

StatusCode TruthHelper::GenAccessIO::store ( McEventCollection *  storee,
const std::string &  key 
) const
inline

Definition at line 79 of file PhysicsAnalysis/TruthParticleID/McParticleTools/src/GenAccessIO.h.

79  {
80  return m_sgSvc->record(storee, key);
81  }

Member Data Documentation

◆ m_sgSvc

ServiceHandle< StoreGateSvc > TruthHelper::GenAccessIO::m_sgSvc {"StoreGateSvc", "GenAccessIO"}
private

Definition at line 80 of file Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h.

The documentation for this class was generated from the following file:
DataModel_detail::const_iterator
Const iterator class for DataVector/DataList.
Definition: DVLIterator.h:82
skel.it
it
Definition: skel.GENtoEVGEN.py:407
PyPoolBrowser.dh
dh
Definition: PyPoolBrowser.py:102
Athena::getMessageSvc
IMessageSvc * getMessageSvc(bool quiet=false)
Definition: getMessageSvc.cxx:20
HepMC::is_simulation_particle
bool is_simulation_particle(const T &p)
Method to establish if a particle (or barcode) was created during the simulation (TODO update to be s...
Definition: MagicNumbers.h:355
endmsg
#define endmsg
Definition: AnalysisConfig_Ntuple.cxx:63
TruthHelper::GenAccessIO::getDH
StatusCode getDH(const McEventCollection *&dh) const
Definition: Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h:70
McEventCollection
This defines the McEventCollection, which is really just an ObjectVector of McEvent objects.
Definition: McEventCollection.h:33
TruthHelper::GenAccessIO::m_sgSvc
ServiceHandle< StoreGateSvc > m_sgSvc
Definition: Generators/FlowAfterburner/FlowAfterburner/GenAccessIO.h:80
DataVector::end
const_iterator end() const noexcept
Return a const_iterator pointing past the end of the collection.
DEBUG
#define DEBUG
Definition: page_access.h:11
python.CaloCondTools.log
log
Definition: CaloCondTools.py:20
SG::ConstIterator
Definition: SGIterator.h:164
DataVector::begin
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
mapkey::key
key
Definition: TElectronEfficiencyCorrectionTool.cxx:37
Generated on Sat Apr 12 2025 22:10:11 for ATLAS Offline Software by   doxygen 1.8.18