PHHepMCGenHelper.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHHepMCGenHelper.h

// $Id: $

#ifndef PHHEPMC_PHHEPMCGENHELPER_H
#define PHHEPMC_PHHEPMCGENHELPER_H

#include <CLHEP/Vector/ThreeVector.h>

#include <gsl/gsl_rng.h>

#include <cmath>
#include <string>
#include <utility>
#include <vector>

class PHCompositeNode;
class PHHepMCGenEvent;
class PHHepMCGenEventMap;

namespace HepMC
{
  class GenEvent;
}

class PHHepMCGenHelper
{
 public:
  PHHepMCGenHelper();
  virtual ~PHHepMCGenHelper();

  enum VTXFUNC
  {
    Uniform,
    Gaus
  };

  void set_vertex_distribution_function(VTXFUNC x, VTXFUNC y, VTXFUNC z, VTXFUNC t);

  void set_vertex_distribution_mean(const double x, const double y, const double z, const double t);

  void set_vertex_distribution_width(const double x, const double y, const double z, const double t);

  int get_embedding_id() const { return _embedding_id; }
  //
  void set_embedding_id(int id) { _embedding_id = id; }
  //
  void set_reuse_vertex(int src_embedding_id)
  {
    _reuse_vertex = true;
    _reuse_vertex_embedding_id = src_embedding_id;
  }

  //
  virtual int create_node_tree(PHCompositeNode *topNode);

  const PHHepMCGenEvent *get_PHHepMCGenEvent_template() const;

  PHHepMCGenEvent *insert_event(HepMC::GenEvent *evt);

  const PHHepMCGenEventMap *get_geneventmap() const
  {
    return _geneventmap;
  }

  PHHepMCGenEventMap *get_geneventmap()
  {
    return _geneventmap;
  }

  gsl_rng *get_random_generator()
  {
    return RandomGenerator;
  }

  void set_geneventmap(PHHepMCGenEventMap *geneventmap)
  {
    _geneventmap = geneventmap;
  }

  void set_beam_direction_theta_phi(
      const double beamA_theta,
      const double beamA_phi,
      const double beamB_theta,
      const double beamB_phi)
  {
    m_beam_direction_theta_phi = {{beamA_theta, beamA_phi}, {beamB_theta, beamB_phi}};
  }

  void set_beam_angular_divergence_hv(
      const double beamA_divergence_h,
      const double beamA_divergence_v,
      const double beamB_divergence_h,
      const double beamB_divergence_v)
  {
    m_beam_angular_divergence_hv = {{beamA_divergence_h, beamA_divergence_v}, {beamB_divergence_h, beamB_divergence_v}};
  }

  void set_beam_angular_z_coefficient_hv(
      const double beamA_h,
      const double beamA_v,
      const double beamB_h,
      const double beamB_v)
  {
    m_beam_angular_z_coefficient_hv = {{beamA_h, beamA_v}, {beamB_h, beamB_v}};
  }

  void use_beam_bunch_sim(bool b) { m_use_beam_bunch_sim = b; }

  void set_beam_bunch_width(const std::vector<double> &beamA, const std::vector<double> &beamB);

  void CopySettings(PHHepMCGenHelper &helper);

  void CopySettings(PHHepMCGenHelper *helper_dest);

  void CopyHelperSettings(PHHepMCGenHelper *helper_src);

  void Print(const std::string &what = "ALL") const;

  void PHHepMCGenHelper_Verbosity(int v) { m_verbosity = v; }

  int PHHepMCGenHelper_Verbosity() { return m_verbosity; }

 protected:
  void HepMC2Lab_boost_rotation_translation(PHHepMCGenEvent *genevent);

  void move_vertex(PHHepMCGenEvent *genevent);

  std::pair<double, double> generate_vertx_with_bunch_interaction(PHHepMCGenEvent *genevent);

 private:
  gsl_rng *RandomGenerator;

  double smear(const double position, const double width, VTXFUNC dist) const;

  static CLHEP::Hep3Vector pair2Hep3Vector(const std::pair<double, double> &theta_phi);

  VTXFUNC _vertex_func_x;
  VTXFUNC _vertex_func_y;
  VTXFUNC _vertex_func_z;
  VTXFUNC _vertex_func_t;

  double _vertex_x;
  double _vertex_y;
  double _vertex_z;
  double _vertex_t;

  double _vertex_width_x;
  double _vertex_width_y;
  double _vertex_width_z;
  double _vertex_width_t;

  std::pair<std::pair<double, double>, std::pair<double, double>> m_beam_direction_theta_phi = {
      {0, 0},    //+z beam
      {M_PI, 0}  //-z beam
  };

  std::pair<std::pair<double, double>, std::pair<double, double>> m_beam_angular_divergence_hv = {
      {0, 0},  //+z beam
      {0, 0}   //-z beam
  };

  std::pair<std::pair<double, double>, std::pair<double, double>> m_beam_angular_z_coefficient_hv = {
      {0, 0},  //+z beam
      {0, 0}   //-z beam
  };

  int _embedding_id;

  bool _reuse_vertex;

  int _reuse_vertex_embedding_id;

  PHHepMCGenEventMap *_geneventmap;

  int m_verbosity = 0;

  bool m_use_beam_bunch_sim = false;

  std::pair<std::vector<double>, std::vector<double>> m_beam_bunch_width = {
      {0, 0, 0},  //+z beam
      {0, 0, 0}   //-z beam
  };

  double get_collision_width(unsigned int hv_index);

};

#endif /* PHHEPMC_PHHEPMCGENHELPER_H */