SQPileupGen.h

Go to the documentation of this file.

#ifndef __SQPileupGen_H__
#define __SQPileupGen_H__
 
#include <vector>
#include <TString.h>
#include <g4main/PHG4ParticleGeneratorBase.h>
 
class PHCompositeNode;
class PHG4ParticleGeneratorBase;
class PHG4InEvent;
class PHG4Particle;
 
class TFile;
class TTree;
class TClonesArray;
class TF1;
class TH1;
 
class SQMCEvent;
class SQPrimaryVertexGen;
class SQEvent;
 
class ExtParticle
{
 public:
  ExtParticle(int evtID, int pdg, const TVector3& pos, const TVector3& mom);
 
 public:
  int _evtID;
  int _pdg;
  TVector3 _pos;
  TVector3 _mom;
};
 
class SQPileupGen: public PHG4ParticleGeneratorBase
{
 public:
  SQPileupGen(const std::string& name = "PileupGen");
  virtual ~SQPileupGen();
 
  int Init(PHCompositeNode* topNode);
  int InitRun(PHCompositeNode* topNode);
  int process_event(PHCompositeNode* topNode);
 
  void setExtInputFile(const TString& name) { _extFileName = name; } 
  void setBucketSize(int n) { _bucketSize = n; } 
  void set_inhibit_threshold (int threshold) {_inhibit_threshold = threshold;} 
  void set_proton_coeff (double coeff) {_proton_coeff = coeff;} 
 
  TF1* get_beam_intensity_profile() const 
  {
    return _beam_intensity_profile;
  }
 
  TH1* get_beam_intensity_profile_histo() const 
  {
    return _beam_intensity_profile_histo;
  }
 
  void set_beam_intensity_profile(TF1* beamIntensityProfile) 
  {
    _beam_intensity_profile = beamIntensityProfile;
  }
 
  void set_beam_intensity_profile_histo(TH1* beamIntensityProfile_histo)
  {
    _beam_intensity_profile_histo = beamIntensityProfile_histo;
  }
 
 
 private:
  bool readExtTree(int nEvents);
 
  SQPrimaryVertexGen* _vertexGen;
 
  PHG4InEvent* _ineve;
  SQMCEvent*   _mcevt;
  SQEvent* _evt; //< An output node
 
 
  TF1* _beam_intensity_profile;
  TH1* _beam_intensity_profile_histo;
  std::vector<ExtParticle> _extParticles;
 
  int _bucketSize;
  int _n_proc_evt; //< N of processed events
 
  TString _extFileName;
 
  TFile* _extFile;
  TTree* _extTree;
 
  int _extEvtID;
  int _readIdx;
  int _nExtPar;
  int _inhibit_threshold;
  double _proton_coeff;
  int _extPDG[10000];
  TClonesArray* _extPos;
  TClonesArray* _extMom;
 
};
 
#endif