TrackingAction.cxx

#include "SimCore/G4User/TrackingAction.h"
 
// LDMX
#include "SimCore/TrackMap.h"
#include "SimCore/UserPrimaryParticleInformation.h"
#include "SimCore/UserRegionInformation.h"
#include "SimCore/UserTrackInformation.h"
 
// Geant4
#include "G4PrimaryParticle.hh"
#include "G4VUserPrimaryParticleInformation.hh"
 
// STL
#include <iostream>
namespace simcore::g4user {
 
void TrackingAction::PreUserTrackingAction(const G4Track* track) {
  if (not trackMap_.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 regionInfo = (UserRegionInformation*)track->GetLogicalVolumeAtVertex()
                          ->GetRegion()
                          ->GetUserInformation();
 
    // Get the gen status if track was primary
    int curGenStatus = -1;
    if (track->GetDynamicParticle()->GetPrimaryParticle()) {
      auto primaryInfo = dynamic_cast<UserPrimaryParticleInformation*>(
          track->GetDynamicParticle()
              ->GetPrimaryParticle()
              ->GetUserInformation());
      if (primaryInfo) {
        curGenStatus = primaryInfo->getHepEvtStatus();
      }
    }
 
    if (curGenStatus == 1 or !regionInfo or regionInfo->getStoreSecondaries()) {
      track_info->setSaveFlag(true);
    }
 
    // insert this track into the event's track map
    trackMap_.insert(track);
  }
 
  // Activate user tracking actions
  for (auto& trackingAction : trackingActions_)
    trackingAction->PreUserTrackingAction(track);
}
 
void TrackingAction::PostUserTrackingAction(const G4Track* track) {
  // Activate user tracking actions
  for (auto& trackingAction : trackingActions_)
    trackingAction->PostUserTrackingAction(track);
 
  auto track_info{UserTrackInformation::get(track)};
  if (track_info->getSaveFlag() and
      track->GetTrackStatus() == G4TrackStatus::fStopAndKill) {
    trackMap_.save(track);
  }
}
 
}  // namespace simcore::g4user