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 **************************************************************************/
18 //-----------------------------------------------------------------
19 // Generate the final state of the interaction as the input
22 //-----------------------------------------------------------------
25 <img src="picts/AliGeneratorClass.gif">
28 <font size=+2 color=red>
29 <p>The responsible person for this module is
30 <a href="mailto:andreas.morsch@cern.ch">Andreas Morsch</a>.
36 ///////////////////////////////////////////////////////////////////
37 #include <TGenerator.h>
38 #include <TMCProcess.h>
41 #include "AliCollisionGeometry.h"
42 #include "AliConfig.h"
43 #include "AliGenerator.h"
47 #include "AliVertexGenerator.h"
49 ClassImp(AliGenerator)
52 //_______________________________________________________________________
53 AliGenerator::AliGenerator():
73 fVertexSmear(kNoSmear),
74 fVertexSource(kInternal),
76 fPileUpTimeWindow(-1.),
87 fCollisionGeometry(0),
97 // Default constructor
100 AliDebug(1, "AliGenerator Default Constructor");
101 AliMC * mc = gAlice->GetMCApp();
102 if (mc) mc->SetGenerator(this);
105 SetThetaRange(); ResetBit(kThetaRange);
106 SetPhiRange(); ResetBit(kPhiRange);
107 SetMomentumRange(); ResetBit(kMomentumRange);
108 SetPtRange(); ResetBit(kPtRange);
109 SetYRange(); ResetBit(kYRange);
110 SetNumberParticles();
115 fOrigin[0]=fOrigin[1]=fOrigin[2]=0;
116 fOsigma[0]=fOsigma[1]=fOsigma[2]=0;
117 fVertex[0]=fVertex[1]=fVertex[2]=0;
119 fVMin[0]=fVMin[1]=fVMin[2]=0;
120 fVMax[0]=fVMax[1]=fVMax[2]=10000;
123 //_______________________________________________________________________
124 AliGenerator::AliGenerator(Int_t npart):
144 fVertexSmear(kNoSmear),
145 fVertexSource(kInternal),
147 fPileUpTimeWindow(-1.),
158 fCollisionGeometry(0),
168 // Standard constructor
171 AliDebug(1, "AliGenerator Constructor initializing number of particles");
172 AliMC * mc = gAlice->GetMCApp();
173 if (mc) mc->SetGenerator(this);
176 SetThetaRange(); ResetBit(kThetaRange);
177 SetPhiRange(); ResetBit(kPhiRange);
178 SetMomentumRange(); ResetBit(kMomentumRange);
179 SetPtRange(); ResetBit(kPtRange);
180 SetYRange(); ResetBit(kYRange);
184 fOrigin[0]=fOrigin[1]=fOrigin[2]=0;
185 fOsigma[0]=fOsigma[1]=fOsigma[2]=0;
186 fVertex[0]=fVertex[1]=fVertex[2]=0;
188 fVMin[0]=fVMin[1]=fVMin[2]=0;
189 fVMax[0]=fVMax[1]=fVMax[2]=10000;
191 SetNumberParticles(npart);
193 AliConfig::Instance()->Add(this);
196 //_______________________________________________________________________
197 AliGenerator::~AliGenerator()
210 //_______________________________________________________________________
211 void AliGenerator::Init()
214 // Dummy initialisation
218 //_______________________________________________________________________
219 void AliGenerator::GenerateN(Int_t ntimes)
222 // Run generator n times
223 // Used by AliGenCocktail for adding extra signals in proportion to
224 // multiplicity/centrality
225 // By default generates one AliGenHeader for each call to Generate()
226 // Can be overloaded (e.g. AliGenBox and AliGenParam) to consolidate headers
228 for (Int_t i=0; i < ntimes; i++) {
232 //_______________________________________________________________________
233 void AliGenerator::SetOrigin(Float_t ox, Float_t oy, Float_t oz)
236 // Set the vertex for the generated tracks
243 //_______________________________________________________________________
244 void AliGenerator::SetOrigin(const TLorentzVector &o)
247 // Set the vertex for the generated tracks
254 //_______________________________________________________________________
255 void AliGenerator::SetSigma(Float_t sx, Float_t sy, Float_t sz)
258 // Set the spread of the vertex
265 //_______________________________________________________________________
266 void AliGenerator::SetMomentumRange(Float_t pmin, Float_t pmax)
269 // Set the momentum range for the generated particles
273 SetBit(kMomentumRange);
276 //_______________________________________________________________________
277 void AliGenerator::SetPtRange(Float_t ptmin, Float_t ptmax)
280 // Set the Pt range for the generated particles
287 //_______________________________________________________________________
288 void AliGenerator::SetPhiRange(Float_t phimin, Float_t phimax)
291 // Set the Phi range for the generated particles
293 fPhiMin = TMath::Pi()*phimin/180;
294 fPhiMax = TMath::Pi()*phimax/180;
298 //_______________________________________________________________________
299 void AliGenerator::SetYRange(Float_t ymin, Float_t ymax)
302 // Set the Rapidity range for the generated particles
309 //_______________________________________________________________________
310 void AliGenerator::SetVRange(Float_t vxmin, Float_t vxmax,
311 Float_t vymin, Float_t vymax,
312 Float_t vzmin, Float_t vzmax)
315 // Set the vertex range for the generated particles
317 fVMin[0]=vxmin; fVMin[1]=vymin; fVMin[2]=vzmin;
318 fVMax[0]=vxmax; fVMax[1]=vymax; fVMax[2]=vzmax;
319 SetBit(kVertexRange);
322 //_______________________________________________________________________
323 void AliGenerator::SetThetaRange(Float_t thetamin, Float_t thetamax)
326 // Set the theta range for the generated particles
328 fThetaMin = TMath::Pi()*thetamin/180;
329 fThetaMax = TMath::Pi()*thetamax/180;
333 void AliGenerator::Vertex()
336 // Obtain vertex for current event from external source or calculated (internal)
338 if (fVertexSource == kInternal) {
340 } else if (fVertexSource == kContainer) {
342 } else if (fVertexSource == kExternal) {
347 //_______________________________________________________________________
348 void AliGenerator::VertexExternal()
351 // Obtain vertex from external source (vertex generator)
353 TVector3 vertex = fVertexGenerator->GetVertex();
354 fVertex[0] = vertex.X();
355 fVertex[1] = vertex.Y();
356 fVertex[2] = vertex.Z();
360 //_______________________________________________________________________
361 void AliGenerator::VertexInternal()
364 // Obtain calculated vertex
365 // Default is gaussian smearing
370 if (!TestBit(kVertexRange)) {
371 while(TMath::Abs(dv[2]) > fCutVertexZ*fOsigma[2]) {
373 for (j=0; j < 3; j++) {
374 dv[j] = fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
375 TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
378 for (j=0; j < 3; j++) fVertex[j] = fOrigin[j] + dv[j];
380 // In case of gaussian smearing we smear also the event/collision time
382 fTime = fTimeOrigin + fOsigma[2]/TMath::Ccgs()*
383 TMath::Cos(2*random[0]*TMath::Pi())*
384 TMath::Sqrt(-2*TMath::Log(random[1]));
388 for (j=0; j < 3; j++)
389 fVertex[j] = fVMin[j] + random[j] * (fVMax[j] - fVMin[j]);
394 //_______________________________________________________________________
395 void AliGenerator::PushTrack(Int_t done, Int_t parent, Int_t pdg,
396 Float_t *pmom, Float_t *vpos, Float_t *polar,
397 Float_t tof, TMCProcess mech, Int_t &ntr,
398 Float_t weight, Int_t is)
401 // Loads one track on the stack
405 fStack->PushTrack(done, parent, pdg, pmom, vpos, polar, tof,
406 mech, ntr, weight, is);
408 gAlice->GetMCApp()->PushTrack(done, parent, pdg, pmom, vpos, polar, tof,
409 mech, ntr, weight, is);
412 //_______________________________________________________________________
413 void AliGenerator::PushTrack(Int_t done, Int_t parent, Int_t pdg,
414 Double_t px, Double_t py, Double_t pz, Double_t e,
415 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
416 Double_t polx, Double_t poly, Double_t polz,
417 TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is)
420 // Loads one track on the stack
424 fStack->PushTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof,
425 polx, poly, polz, mech, ntr, weight, is);
427 gAlice->GetMCApp()->PushTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof,
428 polx, poly, polz, mech, ntr, weight, is);
432 //_______________________________________________________________________
433 void AliGenerator:: KeepTrack(Int_t itrack)
436 // Declare a track permanent on the stack
439 fStack->KeepTrack(itrack);
441 gAlice->GetMCApp()->KeepTrack(itrack);
445 //_______________________________________________________________________
446 void AliGenerator:: SetHighWaterMark(Int_t nt)
449 // Internal function to set the maximum index used in the stack
452 fStack->SetHighWaterMark(nt);
454 gAlice->GetMCApp()->SetHighWaterMark(nt);
457 void AliGenerator::FinishRun()