ATLAS Offline Software
Loading...
Searching...
No Matches
G4UA::AthenaTrackingAction Class Reference

User action for pre/post tracking truth handling. More...

#include <AthenaTrackingAction.h>

Inheritance diagram for G4UA::AthenaTrackingAction:
Collaboration diagram for G4UA::AthenaTrackingAction:

Public Member Functions

 AthenaTrackingAction (MSG::Level lvl, int secondarySavingLevel, int subDetVolLevel)
 Constructor.
virtual void PreUserTrackingAction (const G4Track *) override final
 Called before tracking a new particle.
virtual void PostUserTrackingAction (const G4Track *) override final
 Called after tracking a particle.
bool msgLvl (const MSG::Level lvl) const
 Test the output level.
MsgStream & msg () const
 The standard message stream.
MsgStream & msg (const MSG::Level lvl) const
 The standard message stream.
void setLevel (MSG::Level lvl)
 Change the current logging level.

Private Member Functions

void initMessaging () const
 Initialize our message level and MessageSvc.

Private Attributes

int m_secondarySavingLevel
 The saving level for secondaries.
int m_subDetVolLevel
 The level in the G4 volume hierarchy at which can we find the sub-detector name.
std::string m_nm
 Message source name.
boost::thread_specific_ptr< MsgStream > m_msg_tls
 MsgStream instance (a std::cout like with print-out levels)
std::atomic< IMessageSvc * > m_imsg { nullptr }
 MessageSvc pointer.
std::atomic< MSG::Level > m_lvl { MSG::NIL }
 Current logging level.
std::atomic_flag m_initialized ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
 Messaging initialized (initMessaging)

Detailed Description

User action for pre/post tracking truth handling.

Definition at line 18 of file AthenaTrackingAction.h.

Constructor & Destructor Documentation

◆ AthenaTrackingAction()

G4UA::AthenaTrackingAction::AthenaTrackingAction ( MSG::Level lvl,
int secondarySavingLevel,
int subDetVolLevel )

Constructor.

Definition at line 24 of file AthenaTrackingAction.cxx.

26 : AthMessaging("AthenaTrackingAction")
27 , m_secondarySavingLevel(secondarySavingLevel)
28 , m_subDetVolLevel(subDetVolLevel)
29 {
30 setLevel(lvl);
31 }
void setLevel(MSG::Level lvl)
Change the current logging level.
AthMessaging()
Default constructor:
int m_subDetVolLevel
The level in the G4 volume hierarchy at which can we find the sub-detector name.
int m_secondarySavingLevel
The saving level for secondaries.

Member Function Documentation

◆ initMessaging()

void AthMessaging::initMessaging ( ) const
privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

40{
42 // If user did not set an explicit level, set a default
43 if (m_lvl == MSG::NIL) {
44 m_lvl = m_imsg ?
45 static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
46 MSG::INFO;
47 }
48}
std::string m_nm
Message source name.
std::atomic< IMessageSvc * > m_imsg
MessageSvc pointer.
std::atomic< MSG::Level > m_lvl
Current logging level.
IMessageSvc * getMessageSvc(bool quiet=false)

◆ msg() [1/2]

MsgStream & AthMessaging::msg ( ) const
inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 163 of file AthMessaging.h.

164{
165 MsgStream* ms = m_msg_tls.get();
166 if (!ms) {
167 if (!m_initialized.test_and_set()) initMessaging();
168 ms = new MsgStream(m_imsg,m_nm);
169 m_msg_tls.reset( ms );
170 }
171
172 ms->setLevel (m_lvl);
173 return *ms;
174}
boost::thread_specific_ptr< MsgStream > m_msg_tls
MsgStream instance (a std::cout like with print-out levels)
void initMessaging() const
Initialize our message level and MessageSvc.

◆ msg() [2/2]

MsgStream & AthMessaging::msg ( const MSG::Level lvl) const
inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 178 of file AthMessaging.h.

179{ return msg() << lvl; }
MsgStream & msg() const
The standard message stream.

◆ msgLvl()

bool AthMessaging::msgLvl ( const MSG::Level lvl) const
inlineinherited

Test the output level.

Parameters
lvlThe message level to test against
Returns
boolean Indicating if messages at given level will be printed
Return values
trueMessages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

