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 saveFlag_; }
 
  void setSaveFlag(bool saveFlag) { saveFlag_ = saveFlag; }
 
  bool isBremCandidate() const { return isBremCandidate_; }
 
  void tagBremCandidate(bool isBremCandidate = true) {
    isBremCandidate_ = isBremCandidate;
  }
 
  bool isRecoilElectron() const { return isRecoilElectron_; }
 
  void tagRecoilElectron(bool isRecoilElectron = true) {
    isRecoilElectron_ = isRecoilElectron;
  }
 
  bool isPNGamma() const { return isPNGamma_; }
 
  void tagPNGamma(bool isPNGamma = true) { isPNGamma_ = isPNGamma; }
 
  const G4ThreeVector& getInitialMomentum() const { return initialMomentum_; }
 
  std::string getVertexVolume() const { return vertexVolume_; }
 
  double getVertexTime() const { return vertex_time_; }
 
 private:
  bool saveFlag_{false};
 
  bool isBremCandidate_{false};
 
  bool isRecoilElectron_{false};
 
  bool isPNGamma_{false};
 
  std::string vertexVolume_{""};
 
  double vertex_time_{0.};
 
  G4ThreeVector initialMomentum_;
};
}  // namespace simcore
 
#endif