PhotonuclearTracker.h

Go to the documentation of this file.

 
#ifndef SIMCORE_G4USER_PHOTONUCLEARTRACKER_H_
#define SIMCORE_G4USER_PHOTONUCLEARTRACKER_H_
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <map>
#include <memory>
#include <vector>
 
/*~~~~~~~~~~~~*/
/*   Geant4   */
/*~~~~~~~~~~~~*/
#include "G4Step.hh"
#include "G4Track.hh"
 
/*~~~~~~~~~~~~~*/
/*   SimCore   */
/*~~~~~~~~~~~~~*/
#include "SimCore/Event/PhotonuclearInteraction.h"
#include "SimCore/G4User/UserAction.h"
 
namespace simcore {
 
class PhotonuclearTracker : public UserAction {
 public:
  PhotonuclearTracker(const std::string& name,
                      framework::config::Parameters& parameters);
 
  virtual ~PhotonuclearTracker() = default;
 
  std::vector<TYPE> getTypes() override {
    return {TYPE::EVENT, TYPE::STEPPING, TYPE::TRACKING};
  }
 
  void BeginOfEventAction(const G4Event* event) override;
 
  void EndOfEventAction(const G4Event* event) override;
 
  void stepping(const G4Step* step) override;
 
  void PreUserTrackingAction(const G4Track* track) override;
 
  void PostUserTrackingAction(const G4Track* track) override;
 
  std::vector<ldmx::PhotonuclearInteraction> getInteractions() const {
    return pn_interactions_;
  }
 
  static PhotonuclearTracker* get() { return instance; }
 
 private:
  void extractTargetInfo(const G4Step* step, int& Z, int& A,
                         std::string& material);
 
  ldmx::PhotonuclearInteraction::ParticleInfo createParticleInfo(
      const G4Track* track);
 
 private:
  std::vector<ldmx::PhotonuclearInteraction> pn_interactions_;
 
  std::map<int, int> secondary_to_pn_index_;
 
  std::map<int, int> descendant_ancestry_;
 
  static PhotonuclearTracker* instance;
 
};  // PhotonuclearTracker
 
}  // namespace simcore
 
#endif  // SIMCORE_G4USER_PHOTONUCLEARTRACKER_H_