PndFixStepParticleGun.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                PndFixStepParticleGun header file                    -----
// -----          Created 30/10/08  by Tobias Stockmanns                      -----
// -------------------------------------------------------------------------

#ifndef Pnd_FLATPARTICLEGENERATOR_H
#define Pnd_FLATPARTICLEGENERATOR_H

#include "FairGenerator.h"

#include <iostream>

class FairPrimaryGenerator;

class PndRangeValues
{
public :
        PndRangeValues(Double32_t start = 0, Double32_t stop = 0, Double32_t step = 1): fStart(start), fStop(stop), fStep(step), fActualValue(start)
        {}
  virtual ~PndRangeValues(){};
        Double32_t fStart;
        Double32_t fStop;
        Double32_t fStep;
        Double32_t fActualValue;

        void SetStart(Double32_t val){fStart = val; fActualValue = val;};
        void SetStop(Double32_t val){fStop = val;};
        void SetStep(Double32_t val){fStep = val;};
        void SetActualValue(Double32_t val){fActualValue = val;};

        int GetNSteps(){
            if (fStep != 0)
                return ((fStop - fStart) / fStep) + 1;
            else
                return 1;
        }

        virtual Double32_t GetStep(){return fStep;}

        ClassDef(PndRangeValues,1);
};


class PndFixStepParticleGun : public FairGenerator
{
public:

  PndFixStepParticleGun();

  PndFixStepParticleGun(Int_t pdgid, Int_t mult=1);

  virtual ~PndFixStepParticleGun() {};

  void SetPDGType      (Int_t pdg)  {fPDGType = pdg;  };

  void SetMultiplicity (Int_t mult) {fMult    = mult; };

  void SetPRange(Double32_t pmin=0   , Double32_t pmax=10, Double32_t pstep = 1)
    {fP.SetStart(pmin); fP.SetStop(pmax); fP.SetStep(pstep); fPRangeIsSet=kTRUE;}

  void SetPtRange      (Double32_t ptmin=0   , Double32_t ptmax=10, Double32_t ptstep=1)
    {fPt.SetStart(ptmin); fPt.SetStop(ptmax); fPt.SetStep(ptstep); fPtRangeIsSet=kTRUE;};

  void SetPhiRange     (Double32_t phimin=0  , Double32_t phimax=360, Double32_t phistep = 1)
    {fPhi.SetStart(phimin); fPhi.SetStop(phimax); fPhi.SetStep(phistep);};

  void SetEtaRange     (Double32_t etamin=-5 , Double32_t etamax=7, Double32_t etastep=1)
    {fEta.SetStart(etamin); fEta.SetStop(etamax); fEta.SetStep(etastep); fEtaRangeIsSet=kTRUE;};

  void SetYRange       (Double32_t ymin=-5, Double32_t ymax=7, Double32_t ystep=1)
    {fRapidity.SetStart(ymin); fRapidity.SetStop(ymax); fRapidity.SetStep(ystep); fYRangeIsSet=kTRUE;};

  void SetThetaRange   (Double32_t thetamin=0, Double32_t thetamax=90, Double32_t thetastep=1)
    {fTheta.SetStart(thetamin); fTheta.SetStop(thetamax); fTheta.SetStep(thetastep); fThetaRangeIsSet=kTRUE;};

  void SetCosThetaRange (Double32_t costhetamin=-1, Double32_t costhetamax=1, Double32_t costhetastep=0.1)
    {fCosTheta.SetStart(costhetamin); fCosTheta.SetStop(costhetamax); fCosTheta.SetStep(costhetastep); fCosThetaIsSet=kTRUE; fThetaRangeIsSet=kTRUE;};

  void SetXYZ   (Double32_t x=0, Double32_t y=0, Double32_t z=0) {
    fX=x; fY=y; fZ=z; fPointVtxIsSet=kTRUE;}

  void SetBoxXYZ (Double32_t x1=0, Double32_t y1=0, Double32_t x2=0, Double32_t y2=0, Double32_t z=0) {
    fX1=x1; fY1=y1; fX2=x2; fY2=y2; fZ=z; fBoxVtxIsSet=kTRUE;}

  void SetDebug(Bool_t debug=0) {fDebug = debug;}

  int GetNEvents();

  Bool_t Init();

  virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);

private:
  Int_t      fPDGType;             // Particle type (PDG encoding)
  Int_t      fMult;                // Multiplicity
  Int_t          fEvent;                           // EventCounter

  Double32_t fPDGMass;             // Particle mass [GeV]
//  Double32_t fPtMin,  fPtMax, fPtStep;      // Transverse momentum range [GeV]
  PndRangeValues fPt;
// Double32_t fPhiMin, fPhiMax, fPhiStep;     // Azimuth angle range [degree]
  PndRangeValues fPhi;
//  Double32_t fEtaMin, fEtaMax,        fEtaStep;     // Pseudorapidity range in lab system
  PndRangeValues fEta;
//  Double32_t fYMin,   fYMax,  fYStep;       // Rapidity range in lab system
  PndRangeValues fRapidity;
//  Double32_t fPMin,   fPMax,  fPStep;       // Momentum range in lab system
  PndRangeValues fP;
//  Double32_t fThetaMin, fThetaMax, fThetaStep; // Polar angle range in lab system [degree]
  PndRangeValues fTheta;
  PndRangeValues fCosTheta;
  Double32_t fX, fY, fZ;           // Point vertex coordinates [cm]
  Double32_t fX1, fY1, fX2, fY2;   // Box vertex coords (x1,y1)->(x2,y2)

  Bool_t     fEtaRangeIsSet;       // True if eta range is set
  Bool_t     fYRangeIsSet;         // True if rapidity range is set
  Bool_t     fThetaRangeIsSet;     // True if theta range is set
  Bool_t     fCosThetaIsSet;       // True if uniform distribution in
                                   //cos(theta) is set (default -> not set)
  Bool_t     fPtRangeIsSet;        // True if transverse momentum range is set
  Bool_t     fPRangeIsSet;         // True if abs.momentum range is set
  Bool_t     fPointVtxIsSet;       // True if point vertex is set
  Bool_t     fBoxVtxIsSet;         // True if box vertex is set
  Bool_t     fDebug;               // Debug switch
  Bool_t         fDoit;                                 // Stop generating particles at the end of the range
  Bool_t     fFirstRun;            // Used to indicate the first run. Otherwise the first event would be skipped
  void CalcActValues(PndRangeValues* val1, PndRangeValues* val2, PndRangeValues* val3);
  bool IsEndOfRanges(PndRangeValues* val1, PndRangeValues* val2, PndRangeValues* val3);

  ClassDef(PndFixStepParticleGun,1);

};


#endif