152{
153 if (m_lvl <= lvl) {
154 msg() << lvl;
155 return true;
156 } else {
157 return false;
158 }
159}

◆ PostUserTrackingAction()

void G4UA::AthenaTrackingAction::PostUserTrackingAction ( const G4Track * )
finaloverridevirtual

Called after tracking a particle.

Here we reset the AtlasTrajectory if it was created.

Definition at line 77 of file AthenaTrackingAction.cxx.

78 {
79 ATH_MSG_DEBUG("Finished tracking a particle");
80
81 // We are done tracking this particle, so reset the trajectory.
82 // TODO: consider caching the tracking manager once to reduce overhead.
83 G4EventManager::GetEventManager()->GetTrackingManager()->
84 SetStoreTrajectory(false);
85 }
#define ATH_MSG_DEBUG(x)

◆ PreUserTrackingAction()

void G4UA::AthenaTrackingAction::PreUserTrackingAction ( const G4Track * track)
finaloverridevirtual

Called before tracking a new particle.

If the track meets certain conditions, we save it in the AtlasG4EventUserInfo and possibly construct a new AtlasTrajectory which will be used for writing out truth particles later.

Definition at line 36 of file AthenaTrackingAction.cxx.

37 {
38 ATH_MSG_DEBUG("Starting to track a new particle");
39
40 // Use the TrackHelper code to identify the kind of particle.
41 TrackHelper trackHelper(track);
42
43 // Condition for storing the GenParticle in the AtlasG4EventUserInfo for later.
44 if (trackHelper.IsPrimary() || trackHelper.IsRegisteredSecondary())
45 {
46 HepMC::GenParticlePtr currentGenParticle = trackHelper.GetTrackInformation()->GetCurrentGenParticle();
47
48 // Assign the GenParticle to the AtlasG4EventUserInfo.
49 AtlasG4EventUserInfo* atlasG4EvtUserInfo = static_cast<AtlasG4EventUserInfo*>
50 (G4EventManager::GetEventManager()->GetConstCurrentEvent()->GetUserInformation());
51 if (trackHelper.IsPrimary()) atlasG4EvtUserInfo->SetCurrentPrimaryGenParticle(currentGenParticle);
52 atlasG4EvtUserInfo->SetCurrentGenParticle(std::move(currentGenParticle));
53 }
54
55 // Condition for creating a trajectory object to store truth.
56 if (trackHelper.IsPrimary() ||
57 (trackHelper.IsRegisteredSecondary() && m_secondarySavingLevel>1) ||
58 (trackHelper.IsSecondary() && m_secondarySavingLevel>2))
59 {
60 ATH_MSG_DEBUG("Preparing an AtlasTrajectory for saving truth");
61
62 // Create a new AtlasTrajectory for this particle
63 AtlasTrajectory* trajectory = new AtlasTrajectory(track, m_subDetVolLevel);
64
65 // Assign the trajectory to the tracking manager.
66 // TODO: consider caching the tracking manager once to reduce overhead.
67 auto trkMgr = G4EventManager::GetEventManager()->GetTrackingManager();
68 //trajectory->setTrackingManager(trkMgr);
69 trkMgr->SetStoreTrajectory(true);
70 trkMgr->SetTrajectory(trajectory);
71 }
72 }
void SetCurrentPrimaryGenParticle(HepMC::ConstGenParticlePtr p)
set m_currentPrimaryGenParticle, the pointer to the HepMC::GenParticle used to create the current G4P...
void SetCurrentGenParticle(HepMC::GenParticlePtr p)
set m_currentGenParticle, the pointer to the GenParticle corresponding to the current G4Track.
GenParticle * GenParticlePtr
Definition GenParticle.h:37

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl)
inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

29{
30 m_lvl = lvl;
31}

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }
mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

135{ nullptr };

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }
mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

138{ MSG::NIL };

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls
mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm
privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

◆ m_secondarySavingLevel

int G4UA::AthenaTrackingAction::m_secondarySavingLevel
private

The saving level for secondaries.

Definition at line 41 of file AthenaTrackingAction.h.

◆ m_subDetVolLevel

int G4UA::AthenaTrackingAction::m_subDetVolLevel
private

The level in the G4 volume hierarchy at which can we find the sub-detector name.

Definition at line 43 of file AthenaTrackingAction.h.


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