PrintTrackNumberAction.cc

Go to the documentation of this file.

#include "SimG4Core/PrintTrackNumber/interface/PrintTrackNumberAction.h"

#include "SimG4Core/Notification/interface/EndOfTrack.h"
#include "SimG4Core/Notification/interface/EndOfEvent.h"

#include "G4Track.hh"
#include "G4Event.hh"
#include "G4VProcess.hh"

PrintTrackNumberAction::PrintTrackNumberAction(edm::ParameterSet const & p)
    : theNoTracks(0), theNoTracksThisEvent(0),
      theNoTracksNoUL(0), theNoTracksThisEventNoUL(0)
{
    theNoTracksToPrint = p.getUntrackedParameter<int>("EachNTrack",-1); 
    // do not count tracks killed by user limits (MinEkineCut for the moment only)
    bNoUserLimits = p.getUntrackedParameter<bool>("NoUserLimits", true); 
    std::cout << " PrintTrackNumberAction::bNoUserLimits " << bNoUserLimits << std::endl; 
}
 
PrintTrackNumberAction::~PrintTrackNumberAction() {}
 
void PrintTrackNumberAction::update(const EndOfTrack * trk)
{
    const G4Track * aTrack = (*trk)(); 

    theNoTracks++;
    theNoTracksThisEvent++;

    if (bNoUserLimits) 
    {
    bool countTrk = 1;
    // tracks that have been killed before first step (by MinEkineCut). 
    // In fact the track makes the first step, MinEkineCut process determines 
    // that the energy is too low, set it to 0, and then at the next step 
    // the 0-energy particle dies
    if (aTrack->GetCurrentStepNumber() == 2) 
    {
        const G4VProcess* proccur = 
        aTrack->GetStep()->GetPostStepPoint()->GetProcessDefinedStep();
        if (proccur != 0)
        {
        if (proccur->GetProcessName() == "MinEkineCut") 
        {
            countTrk = false;
        } 
        else 
        {
            // for e+, last step is annihil, while previous is MinEkineCut
            const G4VProcess* procprev = 
            aTrack->GetStep()->GetPreStepPoint()->GetProcessDefinedStep();
            if (procprev != 0)
            {
            if (procprev->GetProcessName() == "MinEkineCut") 
            {
                countTrk = false;
            }
            }
        }
        }
    }
    if (countTrk) 
    {
        theNoTracksNoUL++;
        theNoTracksThisEventNoUL++;
        if (theNoTracksToPrint > 0) 
        {
        if (theNoTracksThisEventNoUL%theNoTracksToPrint == 0) 
        {
            std::cout << "PTNA: Simulating Track Number = " 
                  << theNoTracksThisEventNoUL << std::endl;
        }
        }
    }
    } 
    else 
    {
    if (theNoTracksToPrint > 0) 
    {
        if (theNoTracksThisEvent%theNoTracksToPrint == 0) 
        {
        std::cout << "PTNA: Simulating Track Number = " 
              << theNoTracksThisEvent << std::endl;
        }
    }
    }
}


void PrintTrackNumberAction::update(const EndOfEvent * e)
{
    const G4Event * g4e = (*e)(); 
    std::cout << "PTNA: Event simulated= " << g4e->GetEventID() << " #tracks= ";
    if (bNoUserLimits) 
    {
    std::cout << theNoTracksThisEventNoUL << "  Total #tracks in run= " 
          << theNoTracksNoUL << " counting killed by UL= " << theNoTracks 
          << std::endl;
    theNoTracksThisEventNoUL = 0;
    } 
    else 
    {
    std::cout << theNoTracksThisEvent << "  Total #tracks in run= " 
          << theNoTracks << std::endl;
    theNoTracksThisEvent = 0;
    }
}