PHHepMCGenHelper.h

Go to the documentation of this file.

// $Id: $
 
#ifndef SIMULATION_CORESOFTWARE_SIMULATION_G4SIMULATION_PHHEPMC_PHHEPMCGENHELPER_H_
#define SIMULATION_CORESOFTWARE_SIMULATION_G4SIMULATION_PHHEPMC_PHHEPMCGENHELPER_H_
 
 
 
class PHCompositeNode;
class PHHepMCGenEvent;
class PHHepMCGenEventMap;
class SQPrimaryVertexGen; //Abi
 
namespace HepMC
{
  class GenEvent;
};
 
#ifndef __CINT__
#include <gsl/gsl_rng.h>
#endif
 
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;
  }
 
  //
  int create_node_tree(PHCompositeNode *topNode);
 
  PHHepMCGenEvent *insert_event(HepMC::GenEvent *evt);
 
  void move_vertex(PHHepMCGenEvent *genevent);
 
  const PHHepMCGenEventMap *get_geneventmap() const
  {
    return _geneventmap;
  }
 
  PHHepMCGenEventMap *get_geneventmap()
  {
    return _geneventmap;
  }
 
#ifndef __CINT__
  gsl_rng * get_random_generator() {return RandomGenerator;}
#endif
 
  void set_geneventmap(PHHepMCGenEventMap *geneventmap)
  {
    _geneventmap = geneventmap;
  }
 
  void enableLegacyVtxGen() { _legacy_vertexgenerator = true; }
  double get_LegacyPARatio(){return _paratio;}
 
 protected:
#ifndef __CINT__
  gsl_rng *RandomGenerator;
#endif
 
  double smear(const double position, const double width, VTXFUNC dist) const;
 
 private:
  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;
 
  int _embedding_id;
 
  bool _reuse_vertex;
 
  int _reuse_vertex_embedding_id;
 
  PHHepMCGenEventMap *_geneventmap;
 
  SQPrimaryVertexGen* _vertexGen;
  bool _legacy_vertexgenerator;
  double _paratio;
};
 
#endif /* SIMULATION_CORESOFTWARE_SIMULATION_G4SIMULATION_PHHEPMC_PHHEPMCGENHELPER_H_ */