G4TrackToParticleID.cc

Go to the documentation of this file.

#include "SimG4Core/Notification/interface/G4TrackToParticleID.h"

#include "G4Track.hh"

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

int G4TrackToParticleID::particleID(const G4Track * g4trk)
{
    int particleID_ = g4trk->GetDefinition()->GetPDGEncoding();
    if (0 == particleID_) {
      edm::LogWarning("SimG4CoreNotification") 
    << "G4TrackToParticleID: unknown code 0 for track Id = " << g4trk->GetTrackID();
      particleID_ = -99;
    }
    return particleID_;
}

bool G4TrackToParticleID::isGammaElectronPositron(int pdgCode)
{
  int pdg = std::abs(pdgCode);
  return (pdg == 11 || pdg == 22);
}

bool G4TrackToParticleID::isGammaElectronPositron(const G4Track * g4trk)
{
  int pdg = std::abs(g4trk->GetDefinition()->GetPDGEncoding());
  return (pdg == 11 || pdg == 22);
}

bool G4TrackToParticleID::isMuon(int pdgCode)
{
  return (std::abs(pdgCode) == 13);
}

bool G4TrackToParticleID::isMuon(const G4Track * g4trk)
{
  return (std::abs(g4trk->GetDefinition()->GetPDGEncoding()) == 13);
}

bool G4TrackToParticleID::isStableHadron(int pdgCode)
{
  // pi+-, p, pbar, n, nbar, KL, K+-, light ions and anti-ions 
  int pdg = std::abs(pdgCode);
  return (pdg == 211 || pdg == 2212 || pdg == 2112 || pdg == 130 || pdg == 321
      || pdg == 1000010020 || pdg == 1000010030 
      || pdg == 1000020030 || pdg == 1000020040);
}

bool G4TrackToParticleID::isStableHadronIon(const G4Track * g4trk)
{
  // pi+-, p, pbar, n, nbar, KL, K+-, light ion and anti-ion, generic ion
  int pdg = std::abs(g4trk->GetDefinition()->GetPDGEncoding());
  return (pdg == 211 || pdg == 2212 || pdg == 2112 || pdg == 130 || pdg == 321
      || pdg == 1000010020 || pdg == 1000010030 
      || pdg == 1000020030 || pdg == 1000020040
      || g4trk->GetDefinition()->IsGeneralIon());
}