TrackingAction.h

Go to the documentation of this file.

 
#ifndef SIMCORE_G4USER_USERTRACKINGACTION_H_
#define SIMCORE_G4USER_USERTRACKINGACTION_H_
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <vector>
 
// LDMX
#include "SimCore/G4User/TrackMap.h"
 
// Geant4
#include "G4RunManager.hh"
#include "G4UserTrackingAction.hh"
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/G4User/UserAction.h"
 
namespace simcore::g4user {
 
class TrackingAction : public G4UserTrackingAction {
 public:
  void PreUserTrackingAction(const G4Track* aTrack) override;
 
  void PostUserTrackingAction(const G4Track* aTrack) override;
 
  TrackMap& getTrackMap() { return track_map_; }
 
  static TrackingAction* get() {
    return static_cast<TrackingAction*>(const_cast<G4UserTrackingAction*>(
        G4RunManager::GetRunManager()->GetUserTrackingAction()));
  }
 
  void registerAction(std::shared_ptr<UserAction> trackingAction) {
    tracking_actions_.push_back(trackingAction);
  }
 
 private:
  std::vector<std::shared_ptr<UserAction>> tracking_actions_;
 
  TrackMap track_map_;
};  // TrackingAction
 
}  // namespace simcore::g4user
 
#endif