G4UA::TestActionEHist Class Reference

#include <TestActionEHist.h>

Inheritance diagram for G4UA::TestActionEHist:

Collaboration diagram for G4UA::TestActionEHist:

Classes
struct	Config

Public Member Functions
	TestActionEHist (const Config &config)
virtual void	PreUserTrackingAction (const G4Track *) override
virtual void	PostUserTrackingAction (const G4Track *) override
virtual void	BeginOfRunAction (const G4Run *) override
virtual void	EndOfRunAction (const G4Run *) override
virtual void	UserSteppingAction (const G4Step *) override

Private Member Functions
void	BuildHists (const std::string &vol_tag, const std::string &part_tag, int &hLeft, double xfill=-1, double yfill=-1, const int nbins=3000, const int binsize=1)
	Size of bins in histogram, in MeV.
bool	BuildDirs (const std::string &vol_tag, const std::string &dirTitle, int &dLeft)
	Remaining directories to create.

Private Attributes
Config	m_config
	holds the python configuration
TFile *	m_world
	File in which to store neutron & electron info.
bool	m_firstStep
	Flag indicating whether step is first in current volume.
std::string	m_p_tag
	Used to specify current particle in tracking.
std::vector< std::string >	m_trajectory
	Used to store volume names which the current track has entered.

Detailed Description

Definition at line 46 of file TestActionEHist.h.

Constructor & Destructor Documentation

TestActionEHist()

G4UA::TestActionEHist::TestActionEHist

(

const Config &

config

)

Definition at line 44 of file TestActionEHist.cxx.

    : m_config(config), m_world(0), m_firstStep(false), m_p_tag("")
  {
  }

Member Function Documentation

BeginOfRunAction()

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

overridevirtual

Definition at line 72 of file TestActionEHist.cxx.

  {
    // initialize histogramming file (DON'T USE GAUDI) & directories
    delete m_world;
    m_world = new TFile(m_config.name.c_str(), "RECREATE");
    G4cout<<m_config.name<<" initialized, in directory "<<gDirectory->GetPath()<<G4endl;
  }

BuildDirs()

bool G4UA::TestActionEHist::BuildDirs ( const std::string & vol_tag, const std::string & dirTitle, int & dLeft )

private

Remaining directories to create.

Parameters

vol_tag	Function to generate TDirectory objects Tag to identify object by volume
dirTitle	Directory title

Definition at line 180 of file TestActionEHist.cxx.

  {
    bool enter = false;
    TDirectory* dExists = (TDirectory*)gDirectory->FindObjectAny(vol_tag.c_str());
    if (!dLeft && !dExists) { return enter; }
    if (!dExists) {
      dExists = new TDirectoryFile(vol_tag.c_str(),
                                   dirTitle.c_str());
      dLeft--;
 
      if (!dLeft) G4cout<<"Last directory created"<<G4endl;
    }
 
    if (dExists) enter = (bool)dExists->cd();
    //    if (enter) ATH_MSG_DEBUG("Current directory: "<<gDirectory->GetPath());
    return enter;
  }

BuildHists()

void G4UA::TestActionEHist::BuildHists ( const std::string & vol_tag, const std::string & part_tag, int & hLeft, double xfill = -1, double yfill = -1, const int nbins = 3000, const int binsize = 1 )

private

Size of bins in histogram, in MeV.

Parameters

vol_tag	Function to generate TH1-type histograms Tag to identify object by volume
part_tag	Tag to identify object by particle
hLeft	Amount of histogram space remaining, set by maxhists
xfill	Parameter to fill histogram along x
yfill	Paramteter to fill histogram along y
nbins	Number of bins in histogram

Definition at line 158 of file TestActionEHist.cxx.

  {
    TH1F* hExists =
      (TH1F*)gDirectory->FindObjectAny((part_tag+"_"+vol_tag+"_hist").c_str());
    if (!hLeft && !hExists) { return; }
    else if (!hExists) {
      hExists = new TH1F((part_tag+"_"+vol_tag+"_hist").c_str(),
                         (part_tag+" KE in "+vol_tag).c_str(),
                         nbins,0,nbins*binsize);
      hLeft--;
      //G4cout<<"Histogram "<<gDirectory->GetPath()<<"/"<<hExists->GetName()<<" created"<<G4endl;
      if (!hLeft) G4cout<<"Last histogram reached"<<G4endl;
    }
 
    if (xfill >= 0 && weight >= 0) { hExists->Fill(xfill,weight); }
    else if (xfill >= 0) { hExists->Fill(xfill); }
    else { return; }
  }

EndOfRunAction()

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

overridevirtual

Definition at line 80 of file TestActionEHist.cxx.

  {
    m_world->Write();
    m_world->Close();
    delete m_world;
    G4cout<<m_config.name<<" saved & closed."<<G4endl;
  }

PostUserTrackingAction()

void G4UA::TestActionEHist::PostUserTrackingAction ( const G4Track * aTrack )

overridevirtual

