KillSecondariesTrackAction.cc

Go to the documentation of this file.

#include "SimG4Core/KillSecondaries/interface/KillSecondariesTrackAction.h"

#include "SimG4Core/Notification/interface/BeginOfTrack.h"
#include "SimG4Core/Notification/interface/TrackInformation.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "G4Track.hh"

using namespace CLHEP;

KillSecondariesTrackAction::KillSecondariesTrackAction(edm::ParameterSet const &p) {
  killHeavy = p.getParameter<bool>("KillHeavy");
  kmaxIon = p.getParameter<double>("IonThreshold") * MeV;
  kmaxProton = p.getParameter<double>("ProtonThreshold") * MeV;
  kmaxNeutron = p.getParameter<double>("NeutronThreshold") * MeV;

  edm::LogInfo("KillSecondaries") << "KillSecondariesTrackAction:: Killing"
                                  << " Flag " << killHeavy << " protons below " << kmaxProton << " MeV, neutrons below "
                                  << kmaxNeutron << " MeV and ions below " << kmaxIon << " MeV\n";
}

KillSecondariesTrackAction::~KillSecondariesTrackAction() {}

void KillSecondariesTrackAction::update(const BeginOfTrack *trk) {
  if (killHeavy) {
    G4Track *theTrack = (G4Track *)((*trk)());
    TrackInformation *trkInfo = (TrackInformation *)(theTrack->GetUserInformation());
    if (trkInfo) {
      int pdg = theTrack->GetDefinition()->GetPDGEncoding();
      if (!(trkInfo->isPrimary())) {  // Only secondary particles
        double ke = theTrack->GetKineticEnergy() / MeV;
        if ((((pdg / 1000000000 == 1 && ((pdg / 10000) % 100) > 0 && ((pdg / 10) % 100) > 0)) && (ke < kmaxIon)) ||
            ((pdg == 2212) && (ke < kmaxProton)) || ((pdg == 2112) && (ke < kmaxNeutron))) {
          theTrack->SetTrackStatus(fStopAndKill);
          edm::LogInfo("KillSecondaries")
              << "Kill Track " << theTrack->GetTrackID() << " Type " << theTrack->GetDefinition()->GetParticleName()
              << " Kinetic Energy " << ke << " MeV";
        }
      }
    }
  }
}