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;

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;
  }

 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;
};

#endif /* SIMULATION_CORESOFTWARE_SIMULATION_G4SIMULATION_PHHEPMC_PHHEPMCGENHELPER_H_ */