#include <ScoringPlane.h>

Inheritance diagram for G4UA::ScoringPlane:

Collaboration diagram for G4UA::ScoringPlane:

Classes
struct	Config

Public Member Functions
	ScoringPlane (const Config &config)

	~ScoringPlane ()

virtual void	BeginOfRunAction (const G4Run *) override

virtual void	EndOfRunAction (const G4Run *) override

virtual void	UserSteppingAction (const G4Step *) override

virtual void	BeginOfEventAction (const G4Event *) override

virtual void	EndOfEventAction (const G4Event *) override

Private Attributes
Config	m_config

std::unique_ptr< TTree >	m_tree0

std::unique_ptr< TTree >	m_tree1

int	m_evt

int	m_ntr

int	m_pdg

double	m_cha

double	m_ene

double	m_vx

double	m_vy

double	m_vz

double	m_x0

double	m_y0

double	m_z0

double	m_t0

double	m_px0

double	m_py0

double	m_pz0

double	m_x1

double	m_y1

double	m_z1

double	m_t1

double	m_px1

double	m_py1

double	m_pz1

double	m_x

double	m_y

double	m_z

Detailed Description

Todo:: TODO NEEDS DOCUMENTATION

Definition at line 23 of file ScoringPlane.h.

Constructor & Destructor Documentation

◆ ScoringPlane()

G4UA::ScoringPlane::ScoringPlane ( const Config & config )

Definition at line 18 of file ScoringPlane.cxx.

     : m_config(config),
       m_evt(0), m_ntr(0), m_pdg(0), m_cha(0), m_ene(0),
       m_vx(0), m_vy(0), m_vz(0),
       m_x0(0), m_y0(0), m_z0(0), m_t0(0),
       m_px0(0), m_py0(0), m_pz0(0),
       m_x1(0), m_y1(0), m_z1(0), m_t1(0),
       m_px1(0), m_py1(0), m_pz1(0),
       m_x(0), m_y(0), m_z(0)
   {
   }

◆ ~ScoringPlane()

G4UA::ScoringPlane::~ScoringPlane ( )

Definition at line 30 of file ScoringPlane.cxx.

31 {

32 }

Member Function Documentation

◆ BeginOfEventAction()

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

overridevirtual

Definition at line 140 of file ScoringPlane.cxx.

   {
     m_evt++;
     m_ntr = 0;
   }

◆ BeginOfRunAction()

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

overridevirtual

Definition at line 34 of file ScoringPlane.cxx.

   {
  
     m_tree0 = std::make_unique<TTree>("t0", "ATHENA event tree");
  
     m_tree0->Branch("evt", &m_evt, "evt/I");
     m_tree0->Branch("ntr", &m_ntr, "ntr/I");
  
     m_tree1 = std::make_unique<TTree>("t1", "ATHENA particle tree");
  
     m_tree1->Branch("evt", &m_evt, "evt/I");
     m_tree1->Branch("ntr", &m_ntr, "ntr/I");
     m_tree1->Branch("pdg", &m_pdg, "pdg/I");
     m_tree1->Branch("cha", &m_cha, "cha/D");
     m_tree1->Branch("ene", &m_ene, "ene/D");
     m_tree1->Branch("x0",  &m_x0,  "x0/D");
     m_tree1->Branch("y0",  &m_y0,  "y0/D");
     m_tree1->Branch("z0",  &m_z0,  "z0/D");
     m_tree1->Branch("t0",  &m_t0,  "t0/D");
     m_tree1->Branch("px0", &m_px0, "px0/D");
     m_tree1->Branch("py0", &m_py0, "py0/D");
     m_tree1->Branch("pz0", &m_pz0, "pz0/D");
     m_tree1->Branch("x1",  &m_x1,  "x1/D");
     m_tree1->Branch("y1",  &m_y1,  "y1/D");
     m_tree1->Branch("z1",  &m_z1,  "z1/D");
     m_tree1->Branch("t1",  &m_t1,  "t1/D");
     m_tree1->Branch("px1", &m_px1, "px1/D");
     m_tree1->Branch("py1", &m_py1, "py1/D");
     m_tree1->Branch("pz1", &m_pz1, "pz1/D");
     m_tree1->Branch("vx",  &m_vx,  "vx/D");
     m_tree1->Branch("vy",  &m_vy,  "vy/D");
     m_tree1->Branch("vz",  &m_vz,  "vz/D");
     m_tree1->Branch("x",   &m_x,   "x/D");
     m_tree1->Branch("y",   &m_y,   "y/D");
     m_tree1->Branch("z",   &m_z,   "z/D");
  
     m_evt = 0;
  
     G4cout<<"ScoringPlane: placing scoring plane at [mm]: " << m_config.plane << G4endl;
     G4cout<<"ScoringPlane: stop and kill particles: "       << m_config.pkill << G4endl;
     G4cout<<"ScoringPlane: output root filename: "          << m_config.fname << G4endl;
   }

