EVGEN/AliGenTunedOnPbPb.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id: AliGenTunedOnPbPb.cxx 51126 2013-08-19 13:37:49Z fnoferin $ */
  17
  18 // Parameterisation based on 5.5 ATeV PbPb data
  19 // pi,K, p , K0, lambda, phi, Xi, Omega spectra, v2, v3 (no jets!)
  20 // Author: fnoferin@cern.ch
  21
  22 #include <TArrayF.h>
  23 #include <TCanvas.h>
  24 #include <TClonesArray.h>
  25 #include <TDatabasePDG.h>
  26 #include <TF1.h>
  27 #include <TH1.h>
  28 #include <TPDGCode.h>
  29 #include <TParticle.h>
  30 #include <TROOT.h>
  31 #include <TVirtualMC.h>
  32
  33 #include "AliConst.h"
  34 #include "AliDecayer.h"
  35 #include "AliGenEventHeaderTunedPbPb.h"
  36 #include "AliLog.h"
  37 #include "AliRun.h"
  38 #include "AliGenTunedOnPbPb.h"
  39
  40 ClassImp(AliGenTunedOnPbPb)
  41
  42 // set default parameters for 10-20% centrality
  43 Int_t AliGenTunedOnPbPb::fgPdgInput[fgNspecies] = {211,-211,111,321,-321,2212,-2212,310,3122,-3122,333,3312,-3312,3334,-3334};
  44 Float_t AliGenTunedOnPbPb::fgMult[fgNspecies] = {450,450,450,70,70,21,21,70,20,20,8,2.4,2.4,0.4,0.4};
  45
  46 Float_t AliGenTunedOnPbPb::fgV3Overv2 = 6.25000000000000000e-01;
  47 Float_t AliGenTunedOnPbPb::fgEventplane=0;
  48
  49 TF1 *AliGenTunedOnPbPb::fgV2 = NULL;
  50
  51 //_____________________________________________________________________________
  52 AliGenTunedOnPbPb::AliGenTunedOnPbPb()
  53   :AliGenerator(),
  54    fCmin(0.),
  55    fCmax(100.),
  56    fChangeWithCentrality(kFALSE),
  57    fYMaxValue(2.0)
  58 {
  59     //
  60     // Default constructor
  61     //
  62     SetCutVertexZ();
  63     SetPtRange();
  64
  65     for(Int_t i=0;i < fgNspecies;i++){
  66       fgHSpectrum[i] = NULL;
  67       fgHv2[i] = NULL;
  68     }
  69 }
  70
  71 //_____________________________________________________________________________
  72 AliGenTunedOnPbPb::~AliGenTunedOnPbPb()
  73 {
  74   //
  75   // Standard destructor
  76   //
  77 }
  78
  79 //_____________________________________________________________________________
  80 void AliGenTunedOnPbPb::Init()
  81 {
  82   //
  83   // Initialise the generator
  84   //
  85
  86   // define histos
  87 }
  88
  89
  90 //_____________________________________________________________________________
  91 void AliGenTunedOnPbPb::Generate()
  92 {
  93   //
  94   // Generate one trigger
  95   //
  96
  97   Float_t avCentr = (fCmin+fCmax)*0.5;
  98
  99   Float_t centrality = avCentr;
 100
 101   if(fChangeWithCentrality) centrality = fCmin + gRandom->Rndm()*(fCmax-fCmin);
 102
 103   SetParameters(centrality);
 104
 105   if(!fChangeWithCentrality){
 106     Float_t in=0;
 107     for(Int_t i=0;i < fgNspecies;i++){
 108       in=0;
 109       if(fgHSpectrum[i]){
 110         for(Int_t j=1;j<=fgHSpectrum[i]->GetNbinsX();j++){
 111           in += fgHSpectrum[i]->GetBinContent(j)*fgHSpectrum[i]->GetBinWidth(j);
 112         }
 113       }
 114
 115       // replace n-particles with the one in input file if centralidy dependece was disable
 116       fgMult[i] = in;
 117     }
 118   }
 119
 120
 121   TMCProcess statusPdg[fgNspecies] = {kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary};
 122
 123   Float_t parV2scaling[3] = {1,0.202538,-0.00214468};
 124
 125   Float_t scaleV2 = 1.0;
 126
 127   TDatabasePDG *pdgD = TDatabasePDG::Instance();
 128
 129   if(fChangeWithCentrality){
 130     parV2scaling[0] = 1. / (1 + parV2scaling[1]*avCentr + parV2scaling[2]*avCentr*avCentr); // normalize to average centrality
 131
 132     scaleV2 = parV2scaling[0]*(1 + parV2scaling[1]*centrality + parV2scaling[2]*centrality*centrality); // apply the trand of v2 w.r.t. centrality
 133   }
 134
 135   fgV2->SetParameter(2,fgV3Overv2);
 136
 137   Float_t psi = gRandom->Rndm()*TMath::Pi();
 138   fgEventplane = psi;
 139   Float_t psi3 = gRandom->Rndm()*TMath::Pi()*2/3;
 140   fgV2->SetParameter(1,psi);
 141   fgV2->SetParameter(3,psi3);
 142
 143   Int_t npart = 0;
 144
 145   Float_t origin0[3];
 146   for (Int_t j=0;j<3;j++) origin0[j]=fOrigin[j];
 147   Float_t time;
 148   time = fTimeOrigin;
 149   if(fVertexSmear == kPerEvent) {
 150     Vertex();
 151     for (Int_t j=0; j < 3; j++) origin0[j] = fVertex[j];
 152     time = fTime;
 153   } // if kPerEvent
 154
 155   printf("Generate event with centrality = %3.1f%c, |y|<%4.1f\n",centrality,'%',fYMaxValue);
 156
 157   for(Int_t isp=0;isp < fgNspecies;isp++){
 158     if(! fgHSpectrum[isp]) continue;
 159
 160     Int_t npartSp = Int_t(fgMult[isp]*2*fYMaxValue + gRandom->Rndm());
 161
 162     printf("Total number of %i = %i\n",fgPdgInput[isp],npartSp);
 163
 164     for(Int_t ipart =0; ipart < npartSp; ipart++){
 165       Int_t pdg = fgPdgInput[isp];
 166
 167       Double_t y = gRandom->Rndm()*2*fYMaxValue - fYMaxValue;
 168       Double_t pt = fgHSpectrum[isp]->GetRandom();
 169       if(fgHv2[isp]) fgV2->SetParameter(0,fgHv2[isp]->Interpolate(pt) * scaleV2);
 170       else fgV2->SetParameter(0,0.);
 171       Double_t mass = pdgD->GetParticle(pdg)->Mass();
 172       Double_t mt2 = pt*pt + mass*mass;
 173       Double_t phi = fgV2->GetRandom(-TMath::Pi(),TMath::Pi());
 174       Double_t px = pt*TMath::Cos(phi);
 175       Double_t py = pt*TMath::Sin(phi);
 176       y = TMath::TanH(y);
 177       Double_t pz = y*TMath::Sqrt(mt2)/TMath::Sqrt(1-y*y);
 178 //       Double_t etot = TMath::Sqrt(mt2 + pz*pz);
 179       Float_t p[3] = {px,py,pz};
 180       Float_t polar[3] = {0.,0.,0.};
 181
 182       PushTrack(1, -1, pdg, p, origin0, polar, time, statusPdg[isp], npart, 1., 1);
 183       KeepTrack(npart);
 184       npart++;
 185     }
 186   }
 187
 188   TArrayF eventVertex;
 189   eventVertex.Set(3);
 190   eventVertex[0] = origin0[0];
 191   eventVertex[1] = origin0[1];
 192   eventVertex[2] = origin0[2];
 193
 194 // Header
 195     AliGenEventHeaderTunedPbPb* header = new AliGenEventHeaderTunedPbPb("tunedOnPbPb");
 196 // Event Vertex
 197     header->SetPrimaryVertex(eventVertex);
 198     header->SetInteractionTime(time);
 199     header->SetNProduced(npart);
 200     header->SetCentrality(centrality);
 201     header->SetPsi2(psi);
 202     header->SetPsi3(psi3);
 203     gAlice->SetGenEventHeader(header);
 204 }
 205
 206 void AliGenTunedOnPbPb::SetPtRange(Float_t ptmin, Float_t ptmax) {
 207     AliGenerator::SetPtRange(ptmin, ptmax);
 208 }
 209 //_____________________________________________________________________________
 210 TH1F *AliGenTunedOnPbPb::GetMultVsCentrality(Int_t species){
 211   char title[100];
 212   snprintf(title,100,"pdg = %i;centrality;dN/dy",fgPdgInput[species]);
 213   TH1F *h = new TH1F("multVsCentr",title,100,0,100);
 214
 215   for(Int_t i=1;i<=100;i++){
 216     Float_t x = i+0.5;
 217     SetParameters(x);
 218     h->SetBinContent(i,fgMult[species]);
 219   }
 220
 221   return h;
 222 }
 223 //_____________________________________________________________________________
 224 void AliGenTunedOnPbPb::SetParameters(Float_t centrality){
 225
 226   if(!fgV2) fgV2 = new TF1("fv2Par","TMath::Max(0.,(1 + 2*[0]*cos(2*(x-[1])) + 2*[0]*[2]*cos(3*(x-[3]))))",-TMath::Pi(),TMath::Pi());
 227
 228   Float_t fr[9] = {0.,0.,0.,0.,0.,0.,0.,0.,0.};
 229
 230   if(centrality < 7.5){
 231     fr[0] = (7.5 - centrality)/5.;
 232     fr[1] = (centrality-2.5)/5.;
 233   }
 234   else if(centrality < 15){
 235     fr[1] = (15-centrality)/7.5;
 236     fr[2] = (centrality-7.5)/7.5;
 237   }
 238   else if(centrality < 25){
 239     fr[2] = (25-centrality)/10.;
 240     fr[3] = (centrality-15)/10.;
 241   }
 242   else if(centrality < 35){
 243     fr[3] = (35-centrality)/10.;
 244     fr[4] = (centrality-25)/10.;
 245   }
 246   else if(centrality < 45){
 247     fr[4] = (45-centrality)/10.;
 248     fr[5] = (centrality-35)/10.;
 249   }
 250   else if(centrality < 55){
 251     fr[5] = (55-centrality)/10.;
 252     fr[6] = (centrality-45)/10.;
 253   }
 254   else if(centrality < 65){
 255     fr[6] = (65-centrality)/10.;
 256     fr[7] = (centrality-55)/10.;
 257   }
 258   else if(centrality < 75){
 259     fr[7] = (75-centrality)/10.;
 260     fr[8] = (centrality-65)/10.;
 261   }
 262   else{
 263     fr[8] = 1.0;
 264   }
 265
 266   // parameters as a function of centrality
 267   Float_t multCent[9*fgNspecies] = {680.,680.,680.,110.,110.,34.,34.,110.,28.,28.,11.5,3.1,3.1,0.5,0.5,
 268                                     560.,560.,560.,92.,92.,28.,28.,92.,24.,24.,9.,2.7,2.7,0.45,0.45,
 269                                     450.,450.,450.,70.,70.,21.,21.,70.,20.,20.,8.,2.4,2.4,0.4,0.4,
 270                                     302.,302.,302.,47.,47.,14.5,14.5,47.,14.,14.,5.5,1.5,1.5,0.2,0.2,
 271                                     200.,200.,200.,30.,30.,9.6,9.6,30.,9.,9.,3.5,0.9,0.9,0.08,0.08,
 272                                     120.,120.,120.,18.,18.,6.1,6.1,18.,5.1,5.1,2.2,0.6,0.6,0.055,0.055,
 273                                     70.,70.,70.,10.4,10.4,3.6,3.6,10.4,2.6,2.6,1.4,0.36,0.36,0.035,0.035,
 274                                     36.,36.,36.,5.25,5.25,2.,2.,5.25,1.5,1.5,0.5,0.02,0.02,0.015,0.015,
 275                                     17.,17.,17.,2.3,2.3,0.9,0.9,2.3,0.6,0.6,0.16,0.006,0.006,0.005,0.005};
 276
 277   Float_t v3Overv2Cent[9] = {1.2,0.82,0.625,0.5,0.45,0.4,0.37,0.3,0.3};
 278
 279   fgV3Overv2 = 0;
 280   for(Int_t j=0;j < 9;j++)
 281       fgV3Overv2 += fr[j]*v3Overv2Cent[j];
 282
 283   // set parameters for current centrality
 284   for(Int_t i=0;i < fgNspecies;i++){
 285     fgMult[i] = 0;
 286
 287     for(Int_t j=0;j < 9;j++){
 288       fgMult[i] += fr[j]*multCent[i+j*fgNspecies];
 289     }
 290   }
 291
 292   if(centrality > 80){
 293     for(Int_t i=0;i < fgNspecies;i++)
 294       fgMult[i] /= TMath::Log(centrality-77.);
 295   }
 296 }