SaveSimTrack.cc

Go to the documentation of this file.

#include "SimG4Core/SaveSimTrackAction/interface/SaveSimTrack.h"

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

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

#include "G4Track.hh"
#include "G4PhysicalConstants.hh"
#include "G4SystemOfUnits.hh"
#include <algorithm>

SaveSimTrack::SaveSimTrack(edm::ParameterSet const & p) {

  edm::ParameterSet ps = p.getParameter<edm::ParameterSet>("SaveSimTrack");
  pdgs_ = ps.getUntrackedParameter<std::vector<int>>("PDGCodes");

  edm::LogInfo("SaveSimTrack") << "SaveSimTrack:: Save Sim Track if PDG code "
                   << "is one from the list of "  << pdgs_.size()
                   << " items";
  for (unsigned int k=0; k<pdgs_.size(); ++k)
    edm::LogInfo("SaveSimTrack") << "[" << k << "] " << pdgs_[k];
}

SaveSimTrack::~SaveSimTrack() {}
 
void SaveSimTrack::update(const BeginOfTrack * trk) {

  G4Track* theTrack = (G4Track*)((*trk)());
  TrackInformation * trkInfo = (TrackInformation *)(theTrack->GetUserInformation());
  if (trkInfo) {
    int pdg = theTrack->GetDefinition()->GetPDGEncoding();
    if (std::find(pdgs_.begin(),pdgs_.end(),pdg) != pdgs_.end()) {
      trkInfo->storeTrack(true);
      LogDebug("SaveSimTrack") << "Save SimTrack the Track " 
                   << theTrack->GetTrackID() << " Type " 
                   << theTrack->GetDefinition()->GetParticleName()
                   << " Momentum " << theTrack->GetMomentum()/MeV 
                   << " MeV/c";
    }
  }
}