PHG4SimpleEventGenerator.h

Go to the documentation of this file.

#ifndef PHG4SimpleEventGenerator_H__
#define PHG4SimpleEventGenerator_H__
 
#include "PHG4ParticleGeneratorBase.h"
 
#include <map>
#include <vector>
 
#include <TGeoManager.h>
#include <phgeom/PHGeomUtility.h>
 
class PHG4InEvent;
class PHCompositeNode;
class SQEvent;
class SQMCEvent;
class SQPrimaryVertexGen;
 
class PHG4SimpleEventGenerator : public PHG4ParticleGeneratorBase {
 
public:
 
  enum FUNCTION {Uniform,Gaus};
 
  PHG4SimpleEventGenerator(const std::string &name="EVTGENERATOR");
  virtual ~PHG4SimpleEventGenerator(){}
 
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
  int End(PHCompositeNode *topNode);
 
  void add_particles(const std::string &name, const unsigned int count);
 
  void add_particles(const int pid, const unsigned int count);
 
  void set_t0(const double t0);
  
  void set_eta_range(const double eta_min, const double eta_max);
 
  void set_phi_range(const double phi_min, const double phi_max);
 
  void set_pt_range(const double pt_min, const double pt_max, const double pt_gaus_width = 0);
 
  void set_p_range(const double p_min, const double p_max, const double p_gaus_width = 0);
 
  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_pxpypz_range(
                const double x_min, const double x_max,
                        const double y_min, const double y_max,
                        const double z_min, const double z_max
                        );
 
   void enableLegacyVtxGen() { _legacy_vertexgenerator = true; } //Abi
 
private:
 
  double smearvtx(const double position, const double width, FUNCTION dist) const;
  // these need to be stored separately until run time when the names
  // can be translated using the GEANT4 lookup
  std::vector<std::pair<int, unsigned int> > _particle_codes; // <pdgcode, count>
  std::vector<std::pair<std::string, unsigned int> > _particle_names; // <names, count>  
  FUNCTION _vertex_func_x;
  FUNCTION _vertex_func_y;
  FUNCTION _vertex_func_z;
  double _t0;
  double _vertex_x;
  double _vertex_y;
  double _vertex_z;
  double _vertex_width_x;
  double _vertex_width_y;
  double _vertex_width_z;
  double _vertex_offset_x;
  double _vertex_offset_y;
  double _vertex_offset_z;
  FUNCTION _vertex_size_func_r;
  double _vertex_size_mean;
  double _vertex_size_width;
  double _eta_min;
  double _eta_max;
  double _phi_min;
  double _phi_max;
  double _pt_min;
  double _pt_max;
  double _pt_gaus_width;
  double _p_min;
  double _p_max; 
  double _p_gaus_width;
 
  double _px_min, _px_max;
  double _py_min, _py_max;
  double _pz_min, _pz_max;
  int    _eventcount;
  PHG4InEvent* _ineve;
  SQEvent* _evt; //< An output node
  SQMCEvent* _mcevt; //< An output node
 
  bool _legacy_vertexgenerator;
  SQPrimaryVertexGen* _vertexGen;
};
 
#endif