/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_14/src/SimG4Core/CountProcesses/src/CountProcessesAction.cc

Go to the documentation of this file.

#include "SimG4Core/CountProcesses/interface/CountProcessesAction.h"

#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/EndOfRun.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"

#include "G4Track.hh"
#include "G4Run.hh"
#include "G4Event.hh"
#include "G4Step.hh"
#include "G4ProcessManager.hh"
#include "G4ParticleTable.hh"

CountProcessesAction::CountProcessesAction(edm::ParameterSet const & p)
    : fDEBUG(p.getUntrackedParameter<bool>("DEBUG",false))
{}

CountProcessesAction::~CountProcessesAction() {}

void CountProcessesAction::update(const BeginOfRun * run)
{
    G4ParticleTable * partTable = G4ParticleTable::GetParticleTable();
    int siz = partTable->size();
    for (int ii= 0; ii < siz; ii++)
    {
        G4ParticleDefinition * particle = partTable->GetParticle(ii);
        std::string particleName = particle->GetParticleName();
        if (fDEBUG)
            std::cout << ii << " PCA " << particleName<< " " << particle->GetPDGStable() 
                      << " " << particle->IsShortLived() << std::endl;
        theParticleList[particleName] = 0;

        //--- All processes of this particle 
        G4ProcessManager * pmanager = particle->GetProcessManager();
        G4ProcessVector * pvect = pmanager->GetProcessList();
        int sizproc = pvect->size();
        for (int jj = 0; jj < sizproc; jj++) 
        {
            std::string processName = (*pvect)[jj]->GetProcessName();
            if (fDEBUG)
                std::cout << jj << " PCR " << processName<< std::endl;
            theProcessList[pss(particleName,processName)] = 0;
        }
    }
    DumpProcessList(0);
}

void CountProcessesAction::update(const BeginOfTrack * trk)
{
    //----- Fill counter of particles
    const G4Track * aTrack = (*trk)();
    std::string particleName = aTrack->GetDefinition()->GetParticleName();
    theParticleList[particleName]++;

    //----- Fill counter of Creator Processes
    const G4VProcess * proc = aTrack->GetCreatorProcess();
    std::string processName;
    if (proc != 0) processName = proc->GetProcessName();
    else processName = "Primary";
    pss parproc(particleName,processName);
    mpssi::iterator ite = theCreatorProcessList.find(parproc);
    if (ite == theCreatorProcessList.end()) theCreatorProcessList[ parproc ] = 1;
    else (*ite).second = (*ite).second +1; 
    if (fDEBUG) 
        std::cout << " creator " << particleName << " " << processName 
                  << theCreatorProcessList.size() << std::endl;
}

void CountProcessesAction::update(const G4Step* aStep )
{
    std::string processName;
    if(aStep->GetPostStepPoint()->GetProcessDefinedStep() != 0)
        processName = aStep->GetPostStepPoint()->GetProcessDefinedStep()->GetProcessName();
    else processName = "User Limit";
    std::string particleName = aStep->GetTrack()->GetDefinition()->GetParticleName();
    theProcessList[pss(particleName,processName)] = theProcessList[pss(particleName,processName)] + 1;
}

void CountProcessesAction::update(const EndOfRun * run)
{
    DumpProcessList(1);
    DumpCreatorProcessList(1);
    DumpParticleList();
}

void CountProcessesAction::DumpProcessList(bool printNsteps, std::ostream& out)
{    
    mpssi::iterator ite;
    for (ite = theProcessList.begin(); ite != theProcessList.end(); ite++) 
    {
        if (!printNsteps) 
            out << "PROC_LIST " << (*ite).first.first << " : " 
                << (*ite) .first.second << std::endl; 
        else if ((*ite).second != 0)
            out << "PROC_COUNT " << (*ite).first.first << " : " 
                << (*ite) .first.second << " = " << (*ite).second << std::endl; 
    }
}

void CountProcessesAction::DumpCreatorProcessList(bool printNsteps, std::ostream& out)
{    
    mpssi::iterator ite;
    for (ite = theCreatorProcessList.begin(); ite != theCreatorProcessList.end(); ite++) 
    {
        if (!printNsteps) 
            out << "PROC-CREATOR_LIST " << (*ite).first.first << " : " 
                <<(*ite) .first.second << std::endl; 
        else if ((*ite).second != 0) 
            out << "PROC_CREATOR_COUNT " << (*ite).first.first << " : " 
                <<(*ite) .first.second << " = " << (*ite).second << std::endl; 
    }
}

void CountProcessesAction::DumpParticleList(std::ostream& out)
{    
    psi::iterator ite;
    for (ite = theParticleList.begin(); ite != theParticleList.end(); ite++) 
    {
        if ((*ite).second != 0) 
            out << "PART_LIST: " << (*ite).first << " = " << (*ite).second << std::endl; 
    }
}