ZDC/AliGenZDC.h

   1 #ifndef ALIGENZDC_H
   2 #define ALIGENZDC_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ////////////////////////////////////////////////////
   9 //                                                //
  10 // Test pc generator for ZDC (taking into account //
  11 // Fermi smearing, beam divergence and crossing)  //
  12 //                                                //
  13 ////////////////////////////////////////////////////
  14
  15
  16 #include <TMath.h>
  17
  18 #include "AliGenerator.h"
  19
  20
  21 class AliGenZDC : public AliGenerator {
  22
  23 public:
  24   AliGenZDC();
  25   AliGenZDC(Int_t npart);
  26   virtual      ~AliGenZDC() {}
  27   void Init();
  28   void Generate();
  29
  30   // Fermi smearing, beam divergence and crossing angle
  31   void FermiTwoGaussian(Float_t A);
  32   void ExtractFermi(Int_t id, Double_t *ddp);
  33   void BeamDivCross(Int_t icross, Double_t *pLab);
  34   void AddAngle(Double_t theta1, Double_t phi1, Double_t theta2,
  35                 Double_t phi2, Double_t *angle);
  36
  37
  38   // Parameters that could be set for generation
  39   void SetParticle(Int_t ipart) {fIpart=ipart;};
  40   void SetMomentum(Float_t ptot) {fPMin=ptot; fPMax=ptot;};
  41   void SetDirection(Float_t zpsrp, Float_t cosx, Float_t cosy, Float_t cosz)
  42           {fPseudoRapidity=zpsrp; fCosx=cosx; fCosy=cosy; fCosz=cosz;};
  43   void SetFermi(Int_t Fflag) {fFermiflag=Fflag;};
  44   void SetDiv(Float_t bmdiv, Float_t bmcra, Int_t iflcr)
  45           {fBeamDiv=bmdiv; fBeamCrossAngle=bmcra; fBeamCrossPlane=iflcr;};
  46   void SetDebug(Int_t idebu) {fDebugOpt = idebu;};
  47
  48   // Getters
  49   Double_t GetFermi2p(Int_t key) const {return fProbintp[key];}
  50   Double_t GetFermi2n(Int_t key) const {return fProbintn[key];}
  51
  52 protected:
  53   Int_t    fIpart;              // Particle to be generated
  54   Float_t  fCosx;               // Director cos of the track - x direction
  55   Float_t  fCosy;               // Director cos of the track - y direction
  56   Float_t  fCosz;               // Director cos of the track - z direction
  57   Float_t  fPseudoRapidity;     // Pseudorapidity (!=0 -> eta of the particle)
  58                                 // (=0 -> director cos of the track)
  59   Int_t    fFermiflag;          // Fermi momentum flag (=1 -> Fermi smearing)
  60   Float_t  fBeamDiv;            // Beam divergence (angle in rad)
  61   Float_t  fBeamCrossAngle;     // Beam crossing angle (angle in rad)
  62   Int_t    fBeamCrossPlane;     // Beam crossing plane
  63                                 // (=1 -> horizontal, =2 -> vertical plane)
  64   Double_t fProbintp[201];      // Protons momentum distribution due to Fermi
  65   Double_t fProbintn[201];      // Neutrons momentum distribution due to Fermi
  66   Double_t fPp[201];            // Spectator momenta
  67   Int_t    fDebugOpt;           // Option for debugging [0->No debug, 1->Screen
  68                                 //  prints, 2->ASCII data file]
  69
  70  private:
  71   AliGenZDC(const AliGenZDC & gen);
  72   AliGenZDC & operator=(const AliGenZDC & gen);
  73
  74    ClassDef(AliGenZDC,1)        // Generator for AliZDC class
  75 };
  76
  77 #endif