FairPrimaryGenerator.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file FairPrimaryGenerator.h

#ifndef FAIRPRIMARYGENERATOR_H
#define FAIRPRIMARYGENERATOR_H

#include "FairGenerator.h"
#include "FairLogger.h"

#include "TNamed.h"
#include "TObjArray.h"
#include "TVector3.h"

#include <iostream>

class TF1;
class FairGenericStack;
class FairMCEventHeader;
//class FairGenerator;

class FairPrimaryGenerator : public TNamed

{

  public:

    FairPrimaryGenerator();


    FairPrimaryGenerator(const char* name, const char* title="FAIR Generator");


    virtual ~FairPrimaryGenerator();

    virtual Bool_t Init();

    void AddGenerator(FairGenerator* generator) {
      if ( ! fGenList ) {
        std::cout << "Empty fGenList pointer ! " << std::endl;
        return;
      }
      fGenList->Add(generator);
    }


    virtual Bool_t GenerateEvent(FairGenericStack* pStack);


    virtual void AddTrack(Int_t pdgid, Double_t px, Double_t py, Double_t pz,
                          Double_t vx, Double_t vy, Double_t vz, Int_t parent=-1,Bool_t wanttracking=true,Double_t e=-9e9);


    void SetBeam(Double_t beamX0, Double_t beamY0,
                 Double_t beamSigmaX, Double_t beamSigmaY);


    void SetTarget(Double_t targetZ, Double_t targetDz);

    void SetMultTarget(Int_t nroftargets, Double_t* targetZ, Double_t targetDz);


    void SmearVertexZ(Bool_t flag)  { fSmearVertexZ  = flag; };
    void SmearGausVertexZ(Bool_t flag)  { fSmearGausVertexZ  = flag; };
    void SmearVertexXY(Bool_t flag) { fSmearVertexXY = flag; };

    TObjArray* GetListOfGenerators() { return fGenList;}


    void SetEvent(FairMCEventHeader* event) { fEvent = event; };


    FairMCEventHeader* GetEvent() { return fEvent; };

    void DoTracking(Bool_t doTracking= kTRUE) {fdoTracking=doTracking ;}

    void SetEventTimeInterval(Double_t min, Double_t max);

    void SetEventMeanTime(Double_t mean);

    void SetEventTime(TF1* timeProb);

    Int_t GetTotPrimary() { return fTotPrim;}




  protected:
    Double_t    fBeamX0;
    Double_t    fBeamY0;
    Double_t    fBeamSigmaX;
    Double_t    fBeamSigmaY;
    Double_t*   fTargetZ;       

    Int_t       fNrTargets;
    Double_t    fTargetDz;
    TVector3    fVertex;
    Int_t       fNTracks;
    Bool_t      fSmearVertexZ;
    Bool_t      fSmearGausVertexZ;
    Bool_t      fSmearVertexXY;
    FairGenericStack*   fStack; 

    TObjArray*  fGenList;
    TIterator*  fListIter;      

    FairMCEventHeader* fEvent;  

    Bool_t      fdoTracking;    

    Double_t    fEventTimeMin;  

    Double_t    fEventTimeMax;  

    Double_t    fEventTime;     

    Double_t    fEventMeanTime; 

    TF1*        fTimeProb;      

    Int_t       fMCIndexOffset; 

    FairLogger*            fLogger;

    static Int_t fTotPrim; 

    Int_t fEventNr;

    void MakeVertex();

  private:
    FairPrimaryGenerator(const FairPrimaryGenerator&);
    FairPrimaryGenerator& operator=(const  FairPrimaryGenerator&);

    ClassDef(FairPrimaryGenerator,4);

};

#endif