/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
* *
* Author: The ALICE Off-line Project. *
* Contributors are mentioned in the code where appropriate. *
* *
* Permission to use, copy, modify and distribute this software and its *
* documentation strictly for non-commercial purposes is hereby granted *
* without fee, provided that the above copyright notice appears in all *
* copies and that both the copyright notice and this permission notice *
* appear in the supporting documentation. The authors make no claims *
* about the suitability of this software for any purpose. It is *
* provided "as is" without express or implied warranty. *
**************************************************************************/
/*
$Log$
Revision 1.20 2002/10/29 14:26:49 hristov
Code clean-up (F.Carminati)
Revision 1.19 2002/03/12 11:06:03 morsch
Add particle status code to argument list of SetTrack(..).
Revision 1.18 2002/02/18 19:23:00 hristov
Put protection in the destructor
Revision 1.17 2001/11/26 15:46:35 morsch
Set default value for vertex truncation in constructor.
Revision 1.16 2001/10/16 08:45:37 morsch
All vertex truncation and event vertex moved here.
Revision 1.15 2001/07/28 10:43:41 hristov
Default argument initialised once
Revision 1.14 2001/07/27 17:13:12 morsch
- SetTrack, KeepTrack and SetHighWaterMark methods added to delegate to local
stack or stack owned by AliRun. (Piotr Skowronski, A.M.)
- Changes needed for afterburners (Piotr Skowronski)
Revision 1.13 2001/05/16 14:57:22 alibrary
New files for folders and Stack
Revision 1.12 2001/02/02 11:12:50 morsch
Add Vertex() method that allows to get vertex from merging manager, if needed.
Revision 1.11 2001/01/26 19:58:48 hristov
Major upgrade of AliRoot code
Revision 1.10 2000/12/21 15:30:18 fca
Correcting coding convention violations
Revision 1.9 2000/10/04 10:08:01 fca
Correction of minor typing mistakes
Revision 1.8 2000/10/02 21:28:14 fca
Removal of useless dependecies via forward declarations
Revision 1.7 2000/07/12 08:56:25 fca
Coding convention correction and warning removal
Revision 1.6 2000/07/11 18:24:59 fca
Coding convention corrections + few minor bug fixes
Revision 1.5 2000/06/08 13:34:50 fca
Better control of momentum range in GenBox
Revision 1.4 1999/09/29 09:24:29 fca
Introduction of the Copyright and cvs Log
*/
///////////////////////////////////////////////////////////////////
// //
// Generate the final state of the interaction as the input //
// to the MonteCarlo //
//
//Begin_Html
/*
The responsible person for this module is
Andreas Morsch.
*/ //End_Html // // /////////////////////////////////////////////////////////////////// #include "TGenerator.h" #include "AliConfig.h" #include "AliGenerator.h" #include "AliRun.h" #include "AliStack.h" ClassImp(AliGenerator) TGenerator* AliGenerator::fgMCEvGen=0; //_______________________________________________________________________ AliGenerator::AliGenerator(): fThetaMin(0), fThetaMax(0), fPhiMin(0), fPhiMax(0), fPMin(0), fPMax(0), fPtMin(0), fPtMax(0), fYMin(0), fYMax(0), fVMin(3), fVMax(3), fNpart(0), fParentWeight(0), fChildWeight(0), fAnalog(0), fVertexSmear(kNoSmear), fVertexSource(kExternal), fCutVertexZ(0), fTrackIt(0), fOrigin(3), fOsigma(3), fVertex(3), fEventVertex(0), fStack(0) { // // Default constructor // if (gAlice) { if (gAlice->GetDebug()>0) printf("\n AliGenerator Default Constructor\n\n"); gAlice->SetGenerator(this); } SetThetaRange(); ResetBit(kThetaRange); SetPhiRange(); ResetBit(kPhiRange); SetMomentumRange(); ResetBit(kMomentumRange); SetPtRange(); ResetBit(kPtRange); SetYRange(); ResetBit(kYRange); SetNumberParticles(); SetTrackingFlag(); SetCutVertexZ(); fOrigin[0]=fOrigin[1]=fOrigin[2]=0; fOsigma[0]=fOsigma[1]=fOsigma[2]=0; fVertex[0]=fVertex[1]=fVertex[2]=0; fVMin[0]=fVMin[1]=fVMin[2]=0; fVMax[0]=fVMax[1]=fVMax[2]=10000; } //_______________________________________________________________________ AliGenerator::AliGenerator(Int_t npart): fThetaMin(0), fThetaMax(0), fPhiMin(0), fPhiMax(0), fPMin(0), fPMax(0), fPtMin(0), fPtMax(0), fYMin(0), fYMax(0), fVMin(3), fVMax(3), fNpart(0), fParentWeight(0), fChildWeight(0), fAnalog(0), fVertexSmear(kNoSmear), fVertexSource(kExternal), fCutVertexZ(0), fTrackIt(0), fOrigin(3), fOsigma(3), fVertex(3), fEventVertex(0), fStack(0) { // // Standard constructor // if (gAlice) { if (gAlice->GetDebug()>0) printf("\n AliGenerator Constructor initializing number of particles \n\n"); gAlice->SetGenerator(this); } SetThetaRange(); ResetBit(kThetaRange); SetPhiRange(); ResetBit(kPhiRange); SetMomentumRange(); ResetBit(kMomentumRange); SetPtRange(); ResetBit(kPtRange); SetYRange(); ResetBit(kYRange); SetTrackingFlag(); SetCutVertexZ(); fOrigin[0]=fOrigin[1]=fOrigin[2]=0; fOsigma[0]=fOsigma[1]=fOsigma[2]=0; fVertex[0]=fVertex[1]=fVertex[2]=0; fVMin[0]=fVMin[1]=fVMin[2]=0; fVMax[0]=fVMax[1]=fVMax[2]=10000; SetNumberParticles(npart); AliConfig::Instance()->Add(this); } //_______________________________________________________________________ AliGenerator::AliGenerator(const AliGenerator &gen): TNamed(gen), AliRndm(gen), fThetaMin(0), fThetaMax(0), fPhiMin(0), fPhiMax(0), fPMin(0), fPMax(0), fPtMin(0), fPtMax(0), fYMin(0), fYMax(0), fVMin(3), fVMax(3), fNpart(0), fParentWeight(0), fChildWeight(0), fAnalog(0), fVertexSmear(kNoSmear), fVertexSource(kExternal), fCutVertexZ(0), fTrackIt(0), fOrigin(3), fOsigma(3), fVertex(3), fEventVertex(0), fStack(0) { // // Copy constructor // gen.Copy(*this); } //_______________________________________________________________________ AliGenerator & AliGenerator::operator=(const AliGenerator &gen) { // // Assignment operator // gen.Copy(*this); return (*this); } //_______________________________________________________________________ void AliGenerator::Copy(AliGenerator &/* gen */) const { // // Copy *this onto gen // Fatal("Copy","Not implemented!\n"); } //_______________________________________________________________________ AliGenerator::~AliGenerator() { // // Destructor // fOrigin.Set(0); fOsigma.Set(0); if (fgMCEvGen) { delete fgMCEvGen; fgMCEvGen=0; } } //_______________________________________________________________________ void AliGenerator::Init() { // // Dummy initialisation // } //_______________________________________________________________________ void AliGenerator::SetOrigin(Float_t ox, Float_t oy, Float_t oz) { // // Set the vertex for the generated tracks // fOrigin[0]=ox; fOrigin[1]=oy; fOrigin[2]=oz; } //_______________________________________________________________________ void AliGenerator::SetOrigin(const TLorentzVector &o) { // // Set the vertex for the generated tracks // fOrigin[0]=o[0]; fOrigin[1]=o[1]; fOrigin[2]=o[2]; } //_______________________________________________________________________ void AliGenerator::SetSigma(Float_t sx, Float_t sy, Float_t sz) { // // Set the spread of the vertex // fOsigma[0]=sx; fOsigma[1]=sy; fOsigma[2]=sz; } //_______________________________________________________________________ void AliGenerator::SetMomentumRange(Float_t pmin, Float_t pmax) { // // Set the momentum range for the generated particles // fPMin = pmin; fPMax = pmax; SetBit(kMomentumRange); } //_______________________________________________________________________ void AliGenerator::SetPtRange(Float_t ptmin, Float_t ptmax) { // // Set the Pt range for the generated particles // fPtMin = ptmin; fPtMax = ptmax; SetBit(kPtRange); } //_______________________________________________________________________ void AliGenerator::SetPhiRange(Float_t phimin, Float_t phimax) { // // Set the Phi range for the generated particles // fPhiMin = TMath::Pi()*phimin/180; fPhiMax = TMath::Pi()*phimax/180; SetBit(kPhiRange); } //_______________________________________________________________________ void AliGenerator::SetYRange(Float_t ymin, Float_t ymax) { // // Set the Rapidity range for the generated particles // fYMin=ymin; fYMax=ymax; SetBit(kYRange); } //_______________________________________________________________________ void AliGenerator::SetVRange(Float_t vxmin, Float_t vxmax, Float_t vymin, Float_t vymax, Float_t vzmin, Float_t vzmax) { // // Set the vertex range for the generated particles // fVMin[0]=vxmin; fVMin[1]=vymin; fVMin[2]=vzmin; fVMax[0]=vxmax; fVMax[1]=vymax; fVMax[2]=vzmax; SetBit(kVertexRange); } //_______________________________________________________________________ void AliGenerator::SetThetaRange(Float_t thetamin, Float_t thetamax) { // // Set the theta range for the generated particles // fThetaMin = TMath::Pi()*thetamin/180; fThetaMax = TMath::Pi()*thetamax/180; SetBit(kThetaRange); } void AliGenerator::Vertex() { // // Obtain vertex for current event from external source or calculated (internal) // if (fVertexSource == kInternal) { VertexInternal(); } else { VertexExternal(); } } //_______________________________________________________________________ void AliGenerator::VertexExternal() { // Dummy !!!!!! // Obtain vertex from external source // // Should be something like fVertex = gAlice->GetVertex() fVertex[0]=fVertex[1]=fVertex[2]=0; } //_______________________________________________________________________ void AliGenerator::VertexInternal() { // // Obtain calculated vertex // Default is gaussian smearing Float_t random[6]; Rndm(random,6); for (Int_t j = 0; j<3 ; j++) { fVertex[j]= fOrigin[j]+fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())* TMath::Sqrt(-2*TMath::Log(random[2*j+1])); } } //_______________________________________________________________________ void AliGenerator::SetTrack(Int_t done, Int_t parent, Int_t pdg, Float_t *pmom, Float_t *vpos, Float_t *polar, Float_t tof, TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is) { if (fStack) fStack->SetTrack(done, parent, pdg, pmom, vpos, polar, tof, mech, ntr, weight, is); else gAlice->SetTrack(done, parent, pdg, pmom, vpos, polar, tof, mech, ntr, weight, is); } //_______________________________________________________________________ void AliGenerator::SetTrack(Int_t done, Int_t parent, Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vx, Double_t vy, Double_t vz, Double_t tof, Double_t polx, Double_t poly, Double_t polz, TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is) { if (fStack) fStack->SetTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof, polx, poly, polz, mech, ntr, weight, is); else gAlice->SetTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof, polx, poly, polz, mech, ntr, weight, is); } //_______________________________________________________________________ void AliGenerator:: KeepTrack(Int_t itrack) { if (fStack) fStack->KeepTrack(itrack); else gAlice->KeepTrack(itrack); } //_______________________________________________________________________ void AliGenerator:: SetHighWaterMark(Int_t nt) { if (fStack) fStack->SetHighWaterMark(nt); else gAlice->SetHighWaterMark(nt); }