UserEventInformation.h

#ifndef SIMCORE_USEREVENTINFORMATION_H
#define SIMCORE_USEREVENTINFORMATION_H
 
#include <vector>
 
#include "G4VUserEventInformation.hh"
#include "SimCore/Event/HepMC3GenEvent.h"
 
namespace simcore {
 
class UserEventInformation : public G4VUserEventInformation {
 public:
  UserEventInformation() = default;
 
  virtual ~UserEventInformation() = default;
 
  void Print() const override;
 
  void incBremCandidateCount() { brem_candidate_count_ += 1; }
 
  void decBremCandidateCount() { brem_candidate_count_ -= 1; }
 
  void setDarkBremMaterialZ(double z_) { db_material_z_ = z_; }
 
  double getDarkBremMaterialZ() const { return db_material_z_; }
 
  void setAPrimeConversionMaterialZ(double z_) {
    aprime_conversion_material_z_ = z_;
  }
 
  double getAPrimeConversionMaterialZ() const {
    return aprime_conversion_material_z_;
  }
 
  void setWeight(double weight) { weight_ = weight; }
 
  double getWeight() { return weight_; }
 
  void incWeight(double step_weight) { weight_ *= step_weight; }
 
  int bremCandidateCount() { return brem_candidate_count_; }
 
  void addPNEnergy(double delta) { total_photonuclear_energy_ += delta; }
 
  void addENEnergy(double delta) { total_electronuclear_energy_ += delta; }
 
  double getPNEnergy() const { return total_photonuclear_energy_; }
 
  double getENEnergy() const { return total_electronuclear_energy_; }
 
  void lastStepWasPN(bool yes) { last_step_pn_ = yes; }
 
  bool wasLastStepPN() const { return last_step_pn_; }
 
  void lastStepWasEN(bool yes) { last_step_en_ = yes; }
 
  bool wasLastStepEN() const { return last_step_en_; }
 
  void setPNTargetZ(int z) { pn_target_z_ = z; }
 
  int getPNTargetZ() const { return pn_target_z_; }
 
  void setPNTargetA(int a) { pn_target_a_ = a; }
 
  int getPNTargetA() const { return pn_target_a_; }
 
  void setPNResampleCount(int count) { pn_resample_count_ = count; }
 
  int getPNResampleCount() const { return pn_resample_count_; }
 
  void addHepMC3GenEvent(ldmx::HepMC3GenEvent event) {
    hepmc3_events_.push_back(event);
  }
  std::vector<ldmx::HepMC3GenEvent> getHepMC3GenEvents() {
    return hepmc3_events_;
  }
 
 private:
  int brem_candidate_count_{0};
 
  double weight_{1.};
 
  double total_photonuclear_energy_{0.};
 
  bool last_step_pn_{false};
 
  double total_electronuclear_energy_{0.};
 
  bool last_step_en_{false};
 
  double db_material_z_{-1.};
 
  double aprime_conversion_material_z_{-1.};
 
  int pn_target_z_{-1};
 
  int pn_target_a_{-1};
 
  int pn_resample_count_{1};
 
  std::vector<ldmx::HepMC3GenEvent> hepmc3_events_;
};
}  // namespace simcore
 
#endif  // SIMCORE_USEREVENTINFORMATION_H