G4UA::StepNtuple Class Reference

#include <StepNtuple.h>

Inheritance diagram for G4UA::StepNtuple:

Collaboration diagram for G4UA::StepNtuple:

Classes
struct	stepdata
	simple struct to hold step information More...

Public Member Functions
	StepNtuple (const MSG::Level lvl=MSG::INFO)
	Constructor with message level argument for AthMessaging. More...
virtual void	BeginOfEventAction (const G4Event *) override
	the hooks for G4 UA handling More...
virtual void	EndOfEventAction (const G4Event *) override
virtual void	UserSteppingAction (const G4Step *) override
virtual void	BeginOfRunAction (const G4Run *) override
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
std::vector< stepdata >	eventSteps
	holds data extracted from steps More...
NTuple::Item< long >	m_nsteps
	handles for ntuple writing More...
NTuple::Array< float >	m_pdgcode
NTuple::Array< float >	m_step_x
NTuple::Array< float >	m_step_y
NTuple::Array< float >	m_step_z
NTuple::Array< float >	m_time
NTuple::Array< float >	m_dep
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

Definition at line 21 of file StepNtuple.h.

Constructor & Destructor Documentation

StepNtuple()

G4UA::StepNtuple::StepNtuple

(

const MSG::Level

lvl = MSG::INFO

)

Constructor with message level argument for AthMessaging.

Definition at line 22 of file StepNtuple.cxx.

    : AthMessaging(Gaudi::svcLocator()->service< IMessageSvc >( "MessageSvc" ),
                   "StepNtuple")
  {
    AthMessaging::setLevel(lvl);
  }

Member Function Documentation

BeginOfEventAction()

void G4UA::StepNtuple::BeginOfEventAction ( const G4Event *  )

overridevirtual

the hooks for G4 UA handling

Definition at line 29 of file StepNtuple.cxx.

  {
    eventSteps.clear();
  }

BeginOfRunAction()

void G4UA::StepNtuple::BeginOfRunAction ( const G4Run *  )

overridevirtual

Definition at line 72 of file StepNtuple.cxx.

  {
    NTupleFilePtr file1(ntupleSvc(), "/NTUPLES/FILE1");
 
    SmartDataPtr<NTuple::Directory>
      ntdir(ntupleSvc(),"/NTUPLES/FILE1/StepNtuple");
    if ( !ntdir ) {
      // otherwise create the directory
      ntdir = ntupleSvc()->createDirectory(file1,"StepNtuple");
    }
    if ( ! ntdir ) {
      ATH_MSG_ERROR( " failed to get ntuple directory" );
    }
 
    NTuplePtr nt(ntupleSvc(), "/NTUPLES/FILE1/StepNtuple/10");
 
    // Check if already booked
    if ( !nt ) {
      nt = ntupleSvc()->book( ntdir.ptr(), 10, CLID_ColumnWiseTuple, "G4 Step Ntuple" );
 
      if ( nt ) {
    ATH_MSG_DEBUG( "booked ntuple " );
    
    // WARNING!! Force limit to 50k tracks
    if( nt->addItem ("NSteps", m_nsteps,0, 50000).isFailure() ||
            nt->addItem ("PCode", m_nsteps, m_pdgcode ).isFailure() ||
            nt->addItem ("step_x", m_nsteps, m_step_x).isFailure() ||
            nt->addItem ("step_y", m_nsteps, m_step_y).isFailure() ||
            nt->addItem ("step_z", m_nsteps, m_step_z).isFailure() ||
            nt->addItem ("dep", m_nsteps, m_dep).isFailure() ||
            nt->addItem ("time", m_nsteps, m_time).isFailure() ) {
      ATH_MSG_ERROR( "Could not book ntuple!! " );
        }
      }
    }
  }

EndOfEventAction()

void G4UA::StepNtuple::EndOfEventAction ( const G4Event *  )

overridevirtual

Definition at line 34 of file StepNtuple.cxx.

  {
    ATH_MSG_DEBUG("Start end of event; size is " << eventSteps.size());
    m_nsteps = eventSteps.size();
 
    for(unsigned int k=0; k<eventSteps.size(); k++){
      m_pdgcode[k] = eventSteps[k].code;
      m_step_x[k] = eventSteps[k].x;
      m_step_y[k] = eventSteps[k].y;
      m_step_z[k] = eventSteps[k].z;
      m_time[k] = eventSteps[k].time;
      m_dep[k] = eventSteps[k].dep;
    }
 
    if(! ntupleSvc()->writeRecord("/NTUPLES/FILE1/StepNtuple/10").isSuccess()){
      ATH_MSG_ERROR( " failed to write record for this event" );
    }
  }

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.

{
  m_imsg = Athena::getMessageSvc();
  m_lvl = m_imsg ?
    static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
    MSG::INFO;
}

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.

{
  MsgStream* ms = m_msg_tls.get();
  if (!ms) {
    if (!m_initialized.test_and_set()) initMessaging();
    ms = new MsgStream(m_imsg,m_nm);
    m_msg_tls.reset( ms );
  }
 
  ms->setLevel (m_lvl);
  return *ms;
}

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.

{ return msg() << lvl; }

msgLvl()

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

inlineinherited

Test the output level.

Parameters

lvl	The message level to test against

Returns

boolean Indicating if messages at given level will be printed

Return values

true	Messages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

{
  if (!m_initialized.test_and_set()) initMessaging();
  if (m_lvl <= lvl) {
    msg() << lvl;
    return true;
  } else {
    return false;
  }
}

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.

{
  m_lvl = lvl;
}

UserSteppingAction()

void G4UA::StepNtuple::UserSteppingAction ( const G4Step *  aStep )

overridevirtual

Definition at line 53 of file StepNtuple.cxx.

  {
    if(eventSteps.size()<49000){
 
      stepdata theInfo{};
 
      theInfo.dep=aStep->GetTotalEnergyDeposit();
      theInfo.time=aStep->GetPreStepPoint()->GetGlobalTime();
      theInfo.code=aStep->GetTrack()->GetDefinition()->GetPDGEncoding();
      G4ThreeVector pos=aStep->GetPreStepPoint()->GetPosition();
      theInfo.x=pos.x();
      theInfo.y=pos.y();
      theInfo.z=pos.z();
 
      eventSteps.push_back(theInfo);
      ATH_MSG_VERBOSE("Stepping; size is " << eventSteps.size());
    }
  }

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.

eventSteps

std::vector<stepdata> G4UA::StepNtuple::eventSteps

private

holds data extracted from steps

Definition at line 44 of file StepNtuple.h.

m_dep

NTuple::Array<float> G4UA::StepNtuple::m_dep

private

Definition at line 53 of file StepNtuple.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_nsteps

NTuple::Item<long> G4UA::StepNtuple::m_nsteps

private

handles for ntuple writing

Definition at line 47 of file StepNtuple.h.

m_pdgcode

NTuple::Array<float> G4UA::StepNtuple::m_pdgcode

private

Definition at line 48 of file StepNtuple.h.

m_step_x

NTuple::Array<float> G4UA::StepNtuple::m_step_x

private

Definition at line 49 of file StepNtuple.h.

m_step_y

NTuple::Array<float> G4UA::StepNtuple::m_step_y

private

Definition at line 50 of file StepNtuple.h.

m_step_z

NTuple::Array<float> G4UA::StepNtuple::m_step_z

private

Definition at line 51 of file StepNtuple.h.

m_time

NTuple::Array<float> G4UA::StepNtuple::m_time

private

Definition at line 52 of file StepNtuple.h.

---

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

• StepNtuple.h
• StepNtuple.cxx