SQSingleMuonGen.h

Go to the documentation of this file.

#ifndef SQSingleMuonGen_H__
#define SQSingleMuonGen_H__
 
#include <g4main/PHG4ParticleGeneratorBase.h>
#include <string>
 
#include <TVector3.h>
#include <TRandom2.h>
#include <TF1.h>
 
#include "SQSingleMuonTruthInfo.h"
 
class PHG4InEvent;
class PHG4Particle;
class PHG4VtxPoint;
class PHCompositeNode;
 
class SQSingleMuonGen: public PHG4ParticleGeneratorBase 
{
public:
 
  SQSingleMuonGen(const std::string& name = "SINGLEGEN");
  virtual ~SQSingleMuonGen();
 
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
 
  const void set_target_only(bool en = true)   { _target_only = en; }
  const void enable_geom_cut(bool en = true)   { _enable_geom_cut = en; }
  const void set_mother_decay_length(double l) { _ctau = l; }
  const void set_charge_ratio(double v)        { _charge_ratio = v/(v+1.); }
  const void set_pt_max(double v)              { _pt_max = v; }
  const void set_mom_range(double min, double max)  { _mom_min = min; _mom_max = max; }
  const void enable_real_mom_dist(bool en = true)   { _enable_real_mom_dist = en; }
 
private:
  bool generatePrimaryVtx();
 
  void generateMotherParticle();
  void decayMotherParticle();
 
  bool geometryCut();
 
  PHG4InEvent* _ineve;
 
  SQSingleMuonTruthInfo* _truth;
 
  double _mom_min;
  double _mom_max;
  double _pt_max;
 
  bool _target_only;
 
  bool _enable_geom_cut;
 
  bool _enable_real_mom_dist;
 
  double _ctau;
 
  double _charge_ratio;
 
  TRandom2 _rndm;
 
  const double _m_muon;
  const double _m_pion;
 
  TF1* _pz_dist[6];
  TF1* _pt_dist;
};
 
#endif