◆ EndOfEventAction()

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

overridevirtual

Definition at line 146 of file ScoringPlane.cxx.

   {
     m_tree0->Fill();
   }

◆ EndOfRunAction()

void G4UA::ScoringPlane::EndOfRunAction ( const G4Run * )

overridevirtual

Definition at line 77 of file ScoringPlane.cxx.

                                                {
     TFile* file = new TFile(m_config.fname.c_str(), "RECREATE", "ATHENA ufo simulation");
  
     m_tree0->Write();
     m_tree1->Write();
  
     file->Close();
   }

◆ UserSteppingAction()

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

overridevirtual

Definition at line 86 of file ScoringPlane.cxx.

   {
     m_z0 = aStep->GetPreStepPoint()->GetPosition().z();
     m_z1 = aStep->GetPostStepPoint()->GetPosition().z();
  
     if (m_z0*m_config.plane < 0)                     return; // take only particles on one side
     if (m_z1*m_config.plane < 0)                     return; // take only particles on one side
     if (std::fabs(m_z0) < std::fabs(m_config.plane)) return; // take only particles flowing towards the IP
     if (std::fabs(m_z1) > std::fabs(m_config.plane)) return; // take only particles flowing towards the IP
  
     m_ntr++;
  
     m_pdg = aStep->GetTrack()->GetDefinition()->GetPDGEncoding();
     m_cha = aStep->GetTrack()->GetDefinition()->GetPDGCharge();
     m_ene = aStep->GetTrack()->GetTotalEnergy();
     m_vx  = aStep->GetTrack()->GetVertexPosition().x();
     m_vy  = aStep->GetTrack()->GetVertexPosition().y();
     m_vz  = aStep->GetTrack()->GetVertexPosition().z();
  
     m_x0  = aStep->GetPreStepPoint()->GetPosition().x();
     m_y0  = aStep->GetPreStepPoint()->GetPosition().y();
     m_t0  = aStep->GetPreStepPoint()->GetGlobalTime()/CLHEP::ns;
     m_px0 = aStep->GetPreStepPoint()->GetMomentum().x();
     m_py0 = aStep->GetPreStepPoint()->GetMomentum().y();
     m_pz0 = aStep->GetPreStepPoint()->GetMomentum().z();
  
     m_x1  = aStep->GetPostStepPoint()->GetPosition().x();
     m_y1  = aStep->GetPostStepPoint()->GetPosition().y();
     m_t1  = aStep->GetPostStepPoint()->GetGlobalTime()/CLHEP::ns;
     m_px1 = aStep->GetPostStepPoint()->GetMomentum().x();
     m_py1 = aStep->GetPostStepPoint()->GetMomentum().y();
     m_pz1 = aStep->GetPostStepPoint()->GetMomentum().z();
  
     m_z = m_config.plane;
     m_x = m_x0 + (m_z0!=m_z1 ? (m_x1-m_x0)/(m_z1-m_z0)*(m_z-m_z0) : (m_x1-m_x0)*0.5);
     m_y = m_y0 + (m_z0!=m_z1 ? (m_y1-m_y0)/(m_z1-m_z0)*(m_z-m_z0) : (m_y1-m_y0)*0.5);
  
     m_tree1->Fill();
  
     if      (m_config.pkill == 1) aStep->GetTrack()->SetTrackStatus(fStopAndKill);
     else if (m_config.pkill == 2) aStep->GetTrack()->SetTrackStatus(fKillTrackAndSecondaries);
  
     //G4cout<<
     //  " z0: " << std::setw(10) << m_z0
     //        << " z1: " << std::setw(10) << m_z1
     //        << " x0: " << std::setw(10) << m_x0
     //        << " x1: " << std::setw(10) << m_x1
     //        << " x:  " << std::setw(10) << m_x
     //        << " y0: " << std::setw(10) << m_y0
     //        << " y1: " << std::setw(10) << m_y1
     //        << " y:  " << std::setw(10) << m_y << G4endl;
  
   }

Member Data Documentation

