PhotoNuclearTopologyFilters.cxx

#include "Biasing/PhotoNuclearTopologyFilters.h"
 
#include <G4RunManager.hh>
 
#include "SimCore/G4User/UserTrackInformation.h"
 
namespace biasing {
 
bool NothingHardFilter::rejectEvent(
    const std::vector<G4Track*>& secondaries) const {
  for (const auto& secondary : secondaries) {
    // Get the PDG ID of the track
    const auto pdg_id{
        std::abs(secondary->GetParticleDefinition()->GetPDGEncoding())};
    if (skipCountingParticle(pdg_id)) {
      continue;
    }
    auto energy{secondary->GetKineticEnergy()};
    if (energy > hard_particle_threshold_) {
      return true;
    }
  }
  return false;
}
bool SingleNeutronFilter::rejectEvent(
    const std::vector<G4Track*>& secondaries) const {
  int hard_particles{0};
  int hard_neutrons{0};
  for (const auto& secondary : secondaries) {
    // Get the PDG ID of the track
    const auto pdg_id{
        std::abs(secondary->GetParticleDefinition()->GetPDGEncoding())};
    if (skipCountingParticle(pdg_id)) {
      continue;
    }
    auto energy{secondary->GetKineticEnergy()};
    if (energy > hard_particle_threshold_) {
      hard_particles++;
      if (isNeutron(pdg_id)) {
        hard_neutrons++;
      }
    }
  }
  auto reject{hard_particles != hard_neutrons || hard_particles != 1};
  return reject;
}
 
PhotoNuclearTopologyFilter::PhotoNuclearTopologyFilter(
    const std::string& name, framework::config::Parameters& parameters)
    : UserAction{name, parameters},
      count_light_ions_{parameters.get<bool>("count_light_ions")},
      hard_particle_threshold_{
          parameters.get<double>("hard_particle_threshold")} {}
 
void PhotoNuclearTopologyFilter::stepping(const G4Step* step) {
  // Get the track associated with this step.
  auto track{step->GetTrack()};
 
  // Get the track info and check if this track has been tagged as the
  // photon that underwent a photo-nuclear reaction. Only those tracks
  // tagged as PN photos will be processed. The track is currently only
  // tagged by the UserAction ECalProcessFilter which needs to be run
  // before this UserAction.
  auto track_info{simcore::UserTrackInformation::get(track)};
  if ((track_info != nullptr) && !track_info->isPNGamma()) return;
 
  // Get the PN photon daughters.
  auto secondaries{step->GetSecondary()};
 
  if (rejectEvent(*secondaries)) {
    track->SetTrackStatus(fKillTrackAndSecondaries);
    G4RunManager::GetRunManager()->AbortEvent();
  }
 
  // Once the PN gamma has been procesed, untag it so its not reprocessed
  // again.
  if (track_info) {
    track_info->tagPNGamma(false);
  }
}
 
}  // namespace biasing
 
DECLARE_ACTION(biasing::NothingHardFilter)
DECLARE_ACTION(biasing::SingleNeutronFilter)