1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 /* $Id: AliGenTunedOnPbPb.cxx 51126 2013-08-19 13:37:49Z fnoferin $ */
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
24 #include <TClonesArray.h>
25 #include <TDatabasePDG.h>
29 #include <TParticle.h>
31 #include <TVirtualMC.h>
34 #include "AliDecayer.h"
35 #include "AliGenEventHeaderTunedPbPb.h"
38 #include "AliGenTunedOnPbPb.h"
40 ClassImp(AliGenTunedOnPbPb)
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};
46 Float_t AliGenTunedOnPbPb::fgV3Overv2 = 6.25000000000000000e-01;
47 Float_t AliGenTunedOnPbPb::fgEventplane=0;
49 TF1 *AliGenTunedOnPbPb::fgV2 = NULL;
51 //_____________________________________________________________________________
52 AliGenTunedOnPbPb::AliGenTunedOnPbPb()
56 fChangeWithCentrality(kFALSE),
60 // Default constructor
65 for(Int_t i=0;i < fgNspecies;i++){
66 fgHSpectrum[i] = NULL;
71 //_____________________________________________________________________________
72 AliGenTunedOnPbPb::~AliGenTunedOnPbPb()
75 // Standard destructor
79 //_____________________________________________________________________________
80 void AliGenTunedOnPbPb::Init()
83 // Initialise the generator
90 //_____________________________________________________________________________
91 void AliGenTunedOnPbPb::Generate()
94 // Generate one trigger
97 Float_t avCentr = (fCmin+fCmax)*0.5;
99 Float_t centrality = avCentr;
101 if(fChangeWithCentrality) centrality = fCmin + gRandom->Rndm()*(fCmax-fCmin);
103 SetParameters(centrality);
105 TMCProcess statusPdg[fgNspecies] = {kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary};
107 Float_t parV2scaling[3] = {1,0.202538,-0.00214468};
109 Float_t scaleV2 = 1.0;
111 TDatabasePDG *pdgD = TDatabasePDG::Instance();
113 if(fChangeWithCentrality){
114 parV2scaling[0] = 1. / (1 + parV2scaling[1]*avCentr + parV2scaling[2]*avCentr*avCentr); // normalize to average centrality
116 scaleV2 = parV2scaling[0]*(1 + parV2scaling[1]*centrality + parV2scaling[2]*centrality*centrality); // apply the trand of v2 w.r.t. centrality
119 fgV2->SetParameter(2,fgV3Overv2);
121 Float_t psi = gRandom->Rndm()*TMath::Pi();
123 Float_t psi3 = gRandom->Rndm()*TMath::Pi()*2/3;
124 fgV2->SetParameter(1,psi);
125 fgV2->SetParameter(3,psi3);
130 for (Int_t j=0;j<3;j++) origin0[j]=fOrigin[j];
133 if(fVertexSmear == kPerEvent) {
135 for (Int_t j=0; j < 3; j++) origin0[j] = fVertex[j];
139 printf("Generate event with centrality = %3.1f%c, |y|<%4.1f\n",centrality,'%',fYMaxValue);
141 for(Int_t isp=0;isp < fgNspecies;isp++){
142 if(! fgHSpectrum[isp]) continue;
144 printf("Total number of %i = %f\n",fgPdgInput[isp],fgMult[isp]*2*fYMaxValue);
146 for(Int_t ipart =0; ipart < fgMult[isp]*2*fYMaxValue; ipart++){
147 Int_t pdg = fgPdgInput[isp];
149 Double_t y = gRandom->Rndm()*2*fYMaxValue - fYMaxValue;
150 Double_t pt = fgHSpectrum[isp]->GetRandom();
151 if(fgHv2[isp]) fgV2->SetParameter(0,fgHv2[isp]->Interpolate(pt) * scaleV2);
152 else fgV2->SetParameter(0,0.);
153 Double_t mass = pdgD->GetParticle(pdg)->Mass();
154 Double_t mt2 = pt*pt + mass*mass;
155 Double_t phi = fgV2->GetRandom(-TMath::Pi(),TMath::Pi());
156 Double_t px = pt*TMath::Cos(phi);
157 Double_t py = pt*TMath::Sin(phi);
159 Double_t pz = y*TMath::Sqrt(mt2)/TMath::Sqrt(1-y*y);
160 // Double_t etot = TMath::Sqrt(mt2 + pz*pz);
161 Float_t p[3] = {px,py,pz};
162 Float_t polar[3] = {0.,0.,0.};
164 PushTrack(1, -1, pdg, p, origin0, polar, time, statusPdg[isp], npart, 1., 1);
172 eventVertex[0] = origin0[0];
173 eventVertex[1] = origin0[1];
174 eventVertex[2] = origin0[2];
177 AliGenEventHeaderTunedPbPb* header = new AliGenEventHeaderTunedPbPb("tunedOnPbPb");
179 header->SetPrimaryVertex(eventVertex);
180 header->SetInteractionTime(time);
181 header->SetNProduced(npart);
182 header->SetCentrality(centrality);
183 header->SetPsi2(psi);
184 header->SetPsi3(psi3);
185 gAlice->SetGenEventHeader(header);
188 void AliGenTunedOnPbPb::SetPtRange(Float_t ptmin, Float_t ptmax) {
189 AliGenerator::SetPtRange(ptmin, ptmax);
191 //_____________________________________________________________________________
192 TH1F *AliGenTunedOnPbPb::GetMultVsCentrality(Int_t species){
194 snprintf(title,100,"pdg = %i;centrality;dN/dy",fgPdgInput[species]);
195 TH1F *h = new TH1F("multVsCentr",title,100,0,100);
197 for(Int_t i=1;i<=100;i++){
200 h->SetBinContent(i,fgMult[species]);
205 //_____________________________________________________________________________
206 void AliGenTunedOnPbPb::SetParameters(Float_t centrality){
208 if(!fgV2) fgV2 = new TF1("fv2Par","(1 + 2*[0]*cos(2*(x-[1])) + 2*[0]*[2]*cos(3*(x-[3])))",-TMath::Pi(),TMath::Pi());
210 Float_t fr[9] = {0.,0.,0.,0.,0.,0.,0.,0.,0.};
212 if(centrality < 7.5){
213 fr[0] = (7.5 - centrality)/5.;
214 fr[1] = (centrality-2.5)/5.;
216 else if(centrality < 15){
217 fr[1] = (15-centrality)/7.5;
218 fr[2] = (centrality-7.5)/7.5;
220 else if(centrality < 25){
221 fr[2] = (25-centrality)/10.;
222 fr[3] = (centrality-15)/10.;
224 else if(centrality < 35){
225 fr[3] = (35-centrality)/10.;
226 fr[4] = (centrality-25)/10.;
228 else if(centrality < 45){
229 fr[4] = (45-centrality)/10.;
230 fr[5] = (centrality-35)/10.;
232 else if(centrality < 55){
233 fr[5] = (55-centrality)/10.;
234 fr[6] = (centrality-45)/10.;
236 else if(centrality < 65){
237 fr[6] = (65-centrality)/10.;
238 fr[7] = (centrality-55)/10.;
240 else if(centrality < 75){
241 fr[7] = (75-centrality)/10.;
242 fr[8] = (centrality-65)/10.;
248 // parameters as a function of centrality
249 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,
250 560.,560.,560.,92.,92.,28.,28.,92.,24.,24.,9.,2.7,2.7,0.45,0.45,
251 450.,450.,450.,70.,70.,21.,21.,70.,20.,20.,8.,2.4,2.4,0.4,0.4,
252 302.,302.,302.,47.,47.,14.5,14.5,47.,14.,14.,5.5,1.5,1.5,0.2,0.2,
253 200.,200.,200.,30.,30.,9.6,9.6,30.,9.,9.,3.5,0.9,0.9,0.08,0.08,
254 120.,120.,120.,18.,18.,6.1,6.1,18.,5.1,5.1,2.2,0.6,0.6,0.055,0.055,
255 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,
256 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,
257 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};
259 Float_t v3Overv2Cent[9] = {1.2,0.82,0.625,0.5,0.45,0.4,0.37,0.3,0.3};
261 // set parameters for current centrality
262 for(Int_t i=0;i < fgNspecies;i++){
265 for(Int_t j=0;j < 9;j++){
266 fgMult[i] += fr[j]*multCent[i+j*fgNspecies];
271 for(Int_t j=0;j < 9;j++)
272 fgV3Overv2 += fr[j]*v3Overv2Cent[j];
275 for(Int_t i=0;i < fgNspecies;i++)
276 fgMult[i] /= TMath::Log(centrality-77.);
280 /**************************************************************************
281 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
283 * Author: The ALICE Off-line Project. *
284 * Contributors are mentioned in the code where appropriate. *
286 * Permission to use, copy, modify and distribute this software and its *
287 * documentation strictly for non-commercial purposes is hereby granted *
288 * without fee, provided that the above copyright notice appears in all *
289 * copies and that both the copyright notice and this permission notice *
290 * appear in the supporting documentation. The authors make no claims *
291 * about the suitability of this software for any purpose. It is *
292 * provided "as is" without express or implied warranty. *
293 **************************************************************************/
295 /* $Id: AliGenTunedOnPbPb.cxx 51126 2013-08-19 13:37:49Z fnoferin $ */
297 // Parameterisation based on 5.5 ATeV PbPb data
298 // pi,K, p , K0, lambda, phi, Xi, Omega spectra, v2, v3 (no jets!)
299 // Author: fnoferin@cern.ch
303 #include <TClonesArray.h>
304 #include <TDatabasePDG.h>
307 #include <TPDGCode.h>
308 #include <TParticle.h>
310 #include <TVirtualMC.h>
312 #include "AliConst.h"
313 #include "AliDecayer.h"
314 #include "AliGenEventHeaderTunedPbPb.h"
317 #include "AliGenTunedOnPbPb.h"
319 ClassImp(AliGenTunedOnPbPb)
321 // set default parameters for 10-20% centrality
322 Int_t AliGenTunedOnPbPb::fgPdgInput[fgNspecies] = {211,-211,111,321,-321,2212,-2212,310,3122,-3122,333,3312,-3312,3334,-3334};
323 Float_t AliGenTunedOnPbPb::fgMult[fgNspecies] = {450,450,450,70,70,21,21,70,20,20,8,2.4,2.4,0.4,0.4};
325 Float_t AliGenTunedOnPbPb::fgV3Overv2 = 6.25000000000000000e-01;
326 Float_t AliGenTunedOnPbPb::fgEventplane=0;
328 TF1 *AliGenTunedOnPbPb::fgV2 = NULL;
330 //_____________________________________________________________________________
331 AliGenTunedOnPbPb::AliGenTunedOnPbPb()
335 fChangeWithCentrality(kFALSE),
339 // Default constructor
344 for(Int_t i=0;i < fgNspecies;i++){
345 fgHSpectrum[i] = NULL;
350 //_____________________________________________________________________________
351 AliGenTunedOnPbPb::~AliGenTunedOnPbPb()
354 // Standard destructor
358 //_____________________________________________________________________________
359 void AliGenTunedOnPbPb::Init()
362 // Initialise the generator
369 //_____________________________________________________________________________
370 void AliGenTunedOnPbPb::Generate()
373 // Generate one trigger
376 Float_t avCentr = (fCmin+fCmax)*0.5;
378 Float_t centrality = avCentr;
380 if(fChangeWithCentrality) centrality = fCmin + gRandom->Rndm()*(fCmax-fCmin);
382 SetParameters(centrality);
384 TMCProcess statusPdg[fgNspecies] = {kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary,kPPrimary};
386 Float_t parV2scaling[3] = {1,0.202538,-0.00214468};
388 Float_t scaleV2 = 1.0;
390 TDatabasePDG *pdgD = TDatabasePDG::Instance();
392 if(fChangeWithCentrality){
393 parV2scaling[0] = 1. / (1 + parV2scaling[1]*avCentr + parV2scaling[2]*avCentr*avCentr); // normalize to average centrality
395 scaleV2 = parV2scaling[0]*(1 + parV2scaling[1]*centrality + parV2scaling[2]*centrality*centrality); // apply the trand of v2 w.r.t. centrality
398 fgV2->SetParameter(2,fgV3Overv2);
400 Float_t psi = gRandom->Rndm()*TMath::Pi();
402 Float_t psi3 = gRandom->Rndm()*TMath::Pi()*2/3;
403 fgV2->SetParameter(1,psi);
404 fgV2->SetParameter(3,psi3);
409 for (Int_t j=0;j<3;j++) origin0[j]=fOrigin[j];
412 if(fVertexSmear == kPerEvent) {
414 for (Int_t j=0; j < 3; j++) origin0[j] = fVertex[j];
418 printf("Generate event with centrality = %3.1f%c, |y|<%4.1f\n",centrality,'%',fYMaxValue);
420 for(Int_t isp=0;isp < fgNspecies;isp++){
421 if(! fgHSpectrum[isp]) continue;
423 printf("Total number of %i = %f\n",fgPdgInput[isp],fgMult[isp]*2*fYMaxValue);
425 for(Int_t ipart =0; ipart < fgMult[isp]*2*fYMaxValue; ipart++){
426 Int_t pdg = fgPdgInput[isp];
428 Double_t y = gRandom->Rndm()*2*fYMaxValue - fYMaxValue;
429 Double_t pt = fgHSpectrum[isp]->GetRandom();
430 if(fgHv2[isp]) fgV2->SetParameter(0,fgHv2[isp]->Interpolate(pt) * scaleV2);
431 else fgV2->SetParameter(0,0.);
432 Double_t mass = pdgD->GetParticle(pdg)->Mass();
433 Double_t mt2 = pt*pt + mass*mass;
434 Double_t phi = fgV2->GetRandom(-TMath::Pi(),TMath::Pi());
435 Double_t px = pt*TMath::Cos(phi);
436 Double_t py = pt*TMath::Sin(phi);
438 Double_t pz = y*TMath::Sqrt(mt2)/TMath::Sqrt(1-y*y);
439 // Double_t etot = TMath::Sqrt(mt2 + pz*pz);
440 Float_t p[3] = {px,py,pz};
441 Float_t polar[3] = {0.,0.,0.};
443 PushTrack(1, -1, pdg, p, origin0, polar, time, statusPdg[isp], npart, 1., 1);
451 eventVertex[0] = origin0[0];
452 eventVertex[1] = origin0[1];
453 eventVertex[2] = origin0[2];
456 AliGenEventHeaderTunedPbPb* header = new AliGenEventHeaderTunedPbPb("tunedOnPbPb");
458 header->SetPrimaryVertex(eventVertex);
459 header->SetInteractionTime(time);
460 header->SetNProduced(npart);
461 header->SetCentrality(centrality);
462 header->SetPsi2(psi);
463 header->SetPsi3(psi3);
464 gAlice->SetGenEventHeader(header);
467 void AliGenTunedOnPbPb::SetPtRange(Float_t ptmin, Float_t ptmax) {
468 AliGenerator::SetPtRange(ptmin, ptmax);
470 //_____________________________________________________________________________
471 TH1F *AliGenTunedOnPbPb::GetMultVsCentrality(Int_t species){
473 snprintf(title,100,"pdg = %i;centrality;dN/dy",fgPdgInput[species]);
474 TH1F *h = new TH1F("multVsCentr",title,100,0,100);
476 for(Int_t i=1;i<=100;i++){
479 h->SetBinContent(i,fgMult[species]);
484 //_____________________________________________________________________________
485 void AliGenTunedOnPbPb::SetParameters(Float_t centrality){
487 if(!fgV2) fgV2 = new TF1("fv2Par","(1 + 2*[0]*cos(2*(x-[1])) + 2*[0]*[2]*cos(3*(x-[3])))",-TMath::Pi(),TMath::Pi());
489 Float_t fr[9] = {0.,0.,0.,0.,0.,0.,0.,0.,0.};
491 if(centrality < 7.5){
492 fr[0] = (7.5 - centrality)/5.;
493 fr[1] = (centrality-2.5)/5.;
495 else if(centrality < 15){
496 fr[1] = (15-centrality)/7.5;
497 fr[2] = (centrality-7.5)/7.5;
499 else if(centrality < 25){
500 fr[2] = (25-centrality)/10.;
501 fr[3] = (centrality-15)/10.;
503 else if(centrality < 35){
504 fr[3] = (35-centrality)/10.;
505 fr[4] = (centrality-25)/10.;
507 else if(centrality < 45){
508 fr[4] = (45-centrality)/10.;
509 fr[5] = (centrality-35)/10.;
511 else if(centrality < 55){
512 fr[5] = (55-centrality)/10.;
513 fr[6] = (centrality-45)/10.;
515 else if(centrality < 65){
516 fr[6] = (65-centrality)/10.;
517 fr[7] = (centrality-55)/10.;
519 else if(centrality < 75){
520 fr[7] = (75-centrality)/10.;
521 fr[8] = (centrality-65)/10.;
527 // parameters as a function of centrality
528 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,
529 560.,560.,560.,92.,92.,28.,28.,92.,24.,24.,9.,2.7,2.7,0.45,0.45,
530 450.,450.,450.,70.,70.,21.,21.,70.,20.,20.,8.,2.4,2.4,0.4,0.4,
531 302.,302.,302.,47.,47.,14.5,14.5,47.,14.,14.,5.5,1.5,1.5,0.2,0.2,
532 200.,200.,200.,30.,30.,9.6,9.6,30.,9.,9.,3.5,0.9,0.9,0.08,0.08,
533 120.,120.,120.,18.,18.,6.1,6.1,18.,5.1,5.1,2.2,0.6,0.6,0.055,0.055,
534 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,
535 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,
536 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};
538 Float_t v3Overv2Cent[9] = {1.2,0.82,0.625,0.5,0.45,0.4,0.37,0.3,0.3};
540 // set parameters for current centrality
541 for(Int_t i=0;i < fgNspecies;i++){
544 for(Int_t j=0;j < 9;j++){
545 fgMult[i] += fr[j]*multCent[i+j*fgNspecies];
550 for(Int_t j=0;j < 9;j++)
551 fgV3Overv2 += fr[j]*v3Overv2Cent[j];
554 for(Int_t i=0;i < fgNspecies;i++)
555 fgMult[i] /= TMath::Log(centrality-77.);