PhotoNuclearProductsFilter.cxx

#include "Biasing/PhotoNuclearProductsFilter.h"
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <cmath>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4RunManager.hh"
#include "G4Step.hh"
#include "G4Track.hh"
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/G4User/UserTrackInformation.h"
 
namespace biasing {
 
PhotoNuclearProductsFilter::PhotoNuclearProductsFilter(
    const std::string& name, framework::config::Parameters& parameters)
    : simcore::UserAction(name, parameters) {
  products_pdg_id_ = parameters.get<std::vector<int> >("pdg_ids");
  min_e_ = parameters.get<double>("min_e");
}
 
PhotoNuclearProductsFilter::~PhotoNuclearProductsFilter() {}
 
void PhotoNuclearProductsFilter::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()};
 
  // Loop through all of the secondaries and check for the product of
  // interest.  This is done by getting the PDG ID of a daughter and
  // checking that it's in the vector of PDG IDs passed to this
  // UserAction.
  bool product_found{false};
  for (const auto& secondary : *secondaries) {
    // Get the PDG ID of the track
    auto pdg_id{std::abs(secondary->GetParticleDefinition()->GetPDGEncoding())};
 
    // Check if the PDG ID is in the list of products of interest
    if (std::find(products_pdg_id_.begin(), products_pdg_id_.end(), pdg_id) !=
        products_pdg_id_.end()) {
      if (secondary->GetKineticEnergy() > min_e_) {
        product_found = true;
      }
      break;
    }
  }
 
  // If the product of interest was not found, kill the track and abort
  // the event.
  if (!product_found) {
    track->SetTrackStatus(fKillTrackAndSecondaries);
    G4RunManager::GetRunManager()->AbortEvent();
    return;
  }
 
  // 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::PhotoNuclearProductsFilter)