HepMCNodeReader.h

Go to the documentation of this file.

#ifndef HEPMCNODEREADER_H__
#define HEPMCNODEREADER_H__
 
#include <fun4all/SubsysReco.h>
 
// rootcint barfs with this header so we need to hide it
#ifndef __CINT__
#include <gsl/gsl_rng.h>
#endif
 
#include <string>
#include <vector>
 
class PHHepMCGenEvent;
class PHCompositeNode;
namespace HepMC {
  class GenParticle;
}
 
class HepMCNodeReader : public SubsysReco
{
 public:
  HepMCNodeReader(const std::string &name = "HEPMCREADER");
  virtual ~HepMCNodeReader();
 
  int Init(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
 
  void Embed(const int i = 1);
 
  void VertexPosition(const double v_x, const double v_y, const double v_z);
 
  void SmearVertex(const double s_x, const double s_y, const double s_z);
 
  void SetT0(const double t0) { vertex_t0 = t0; }
  //
  void SetSeed(const unsigned int i)
  {
    seed = i;
    use_seed = 1;
  }
 
  void set_particle_filter_on(const bool a) {_particle_filter_on = a;}
  void insert_particle_filter_pid(const int a) {_particle_filter_pid.push_back(a);}
 
  void enable_position_filter(const double x_min, const double x_max, const double y_min, const double y_max, const double z_min, const double z_max);
 
  bool PassParticleFilter(HepMC::GenParticle * p);
 
  void Set_bkg_mode(){_bkg_mode = true;}// abi
  void Set_pxy2pz_rat (const double rat){_pxy2pz_rat = rat;}
 
 private:
  double smeargauss(const double width);
  double smearflat(const double width);
  int use_seed;
  unsigned int seed;
  double vertex_pos_x;
  double vertex_pos_y;
  double vertex_pos_z;
  double vertex_t0;
  double width_vx;
  double width_vy;
  double width_vz;
 
  bool _particle_filter_on;
  std::vector<int> _particle_filter_pid;
  
  bool _bkg_mode; 
  double _pxy2pz_rat;//Abi
 
  bool _position_filter_on;
  double _pos_filter_x_min;
  double _pos_filter_x_max;
  double _pos_filter_y_min;
  double _pos_filter_y_max;
  double _pos_filter_z_min;
  double _pos_filter_z_max;
 
#ifndef __CINT__
  gsl_rng *RandomGenerator;
#endif
};
 
#endif