◆ m_cha

double G4UA::ScoringPlane::m_cha

private

Definition at line 54 of file ScoringPlane.h.

◆ m_config

Config G4UA::ScoringPlane::m_config

private

Definition at line 45 of file ScoringPlane.h.

◆ m_ene

double G4UA::ScoringPlane::m_ene

private

Definition at line 55 of file ScoringPlane.h.

◆ m_evt

int G4UA::ScoringPlane::m_evt

private

Definition at line 50 of file ScoringPlane.h.

◆ m_ntr

int G4UA::ScoringPlane::m_ntr

private

Definition at line 51 of file ScoringPlane.h.

◆ m_pdg

int G4UA::ScoringPlane::m_pdg

private

Definition at line 53 of file ScoringPlane.h.

◆ m_px0

double G4UA::ScoringPlane::m_px0

private

Definition at line 63 of file ScoringPlane.h.

◆ m_px1

double G4UA::ScoringPlane::m_px1

private

Definition at line 70 of file ScoringPlane.h.

◆ m_py0

double G4UA::ScoringPlane::m_py0

private

Definition at line 64 of file ScoringPlane.h.

◆ m_py1

double G4UA::ScoringPlane::m_py1

private

Definition at line 71 of file ScoringPlane.h.

◆ m_pz0

double G4UA::ScoringPlane::m_pz0

private

Definition at line 65 of file ScoringPlane.h.

◆ m_pz1

double G4UA::ScoringPlane::m_pz1

private

Definition at line 72 of file ScoringPlane.h.

◆ m_t0

double G4UA::ScoringPlane::m_t0

private

Definition at line 62 of file ScoringPlane.h.

◆ m_t1

double G4UA::ScoringPlane::m_t1

private

Definition at line 69 of file ScoringPlane.h.

◆ m_tree0

std::unique_ptr<TTree> G4UA::ScoringPlane::m_tree0

private

Definition at line 47 of file ScoringPlane.h.

◆ m_tree1

std::unique_ptr<TTree> G4UA::ScoringPlane::m_tree1

private

Definition at line 48 of file ScoringPlane.h.

◆ m_vx

double G4UA::ScoringPlane::m_vx

private

Definition at line 56 of file ScoringPlane.h.

◆ m_vy

double G4UA::ScoringPlane::m_vy

private

Definition at line 57 of file ScoringPlane.h.

◆ m_vz

double G4UA::ScoringPlane::m_vz

private

Definition at line 58 of file ScoringPlane.h.

◆ m_x

double G4UA::ScoringPlane::m_x

private

Definition at line 73 of file ScoringPlane.h.

◆ m_x0

double G4UA::ScoringPlane::m_x0

private

Definition at line 59 of file ScoringPlane.h.

◆ m_x1

double G4UA::ScoringPlane::m_x1

private

Definition at line 66 of file ScoringPlane.h.

◆ m_y

double G4UA::ScoringPlane::m_y

private

Definition at line 74 of file ScoringPlane.h.

◆ m_y0

double G4UA::ScoringPlane::m_y0

private

Definition at line 60 of file ScoringPlane.h.

◆ m_y1

double G4UA::ScoringPlane::m_y1

private

Definition at line 67 of file ScoringPlane.h.

◆ m_z

double G4UA::ScoringPlane::m_z

private

Definition at line 75 of file ScoringPlane.h.

◆ m_z0

double G4UA::ScoringPlane::m_z0

private

Definition at line 61 of file ScoringPlane.h.

◆ m_z1

double G4UA::ScoringPlane::m_z1

private

Definition at line 68 of file ScoringPlane.h.

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

Classes

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ ScoringPlane()

◆ ~ScoringPlane()

Member Function Documentation

◆ BeginOfEventAction()

◆ BeginOfRunAction()

◆ EndOfEventAction()

◆ EndOfRunAction()

◆ UserSteppingAction()

Member Data Documentation

◆ m_cha

◆ m_config

◆ m_ene

◆ m_evt

◆ m_ntr

◆ m_pdg

◆ m_px0

◆ m_px1

◆ m_py0

◆ m_py1

◆ m_pz0

◆ m_pz1

◆ m_t0

◆ m_t1

◆ m_tree0

◆ m_tree1

◆ m_vx

◆ m_vy

◆ m_vz

◆ m_x

◆ m_x0

◆ m_x1

◆ m_y

◆ m_y0

◆ m_y1

◆ m_z

◆ m_z0

◆ m_z1