/**************************************************************************
* 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. *
**************************************************************************/
/* $Id$ */
//-----------------------------------------------------------------
// Generate the final state of the interaction as the input
// to the MonteCarlo
// Author: A.Morsch
//-----------------------------------------------------------------
//Begin_Html
/*
The responsible person for this module is
Andreas Morsch.
*/ //End_Html // // /////////////////////////////////////////////////////////////////// #include#include "AliCollisionGeometry.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), fCollisionGeometry(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), fCollisionGeometry(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) { // // Loads one track on the stack // 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) { // // Loads one track on the stack // 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) { // // Declare a track permanent on the stack // if (fStack) fStack->KeepTrack(itrack); else gAlice->KeepTrack(itrack); } //_______________________________________________________________________ void AliGenerator:: SetHighWaterMark(Int_t nt) { // // Internal function to set the maximum index used in the stack // if (fStack) fStack->SetHighWaterMark(nt); else gAlice->SetHighWaterMark(nt); }