UserTrackInformation.h

#ifndef SIMCORE_USERTRACKINFORMATION_H
#define SIMCORE_USERTRACKINFORMATION_H
 
#include "G4ThreeVector.hh"
#include "G4Track.hh"
#include "G4VUserTrackInformation.hh"
 
namespace simcore {
 
class UserTrackInformation : public G4VUserTrackInformation {
 public:
  UserTrackInformation() = default;
 
  virtual ~UserTrackInformation() = default;
  static UserTrackInformation* get(const G4Track* track);
 
  void initialize(const G4Track* track);
 
  void Print() const override;
 
  bool getSaveFlag() const { return save_flag_; }
 
  void setSaveFlag(bool saveFlag) { save_flag_ = saveFlag; }
 
  bool isBremCandidate() const { return is_brem_candidate_; }
 
  void tagBremCandidate(bool isBremCandidate = true) {
    is_brem_candidate_ = isBremCandidate;
  }
 
  bool isRecoilElectron() const { return is_recoil_electron_; }
 
  void tagRecoilElectron(bool isRecoilElectron = true) {
    is_recoil_electron_ = isRecoilElectron;
  }
 
  bool isPNGamma() const { return is_pn_gamma_; }
 
  void tagPNGamma(bool isPNGamma = true) { is_pn_gamma_ = isPNGamma; }
 
  const G4ThreeVector& getInitialMomentum() const { return initial_momentum_; }
 
  std::string getVertexVolume() const { return vertex_volume_; }
 
  double getVertexTime() const { return vertex_time_; }
 
 private:
  bool save_flag_{false};
 
  bool is_brem_candidate_{false};
 
  bool is_recoil_electron_{false};
 
  bool is_pn_gamma_{false};
 
  std::string vertex_volume_{""};
 
  double vertex_time_{0.};
 
  G4ThreeVector initial_momentum_;
};
}  // namespace simcore
 
#endif