ATLAS Offline Software
Loading...
Searching...
No Matches
McAodWriterTool Class Reference

#include <McAodWriterTool.h>

Inheritance diagram for McAodWriterTool:
Collaboration diagram for McAodWriterTool:

Public Member Functions

 McAodWriterTool (const std::string &type, const std::string &name, const IInterface *parent)
 Constructor with parameters:
virtual ~McAodWriterTool ()
 Destructor:
StatusCode initialize ()
StatusCode execute ()
StatusCode finalize ()
StatusCode write (const TruthParticleContainer *mcParts)
 Process the TruthParticleContainer through the I/O backend.
ServiceHandle< StoreGateSvc > & evtStore ()
 The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > & detStore () const
 The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode sysInitialize () override
 Perform system initialization for an algorithm.
virtual StatusCode sysStart () override
 Handle START transition.
virtual std::vector< Gaudi::DataHandle * > inputHandles () const override
 Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * > outputHandles () const override
 Return this algorithm's output handles.
Gaudi::Details::PropertyBase & declareProperty (Gaudi::Property< T, V, H > &t)
void updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream & msg () const
bool msgLvl (const MSG::Level lvl) const

Static Public Member Functions

static const InterfaceID & interfaceID ()
 Inline methods:

Protected Member Functions

 McAodWriterTool ()
 Default constructor:
void setupBackend (Gaudi::Details::PropertyBase &ioBackendURL)
 Method to configure the back-end to write out the TruthParticleContainer.
void renounceArray (SG::VarHandleKeyArray &handlesArray)
 remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce (T &h)
void extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
 Add StoreName to extra input/output deps as needed.

Protected Attributes

StringProperty m_ioBackendURL
 URL of the I/O back-end (only "ASCII" for now...) glued with the name of the output file name.
StringProperty m_truthParticlesName
 Location of the TruthParticleContainer to be written out.
std::ofstream * m_ioBackend
 Abstract base class for the back-end.

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t

Private Member Functions

Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

StoreGateSvc_t m_evtStore
 Pointer to StoreGate (event store by default)
StoreGateSvc_t m_detStore
 Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * > m_vhka
bool m_varHandleArraysDeclared

Detailed Description

Definition at line 28 of file McAodWriterTool.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t
privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ McAodWriterTool() [1/2]

McAodWriterTool::McAodWriterTool ( const std::string & type,
const std::string & name,
const IInterface * parent )

Constructor with parameters:

Public methods:

Constructors

Definition at line 41 of file McAodWriterTool.cxx.

43 :
44 AthAlgTool( type, name, parent ),
45 m_ioBackend( nullptr )
46{
47 //
48 // Property declaration
49 //
50
51 declareProperty( "Output",
52 m_ioBackendURL = "ascii:mcaod.txt",
53 "Name of the back-end we'll use to write out the "
54 "TruthParticleContainer.\nEx: ascii:mcaod.txt" );
55 m_ioBackendURL.declareUpdateHandler( &McAodWriterTool::setupBackend,
56 this );
57
58 declareProperty( "TruthParticles",
59 m_truthParticlesName = "SpclMC",
60 "Input location of the TruthParticleContainer to write out" );
61
62 declareInterface<IIOMcAodTool>(this);
63}
AthAlgTool()
Default constructor:
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
std::ofstream * m_ioBackend
Abstract base class for the back-end.
void setupBackend(Gaudi::Details::PropertyBase &ioBackendURL)
Method to configure the back-end to write out the TruthParticleContainer.
StringProperty m_ioBackendURL
URL of the I/O back-end (only "ASCII" for now...) glued with the name of the output file name.
StringProperty m_truthParticlesName
Location of the TruthParticleContainer to be written out.

◆ ~McAodWriterTool()

McAodWriterTool::~McAodWriterTool ( )
virtual

Destructor:

Destructor.

Definition at line 67 of file McAodWriterTool.cxx.

68{
69 ATH_MSG_DEBUG("Calling destructor");
70
71 if ( m_ioBackend ) {
72 delete m_ioBackend;
73 m_ioBackend = nullptr;
74 }
75}
#define ATH_MSG_DEBUG(x)

◆ McAodWriterTool() [2/2]

