/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/SimG4Core/PrintTrackNumber/src/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;
    }
}