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

  bool PassParticleFilter(HepMC::GenParticle * p);

 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;


#ifndef __CINT__
  gsl_rng *RandomGenerator;
#endif
};

#endif