PHG4ParticleGeneratorVectorMeson.h

Go to the documentation of this file.

#ifndef PHG4ParticleGeneratorVectorMeson_H__
#define PHG4ParticleGeneratorVectorMeson_H__
 
#include "PHG4ParticleGeneratorBase.h"
#include <map>
 
class PHG4InEvent;
class TRandom;
class TF1;
 
class PHG4ParticleGeneratorVectorMeson: public PHG4ParticleGeneratorBase
{
 public:
      enum FUNCTION {Uniform,Gaus};
 
  explicit PHG4ParticleGeneratorVectorMeson(const std::string &name="PGUN");
  virtual ~PHG4ParticleGeneratorVectorMeson() {}
 
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
 
  void add_decay_particles(const std::string &name1, const std::string &name2, const unsigned int decay_id);
 
  void set_decay_vertex_offset(double dx, double dy, double dz, const unsigned int decay_id);
  void set_eta_range(const double eta_min, const double eta_max);
  void set_rapidity_range(const double y_min, const double y_max);
  void set_mom_range(const double mom_min, const double mom_max);
  void set_pt_range(const double pt_min, const double pt_max);
  void set_vertex_distribution_function(FUNCTION x, FUNCTION y, FUNCTION z);
 
  void set_vertex_distribution_mean(const double x, const double y, const double z);
 
  void set_vertex_distribution_width(const double x, const double y, const double z);
 
  void set_existing_vertex_offset_vector(const double x, const double y, const double z);
 
  void set_vertex_size_function(FUNCTION r);
 
  void set_vertex_size_parameters(const double mean, const double width);
 
  void set_read_vtx_from_hepmc(bool read_vtx) { read_vtx_from_hepmc = read_vtx;}
 
  void set_mass(const double mass);
  void set_width(const double width);
  void set_decay_types(const std::string &decay1, const std::string &decay2);
  void set_histrand_init(const int initflag) {_histrand_init = initflag;}
 
 private:
 
  double smearvtx(const double position, const double width, FUNCTION dist) const;
  std::map<unsigned int, int> decay1_codes; // <pdgcode, count>
  std::map<unsigned int, std::string> decay1_names; // <names, count>
  std::map<unsigned int, int> decay2_codes; // <pdgcode, count>
  std::map<unsigned int, std::string> decay2_names; // <names, count>
  std::map<unsigned int, double> decay_vtx_offset_x;
  std::map<unsigned int, double> decay_vtx_offset_y;
  std::map<unsigned int, double> decay_vtx_offset_z;
 
  protected:
 
  FUNCTION _vertex_func_x;
  FUNCTION _vertex_func_y;
  FUNCTION _vertex_func_z;
  double _t0;
  double _vertex_x;// primary vertex (or mean) x component, cf. vtx_x = track-by-track vertex x component
  double _vertex_y;// primary vertex (or mean) y component, cf. vtx_y = track-by-track vertex y component
  double _vertex_z;// primary vertex (or mean)z component, cf. vtx_z = track-by-track vertex z component
  double _vertex_width_x; // sigma x if not use existing vtx
  double _vertex_width_y; // sigma y if not use existing vtx
  double _vertex_width_z; // sigma z if not use existing vtx
  double _vertex_offset_x; // track-by-track decay vertex offset if use existing vtx
  double _vertex_offset_y; // track-by-track decay vertex offset if use existing vtx
  double _vertex_offset_z; // track-by-track decay vertex offset if use existing vtx
  FUNCTION _vertex_size_func_r;
  double _vertex_size_mean;
  double _vertex_size_width;
  bool   read_vtx_from_hepmc;
 
  double y_min;
  double y_max;
  double eta_min;
  double eta_max;
  double mom_min;
  double mom_max;
  double pt_min;
  double pt_max;
  double mass;
  double width;
  double m1;
  double m2;
  int _histrand_init;
  std::string decay1;
  std::string decay2;
 
  TF1 *fsin;
  TF1 *frap;
  TF1 *fpt;
  TRandom *trand;
 
  PHG4InEvent* ineve;
 
};
 
#endif