#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.
virtual void	BeginOfEventAction (const G4Event *) override
	the hooks for G4 UA handling
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.
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
std::vector< stepdata >	eventSteps
	holds data extracted from steps
NTuple::Item< long >	m_nsteps
	handles for ntuple writing
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.
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

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();
  // If user did not set an explicit level, set a default
  if (m_lvl == MSG::NIL) {
    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 163 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 178 of file AthMessaging.h.

179{ return msg() << lvl; }

AthMessaging::msg

MsgStream & msg() const

The standard message stream.

Definition AthMessaging.h:163

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

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_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:

Classes

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ StepNtuple()

Member Function Documentation

◆ BeginOfEventAction()

◆ BeginOfRunAction()

◆ EndOfEventAction()

◆ initMessaging()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ setLevel()

◆ UserSteppingAction()

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ eventSteps

◆ m_dep

◆ m_imsg

◆ m_lvl

◆ m_msg_tls

◆ m_nm

◆ m_nsteps

◆ m_pdgcode

◆ m_step_x

◆ m_step_y

◆ m_step_z

◆ m_time