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/TrackMap.h"
 
// Geant4
#include "G4RunManager.hh"
#include "G4UserTrackingAction.hh"
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/UserAction.h"
 
namespace simcore::g4user {
 
class TrackingAction : public G4UserTrackingAction {
 public:
  TrackingAction() {}
 
  virtual ~TrackingAction() {}
 
  void PreUserTrackingAction(const G4Track* aTrack);
 
  void PostUserTrackingAction(const G4Track* aTrack);
 
  TrackMap& getTrackMap() { return trackMap_; }
 
  static TrackingAction* get() {
    return static_cast<TrackingAction*>(const_cast<G4UserTrackingAction*>(
        G4RunManager::GetRunManager()->GetUserTrackingAction()));
  }
 
  void registerAction(UserAction* trackingAction) {
    trackingActions_.push_back(trackingAction);
  }
 
 private:
  std::vector<UserAction*> trackingActions_;
 
  TrackMap trackMap_;
};  // TrackingAction
 
}  // namespace simcore::g4user
 
#endif