PhotonuclearInteraction.h

#ifndef SIMCORE_EVENT_PHOTONUCLEARINTERACTION_H
#define SIMCORE_EVENT_PHOTONUCLEARINTERACTION_H
 
/*~~~~~~~~~~*/
/*   ROOT   */
/*~~~~~~~~~~*/
 
/*~~~~~~~~~~~~~~~~*/
/*   C++ StdLib   */
/*~~~~~~~~~~~~~~~~*/
#include <Rtypes.h>
 
#include <map>
#include <string>
#include <vector>
 
namespace ldmx {
 
class PhotonuclearInteraction {
 public:
  struct ParticleInfo {
    int track_id_{0};
 
    int pdg_id_{0};
 
    double energy_{0.0};
 
    double px_{0.0};
    double py_{0.0};
    double pz_{0.0};
 
    double x_{0.0};
    double y_{0.0};
    double z_{0.0};
 
    double time_{0.0};
 
    ParticleInfo() = default;
 
    virtual ~ParticleInfo() = default;
 
    ClassDef(ParticleInfo, 1);
  };
 
  PhotonuclearInteraction() = default;
 
  virtual ~PhotonuclearInteraction() = default;
 
  void clear();
 
  void setIncidentPhoton(int track_id, int pdg_id, double energy, double px,
                         double py, double pz, double x, double y, double z,
                         double time);
 
  ParticleInfo getIncidentPhoton() const { return incident_photon_; }
 
  void setTarget(int Z, int A, const std::string& material);
 
  int getTargetZ() const { return target_z_; }
 
  int getTargetA() const { return target_a_; }
 
  std::string getTargetMaterial() const { return target_material_; }
 
  void addImmediateSecondary(const ParticleInfo& particle);
 
  std::vector<ParticleInfo> getImmediateSecondaries() const {
    return immediate_secondaries_;
  }
 
  int getNumImmediateSecondaries() const {
    return immediate_secondaries_.size();
  }
 
  void addDescendant(int immediate_secondary_id, int final_state_track_id);
 
  std::map<int, std::vector<int>> getDescendantMap() const {
    return descendant_map_;
  }
 
  std::vector<int> getDescendants(int immediate_secondary_id) const;
 
  void setInteractionVolume(const std::string& volume) {
    interaction_volume_ = volume;
  }
 
  std::string getInteractionVolume() const { return interaction_volume_; }
 
  void setProcessName(const std::string& process) { process_name_ = process; }
 
  std::string getProcessName() const { return process_name_; }
 
  void print() const;
 
  friend std::ostream& operator<<(std::ostream& o,
                                  const PhotonuclearInteraction& pn);
 
 private:
  ParticleInfo incident_photon_;
 
  int target_z_{0};
 
  int target_a_{0};
 
  std::string target_material_{""};
 
  std::vector<ParticleInfo> immediate_secondaries_;
 
  std::map<int, std::vector<int>> descendant_map_;
 
  std::string interaction_volume_{""};
 
  std::string process_name_{""};
 
  ClassDef(PhotonuclearInteraction, 1);
 
};  // PhotonuclearInteraction
 
}  // namespace ldmx
 
#endif  // SIMCORE_EVENT_PHOTONUCLEARINTERACTION_H