ATLAS Offline Software
Public Member Functions | Private Member Functions | Private Attributes | List of all members
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. More...
 
virtual void PreUserTrackingAction (const G4Track *) override final
 Called before tracking a new particle. More...
 
virtual void PostUserTrackingAction (const G4Track *) override final
 Called after tracking a particle. More...
 
bool msgLvl (const MSG::Level lvl) const
 Test the output level. More...
 
MsgStream & msg () const
 The standard message stream. More...
 
MsgStream & msg (const MSG::Level lvl) const
 The standard message stream. More...
 
void setLevel (MSG::Level lvl)
 Change the current logging level. More...
 

Private Member Functions

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

Private Attributes

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

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  }

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  m_lvl = m_imsg ?
43  static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
44  MSG::INFO;
45 }

◆ 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 164 of file AthMessaging.h.

165 {
166  MsgStream* ms = m_msg_tls.get();
167  if (!ms) {
168  if (!m_initialized.test_and_set()) initMessaging();
169  ms = new MsgStream(m_imsg,m_nm);
170  m_msg_tls.reset( ms );
171  }
172 
173  ms->setLevel (m_lvl);
174  return *ms;
175 }

◆ 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 179 of file AthMessaging.h.

180 { return msg() << lvl; }

◆ 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_initialized.test_and_set()) initMessaging();
154  if (m_lvl <= lvl) {
155  msg() << lvl;
156  return true;
157  } else {
158  return false;
159  }
160 }

◆ 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  }

◆ 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(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
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  }

◆ 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.

◆ m_lvl

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

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ 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:
AtlasG4EventUserInfo::SetCurrentPrimaryGenParticle
void SetCurrentPrimaryGenParticle(HepMC::ConstGenParticlePtr p)
set m_currentPrimaryGenParticle, the pointer to the HepMC::GenParticle used to create the current G4P...
Definition: AtlasG4EventUserInfo.h:53
AthMessaging::m_lvl
std::atomic< MSG::Level > m_lvl
Current logging level.
Definition: AthMessaging.h:138
AtlasG4EventUserInfo
This class is attached to G4Event objects as UserInformation. It holds a pointer to the HepMC::GenEve...
Definition: AtlasG4EventUserInfo.h:21
G4UA::AthenaTrackingAction::m_subDetVolLevel
int m_subDetVolLevel
The level in the G4 volume hierarchy at which can we find the sub-detector name.
Definition: AthenaTrackingAction.h:43
HepMC::GenParticlePtr
GenParticle * GenParticlePtr
Definition: GenParticle.h:37
AtlasTrajectory
Class to store G4 trajectory information.
Definition: AtlasTrajectory.h:15
AthMessaging::m_imsg
std::atomic< IMessageSvc * > m_imsg
MessageSvc pointer.
Definition: AthMessaging.h:135
python.SystemOfUnits.ms
int ms
Definition: SystemOfUnits.py:132
Athena::getMessageSvc
IMessageSvc * getMessageSvc(bool quiet=false)
Definition: getMessageSvc.cxx:20
AthMessaging::setLevel
void setLevel(MSG::Level lvl)
Change the current logging level.
Definition: AthMessaging.cxx:28
TrackHelper
Definition: TrackHelper.h:14
AthMessaging::AthMessaging
AthMessaging()
Default constructor:
TrigConf::MSGTC::Level
Level
Definition: Trigger/TrigConfiguration/TrigConfBase/TrigConfBase/MsgStream.h:21
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition: AthMsgStreamMacros.h:29
AthMessaging::msg
MsgStream & msg() const
The standard message stream.
Definition: AthMessaging.h:164
G4UA::AthenaTrackingAction::m_secondarySavingLevel
int m_secondarySavingLevel
The saving level for secondaries.
Definition: AthenaTrackingAction.h:41
AthMessaging::m_nm
std::string m_nm
Message source name.
Definition: AthMessaging.h:129
xAOD::track
@ track
Definition: TrackingPrimitives.h:512
AthMessaging::initMessaging
void initMessaging() const
Initialize our message level and MessageSvc.
Definition: AthMessaging.cxx:39
AthMessaging::m_msg_tls
boost::thread_specific_ptr< MsgStream > m_msg_tls
MsgStream instance (a std::cout like with print-out levels)
Definition: AthMessaging.h:132
AtlasG4EventUserInfo::SetCurrentGenParticle
void SetCurrentGenParticle(HepMC::GenParticlePtr p)
set m_currentGenParticle, the pointer to the GenParticle corresponding to the current G4Track.
Definition: AtlasG4EventUserInfo.h:67