Definition at line 65 of file TestActionEHist.cxx.

  {
    if (aTrack) {
      m_trajectory.clear();
    }
  }

PreUserTrackingAction()

void G4UA::TestActionEHist::PreUserTrackingAction ( const G4Track * aTrack )

overridevirtual

Definition at line 50 of file TestActionEHist.cxx.

  {
    if (aTrack) {
 
      // Set Particle label and empty trajectory
      G4ParticleDefinition* pDef = aTrack->GetDefinition();
      const G4String& pName    = pDef->GetParticleName();
      const G4String& pSubType = pDef->GetParticleSubType();
      if (pName == "neutron" || pName == "proton" ) { m_p_tag = pName; }
      else if (pSubType =="e" || pSubType == "pi" ) { m_p_tag = pSubType; }
      else { m_p_tag = pDef->GetParticleType(); }
      if (!m_trajectory.empty()) m_trajectory.clear();
    }
  }

UserSteppingAction()

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

overridevirtual

Definition at line 88 of file TestActionEHist.cxx.

  {
    if (aStep) {
 
      // Create tree structure for current step
      VolumeTreeNavigator currentTree( aStep );
 
      // check for processes which don't need to be histogrammed
      //if ( currentTree.KillProcesses( 2, "nKiller", "G4FastSimulationManagerProcess" ) )  return;
 
      // Stores a list of the names of leaf-node volumes which the track has entered
      // and checks it for earlier occurences of the same volume
      m_firstStep = true;
      if (currentTree.GetStepNumber() > 1) {
        for ( vector<string>::const_iterator it = m_trajectory.begin();
              it != m_trajectory.end(); ++it ) {
          if ( *it == stringify( currentTree.GetVolume()->GetName() ) ) {
            m_firstStep = false;
            break;
          }
        }
      }
 
      // For particles' first step in the current volume
      if (m_firstStep) {
        // push_back trajectory element
        m_trajectory.push_back( stringify( currentTree.GetVolume()->GetName() ) );
 
        // set depth cut (MUST implement Simple if Detail is used)
        currentTree.SetDepthCutSimple(m_config.dCALO, m_config.dBeam, m_config.dIDET, m_config.dMUON);
        // Detailed depth cut format: /Atlas::Atlas/[level 1]/[level 2]/.../[last directory level]
        if ( !m_config.dDetail.empty() ) {
          currentTree.SetDepthCutDetail( m_config.dDetail.c_str() );
        }
 
        //for( auto histelement : currentTree.GetHistory())
        //    std::cout<<histelement.first->GetName()<< std::endl;
        //std::cout<<"end dump tree "<<std::endl;
        //std::cout<<"before ascending "<<currentTree.GetVolume()->GetName()<<std::endl;
 
        // go to ATLAS::ATLAS
        currentTree.Ascend( currentTree.GetCurrentDepth() );
 
        while ( true ) {
          // collect naming info for histogram
          G4VPhysicalVolume* pv = currentTree.GetVolume();
          string thPV = cleanstr(pv->GetName());
          string thNoDau = stringify(pv->GetLogicalVolume()->GetNoDaughters());
          int thRep = currentTree.GetCopyNumber();
 
          // construct keyname & directory title
          string v_name = (thRep == 16969 ? std::move(thPV) : thPV+"_"+stringify(thRep));
          string p_name = m_p_tag + ( currentTree.GetStepNumber() != 1 ? "_entred" : "_madein" );
          string title = thNoDau+" daughters, "+stringify(currentTree.GetCurrentDepth())+" from base";
 
          // Build and fill histogram at bottom of tree, then build dir and cd to it
          if ( currentTree.GetStepNumber() == 1 || currentTree.GetCurrentDepth() == currentTree.GetFullDepth() ) {
            BuildHists(v_name, p_name, m_config.maxhists, currentTree.GetPreStepPoint()->GetKineticEnergy());
          }
          if ( !currentTree.Descend() )  break;
          if ( !BuildDirs(v_name, title, m_config.maxdirs) ) break;
        }
 
        // Return to world to get ready for next step... FINALLY
        m_world->cd();
      }
    }
  }

Member Data Documentation

m_config

Config G4UA::TestActionEHist::m_config

private

holds the python configuration

Definition at line 71 of file TestActionEHist.h.

m_firstStep

bool G4UA::TestActionEHist::m_firstStep

private

Flag indicating whether step is first in current volume.

Definition at line 75 of file TestActionEHist.h.

m_p_tag

std::string G4UA::TestActionEHist::m_p_tag

private

Used to specify current particle in tracking.

Definition at line 77 of file TestActionEHist.h.

m_trajectory

std::vector<std::string> G4UA::TestActionEHist::m_trajectory

private

Used to store volume names which the current track has entered.

Definition at line 79 of file TestActionEHist.h.

m_world

TFile* G4UA::TestActionEHist::m_world

private

File in which to store neutron & electron info.

Definition at line 73 of file TestActionEHist.h.

---

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

• TestActionEHist.h
• TestActionEHist.cxx