EventAction.cc

Go to the documentation of this file.

//! \file
//!
//! Source file for the EventAction class.
//!
//! Defines the EventAction class and the member routines which
//! control the actions taken during the event.
//!
//! \author D.K. Hasell
//! \version 1.0
//! \date 2010-10-14
//!
//! \ingroup control

// Include user header files referenced in this file.

#include "EventAction.h"
#include "RunAction.h"

#include "GN_Data.h"
#include "EV_Data.h"
#include "GT_Data.h"
#include "WC_Data.h"
#include "TF_Data.h"
#include "LM_Data.h"
#include "MW_Data.h"
#include "SM_Data.h"

// Include the GEANT4 header files referenced in this file.

#include "G4Event.hh"
#include "G4PrimaryVertex.hh"
#include "G4PrimaryParticle.hh"
#include "G4ParticleDefinition.hh"

// Include required C++ header files.

#include <iomanip>

// Include the ROOT header files referenced in this file.

#include "TTree.h"

// Initialise the static data members.

GN_Data * EventAction::gndata = new GN_Data();
EV_Data * EventAction::evdata = new EV_Data();
GT_Data * EventAction::gtdata = new GT_Data();
WC_Data * EventAction::wcdata = new WC_Data();
TF_Data * EventAction::tfdata = new TF_Data();
LM_Data * EventAction::lmdata = new LM_Data();
MW_Data * EventAction::mwdata = new MW_Data();
SM_Data * EventAction::smdata = new SM_Data();

// Define the EventAction class.

// Constructor.

EventAction::EventAction() {}

// Destructor.

EventAction::~EventAction() {}

// Define what happens at the beginning of an event.

void EventAction::BeginOfEventAction( const G4Event * event ) {

   // Print the event information at the beginning of the event.
   // Note this happens after the primary particles are generated.

//   G4cout << "Beginning of Event " << event->GetEventID() << "\n" << G4endl;
//   Print( event );

   // Get event number.

   G4int id = event->GetEventID();
   
   // Print silly display to show it is working.

   if( id == 0 ) G4cout << "\nEvent processing status. One dot = 100 events."
                        << endl;
   if( id % 5000 == 0 ) G4cout << "\nEvent " << setw(10) << id << " " << flush;
   else if( id % 1000 == 0 ) G4cout << "|" << flush;
   else if( id %  500 == 0 ) G4cout << "+" << flush;
   else if( id %  100 == 0 ) G4cout << "." << flush;

}

// Define what happens at the end of an event.

void EventAction::EndOfEventAction( const G4Event * event ) {

   // Print the event information at the end of the event.
   // Note this happens after the sensitive detector end of event routines.

//   G4cout << "End of Event " << event->GetEventID() << "\n" << G4endl;
//   Print( event );

   // Get pointer to TTree.

   TTree * Tree = RunAction::Tree;

   // Define the various branches if not already defined.

   if( !Tree->GetBranchStatus("GN") ) Tree->Branch( "GN", "GN_Data", &gndata );

   if( !Tree->GetBranchStatus("EV") ) Tree->Branch( "EV", "EV_Data", &evdata );

   if( !Tree->GetBranchStatus("GT") ) Tree->Branch( "GT", "GT_Data", &gtdata );

   if( !Tree->GetBranchStatus("WC") ) Tree->Branch( "WC", "WC_Data", &wcdata );

   if( !Tree->GetBranchStatus("TF") ) Tree->Branch( "TF", "TF_Data", &tfdata );

   if( !Tree->GetBranchStatus("LM") ) Tree->Branch( "LM", "LM_Data", &lmdata );

   if( !Tree->GetBranchStatus("MW") ) Tree->Branch( "MW", "MW_Data", &mwdata );

   if( !Tree->GetBranchStatus("SM") ) Tree->Branch( "SM", "SM_Data", &smdata );

   // Reset and fill the EV event information.

   evdata->Reset();

   evdata->nEV = event->GetEventID();

   // Reset and fill the GN event information.

   gndata->Reset();

   G4int nPrim = event->GetNumberOfPrimaryVertex();

   gndata->nGN = nPrim;

   G4PrimaryVertex * vertex;
   G4PrimaryParticle * particle;
   G4ParticleDefinition * definition;

   for( G4int i = 0; i < nPrim; ++i ) {

      vertex = event->GetPrimaryVertex(i);

      particle = vertex->GetPrimary();

      definition = particle->GetG4code();

      gndata->id.push_back( particle->GetPDGcode() );
      gndata->q.push_back( particle->GetCharge() );
      gndata->tr.push_back( particle->GetTrackID() );

      gndata->x.push_back( vertex->GetX0()/cm );
      gndata->y.push_back( vertex->GetY0()/cm );
      gndata->z.push_back( vertex->GetZ0()/cm );

      gndata->px.push_back( particle->GetPx()/GeV );
      gndata->py.push_back( particle->GetPy()/GeV );
      gndata->pz.push_back( particle->GetPz()/GeV );

   }

   // Write the data to the Tree.

   RunAction::Tree->Fill();

}

// Print generated event information.

void EventAction::Print( const G4Event * event ) {

   G4int nPrim = event->GetNumberOfPrimaryVertex();
   G4PrimaryVertex * vertex;
   G4PrimaryParticle * particle;
   G4ParticleDefinition * definition;

   G4cout << "   Number of primary vertices " << nPrim << "\n" << G4endl;

   for( G4int i = 0; i < nPrim; ++i ) {

      vertex = event->GetPrimaryVertex(i);

      particle = vertex->GetPrimary();

      definition = particle->GetG4code();

      G4cout << "   Vertex " << i 
             << " PDG ID " << particle->GetPDGcode()
             << " Charge " << particle->GetCharge()
             << " Name " << definition->GetParticleName()
             << G4endl;

      G4cout << "      Origin "
             << vertex->GetX0()/cm << " " 
             << vertex->GetY0()/cm << " " 
             << vertex->GetZ0()/cm << " [cm]" << G4endl;

      G4cout << "      Momentum "
             << particle->GetPx()/GeV << " " 
             << particle->GetPy()/GeV << " " 
             << particle->GetPz()/GeV << " [GeV/c]\n" << G4endl;
   }
}