McAodWriterTool::McAodWriterTool ( )
protected

Default constructor:

Member Function Documentation

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl,
const SG::VarHandleKeyType &  )
inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

158 {
160 hndl.value(),
161 hndl.documentation());
162
163 }

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T, V, H > & t)
inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

145 {
146 typedef typename SG::HandleClassifier<T>::type htype;
148 }
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const
inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )
inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

◆ execute()

StatusCode McAodWriterTool::execute ( )
virtual

Implements IIOMcAodTool.

Definition at line 105 of file McAodWriterTool.cxx.

106{
107 // retrieve the TruthParticles
108 const TruthParticleContainer * mcParts = nullptr;
109 if ( evtStore()->retrieve( mcParts, m_truthParticlesName ).isFailure() ||
110 nullptr == mcParts ) {
111 ATH_MSG_WARNING("Could not retrieve a TruthParticleContainer at ["
112 << m_truthParticlesName << "] !!");
113 return StatusCode::RECOVERABLE;
114 }
115
116 return write(mcParts);
117}
#define ATH_MSG_WARNING(x)
Athena::TPCnvVers::Current TruthParticleContainer
ServiceHandle< StoreGateSvc > & evtStore()
StatusCode write(const TruthParticleContainer *mcParts)
Process the TruthParticleContainer through the I/O backend.
retrieve(aClass, aKey=None)
Definition PyKernel.py:110

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps)
protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ finalize()

StatusCode McAodWriterTool::finalize ( )
virtual

Implements IIOMcAodTool.

Definition at line 96 of file McAodWriterTool.cxx.

97{
98 ATH_MSG_INFO("Finalizing " << name() << "...");
99 ATH_MSG_DEBUG("Closing backend...");
100 m_ioBackend->close();
101
102 return StatusCode::SUCCESS;
103}
#define ATH_MSG_INFO(x)

◆ initialize()

StatusCode McAodWriterTool::initialize ( )
virtual

Implements IIOMcAodTool.

Definition at line 79 of file McAodWriterTool.cxx.

80{
81 ATH_MSG_INFO("Initializing " << name() << "...");
82 // Get pointer to StoreGateSvc and cache it :
83 if ( !evtStore().retrieve().isSuccess() ) {
84 ATH_MSG_ERROR("Unable to retrieve pointer to StoreGateSvc");
85 return StatusCode::FAILURE;
86 }
87
88 // setup backend
89 if ( nullptr == m_ioBackend ) {
91 }
92
93 return StatusCode::SUCCESS;
94}
#define ATH_MSG_ERROR(x)

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const
overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ interfaceID()

const InterfaceID & IIOMcAodTool::interfaceID ( )
inlinestaticinherited

Inline methods:

Definition at line 59 of file IIOMcAodTool.h.

60{
61 return IID_IIOMcAodTool;
62}
static const InterfaceID IID_IIOMcAodTool("IIOMcAodTool", 1, 0)

◆ msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const
inlineinherited

Definition at line 24 of file AthCommonMsg.h.

24 {
25 return this->msgStream();
26 }

◆ msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl) const
inlineinherited

Definition at line 30 of file AthCommonMsg.h.

30 {
31 return this->msgLevel(lvl);
32 }

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const
overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h)
inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

381 {
382 h.renounce();
384 }
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce(T &h)

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ setupBackend()

void McAodWriterTool::setupBackend ( Gaudi::Details::PropertyBase & ioBackendURL)
protected

Method to configure the back-end to write out the TruthParticleContainer.

Definition at line 159 of file McAodWriterTool.cxx.

