StackingAction.h

 
#ifndef SIMCORE_G4USER_STACKINGACTION_H
#define SIMCORE_G4USER_STACKINGACTION_H
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <vector>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4UserStackingAction.hh"
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/G4User/UserAction.h"
 
namespace simcore {
namespace g4user {
 
class StackingAction : public G4UserStackingAction {
 public:
  StackingAction() = default;
 
  virtual ~StackingAction() = default;
 
  G4ClassificationOfNewTrack ClassifyNewTrack(const G4Track* track) override;
 
  void NewStage() override;
 
  void PrepareNewEvent() override;
 
  void registerAction(std::shared_ptr<UserAction> stackingAction) {
    stacking_actions_.push_back(stackingAction);
  }
 
 private:
  std::vector<std::shared_ptr<UserAction>> stacking_actions_;
 
};  // StackingAction
 
}  // namespace g4user
}  // namespace simcore
 
#endif  // SIMCORE_G4USER_STACKINGACTION_H