TrackingAction.cxx

#include "SimCore/G4User/TrackingAction.h"
 
// LDMX
#include "SimCore/G4User/TrackMap.h"
#include "SimCore/G4User/UserPrimaryParticleInformation.h"
#include "SimCore/G4User/UserRegionInformation.h"
#include "SimCore/G4User/UserTrackInformation.h"
 
// Geant4
#include "G4PrimaryParticle.hh"
 
// STL
namespace simcore::g4user {
 
void TrackingAction::PreUserTrackingAction(const G4Track* track) {
  if (not track_map_.contains(track)) {
    // New Track
 
    // get track information and initialize our new track
    //  this will create a new track info object if it doesn't exist
    auto track_info{UserTrackInformation::get(track)};
    track_info->initialize(track);
 
    // Get the region info for where the track was created (could be NULL)
    auto region_info = static_cast<UserRegionInformation*>(
        track->GetLogicalVolumeAtVertex()->GetRegion()->GetUserInformation());
 
    // Get the gen status if track was primary
    int cur_gen_status = -1;
    if (track->GetDynamicParticle()->GetPrimaryParticle()) {
      auto primary_info = dynamic_cast<UserPrimaryParticleInformation*>(
          track->GetDynamicParticle()
              ->GetPrimaryParticle()
              ->GetUserInformation());
      if (primary_info) {
        cur_gen_status = primary_info->getHepEvtStatus();
      }
    }
 
    if (cur_gen_status == 1 or !region_info or
        region_info->getStoreSecondaries()) {
      track_info->setSaveFlag(true);
    }
 
    // insert this track into the event's track map
    track_map_.insert(track);
  }
 
  // Activate user tracking actions
  for (auto& tracking_action : tracking_actions_)
    tracking_action->preUserTrackingAction(track);
}
 
void TrackingAction::PostUserTrackingAction(const G4Track* track) {
  // Activate user tracking actions
  for (auto& tracking_action : tracking_actions_)
    tracking_action->postUserTrackingAction(track);
 
  auto track_info{UserTrackInformation::get(track)};
  if (track_info->getSaveFlag() and
      track->GetTrackStatus() == G4TrackStatus::fStopAndKill) {
    track_map_.save(track);
  }
}
 
}  // namespace simcore::g4user