160{
161 // defaults
162 std::string protocol = "ascii";
163 std::string fileName = "mcaod.txt";
164
165 // reset internal state
166 if ( m_ioBackend ) {
167 delete m_ioBackend;
168 m_ioBackend = nullptr;
169 }
170
171 // caching URL
172 const std::string& url = m_ioBackendURL.value();
173
174 std::string::size_type protocolPos = url.find(s_protocolSep);
175
176 if ( std::string::npos != protocolPos ) {
177 protocol = url.substr( 0, protocolPos );
178 fileName = url.substr( protocolPos+1, std::string::npos );
179 } else {
180 //protocol = "ascii";
181 fileName = url;
182 }
183
184 // get the protocol name in lower cases
185 std::transform( protocol.begin(), protocol.end(),
186 protocol.begin(),
187 ToLower() );
188
189 if ( "ascii" == protocol ) {
190 m_ioBackend = new std::ofstream( fileName.c_str(),
191 std::ios::out | std::ios::trunc );
192
193 } else {
194 ATH_MSG_WARNING("UNKNOWN protocol [" << protocol << "] !!" << endmsg
195 << "Will use [ascii] instead...");
196 protocol = "ascii";
197 m_ioBackend = new std::ofstream( fileName.c_str(),
198 std::ios::out | std::ios::trunc );
199 }
200
201 ATH_MSG_DEBUG("Using protocol [" << protocol << "] and write to ["
202 << fileName << "]");
203}
#define endmsg
static const char *const s_protocolSep

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )
overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )
overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )
inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

308 {
309 // debug() << "updateVHKA for property " << p.name() << " " << p.toString()
310 // << " size: " << m_vhka.size() << endmsg;
311 for (auto &a : m_vhka) {
313 for (auto k : keys) {
314 k->setOwner(this);
315 }
316 }
317 }
std::vector< SG::VarHandleKeyArray * > m_vhka

◆ write()

StatusCode McAodWriterTool::write ( const TruthParticleContainer * mcParts)

Process the TruthParticleContainer through the I/O backend.

Non-const methods:

Definition at line 123 of file McAodWriterTool.cxx.

124{
125 if ( nullptr == mcParts ) {
126 ATH_MSG_WARNING("NULL pointer to TruthParticleContainer !!");
127 return StatusCode::RECOVERABLE;
128 }
129
130 std::ofstream& out = *m_ioBackend;
131 out << std::fixed;
132
133 out << "---" << std::setw(5) << mcParts->size() << std::endl;
134 for ( unsigned int i = 0; i != mcParts->size(); ++i ) {
135 const TruthParticle* mc = (*mcParts)[i];
136 out << "P " << std::setprecision(3) << std::setw(10) << mc->px()
137 << " " << std::setprecision(3) << std::setw(10) << mc->py()
138 << " " << std::setprecision(3) << std::setw(10) << mc->pz()
139 << " " << std::setprecision(3) << std::setw(10) << mc->e()
140 << " " << std::setprecision(3) << std::setw(10) << mc->pdgId()
141 << " " << std::setprecision(3) << std::setw(10) << mc->charge()
142 << " " << std::setprecision(3) << std::setw(10) << HepMC::barcode(mc->genParticle())
143 << std::endl;
144 out << "EtIsol "
145 << std::setw(3) << TruthParticleParameters::NbrOfCones
146 << " : ";
147 for ( unsigned int iCone = 0;
149 ++iCone ) {
150 out << " " << std::setprecision(3) << std::setw(10)
151 << mc->etIsol(static_cast<TruthParticleParameters::ConeSize>(iCone));
152 }
153 out << std::endl;
154 }
155
156 return StatusCode::SUCCESS;
157}
size_type size() const noexcept
Returns the number of elements in the collection.
int barcode(const T *p)
Definition Barcode.h:16
ConeSize
Enum for Cone size indexes (for isolation)
TruthParticle_v1 TruthParticle
Typedef to implementation.

Member Data Documentation

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_ioBackend

std::ofstream* McAodWriterTool::m_ioBackend
protected

Abstract base class for the back-end.

Definition at line 96 of file McAodWriterTool.h.

◆ m_ioBackendURL

StringProperty McAodWriterTool::m_ioBackendURL
protected

URL of the I/O back-end (only "ASCII" for now...) glued with the name of the output file name.

Ex: "ascii:/home/foo/mcaod.txt" If no protocol separator ':' is found, fallback is "ASCII"

Definition at line 88 of file McAodWriterTool.h.

◆ m_truthParticlesName

StringProperty McAodWriterTool::m_truthParticlesName
protected

Location of the TruthParticleContainer to be written out.

Definition at line 92 of file McAodWriterTool.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka
privateinherited

Definition at line 398 of file AthCommonDataStore.h.


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