* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-Revision 1.16 2002/11/28 11:46:15 morsch
-Don't track pi0 if already decayed.
-
-Revision 1.15 2002/11/28 11:38:53 morsch
-Typo corrected.
-
-Revision 1.14 2002/11/26 17:12:36 morsch
-Decay pi0 if requested.
-
-Revision 1.13 2002/10/14 14:55:35 hristov
-Merging the VirtualMC branch to the main development branch (HEAD)
-
-Revision 1.11.4.1 2002/07/24 08:56:28 alibrary
-Updating EVGEN on TVirtulaMC
-
-Revision 1.12 2002/06/19 06:56:34 hristov
-Memory leak corrected
-
-Revision 1.11 2002/03/20 10:21:13 hristov
-Set fPtMax to 15 GeV in order to avoid some numerical problems
-
-Revision 1.10 2001/10/15 16:44:46 morsch
-- Possibility for vertex distribution truncation.
-- Write mc header with vertex position.
-
-Revision 1.9 2001/07/27 17:09:36 morsch
-Use local SetTrack, KeepTrack and SetHighWaterMark methods
-to delegate either to local stack or to stack owned by AliRun.
-(Piotr Skowronski, A.M.)
-
-Revision 1.8 2001/07/20 11:03:58 morsch
-Issue warning message if used outside allowed eta range (-8 to 8).
-
-Revision 1.7 2001/07/17 12:41:01 morsch
-- Calculation of fraction of event corresponding to selected pt-range corrected
-(R. Turrisi)
-- Parent weight corrected.
-
-Revision 1.6 2001/05/16 14:57:10 alibrary
-New files for folders and Stack
-
-Revision 1.5 2000/12/21 16:24:06 morsch
-Coding convention clean-up
-
-Revision 1.4 2000/11/30 07:12:50 alibrary
-Introducing new Rndm and QA classes
-
-Revision 1.3 2000/10/02 21:28:06 fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.2 2000/07/11 18:24:55 fca
-Coding convention corrections + few minor bug fixes
-
-Revision 1.1 2000/06/09 20:20:30 morsch
-Same class as previously in AliSimpleGen.cxx
-All coding rule violations except RS3 corrected (AM)
-
-*/
+/* $Id$ */
// Parameterisation of pi and K, eta and pt distributions
// used for the ALICE TDRs.
// //
///////////////////////////////////////////////////////////////////
-#include "AliGenHIJINGpara.h"
-#include "AliGenEventHeader.h"
-#include "AliRun.h"
-#include "AliConst.h"
-#include "AliDecayer.h"
-#include "AliDecayerPythia.h"
-#include "AliPDG.h"
-
-#include <TF1.h>
#include <TArrayF.h>
-#include <TDatabasePDG.h>
+#include <TCanvas.h>
#include <TClonesArray.h>
+#include <TDatabasePDG.h>
+#include <TF1.h>
+#include <TH1.h>
+#include <TPDGCode.h>
#include <TParticle.h>
+#include <TROOT.h>
+#include <TVirtualMC.h>
+
+#include "AliConst.h"
+#include "AliDecayer.h"
+#include "AliGenEventHeader.h"
+#include "AliGenHIJINGpara.h"
+#include "AliLog.h"
+#include "AliRun.h"
ClassImp(AliGenHIJINGpara)
-AliGenHIJINGpara::AliGenHIJINGpara(const AliGenHIJINGpara & para)
-{
-// copy constructor
-}
+
//_____________________________________________________________________________
static Double_t ptpi(Double_t *px, Double_t *)
// POWER LAW FOR PT > 500 MEV
// MT SCALING BELOW (T=160 MEV)
//
- const Double_t kp0 = 1.3;
- const Double_t kxn = 8.28;
- const Double_t kxlim=0.5;
- const Double_t kt=0.160;
- const Double_t kxmpi=0.139;
- const Double_t kb=1.;
+ const Double_t kp0 = 1.3;
+ const Double_t kxn = 8.28;
+ const Double_t kxlim = 0.5;
+ const Double_t kt = 0.160;
+ const Double_t kxmpi = 0.139;
+ const Double_t kb = 1.;
Double_t y, y1, xmpi2, ynorm, a;
- Double_t x=*px;
+ Double_t x = *px;
//
- y1=TMath::Power(kp0/(kp0+kxlim),kxn);
- xmpi2=kxmpi*kxmpi;
- ynorm=kb*(TMath::Exp(-sqrt(kxlim*kxlim+xmpi2)/kt));
- a=ynorm/y1;
+ y1 = TMath::Power(kp0 / (kp0 + kxlim), kxn);
+ xmpi2 = kxmpi * kxmpi;
+ ynorm = kb * (TMath::Exp(-sqrt(kxlim * kxlim + xmpi2) / kt ));
+ a = ynorm / y1;
if (x > kxlim)
- y=a*TMath::Power(kp0/(kp0+x),kxn);
+ y = a * TMath::Power(kp0 / (kp0 + x), kxn);
else
- y=kb*TMath::Exp(-sqrt(x*x+xmpi2)/kt);
+ y = kb* TMath::Exp(-sqrt(x * x + xmpi2) / kt);
+
return y*x;
}
//_____________________________________________________________________________
AliGenHIJINGpara::AliGenHIJINGpara()
- :AliGenerator()
+ :AliGenerator(),
+ fNt(-1),
+ fNpartProd(0),
+ fPi0Decays(kFALSE),
+ fPtWgtPi(0.),
+ fPtWgtKa(0.),
+ fPtpi(0),
+ fPtka(0),
+ fETApic(0),
+ fETAkac(0),
+ fDecayer(0)
{
//
// Default constructor
//
- fPtpi = 0;
- fPtka = 0;
- fETApic = 0;
- fETAkac = 0;
- fDecayer = 0;
- fNt = -1;
SetCutVertexZ();
SetPtRange();
- SetPi0Decays();
}
//_____________________________________________________________________________
AliGenHIJINGpara::AliGenHIJINGpara(Int_t npart)
- :AliGenerator(npart)
+ :AliGenerator(npart),
+ fNt(-1),
+ fNpartProd(npart),
+ fPi0Decays(kFALSE),
+ fPtWgtPi(0.),
+ fPtWgtKa(0.),
+ fPtpi(0),
+ fPtka(0),
+ fETApic(0),
+ fETAkac(0),
+ fDecayer(0)
{
//
// Standard constructor
//
- fName="HIGINGpara";
+ fName="HIJINGpara";
fTitle="HIJING Parametrisation Particle Generator";
- fPtpi = 0;
- fPtka = 0;
- fETApic = 0;
- fETAkac = 0;
- fDecayer = 0;
- fNt = -1;
SetCutVertexZ();
SetPtRange();
- SetPi0Decays();
}
//_____________________________________________________________________________
Float_t etaMax = -TMath::Log(TMath::Tan(
TMath::Max((Double_t)fThetaMin/2,1.e-10)));
fPtpi = new TF1("ptpi",&ptpi,0,20,0);
+ gROOT->GetListOfFunctions()->Remove(fPtpi);
fPtka = new TF1("ptka",&ptka,0,20,0);
+ gROOT->GetListOfFunctions()->Remove(fPtka);
+ fPtpi->SetNpx(1000);
+ fPtka->SetNpx(1000);
fETApic = new TF1("etapic",&etapic,etaMin,etaMax,0);
+ gROOT->GetListOfFunctions()->Remove(fETApic);
fETAkac = new TF1("etakac",&etakac,etaMin,etaMax,0);
+ gROOT->GetListOfFunctions()->Remove(fETAkac);
- TF1 etaPic0("etapic",&etapic,-7,7,0);
- TF1 etaKac0("etakac",&etakac,-7,7,0);
+ TF1 etaPic0("etaPic0",&etapic,-7,7,0);
+ TF1 etaKac0("etaKac0",&etakac,-7,7,0);
- TF1 ptPic0("ptpi",&ptpi,0.,15.,0);
- TF1 ptKac0("ptka",&ptka,0.,15.,0);
+ TF1 ptPic0("ptPic0",&ptpi,0.,15.,0);
+ TF1 ptKac0("ptKac0",&ptka,0.,15.,0);
Float_t intETApi = etaPic0.Integral(-0.5, 0.5);
Float_t intETAka = etaKac0.Integral(-0.5, 0.5);
// Fraction of events corresponding to the selected phi-range
Float_t phiFrac = (fPhiMax-fPhiMin)/2/TMath::Pi();
- fParentWeight = Float_t(fNpart)/(intETASel*ptFrac*phiFrac);
- printf("%s: The number of particles in the selected kinematic region corresponds to %f percent of a full event\n ",
- ClassName(),100.*fParentWeight);
+ fParentWeight = (intETASel*ptFrac*phiFrac) / Float_t(fNpart);
+
+ if (fAnalog != 0) {
+ fPtWgtPi = (fPtMax - fPtMin) / fPtpi->Integral(0., 20.);
+ fPtWgtKa = (fPtMax - fPtMin) / fPtka->Integral(0., 20.);
+ fParentWeight = (intETASel*phiFrac) / Float_t(fNpart);
+ }
+
+
+ AliInfo(Form("The number of particles in the selected kinematic region corresponds to %f percent of a full event",
+ 100./ fParentWeight));
// Issue warning message if etaMin or etaMax are outside the alowed range
// of the parametrization
if (etaMin < -8.001 || etaMax > 8.001) {
- printf("\n \n WARNING FROM AliGenHIJINGPara !");
- printf("\n YOU ARE USING THE PARAMETERISATION OUTSIDE ");
- printf("\n THE ALLOWED PSEUDORAPIDITY RANGE (-8. - 8.)");
- printf("\n YOUR LIMITS: %f %f \n \n ", etaMin, etaMax);
+ AliWarning("\nYOU ARE USING THE PARAMETERISATION OUTSIDE ");
+ AliWarning("THE ALLOWED PSEUDORAPIDITY RANGE (-8. - 8.)");
+ AliWarning(Form("YOUR LIMITS: %f %f \n ", etaMin, etaMax));
}
//
//
- if (fPi0Decays) fDecayer = new AliDecayerPythia();
+ if (fPi0Decays && gMC)
+ fDecayer = gMC->GetDecayer();
+
+ if (fPi0Decays)
+ {
+ fDecayer->SetForceDecay(kNeutralPion);
+ fDecayer->Init();
+ }
+
}
const Int_t kKaons[4] = {kK0Long, kK0Short, kKPlus, kKMinus};
//
Float_t origin[3];
- Float_t pt, pl, ptot;
+ Float_t pt, pl, ptot, wgt;
Float_t phi, theta;
Float_t p[3];
Int_t i, part, j;
for (j=0;j<3;j++) origin[j]=fOrigin[j];
if(fVertexSmear == kPerEvent) {
- Float_t dv[3];
- dv[2] = 1.e10;
- 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++) origin[j] += dv[j];
+ Vertex();
+ for (j=0; j < 3; j++) origin[j] = fVertex[j];
} // if kPerEvent
TArrayF eventVertex;
eventVertex.Set(3);
eventVertex[0] = origin[0];
eventVertex[1] = origin[1];
eventVertex[2] = origin[2];
-
+
for(i=0;i<fNpart;i++) {
while(1) {
- Rndm(random,3);
+ Rndm(random,4);
if(random[0]<kBorne) {
part=kPions[Int_t (random[1]*3)];
ptf=fPtpi;
etaf=fETApic;
+ wgt = fPtWgtPi;
} else {
part=kKaons[Int_t (random[1]*4)];
ptf=fPtka;
etaf=fETAkac;
+ wgt = fPtWgtKa;
}
phi=fPhiMin+random[2]*(fPhiMax-fPhiMin);
theta=2*TMath::ATan(TMath::Exp(-etaf->GetRandom()));
if(theta<fThetaMin || theta>fThetaMax) continue;
- pt=ptf->GetRandom();
+
+ if (fAnalog == 0) {
+ pt = ptf->GetRandom();
+ } else {
+ pt = fPtMin + random[3] * (fPtMax - fPtMin);
+ }
+
+
pl=pt/TMath::Tan(theta);
ptot=TMath::Sqrt(pt*pt+pl*pl);
if(ptot<fPMin || ptot>fPMax) continue;
TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
}
}
+
+ if (fAnalog == 0) {
+ wgt = fParentWeight;
+ } else {
+ wgt *= (fParentWeight * ptf->Eval(pt));
+ }
+
+
if (part == kPi0 && fPi0Decays){
//
// Decay pi0 if requested
- SetTrack(0,-1,part,p,origin,polar,0,kPPrimary,fNt,fParentWeight);
+ PushTrack(0,-1,part,p,origin,polar,0,kPPrimary,fNt,wgt);
KeepTrack(fNt);
DecayPi0(origin, p);
} else {
- SetTrack(fTrackIt,-1,part,p,origin,polar,0,kPPrimary,fNt,fParentWeight);
+ // printf("fNt %d", fNt);
+ PushTrack(fTrackIt,-1,part,p,origin,polar,0,kPPrimary,fNt,wgt);
+
KeepTrack(fNt);
}
AliGenEventHeader* header = new AliGenEventHeader("HIJINGparam");
// Event Vertex
header->SetPrimaryVertex(eventVertex);
+ header->SetNProduced(fNpartProd);
gAlice->SetGenEventHeader(header);
}
-AliGenHIJINGpara& AliGenHIJINGpara::operator=(const AliGenHIJINGpara& rhs)
-{
-// Assignment operator
- return *this;
-}
-
void AliGenHIJINGpara::SetPtRange(Float_t ptmin, Float_t ptmax) {
AliGenerator::SetPtRange(ptmin, ptmax);
}
if(!particles) particles = new TClonesArray("TParticle",1000);
//
const Float_t kMass = TDatabasePDG::Instance()->GetParticle(kPi0)->Mass();
- Float_t e = TMath::Sqrt(p[0] * p[0] + p[1] * p[1] +p[2] * p[2] * kMass * kMass);
+ Float_t e = TMath::Sqrt(p[0] * p[0] + p[1] * p[1] + p[2] * p[2]+ kMass * kMass);
//
// Decay the pi0
TLorentzVector pmom(p[0], p[1], p[2], e);
fDecayer->Decay(kPi0, &pmom);
+
//
// Put decay particles on the stack
//
Float_t polar[3] = {0., 0., 0.};
Int_t np = fDecayer->ImportParticles(particles);
+ fNpartProd += (np-1);
Int_t nt;
for (Int_t i = 1; i < np; i++)
{
p[2] = iParticle->Pz();
Int_t part = iParticle->GetPdgCode();
- SetTrack(fTrackIt, fNt, part, p, orig, polar, 0, kPDecay, nt, fParentWeight);
+ PushTrack(fTrackIt, fNt, part, p, orig, polar, 0, kPDecay, nt, fParentWeight);
KeepTrack(nt);
}
fNt = nt;
}
+
+void AliGenHIJINGpara::Draw( const char * /*opt*/)
+{
+ //
+ // Draw the pT and y Distributions
+ //
+ TCanvas *c0 = new TCanvas("c0","Canvas 0",400,10,600,700);
+ c0->Divide(2,1);
+ c0->cd(1);
+ fPtpi->Draw();
+ fPtpi->GetHistogram()->SetXTitle("p_{T} (GeV)");
+ c0->cd(2);
+ fPtka->Draw();
+ fPtka->GetHistogram()->SetXTitle("p_{T} (GeV)");
+
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff