FASTSIM/AliFastGlauber.h

   1 #ifndef ALIFASTGLAUBER_H
   2 #define ALIFASTGLAUBER_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 // Utility class to make simple Glauber type calculations for collision geometries:
   9 // Impact parameter, production points, reaction plane dependence
  10 // Author: Andreas Morsch
  11 // andreas.morsch@cern.ch
  12
  13 #include <TObject.h>
  14 #include <TString.h>
  15 class TF1;
  16 class TF2;
  17
  18
  19 class AliFastGlauber : public TObject {
  20  public:
  21     AliFastGlauber();
  22     AliFastGlauber(const AliFastGlauber& glauber);
  23     virtual ~AliFastGlauber();
  24     void Init(Int_t mode = 0);
  25
  26     void SetWoodSaxonParameters(Double_t r0, Double_t d, Double_t w, Double_t n)
  27         {fWSr0 = r0; fWSd = d; fWSw = w; fWSn = n;}
  28     void SetWoodSaxonParametersAu()
  29         {fWSr0 = 6.38; fWSd = 0.535; fWSw = 0.; fWSn = 8.59e-4;}
  30     void SetWoodSaxonParametersPb()
  31         {fWSr0 = 6.78; fWSd = 0.54; fWSw = 0.; fWSn = 7.14e-4;}
  32     void SetMaxImpact(Float_t bmax = 20.) {fgBMax = bmax;};
  33     void SetHardCrossSection(Float_t xs = 1.0) {fSigmaHard = xs;}
  34     void SetNNCrossSection  (Float_t xs = 55.6) {fSigmaNN = xs;}
  35     void SetNucleus(Int_t n=208) {fA=n;}
  36     void SetAuAuRhic();
  37     void SetPbPbLHC();
  38     void SetFileName(TString &fn){fName=fn;}
  39     void SetFileName(const char *fn="$(ALICE_ROOT)/FASTSIM/data/glauberPbPb.root"){fName=fn;}
  40
  41     const TF1* GetWSB()            const {return fgWSb;}
  42     const TF2* GetWSbz()           const {return fgWSbz;}
  43     const TF1* GetWSz()            const {return fgWSz;}
  44     const TF1* GetWSta()           const {return fgWSta;}
  45     const TF2* Kernel()            const {return fgWStarfi;}
  46     const TF2* GetWStarfi()        const {return fgWStarfi;}
  47     const TF2* GetWKParticipants() const {return fgWKParticipants;}
  48     const TF1* GetWParticipants()  const {return fgWParticipants;}
  49     const TF1* Overlap()           const {return fgWStaa;}
  50     const TF1* GetWStaa()          const {return fgWStaa;}
  51     const TF2* GetWAlmond()        const {return fgWAlmond;}
  52     const TF1* GetWPathLength0()   const {return fgWPathLength0;}
  53     const TF1* GetWPathLength()    const {return fgWPathLength;}
  54     const TF1* GetWIntRadius()     const {return fgWIntRadius;}
  55     const TF1* GetWSgeo()          const {return fgWSgeo;}
  56     const TF1* GetWSbinary()       const {return fgWSbinary;}
  57     const TF1* GetWSN()            const {return fgWSN;}
  58     const TF1* GetWEnergyDensity() const {return fgWEnergyDensity;}
  59     const TF2* GetWAlmondFixedB(Int_t i) const {return fgWAlmondFixedB[i];}
  60
  61     Float_t GetWr0() const {return fWSr0;}
  62     Float_t GetWSd() const {return fWSd;}
  63     Float_t GetWSw() const {return fWSw;}
  64     Float_t GetWSn() const {return fWSn;}
  65     Float_t GetSigmaHard()       const {return fSigmaHard;}
  66     Float_t GetSigmaNN()         const {return fSigmaNN;}
  67     Int_t GetA()                 const {return fA;}
  68     const TString* GetFileName() const {return &fName;}
  69     Float_t GetBmin() const {return fBmin;}
  70     Float_t GetBmax() const {return fBmax;}
  71
  72     void DrawWSb()          const;
  73     void DrawThickness()    const;
  74     void DrawOverlap()      const;
  75     void DrawParticipants() const;
  76     void DrawGeo()          const;
  77     void DrawBinary()       const;
  78     void DrawN()            const;
  79     void DrawKernel(Double_t b = 0.) const;
  80     void DrawAlmond(Double_t b = 0.) const;
  81     void DrawPathLength0(Double_t b = 0., Int_t iopt = 0)            const;
  82     void DrawPathLength(Double_t b, Int_t ni = 1000, Int_t iopt = 0) const;
  83     void DrawIntRadius(Double_t b = 0.) const;
  84     void DrawEnergyDensity()            const;
  85
  86     Double_t CrossSection(Double_t b1, Double_t b2)               const;
  87     Double_t HardCrossSection(Double_t b1, Double_t b2)           const;
  88     Double_t NHard(Double_t b1, Double_t b2)                      const;
  89     Double_t FractionOfHardCrossSection(Double_t b1, Double_t b2) const;
  90     Double_t Binaries(Double_t b)                 const;
  91     Double_t GetNumberOfBinaries(Double_t b)      const;
  92     Double_t Participants(Double_t b)             const;
  93     Double_t GetNumberOfParticipants(Double_t  b) const;
  94     Double_t GetNumberOfCollisions(Double_t  b)   const;
  95     Double_t GetNumberOfCollisionsPerEvent(Double_t  b) const;
  96     Double_t MeanOverlap(Double_t b1, Double_t b2);
  97     Double_t MeanNumberOfCollisionsPerEvent(Double_t b1, Double_t b2);
  98     void SimulateTrigger(Int_t n);
  99     void GetRandom(Float_t& b, Float_t& p, Float_t& mult);
 100     void GetRandom(Int_t& bin, Bool_t& hard);
 101     Double_t GetRandomImpactParameter(Double_t bmin, Double_t bmax);
 102
 103     void StoreFunctions() const;
 104     void StoreAlmonds()   const;
 105
 106     void SetLengthDefinition(Int_t def=1) {fEllDef=def;}
 107     Int_t GetLengthDef() const {return fEllDef;}
 108     void SetCentralityClass(Double_t xsecFrLow=0.0,Double_t xsecFrUp=0.1);
 109     void GetRandomBHard(Double_t& b);
 110     void GetRandomXY(Double_t& x,Double_t& y);
 111     void GetRandomPhi(Double_t& phi);
 112     Double_t CalculateLength(Double_t b=0.,Double_t x0=0.,Double_t y0=0.,
 113                              Double_t phi0=0.);
 114     void GetLengthAndPhi(Double_t& ell,Double_t &phi,Double_t b=-1.);
 115     void GetLength(Double_t& ell,Double_t b=-1.);
 116     void GetLengthsBackToBackAndPhi(Double_t& ell1,Double_t& ell2,
 117                                     Double_t &phi,
 118                                     Double_t b=-1.);
 119     void GetLengthsBackToBack(Double_t& ell1,Double_t& ell2,
 120                               Double_t b=-1.);
 121     void GetLengthsForPythia(Int_t n,Double_t* phi,Double_t* ell,
 122                              Double_t b=-1.);
 123     void PlotBDistr(Int_t n=1000);
 124     void PlotLengthDistr(Int_t n=1000,Bool_t save=kFALSE,
 125                          const char *fname="length.root");
 126     void PlotLengthB2BDistr(Int_t n=1000,Bool_t save=kFALSE,
 127                             const char *fname="lengthB2B.root");
 128     void CalculateI0I1(Double_t& integral0,Double_t& integral1,
 129                        Double_t b=0.,
 130                        Double_t x0=0.,Double_t y0=0.,Double_t phi0=0.,
 131                        Double_t ellCut=20.) const;
 132     void GetI0I1AndPhi(Double_t& integral0,Double_t& integral1,Double_t &phi,
 133                  Double_t ellCut=20.,Double_t b=-1.);
 134     void GetI0I1(Double_t& integral0,Double_t& integral1,
 135                  Double_t ellCut=20.,Double_t b=-1.);
 136     void GetI0I1BackToBackAndPhi(Double_t& integral01,Double_t& integral11,
 137                                  Double_t& integral02,Double_t& integral12,
 138                                  Double_t& phi,
 139                                  Double_t ellCut=20.,Double_t b=-1.);
 140     void GetI0I1BackToBackAndPhiAndXY(Double_t& integral01,Double_t& integral11,
 141                                       Double_t& integral02,Double_t& integral12,
 142                                       Double_t& phi,Double_t& x,Double_t&y,
 143                                       Double_t ellCut=20.,Double_t b=-1.);
 144     void GetI0I1BackToBack(Double_t& integral01,Double_t& integral11,
 145                            Double_t& integral02,Double_t& integral12,
 146                            Double_t ellCut=20.,Double_t b=-1.);
 147     void GetI0I1ForPythia(Int_t n,Double_t* phi,
 148                           Double_t* integral0,Double_t* integral1,
 149                           Double_t ellCut=20.,Double_t b=-1.);
 150     void GetI0I1ForPythiaAndXY(Int_t n,Double_t* phi,
 151                                Double_t* integral0,Double_t* integral1,
 152                                Double_t&x, Double_t &y,
 153                                Double_t ellCut=20.,Double_t b=-1.);
 154     void PlotI0I1Distr(Int_t n=1000,Double_t ellCut=20.,Bool_t save=kFALSE,
 155                        const char *fname="i0i1.root");
 156     void PlotI0I1B2BDistr(Int_t n=1000,Double_t ellCut=20.,Bool_t save=kFALSE,
 157                           const char *fname="i0i1B2B.root");
 158     void PlotAlmonds() const;
 159     // Copy
 160     AliFastGlauber& operator=(const AliFastGlauber & rhs);
 161     void Copy(TObject&) const;
 162  protected:
 163     static Double_t WSb            (Double_t *xx, Double_t *par);
 164     static Double_t WSbz           (Double_t *xx, Double_t *par);
 165     static Double_t WSz            (Double_t *xx, Double_t *par);
 166     static Double_t WSta           (Double_t *xx, Double_t *par);
 167     static Double_t WStarfi        (Double_t *xx, Double_t *par);
 168     static Double_t WStaa          (Double_t *xx, Double_t *par);
 169     static Double_t WKParticipants (Double_t *xx, Double_t *par);
 170     static Double_t WParticipants  (Double_t *xx, Double_t *par);
 171     static Double_t WSgeo          (Double_t *xx, Double_t *par);
 172     static Double_t WSbinary       (Double_t *xx, Double_t *par);
 173     static Double_t WSN            (Double_t *xx, Double_t *par);
 174     static Double_t WAlmond        (Double_t *xx, Double_t *par);
 175     static Double_t WPathLength0   (Double_t *xx, Double_t *par);
 176     static Double_t WPathLength    (Double_t *xx, Double_t *par);
 177     static Double_t WIntRadius     (Double_t *xx, Double_t *par);
 178     static Double_t WEnergyDensity (Double_t *xx, Double_t *par);
 179
 180     void Reset() const;
 181
 182     static Float_t fgBMax;           // Maximum Impact Parameter
 183     static Int_t fgCounter;          // Counter to protect double instantiation
 184     static const Int_t fgkMCInts;    // Number of MC integrations
 185
 186     static TF1*    fgWSb;            // Wood-Saxon Function (b)
 187     static TF2*    fgWSbz;           // Wood-Saxon Function (b, z)
 188     static TF1*    fgWSz;            // Wood-Saxon Function (b = b0, z)
 189     static TF1*    fgWSta;           // Thickness Function
 190     static TF2*    fgWStarfi;        // Kernel for Overlap Function
 191     static TF2*    fgWKParticipants; // Kernel for number of participants
 192     static TF1*    fgWParticipants;  // Number of participants
 193     static TF1*    fgWStaa;          // Overlap Function
 194     static TF2*    fgWAlmond;        // Interaction Almond
 195     static TF1*    fgWPathLength0;   // Path Length as a function of phi
 196     static TF1*    fgWPathLength;    // Path Length as a function of phi
 197     static TF1*    fgWIntRadius;     // Interaction Radius
 198     static TF1*    fgWSgeo;          // dSigma/db geometric
 199     static TF1*    fgWSbinary;       // dSigma/db binary
 200     static TF1*    fgWSN;            // dN/db binary
 201     static TF1*    fgWEnergyDensity; // Energy density as a function of impact parameter
 202     static TF2*    fgWAlmondFixedB[40]; // Interaction Almonds read from file
 203     static TF2*    fgWAlmondCurrent;    // Interaction Almond used for length
 204
 205     Float_t fWSr0;      // Wood-Saxon Parameter r0
 206     Float_t fWSd;       // Wood-Saxon Parameter d
 207     Float_t fWSw;       // Wood-Saxon Parameter w
 208     Float_t fWSn;       // Wood-Saxon Parameter n
 209                         // (chosen such that integral is one)
 210     Float_t fSigmaHard; // Hard Cross Section [mbarn]
 211     Float_t fSigmaNN;   // NN Cross Section [mbarn]
 212     Int_t fA;           // Nucleon number of nucleus A
 213
 214     Float_t fBmin;      // Minimum b (set through centrality selection)
 215     Float_t fBmax;      // Coresponding maximum b
 216
 217     Int_t fEllDef;      // definition of length (see CalculateLength())
 218     TString fName;     // filename of stored distributions
 219     ClassDef(AliFastGlauber,2) // Event geometry simulation in the Glauber Model
 220 };
 221
 222 #endif