SQCosmicGen.h

Go to the documentation of this file.

#ifndef SQCosmicGen_H__
#define SQCosmicGen_H__
 
#include "PHG4ParticleGeneratorBase.h"
 
#include <string>
 
#include <TRandom1.h>
 
class PHG4InEvent;
class PHCompositeNode;
 
class SQCosmicGen: public PHG4ParticleGeneratorBase 
{
public:
 
  SQCosmicGen(const std::string& name = "COSMICGEN");
  virtual ~SQCosmicGen() {}
 
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
 
  void set_mom_range(const double lo, const double hi);
  void set_theta_range(const double lo, const double hi);
  void set_charge_ratio(const double p, const double n);
 
  void set_acceptance_st(int stationID, bool b = true) { _req_st[stationID] = b; } 
 
private:
  double cosmicProb(double p, double theta);
 
  double uniformRand(const double lo, const double hi);
 
  void getZVtxLimits(int stationID, double ty, double y, double& min, double& max);
  void getXVtxLimits(int stationID, double tx, double z, double& min, double& max);
 
  bool generateVtx(double tx, double ty, double& x, double& y, double& z);
 
  inline bool acceptedInSt(int stationID, double x, double y, double z, double tx, double ty);
 
  PHG4InEvent* _ineve;
 
  double _prob_mup;
  double _prob_mum;
 
  double _mom_min;
  double _mom_max;
 
  double _theta_min;
  double _theta_max;
 
  bool   _req_st[5];
  double _z_st[5];
  double _size_x_st[5];
  double _size_y_st[5];
 
  double _p0;
  double _p1;
  double _p2;
 
  double _altitude;
 
  double _size_x;
  double _size_z;
 
  double _prob_max;
 
  TRandom1 _rndm;
};
 
#endif