X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliGenerator.cxx;h=e94345f16bf6e1b23611cfdc953c3734c69edbad;hb=e0919e877c6f962639a2af56bc822a7d58a7f33b;hp=13201a61c81cb68d6c343e9a80357520c64b8ba7;hpb=1439f98e64c177f2d0122bfcc85e4162b29f1f2f;p=u%2Fmrichter%2FAliRoot.git
diff --git a/STEER/AliGenerator.cxx b/STEER/AliGenerator.cxx
index 13201a61c81..e94345f16bf 100644
--- a/STEER/AliGenerator.cxx
+++ b/STEER/AliGenerator.cxx
@@ -1,8 +1,25 @@
-///////////////////////////////////////////////////////////////////
-// //
-// Generate the final state of the interaction as the input //
-// to the MonteCarlo //
-//
+/**************************************************************************
+ * 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
/*
@@ -17,75 +34,398 @@
//End_Html
// //
///////////////////////////////////////////////////////////////////
+#include
+#include
+#include "AliLog.h"
+#include "AliCollisionGeometry.h"
+#include "AliConfig.h"
#include "AliGenerator.h"
#include "AliRun.h"
+#include "AliStack.h"
+#include "AliMC.h"
+#include "AliVertexGenerator.h"
ClassImp(AliGenerator)
-TGenerator* AliGenerator::fgMCEvGen=0;
-//____________________________________________________________
-AliGenerator::AliGenerator()
+//_______________________________________________________________________
+AliGenerator::AliGenerator():
+ TNamed(),
+ AliRndm(),
+ fMCEvGen(0),
+ 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(kInternal),
+ fCutVertexZ(0),
+ fPileUpTimeWindow(-1.),
+ fTrackIt(0),
+ fVertexGenerator(0),
+ fOrigin(3),
+ fOsigma(3),
+ fVertex(3),
+ fStack(0),
+ fContainer(0),
+ fCollisionGeometry(0),
+ fEnergyCMS(0),
+ fAProjectile(0),
+ fZProjectile(0),
+ fATarget(0),
+ fZTarget(0),
+ fProjectile(""),
+ fTarget("")
{
- printf("\n Initialising AliGenerator\n\n");
-
- gAlice->SetGenerator(this);
- SetThetaRange();
- SetPhiRange();
- SetMomentumRange();
- SetPtRange();
- SetYRange();
- SetNumberParticles();
- SetTrackingFlag();
//
- // fName="Default";
- // fTitle="Base Generator Class";
+ // Default constructor
//
- fOrigin.Set(3);
- fOsigma.Set(3);
+ if (gAlice) {
+ AliDebug(1, "AliGenerator Default Constructor");
+ AliMC * mc = gAlice->GetMCApp();
+ if (mc) mc->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)
- : TNamed(" "," ")
+//_______________________________________________________________________
+AliGenerator::AliGenerator(Int_t npart):
+ TNamed(),
+ AliRndm(),
+ fMCEvGen(0),
+ 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(kInternal),
+ fCutVertexZ(0),
+ fPileUpTimeWindow(-1.),
+ fTrackIt(0),
+ fVertexGenerator(0),
+ fOrigin(3),
+ fOsigma(3),
+ fVertex(3),
+ fStack(0),
+ fContainer(0),
+ fCollisionGeometry(0),
+ fEnergyCMS(0),
+ fAProjectile(0),
+ fZProjectile(0),
+ fATarget(0),
+ fZTarget(0),
+ fProjectile(""),
+ fTarget("")
{
- printf("\n Initialising AliGenerator\n\n");
- gAlice->SetGenerator(this);
- SetThetaRange();
- SetPhiRange();
- SetMomentumRange();
- SetPtRange();
- SetYRange();
- SetNumberParticles(npart);
- SetTrackingFlag();
//
- // fName="Default";
- // fTitle="Base Generator Class";
+ // Standard constructor
//
- fVertexSmear=none;
- fOrigin.Set(3);
- fOsigma.Set(3);
+ if (gAlice) {
+ AliDebug(1, "AliGenerator Constructor initializing number of particles");
+ AliMC * mc = gAlice->GetMCApp();
+ if (mc) mc->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;
- fVMin.Set(3);
+ fVertex[0]=fVertex[1]=fVertex[2]=0;
+
fVMin[0]=fVMin[1]=fVMin[2]=0;
- fVMax.Set(3);
fVMax[0]=fVMax[1]=fVMax[2]=10000;
+
+ SetNumberParticles(npart);
+
+ AliConfig::Instance()->Add(this);
}
-//____________________________________________________________
+//_______________________________________________________________________
AliGenerator::~AliGenerator()
{
+ //
+ // Destructor
+ //
fOrigin.Set(0);
fOsigma.Set(0);
- delete fgMCEvGen;
+ if (fMCEvGen) {
+ delete fMCEvGen;
+ fMCEvGen=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 if (fVertexSource == kContainer) {
+ ;
+ } else if (fVertexSource == kExternal) {
+ VertexExternal();
+ }
+}
+
+//_______________________________________________________________________
+void AliGenerator::VertexExternal()
+{
+ //
+ // Obtain vertex from external source (vertex generator)
+ //
+ TVector3 vertex = fVertexGenerator->GetVertex();
+ fVertex[0] = vertex.X();
+ fVertex[1] = vertex.Y();
+ fVertex[2] = vertex.Z();
+}
+//_______________________________________________________________________
+void AliGenerator::VertexInternal()
+{
+ //
+ // Obtain calculated vertex
+ // Default is gaussian smearing
+ Float_t random[6];
+ Float_t dv[3];
+ Int_t j;
+ dv[2] = 1.e10;
+ if (!TestBit(kVertexRange)) {
+ while(TMath::Abs(dv[2]) > fCutVertexZ*fOsigma[2]) {
+ Rndm(random,6);
+ for (j=0; j < 3; j++) {
+ dv[j] = fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
+ TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
+ }
+ }
+ for (j=0; j < 3; j++) fVertex[j] = fOrigin[j] + dv[j];
+ } else {
+ Rndm(random,3);
+ for (j=0; j < 3; j++)
+ fVertex[j] = fVMin[j] + random[j] * (fVMax[j] - fVMin[j]);
+ }
+}
+
+//_______________________________________________________________________
+void AliGenerator::PushTrack(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->PushTrack(done, parent, pdg, pmom, vpos, polar, tof,
+ mech, ntr, weight, is);
+ else
+ gAlice->GetMCApp()->PushTrack(done, parent, pdg, pmom, vpos, polar, tof,
+ mech, ntr, weight, is);
+}
+
+//_______________________________________________________________________
+void AliGenerator::PushTrack(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->PushTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof,
+ polx, poly, polz, mech, ntr, weight, is);
+ else
+ gAlice->GetMCApp()->PushTrack(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->GetMCApp()->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->GetMCApp()->SetHighWaterMark(nt);
+
+}
+void AliGenerator::FinishRun()
+{
+ ;
+}